feat: allow multiple roots
This commit is contained in:
@@ -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))
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user