package order import ( "backend/internal/dto" orderservice "backend/internal/service/order" "backend/utils" "encoding/json" "io" "net/http" "strconv" "strings" "github.com/gin-gonic/gin" "gorm.io/gorm" ) // OrderController 订单控制器 // Logan.Li type OrderController struct { svc *orderservice.OrderService db *gorm.DB } // NewOrderController 创建控制器实例 // Logan.Li func NewOrderController(db *gorm.DB) *OrderController { return &OrderController{svc: orderservice.NewOrderService(db), db: db} } // List 订单列表(管理后台) // Logan.Li func (ctrl *OrderController) List(c *gin.Context) { page, _ := strconv.Atoi(c.DefaultQuery("page", "1")) pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "10")) status := c.Query("status") productID, _ := strconv.ParseUint(c.Query("product_id"), 10, 32) orders, total, err := ctrl.svc.List(page, pageSize, status, 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": orders, "total": total, "page": page, "page_size": pageSize, "total_pages": (total + int64(pageSize) - 1) / int64(pageSize), }}) } // GetDetail 订单详情 // Logan.Li func (ctrl *OrderController) GetDetail(c *gin.Context) { id, _ := strconv.ParseUint(c.Param("id"), 10, 32) order, 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": order}) } // CreateOrder 客户端下单(公开接口) // Logan.Li func (ctrl *OrderController) CreateOrder(c *gin.Context) { var req dto.CreateOrderRequest if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误: " + err.Error()}) return } resp, err := ctrl.svc.CreateOrder(req, utils.GetClientIP(c), c.GetHeader("User-Agent")) if err != nil { c.JSON(http.StatusOK, gin.H{"code": 2001, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "message": "订单创建成功", "data": resp}) } // Checkout desktop-friendly checkout entry. // Logan.Li func (ctrl *OrderController) Checkout(c *gin.Context) { var req dto.CheckoutRequest if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误: " + err.Error()}) return } resp, err := ctrl.svc.Checkout(req, utils.GetClientIP(c), c.GetHeader("User-Agent")) if err != nil { c.JSON(http.StatusOK, gin.H{"code": 2001, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "message": "订单创建成功", "data": resp}) } // ClaimPurchase fetch license after payment. // Logan.Li func (ctrl *OrderController) ClaimPurchase(c *gin.Context) { var req dto.ClaimPurchaseRequest if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误: " + err.Error()}) return } resp, err := ctrl.svc.ClaimPurchase(req) if err != nil { c.JSON(http.StatusOK, gin.H{"code": 2002, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "data": resp}) } // PaymentWebhook 支付回调(公开接口,供支付渠道调用) // Logan.Li func (ctrl *OrderController) PaymentWebhook(c *gin.Context) { var req dto.PaymentWebhookRequest if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "Invalid payload"}) return } if err := ctrl.svc.PaymentWebhook(req, utils.GetClientIP(c)); err != nil { c.JSON(http.StatusOK, gin.H{"code": 1, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "message": "Webhook processed"}) } // WaffoWebhook accepts loosely-shaped Waffo payment events and maps to PaymentWebhook. // Logan.Li func (ctrl *OrderController) WaffoWebhook(c *gin.Context) { body, err := io.ReadAll(c.Request.Body) if err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "无法读取请求体"}) return } // Optional: accept X-Waffo-Signature if present (no strict verification yet). // Logan.Li _ = c.GetHeader("X-Waffo-Signature") var payload map[string]interface{} if len(body) > 0 { if err := json.Unmarshal(body, &payload); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "Invalid JSON"}) return } } else { payload = map[string]interface{}{} } orderNo := firstNonEmpty( asString(payload["order_no"]), asString(payload["orderNo"]), asString(payload["order_id"]), c.Query("order_no"), c.Query("orderNo"), ) if orderNo == "" { if meta, ok := payload["metadata"].(map[string]interface{}); ok { orderNo = firstNonEmpty(asString(meta["orderNo"]), asString(meta["order_no"])) } } if orderNo == "" { if data, ok := payload["data"].(map[string]interface{}); ok { orderNo = firstNonEmpty(asString(data["order_no"]), asString(data["orderNo"])) if orderNo == "" { if meta, ok := data["metadata"].(map[string]interface{}); ok { orderNo = firstNonEmpty(asString(meta["orderNo"]), asString(meta["order_no"])) } } } } if orderNo == "" { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "missing order_no"}) return } eventType := firstNonEmpty( asString(payload["event_type"]), asString(payload["eventType"]), asString(payload["type"]), asString(payload["event"]), "payment", ) statusRaw := strings.ToLower(firstNonEmpty( asString(payload["status"]), asString(payload["payment_status"]), asString(payload["paymentStatus"]), )) gatewayOrderNo := firstNonEmpty( asString(payload["gateway_order_no"]), asString(payload["gatewayOrderNo"]), asString(payload["transaction_id"]), asString(payload["transactionId"]), asString(payload["id"]), ) mappedStatus := "paid" switch { case strings.Contains(statusRaw, "fail"), strings.Contains(statusRaw, "cancel"), strings.Contains(eventType, "fail"), strings.Contains(eventType, "cancel"): mappedStatus = "failed" case strings.Contains(statusRaw, "refund"), strings.Contains(eventType, "refund"): mappedStatus = "refunded" case statusRaw == "" || strings.Contains(statusRaw, "paid") || strings.Contains(statusRaw, "complete") || strings.Contains(statusRaw, "success") || strings.Contains(eventType, "complete") || strings.Contains(eventType, "paid") || strings.Contains(eventType, "success"): mappedStatus = "paid" default: // Treat unknown completed-like events as paid for now. // Logan.Li mappedStatus = "paid" } req := dto.PaymentWebhookRequest{ EventType: eventType, OrderNo: orderNo, GatewayOrderNo: gatewayOrderNo, Status: mappedStatus, Gateway: "waffo", Payload: string(body), Signature: c.GetHeader("X-Waffo-Signature"), } if err := ctrl.svc.PaymentWebhook(req, utils.GetClientIP(c)); err != nil { c.JSON(http.StatusOK, gin.H{"code": 1, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "message": "Webhook processed"}) } func asString(v interface{}) string { switch t := v.(type) { case string: return strings.TrimSpace(t) case float64: if t == float64(int64(t)) { return strconv.FormatInt(int64(t), 10) } return strconv.FormatFloat(t, 'f', -1, 64) case json.Number: return t.String() default: return "" } } func firstNonEmpty(values ...string) string { for _, v := range values { if strings.TrimSpace(v) != "" { return strings.TrimSpace(v) } } return "" } // ManualGrant 手动发放 License(管理后台) // Logan.Li func (ctrl *OrderController) ManualGrant(c *gin.Context) { var req dto.ManualGrantRequest if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误: " + err.Error()}) return } userID, _ := c.Get("user_id") username, _ := c.Get("username") data, err := ctrl.svc.ManualGrant(req, userID.(uint), username.(string), utils.GetClientIP(c)) if err != nil { c.JSON(http.StatusOK, gin.H{"code": 1, "message": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"code": 0, "message": "License 发放成功", "data": data}) } // CheckDuplicatePurchase 防重复购买检测(公开接口) // Logan.Li func (ctrl *OrderController) CheckDuplicatePurchase(c *gin.Context) { var req dto.DuplicatePurchaseCheckRequest if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"}) return } // 通过 CheckPurchaseStatus 间接判断:有记录且有效则不允许重复购买 // Logan.Li status, _ := ctrl.svc.CheckPurchaseStatus(req.CustomerEmail, req.ProductID) result := dto.DuplicatePurchaseCheckResponse{AllowPurchase: true} if status.HasActiveLicense { result.AllowPurchase = false result.Reason = status.Message if result.Reason == "" { result.Reason = "您已拥有该产品的有效授权" } result.ExistingLicenseKey = status.LicenseKey result.ExistingPlanType = status.PlanType result.CanUpgrade = true } c.JSON(http.StatusOK, gin.H{"code": 0, "data": result}) } // CheckPurchaseStatus 查询购买状态(公开接口) // Logan.Li func (ctrl *OrderController) CheckPurchaseStatus(c *gin.Context) { var req dto.CheckPurchaseStatusRequest if err := c.ShouldBindJSON(&req); err != nil { c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "参数错误"}) return } resp, _ := ctrl.svc.CheckPurchaseStatus(req.CustomerEmail, req.ProductID) c.JSON(http.StatusOK, gin.H{"code": 0, "data": resp}) }