spend-sparrow/internal/budget/handler.go

package budget

import (
	"fmt"
	"math"
	"net/http"
	"spend-sparrow/internal/core"
	"strconv"

	"github.com/google/uuid"
)

const (
	DECIMALS_MULTIPLIER = 100
)

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

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

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

func (h HandlerImpl) Handle(r *http.ServeMux) {
	r.Handle("GET /budget", h.handlePage())
	r.Handle("GET /budget/new", h.handleNew())
	r.Handle("GET /budget/{id}", h.handleEdit())
	r.Handle("POST /budget/{id}", h.handlePost())
	r.Handle("DELETE /budget/{id}", h.handleDelete())
}

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

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

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

		comp := page(budgets)
		h.r.RenderLayout(r, w, comp, user)
	}
}

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

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

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

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

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

		id, err := uuid.Parse(r.PathValue("id"))
		if err != nil {
			core.HandleError(w, r, fmt.Errorf("could not parse Id: %w", core.ErrBadRequest))
			return
		}

		budget, err := h.s.Get(r.Context(), user, id)
		if err != nil {
			core.HandleError(w, r, fmt.Errorf("could not parse Id: %w", core.ErrBadRequest))
			return
		}

		comp := edit(*budget)
		h.r.RenderLayout(r, w, comp, user)
	}
}

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

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

		var (
			id  uuid.UUID
			err error
		)

		idStr := r.PathValue("id")
		if idStr != "new" {
			id, err = uuid.Parse(idStr)
			if err != nil {
				core.HandleError(w, r, fmt.Errorf("could not parse Id: %w", core.ErrBadRequest))
				return
			}
		}

		valueF, err := strconv.ParseFloat(r.FormValue("value"), 64)
		if err != nil {
			core.HandleError(w, r, fmt.Errorf("could not parse value: %w", core.ErrBadRequest))
			return
		}
		value := int64(math.Round(valueF * DECIMALS_MULTIPLIER))

		input := Budget{
			Id:    id,
			Name:  r.FormValue("name"),
			Value: value,
		}

		if idStr == "new" {
			_, err = h.s.Add(r.Context(), user, input)
			if err != nil {
				core.HandleError(w, r, err)
				return
			}
		} else {
			_, err = h.s.Update(r.Context(), user, input)
			if err != nil {
				core.HandleError(w, r, err)
				return
			}
		}

		core.DoRedirect(w, r, "/budget")
	}
}

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

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

		idStr := r.PathValue("id")
		id, err := uuid.Parse(idStr)
		if err != nil {
			core.HandleError(w, r, fmt.Errorf("could not parse Id: %w", core.ErrBadRequest))
			return
		}

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

		core.DoRedirect(w, r, "/budget")
	}
}