package admin

import (
	"strconv"

	"github.com/gofiber/fiber/v2"

	"github.com/break/junhong_cmp_fiber/internal/model/dto"
	iotCardImportService "github.com/break/junhong_cmp_fiber/internal/service/iot_card_import"
	"github.com/break/junhong_cmp_fiber/pkg/errors"
	"github.com/break/junhong_cmp_fiber/pkg/response"
)

type IotCardImportHandler struct {
	service *iotCardImportService.Service
}

func NewIotCardImportHandler(service *iotCardImportService.Service) *IotCardImportHandler {
	return &IotCardImportHandler{service: service}
}

func (h *IotCardImportHandler) Import(c *fiber.Ctx) error {
	var req dto.ImportIotCardRequest
	if err := c.BodyParser(&req); err != nil {
		return errors.New(errors.CodeInvalidParam, "请求参数解析失败")
	}

	file, err := c.FormFile("file")
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "请上传 CSV 文件")
	}

	f, err := file.Open()
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无法读取上传文件")
	}
	defer f.Close()

	result, err := h.service.CreateImportTask(c.UserContext(), &req, f, file.Filename)
	if err != nil {
		return err
	}

	return response.Success(c, result)
}

func (h *IotCardImportHandler) List(c *fiber.Ctx) error {
	var req dto.ListImportTaskRequest
	if err := c.QueryParser(&req); err != nil {
		return errors.New(errors.CodeInvalidParam, "请求参数解析失败")
	}

	result, err := h.service.List(c.UserContext(), &req)
	if err != nil {
		return err
	}

	return response.SuccessWithPagination(c, result.List, result.Total, result.Page, result.PageSize)
}

func (h *IotCardImportHandler) GetByID(c *fiber.Ctx) error {
	idStr := c.Params("id")
	id, err := strconv.ParseUint(idStr, 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无效的任务ID")
	}

	result, err := h.service.GetByID(c.UserContext(), uint(id))
	if err != nil {
		return err
	}

	return response.Success(c, result)
}