chore(auth): add test for retrieving session from db #181

handler/default.go

@@ -3,17 +3,19 @@ package handler
 import (
 	"me-fit/middleware"
 	"me-fit/service"
+	"me-fit/template"
+	"me-fit/utils"
 
 	"database/sql"
 	"net/http"
 )
 
 func GetHandler(db *sql.DB) http.Handler {
-	var router = http.NewServeMux()
+	router := http.NewServeMux()
 
-	router.HandleFunc("/", service.HandleIndexAnd404(db))
+	router.HandleFunc("/$", handleIndex(db))
+	router.HandleFunc("/", handleNotFound(db))
 
-	// Serve static files (CSS, JS and images)
 	router.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("./static/"))))
 
 	router.Handle("/auth/", authUi(db))
@@ -32,3 +34,33 @@ func GetHandler(db *sql.DB) http.Handler {
 func auth(db *sql.DB, h http.Handler) http.Handler {
 	return middleware.EnsureValidSession(db, h)
 }
+
+func handleIndex(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		user := service.GetUserFromRequest(db, r)
+
+		err := template.
+			Layout(template.Index(), service.UserInfoComp(user)).
+			Render(r.Context(), w)
+		if err != nil {
+			utils.LogError("Failed to render index", err)
+			http.Error(w, "Failed to render index", http.StatusInternalServerError)
+		}
+	}
+}
+
+func handleNotFound(db *sql.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		user := service.GetUserFromRequest(db, r)
+
+		err := template.
+			Layout(template.NotFound(), service.UserInfoComp(user)).
+			Render(r.Context(), w)
+		if err != nil {
+			utils.LogError("Failed to render index", err)
+			http.Error(w, "Failed to render index", http.StatusInternalServerError)
+		}
+
+		w.WriteHeader(http.StatusNotFound)
+	}
+}