controller.go

package user_http

import (
	"errors"
	"net/http"

	"app/adapter/featureflags"
	"app/user"

	"github.com/go-playground/validator/v10"
	"github.com/gofiber/fiber/v2"
)

func NewController(service *user.Service, validator *validator.Validate) *Controller {
	return &Controller{service: service, validator: validator}
}

type Controller struct {
	service   *user.Service
	validator *validator.Validate
}

func (controller *Controller) Register(app *fiber.App) {
	app.Group("/users").
		Post("/", controller.Create).
		Get("/", controller.List)

	app.Group("/users/:userID", controller.middlewareGetUser, featureflags.Middleware).
		Get("/", controller.Get).
		Delete("/", controller.Delete).
		Get("/feature", controller.GetFeature)
}

func (controller *Controller) List(ctx *fiber.Ctx) error {
	users, err := controller.service.List(ctx.UserContext())
	if err != nil {
		return ctx.SendStatus(http.StatusInternalServerError)
	}

	return ctx.JSON(users)
}

func (controller *Controller) Create(ctx *fiber.Ctx) error {
	var body UserCreateDTO
	if err := ctx.BodyParser(&body); err != nil {
		return ctx.SendStatus(http.StatusBadRequest)
	}

	if err := controller.validator.Struct(body); err != nil {
		var errorMessages []string
		for _, err := range err.(validator.ValidationErrors) {
			errorMessages = append(errorMessages, err.Error())
		}
		return ctx.Status(http.StatusBadRequest).JSON(fiber.Map{"errors": errorMessages})
	}

	createdUser, err := controller.service.CreateUser(ctx.UserContext(), body.Email)
	if err != nil {
		return ctx.SendStatus(http.StatusInternalServerError)
	}

	return ctx.Status(http.StatusCreated).JSON(createdUser)
}

func (controller *Controller) Get(ctx *fiber.Ctx) error {
	return ctx.JSON(ctx.Locals("user"))
}

func (controller *Controller) GetFeature(ctx *fiber.Ctx) error {
	return ctx.JSON(ctx.Locals("flags"))
}

func (controller *Controller) Delete(ctx *fiber.Ctx) error {
	foundUser := ctx.Locals("user").(user.User)

	if err := controller.service.Remove(ctx.UserContext(), foundUser); err != nil {
		return ctx.SendStatus(http.StatusInternalServerError)
	}

	return ctx.SendStatus(http.StatusOK)
}

func (controller *Controller) middlewareGetUser(ctx *fiber.Ctx) error {
	userID := ctx.Params("userID")
	ctx.AllParams()
	if userID == "" {
		return ctx.SendStatus(http.StatusBadRequest)
	}

	userFound, err := controller.service.FindByID(ctx.UserContext(), userID)
	if err != nil {
		if errors.Is(err, user.ErrNotFound) {
			return ctx.SendStatus(http.StatusNotFound)
		} else {
			return ctx.SendStatus(http.StatusInternalServerError)
		}
	}

	ctx.Locals("user", userFound)
	return ctx.Next()
}