package agent import ( "context" "database/sql" "fmt" "os" "strconv" "time" "ycc-server/app/main/model" "ycc-server/common/ctxdata" "ycc-server/common/globalkey" "ycc-server/common/xerr" "ycc-server/pkg/lzkit/crypto" "github.com/google/uuid" "github.com/pkg/errors" "github.com/zeromicro/go-zero/core/stores/redis" "github.com/zeromicro/go-zero/core/stores/sqlx" "ycc-server/app/main/api/internal/svc" "ycc-server/app/main/api/internal/types" "github.com/zeromicro/go-zero/core/logx" ) type RegisterByInviteCodeLogic struct { logx.Logger ctx context.Context svcCtx *svc.ServiceContext } func NewRegisterByInviteCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterByInviteCodeLogic { return &RegisterByInviteCodeLogic{ Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx, } } func (l *RegisterByInviteCodeLogic) RegisterByInviteCode(req *types.RegisterByInviteCodeReq) (resp *types.RegisterByInviteCodeResp, err error) { secretKey := l.svcCtx.Config.Encrypt.SecretKey encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey) if err != nil { return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err) } // 校验验证码(开发环境下跳过验证码校验) if os.Getenv("ENV") != "development" { redisKey := fmt.Sprintf("%s:%s", "agentApply", encryptedMobile) cacheCode, err := l.svcCtx.Redis.Get(redisKey) if err != nil { if errors.Is(err, redis.Nil) { return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "") } return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "读取验证码失败, %v", err) } if cacheCode != req.Code { return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "") } } // 获取当前登录态(可能为空) claims, err := ctxdata.GetClaimsFromCtx(l.ctx) if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) { return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败: %v", err) } // 前置检查:如果当前用户是正式用户(有手机号),进行拦截检查 if claims != nil { currentUser, err := l.svcCtx.UserModel.FindOne(l.ctx, claims.UserId) if err != nil && !errors.Is(err, model.ErrNotFound) { return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败: %v", err) } if currentUser != nil && currentUser.Mobile.Valid && currentUser.Mobile.String != "" { // 当前用户是正式用户,检查是否已是代理 agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, claims.UserId) if err != nil && !errors.Is(err, model.ErrNotFound) { return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理失败: %v", err) } if agent != nil { return nil, errors.Wrapf(xerr.NewErrMsg("您已经是代理,不能重复注册"), "") } // 正式用户手机号必须匹配 if currentUser.Mobile.String != encryptedMobile { return nil, errors.Wrapf(xerr.NewErrMsg("请输入当前账号的手机号码"), "") } } } // 验证邀请码是否有效 var inviteCodeModel *model.AgentInviteCode inviteCodeModel, err = l.svcCtx.AgentInviteCodeModel.FindOneByCode(l.ctx, req.Referrer) if err != nil && !errors.Is(err, model.ErrNotFound) { return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码失败, %v", err) } if inviteCodeModel != nil { if inviteCodeModel.Status != 0 { if inviteCodeModel.Status == 1 { return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已使用"), "") } return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已失效"), "") } if inviteCodeModel.ExpireTime.Valid && inviteCodeModel.ExpireTime.Time.Before(time.Now()) { return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已过期"), "") } } // 使用事务处理注册 var userID string var agentID string var agentLevel int64 err = l.svcCtx.AgentInviteCodeModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { // 1. 查找目标用户(通过手机号) targetUser, findUserErr := l.svcCtx.UserModel.FindOneByMobile(transCtx, sql.NullString{String: encryptedMobile, Valid: true}) if findUserErr != nil && !errors.Is(findUserErr, model.ErrNotFound) { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败, %v", findUserErr) } // 2. 获取当前登录态信息 var currentUserID string var currentAuthType string var currentAuthKey string if claims != nil { currentUserID = claims.UserId currentAuthType = claims.AuthType currentAuthKey = claims.AuthKey } // 3. 根据目标用户是否存在,处理用户和认证 if targetUser == nil { // 场景1: 手机号不存在 userID, err = l.handleMobileNotExists(transCtx, session, encryptedMobile, currentUserID) if err != nil { return err } } else { // 场景2: 手机号已存在 userID, err = l.handleMobileExists(transCtx, session, targetUser, currentUserID, currentAuthType, currentAuthKey) if err != nil { return err } } // 4. 处理邀请码和上级关系 var targetLevel int64 var parentAgentId string if inviteCodeModel != nil { targetLevel = inviteCodeModel.TargetLevel if inviteCodeModel.AgentId.Valid { parentAgentId = inviteCodeModel.AgentId.String } } else { if codeVal, parseErr := strconv.ParseInt(req.Referrer, 10, 64); parseErr == nil && codeVal > 0 { parentAgent, err := l.findAgentByCode(transCtx, codeVal) if err != nil { return errors.Wrapf(err, "") } parentAgentId = parentAgent.Id targetLevel = 1 } else { encRefMobile, _ := crypto.EncryptMobile(req.Referrer, l.svcCtx.Config.Encrypt.SecretKey) agents, findErr := l.svcCtx.AgentModel.FindAll(transCtx, l.svcCtx.AgentModel.SelectBuilder().Where("mobile = ? AND del_state = ?", encRefMobile, globalkey.DelStateNo).Limit(1), "") if findErr != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", findErr) } if len(agents) == 0 { return errors.Wrapf(xerr.NewErrMsg("邀请信息无效"), "") } parentAgentId = agents[0].Id targetLevel = 1 } } // 5. 创建代理记录 newAgent := &model.Agent{Id: uuid.NewString(), UserId: userID, Level: targetLevel, Mobile: encryptedMobile} if req.Region != "" { newAgent.Region = sql.NullString{String: req.Region, Valid: true} } if req.WechatId != "" { newAgent.WechatId = sql.NullString{String: req.WechatId, Valid: true} } // 6. 处理上级关系 if parentAgentId != "" { // 查找上级代理 parentAgent, err := l.svcCtx.AgentModel.FindOne(transCtx, parentAgentId) if err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err) } // 验证关系是否允许(下级不能比上级等级高) if newAgent.Level > parentAgent.Level { return errors.Wrapf(xerr.NewErrMsg("代理等级不能高于上级代理"), "") } // 查找团队首领(钻石代理) teamLeaderId, err := l.findTeamLeader(transCtx, parentAgent.Id) if err != nil { return errors.Wrapf(err, "查找团队首领失败") } if teamLeaderId != "" { newAgent.TeamLeaderId = sql.NullString{String: teamLeaderId, Valid: true} } newAgent.AgentCode = 0 _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) if err != nil { return errors.Wrapf(err, "创建代理记录失败") } agentID = newAgent.Id // 建立关系 relation := &model.AgentRelation{Id: uuid.NewString(), ParentId: parentAgent.Id, ChildId: agentID, RelationType: 1} if _, err := l.svcCtx.AgentRelationModel.Insert(transCtx, session, relation); err != nil { return errors.Wrapf(err, "建立代理关系失败") } } else { // 平台发放的钻石邀请码,独立成团队 if targetLevel == 3 { newAgent.AgentCode = 0 _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) if err != nil { return errors.Wrapf(err, "创建代理记录失败") } agentID = newAgent.Id // 设置自己为团队首领 newAgent.TeamLeaderId = sql.NullString{String: agentID, Valid: true} if err := l.svcCtx.AgentModel.UpdateInTransaction(transCtx, session, newAgent); err != nil { return errors.Wrapf(err, "更新团队首领失败") } } else { // 普通/黄金代理,但没有上级(异常情况) newAgent.AgentCode = 0 _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) if err != nil { return errors.Wrapf(err, "创建代理记录失败") } agentID = newAgent.Id } } // 7. 初始化钱包 wallet := &model.AgentWallet{Id: uuid.NewString(), AgentId: agentID} if _, err := l.svcCtx.AgentWalletModel.Insert(transCtx, session, wallet); err != nil { return errors.Wrapf(err, "初始化钱包失败") } // 8. 更新邀请码状态 // 钻石级别的邀请码只能使用一次,使用后立即失效 // 普通级别的邀请码可以无限使用,不更新状态 if targetLevel == 3 { // 钻石邀请码:使用后失效 inviteCodeModel.Status = 1 // 已使用(使用后立即失效) } if inviteCodeModel != nil { inviteCodeModel.UsedUserId = sql.NullString{String: userID, Valid: true} inviteCodeModel.UsedAgentId = sql.NullString{String: agentID, Valid: true} inviteCodeModel.UsedTime = sql.NullTime{Time: time.Now(), Valid: true} if err := l.svcCtx.AgentInviteCodeModel.UpdateWithVersion(transCtx, session, inviteCodeModel); err != nil { return errors.Wrapf(err, "更新邀请码状态失败") } } // 9. 记录邀请码使用历史(用于统计和查询) if inviteCodeModel != nil { usage := &model.AgentInviteCodeUsage{Id: uuid.NewString(), InviteCodeId: inviteCodeModel.Id, Code: inviteCodeModel.Code, UserId: userID, AgentId: agentID, AgentLevel: targetLevel, UsedTime: time.Now()} if _, err := l.svcCtx.AgentInviteCodeUsageModel.Insert(transCtx, session, usage); err != nil { return errors.Wrapf(err, "记录邀请码使用历史失败") } } agentLevel = targetLevel return nil }) if err != nil { return nil, err } // 10. 生成并返回token token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) if err != nil { return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成token失败: %v", err) } now := time.Now().Unix() // 获取等级名称 levelName := "" switch agentLevel { case 1: levelName = "普通" case 2: levelName = "黄金" case 3: levelName = "钻石" } agent, _ := l.svcCtx.AgentModel.FindOne(l.ctx, agentID) return &types.RegisterByInviteCodeResp{ AccessToken: token, AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, AgentId: agentID, Level: agentLevel, LevelName: levelName, AgentCode: func() int64 { if agent != nil { return agent.AgentCode } return 0 }(), }, nil } // findTeamLeader 查找团队首领(钻石代理) func (l *RegisterByInviteCodeLogic) findTeamLeader(ctx context.Context, agentId string) (string, error) { currentId := agentId maxDepth := 100 depth := 0 for depth < maxDepth { builder := l.svcCtx.AgentRelationModel.SelectBuilder(). Where("child_id = ? AND relation_type = ? AND del_state = ?", currentId, 1, 0) relations, err := l.svcCtx.AgentRelationModel.FindAll(ctx, builder, "") if err != nil { return "", err } if len(relations) == 0 { agent, err := l.svcCtx.AgentModel.FindOne(ctx, currentId) if err != nil { return "", err } if agent.Level == 3 { return agent.Id, nil } return "", nil } parentAgent, err := l.svcCtx.AgentModel.FindOne(ctx, relations[0].ParentId) if err != nil { return "", err } if parentAgent.Level == 3 { return parentAgent.Id, nil } currentId = parentAgent.Id depth++ } return "", nil } func (l *RegisterByInviteCodeLogic) findAgentByCode(ctx context.Context, code int64) (*model.Agent, error) { builder := l.svcCtx.AgentModel.SelectBuilder().Where("agent_code = ? AND del_state = ?", code, globalkey.DelStateNo).Limit(1) agents, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "") if err != nil { return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err) } if len(agents) == 0 { return nil, errors.Wrapf(xerr.NewErrMsg("上级邀请码不存在"), "") } return agents[0], nil } func (l *RegisterByInviteCodeLogic) allocateAgentCode(ctx context.Context, session sqlx.Session) (int64, error) { builder := l.svcCtx.AgentModel.SelectBuilder().OrderBy("agent_code DESC").Limit(1) rows, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "") if err != nil { return 0, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理编码失败, %v", err) } var next int64 = 16800 if len(rows) > 0 && rows[0].AgentCode > 0 { next = rows[0].AgentCode + 1 } return next, nil } // handleMobileNotExists 处理手机号不存在的情况 func (l *RegisterByInviteCodeLogic) handleMobileNotExists(ctx context.Context, session sqlx.Session, encryptedMobile string, currentUserID string) (string, error) { if currentUserID == "" { // 场景1.1: 未登录 + 手机号不存在 -> 创建新用户 newUser := &model.User{Id: uuid.NewString(), Mobile: sql.NullString{String: encryptedMobile, Valid: true}} if _, err := l.svcCtx.UserModel.Insert(ctx, session, newUser); err != nil { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err) } // 创建 mobile 认证 if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{ Id: uuid.NewString(), UserId: newUser.Id, AuthType: model.UserAuthTypeMobile, AuthKey: encryptedMobile, }); err != nil { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建手机号认证失败: %v", err) } return newUser.Id, nil } else { // 场景1.2: 已登录临时用户 + 手机号不存在 -> 升级为正式用户 // 前置检查已保证不是正式用户,所以这里一定是临时用户 currentUser, err := l.svcCtx.UserModel.FindOne(ctx, currentUserID) if err != nil { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询当前用户失败: %v", err) } // 升级为正式用户 currentUser.Mobile = sql.NullString{String: encryptedMobile, Valid: true} if _, err := l.svcCtx.UserModel.Update(ctx, session, currentUser); err != nil { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新手机号失败: %v", err) } // 创建 mobile 认证 if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{ Id: uuid.NewString(), UserId: currentUserID, AuthType: model.UserAuthTypeMobile, AuthKey: encryptedMobile, }); err != nil { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建手机号认证失败: %v", err) } return currentUserID, nil } } // handleMobileExists 处理手机号已存在的情况 func (l *RegisterByInviteCodeLogic) handleMobileExists(ctx context.Context, session sqlx.Session, targetUser *model.User, currentUserID string, currentAuthType string, currentAuthKey string) (string, error) { userID := targetUser.Id // 检查目标用户是否已是代理 existingAgent, err := l.svcCtx.AgentModel.FindOneByUserId(ctx, userID) if err != nil && !errors.Is(err, model.ErrNotFound) { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败: %v", err) } if existingAgent != nil { return "", errors.Wrapf(xerr.NewErrMsg("该手机号已经是代理,不能重复注册"), "") } if currentUserID == "" { // 场景2.1: 未登录 + 手机号存在 -> 直接使用目标用户(验证码已确认身份) return userID, nil } else if currentUserID == userID { // 场景2.2: 已登录正式用户 + 手机号匹配 -> 直接使用 // 前置检查已保证手机号匹配且不是代理 return userID, nil } else { // 场景2.3: 已登录临时用户 + 手机号存在 -> 需要合并账号 // 前置检查已保证是临时用户(不是正式用户) return l.mergeTempUserToTarget(ctx, session, currentUserID, userID, currentAuthType, currentAuthKey) } } // mergeTempUserToTarget 合并临时用户到目标用户 func (l *RegisterByInviteCodeLogic) mergeTempUserToTarget(ctx context.Context, session sqlx.Session, sourceUserID string, targetUserID string, currentAuthType string, currentAuthKey string) (string, error) { // 检查目标用户是否已有该认证(除了UUID) if currentAuthType != model.UserAuthTypeUUID { targetAuth, err := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, currentAuthType) if err != nil && !errors.Is(err, model.ErrNotFound) { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err) } if targetAuth != nil && targetAuth.AuthKey != currentAuthKey { // 目标用户已有该类型的其他认证,证明手机号绑定过其他微信等 if currentAuthType == model.UserAuthTypeWxh5OpenID { return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他微信号"), "") } if currentAuthType == model.UserAuthTypeWxMiniOpenID { return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他微信号"), "") } return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他终端"), "") } } // 查找当前认证 existingAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(ctx, currentAuthType, currentAuthKey) if err != nil && !errors.Is(err, model.ErrNotFound) { return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err) } // 执行账号合并 if err := l.mergeUserAccounts(ctx, session, sourceUserID, targetUserID, currentAuthType, currentAuthKey, existingAuth); err != nil { return "", err } return targetUserID, nil } // mergeUserAccounts 合并账号:迁移认证、业务数据,删除临时用户 func (l *RegisterByInviteCodeLogic) mergeUserAccounts(ctx context.Context, session sqlx.Session, sourceUserID string, targetUserID string, currentAuthType string, currentAuthKey string, existingAuth *model.UserAuth) error { // 1) 认证绑定处理 if existingAuth != nil && existingAuth.UserId != targetUserID { // 认证存在但不属于目标用户,迁移到目标用户 if currentAuthType == model.UserAuthTypeUUID { // UUID替换策略:如果目标用户已有UUID认证,替换UUID;否则迁移认证 targetUUIDAuth, _ := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, model.UserAuthTypeUUID) if targetUUIDAuth != nil { // 目标用户已有UUID认证,删除源认证并更新目标UUID if err := l.svcCtx.UserAuthModel.Delete(ctx, session, existingAuth.Id); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除旧UUID认证失败: %v", err) } if targetUUIDAuth.AuthKey != currentAuthKey { targetUUIDAuth.AuthKey = currentAuthKey if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err) } } } else { // 目标用户没有UUID认证,迁移源认证 existingAuth.UserId = targetUserID if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "迁移UUID认证失败: %v", err) } } } else { // 其他认证类型,直接迁移 existingAuth.UserId = targetUserID if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "迁移认证失败: %v", err) } } } else if existingAuth == nil { // 认证不存在,创建新认证 if currentAuthType == model.UserAuthTypeUUID { // UUID特殊处理:如果目标用户已有UUID认证,更新UUID;否则创建新认证 targetUUIDAuth, _ := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, model.UserAuthTypeUUID) if targetUUIDAuth != nil { if targetUUIDAuth.AuthKey != currentAuthKey { targetUUIDAuth.AuthKey = currentAuthKey if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err) } } } else { if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{ Id: uuid.NewString(), UserId: targetUserID, AuthType: currentAuthType, AuthKey: currentAuthKey, }); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建UUID认证失败: %v", err) } } } else { // 其他认证类型,创建新认证 if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{ Id: uuid.NewString(), UserId: targetUserID, AuthType: currentAuthType, AuthKey: currentAuthKey, }); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建认证失败: %v", err) } } } // 2) 业务数据迁移:迁移订单和报告到目标用户 if err := l.svcCtx.OrderModel.UpdateUserIDWithSession(ctx, session, sourceUserID, targetUserID); err != nil { return errors.Wrapf(err, "迁移订单失败") } if err := l.svcCtx.QueryModel.UpdateUserIDWithSession(ctx, session, sourceUserID, targetUserID); err != nil { return errors.Wrapf(err, "迁移报告失败") } // 3) 删除临时用户 sourceUser, err := l.svcCtx.UserModel.FindOne(ctx, sourceUserID) if err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找源用户失败: %v", err) } if err := l.svcCtx.UserModel.Delete(ctx, session, sourceUser.Id); err != nil { return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除源用户失败: %v", err) } return nil }