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
|
}
|
