add购买记录功能

internal/shared/component_report/check_payment_status_fix.go

@@ -0,0 +1,102 @@
+package component_report
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+	"go.uber.org/zap"
+
+	finance_entities "tyapi-server/internal/domains/finance/entities"
+)
+
+// CheckPaymentStatusFixed 修复版检查支付状态方法
+func (h *ComponentReportHandler) CheckPaymentStatusFixed(c *gin.Context) {
+	userID := c.GetString("user_id")
+	if userID == "" {
+		c.JSON(http.StatusUnauthorized, gin.H{
+			"code":    401,
+			"message": "用户未登录",
+		})
+		return
+	}
+
+	orderID := c.Param("orderId")
+	if orderID == "" {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"code":    400,
+			"message": "订单ID不能为空",
+		})
+		return
+	}
+
+	// 根据订单ID查询下载记录
+	download, err := h.componentReportRepo.GetDownloadByID(c.Request.Context(), orderID)
+	if err != nil {
+		h.logger.Error("查询下载记录失败", zap.Error(err), zap.String("order_id", orderID))
+		c.JSON(http.StatusNotFound, gin.H{
+			"code":    404,
+			"message": "订单不存在",
+		})
+		return
+	}
+
+	// 验证订单是否属于当前用户
+	if download.UserID != userID {
+		c.JSON(http.StatusForbidden, gin.H{
+			"code":    403,
+			"message": "无权访问此订单",
+		})
+		return
+	}
+
+	// 使用购买订单状态来判断支付状态
+	var paymentStatus string
+	var canDownload bool
+
+	// 优先使用OrderID查询购买订单状态
+	if download.OrderID != nil {
+		// 查询购买订单状态
+		purchaseOrder, err := h.purchaseOrderRepo.GetByID(c.Request.Context(), *download.OrderID)
+		if err != nil {
+			h.logger.Error("查询购买订单失败", zap.Error(err), zap.String("order_id", *download.OrderID))
+			paymentStatus = "unknown"
+		} else {
+			// 根据购买订单状态设置支付状态
+			switch purchaseOrder.Status {
+			case finance_entities.PurchaseOrderStatusPaid:
+				paymentStatus = "success"
+				canDownload = true
+			case finance_entities.PurchaseOrderStatusCreated:
+				paymentStatus = "pending"
+				canDownload = false
+			case finance_entities.PurchaseOrderStatusCancelled:
+				paymentStatus = "cancelled"
+				canDownload = false
+			case finance_entities.PurchaseOrderStatusFailed:
+				paymentStatus = "failed"
+				canDownload = false
+			default:
+				paymentStatus = "unknown"
+				canDownload = false
+			}
+		}
+	} else if download.OrderNumber != nil {
+		// 兼容旧的支付订单逻辑
+		paymentStatus = "success" // 简化处理，有支付订单号就认为已支付
+		canDownload = true
+	} else {
+		paymentStatus = "pending"
+		canDownload = false
+	}
+
+	// 检查是否过期
+	if download.IsExpired() {
+		canDownload = false
+	}
+
+	c.JSON(http.StatusOK, CheckPaymentStatusResponse{
+		OrderID:       download.ID,
+		PaymentStatus: paymentStatus,
+		CanDownload:   canDownload,
+	})
+}