feat: Translate save process --story=1

Author: studyzy

cmd/codei18n/init.go

@@ -54,7 +54,7 @@ func runInit(cmd *cobra.Command) {
 	// 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
 	}
@@ -73,7 +73,7 @@ func runInit(cmd *cobra.Command) {
 	if _, err := os.Stat(".codei18n/config.json"); err == nil && !initForce {
 		log.Fatal("配置文件已存在 (使用 --force 覆盖)")
 	}
-	
+
 	if err := config.SaveConfig(projectCfg); err != nil {
 		log.Fatal("保存配置失败: %v", err)
 	}
@@ -106,7 +106,7 @@ func runInit(cmd *cobra.Command) {
 	} else {
 		log.Info("提示: 运行 'codei18n translate' 可生成翻译")
 	}
-	
+
 	log.Success("项目初始化完成！")
 }
 

core/workflow/translate.go

@@ -175,7 +175,6 @@ func Translate(cfg *config.Config, opts TranslateOptions) (*TranslateResult, err
 			}
 
 			countMu.Lock()
-			defer countMu.Unlock()
 
 			if err != nil {
 				failCount += len(currentBatch)
@@ -186,8 +185,14 @@ func Translate(cfg *config.Config, opts TranslateOptions) (*TranslateResult, err
 					store.Set(t.id, t.toLang, res)
 					successCount++
 				}
+				// Save progress immediately
+				if err := store.Save(); err != nil {
+					log.Warn("保存进度失败: %v", err)
+				}
 			}
 
+			countMu.Unlock()
+
 			s.Suffix = fmt.Sprintf(" 正在翻译... (%d/%d 成功, %d 失败)", successCount, len(tasks), failCount)
 		}(batch)
 	}

tests/init_integration_test.go

@@ -21,7 +21,7 @@ func TestInitEnhancedFlow(t *testing.T) {
 
 	// 1. Setup a fake Git repo
 	require.NoError(t, exec.Command("git", "init", tempDir).Run())
-	
+
 	// Create a sample file
 	sampleContent := `package main
 // Test Init
@@ -51,7 +51,7 @@ func main() {}`
 	// JSON marshaling escapes '>', so we unmarshal to verify content properly
 	var mappingData map[string]interface{}
 	require.NoError(t, json.Unmarshal(mappingContent, &mappingData))
-	
+
 	comments := mappingData["comments"].(map[string]interface{})
 	var foundTranslation bool
 	for _, v := range comments {
@@ -91,7 +91,7 @@ func TestInitGlobalConfigInheritance(t *testing.T) {
 	// Create a fake global config in a subfolder
 	fakeHome := filepath.Join(tempDir, "fake_home")
 	os.MkdirAll(filepath.Join(fakeHome, ".codei18n"), 0755)
-	
+
 	globalConfig := `{
 		"sourceLanguage": "fr",
 		"localLanguage": "de",
@@ -103,7 +103,7 @@ func TestInitGlobalConfigInheritance(t *testing.T) {
 	}`
 	os.WriteFile(filepath.Join(fakeHome, ".codei18n", "config.json"), []byte(globalConfig), 0644)
 
-	// Run init in a project dir, pointing to fake config via --config flag? 
+	// Run init in a project dir, pointing to fake config via --config flag?
 	// The init command logic uses config.LoadConfig() which looks at HOME.
 	// We cannot easily mock HOME env var for the subprocess unless we set it.
 	projectDir := filepath.Join(tempDir, "project")
@@ -113,14 +113,14 @@ func TestInitGlobalConfigInheritance(t *testing.T) {
 	cmd.Dir = projectDir
 	// Mock HOME to point to fake_home
 	cmd.Env = append(os.Environ(), "HOME="+fakeHome)
-	
+
 	out, err := cmd.CombinedOutput()
 	require.NoError(t, err, string(out))
 
 	// Verify project config
 	projectConfigPath := filepath.Join(projectDir, ".codei18n", "config.json")
 	require.FileExists(t, projectConfigPath)
-	
+
 	content, _ := os.ReadFile(projectConfigPath)
 	sContent := string(content)