增加使用命令初始化用户密码功能

This commit is contained in:
dushixiang
2021-03-05 20:02:17 +08:00
parent d16295baa5
commit af74e18c9a
3 changed files with 59 additions and 6 deletions

23
main.go
View File

@ -27,7 +27,10 @@ import (
const Version = "v0.3.0"
func main() {
log.Fatal(Run())
err := Run()
if err != nil {
log.Fatal(err)
}
}
func Run() error {
@ -85,6 +88,24 @@ func Run() error {
return err
}
if global.Config.ResetPassword != "" {
user, err := model.FindUserByUsername(global.Config.ResetPassword)
if err != nil {
return err
}
password := "next-terminal"
passwd, err := utils.Encoder.Encode([]byte(password))
if err != nil {
return err
}
u := &model.User{
Password: string(passwd),
}
model.UpdateUserById(u, user.ID)
logrus.Debugf("用户「%v」密码初始化为: %v", user.Username, password)
return nil
}
if err := global.DB.AutoMigrate(&model.User{}); err != nil {
return err
}