package handler

import (
	"me-fit/handler/middleware"
	"me-fit/service"
	"me-fit/template"

	"net/http"

	"github.com/a-h/templ"
)

type Index interface {
	Handle(router *http.ServeMux)
}

type IndexImpl struct {
	service service.Auth
	render  *Render
}

func NewIndex(service service.Auth, render *Render) Index {
	return IndexImpl{
		service: service,
		render:  render,
	}
}

func (handler IndexImpl) Handle(router *http.ServeMux) {
	router.Handle("/", handler.handleIndexAnd404())
}

func (handler IndexImpl) handleIndexAnd404() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		session := r.Context().Value(middleware.SessionKey).(*service.Session)
		var user *service.User
		if session != nil {
			user = session.User
		}

		var comp templ.Component

		if r.URL.Path != "/" {
			comp = template.NotFound()
			w.WriteHeader(http.StatusNotFound)
		} else {
			comp = template.Index()
		}

		handler.render.RenderLayout(r, w, comp, user)
	}
}