refactor webdav subdirectory, fixes #4967 (#4969)

* refactor webdav subdirectory, fixes #4967

* fix bug where Name() was not called on delegate wrappedFileInfo
This commit is contained in:
Nico D'Cotta 2023-11-01 19:12:49 -01:00 committed by GitHub
parent 4ed06e9ba5
commit 3834ad5c48
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 101 additions and 4 deletions

View file

@ -63,6 +63,10 @@ func NewWebDavServer(option *WebDavOption) (ws *WebDavServer, err error) {
if option.FilerRootPath == "/" { if option.FilerRootPath == "/" {
option.FilerRootPath = "" option.FilerRootPath = ""
} }
// filer.path non "/" option means we are accessing filer's sub-folders
if option.FilerRootPath != "" {
fs = NewWrappedFs(fs, path.Clean(option.FilerRootPath))
}
ws = &WebDavServer{ ws = &WebDavServer{
option: option, option: option,
@ -204,8 +208,6 @@ func (fs *WebDavFileSystem) Mkdir(ctx context.Context, fullDirPath string, perm
} }
func (fs *WebDavFileSystem) OpenFile(ctx context.Context, fullFilePath string, flag int, perm os.FileMode) (webdav.File, error) { func (fs *WebDavFileSystem) OpenFile(ctx context.Context, fullFilePath string, flag int, perm os.FileMode) (webdav.File, error) {
// Add filer.path
fullFilePath = fs.option.FilerRootPath + fullFilePath
glog.V(2).Infof("WebDavFileSystem.OpenFile %v %x", fullFilePath, flag) glog.V(2).Infof("WebDavFileSystem.OpenFile %v %x", fullFilePath, flag)
var err error var err error
@ -377,8 +379,6 @@ func (fs *WebDavFileSystem) stat(ctx context.Context, fullFilePath string) (os.F
} }
func (fs *WebDavFileSystem) Stat(ctx context.Context, name string) (os.FileInfo, error) { func (fs *WebDavFileSystem) Stat(ctx context.Context, name string) (os.FileInfo, error) {
// Add filer.path
name = fs.option.FilerRootPath + name
glog.V(2).Infof("WebDavFileSystem.Stat %v", name) glog.V(2).Infof("WebDavFileSystem.Stat %v", name)
return fs.stat(ctx, name) return fs.stat(ctx, name)

View file

@ -0,0 +1,97 @@
package weed_server
import (
"context"
"golang.org/x/net/webdav"
"io/fs"
"os"
"strings"
)
type wrappedFs struct {
subFolder string
webdav.FileSystem
}
// NewWrappedFs returns a webdav.FileSystem identical to fs, except it
// provides access to a sub-folder of fs that is denominated by subFolder.
// It transparently handles renaming paths and filenames so that the outer part of the wrapped filesystem
// does not leak out.
func NewWrappedFs(fs webdav.FileSystem, subFolder string) webdav.FileSystem {
return wrappedFs{
subFolder: subFolder,
FileSystem: fs,
}
}
func (w wrappedFs) Mkdir(ctx context.Context, name string, perm os.FileMode) error {
name = w.subFolder + name
return w.FileSystem.Mkdir(ctx, name, perm)
}
func (w wrappedFs) OpenFile(ctx context.Context, name string, flag int, perm os.FileMode) (webdav.File, error) {
name = w.subFolder + name
file, err := w.FileSystem.OpenFile(ctx, name, flag, perm)
file = wrappedFile{
File: file,
subFolder: &w.subFolder,
}
return file, err
}
func (w wrappedFs) RemoveAll(ctx context.Context, name string) error {
name = w.subFolder + name
return w.FileSystem.RemoveAll(ctx, name)
}
func (w wrappedFs) Rename(ctx context.Context, oldName, newName string) error {
oldName = w.subFolder + oldName
newName = w.subFolder + newName
return w.FileSystem.Rename(ctx, oldName, newName)
}
func (w wrappedFs) Stat(ctx context.Context, name string) (os.FileInfo, error) {
name = w.subFolder + name
info, err := w.FileSystem.Stat(ctx, name)
info = wrappedFileInfo{
subFolder: &w.subFolder,
FileInfo: info,
}
return info, err
}
type wrappedFile struct {
webdav.File
subFolder *string
}
func (w wrappedFile) Readdir(count int) ([]fs.FileInfo, error) {
infos, err := w.File.Readdir(count)
for i, info := range infos {
infos[i] = wrappedFileInfo{
subFolder: w.subFolder,
FileInfo: info,
}
}
return infos, err
}
func (w wrappedFile) Stat() (fs.FileInfo, error) {
info, err := w.File.Stat()
info = wrappedFileInfo{
subFolder: w.subFolder,
FileInfo: info,
}
return info, err
}
type wrappedFileInfo struct {
subFolder *string
fs.FileInfo
}
func (w wrappedFileInfo) Name() string {
name := w.FileInfo.Name()
return strings.TrimPrefix(name, *w.subFolder)
}