package handler

import (
	"fmt"
	"net/http"
	"strconv"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"

	"github.com/wangjia/jiu/backend/internal/middleware"
	"github.com/wangjia/jiu/backend/internal/model"
)

type PartnerHandler struct {
	db *gorm.DB
}

func NewPartnerHandler(db *gorm.DB) *PartnerHandler {
	return &PartnerHandler{db: db}
}

func (h *PartnerHandler) List(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))

	query := h.db.Model(&model.Partner{}).
		Where("shop_id = ? AND deleted_at IS NULL", shopID)

	if t := c.Query("type"); t != "" {
		query = query.Where("FIND_IN_SET(?, type)", t)
	}
	if kw := c.Query("keyword"); kw != "" {
		query = query.Where("name LIKE ? OR phone LIKE ?", "%"+kw+"%", "%"+kw+"%")
	}

	var total int64
	query.Count(&total)

	partners := make([]model.Partner, 0)
	query.Offset((page - 1) * pageSize).Limit(pageSize).Order("id DESC").Find(&partners)

	c.JSON(http.StatusOK, gin.H{"data": partners, "total": total, "page": page, "page_size": pageSize})
}

func (h *PartnerHandler) Create(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	var p model.Partner
	if err := c.ShouldBindJSON(&p); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}
	p.ShopID = shopID
	// 编码未提供时自动生成：供应商 S001/S002…，客户 C001/C002…
	if p.Code == "" {
		prefix := "S"
		if p.Type == "customer" {
			prefix = "C"
		}
		var count int64
		h.db.Model(&model.Partner{}).
			Where("shop_id = ? AND type = ? AND deleted_at IS NULL", shopID, p.Type).
			Count(&count)
		p.Code = fmt.Sprintf("%s%03d", prefix, count+1)
	}
	if err := h.db.Create(&p).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}
	c.JSON(http.StatusCreated, gin.H{"data": p})
}

func (h *PartnerHandler) Update(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	var p model.Partner
	if err := h.db.Where("id = ? AND shop_id = ? AND deleted_at IS NULL", c.Param("id"), shopID).
		First(&p).Error; err != nil {
		c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
		return
	}
	if err := c.ShouldBindJSON(&p); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}
	p.ShopID = shopID
	h.db.Save(&p)
	c.JSON(http.StatusOK, gin.H{"data": p})
}

func (h *PartnerHandler) Delete(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	now := timeNow()
	result := h.db.Model(&model.Partner{}).
		Where("id = ? AND shop_id = ? AND deleted_at IS NULL", c.Param("id"), shopID).
		Update("deleted_at", now)
	if result.RowsAffected == 0 {
		c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
		return
	}
	c.JSON(http.StatusOK, gin.H{"message": "deleted"})
}