jiu/backend/internal/handler/stock_in.go

package handler

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

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

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

func timeNow() *time.Time {
	t := time.Now()
	return &t
}

type StockInHandler struct {
	db       *gorm.DB
	stockSvc *service.StockService
}

func NewStockInHandler(db *gorm.DB, svc *service.StockService) *StockInHandler {
	return &StockInHandler{db: db, stockSvc: svc}
}

// List GET /api/v1/stock-in/orders
func (h *StockInHandler) 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.StockInOrder{}).
		Where("shop_id = ? AND deleted_at IS NULL", shopID)

	if status := c.Query("status"); status != "" {
		query = query.Where("status = ?", status)
	}
	if startDate := c.Query("start_date"); startDate != "" {
		query = query.Where("order_date >= ?", startDate)
	}
	if endDate := c.Query("end_date"); endDate != "" {
		query = query.Where("order_date <= ?", endDate)
	}

	var total int64
	query.Count(&total)

	orders := make([]model.StockInOrder, 0)
	query.Preload("Warehouse").Preload("Partner").Preload("Operator").Preload("Reviewer").
		Offset((page - 1) * pageSize).Limit(pageSize).
		Order("id DESC").Find(&orders)

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

// Get GET /api/v1/stock-in/orders/:id
func (h *StockInHandler) Get(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	var order model.StockInOrder
	if err := h.db.Preload("Items.Product").Preload("Items.Product.Origin").
		Preload("Items.Product.ShelfLife").Preload("Items.Product.Storage").
		Preload("Warehouse").Preload("Partner").Preload("Operator").Preload("Reviewer").
		Where("id = ? AND shop_id = ? AND deleted_at IS NULL", c.Param("id"), shopID).
		First(&order).Error; err != nil {
		c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
		return
	}
	c.JSON(http.StatusOK, gin.H{"data": order})
}

// Create POST /api/v1/stock-in/orders
func (h *StockInHandler) Create(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	operatorID := middleware.GetUserID(c)

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

	req.ShopID = shopID
	req.OperatorID = operatorID
	req.Status = "draft"

	// 生成单号
	orderNo, err := h.stockSvc.GenerateOrderNo(shopID, "stock_in")
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}
	req.OrderNo = orderNo

	// 计算总金额；自动生成批次号
	var total float64
	for i := range req.Items {
		req.Items[i].ShopID = shopID
		req.Items[i].TotalPrice = req.Items[i].Quantity * req.Items[i].UnitPrice
		total += req.Items[i].TotalPrice
		if req.Items[i].BatchNo == "" {
			req.Items[i].BatchNo = fmt.Sprintf("%s-%02d", req.OrderNo, i+1)
		}
	}
	req.TotalAmount = total

	if err := h.db.Create(&req).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}
	c.JSON(http.StatusCreated, gin.H{"data": req})
}

// Update PUT /api/v1/stock-in/orders/:id （只允许草稿状态）
func (h *StockInHandler) Update(c *gin.Context) {
	shopID := middleware.GetShopID(c)

	var order model.StockInOrder
	if err := h.db.Where("id = ? AND shop_id = ? AND status = 'draft' AND deleted_at IS NULL", c.Param("id"), shopID).
		First(&order).Error; err != nil {
		c.JSON(http.StatusNotFound, gin.H{"error": "入库单不存在或不可修改"})
		return
	}

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

	err := h.db.Transaction(func(tx *gorm.DB) error {
		if err := tx.Where("order_id = ?", order.ID).Delete(&model.StockInItem{}).Error; err != nil {
			return err
		}
		var total float64
		for i := range req.Items {
			req.Items[i].ShopID = shopID
			req.Items[i].OrderID = order.ID
			req.Items[i].TotalPrice = req.Items[i].Quantity * req.Items[i].UnitPrice
			total += req.Items[i].TotalPrice
			if req.Items[i].BatchNo == "" {
				req.Items[i].BatchNo = fmt.Sprintf("%s-%02d", order.OrderNo, i+1)
			}
		}
		updates := map[string]interface{}{
			"warehouse_id": req.WarehouseID,
			"partner_id":   req.PartnerID,
			"order_date":   req.OrderDate,
			"remark":       req.Remark,
			"total_amount": total,
		}
		if err := tx.Model(&order).Updates(updates).Error; err != nil {
			return err
		}
		if len(req.Items) > 0 {
			if err := tx.Create(&req.Items).Error; err != nil {
				return err
			}
		}
		return nil
	})
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{"message": "updated"})
}

// Delete DELETE /api/v1/stock-in/orders/:id （只允许草稿状态）
func (h *StockInHandler) Delete(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	now := timeNow()
	result := h.db.Model(&model.StockInOrder{}).
		Where("id = ? AND shop_id = ? AND status = 'draft' AND deleted_at IS NULL", c.Param("id"), shopID).
		Update("deleted_at", now)
	if result.RowsAffected == 0 {
		c.JSON(http.StatusNotFound, gin.H{"error": "入库单不存在或不可删除"})
		return
	}
	c.JSON(http.StatusOK, gin.H{"message": "deleted"})
}

// Submit PUT /api/v1/stock-in/orders/:id/submit （草稿→待审核）
func (h *StockInHandler) Submit(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	result := h.db.Model(&model.StockInOrder{}).
		Where("id = ? AND shop_id = ? AND status = 'draft'", c.Param("id"), shopID).
		Update("status", "pending")
	if result.RowsAffected == 0 {
		c.JSON(http.StatusBadRequest, gin.H{"error": "order not found or not in draft status"})
		return
	}
	c.JSON(http.StatusOK, gin.H{"message": "submitted"})
}

// Approve PUT /api/v1/stock-in/orders/:id/approve
func (h *StockInHandler) Approve(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	reviewerID := middleware.GetUserID(c)

	id, _ := strconv.ParseUint(c.Param("id"), 10, 64)
	if err := h.stockSvc.ApproveStockIn(shopID, id, reviewerID); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{"message": "approved"})
}

// Reject PUT /api/v1/stock-in/orders/:id/reject
func (h *StockInHandler) Reject(c *gin.Context) {
	shopID := middleware.GetShopID(c)
	reviewerID := middleware.GetUserID(c)
	now := timeNow()

	result := h.db.Model(&model.StockInOrder{}).
		Where("id = ? AND shop_id = ? AND status = 'pending'", c.Param("id"), shopID).
		Updates(map[string]interface{}{
			"status":      "rejected",
			"reviewer_id": reviewerID,
			"reviewed_at": now,
		})
	if result.RowsAffected == 0 {
		c.JSON(http.StatusBadRequest, gin.H{"error": "order not found or not in pending status"})
		return
	}
	c.JSON(http.StatusOK, gin.H{"message": "rejected"})
}