feat: Add 'codei18n init' subcommand --story=1

Author: studyzy

.gitignore

@@ -65,4 +65,5 @@ release/
 # C/C++ artifacts (for CGO)
 *.o
 *.a
-coverage-data
+coverage-data
+./codei18n

cmd/codei18n/hook.go

@@ -36,9 +36,16 @@ func init() {
 }
 
 func runHookInstall() {
+	if err := installHook(); err != nil {
+		log.Fatal("安装 hook 失败: %v", err)
+	}
+	log.Success("Hook 安装成功")
+}
+
+func installHook() error {
 	gitDir := ".git"
 	if _, err := os.Stat(gitDir); os.IsNotExist(err) {
-		log.Fatal("当前目录不是 Git 仓库根目录")
+		return os.ErrNotExist
 	}
 
 	hookPath := filepath.Join(gitDir, "hooks", "pre-commit")
@@ -91,10 +98,9 @@ exit 0
 `
 
 	if err := os.WriteFile(hookPath, []byte(hookContent), 0755); err != nil {
-		log.Fatal("安装 hook 失败: %v", err)
+		return err
 	}
-
-	log.Success("Hook 安装成功: %s", hookPath)
+	return nil
 }
 
 func runHookUninstall() {

cmd/codei18n/init.go

@@ -1,22 +1,29 @@
 package main
 
 import (
+	"os"
+	"strings"
+
 	"github.com/spf13/cobra"
 	"github.com/studyzy/codei18n/core/config"
+	"github.com/studyzy/codei18n/core/workflow"
 	"github.com/studyzy/codei18n/internal/log"
 )
 
 var (
-	initSourceLang string
-	initTargetLang string
-	initProvider   string
+	initSourceLang    string
+	initTargetLang    string
+	initProvider      string
+	initWithTranslate bool
+	initForce         bool
 )
 
 var initCmd = &cobra.Command{
 	Use:   "init",
 	Short: "初始化项目配置",
+	Long:  `初始化项目配置，继承全局设置，并自动配置 Git 环境。可选执行初次扫描和翻译。`,
 	Run: func(cmd *cobra.Command, args []string) {
-		runInit()
+		runInit(cmd)
 	},
 }
 
@@ -25,19 +32,112 @@ func init() {
 
 	initCmd.Flags().StringVar(&initSourceLang, "source-lang", "en", "源码语言")
 	initCmd.Flags().StringVar(&initTargetLang, "target-lang", "zh-CN", "本地目标语言")
-	// Default to using LLM-based remote translation (OpenAI-compatible protocol)
-	initCmd.Flags().StringVar(&initProvider, "provider", "openai", "翻译提供商 (openai/llm/ollama)")
+	initCmd.Flags().StringVar(&initProvider, "provider", "", "翻译提供商 (openai/llm/ollama)")
+	initCmd.Flags().BoolVar(&initWithTranslate, "with-translate", false, "初始化后立即执行翻译")
+	initCmd.Flags().BoolVar(&initForce, "force", false, "如果配置已存在，强制覆盖")
 }
 
-func runInit() {
-	cfg := config.DefaultConfig()
-	cfg.SourceLanguage = initSourceLang
-	cfg.LocalLanguage = initTargetLang
-	cfg.TranslationProvider = initProvider
+func runInit(cmd *cobra.Command) {
+	// 1. Load effective config (Global + Defaults)
+	cfg, err := config.LoadConfig()
+	if err != nil {
+		log.Warn("无法加载配置，使用默认值: %v", err)
+		cfg = config.DefaultConfig()
+	}
+
+	// 2. Apply flags
+	// Note: Flags defined in init() act as overrides or defaults.
+	// Cobra flags don't automatically populate cfg, we map them here.
+	// But we need to distinguish between "default value" and "user provided value".
+	// Since we set default values in flags, we might overwrite global config if we are not careful.
+	// Actually, `initSourceLang` defaults to "en". If global config has "fr", `initSourceLang` will still be "en" if user didn't provide it?
+	// No, cobra flags have default values.
+	// We should only override if user EXPLICITLY provided the flag.
+	// How to check if flag was changed? cmd.Flags().Changed("source-lang")
+	
+	if cmd.Flags().Changed("source-lang") {
+		cfg.SourceLanguage = initSourceLang
+	}
+	if cmd.Flags().Changed("target-lang") {
+		cfg.LocalLanguage = initTargetLang
+	}
+	if cmd.Flags().Changed("provider") {
+		cfg.TranslationProvider = initProvider
+	}
+
+	// 3. Sanitize (Strip secrets inherited from global)
+	// We create a new sanitized config object to save locally
+	projectCfg := cfg.Sanitize()
 
-	if err := config.SaveConfig(cfg); err != nil {
+	// 4. Save Config
+	if _, err := os.Stat(".codei18n/config.json"); err == nil && !initForce {
+		log.Fatal("配置文件已存在 (使用 --force 覆盖)")
+	}
+	
+	if err := config.SaveConfig(projectCfg); err != nil {
 		log.Fatal("保存配置失败: %v", err)
 	}
+	log.Success("项目配置文件已生成")
 
-	log.Success("项目初始化成功")
+	// 5. Git Integration
+	setupGitIntegration()
+
+	// 6. Map Update (Auto Scan)
+	log.Info("正在初始化注释映射...")
+	mapResult, err := workflow.MapUpdate(projectCfg, ".", false)
+	if err != nil {
+		log.Warn("映射初始化失败: %v", err)
+	} else {
+		log.Success("已扫描 %d 条注释，生成初始映射", mapResult.TotalComments)
+	}
+
+	// 7. Conditional Translation
+	if initWithTranslate {
+		log.Info("正在执行初次翻译...")
+		transResult, err := workflow.Translate(projectCfg, workflow.TranslateOptions{
+			Concurrency: 5,
+			// Use provider/model from config unless overridden (which we handled in step 2)
+		})
+		if err != nil {
+			log.Warn("翻译失败: %v", err)
+		} else {
+			log.Success("翻译完成！共处理 %d 条注释", transResult.SuccessCount)
+		}
+	} else {
+		log.Info("提示: 运行 'codei18n translate' 可生成翻译")
+	}
+	
+	log.Success("项目初始化完成！")
+}
+
+func setupGitIntegration() {
+	if _, err := os.Stat(".git"); os.IsNotExist(err) {
+		return // Not a git repo
+	}
+
+	// 1. Update .gitignore
+	gitignorePath := ".gitignore"
+	content, _ := os.ReadFile(gitignorePath) // ignore error, treat as empty if missing
+	sContent := string(content)
+	if !strings.Contains(sContent, ".codei18n/") {
+		f, err := os.OpenFile(gitignorePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
+		if err != nil {
+			log.Warn("更新 .gitignore 失败: %v", err)
+		} else {
+			if len(content) > 0 && !strings.HasSuffix(sContent, "\n") {
+				f.WriteString("\n")
+			}
+			f.WriteString(".codei18n/\n")
+			f.Close()
+			log.Success("已将 .codei18n/ 添加到 .gitignore")
+		}
+	}
+
+	// 2. Install hook
+	// Assuming installHook is available in package main (from hook.go)
+	if err := installHook(); err != nil {
+		log.Warn("安装 git hook 失败: %v", err)
+	} else {
+		log.Success("Git Pre-commit Hook 已自动安装")
+	}
 }

cmd/codei18n/map.go

@@ -8,7 +8,7 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/studyzy/codei18n/core/config"
 	"github.com/studyzy/codei18n/core/mapping"
-	"github.com/studyzy/codei18n/core/utils"
+	"github.com/studyzy/codei18n/core/workflow"
 	"github.com/studyzy/codei18n/internal/log"
 )
 
@@ -64,63 +64,17 @@ func runMapUpdate() {
 		cfg = config.DefaultConfig()
 	}
 
-	// 1. Scan for comments
-	log.Info("正在扫描目录: %s", mapScanDir)
-	comments, err := scanDirectory(mapScanDir)
+	result, err := workflow.MapUpdate(cfg, mapScanDir, mapDryRun)
 	if err != nil {
-		log.Fatal("扫描失败: %v", err)
+		log.Fatal("Map update failed: %v", err)
 	}
 
-	// 2. Generate IDs
-	for _, c := range comments {
-		if c.ID == "" {
-			c.ID = utils.GenerateCommentID(c)
-		}
-	}
-
-	// 3. Load Store
-	storePath := filepath.Join(".codei18n", "mappings.json")
-	store := mapping.NewStore(storePath)
-	if err := store.Load(); err != nil {
-		log.Fatal("加载映射文件失败: %v", err)
-	}
-
-	// 4. Update
-	m := store.GetMapping()
-	m.SourceLanguage = cfg.SourceLanguage
-	m.TargetLanguage = cfg.LocalLanguage
-
-	addedCount := 0
-	for _, c := range comments {
-		// Check if ID exists
-		if _, exists := m.Comments[c.ID]; !exists {
-			// [MOCK zh-CN->en] // Intelligently detect comment language
-			detectedLang := utils.DetectLanguage(c.SourceText)
-
-			if detectedLang == cfg.LocalLanguage {
-				// [MOCK zh-CN->en] // The comment is in the local language (Chinese), stored as LocalLanguage
-				store.Set(c.ID, cfg.LocalLanguage, c.SourceText)
-				log.Info("检测到中文注释: ID=%s, Text=%s", c.ID, c.SourceText)
-			} else {
-				// [MOCK zh-CN->en] // The comment is in the source language (English), stored as SourceLanguage
-				store.Set(c.ID, cfg.SourceLanguage, c.SourceText)
-			}
-			addedCount++
-		}
-	}
-
-	log.Success("发现 %d 条注释，新增 %d 条映射", len(comments), addedCount)
-
+	log.Success("发现 %d 条注释，新增 %d 条映射", result.TotalComments, result.AddedCount)
 	if mapDryRun {
 		log.Info("Dry run 模式，不保存文件")
-		return
-	}
-
-	// 5. Save
-	if err := store.Save(); err != nil {
-		log.Fatal("保存映射文件失败: %v", err)
+	} else {
+		log.Success("映射文件已更新: %s", result.StorePath)
 	}
-	log.Success("映射文件已更新: %s", storePath)
 }
 
 func runMapGet(commentID string) {

cmd/codei18n/scan.go

@@ -3,15 +3,14 @@ package main
 import (
 	"encoding/json"
 	"fmt"
-	"io"
 	"os"
 	"path/filepath"
 
 	"github.com/spf13/cobra"
-	"github.com/studyzy/codei18n/adapters"
 	"github.com/studyzy/codei18n/core/config"
 	"github.com/studyzy/codei18n/core/domain"
 	"github.com/studyzy/codei18n/core/mapping"
+	"github.com/studyzy/codei18n/core/scanner"
 	"github.com/studyzy/codei18n/core/utils"
 	"github.com/studyzy/codei18n/internal/log"
 )
@@ -60,11 +59,11 @@ func runScan() {
 
 	// Determine strategy
 	if scanStdin {
-		comments, err = scanFromStdin(scanFile)
+		comments, err = scanner.FromStdin(scanFile)
 	} else if scanFile != "" {
-		comments, err = scanSingleFile(scanFile)
+		comments, err = scanner.SingleFile(scanFile)
 	} else {
-		comments, err = scanDirectory(scanDir)
+		comments, err = scanner.Directory(scanDir)
 	}
 
 	if err != nil {
@@ -114,85 +113,6 @@ func runScan() {
 	}
 }
 
-func scanFromStdin(filename string) ([]*domain.Comment, error) {
-	// Read stdin
-	src, err := io.ReadAll(os.Stdin)
-	if err != nil {
-		return nil, fmt.Errorf("读取 stdin 失败: %w", err)
-	}
-
-	adapter, err := adapters.GetAdapter(filename)
-	if err != nil {
-		return nil, err
-	}
-	return adapter.Parse(filename, src)
-}
-
-func scanSingleFile(filename string) ([]*domain.Comment, error) {
-	adapter, err := adapters.GetAdapter(filename)
-	if err != nil {
-		return nil, err
-	}
-
-	// Read file content manually to ensure consistency with directory scan
-	// and to debug potential read issues
-	content, err := os.ReadFile(filename)
-	if err != nil {
-		return nil, fmt.Errorf("read file failed: %w", err)
-	}
-
-	return adapter.Parse(filename, content)
-}
-
-func scanDirectory(dir string) ([]*domain.Comment, error) {
-	var comments []*domain.Comment
-	var walkErr error
-
-	err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
-		if err != nil {
-			return err
-		}
-		if info.IsDir() {
-			// Skip .git, vendor, .codei18n
-			if info.Name() == ".git" || info.Name() == "vendor" || info.Name() == ".codei18n" {
-				return filepath.SkipDir
-			}
-			return nil
-		}
-
-		// Try to get adapter for file
-		adapter, err := adapters.GetAdapter(path)
-		if err == nil {
-			// Supported file
-			// Calculate relative path for ID stability
-			relPath, err := filepath.Rel(dir, path)
-			if err != nil {
-				relPath = path
-			}
-
-			// Read file content manually to ensure we access the correct file
-			content, err := os.ReadFile(path)
-			if err != nil {
-				log.Warn("读取文件 %s 失败: %v", path, err)
-				return nil
-			}
-
-			fileComments, err := adapter.Parse(relPath, content)
-			if err != nil {
-				log.Warn("解析文件 %s 失败: %v", path, err)
-				return nil // Continue scanning other files
-			}
-			comments = append(comments, fileComments...)
-		}
-		return nil
-	})
-
-	if walkErr != nil {
-		return nil, walkErr
-	}
-	return comments, err
-}
-
 func outputResults(comments []*domain.Comment) error {
 	// Prepare output data
 	type Output struct {