mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-02 23:40:24 +08:00
129 lines
3.9 KiB
Go
129 lines
3.9 KiB
Go
package api
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"mayfly-go/internal/auth/config"
|
||
"mayfly-go/internal/auth/imsg"
|
||
"mayfly-go/internal/auth/pkg/otp"
|
||
msgapp "mayfly-go/internal/msg/application"
|
||
msgentity "mayfly-go/internal/msg/domain/entity"
|
||
sysapp "mayfly-go/internal/sys/application"
|
||
sysentity "mayfly-go/internal/sys/domain/entity"
|
||
"mayfly-go/pkg/biz"
|
||
"mayfly-go/pkg/cache"
|
||
"mayfly-go/pkg/i18n"
|
||
"mayfly-go/pkg/req"
|
||
"mayfly-go/pkg/utils/collx"
|
||
"mayfly-go/pkg/utils/jsonx"
|
||
"mayfly-go/pkg/utils/netx"
|
||
"mayfly-go/pkg/utils/stringx"
|
||
"mayfly-go/pkg/utils/timex"
|
||
"time"
|
||
)
|
||
|
||
const (
|
||
OtpStatusNone = -1 // 未启用otp校验
|
||
OtpStatusReg = 1 // 用户otp secret已注册
|
||
OtpStatusNoReg = 2 // 用户otp secret未注册
|
||
)
|
||
|
||
// 最后的登录校验（共用）。校验通过返回登录成功响应结果map
|
||
func LastLoginCheck(ctx context.Context, account *sysentity.Account, accountLoginSecurity *config.AccountLoginSecurity, loginIp string) map[string]any {
|
||
biz.IsTrueI(ctx, account.IsEnable(), imsg.ErrAccountNotAvailable)
|
||
username := account.Username
|
||
|
||
res := collx.M{
|
||
"name": account.Name,
|
||
"username": username,
|
||
"lastLoginTime": account.LastLoginTime,
|
||
"lastLoginIp": account.LastLoginIp,
|
||
}
|
||
|
||
// 默认为不校验otp
|
||
otpStatus := OtpStatusNone
|
||
// 访问系统使用的token
|
||
accessToken, refreshToken, err := req.CreateToken(account.Id, username)
|
||
biz.ErrIsNilAppendErr(err, "token create failed: %s")
|
||
|
||
// 若系统配置中设置开启otp双因素校验，则进行otp校验
|
||
if accountLoginSecurity.UseOtp {
|
||
otpInfo, otpurl, otpToken := useOtp(account, accountLoginSecurity.OtpIssuer, accessToken, refreshToken)
|
||
otpStatus = otpInfo.OptStatus
|
||
if otpurl != "" {
|
||
res["otpUrl"] = otpurl
|
||
}
|
||
accessToken = otpToken
|
||
} else {
|
||
res["refresh_token"] = refreshToken
|
||
// 不进行otp二次校验则直接返回accessToken
|
||
// 保存登录消息
|
||
go saveLogin(ctx, account, loginIp)
|
||
}
|
||
|
||
// 赋值otp状态
|
||
res["otp"] = otpStatus
|
||
res["token"] = accessToken
|
||
return res
|
||
}
|
||
|
||
func useOtp(account *sysentity.Account, otpIssuer, accessToken string, refreshToken string) (*OtpVerifyInfo, string, string) {
|
||
biz.ErrIsNil(account.OtpSecretDecrypt())
|
||
otpSecret := account.OtpSecret
|
||
// 修改状态为已注册
|
||
otpStatus := OtpStatusReg
|
||
otpUrl := ""
|
||
// 该token用于otp双因素校验
|
||
token := stringx.Rand(32)
|
||
// 未注册otp secret或重置了秘钥
|
||
if otpSecret == "" || otpSecret == "-" {
|
||
otpStatus = OtpStatusNoReg
|
||
key, err := otp.NewTOTP(otp.GenerateOpts{
|
||
AccountName: account.Username,
|
||
Issuer: otpIssuer,
|
||
})
|
||
biz.ErrIsNilAppendErr(err, "otp generate failed: %s")
|
||
otpUrl = key.URL()
|
||
otpSecret = key.Secret()
|
||
}
|
||
// 缓存otpInfo, 只有双因素校验通过才可返回真正的token
|
||
otpInfo := &OtpVerifyInfo{
|
||
AccountId: account.Id,
|
||
Username: account.Username,
|
||
OptStatus: otpStatus,
|
||
OtpSecret: otpSecret,
|
||
AccessToken: accessToken,
|
||
RefreshToken: refreshToken,
|
||
}
|
||
cache.SetStr(fmt.Sprintf("otp:token:%s", token), jsonx.ToStr(otpInfo), time.Minute*time.Duration(3))
|
||
return otpInfo, otpUrl, token
|
||
}
|
||
|
||
// 获取ip与归属地信息
|
||
func getIpAndRegion(rc *req.Ctx) string {
|
||
clientIp := rc.ClientIP()
|
||
return fmt.Sprintf("%s %s", clientIp, netx.Ip2Region(clientIp))
|
||
}
|
||
|
||
// 保存更新账号登录信息
|
||
func saveLogin(ctx context.Context, account *sysentity.Account, ip string) {
|
||
// 更新账号最后登录时间
|
||
now := time.Now()
|
||
updateAccount := &sysentity.Account{LastLoginTime: &now}
|
||
updateAccount.Id = account.Id
|
||
updateAccount.LastLoginIp = ip
|
||
// 偷懒为了方便直接获取accountApp
|
||
biz.ErrIsNil(sysapp.GetAccountApp().Update(context.TODO(), updateAccount))
|
||
|
||
// 创建登录消息
|
||
loginMsg := &msgentity.Msg{
|
||
RecipientId: int64(account.Id),
|
||
Msg: i18n.TC(ctx, imsg.LoginMsg, "ip", ip, "time", timex.DefaultFormat(now)),
|
||
Type: 1,
|
||
}
|
||
loginMsg.CreateTime = &now
|
||
loginMsg.Creator = account.Username
|
||
loginMsg.CreatorId = account.Id
|
||
msgapp.GetMsgApp().Create(context.TODO(), loginMsg)
|
||
}
|