ok-2

cmd/init/config.go

@@ -11,31 +11,58 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var configCmd = &cobra.Command{
-	Use:   "config",
-	Short: "生成基础配置文件",
-	Long:  "创建 /etc/sunhpc 目录并生成所有默认配置文件（若目录已存在则跳过）",
-	RunE: func(cmd *cobra.Command, args []string) error {
-		if err := auth.RequireRoot(); err != nil {
-			return err
-		}
+// NewConfigCmd 创建 "init config" 命令
+func NewConfigCmd() *cobra.Command {
+	var (
+		force   bool
+		path    string
+		verbose bool
+	)
 
-		// 检查目录是否已存在
-		if _, err := os.Stat(config.BaseDir); err == nil {
-			log.Warnf("配置目录 %s 已存在，跳过初始化", config.BaseDir)
+	cmd := &cobra.Command{
+		Use:   "config",
+		Short: "生成默认配置文件",
+		Long: `
+  在指定路径生成 SunHPC 默认配置文件 (sunhpc.yaml)
+
+  示例:
+    sunhpc init config    # 生成默认配置文件
+    sunhpc init config -f # 强制覆盖已有配置文件
+    sunhpc init config -p /etc/sunhpc/sunhpc.yaml # 指定路径
+		`,
+
+		Annotations: map[string]string{
+			"require-root": "true", // 假设需要 root（你可自定义策略）
+		},
+
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if err := auth.RequireRoot(); err != nil {
+				return err
+			}
+
+			if path == "" {
+				path = "/etc/sunhpc/sunhpc.yaml"
+			}
+
+			if !force {
+				if _, err := os.Stat(path); err == nil {
+					return fmt.Errorf("配置文件已存在: %s (使用 --force 覆盖)", path)
+				}
+			}
+
+			if err := config.WriteDefaultConfig(path); err != nil {
+				return fmt.Errorf("写入配置失败: %w", err)
+			}
+
+			log.Infof("✅ 配置文件已生成: %s", path)
 			return nil
-		}
+		},
+	}
 
-		log.Info("初始化 SunHPC 配置目录...")
-		if err := config.InitDirs(); err != nil {
-			return fmt.Errorf("创建目录失败: %v", err)
-		}
+	// 定义局部 flags
+	cmd.Flags().BoolVarP(&force, "force", "f", false, "强制覆盖已有配置文件")
+	cmd.Flags().StringVarP(&path, "path", "p", "", "指定配置文件路径")
+	cmd.Flags().BoolVarP(&verbose, "verbose", "v", false, "显示详细日志")
 
-		if err := config.CreateDefaultConfigs(); err != nil {
-			return fmt.Errorf("生成默认配置文件失败: %v", err)
-		}
-
-		log.Info("配置文件已生成，请根据需要编辑 /etc/sunhpc/ 下的 YAML 文件")
-		return nil
-	},
+	return cmd
 }

cmd/init/database.go

@@ -1,193 +1,47 @@
 package initcmd
 
 import (
-	"bufio"
-	"fmt"
-	"os"
-	"path/filepath"
-	"strings"
-
 	"sunhpc/internal/auth"
 	"sunhpc/internal/db"
 	"sunhpc/internal/log"
 
 	"github.com/spf13/cobra"
-	"github.com/spf13/viper"
 )
 
