package cmd
|
|
import (
|
"fmt"
|
|
"github.com/aisim/kb-cli/internal/classify"
|
"github.com/spf13/cobra"
|
)
|
|
var reorgCmd = &cobra.Command{
|
Use: "reorg",
|
Short: "根据分类结果重组知识库目录",
|
Long: `# reorg - 目录重组
|
kb-cli reorg [--plan <文件>] [--dry-run] [--execute] # 根据分类结果重组知识库目录`,
|
RunE: runReorg,
|
}
|
|
var (
|
reorgPlan string
|
reorgDryRun bool
|
reorgExecute bool
|
)
|
|
func init() {
|
rootCmd.AddCommand(reorgCmd)
|
reorgCmd.Flags().StringVar(&reorgPlan, "plan", "classification.json", "分类结果文件")
|
reorgCmd.Flags().BoolVar(&reorgDryRun, "dry-run", false, "仅生成迁移计划，不执行移动")
|
reorgCmd.Flags().BoolVar(&reorgExecute, "execute", false, "执行迁移（需要明确确认）")
|
}
|
|
func runReorg(cmd *cobra.Command, args []string) error {
|
vaultPath := cmd.Flag("vault").Value.String()
|
|
fmt.Println("步骤 1/3: 读取分类结果...")
|
classification, err := classify.LoadClassification(vaultPath, reorgPlan)
|
if err != nil {
|
return fmt.Errorf("读取分类结果失败: %w", err)
|
}
|
fmt.Printf("✓ 读取 %d 个文档分类\n", len(classification.Items))
|
|
fmt.Println("\n步骤 2/3: 生成迁移计划...")
|
plan, err := classify.GenerateMigrationPlan(vaultPath, classification)
|
if err != nil {
|
return fmt.Errorf("生成迁移计划失败: %w", err)
|
}
|
|
if err := classify.SaveMigrationPlan(vaultPath, plan); err != nil {
|
return fmt.Errorf("保存迁移计划失败: %w", err)
|
}
|
fmt.Printf("✓ 已生成 migration-plan.json（%d 个文件）\n", len(plan.Migrations))
|
|
if reorgDryRun {
|
fmt.Println("\n[DRY-RUN] 迁移计划已生成，未执行移动")
|
fmt.Println("使用 --execute 执行迁移")
|
return nil
|
}
|
|
if !reorgExecute {
|
fmt.Println("\n警告: 未指定 --execute，迁移未执行")
|
fmt.Println("请检查 migration-plan.json，然后使用 --execute 执行")
|
return nil
|
}
|
|
fmt.Println("\n步骤 3/3: 执行迁移...")
|
if err := classify.ExecuteMigration(vaultPath, plan); err != nil {
|
return fmt.Errorf("执行迁移失败: %w", err)
|
}
|
fmt.Printf("✓ 迁移完成（%d 个文件）\n", len(plan.Migrations))
|
|
return nil
|
}
|
