package handler

import (
	"BE-MiniERP/database"
	"BE-MiniERP/modules/inventory/models"
	"BE-MiniERP/modules/inventory/repository"
	"strconv"

	"github.com/gofiber/fiber/v2"
)

type SizeHandler struct {
	Repo *repository.SizeRepository
}

func NewSizeHandler() *SizeHandler {
	return &SizeHandler{
		Repo: repository.NewSizeRepository(database.DB),
	}
}

func (h *SizeHandler) GetAll(c *fiber.Ctx) error {
	sizes, err := h.Repo.FindAll()
	if err != nil {
		return c.Status(500).JSON(fiber.Map{"error": "Failed to fetch sizes"})
	}
	return c.JSON(sizes)
}

func (h *SizeHandler) Create(c *fiber.Ctx) error {
	var input models.Size
	if err := c.BodyParser(&input); err != nil {
		return c.Status(400).JSON(fiber.Map{"error": "Invalid input"})
	}
	if err := h.Repo.Create(&input); err != nil {
		return c.Status(500).JSON(fiber.Map{"error": "Failed to create size"})
	}
	return c.JSON(input)
}

func (h *SizeHandler) Update(c *fiber.Ctx) error {
	id, _ := strconv.Atoi(c.Params("id"))
	var input models.Size
	if err := c.BodyParser(&input); err != nil {
		return c.Status(400).JSON(fiber.Map{"error": "Invalid input"})
	}
	if err := h.Repo.Update(uint(id), &input); err != nil {
		return c.Status(500).JSON(fiber.Map{"error": "Failed to update size"})
	}
	return c.JSON(fiber.Map{"message": "Size updated"})
}

func (h *SizeHandler) Delete(c *fiber.Ctx) error {
	id, _ := strconv.Atoi(c.Params("id"))
	if err := h.Repo.Delete(uint(id)); err != nil {
		return c.Status(500).JSON(fiber.Map{"error": "Failed to delete size"})
	}
	return c.JSON(fiber.Map{"message": "Size deleted"})
}