84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
//go:generate go-bindata -pkg assets -o assets/assets.go templates/ assets/css/ assets/js/ assets/fonts/
|
|
|
|
package hugo
|
|
|
|
import (
|
|
"mime"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/hacdias/caddy-hugo/assets"
|
|
"github.com/hacdias/caddy-hugo/browse"
|
|
"github.com/hacdias/caddy-hugo/edit"
|
|
"github.com/hacdias/caddy-hugo/settings"
|
|
"github.com/hacdias/caddy-hugo/utils"
|
|
"github.com/mholt/caddy/config/setup"
|
|
"github.com/mholt/caddy/middleware"
|
|
"github.com/spf13/hugo/commands"
|
|
)
|
|
|
|
// Setup function
|
|
func Setup(c *setup.Controller) (middleware.Middleware, error) {
|
|
commands.Execute()
|
|
|
|
return func(next middleware.Handler) middleware.Handler {
|
|
return &handler{Next: next}
|
|
}, nil
|
|
}
|
|
|
|
type handler struct{ Next middleware.Handler }
|
|
|
|
func (h handler) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
|
|
if middleware.Path(r.URL.Path).Matches("/admin") {
|
|
page := utils.ParseComponents(r)[1]
|
|
code := 404
|
|
var err error
|
|
|
|
if page != "assets" && page != "edit" {
|
|
if r.URL.Path[len(r.URL.Path)-1] != '/' {
|
|
http.Redirect(w, r, r.URL.Path+"/", http.StatusTemporaryRedirect)
|
|
return 0, nil
|
|
}
|
|
}
|
|
|
|
if page == "assets" {
|
|
filename := strings.Replace(r.URL.Path, "/admin/", "", 1)
|
|
file, err := assets.Asset(filename)
|
|
|
|
if err != nil {
|
|
return 404, nil
|
|
}
|
|
|
|
extension := filepath.Ext(filename)
|
|
mime := mime.TypeByExtension(extension)
|
|
|
|
header := w.Header()
|
|
header.Set("Content-Type", mime)
|
|
|
|
w.Write(file)
|
|
return 200, nil
|
|
}
|
|
|
|
if page == "browse" {
|
|
code, err = browse.Execute(w, r)
|
|
}
|
|
|
|
if page == "edit" {
|
|
code, err = edit.Execute(w, r)
|
|
}
|
|
|
|
if r.URL.Path == "/admin/settings/" {
|
|
code, err = settings.Execute(w, r)
|
|
}
|
|
|
|
if r.Header.Get("X-Regenerate") == "true" {
|
|
commands.Execute()
|
|
}
|
|
|
|
return code, err
|
|
}
|
|
|
|
return h.Next.ServeHTTP(w, r)
|
|
}
|