package entitlement import ( entitlementservice "backend/internal/service/entitlement" "backend/utils" "net/http" "strconv" "github.com/gin-gonic/gin" "gorm.io/gorm" ) // EntitlementController 客户权益控制器 // Logan.Li type EntitlementController struct { svc *entitlementservice.EntitlementService db *gorm.DB } // NewEntitlementController 创建控制器实例 // Logan.Li func NewEntitlementController(db *gorm.DB) *EntitlementController { return &EntitlementController{svc: entitlementservice.NewEntitlementService(db), db: db} } // List 权益列表 // Logan.Li func (ctrl *EntitlementController) List(c *gin.Context) { page, _ := strconv.Atoi(c.DefaultQuery("page", "1")) pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "10")) customerID, _ := strconv.ParseUint(c.Query("customer_id"), 10, 32) productID, _ := strconv.ParseUint(c.Query("product_id"), 10, 32) entitlements, total, err := ctrl.svc.List(page, pageSize, uint(customerID), uint(productID)) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "data": gin.H{ "list": entitlements, "total": total, "page": page, "page_size": pageSize, "total_pages": (total + int64(pageSize) - 1) / int64(pageSize), }}) } // Get 权益详情 // Logan.Li func (ctrl *EntitlementController) Get(c *gin.Context) { id, _ := strconv.ParseUint(c.Param("id"), 10, 32) ent, err := ctrl.svc.GetByID(uint(id)) if err != nil { c.JSON(http.StatusOK, gin.H{"code": 404, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "data": ent}) } // Revoke 吊销权益 // Logan.Li func (ctrl *EntitlementController) Revoke(c *gin.Context) { id, _ := strconv.ParseUint(c.Param("id"), 10, 32) if err := ctrl.svc.Revoke(uint(id)); err != nil { c.JSON(http.StatusOK, gin.H{"code": 1, "message": err.Error()}) return } utils.LogAction(ctrl.db, c, "revoke", "entitlement", uint(id), "", "", "", "吊销客户权益") c.JSON(http.StatusOK, gin.H{"code": 0, "message": "权益已吊销"}) } // Renew 续费权益 // Logan.Li func (ctrl *EntitlementController) Renew(c *gin.Context) { id, _ := strconv.ParseUint(c.Param("id"), 10, 32) var req struct { PlanID uint `json:"plan_id" binding:"required"` } if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"}) return } ent, err := ctrl.svc.Renew(uint(id), req.PlanID) if err != nil { c.JSON(http.StatusOK, gin.H{"code": 1, "message": err.Error()}) return } utils.LogAction(ctrl.db, c, "renew", "entitlement", ent.ID, "", "", "", "续费客户权益") c.JSON(http.StatusOK, gin.H{"code": 0, "message": "续费成功", "data": ent}) } // GetByEmail 按邮箱查询客户权益(客户端接口) // Logan.Li func (ctrl *EntitlementController) GetByEmail(c *gin.Context) { email := c.Query("email") if email == "" { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "缺少 email 参数"}) return } entitlements, err := ctrl.svc.GetByEmail(email) if err != nil { c.JSON(http.StatusOK, gin.H{"code": 1, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "data": entitlements}) }