package cmd
|
|
import (
|
"fmt"
|
"os"
|
"strings"
|
|
"github.com/aisim/kb-cli/internal/index"
|
"github.com/aisim/kb-cli/internal/search"
|
"github.com/aisim/kb-cli/internal/output"
|
"github.com/spf13/cobra"
|
)
|
|
var (
|
expanded []string
|
symptom []string
|
topN int
|
jsonOut bool
|
withContent bool
|
withLinks bool
|
)
|
|
var searchCmd = &cobra.Command{
|
Use: "search [keywords...]",
|
Short: "搜索知识库",
|
Long: `# search - 搜索
|
kb-cli search <关键词> [--top N] [--expanded <词>] [--symptom <词>] [--with-content] [--with-links] [--json] # 搜索知识库，支持关键词、扩展词、症状词`,
|
Args: cobra.MinimumNArgs(1),
|
RunE: runSearch,
|
}
|
|
func init() {
|
rootCmd.AddCommand(searchCmd)
|
searchCmd.Flags().StringSliceVar(&expanded, "expanded", nil, "扩展词（提升相关实体权重）")
|
searchCmd.Flags().StringSliceVar(&symptom, "symptom", nil, "症状词（针对具体症状）")
|
searchCmd.Flags().IntVar(&topN, "top", 10, "返回前 N 条结果")
|
searchCmd.Flags().BoolVar(&jsonOut, "json", false, "JSON 格式输出")
|
searchCmd.Flags().BoolVar(&withContent, "with-content", false, "返回完整文件内容")
|
searchCmd.Flags().BoolVar(&withLinks, "with-links", false, "显示关联文档链接")
|
|
// 参数不足时显示 help
|
searchCmd.SetUsageTemplate(searchCmd.Long)
|
}
|
|
func runSearch(cmd *cobra.Command, args []string) error {
|
// 打开索引
|
store, err := index.Open(dbPath)
|
if err != nil {
|
return fmt.Errorf("打开索引失败: %w", err)
|
}
|
defer store.Close()
|
|
// 检查是否需要重建索引
|
needsRebuild, commit, err := index.NeedsRebuild(store, vaultPath)
|
if err != nil {
|
return fmt.Errorf("检查索引状态失败: %w", err)
|
}
|
|
if needsRebuild {
|
fmt.Fprintln(os.Stderr, "索引过期，正在重建...")
|
if err := rebuildIndex(store, commit); err != nil {
|
return fmt.Errorf("重建索引失败: %w", err)
|
}
|
}
|
|
// 合并所有关键词（支持空格分隔和引号分隔）
|
var allKeywords []string
|
for _, arg := range args {
|
// 如果参数包含空格，按空格分割
|
parts := strings.Fields(arg)
|
allKeywords = append(allKeywords, parts...)
|
}
|
|
if len(allKeywords) == 0 {
|
return fmt.Errorf("请提供至少一个关键词")
|
}
|
|
// 执行搜索
|
opts := search.SearchOptions{
|
Expanded: expanded,
|
Symptom: symptom,
|
TopN: topN,
|
WithContent: withContent,
|
WithLinks: withLinks,
|
}
|
|
results, err := search.Search(store, allKeywords, opts)
|
if err != nil {
|
return fmt.Errorf("搜索失败: %w", err)
|
}
|
|
// 输出结果
|
if jsonOut {
|
jsonStr, err := output.FormatJSON(results)
|
if err != nil {
|
return err
|
}
|
fmt.Println(jsonStr)
|
} else {
|
fmt.Print(output.FormatTable(results))
|
}
|
|
return nil
|
}
|
