spend-sparrow/internal/account/handler.go

package account

import (
	"github.com/a-h/templ"
	"net/http"
	"spend-sparrow/internal/core"
	"spend-sparrow/internal/utils"
)

type Handler struct {
	s Service
	r *core.Render
}

func NewHandler(s Service, r *core.Render) Handler {
	return Handler{
		s: s,
		r: r,
	}
}

func (h Handler) Handle(r *http.ServeMux) {
	r.Handle("GET /account", h.handleAccountPage())
	r.Handle("GET /account/{id}", h.handleAccountItemComp())
	r.Handle("POST /account/{id}", h.handleUpdateAccount())
	r.Handle("DELETE /account/{id}", h.handleDeleteAccount())
}

func (h Handler) handleAccountPage() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		core.UpdateSpan(r)

		user := core.GetUser(r)
		if user == nil {
			utils.DoRedirect(w, r, "/auth/signin")
			return
		}

		accounts, err := h.s.GetAll(r.Context(), user)
		if err != nil {
			core.HandleError(w, r, err)
			return
		}

		comp := template(accounts)
		h.r.RenderLayout(r, w, comp, user)
	}
}

func (h Handler) handleAccountItemComp() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		core.UpdateSpan(r)

		user := core.GetUser(r)
		if user == nil {
			utils.DoRedirect(w, r, "/auth/signin")
			return
		}

		id := r.PathValue("id")
		if id == "new" {
			comp := editAccount(nil)
			h.r.Render(r, w, comp)
			return
		}

		account, err := h.s.Get(r.Context(), user, id)
		if err != nil {
			core.HandleError(w, r, err)
			return
		}

		var comp templ.Component
		if r.URL.Query().Get("edit") == "true" {
			comp = editAccount(account)
		} else {
			comp = accountItem(account)
		}
		h.r.Render(r, w, comp)
	}
}

func (h Handler) handleUpdateAccount() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		core.UpdateSpan(r)

		user := core.GetUser(r)
		if user == nil {
			utils.DoRedirect(w, r, "/auth/signin")
			return
		}

		var (
			account *Account
			err     error
		)
		id := r.PathValue("id")
		name := r.FormValue("name")
		if id == "new" {
			account, err = h.s.Add(r.Context(), user, name)
			if err != nil {
				core.HandleError(w, r, err)
				return
			}
		} else {
			account, err = h.s.UpdateName(r.Context(), user, id, name)
			if err != nil {
				core.HandleError(w, r, err)
				return
			}
		}

		comp := accountItem(account)
		h.r.Render(r, w, comp)
	}
}

func (h Handler) handleDeleteAccount() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		core.UpdateSpan(r)

		user := core.GetUser(r)
		if user == nil {
			utils.DoRedirect(w, r, "/auth/signin")
			return
		}

		id := r.PathValue("id")

		err := h.s.Delete(r.Context(), user, id)
		if err != nil {
			core.HandleError(w, r, err)
			return
		}
	}
}