mirror of https://github.com/dirtbags/moth.git
48 lines
744 B
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)
|
|
}
|