initial commit

internal/handlers/errorpage.go

@@ -0,0 +1,28 @@
+package handlers
+
+import (
+	"net/http"
+
+	"git.haelnorr.com/h/golib/hws"
+	"git.haelnorr.com/h/oslstats/internal/view/page"
+	"github.com/pkg/errors"
+)
+
+func ErrorPage(
+	errorCode int,
+) (hws.ErrorPage, error) {
+	messages := map[int]string{
+		400: "The request you made was malformed or unexpected.",
+		401: "You need to login to view this page.",
+		403: "You do not have permission to view this page.",
+		404: "The page or resource you have requested does not exist.",
+		500: `An error occured on the server. Please try again, and if this
+        continues to happen contact an administrator.`,
+		503: "The server is currently down for maintenance and should be back soon. =)",
+	}
+	msg, exists := messages[errorCode]
+	if !exists {
+		return nil, errors.New("No valid message for the given code")
+	}
+	return page.Error(errorCode, http.StatusText(errorCode), msg), nil
+}

internal/handlers/index.go

@@ -0,0 +1,49 @@
+package handlers
+
+import (
+	"net/http"
+
+	"git.haelnorr.com/h/oslstats/internal/view/page"
+
+	"git.haelnorr.com/h/golib/hws"
+)
+
+// Handles responses to the / path. Also serves a 404 Page for paths that
+// don't have explicit handlers
+func Index(server *hws.Server) http.Handler {
+	return http.HandlerFunc(
+		func(w http.ResponseWriter, r *http.Request) {
+			if r.URL.Path != "/" {
+				page, err := ErrorPage(http.StatusNotFound)
+				if err != nil {
+					err = server.ThrowError(w, r, hws.HWSError{
+						StatusCode:      http.StatusInternalServerError,
+						Message:         "An error occured trying to generate the error page",
+						Error:           err,
+						Level:           hws.ErrorLevel("error"),
+						RenderErrorPage: false,
+					})
+					if err != nil {
+						server.ThrowFatal(w, err)
+					}
+					return
+				}
+				err = page.Render(r.Context(), w)
+				if err != nil {
+					err = server.ThrowError(w, r, hws.HWSError{
+						StatusCode:      http.StatusInternalServerError,
+						Message:         "An error occured trying to render the error page",
+						Error:           err,
+						Level:           hws.ErrorLevel("error"),
+						RenderErrorPage: false,
+					})
+					if err != nil {
+						server.ThrowFatal(w, err)
+					}
+					return
+				}
+			}
+			page.Index().Render(r.Context(), w)
+		},
+	)
+}

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",
+}