-const (
-	appName       string = "sunhpc"
-	defaultDBPath string = "/var/lib/sunhpc"
-	defaultDBName string = "sunhpc.db"
-)
+func NewDatabaseCmd() *cobra.Command {
+	var force bool
 
-var (
-	forceDB    bool
-	dbPath     string
-	dbName     string
-	dbFullPath string
-)
-
-func initDBPathWithViper() error {
-	/*
-		从 Viper 配置文件获取数据库路径
-		配置文件里的键要和 Viper.GetXXX 的键对应
-		配置文件格式：
-		db:
-			path: "/tmp/sunhpc"  # 自定义数据库路径
-			name: "my_sunhpc.db" # 自定义数据库名
-	*/
-
-	log.Infof("从 Viper 配置文件获取数据库路径...")
-
-	// ========== 第一步：设置 Viper 配置文件规则（核心） ==========
-	// 1. 设置Viper基础规则
-	viper.SetConfigType("yaml")   // 配置文件类型
-	viper.SetConfigName("config") // 配置文件名（不带后缀）
-	viper.SetEnvPrefix(appName)   // 环境变量前缀：SUNHPC_
-	viper.AutomaticEnv()          // 自动读取环境变量（可选，增强兼容性）
-
-	// 2. 添加配置文件搜索目录（Viper 会按顺序查找，找到第一个就停止）
-	// 优先级：当前目录 → 用户级目录 → 系统级目录
-
-	// ① 当前目录（开发/测试常用）
-	viper.AddConfigPath(".")
-
-	// ② Linux/macOS 用户级目录（~/.config/sunhpc/）
-	if homeDir, err := os.UserHomeDir(); err == nil {
-		viper.AddConfigPath(filepath.Join(homeDir, ".config", appName))
-	}
-	// ③ Linux/macOS 系统级目录（/etc/sunhpc/）
-	viper.AddConfigPath(filepath.Join("/etc", appName))
-
-	// ========== 第二步：设置默认值（最低优先级） ==========
-	viper.SetDefault("db.path", defaultDBPath)
-	viper.SetDefault("db.name", defaultDBName)
-
-	// ========== 第三步：绑定环境变量（优先级高于默认值，低于配置文件） ==========
-	viper.BindEnv("db.path", "DB_PATH") // 绑定 SUNHPC_DB_PATH → db.path
-	viper.BindEnv("db.name", "DB_NAME") // 绑定 SUNHPC_DB_NAME → db.name
-
-	// ========== 第四步：读取配置文件（优先级高于环境变量，低于默认值） ==========
-	if err := viper.ReadInConfig(); err != nil {
-		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
-			log.Info("未找到配置文件,将使用环境变量/默认值")
-			return nil // 配置文件存在但格式错误，返回错误
-		}
-		log.Warnf("读取配置文件失败: %v", err)
-		return fmt.Errorf("读取配置文件失败: %w", err)
-	}
-
-	log.Infof("成功加载配置文件: %s", viper.ConfigFileUsed())
-	return nil
-}
-
-func initDBPath() error {
-
-	// 1. 从 Viper 配置文件获取数据库路径(加载配置文件->环境变量->默认值)
-	if err := initDBPathWithViper(); err != nil {
-		return fmt.Errorf("Viper初始化数据库失败: %v", err)
-	}
-
-	// 2. 从Viper获取数据库路径
-	dbPath = viper.GetString("db.path")
-	dbName = viper.GetString("db.name")
-
-	// 3. 拼接数据库路径
-	dbFullPath = filepath.Join(dbPath, dbName)
-	log.Infof("数据库完整路径: %s", dbFullPath)
-
-	// 3. 检查数据库文件是否存在
-	dir := filepath.Dir(dbFullPath)
-	// 4. 检查数据库目录是否存在
-	if _, err := os.Stat(dir); os.IsNotExist(err) {
-		log.Infof("数据库目录不存在,创建目录: %s", dir)
-
-		if err := os.MkdirAll(dir, 0755); err != nil {
-			return fmt.Errorf("创建数据库目录失败: %v", err)
-		}
-	}
-	return nil
-}
-
-var databaseCmd = &cobra.Command{
-	Use:   "database",
-	Short: "初始化数据库",
-	Long: `初始化SQLite数据库,创建所有表结构和默认数据。
+	cmd := &cobra.Command{
+		Use:   "database",
+		Short: "初始化数据库",
+		Long: `初始化SQLite数据库,创建所有表结构和默认数据。
     
 示例:
   sunhpc init database        # 初始化数据库
   sunhpc init database --force # 强制重新初始化`,
 
-	RunE: func(cmd *cobra.Command, args []string) error {
-		if err := auth.RequireRoot(); err != nil {
-			return err
-		}
-
-		log.Debug("初始化数据库...")
-
-		if err := initDBPath(); err != nil {
-			return fmt.Errorf("初始化数据库路径失败: %w", err)
-		}
-
-		log.Debugf("数据库目录存在: %s", dbPath)
-
-		// 强制模式：用户确认
-		if forceDB {
-			log.Warn("⚠️  警告：强制重新初始化将清空数据库中的所有数据！")
-			fmt.Printf("数据库路径: %s\n", dbFullPath)
-			fmt.Print("确认要重新初始化数据库吗？这将删除所有现有数据。(Y/yes): ")
-
-			reader := bufio.NewReader(os.Stdin)
-			input, err := reader.ReadString('\n')
-			if err != nil {
-				return fmt.Errorf("读取用户输入失败: %v", err)
+		Annotations: map[string]string{
+			"skip-db-check": "true", // 标记此命令跳过数据库检查
+		},
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if err := auth.RequireRoot(); err != nil {
+				return err
 			}
 
-			input = strings.TrimSpace(strings.ToLower(input))
-			if input != "y" && input != "yes" {
-				log.Info("操作已取消")
-				return nil
+			log.Info("初始化数据库...")
+			if force {
+				log.Warn("⚠️  警告：强制重新初始化将清空数据库中的所有数据！")
 			}
 
-			log.Info("用户确认重新初始化数据库")
-		}
-
-		// 数据库存在且不是强制模式则跳过初始化
-		if _, err := os.Stat(dbFullPath); err == nil && !forceDB {
-			log.Infof("数据库文件已存在: %s", dbFullPath)
+			dbInst := db.MustGetDB() // panic if fail (ok for CLI tool)
+			if err := dbInst.InitSchema(force); err != nil {
+				return err
+			}
+			log.Info("数据库初始化完成")
 			return nil
-		}
+		},
+	}
 
-		// 初始化数据库（使用配置的路径）
-		database, err := db.GetInstanceWithConfig(dbPath, dbName)
-		if err != nil {
-			return fmt.Errorf("初始化数据库失败: %v", err)
-		}
-		defer database.Close()
-
-		// 如果是强制模式，设置强制重新初始化标志
-		if forceDB {
-			database.SetForceInit(true)
-			log.Info("强制重新初始化数据库表...")
-
-			// 关闭现有连接以触发重新连接
-			if err := database.CloseConnection(); err != nil {
-				return fmt.Errorf("关闭现有数据库连接失败: %v", err)
-			}
-
-			// 重新连接并初始化
-			if err := database.Connect(); err != nil {
-				return fmt.Errorf("强制重新初始化数据库失败: %v", err)
-			}
-		}
-
-		log.Infof("数据库初始化成功: %s", dbFullPath)
-		return nil
-	},
-}
-
-func init() {
-	databaseCmd.Flags().BoolVarP(&forceDB, "force", "f", false, "强制重新初始化，删除现有数据库")
-	Cmd.AddCommand(databaseCmd)
+	cmd.Flags().BoolVarP(&force, "force", "f", false, "强制重新初始化")
+	return cmd
 }

