initial commit

internal/handlers/static.go

@@ -0,0 +1,70 @@
+package handlers
+
+import (
+	"git.haelnorr.com/h/golib/hws"
+	"net/http"
+	"path/filepath"
+	"strings"
+)
+
+// Handles requests for static files, without allowing access to the
+// directory viewer and returning 404 if an exact file is not found
+func StaticFS(staticFS *http.FileSystem, server *hws.Server) http.Handler {
+	// Create the file server once, not on every request
+	fs, err := hws.SafeFileServer(staticFS)
+	if err != nil {
+		// If we can't create the file server, return a handler that always errors
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			err = server.ThrowError(w, r, hws.HWSError{
+				StatusCode:      http.StatusInternalServerError,
+				Message:         "An error occured trying to load the file system",
+				Error:           err,
+				Level:           hws.ErrorLevel("error"),
+				RenderErrorPage: true,
+			})
+			if err != nil {
+				server.ThrowFatal(w, err)
+			}
+		})
+	}
+
+	return http.HandlerFunc(
+		func(w http.ResponseWriter, r *http.Request) {
+			// Explicitly set Content-Type for CSS files
+			if strings.HasSuffix(r.URL.Path, ".css") {
+				w.Header().Set("Content-Type", "text/css; charset=utf-8")
+			} else if strings.HasSuffix(r.URL.Path, ".js") {
+				w.Header().Set("Content-Type", "application/javascript; charset=utf-8")
+			} else if strings.HasSuffix(r.URL.Path, ".ico") {
+				w.Header().Set("Content-Type", "image/x-icon")
+			} else {
+				// Let Go detect the content type for other files
+				ext := filepath.Ext(r.URL.Path)
+				if contentType := mimeTypes[ext]; contentType != "" {
+					w.Header().Set("Content-Type", contentType)
+				}
+			}
+			fs.ServeHTTP(w, r)
+		},
+	)
+}
+
+// Common MIME types for static files
+var mimeTypes = map[string]string{
+	".html":  "text/html; charset=utf-8",
+	".css":   "text/css; charset=utf-8",
+	".js":    "application/javascript; charset=utf-8",
+	".json":  "application/json; charset=utf-8",
+	".xml":   "application/xml; charset=utf-8",
+	".png":   "image/png",
+	".jpg":   "image/jpeg",
+	".jpeg":  "image/jpeg",
+	".gif":   "image/gif",
+	".svg":   "image/svg+xml",
+	".ico":   "image/x-icon",
+	".webp":  "image/webp",
+	".woff":  "font/woff",
+	".woff2": "font/woff2",
+	".ttf":   "font/ttf",
+	".eot":   "application/vnd.ms-fontobject",
+}