feat: 实现输出格式化

ai_xiaopei

2026-07-25 09c2a03cefa687832257d3b816da50db3cfcc0f2

feat: 实现输出格式化

 internal/output/formatter.go

New file
@@ -0,0 +1,40 @@
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))
    }

    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
}

 internal/output/formatter_test.go

New file
@@ -0,0 +1,41 @@
package output

import (
    "strings"
    "testing"

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

func TestFormatTable(t *testing.T) {
    results := []search.SearchResult{
        {Path: "FAQ/充装类/001.md", Title: "充装问题", Section: "FAQ", Score: 100},
    }
    output := FormatTable(results)
    if !strings.Contains(output, "FAQ") {
        t.Error("output should contain section")
    }
    if !strings.Contains(output, "100") {
        t.Error("output should contain score")
    }
}

func TestFormatTableEmpty(t *testing.T) {
    output := FormatTable(nil)
    if output != "未找到匹配结果" {
        t.Errorf("output = %q, want %q", output, "未找到匹配结果")
    }
}

func TestFormatJSON(t *testing.T) {
    results := []search.SearchResult{
        {Path: "test.md", Title: "Test", Score: 50},
    }
    output, err := FormatJSON(results)
    if err != nil {
        t.Fatalf("FormatJSON failed: %v", err)
    }
    if !strings.Contains(output, "test.md") {
        t.Error("JSON should contain path")
    }
}