Fix some pathing and caching issues with admin controller

This commit is contained in:
Gabe Kangas 2021-10-12 12:47:58 -07:00
parent a58a1f67d6
commit 01b3489287

View file

@ -2,6 +2,7 @@ package admin
import ( import (
"bytes" "bytes"
"io/fs"
"net/http" "net/http"
"os" "os"
"path/filepath" "path/filepath"
@ -14,13 +15,23 @@ import (
// ServeAdmin will return admin web assets. // ServeAdmin will return admin web assets.
func ServeAdmin(w http.ResponseWriter, r *http.Request) { func ServeAdmin(w http.ResponseWriter, r *http.Request) {
// If the ETags match then return a StatusNotModified
if responseCode := middleware.ProcessEtags(w, r); responseCode != 0 {
w.WriteHeader(responseCode)
return
}
adminFiles := static.GetAdmin() adminFiles := static.GetAdmin()
path := strings.TrimPrefix(r.URL.Path, "/")
// Determine if the requested path is a directory. // Determine if the requested path is a directory.
// If so, append index.html to the request. // If so, append index.html to the request.
path := strings.TrimPrefix(r.URL.Path, "/") if info, err := fs.Stat(adminFiles, path); err == nil && info.IsDir() {
if strings.HasSuffix(path, "/") {
path = filepath.Join(path, "index.html") path = filepath.Join(path, "index.html")
} else if _, err := fs.Stat(adminFiles, path+"index.html"); err == nil {
path = filepath.Join(path, "index.html")
} else if err != nil {
log.Errorln(err)
} }
f, err := adminFiles.Open(path) f, err := adminFiles.Open(path)
@ -30,7 +41,6 @@ func ServeAdmin(w http.ResponseWriter, r *http.Request) {
} }
info, err := f.Stat() info, err := f.Stat()
if os.IsNotExist(err) { if os.IsNotExist(err) {
w.WriteHeader(http.StatusNotFound) w.WriteHeader(http.StatusNotFound)
return return
@ -38,7 +48,6 @@ func ServeAdmin(w http.ResponseWriter, r *http.Request) {
// Set a cache control max-age header // Set a cache control max-age header
middleware.SetCachingHeaders(w, r) middleware.SetCachingHeaders(w, r)
d, err := adminFiles.ReadFile(path) d, err := adminFiles.ReadFile(path)
if err != nil { if err != nil {
log.Errorln(err) log.Errorln(err)