pkg/service/cast/media.go

@@ -13,16 +13,11 @@ import (
 	"fonchain-fiee/pkg/e"
 	modelCast "fonchain-fiee/pkg/model/cast"
 	"fonchain-fiee/pkg/service"
-	"fonchain-fiee/pkg/utils"
 	"net/http"
 	"net/url"
-	"path/filepath"
 	"strconv"
-	"strings"
-	"time"
 
 	"github.com/gin-gonic/gin"
-	"github.com/xuri/excelize/v2"
 	"go.uber.org/zap"
 )
 
@@ -164,7 +159,7 @@ func UpdateMediaAccount(ctx *gin.Context) {
 		}
 	}
 	// 字符串转整型
-	artistID, _ := strconv.ParseUint(req.ArtistUuid, 10, 64)
+	artistID, err := strconv.ParseUint(req.ArtistUuid, 10, 64)
 	infoResp, err = GetArtistAccountInfo(artistID)
 	if err != nil {
 		service.Error(ctx, err)
@@ -370,7 +365,8 @@ func SyncAsAuth(artistUuid string) error {
 	if err != nil {
 		return errors.New("获取艺人绑定信息错误")
 	}
-	authReq := &cast.UpdateOAuthReq{Data: make([]*cast.UpdateOAuthReq_Info, 0)}
+	var authReq *cast.UpdateOAuthReq
+	authReq = &cast.UpdateOAuthReq{Data: make([]*cast.UpdateOAuthReq_Info, 0)}
 	if len(userResp.DisplayNames) == 0 {
 		//return errors.New("没有授权信息")
 		authReq.Data = append(authReq.Data, &cast.UpdateOAuthReq_Info{
@@ -519,333 +515,3 @@ func SyncAsProfile(ctx *gin.Context) {
 	service.Success(ctx, infoResp)
 	return
 }
-
-// ImportMediaAccount 导入自媒体账号
-func ImportMediaAccount(ctx *gin.Context) {
-	var failedRecords []FailedRecord
-	var successRecords []SuccessRecord
-
-	// 1. 接收上传的Excel文件
-	excelFile, err := ctx.FormFile("file")
-	if err != nil {
-		service.Error(ctx, errors.New("缺少Excel文件"))
-		return
-	}
-
-	// 2. 保存临时文件
-	tempDir := "./runtime/media"
-	_, err = utils.CheckDirPath(tempDir, true)
-	if err != nil {
-		service.Error(ctx, err)
-		return
-	}
-
-	fileName := fmt.Sprintf("%d_media_account.xlsx", time.Now().UnixMicro())
-	excelPath := filepath.Join(tempDir, fileName)
-	if err = ctx.SaveUploadedFile(excelFile, excelPath); err != nil {
-		service.Error(ctx, err)
-		return
-	}
-
-	// 3. 读取Excel数据
-	mediaAccounts, err := readMediaAccountExcel(excelPath)
-	if err != nil {
-		service.Error(ctx, fmt.Errorf("读取Excel失败: %w", err))
-		return
-	}
-
-	if len(mediaAccounts) == 0 {
-		service.Error(ctx, errors.New("Excel中没有有效数据"))
-		return
-	}
-
-	// 4. 处理每个账号
-	for _, account := range mediaAccounts {
-		for platformID, accountInfo := range account.Account {
-			// 跳过空账号信息
-			if accountInfo.AccountId == "" && accountInfo.NickName == "" {
-				continue
-			}
-
-			// 查询用户信息
-			res, err := service.AccountFieeProvider.UserList(context.Background(), &accountFiee.UserListRequest{
-				Name:   account.Name,
-				SubNum: account.SubNum,
-			})
-			if err != nil {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      fmt.Sprintf("获取用户信息失败: %s", err.Error()),
-				})
-				continue
-			}
-
-			if res.Count == 0 {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      "未找到用户信息",
-				})
-				continue
-			}
-
-			// 获取用户详细信息
-			var infoResp *accountFiee.UserInfoResponse
-			infoResp, err = GetArtistAccountInfo(res.UserList[0].Id)
-			if err != nil {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      fmt.Sprintf("查询用户详细信息失败: %s", err.Error()),
-				})
-				continue
-			}
-
-			if infoResp.SubNum == "" {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      "用户不存在",
-				})
-				continue
-			}
-
-			// 检查并创建AS Profile
-			if err = CheckAsProfile(infoResp); err != nil {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      fmt.Sprintf("创建AS Profile失败: %s", err.Error()),
-				})
-				continue
-			}
-
-			// 验证平台ID
-			if _, ok := cast.PlatformIDENUM_name[int32(platformID)]; !ok {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      "无效的平台ID",
-				})
-				continue
-			}
-
-			artistUuid := strconv.FormatUint(res.UserList[0].Id, 10)
-			userID := int32(res.UserList[0].Id)
-
-			// 查询该艺人是否已存在该平台账号
-			userResp, err := service.CastProvider.MediaUserList(context.Background(), &cast.MediaUserListReq{
-				ArtistUuid: artistUuid,
-				Page:       1,
-				PageSize:   100,
-			})
-			if err != nil {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      fmt.Sprintf("查询已有账号失败: %s", err.Error()),
-				})
-				continue
-			}
-
-			// 检查是否已经存在该平台的账号
-			accountExists := false
-			if userResp != nil && len(userResp.Data) > 0 {
-				for _, v := range userResp.Data {
-					if v.PlatformID == uint32(platformID) {
-						accountExists = true
-						break
-					}
-				}
-			}
-
-			if accountExists {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      "该平台账号已存在",
-				})
-				continue
-			}
-
-			// 检查并消耗账户余额
-			if _, err = CheckUserBundleBalance(userID, modelCast.BalanceTypeAccountValue); err != nil {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      fmt.Sprintf("账户余额不足: %s", err.Error()),
-				})
-				continue
-			}
-
-			// 增加账户消耗数量
-			_, err = service.BundleProvider.AddBundleBalance(context.Background(), &bundle.AddBundleBalanceReq{
-				UserId:                   userID,
-				AccountConsumptionNumber: 1,
-			})
-			if err != nil {
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      fmt.Sprintf("扣除账户余额失败: %s", err.Error()),
-				})
-				continue
-			}
-
-			// 创建自媒体账号
-			mediaAccountResp, err := service.CastProvider.UpdateMediaAccount(ctx, &cast.UpdateMediaAccountReq{
-				ArtistUuid:          artistUuid,
-				PlatformID:          platformID,
-				PlatformUserName:    accountInfo.NickName,
-				PlatformUserID:      accountInfo.AccountId,
-				ArtistName:          infoResp.Name,
-				ArtistPhone:         infoResp.TelNum,
-				ArtistPhoneAreaCode: infoResp.TelAreaCode,
-				ArtistSubNum:        infoResp.SubNum,
-			})
-
-			if err != nil {
-				// 创建失败，回退余额
-				_, _ = service.BundleProvider.AddBundleBalance(context.Background(), &bundle.AddBundleBalanceReq{
-					UserId:                   userID,
-					AccountConsumptionNumber: -1,
-				})
-				failedRecords = append(failedRecords, FailedRecord{
-					Name:     account.Name,
-					SubNum:   account.SubNum,
-					Platform: modelCast.PlatformNameKv[uint32(platformID)],
-					Msg:      fmt.Sprintf("创建账号失败: %s", err.Error()),
-				})
-				continue
-			}
-
-			// 记录成功
-			successRecords = append(successRecords, SuccessRecord{
-				MediaAccountUuid: mediaAccountResp.MediaAccountUuid,
-				Name:             account.Name,
-				SubNum:           account.SubNum,
-				Platform:         modelCast.PlatformNameKv[uint32(platformID)],
-				PlatformUserName: accountInfo.NickName,
-				PlatformUserID:   accountInfo.AccountId,
-			})
-		}
-	}
-
-	// 5. 返回结果
-	service.Success(ctx, map[string]interface{}{
-		"successCount":   len(successRecords),
-		"failedCount":    len(failedRecords),
-		"successRecords": successRecords,
-		"failedRecords":  failedRecords,
-	})
-}
-
-// readMediaAccountExcel 读取自媒体账号Excel文件
-func readMediaAccountExcel(excelPath string) ([]MediaAccountImport, error) {
-	f, err := excelize.OpenFile(excelPath)
-	if err != nil {
-		return nil, err
-	}
-	defer f.Close()
-
-	sheetName := f.GetSheetName(0)
-	rows, err := f.GetRows(sheetName)
-	if err != nil {
-		return nil, err
-	}
-
-	var accounts []MediaAccountImport
-	for i, row := range rows {
-		// 跳过表头
-		if i == 0 || len(row) < 3 {
-			continue
-		}
-
-		// 获取基本信息
-		index := strings.TrimSpace(row[0])
-		name := strings.TrimSpace(row[1])
-		subNum := strings.TrimSpace(row[2])
-
-		if name == "" && subNum == "" {
-			continue
-		}
-
-		tmp := MediaAccountImport{
-			Index:   index,
-			Name:    name,
-			SubNum:  subNum,
-			Account: make(map[cast.PlatformIDENUM]AccountInfo),
-		}
-
-		// 读取TikTok账号信息 (列D, E)
-		tiktokId, _ := f.GetCellValue(sheetName, fmt.Sprintf("D%d", i+1))
-		tiktokNickName, _ := f.GetCellValue(sheetName, fmt.Sprintf("E%d", i+1))
-		tmp.Account[cast.PlatformIDENUM_TIKTOK] = AccountInfo{
-			AccountId: strings.TrimSpace(tiktokId),
-			NickName:  strings.TrimSpace(tiktokNickName),
-		}
-
-		// 读取YouTube账号信息 (列F, G)
-		youtubeId, _ := f.GetCellValue(sheetName, fmt.Sprintf("F%d", i+1))
-		youtubeNickName, _ := f.GetCellValue(sheetName, fmt.Sprintf("G%d", i+1))
-		tmp.Account[cast.PlatformIDENUM_YOUTUBE] = AccountInfo{
-			AccountId: strings.TrimSpace(youtubeId),
-			NickName:  strings.TrimSpace(youtubeNickName),
-		}
-
-		// 读取Instagram账号信息 (列H, I)
-		insId, _ := f.GetCellValue(sheetName, fmt.Sprintf("H%d", i+1))
-		insNickName, _ := f.GetCellValue(sheetName, fmt.Sprintf("I%d", i+1))
-		tmp.Account[cast.PlatformIDENUM_INS] = AccountInfo{
-			AccountId: strings.TrimSpace(insId),
-			NickName:  strings.TrimSpace(insNickName),
-		}
-
-		accounts = append(accounts, tmp)
-	}
-
-	return accounts, nil
-}
-
-// MediaAccountImport 自媒体账号导入结构
-type MediaAccountImport struct {
-	Index   string                              `json:"index"`
-	Name    string                              `json:"name"`
-	SubNum  string                              `json:"subNum"`
-	Account map[cast.PlatformIDENUM]AccountInfo `json:"account"`
-}
-
-// AccountInfo 账号信息
-type AccountInfo struct {
-	NickName  string `json:"nickName"`
-	AccountId string `json:"accountId"`
-}
-
-// FailedRecord 失败记录
-type FailedRecord struct {
-	Name     string `json:"name"`
-	SubNum   string `json:"subNum"`
-	Platform string `json:"platform"`
-	Msg      string `json:"msg"`
-}
-
-// SuccessRecord 成功记录
-type SuccessRecord struct {
-	MediaAccountUuid string `json:"mediaAccountUuid"`
-	Name             string `json:"name"`
-	SubNum           string `json:"subNum"`
-	Platform         string `json:"platform"`
-	PlatformUserName string `json:"platformUserName"`
-	PlatformUserID   string `json:"platformUserID"`
-}

