拆分大型 Repository 文件职责

2026-06-10 14:28:16 +08:00
parent 6ae8f0e830
commit 60ff513463
23 changed files with 3977 additions and 3906 deletions
@@ -0,0 +1,314 @@
+package listing
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"strings"
+	"time"
+
+	"hfb_sys/backend/internal/model"
+
+	"gorm.io/datatypes"
+	"gorm.io/gorm"
+)
+
+func (r *Repository) Create(ctx context.Context, ownerID uint64, req CreateRequest, reviewRequired bool) (*ListingDTO, error) {
+	var dto *ListingDTO
+	err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		listingNo, err := r.nextListingNo(tx, time.Now())
+		if err != nil {
+			return err
+		}
+		screenshots, err := marshalScreenshots(req.ScreenshotURLS)
+		if err != nil {
+			return err
+		}
+		assetSummary, err := marshalAssetSummary(req.AssetSummary)
+		if err != nil {
+			return err
+		}
+		listingStatus, reviewStatus, publishedAt := initialPublishState(reviewRequired)
+		account := model.GameAccount{
+			OwnerID:        ownerID,
+			GameName:       "delta_force",
+			ServerRegion:   req.ServerRegion,
+			LoginPlatform:  req.LoginPlatform,
+			Title:          req.Title,
+			Description:    req.Description,
+			RankLevel:      req.RankLevel,
+			HafCoinAmount:  req.HafCoinAmount,
+			AssetSummary:   assetSummary,
+			ScreenshotURLS: screenshots,
+			Status:         listingStatus,
+		}
+		if err := tx.Create(&account).Error; err != nil {
+			return err
+		}
+		priceCent := normalizedListingPriceCent(req)
+		listing := model.RentalListing{
+			ListingNo:         listingNo,
+			AccountID:         account.ID,
+			OwnerID:           ownerID,
+			PriceCent:         priceCent,
+			DepositAmountCent: req.DepositAmountCent,
+			Status:            listingStatus,
+			ReviewStatus:      reviewStatus,
+			PublishedAt:       publishedAt,
+		}
+		if err := tx.Create(&listing).Error; err != nil {
+			return err
+		}
+		dto = toDTO(account, listing)
+		return nil
+	})
+	return dto, err
+}
+
+type externalUploadCreate struct {
+	UploaderName     string
+	ClientUploadTime *time.Time
+	ClientIP         string
+	RawPayload       []byte
+	ParsedPayload    []byte
+}
+
+func (r *Repository) CreateFromExternalUpload(ctx context.Context, upload externalUploadCreate, req CreateRequest) (*ListingDTO, error) {
+	var dto *ListingDTO
+	err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		listingNo, err := r.nextListingNo(tx, time.Now())
+		if err != nil {
+			return err
+		}
+		admin, err := r.findActiveUploadAdmin(tx, upload.UploaderName)
+		if err != nil {
+			return err
+		}
+		owner, err := r.ensureUploadOwnerUser(tx, admin)
+		if err != nil {
+			return err
+		}
+		screenshots, err := marshalScreenshots(req.ScreenshotURLS)
+		if err != nil {
+			return err
+		}
+		assetSummary, err := marshalAssetSummary(req.AssetSummary)
+		if err != nil {
+			return err
+		}
+		account := model.GameAccount{
+			OwnerID:        owner.ID,
+			GameName:       "delta_force",
+			ServerRegion:   req.ServerRegion,
+			LoginPlatform:  req.LoginPlatform,
+			Title:          req.Title,
+			Description:    req.Description,
+			RankLevel:      req.RankLevel,
+			HafCoinAmount:  req.HafCoinAmount,
+			AssetSummary:   assetSummary,
+			ScreenshotURLS: screenshots,
+			Status:         "draft",
+		}
+		if err := tx.Create(&account).Error; err != nil {
+			return err
+		}
+		priceCent := normalizedListingPriceCent(req)
+		listing := model.RentalListing{
+			ListingNo:         listingNo,
+			AccountID:         account.ID,
+			OwnerID:           owner.ID,
+			PriceCent:         priceCent,
+			DepositAmountCent: req.DepositAmountCent,
+			Status:            "draft",
+			ReviewStatus:      "pending",
+		}
+		if err := tx.Create(&listing).Error; err != nil {
+			return err
+		}
+		matchedAdminID := admin.ID
+		ownerID := owner.ID
+		listingID := listing.ID
+		uploadRow := model.ListingUpload{
+			UploaderName:     upload.UploaderName,
+			MatchedAdminID:   &matchedAdminID,
+			OwnerID:          &ownerID,
+			ClientUploadTime: upload.ClientUploadTime,
+			ClientIP:         upload.ClientIP,
+			RawPayload:       datatypes.JSON(upload.RawPayload),
+			ParsedPayload:    datatypes.JSON(upload.ParsedPayload),
+			ListingID:        &listingID,
+			Status:           "draft_created",
+		}
+		if err := tx.Create(&uploadRow).Error; err != nil {
+			return err
+		}
+		dto = toDTO(account, listing)
+		return nil
+	})
+	return dto, err
+}
+
+func (r *Repository) Update(ctx context.Context, ownerID uint64, listingID uint64, req UpdateRequest, reviewRequired bool) (*ListingDTO, error) {
+	var dto *ListingDTO
+	err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		listing, account, err := r.findOwnedForUpdate(tx, ownerID, listingID)
+		if err != nil {
+			return err
+		}
+		if listing.Status == "rented" || listing.InTransaction {
+			return ErrListingLocked
+		}
+
+		account.Title = req.Title
+		account.Description = req.Description
+		account.ServerRegion = req.ServerRegion
+		account.LoginPlatform = req.LoginPlatform
+		account.RankLevel = req.RankLevel
+		account.HafCoinAmount = req.HafCoinAmount
+		assetSummary, err := marshalAssetSummary(req.AssetSummary)
+		if err != nil {
+			return err
+		}
+		account.AssetSummary = assetSummary
+		screenshots, err := marshalScreenshots(req.ScreenshotURLS)
+		if err != nil {
+			return err
+		}
+		account.ScreenshotURLS = screenshots
+
+		listing.PriceCent = normalizedListingPriceCent(req)
+		listing.DepositAmountCent = req.DepositAmountCent
+		listingStatus, reviewStatus, publishedAt := initialPublishState(reviewRequired)
+		listing.Status = listingStatus
+		listing.ReviewStatus = reviewStatus
+		listing.ReviewReason = ""
+		listing.PublishedAt = publishedAt
+		account.Status = listingStatus
+		if err := tx.Save(account).Error; err != nil {
+			return err
+		}
+		if err := tx.Save(listing).Error; err != nil {
+			return err
+		}
+		dto = toDTO(*account, *listing)
+		return nil
+	})
+	return dto, err
+}
+
+func (r *Repository) SubmitReview(ctx context.Context, ownerID uint64, listingID uint64, reviewRequired bool) (*ListingDTO, error) {
+	var dto *ListingDTO
+	err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		listing, account, err := r.findOwnedForUpdate(tx, ownerID, listingID)
+		if err != nil {
+			return err
+		}
+		if listing.Status == "rented" || listing.InTransaction {
+			return ErrListingLocked
+		}
+		listingStatus, reviewStatus, publishedAt := initialPublishState(reviewRequired)
+		listing.Status = listingStatus
+		listing.ReviewStatus = reviewStatus
+		listing.ReviewReason = ""
+		listing.PublishedAt = publishedAt
+		account.Status = listingStatus
+		if err := tx.Save(account).Error; err != nil {
+			return err
+		}
+		if err := tx.Save(listing).Error; err != nil {
+			return err
+		}
+		dto = toDTO(*account, *listing)
+		return nil
+	})
+	return dto, err
+}
+
+func (r *Repository) findActiveUploadAdmin(tx *gorm.DB, uploaderName string) (*model.AdminUser, error) {
+	uploaderName = strings.TrimSpace(uploaderName)
+	var admin model.AdminUser
+	if err := tx.Where("username = ? AND status = ?", uploaderName, "active").First(&admin).Error; err == nil {
+		return &admin, nil
+	} else if !errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+
+	var admins []model.AdminUser
+	if err := tx.Where("nickname = ? AND status = ?", uploaderName, "active").Limit(2).Find(&admins).Error; err != nil {
+		return nil, err
+	}
+	switch len(admins) {
+	case 0:
+		return nil, ErrUploaderNotFound
+	case 1:
+		return &admins[0], nil
+	default:
+		return nil, ErrUploaderAmbiguous
+	}
+}
+
+func (r *Repository) ensureUploadOwnerUser(tx *gorm.DB, admin *model.AdminUser) (*model.User, error) {
+	phone := fmt.Sprintf("admin:%d", admin.ID)
+	nickname := strings.TrimSpace(admin.Nickname)
+	if nickname == "" {
+		nickname = admin.Username
+	}
+	var user model.User
+	err := tx.Where("phone = ?", phone).First(&user).Error
+	if err == nil {
+		updates := map[string]any{
+			"nickname":        nickname,
+			"status":          "active",
+			"realname_status": "verified",
+		}
+		if err := tx.Model(&user).Updates(updates).Error; err != nil {
+			return nil, err
+		}
+		user.Nickname = nickname
+		user.Status = "active"
+		user.RealnameStatus = "verified"
+		return &user, nil
+	}
+	if !errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+	user = model.User{
+		Phone:          phone,
+		Nickname:       nickname,
+		RealnameStatus: "verified",
+		RiskStatus:     "normal",
+		CreditScore:    100,
+		Status:         "active",
+	}
+	if err := tx.Create(&user).Error; err != nil {
+		return nil, err
+	}
+	return &user, nil
+}
+
+func (r *Repository) Offline(ctx context.Context, ownerID uint64, listingID uint64) (*ListingDTO, error) {
+	var dto *ListingDTO
+	err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		listing, account, err := r.findOwnedForUpdate(tx, ownerID, listingID)
+		if err != nil {
+			return err
+		}
+		if listing.Status == "rented" || listing.InTransaction {
+			return ErrListingLocked
+		}
+		listing.Status = "offline"
+		listing.ReviewStatus = "none"
+		listing.ReviewReason = "号主已手动下架"
+		listing.PublishedAt = nil
+		account.Status = "offline"
+		if err := tx.Save(account).Error; err != nil {
+			return err
+		}
+		if err := tx.Save(listing).Error; err != nil {
+			return err
+		}
+		dto = toDTO(*account, *listing)
+		return nil
+	})
+	return dto, err
+}