feat: allow multiple roots

This commit is contained in:
Max Richter
2025-10-03 21:41:39 +02:00
parent 3986e22dd3
commit ab81c980b5
3 changed files with 120 additions and 22 deletions

View File

@@ -1,16 +1,22 @@
package adapters
import (
"errors"
"os"
"path/filepath"
"strings"
"time"
)
type LocalFsAdapter struct {
root string
roots []string
}
func (l LocalFsAdapter) readDir(path string) (FsResponse, error) {
dirInfo, _ := os.Stat(path)
func (l LocalFsAdapter) readDir(path string, root string) (FsResponse, error) {
dirInfo, err := os.Stat(path)
if err != nil {
return FsResponse{}, err
}
entries, err := os.ReadDir(path)
if err != nil {
@@ -29,20 +35,31 @@ func (l LocalFsAdapter) readDir(path string) (FsResponse, error) {
out = append(out, FsDirEntry{
Name: e.Name(),
Type: entryType,
IsDir: e.IsDir(),
ModTime: info.ModTime(),
})
}
relPath, err := filepath.Rel(root, path)
if err != nil {
return FsResponse{}, err
}
if relPath == "." {
relPath = ""
}
responsePath := "/" + filepath.ToSlash(filepath.Join(filepath.Base(root), relPath))
return FsResponse{
Dir: &FsDir{
Files: out,
Name: ResponsePath(l.root, path),
Name: responsePath,
ModTime: dirInfo.ModTime(),
},
}, nil
}
func (l LocalFsAdapter) readFile(path string) (FsResponse, error) {
func (l LocalFsAdapter) readFile(path string, root string) (FsResponse, error) {
fi, err := os.Stat(path)
if err != nil {
return FsResponse{}, err
@@ -53,9 +70,16 @@ func (l LocalFsAdapter) readFile(path string) (FsResponse, error) {
return FsResponse{}, err
}
relPath, err := filepath.Rel(root, path)
if err != nil {
return FsResponse{}, err
}
responsePath := "/" + filepath.ToSlash(filepath.Join(filepath.Base(root), relPath))
return FsResponse{
File: &FsFile{
Name: ResponsePath(l.root, path),
Name: responsePath,
Type: contentTypeFor(path),
ModTime: fi.ModTime(),
Content: data,
@@ -64,11 +88,66 @@ func (l LocalFsAdapter) readFile(path string) (FsResponse, error) {
}
func (l LocalFsAdapter) Read(path string) (FsResponse, error) {
cleanRel, err := SafeRel(l.root, path)
if path == "/" {
entries := make([]FsDirEntry, 0, len(l.roots))
var latestModTime time.Time
for _, r := range l.roots {
info, err := os.Stat(r)
if err != nil {
continue
}
entries = append(entries, FsDirEntry{
Name: filepath.Base(r),
Type: "dir",
IsDir: true,
ModTime: info.ModTime(),
})
if info.ModTime().After(latestModTime) {
latestModTime = info.ModTime()
}
}
return FsResponse{
Dir: &FsDir{
Files: entries,
Name: "/",
ModTime: latestModTime,
},
}, nil
}
pathParts := strings.Split(strings.Trim(path, "/"), "/")
if len(pathParts) == 0 || pathParts[0] == "" {
return FsResponse{}, ErrNotFound
}
rootIdentifier := pathParts[0]
var targetRoot string
for _, r := range l.roots {
if filepath.Base(r) == rootIdentifier {
targetRoot = r
break
}
}
if targetRoot == "" {
return FsResponse{}, ErrNotFound
}
subPath := filepath.Join(pathParts[1:]...)
target := filepath.Join(targetRoot, subPath)
absTarget, err := filepath.Abs(target)
if err != nil {
return FsResponse{}, err
}
target := filepath.Join(l.root, filepath.FromSlash(cleanRel))
absRoot, err := filepath.Abs(targetRoot)
if err != nil {
return FsResponse{}, err
}
if !strings.HasPrefix(absTarget, absRoot) {
return FsResponse{}, errors.New("path escapes root")
}
fi, err := os.Stat(target)
if err != nil {
@@ -80,18 +159,19 @@ func (l LocalFsAdapter) Read(path string) (FsResponse, error) {
}
if fi.IsDir() {
return l.readDir(target)
return l.readDir(target, targetRoot)
}
return l.readFile(target)
return l.readFile(target, targetRoot)
}
func (LocalFsAdapter) Write(path string, content []byte) error {
return nil
}
func NewLocalFsAdapter(root string) (FileAdapter, error) {
func NewLocalFsAdapter(roots []string) (FileAdapter, error) {
return LocalFsAdapter{
root: root,
roots: roots,
}, nil
}