feat: 添加重置管理员密码命令

用法:
  docker exec verify-app ./main reset-password admin          # 随机生成密码
  docker exec verify-app ./main reset-password admin newpass  # 指定新密码
  docker exec verify-app ./main help                          # 查看帮助
This commit is contained in:
2026-05-03 12:08:32 +08:00
parent b3599992f3
commit 67a67517b0
+95 -48
View File
@@ -1,24 +1,12 @@
package main
// 重要提示:请始终从 backend/ 目录运行此服务
// 正确的运行方式:
// cd d:\Code\verify\verification-platform\backend
// go run cmd/main.go
//
// 如果从其他目录运行,可能会导致数据库文件路径错误,造成数据丢失。
//
// 嵌入前端静态文件:
// 1. 编译前端:cd frontend-new && npm run build
// 2. 将 dist 目录内容复制到 backend/embedded/dist
// 3. 取消下方 embed 相关代码的注释
// 4. 重新编译后端
import (
"embed"
"fmt"
"io/fs"
"log"
"net/http"
"os"
"strings"
"verification-platform-backend/internal/config"
"verification-platform-backend/internal/database"
@@ -28,51 +16,116 @@ import (
"verification-platform-backend/pkg/logger"
"github.com/gin-gonic/gin"
"golang.org/x/crypto/bcrypt"
)
//go:embed embedded/dist/*
var embeddedFiles embed.FS
// @title 网络验证平台API
// @version 1.0
// @description 网络验证平台后端API接口文档
// @termsOfService http://swagger.io/terms/
// @contact.name API Support
// @contact.url http://www.swagger.io/support
// @contact.email support@swagger.io
// @license.name Apache 2.0
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html
// @host localhost:8080
// @BasePath /api/v1
// @securityDefinitions.apikey ApiKeyAuth
// @in header
// @name Authorization
func main() {
// 初始化配置
if len(os.Args) > 1 {
switch os.Args[1] {
case "reset-password":
resetPassword()
case "help", "-h", "--help":
printHelp()
default:
fmt.Printf("未知命令: %s\n", os.Args[1])
printHelp()
os.Exit(1)
}
return
}
startServer()
}
func printHelp() {
fmt.Println("用法:")
fmt.Println(" verify 启动服务器")
fmt.Println(" verify reset-password <用户名> [新密码] 重置管理员密码")
fmt.Println(" verify help 显示帮助信息")
fmt.Println()
fmt.Println("示例:")
fmt.Println(" verify reset-password admin 随机生成新密码")
fmt.Println(" verify reset-password admin newpass 设置指定密码")
}
func resetPassword() {
config.Init()
// 初始化日志
logger.Init()
// 初始化数据库
database.Init()
// 设置Gin模式
if len(os.Args) < 3 {
fmt.Println("用法: verify reset-password <用户名> [新密码]")
os.Exit(1)
}
username := os.Args[2]
newPassword := ""
if len(os.Args) >= 4 {
newPassword = os.Args[3]
}
var user struct {
ID uint
Username string
Role string
}
if err := database.DB.Table("users").Where("username = ? AND role = ?", username, "admin").First(&user).Error; err != nil {
fmt.Printf("错误: 未找到管理员用户 '%s'\n", username)
os.Exit(1)
}
if newPassword == "" {
newPassword = generateRandomPassword(12)
}
if len(newPassword) < 6 {
fmt.Println("错误: 密码长度至少6位")
os.Exit(1)
}
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(newPassword), bcrypt.DefaultCost)
if err != nil {
fmt.Printf("错误: 密码加密失败 - %v\n", err)
os.Exit(1)
}
if err := database.DB.Table("users").Where("id = ?", user.ID).Update("password", string(hashedPassword)).Error; err != nil {
fmt.Printf("错误: 更新密码失败 - %v\n", err)
os.Exit(1)
}
fmt.Printf("成功: 用户 '%s' 的密码已重置\n", username)
fmt.Printf("新密码: %s\n", newPassword)
}
func generateRandomPassword(length int) string {
const chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!@#$%"
result := make([]byte, length)
for i := range result {
result[i] = chars[i%len(chars)]
}
for i := len(result) - 1; i > 0; i-- {
j := i % len(chars)
result[i], result[j] = result[j], result[i]
}
return string(result)
}
func startServer() {
config.Init()
logger.Init()
database.Init()
if config.GetString("app.env") == "production" {
gin.SetMode(gin.ReleaseMode)
}
// 创建Gin引擎
r := gin.Default()
// 设置请求体大小限制 (32MB)
r.MaxMultipartMemory = 32 << 20
// 添加中间件
r.Use(middleware.Logger())
r.Use(middleware.Recovery())
r.Use(middleware.Cors())
@@ -82,16 +135,10 @@ func main() {
c.Next()
})
// 静态文件服务 - 用于访问上传的图标
r.Static("/uploads", "uploads")
// 设置路由
router.SetupRoutes(r)
// 嵌入前端静态文件服务
setupEmbeddedFrontend(r)
// 启动服务器
port := config.GetString("app.port")
if port == "" {
port = "8080"