pkg/service/cast/work.go

@@ -358,25 +358,25 @@ func CheckUserBundleBalance(userID int32, balanceType modelCast.BalanceTypeEnum)
 	fmt.Println("CheckUserBundleBalance resp", resp)
 	switch balanceType {
 	case modelCast.BalanceTypeAccountValue:
-		if resp.AccountConsumptionNumber >= resp.AccountNumber {
+		if resp.AccountNumber-resp.AccountConsumptionNumber <= 0 {
 			confirmFailType = cast.ConfirmFailENUM_ConfirmFailENUM_NOTENOUGH
-			err = errors.New("该艺人无可用账号数")
+			err = errors.New(e.ErrorBalanceInsufficient)
 			return
 		}
 	case modelCast.BalanceTypeImageValue:
-		if resp.ImageExtendConsumptionNumber >= resp.ImageExtendNumber {
+		if resp.ImageExtendConsumptionNumber-resp.ImageExtendNumber <= 0 {
 			confirmFailType = cast.ConfirmFailENUM_ConfirmFailENUM_NOTENOUGH
 			err = errors.New(e.ErrorBalanceInsufficient)
 			return
 		}
 	case modelCast.BalanceTypeVideoValue:
-		if resp.VideoExtendConsumptionNumber >= resp.VideoExtendNumber {
+		if resp.VideoExtendConsumptionNumber-resp.VideoExtendNumber <= 0 {
 			confirmFailType = cast.ConfirmFailENUM_ConfirmFailENUM_NOTENOUGH
 			err = errors.New(e.ErrorBalanceInsufficient)
 			return
 		}
 	case modelCast.BalanceTypeDataValue:
-		if resp.DataAnalysisExtendConsumptionNumber >= resp.DataAnalysisExtendNumber {
+		if resp.DataAnalysisExtendConsumptionNumber-resp.DataAnalysisExtendNumber <= 0 {
 			confirmFailType = cast.ConfirmFailENUM_ConfirmFailENUM_NOTENOUGH
 			err = errors.New(e.ErrorBalanceInsufficient)
 			return