package output

import (
	"encoding/json"
	"fmt"
	"strings"

	"github.com/aisim/kb-cli/internal/search"
)

// FormatTable 表格格式输出
func FormatTable(results []search.SearchResult) string {
	if len(results) == 0 {
		return "未找到匹配结果"
	}

	var sb strings.Builder
	sb.WriteString(fmt.Sprintf("%-50s %-20s %-8s %s\n", "路径", "标题", "板块", "得分"))
	sb.WriteString(strings.Repeat("-", 100) + "\n")

	for _, r := range results {
		title := r.Title
		if len(title) > 18 {
			title = title[:18] + ".."
		}
		sb.WriteString(fmt.Sprintf("%-50s %-20s %-8s %d\n", r.Path, title, r.Section, r.Score))

		// 输出内容（如果有）
		if r.Content != "" {
			sb.WriteString("\n--- 内容 ---\n")
			sb.WriteString(r.Content)
			sb.WriteString("\n")
		}

		// 输出关联文档（如果有）
		if len(r.Links) > 0 {
			sb.WriteString("\n--- 关联文档 ---\n")
			for _, link := range r.Links {
				sb.WriteString(fmt.Sprintf("- %s\n", link))
			}
		}
	}

	sb.WriteString(fmt.Sprintf("\n共 %d 条结果\n", len(results)))
	return sb.String()
}

// FormatJSON JSON 格式输出
func FormatJSON(results []search.SearchResult) (string, error) {
	data, err := json.MarshalIndent(results, "", "  ")
	if err != nil {
		return "", err
	}
	return string(data), nil
}