codeword/internal/controller/promocode/promocode_controller.go

package promocode

import (
	"codeword/internal/models"
	"codeword/internal/repository"
	"codeword/internal/services"
	"errors"
	"github.com/gofiber/fiber/v2"
	"go.uber.org/zap"
)

type PromoCodeController struct {
	logger           *zap.Logger
	promoCodeService *services.PromoCodeService
}

func NewPromoCodeController(logger *zap.Logger, promoCodeService *services.PromoCodeService) *PromoCodeController {
	return &PromoCodeController{
		logger:           logger,
		promoCodeService: promoCodeService,
	}
}

func (p *PromoCodeController) CreatePromoCode(c *fiber.Ctx) error {
	var reqCreatePromoCode models.PromoCode
	if err := c.BodyParser(&reqCreatePromoCode); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
	}

	createdPromoCode, err := p.promoCodeService.CreatePromoCode(c.Context(), &reqCreatePromoCode)
	if err != nil {
		p.logger.Error("Failed to create promocode", zap.Error(err))

		if errors.Is(err, repository.ErrDuplicateCodeword) {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Duplicate Codeword"})
		}

		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
	}

	return c.Status(fiber.StatusCreated).JSON(createdPromoCode)
}

func (p *PromoCodeController) EditPromoCode(c *fiber.Ctx) error {
	var reqEditPromoCode models.ReqEditPromoCode
	if err := c.BodyParser(&reqEditPromoCode); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
	}

	editedPromoCode, err := p.promoCodeService.EditPromoCode(c.Context(), &reqEditPromoCode)
	if err != nil {
		p.logger.Error("Failed to edit promocode", zap.Error(err))

		if errors.Is(err, repository.ErrPromoCodeNotFound) {
			return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "PromoCode not found"})
		}

		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
	}

	return c.Status(fiber.StatusOK).JSON(editedPromoCode)
}