package cmd

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/aisim/kb-cli/internal/llm"
	"github.com/aisim/kb-cli/internal/review"
	"github.com/spf13/cobra"
)

var reviewCmd = &cobra.Command{
	Use:   "review",
	Short: "审阅预览",
	Long: `# review - 审阅预览
kb-cli review <草稿目录> [--json] # 读取草稿和合并指示，使用 LLM 生成审阅预览`,
	Args: cobra.ExactArgs(1),
	RunE: runReview,
}

var reviewJSON bool

func init() {
	reviewCmd.Flags().BoolVar(&reviewJSON, "json", false, "JSON 格式输出")
	rootCmd.AddCommand(reviewCmd)
}

func runReview(cmd *cobra.Command, args []string) error {
	draftDir := args[0]

	// 读取草稿和合并指示
	preview, err := review.ReadDraftAndMerge(vaultPath, draftDir)
	if err != nil {
		return fmt.Errorf("读取草稿失败: %w", err)
	}

	// 如果有目标FAQ，用 LLM 生成3句话概要
	if preview.TargetFAQ != nil && preview.TargetFAQ.Title != "" {
		targetPath := preview.TargetFAQ.Path
		if !filepath.IsAbs(targetPath) {
			targetPath = filepath.Join(vaultPath, targetPath)
		}

		targetContent, err := os.ReadFile(targetPath)
		if err == nil && len(targetContent) > 0 {
			llmClient := llm.NewClient()
			summary, err := llmClient.GenerateReviewPreview(preview.TargetFAQ.Title, string(targetContent))
			if err != nil {
				fmt.Fprintf(os.Stderr, "警告: LLM 生成概要失败: %v\n", err)
			} else {
				preview.TargetFAQ.Summary = summary
			}
		}
	}

	// 输出
	if reviewJSON {
		output, err := review.FormatPreviewJSON(preview)
		if err != nil {
			return fmt.Errorf("格式化输出失败: %w", err)
		}
		fmt.Println(output)
	} else {
		// 表格输出
		fmt.Println(review.FormatPreviewTable(preview))

		// 如果有草稿正文，也展示主要内容
		expandedVault := expandHomePath(vaultPath)
		draftPath := filepath.Join(expandedVault, draftDir, "draft.md")
		draftContent, err := os.ReadFile(draftPath)
		if err == nil {
			content := extractMainContent(string(draftContent))
			if content != "" {
				fmt.Println("### 草稿主要内容\n")
				fmt.Println(content)
			}
		}
	}

	return nil
}

// extractMainContent 从 draft.md 中提取正文（去掉 frontmatter）
func extractMainContent(content string) string {
	if !strings.HasPrefix(content, "---") {
		return content
	}

	// 找到第二个 ---
	endIdx := strings.Index(content[3:], "\n---")
	if endIdx == -1 {
		return content
	}

	// 返回 frontmatter 之后的内容
	mainContent := content[endIdx+6:]
	return strings.TrimSpace(mainContent)
}

// expandHomePath 展开路径中的 ~
func expandHomePath(path string) string {
	if strings.HasPrefix(path, "~/") {
		homeDir, _ := os.UserHomeDir()
		return filepath.Join(homeDir, path[2:])
	}
	return path
}

// 保留 json 包引用
var _ = json.Marshal