48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package initcmd
|
|
|
|
import (
|
|
"sunhpc/internal/auth"
|
|
"sunhpc/internal/db"
|
|
"sunhpc/internal/log"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func NewDatabaseCmd() *cobra.Command {
|
|
var force bool
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "database",
|
|
Short: "初始化数据库",
|
|
Long: `初始化SQLite数据库,创建所有表结构和默认数据。
|
|
|
|
示例:
|
|
sunhpc init database # 初始化数据库
|
|
sunhpc init database --force # 强制重新初始化`,
|
|
|
|
Annotations: map[string]string{
|
|
"skip-db-check": "true", // 标记此命令跳过数据库检查
|
|
},
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if err := auth.RequireRoot(); err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Info("初始化数据库...")
|
|
if force {
|
|
log.Warn("⚠️ 警告：强制重新初始化将清空数据库中的所有数据！")
|
|
}
|
|
|
|
dbInst := db.MustGetDB() // panic if fail (ok for CLI tool)
|
|
if err := dbInst.InitSchema(force); err != nil {
|
|
return err
|
|
}
|
|
log.Info("数据库初始化完成")
|
|
return nil
|
|
},
|
|
}
|
|
|
|
cmd.Flags().BoolVarP(&force, "force", "f", false, "强制重新初始化")
|
|
return cmd
|
|
}
|