v1.1

app/main/api/internal/logic/agent/getinvitelinklogic.go

@@ -4,18 +4,22 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"strconv"
+	"strings"
 	"time"
 	"ycc-server/app/main/model"
 	"ycc-server/common/ctxdata"
 	"ycc-server/common/globalkey"
 	"ycc-server/common/tool"
 	"ycc-server/common/xerr"
+	"ycc-server/pkg/lzkit/crypto"
 
 	"github.com/pkg/errors"
 
 	"ycc-server/app/main/api/internal/svc"
 	"ycc-server/app/main/api/internal/types"
 
+	"github.com/google/uuid"
 	"github.com/zeromicro/go-zero/core/logx"
 )
 
@@ -53,41 +57,51 @@ func (l *GetInviteLinkLogic) GetInviteLink(req *types.GetInviteLinkReq) (resp *t
 		return nil, errors.Wrapf(xerr.NewErrMsg("邀请码不能为空"), "")
 	}
 
-	// 3. 查询邀请码是否存在且属于当前代理
-	inviteCodeRecord, err := l.svcCtx.AgentInviteCodeModel.FindOneByCode(l.ctx, req.InviteCode)
-	if err != nil {
-		if errors.Is(err, model.ErrNotFound) {
-			return nil, errors.Wrapf(xerr.NewErrMsg("邀请码不存在"), "")
-		}
+	ref := strings.TrimSpace(req.InviteCode)
+	var inviteCodeRecord *model.AgentInviteCode
+	inviteCodeRecord, err = l.svcCtx.AgentInviteCodeModel.FindOneByCode(l.ctx, ref)
+	if err != nil && !errors.Is(err, model.ErrNotFound) {
 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码失败, %v", err)
 	}
 
-	// 4. 验证邀请码是否属于当前代理
-	if !inviteCodeRecord.AgentId.Valid || inviteCodeRecord.AgentId.Int64 != agent.Id {
-		return nil, errors.Wrapf(xerr.NewErrMsg("无权使用此邀请码"), "")
-	}
-
-	// 5. 验证邀请码状态
-	if inviteCodeRecord.Status != 0 {
-		if inviteCodeRecord.Status == 1 {
-			return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已使用"), "")
+	if inviteCodeRecord != nil {
+		if !inviteCodeRecord.AgentId.Valid || inviteCodeRecord.AgentId.String != agent.Id {
+			return nil, errors.Wrapf(xerr.NewErrMsg("无权使用此邀请码"), "")
+		}
+		if inviteCodeRecord.Status != 0 {
+			if inviteCodeRecord.Status == 1 {
+				return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已使用"), "")
+			}
+			return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已失效"), "")
+		}
+		if inviteCodeRecord.ExpireTime.Valid && inviteCodeRecord.ExpireTime.Time.Before(time.Now()) {
+			return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已过期"), "")
+		}
+	} else {
+		if codeVal, parseErr := strconv.ParseInt(ref, 10, 64); parseErr == nil && codeVal > 0 {
+			if agent.AgentCode != codeVal {
+				return nil, errors.Wrapf(xerr.NewErrMsg("无权使用此邀请码"), "")
+			}
+		} else {
+			encMobile, _ := crypto.EncryptMobile(ref, l.svcCtx.Config.Encrypt.SecretKey)
+			if encMobile != agent.Mobile {
+				return nil, errors.Wrapf(xerr.NewErrMsg("邀请信息无效"), "")
+			}
 		}
-		return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已失效"), "")
-	}
-
-	// 6. 验证邀请码是否过期
-	if inviteCodeRecord.ExpireTime.Valid && inviteCodeRecord.ExpireTime.Time.Before(time.Now()) {
-		return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已过期"), "")
 	}
 
 	// 7. 使用默认target_path（如果未提供）
 	targetPath := req.TargetPath
 	if targetPath == "" {
-		targetPath = fmt.Sprintf("/register?invite_code=%s", req.InviteCode)
+		targetPath = fmt.Sprintf("/register?invite_code=%s", ref)
 	}
 
-	// 8. 生成短链（类型：2=邀请好友）
-	shortLink, err := l.createInviteShortLink(inviteCodeRecord.Id, req.InviteCode, targetPath)
+	shortLink, err := l.createInviteShortLink(func() string {
+		if inviteCodeRecord != nil {
+			return inviteCodeRecord.Id
+		}
+		return ""
+	}(), ref, targetPath)
 	if err != nil {
 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成短链失败, %v", err)
 	}
@@ -98,7 +112,7 @@ func (l *GetInviteLinkLogic) GetInviteLink(req *types.GetInviteLinkReq) (resp *t
 }
 
 // createInviteShortLink 创建邀请好友短链
-func (l *GetInviteLinkLogic) createInviteShortLink(inviteCodeId int64, inviteCode, targetPath string) (string, error) {
+func (l *GetInviteLinkLogic) createInviteShortLink(inviteCodeId string, inviteCode, targetPath string) (string, error) {
 	promotionConfig := l.svcCtx.Config.Promotion
 
 	// 如果没有配置推广域名，返回空字符串（保持向后兼容）
@@ -108,7 +122,7 @@ func (l *GetInviteLinkLogic) createInviteShortLink(inviteCodeId int64, inviteCod
 	}
 
 	// 先查询是否已存在短链
-	existingShortLink, err := l.svcCtx.AgentShortLinkModel.FindOneByInviteCodeIdTypeDelState(l.ctx, sql.NullInt64{Int64: inviteCodeId, Valid: true}, 2, globalkey.DelStateNo)
+	existingShortLink, err := l.svcCtx.AgentShortLinkModel.FindOneByInviteCodeIdTypeDelState(l.ctx, sql.NullString{String: inviteCodeId, Valid: inviteCodeId != ""}, 2, globalkey.DelStateNo)
 	if err == nil && existingShortLink != nil {
 		// 已存在短链，直接返回
 		return fmt.Sprintf("%s/s/%s", promotionConfig.PromotionDomain, existingShortLink.ShortCode), nil
@@ -118,6 +132,17 @@ func (l *GetInviteLinkLogic) createInviteShortLink(inviteCodeId int64, inviteCod
 		return "", errors.Wrapf(err, "查询短链失败")
 	}
 
+	// 如果没有邀请码ID（例如使用手机号或代理码），按邀请码字符串尝试查找以避免重复创建
+	if inviteCodeId == "" && inviteCode != "" {
+		existingByCode, err2 := l.svcCtx.AgentShortLinkModel.FindOneByInviteCodeTypeDelState(l.ctx, sql.NullString{String: inviteCode, Valid: true}, 2, globalkey.DelStateNo)
+		if err2 == nil && existingByCode != nil {
+			return fmt.Sprintf("%s/s/%s", promotionConfig.PromotionDomain, existingByCode.ShortCode), nil
+		}
+		if err2 != nil && !errors.Is(err2, model.ErrNotFound) {
+			return "", errors.Wrapf(err2, "查询短链失败")
+		}
+	}
+
 	// 生成短链标识（6位随机字符串，大小写字母+数字）
 	var shortCode string
 	maxRetries := 10 // 最大重试次数
@@ -140,8 +165,9 @@ func (l *GetInviteLinkLogic) createInviteShortLink(inviteCodeId int64, inviteCod
 
 	// 创建短链记录（类型：2=邀请好友）
 	shortLink := &model.AgentShortLink{
+		Id:              uuid.NewString(),
 		Type:            2, // 邀请好友
-		InviteCodeId:    sql.NullInt64{Int64: inviteCodeId, Valid: inviteCodeId > 0},
+		InviteCodeId:    sql.NullString{String: inviteCodeId, Valid: inviteCodeId != ""},
 		InviteCode:      sql.NullString{String: inviteCode, Valid: inviteCode != ""},
 		ShortCode:       shortCode,
 		TargetPath:      targetPath,