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)
	}
}