duorameng
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package resetpwd
|
|
|
|
import (
|
|
"bufio"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/engigu/baihu-panel/cmd/clibase"
|
|
"github.com/engigu/baihu-panel/internal/services"
|
|
"github.com/engigu/baihu-panel/internal/utils"
|
|
)
|
|
|
|
func printHelp() {
|
|
clibase.PrintSubCommandUsage("白虎面板用户密码重置工具", "baihu resetpwd [用户名]", " baihu resetpwd admin", nil)
|
|
}
|
|
|
|
func Run(args []string) {
|
|
if len(args) > 0 && (args[0] == "-h" || args[0] == "--help") {
|
|
printHelp()
|
|
return
|
|
}
|
|
|
|
fs := flag.NewFlagSet("resetpwd", flag.ExitOnError)
|
|
fs.Usage = printHelp
|
|
|
|
if err := fs.Parse(args); err != nil {
|
|
return
|
|
}
|
|
|
|
if err := clibase.InitContext(true); err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
userService := services.NewUserService()
|
|
|
|
var username string
|
|
parsedArgs := fs.Args()
|
|
if len(parsedArgs) >= 1 {
|
|
username = parsedArgs[0]
|
|
} else {
|
|
username = "admin"
|
|
}
|
|
|
|
fmt.Printf("此操作将重置用户 [%s] 的密码，是否继续? (y/N): ", username)
|
|
reader := bufio.NewReader(os.Stdin)
|
|
answer, _ := reader.ReadString('\n')
|
|
answer = strings.TrimSpace(strings.ToLower(answer))
|
|
|
|
if answer != "y" && answer != "yes" {
|
|
fmt.Println("操作已取消。")
|
|
return
|
|
}
|
|
|
|
fmt.Printf("请输入用户 [%s] 的新密码 (留空则自动随机生成): ", username)
|
|
inputPwd, _ := reader.ReadString('\n')
|
|
newPassword := strings.TrimSpace(inputPwd)
|
|
if newPassword == "" {
|
|
newPassword = utils.RandomString(12)
|
|
fmt.Println("未输入密码，系统已自动生成。")
|
|
}
|
|
|
|
user := userService.GetUserByUsername(username)
|
|
if user == nil {
|
|
fmt.Printf("找不到用户 [%s]\n", username)
|
|
clibase.PrintDBConfigHint("resetpwd " + username)
|
|
return
|
|
}
|
|
|
|
err := userService.UpdatePassword(user.ID, newPassword)
|
|
if err != nil {
|
|
fmt.Printf("重置密码失败: %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Println("--------------------------------------------------")
|
|
fmt.Printf("用户 [%s] 密码已重置成功:\n", username)
|
|
fmt.Printf("新密码: %s\n", newPassword)
|
|
fmt.Println("请妥善保管您的新密码，并登录后及时修改。")
|
|
fmt.Println("--------------------------------------------------")
|
|
}
|