neko/internal/api/members/handler.go

package members

import (
	"context"
	"errors"
	"net/http"

	"github.com/go-chi/chi"

	"demodesk/neko/internal/http/auth"
	"demodesk/neko/internal/types"
	"demodesk/neko/internal/utils"
)

type key int

const (
	keyMemberCtx key = iota
)

type MembersHandler struct {
	members types.MemberManager
}

func New(
	members types.MemberManager,
) *MembersHandler {
	// Init

	return &MembersHandler{
		members: members,
	}
}

func (h *MembersHandler) Route(r chi.Router) {
	r.Get("/", h.membersList)

	r.With(auth.AdminsOnly).Group(func(r chi.Router) {
		r.Post("/", h.membersCreate)
		r.With(h.ExtractMember).Route("/{memberId}", func(r chi.Router) {
			r.Get("/", h.membersRead)
			r.Post("/", h.membersUpdateProfile)
			r.Post("/password", h.membersUpdatePassword)
			r.Delete("/", h.membersDelete)
		})
	})
}

func (h *MembersHandler) RouteBulk(r chi.Router) {
	r.With(auth.AdminsOnly).Group(func(r chi.Router) {
		r.Post("/update", h.membersBulkUpdate)
	})
}

type MemberData struct {
	ID      string
	Profile types.MemberProfile
}

func SetMember(r *http.Request, session MemberData) *http.Request {
	ctx := context.WithValue(r.Context(), keyMemberCtx, session)
	return r.WithContext(ctx)
}

func GetMember(r *http.Request) MemberData {
	return r.Context().Value(keyMemberCtx).(MemberData)
}

func (h *MembersHandler) ExtractMember(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		memberId := chi.URLParam(r, "memberId")

		profile, err := h.members.Select(memberId)
		if err != nil {
			if errors.Is(err, types.ErrMemberDoesNotExist) {
				utils.HttpNotFound(w, err)
			} else {
				utils.HttpInternalServerError(w, err)
			}

			return
		}

		next.ServeHTTP(w, SetMember(r, MemberData{
			ID:      memberId,
			Profile: profile,
		}))
	})
}