cmd/init/init.go

@@ -4,6 +4,7 @@ import (
 	"github.com/spf13/cobra"
 )
 
+// 仅定义 Cmd， 注册子命令,只负责组装命令树，尽量不包含业务逻辑
 var Cmd = &cobra.Command{
 	Use:   "init",
 	Short: "初始化集群配置",
@@ -11,7 +12,7 @@ var Cmd = &cobra.Command{
 }
 
 func init() {
-	Cmd.AddCommand(configCmd)
-	Cmd.AddCommand(systemCmd)
-	Cmd.AddCommand(serviceCmd)
+	// 注册所有子命令（通过工厂函数创建, 例如 DatabaseCmd()）
+	Cmd.AddCommand(NewDatabaseCmd())
+	Cmd.AddCommand(NewConfigCmd())
 }

cmd/init/service.go

@@ -1,37 +0,0 @@
-package initcmd
-
-import (
-	"fmt"
-
-	"sunhpc/internal/auth"
-	"sunhpc/internal/config"
-	"sunhpc/internal/log"
-	"sunhpc/internal/service"
-
-	"github.com/spf13/cobra"
-)
-
-var serviceCmd = &cobra.Command{
-	Use:   "service",
-	Short: "根据配置文件初始化服务",
-	Long: `读取 /etc/sunhpc/services.yaml 并部署/配置相关服务。
-支持 HTTPD、TFTPD、DHCPD 等。`,
-	RunE: func(cmd *cobra.Command, args []string) error {
-		if err := auth.RequireRoot(); err != nil {
-			return err
-		}
-
-		svcCfg, err := config.LoadServices()
-		if err != nil {
-			return fmt.Errorf("加载 services.yaml 失败: %v", err)
-		}
-
-		log.Info("开始部署服务...")
-		if err := service.Deploy(svcCfg); err != nil {
-			return fmt.Errorf("服务部署失败: %v", err)
-		}
-
-		log.Info("服务初始化完成")
-		return nil
-	},
-}

cmd/init/system.go

@@ -1,49 +0,0 @@
-package initcmd
-
-import (
-	"fmt"
-
-	"sunhpc/internal/auth"
-	"sunhpc/internal/config"
-	"sunhpc/internal/system"
-
-	"github.com/spf13/cobra"
-)
-
-var (
-	dryRun  bool // --dry-run -n: 仅模拟执行，不实际应用
-	verbose bool // --verbose -v: 启用详细日志输出
-)
-
-var systemCmd = &cobra.Command{
-	Use:   "system [flags]",
-	Short: "根据配置文件初始化系统",
-	Long: `读取 /etc/sunhpc/sunhpc.yaml 中的系统配置项并应用到当前节点。
-	示例:
-		sunhpc init system            # 应用所有配置项
-		sunhpc init system --dry-run  # 仅模拟执行，不实际应用
-		sunhpc init system --verbose  # 启用详细日志输出
-	`,
-	RunE: func(cmd *cobra.Command, args []string) error {
-		// 权限检查：必须以 root 或 sudo 运行
-		if err := auth.RequireRoot(); err != nil {
-			return err
-		}
-
-		// 加载主配置
-		cfg, err := config.LoadSunHPC()
-		if err != nil {
-			return fmt.Errorf("加载 sunhpc.yaml 失败: %v", err)
-		}
-
-		// 统一应用所有配置
-		return system.ApplyAll(cfg)
-	},
-}
-
-// init 初始化 systemCmd 的标志,添加长参数和段参数.
-func init() {
-	// 注册长参数, 布尔参数
-	systemCmd.Flags().BoolVarP(&dryRun, "dry-run", "n", false, "仅模拟执行，不实际应用")
-	systemCmd.Flags().BoolVarP(&verbose, "verbose", "v", false, "启用详细日志输出")
-}