增加重置totp的命令

This commit is contained in:
dushixiang
2021-04-14 22:27:12 +08:00
parent 5cee2dd50e
commit 11f2d8a1f4
4 changed files with 54 additions and 4 deletions

View File

@ -275,8 +275,8 @@ func InitDBData() (err error) {
return nil
}
func ResetPassword() error {
user, err := userRepository.FindByUsername(global.Config.ResetPassword)
func ResetPassword(username string) error {
user, err := userRepository.FindByUsername(username)
if err != nil {
return err
}
@ -296,6 +296,22 @@ func ResetPassword() error {
return nil
}
func ResetTotp(username string) error {
user, err := userRepository.FindByUsername(username)
if err != nil {
return err
}
u := &model.User{
TOTPSecret: "-",
ID: user.ID,
}
if err := userRepository.Update(u); err != nil {
return err
}
log.Debugf("用户「%v」已重置TOTP", user.Username)
return nil
}
func SetupCache() *cache.Cache {
// 配置缓存器
mCache := cache.New(5*time.Minute, 10*time.Minute)