codeword/internal/controller/admin/admin_promocode/promocode_controller.go
2025-01-04 17:01:27 +03:00

258 lines
8.4 KiB
Go

package admin_promocode
import (
"errors"
"fmt"
"gitea.pena/PenaSide/codeword/internal/models"
"gitea.pena/PenaSide/codeword/internal/repository"
"gitea.pena/PenaSide/codeword/internal/services"
"gitea.pena/PenaSide/codeword/internal/utils/middleware"
"github.com/gofiber/fiber/v2"
"go.uber.org/zap"
"gitea.pena/PenaSide/common/log_mw"
)
type Deps struct {
Logger *zap.Logger
PromoCodeService *services.PromoCodeService
}
type PromoCodeController struct {
logger *zap.Logger
promoCodeService *services.PromoCodeService
}
func NewPromoCodeController(deps Deps) *PromoCodeController {
return &PromoCodeController{
logger: deps.Logger,
promoCodeService: deps.PromoCodeService,
}
}
func (p *PromoCodeController) CreatePromoCode(c *fiber.Ctx) error {
userID := middleware.ExtractUserID(c)
hlogger := log_mw.ExtractLogger(c)
var req models.PromoCode
if err := c.BodyParser(&req); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
}
if req.Codeword == "" {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "codeword is required"})
}
req.ActivationLimit = req.ActivationCount
createdPromoCode, err := p.promoCodeService.CreatePromoCode(c.Context(), &req)
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"})
}
var keyType string
var ctxFactor float64
var keyTargetType string
var ctxTarget string
var ctxAmount int64
if createdPromoCode.Bonus.Privilege.PrivilegeID != "" && createdPromoCode.Bonus.Discount.Factor != 0 && createdPromoCode.Bonus.Discount.Factor != 1 {
keyType = "privilege,discount"
keyTargetType = "privilege,service"
ctxTarget = fmt.Sprintf("%s,%s", createdPromoCode.Bonus.Privilege.PrivilegeID, createdPromoCode.Bonus.Discount.Target)
ctxAmount = int64(createdPromoCode.Bonus.Privilege.Amount)
ctxFactor = createdPromoCode.Bonus.Discount.Factor
} else if createdPromoCode.Bonus.Privilege.PrivilegeID != "" {
keyType = "privilege"
keyTargetType = "privilege"
ctxTarget = createdPromoCode.Bonus.Privilege.PrivilegeID
ctxAmount = int64(createdPromoCode.Bonus.Privilege.Amount)
} else if createdPromoCode.Bonus.Discount.Factor != 0 && createdPromoCode.Bonus.Discount.Factor != 1 {
keyType = "discount"
keyTargetType = "service"
ctxFactor = createdPromoCode.Bonus.Discount.Factor
ctxTarget = createdPromoCode.Bonus.Discount.Target
}
hlogger.Emit(models.InfoPromocodeCreated{
CtxID: createdPromoCode.ID.String(),
CtxUserID: userID,
KeyType: keyType,
CtxFactor: ctxFactor,
KeyTargetType: keyTargetType,
CtxTarget: ctxTarget,
CtxAmount: ctxAmount,
CtxCode: createdPromoCode.Codeword,
})
return c.Status(fiber.StatusOK).JSON(createdPromoCode)
}
func (p *PromoCodeController) EditPromoCode(c *fiber.Ctx) error {
userID := middleware.ExtractUserID(c)
hlogger := log_mw.ExtractLogger(c)
var req models.ReqEditPromoCode
if err := c.BodyParser(&req); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
}
if req.ID == "" {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "promocode ID is required"})
}
req.ActivationLimit = req.ActivationCount
editedPromoCode, err := p.promoCodeService.EditPromoCode(c.Context(), &req)
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"})
}
var keyType string
var ctxFactor float64
var keyTargetType string
var ctxTarget string
var ctxAmount int64
if editedPromoCode.Bonus.Privilege.PrivilegeID != "" && editedPromoCode.Bonus.Discount.Factor != 0 && editedPromoCode.Bonus.Discount.Factor != 1 {
keyType = "privilege,discount"
keyTargetType = "privilege,service"
ctxTarget = fmt.Sprintf("%s,%s", editedPromoCode.Bonus.Privilege.PrivilegeID, editedPromoCode.Bonus.Discount.Target)
ctxAmount = int64(editedPromoCode.Bonus.Privilege.Amount)
ctxFactor = editedPromoCode.Bonus.Discount.Factor
} else if editedPromoCode.Bonus.Privilege.PrivilegeID != "" {
keyType = "privilege"
keyTargetType = "privilege"
ctxTarget = editedPromoCode.Bonus.Privilege.PrivilegeID
ctxAmount = int64(editedPromoCode.Bonus.Privilege.Amount)
} else if editedPromoCode.Bonus.Discount.Factor != 0 && editedPromoCode.Bonus.Discount.Factor != 1 {
keyType = "discount"
keyTargetType = "service"
ctxFactor = editedPromoCode.Bonus.Discount.Factor
ctxTarget = editedPromoCode.Bonus.Discount.Target
}
hlogger.Emit(models.InfoPromocodeUpdated{
CtxID: editedPromoCode.ID.String(),
CtxUserID: userID,
KeyType: keyType,
CtxFactor: ctxFactor,
KeyTargetType: keyTargetType,
CtxTarget: ctxTarget,
CtxAmount: ctxAmount,
CtxCode: editedPromoCode.Codeword,
})
return c.Status(fiber.StatusOK).JSON(editedPromoCode)
}
func (p *PromoCodeController) GetList(c *fiber.Ctx) error {
var req models.GetPromoCodesListReq
if err := c.BodyParser(&req); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
}
promoCodes, count, err := p.promoCodeService.GetPromoCodesList(c.Context(), &req)
if err != nil {
p.logger.Error("Failed to retrieve promocode list", zap.Error(err))
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
}
return c.Status(fiber.StatusOK).JSON(models.GetPromoCodesListResp{
Count: count,
Items: promoCodes,
})
}
func (p *PromoCodeController) Delete(c *fiber.Ctx) error {
userID := middleware.ExtractUserID(c)
hlogger := log_mw.ExtractLogger(c)
promoCodeID := c.Params("promocodeID")
if promoCodeID == "" {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "PromoCode ID is required"})
}
err := p.promoCodeService.DeletePromoCode(c.Context(), promoCodeID)
if err != nil {
p.logger.Error("Failed to delete 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"})
}
hlogger.Emit(models.InfoPromocodeDeleted{
CtxID: promoCodeID,
CtxUserID: userID,
})
return c.SendStatus(fiber.StatusOK)
}
func (p *PromoCodeController) CreateFastLink(c *fiber.Ctx) error {
userID := middleware.ExtractUserID(c)
hlogger := log_mw.ExtractLogger(c)
var req struct {
PromoCodeID string `json:"id"`
}
if err := c.BodyParser(&req); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
}
if req.PromoCodeID == "" {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "PromoCode ID is required"})
}
fastLink, err := p.promoCodeService.CreateFastLink(c.Context(), req.PromoCodeID)
if err != nil {
p.logger.Error("Failed to create fastlink", 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"})
}
hlogger.Emit(models.InfoFastlinkCreated{
CtxID: fastLink,
CtxPromocodeID: req.PromoCodeID,
CtxUserID: userID,
})
return c.Status(fiber.StatusCreated).JSON(fiber.Map{"fastlink": fastLink})
}
func (p *PromoCodeController) GetStats(c *fiber.Ctx) error {
var req models.PromoStatReq
if err := c.BodyParser(&req); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
}
if req.PromoCodeID == "" {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "PromoCode ID is required"})
}
promoStats, err := p.promoCodeService.GetStats(c.Context(), req)
if err != nil {
p.logger.Error("Failed getting promo stats", zap.Error(err))
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error: " + err.Error()})
}
return c.Status(fiber.StatusOK).JSON(promoStats)
}