portal/renterd/web.go

53 lines
976 B
Go
Raw Normal View History

2023-04-29 17:38:21 +00:00
package renterd
import (
"errors"
"io/fs"
"net/http"
_ "net/http/pprof"
"strings"
"go.sia.tech/web/renterd/ui"
)
type clientRouterFS struct {
fs fs.FS
}
func (cr *clientRouterFS) Open(name string) (fs.File, error) {
f, err := cr.fs.Open(name)
if errors.Is(err, fs.ErrNotExist) {
return cr.fs.Open("index.html")
}
return f, err
}
func createUIHandler() http.Handler {
assets, err := fs.Sub(ui.Assets, "assets")
if err != nil {
panic(err)
}
return http.FileServer(http.FS(&clientRouterFS{fs: assets}))
}
type treeMux struct {
h http.Handler
sub map[string]treeMux
}
func (t treeMux) ServeHTTP(w http.ResponseWriter, req *http.Request) {
if strings.HasPrefix(req.URL.Path, "/debug/pprof") {
http.DefaultServeMux.ServeHTTP(w, req)
return
}
for prefix, c := range t.sub {
if strings.HasPrefix(req.URL.Path, prefix) {
req.URL.Path = strings.TrimPrefix(req.URL.Path, prefix)
c.ServeHTTP(w, req)
return
}
}
t.h.ServeHTTP(w, req)
}