feat: added notification system with toasts #160

service/index_and_404.go

@@ -0,0 +1,28 @@
+package service
+
+import (
+	"database/sql"
+	"me-fit/template"
+	"me-fit/utils"
+	"net/http"
+
+	"github.com/a-h/templ"
+)
+
+func HandleIndexAnd404(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		user := utils.GetUserFromSession(db, r)
+
+		var comp templ.Component = nil
+		userComp := UserInfoComp(user)
+
+		if r.URL.Path != "/" {
+			comp = template.Layout(template.NotFound(), userComp)
+			w.WriteHeader(http.StatusNotFound)
+		} else {
+			comp = template.Layout(template.Index(), userComp)
+		}
+
+		comp.Render(r.Context(), w)
+	}
+}