moth/cmd/mothd/theme.go

48 lines
744 B
Go

package main
import (
"net/http"
"os"
"strings"
)
type Theme struct {
ThemeDir string
}
func NewTheme(themeDir string) *Theme {
return &Theme{
ThemeDir: themeDir,
}
}
func (t *Theme) path(parts ...string) string {
return MothPath(t.ThemeDir, parts...)
}
func (t *Theme) staticHandler(w http.ResponseWriter, req *http.Request) {
path := req.URL.Path
if strings.Contains(path, "/.") {
http.Error(w, "Invalid URL path", http.StatusBadRequest)
return
}
if path == "/" {
path = "/index.html"
}
f, err := os.Open(t.path(path))
if err != nil {
http.NotFound(w, req)
return
}
defer f.Close()
d, err := f.Stat()
if err != nil {
http.NotFound(w, req)
return
}
http.ServeContent(w, req, path, d.ModTime(), f)
}