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
 }