tariffs/internal/controller/tariff_external/controller.go

package tariff_external

import (
	"errors"
	"github.com/gofiber/fiber/v2"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.uber.org/zap"
	"hub_admin_backend_service/internal/controller/middleware"
	our_errors "hub_admin_backend_service/internal/errors"
	"hub_admin_backend_service/internal/models"
	"hub_admin_backend_service/internal/repository/tariff"
	"hub_admin_backend_service/internal/tools"
)

type Deps struct {
	Repo       *tariff.Tariff
	Logger     *zap.Logger
	MiddleWare *middleware.MiddleWare
}

type TariffExternal struct {
	repo       *tariff.Tariff
	logger     *zap.Logger
	middleWare *middleware.MiddleWare
}

func NewTariffExternal(deps Deps) *TariffExternal {
	return &TariffExternal{
		repo:       deps.Repo,
		logger:     deps.Logger,
		middleWare: deps.MiddleWare,
	}
}

func (t *TariffExternal) Get(ctx *fiber.Ctx) error {
	tariffID := ctx.Params("id")
	if tariffID == "" {
		return ctx.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "url field id don't be empty"})
	}

	objID, err := primitive.ObjectIDFromHex(tariffID)
	if err != nil {
		return ctx.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "no valid object tariff id"})
	}

	result, err := t.repo.GetByID(ctx.Context(), objID)
	if err != nil {
		switch {
		case errors.Is(err, our_errors.ErrNotFound):
			return ctx.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "Tariff not found"})
		default:
			return ctx.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": err.Error()})
		}
	}

	return ctx.Status(fiber.StatusOK).JSON(result)
}

func (t *TariffExternal) GetList(ctx *fiber.Ctx) error {
	page, limit := t.middleWare.GetPaginationParameters(ctx)

	userID, ok := t.middleWare.ExtractUserID(ctx)
	if !ok {
		return ctx.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "No auth"})
	}

	result, err := t.repo.GetList(ctx.Context(), page, limit, userID)
	if err != nil {
		return ctx.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": err.Error()})
	}

	return ctx.Status(fiber.StatusOK).JSON(result)
}

func (t *TariffExternal) Create(ctx *fiber.Ctx) error {
	userID, ok := t.middleWare.ExtractUserID(ctx)
	if !ok {
		return ctx.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "No auth"})
	}

	var req models.Tariff
	if err := ctx.BodyParser(&req); err != nil {
		return ctx.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid request payload"})
	}
	req.UserID = userID

	err := tools.ValidateTariff(req)
	if err != nil {
		return ctx.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": err.Error()})
	}

	result, err := t.repo.Create(ctx.Context(), req)
	if err != nil {
		return ctx.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": err.Error()})
	}

	return ctx.Status(fiber.StatusOK).JSON(result)
}