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

@@ -6,31 +6,48 @@ import (
"net/http"
"os"
"path/filepath"
"strings"
"git.max-richter.dev/max/marka/server-new/internal/adapters"
"git.max-richter.dev/max/marka/server-new/internal/handler"
)
type multi []string
func (m *multi) String() string { return strings.Join(*m, ",") }
func (m *multi) Set(v string) error {
*m = append(*m, v)
return nil
}
func main() {
root := flag.String("root", ".", "filesystem root to serve")
var roots multi
flag.Var(&roots, "root", "repeatable; specify multiple -root flags")
addr := flag.String("addr", ":8080", "listen address")
flag.Parse()
absRoot, err := filepath.Abs(*root)
must(err)
info, err := os.Stat(absRoot)
must(err)
if !info.IsDir() {
log.Fatal("root is not a directory")
if len(roots) == 0 {
log.Fatal("at least one -root flag must be specified")
}
fsAdapter, err := adapters.NewLocalFsAdapter(absRoot)
absRoots := make([]string, len(roots))
for i, r := range roots {
abs, err := filepath.Abs(r)
must(err)
info, err := os.Stat(abs)
must(err)
if !info.IsDir() {
log.Fatalf("root %s is not a directory", r)
}
absRoots[i] = abs
}
fsAdapter, err := adapters.NewLocalFsAdapter(absRoots)
must(err)
http.Handle("/", handler.NewHandler(fsAdapter))
log.Printf("listening on %s, root=%s", *addr, absRoot)
log.Printf("listening on %s, roots=%s", *addr, strings.Join(absRoots, ", "))
log.Fatal(http.ListenAndServe(*addr, nil))
}