add购买记录功能

internal/infrastructure/database/repositories/product/gorm_component_report_repository.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"errors"
+	"time"
 
 	"tyapi-server/internal/domains/product/entities"
 	"tyapi-server/internal/domains/product/repositories"
@@ -29,12 +30,8 @@ func NewGormComponentReportRepository(db *gorm.DB, logger *zap.Logger) repositor
 	}
 }
 
-func (r *GormComponentReportRepository) CreateDownload(ctx context.Context, download *entities.ComponentReportDownload) (*entities.ComponentReportDownload, error) {
-	err := r.CreateEntity(ctx, download)
-	if err != nil {
-		return nil, err
-	}
-	return download, nil
+func (r *GormComponentReportRepository) Create(ctx context.Context, download *entities.ComponentReportDownload) error {
+	return r.CreateEntity(ctx, download)
 }
 
 func (r *GormComponentReportRepository) UpdateDownload(ctx context.Context, download *entities.ComponentReportDownload) error {
@@ -55,7 +52,7 @@ func (r *GormComponentReportRepository) GetDownloadByID(ctx context.Context, id
 
 func (r *GormComponentReportRepository) GetUserDownloads(ctx context.Context, userID string, productID *string) ([]*entities.ComponentReportDownload, error) {
 	var downloads []entities.ComponentReportDownload
-	query := r.GetDB(ctx).Where("user_id = ? AND payment_status = ?", userID, "success")
+	query := r.GetDB(ctx).Where("user_id = ?", userID)
 
 	if productID != nil && *productID != "" {
 		query = query.Where("product_id = ?", *productID)
@@ -76,7 +73,7 @@ func (r *GormComponentReportRepository) GetUserDownloads(ctx context.Context, us
 func (r *GormComponentReportRepository) HasUserDownloaded(ctx context.Context, userID string, productCode string) (bool, error) {
 	var count int64
 	err := r.GetDB(ctx).Model(&entities.ComponentReportDownload{}).
-		Where("user_id = ? AND product_code = ? AND payment_status = ?", userID, productCode, "success").
+		Where("user_id = ? AND product_code = ?", userID, productCode).
 		Count(&count).Error
 	if err != nil {
 		return false, err
@@ -88,7 +85,7 @@ func (r *GormComponentReportRepository) GetUserDownloadedProductCodes(ctx contex
 	var downloads []entities.ComponentReportDownload
 	err := r.GetDB(ctx).
 		Select("DISTINCT sub_product_codes").
-		Where("user_id = ? AND payment_status = ?", userID, "success").
+		Where("user_id = ?", userID).
 		Find(&downloads).Error
 	if err != nil {
 		return nil, err
@@ -119,7 +116,7 @@ func (r *GormComponentReportRepository) GetUserDownloadedProductCodes(ctx contex
 
 func (r *GormComponentReportRepository) GetDownloadByPaymentOrderID(ctx context.Context, orderID string) (*entities.ComponentReportDownload, error) {
 	var download entities.ComponentReportDownload
-	err := r.GetDB(ctx).Where("payment_order_id = ?", orderID).First(&download).Error
+	err := r.GetDB(ctx).Where("order_number = ?", orderID).First(&download).Error
 	if err != nil {
 		if errors.Is(err, gorm.ErrRecordNotFound) {
 			return nil, gorm.ErrRecordNotFound
@@ -128,3 +125,65 @@ func (r *GormComponentReportRepository) GetDownloadByPaymentOrderID(ctx context.
 	}
 	return &download, nil
 }
+
+// GetActiveDownload 获取用户有效的下载记录
+func (r *GormComponentReportRepository) GetActiveDownload(ctx context.Context, userID, productID string) (*entities.ComponentReportDownload, error) {
+	var download entities.ComponentReportDownload
+
+	// 先尝试查找有支付订单号的下载记录（已支付）
+	err := r.GetDB(ctx).
+		Where("user_id = ? AND product_id = ? AND order_number IS NOT NULL AND deleted_at IS NULL", userID, productID).
+		Order("created_at DESC").
+		First(&download).Error
+
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			// 如果没有找到有支付订单号的记录，尝试查找任何有效的下载记录
+			err = r.GetDB(ctx).
+				Where("user_id = ? AND product_id = ? AND deleted_at IS NULL", userID, productID).
+				Order("created_at DESC").
+				First(&download).Error
+
+			if err != nil {
+				if errors.Is(err, gorm.ErrRecordNotFound) {
+					return nil, nil
+				}
+				return nil, err
+			}
+		} else {
+			return nil, err
+		}
+	}
+
+	// 如果找到了下载记录，检查关联的购买订单状态
+	if download.OrderID != nil {
+		// 这里需要查询购买订单状态，但当前仓库没有依赖购买订单仓库
+		// 所以只检查是否有过期时间设置，如果有则认为已支付
+		if download.ExpiresAt == nil {
+			return nil, nil // 没有过期时间，表示未支付
+		}
+	}
+
+	// 检查是否已过期
+	if download.IsExpired() {
+		return nil, nil
+	}
+
+	return &download, nil
+}
+
+// UpdateFilePath 更新下载记录文件路径
+func (r *GormComponentReportRepository) UpdateFilePath(ctx context.Context, downloadID, filePath string) error {
+	return r.GetDB(ctx).Model(&entities.ComponentReportDownload{}).Where("id = ?", downloadID).Update("file_path", filePath).Error
+}
+
+// IncrementDownloadCount 增加下载次数
+func (r *GormComponentReportRepository) IncrementDownloadCount(ctx context.Context, downloadID string) error {
+	now := time.Now()
+	return r.GetDB(ctx).Model(&entities.ComponentReportDownload{}).
+		Where("id = ?", downloadID).
+		Updates(map[string]interface{}{
+			"download_count":   gorm.Expr("download_count + 1"),
+			"last_download_at": &now,
+		}).Error
+}