增加重置totp的命令

docs/faq.md

@@ -18,7 +18,8 @@ location / {
 <details>
     <summary>访问realvnc提示验证失败？</summary>
 
-把加密类型修改为 Prefer On
+1. 把密码类型修改为VNC
+2. 把加密类型修改为 Prefer On
 
 </details>
 
@@ -88,3 +89,31 @@ Mar  5 20:00:16.923 [DEBU] 用户「admin」密码初始化为: next-terminal
 
 </details>
 
+<details>
+    <summary>TOTP客户端丢了怎么办？</summary>
+首先需要进入程序所在目录，使用docker安装的程序目录为：/usr/local/next-terminal
+
+执行命令
+
+```shell
+./next-terminal --reset-totp admin
+```
+
+其中 admin 为用户登录账号，成功之后会输出
+
+``` shell
+
+ _______                   __    ___________                  .__              .__   
+ \      \   ____ ___  ____/  |_  \__    ___/__________  _____ |__| ____ _____  |  |  
+ /   |   \_/ __ \\  \/  /\   __\   |    |_/ __ \_  __ \/     \|  |/    \\__  \ |  |  
+/    |    \  ___/ >    <  |  |     |    |\  ___/|  | \/  Y Y  \  |   |  \/ __ \|  |__
+\____|__  /\___  >__/\_ \ |__|     |____| \___  >__|  |__|_|  /__|___|  (____  /____/
+        \/     \/      \/                     \/            \/        \/     \/      v0.3.0
+
+当前数据库模式为：mysql
+Mar  5 20:00:16.923 [DEBU] 用户「admin」已重置TOTP
+
+```
+
+</details>
+

main.go

@@ -39,7 +39,10 @@ func Run() error {
 	e := api.SetupRoutes(db)
 
 	if global.Config.ResetPassword != "" {
-		return api.ResetPassword()
+		return api.ResetPassword(global.Config.ResetPassword)
+	}
+	if global.Config.ResetTotp != "" {
+		return api.ResetTotp(global.Config.ResetTotp)
 	}
 	sessionRepo := repository.NewSessionRepository(db)
 	propertyRepo := repository.NewPropertyRepository(db)

pkg/config/config.go

@@ -18,6 +18,7 @@ type Config struct {
 	Mysql         *Mysql
 	Sqlite        *Sqlite
 	ResetPassword string
+	ResetTotp     string
 }
 
 type Mysql struct {
@@ -83,6 +84,7 @@ func SetupConfig() *Config {
 			Key:  viper.GetString("server.key"),
 		},
 		ResetPassword: viper.GetString("reset-password"),
+		ResetTotp:     viper.GetString("reset-totp"),
 		Debug:         viper.GetBool("debug"),
 		Demo:          viper.GetBool("demo"),
 	}

server/api/routes.go

@@ -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)