重构架构

internal/cli/init/db.go

@@ -0,0 +1,53 @@
+package initcmd
+
+import (
+	"fmt"
+	"sunhpc/internal/middler/auth"
+	"sunhpc/pkg/config"
+	"sunhpc/pkg/database"
+	"sunhpc/pkg/logger"
+
+	"github.com/spf13/cobra"
+)
+
+func NewInitDBCmd() *cobra.Command {
+	var force bool
+
+	cmd := &cobra.Command{
+		Use:   "db",
+		Short: "初始化数据库",
+		Long: `初始化SQLite数据库,创建所有表结构和默认数据。
+    
+示例:
+  sunhpc init db         # 初始化数据库
+  sunhpc init db --force # 强制重新初始化`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if err := auth.RequireRoot(); err != nil {
+				return err
+			}
+
+			logger.Debug("执行数据库初始化...")
+
+			cfg, err := config.LoadConfig()
+			if err != nil {
+				return fmt.Errorf("加载配置失败: %w", err)
+			}
+
+			// 初始化数据库
+			db, err := database.GetInstance(&cfg.Database, nil)
+			if err != nil {
+				return fmt.Errorf("数据库连接失败: %w", err)
+			}
+			defer db.Close()
+
+			if err := db.InitTables(force); err != nil {
+				return fmt.Errorf("数据库初始化失败: %w", err)
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().BoolVarP(&force, "force", "f", false, "强制重新初始化")
+	return cmd
+}