jiu/backend/internal/handler/product.go

package handler

import (
	"errors"
	"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 ProductHandler struct {
	db *gorm.DB
}

func NewProductHandler(db *gorm.DB) *ProductHandler {
	return &ProductHandler{db: db}
}

// List GET /api/v1/products
func (h *ProductHandler) List(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
	keyword := c.Query("keyword")
	categoryID := c.Query("category_id")

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

	if keyword != "" {
		query = query.Where("name LIKE ? OR code LIKE ? OR barcode LIKE ?",
			"%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%")
	}
	if categoryID != "" {
		query = query.Where("category_id = ?", categoryID)
	}

	var total int64
	query.Count(&total)

	var products []model.Product
	offset := (page - 1) * pageSize
	query.Preload("Category").Offset(offset).Limit(pageSize).
		Order("id DESC").Find(&products)

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

// Create POST /api/v1/products
func (h *ProductHandler) Create(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	var product model.Product
	if err := c.ShouldBindJSON(&product); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}
	product.ShopID = shopID

	// Auto-generate product code if not provided (e.g. P001, P002)
	// Retry up to 5 times on duplicate key to handle concurrent creates
	if product.Code == "" {
		var count int64
		h.db.Model(&model.Product{}).
			Where("shop_id = ? AND deleted_at IS NULL", shopID).
			Count(&count)
		product.Code = fmt.Sprintf("P%03d", count+1)
	}

	var createErr error
	for attempt := 0; attempt < 5; attempt++ {
		if createErr = h.db.Create(&product).Error; createErr == nil {
			break
		}
		if !errors.Is(createErr, gorm.ErrDuplicatedKey) {
			break
		}
		// Duplicate code: try next slot
		var count int64
		h.db.Model(&model.Product{}).
			Where("shop_id = ? AND deleted_at IS NULL", shopID).
			Count(&count)
		product.ID = 0
		product.Code = fmt.Sprintf("P%03d", count+int64(attempt)+2)
	}
	if createErr != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": createErr.Error()})
		return
	}
	c.JSON(http.StatusCreated, gin.H{"data": product})
}

// Update PUT /api/v1/products/:id
func (h *ProductHandler) Update(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	id := c.Param("id")

	var product model.Product
	if err := h.db.Where("id = ? AND shop_id = ? AND deleted_at IS NULL", id, shopID).
		First(&product).Error; err != nil {
		c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
		return
	}

	var req model.Product
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}

	// 只更新业务字段，防止 Save() 覆盖 shop_id / created_at 等系统字段
	if err := h.db.Model(&product).Updates(map[string]interface{}{
		"code":           req.Code,
		"barcode":        req.Barcode,
		"name":           req.Name,
		"series":         req.Series,
		"spec":           req.Spec,
		"unit":           req.Unit,
		"category_id":    req.CategoryID,
		"brand":          req.Brand,
		"purchase_price": req.PurchasePrice,
		"sale_price":     req.SalePrice,
		"min_stock":      req.MinStock,
		"remark":         req.Remark,
		"custom_fields":  req.CustomFields,
	}).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}

	// 重新读取完整数据返回
	h.db.Preload("Category").First(&product, product.ID)
	c.JSON(http.StatusOK, gin.H{"data": product})
}

// Delete DELETE /api/v1/products/:id （软删除）
func (h *ProductHandler) Delete(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	id := c.Param("id")

	now := timeNow()
	result := h.db.Model(&model.Product{}).
		Where("id = ? AND shop_id = ? AND deleted_at IS NULL", 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"})
}