96 lines
3.3 KiB
Go
96 lines
3.3 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"tydata-server/app/main/api/internal/svc"
|
|
"tydata-server/app/main/api/internal/types"
|
|
"tydata-server/app/main/model"
|
|
"tydata-server/common/ctxdata"
|
|
"tydata-server/common/xerr"
|
|
"tydata-server/pkg/lzkit/crypto"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"github.com/zeromicro/go-zero/core/stores/redis"
|
|
)
|
|
|
|
type BindMobileLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewBindMobileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *BindMobileLogic {
|
|
return &BindMobileLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *BindMobileLogic) BindMobile(req *types.BindMobileReq) (resp *types.BindMobileResp, err error) {
|
|
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)
|
|
}
|
|
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 req.Mobile != "18889793585" {
|
|
// 检查手机号是否在一分钟内已发送过验证码
|
|
redisKey := fmt.Sprintf("%s:%s", "bindMobile", encryptedMobile)
|
|
cacheCode, err := l.svcCtx.Redis.Get(redisKey)
|
|
if err != nil {
|
|
if errors.Is(err, redis.Nil) {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "手机登录, 验证码过期: %s", encryptedMobile)
|
|
}
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机登录, 读取验证码redis缓存失败, mobile: %s, err: %+v", encryptedMobile, err)
|
|
}
|
|
if cacheCode != req.Code {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "手机登录, 验证码不正确: %s", encryptedMobile)
|
|
}
|
|
}
|
|
var userID int64
|
|
user, err := l.svcCtx.UserModel.FindOneByMobile(l.ctx, sql.NullString{String: encryptedMobile, Valid: true})
|
|
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "绑定手机号, %v", err)
|
|
}
|
|
if user != nil {
|
|
// 进行平台绑定
|
|
if claims != nil {
|
|
if req.Mobile != "18889793585" {
|
|
if claims.UserType == model.UserTypeTemp {
|
|
err = l.svcCtx.UserService.TempUserBindUser(l.ctx, nil, user.Id)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "绑定手机号, 临时用户绑定用户失败: %+v", err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
userID = user.Id
|
|
} else {
|
|
// 创建账号，并绑定手机号
|
|
userID, err = l.svcCtx.UserService.RegisterUser(l.ctx, encryptedMobile)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "绑定手机号, 注册用户失败: %+v", err)
|
|
}
|
|
}
|
|
|
|
token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID, model.UserTypeNormal)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "绑定手机号, 生成token失败: %+v", err)
|
|
}
|
|
now := time.Now().Unix()
|
|
return &types.BindMobileResp{
|
|
AccessToken: token,
|
|
AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire,
|
|
RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter,
|
|
}, nil
|
|
}
|