package cmd
|
|
import (
|
"fmt"
|
"os"
|
"strings"
|
|
"github.com/spf13/cobra"
|
"github.com/aisim/kb-cli/internal/llm"
|
)
|
|
var (
|
vaultPath string
|
dbPath string
|
)
|
|
// buildLong 为父命令动态生成 Long，格式：# cmd - Short\n子命令详细用法
|
func buildLong(cmd *cobra.Command) string {
|
var sb strings.Builder
|
sb.WriteString(fmt.Sprintf("# %s - %s\n", cmd.Name(), cmd.Short))
|
for _, sub := range cmd.Commands() {
|
if sub.Hidden {
|
continue
|
}
|
if sub.Long != "" {
|
sb.WriteString(sub.Long)
|
// 确保 Long 末尾有换行
|
if !strings.HasSuffix(sub.Long, "\n") {
|
sb.WriteString("\n")
|
}
|
} else {
|
sb.WriteString(fmt.Sprintf("%s # %s\n", sub.CommandPath(), sub.Short))
|
}
|
}
|
return sb.String()
|
}
|
|
var rootCmd = &cobra.Command{
|
Use: "kb-cli",
|
Short: "知识库搜索与管理工具",
|
Long: `kb-cli - 知识库搜索与管理工具
|
|
Global: [--vault <路径>] [--db <路径>]`,
|
}
|
|
func Execute() {
|
// 为每个父命令动态生成 Long（包含子命令详细用法）
|
for _, cmd := range rootCmd.Commands() {
|
if cmd.HasAvailableSubCommands() && !cmd.Hidden {
|
cmd.Long = buildLong(cmd)
|
}
|
}
|
|
// root Long = 头部 + 所有一级子命令的 Long
|
var sb strings.Builder
|
sb.WriteString(rootCmd.Long)
|
for _, cmd := range rootCmd.Commands() {
|
if cmd.Hidden {
|
continue
|
}
|
sb.WriteString("\n")
|
sb.WriteString(cmd.Long)
|
}
|
rootCmd.Long = sb.String()
|
|
if err := rootCmd.Execute(); err != nil {
|
// 如果是参数不足的错误，显示对应命令的 help
|
errMsg := err.Error()
|
if strings.Contains(errMsg, "arg(s)") || strings.Contains(errMsg, "received") || strings.Contains(errMsg, "required flag") {
|
// 找到当前命令并显示其 help
|
cmd, _, findErr := rootCmd.Find(os.Args[1:])
|
if findErr == nil {
|
cmd.Help()
|
}
|
fmt.Println() // 空行
|
} else {
|
cobra.CheckErr(err)
|
}
|
}
|
}
|
|
func init() {
|
homeDir, _ := os.UserHomeDir()
|
|
// 从配置文件读取默认值（优先级：配置文件 > 环境变量 > 内置默认值）
|
cfg := llm.LoadConfig()
|
defaultVault := cfg.KnowledgeBase.VaultPath
|
defaultDB := cfg.KnowledgeBase.DBPath
|
|
// 如果配置文件没有设置，使用环境变量或默认值
|
if defaultVault == "" {
|
defaultVault = os.Getenv("KB_VAULT")
|
if defaultVault == "" {
|
defaultVault = homeDir + "/aisim/note/001/笔记001"
|
}
|
}
|
if defaultDB == "" {
|
defaultDB = homeDir + "/.cache/kb-cli/kb.db"
|
}
|
|
rootCmd.PersistentFlags().StringVar(&vaultPath, "vault", defaultVault, "知识库根目录")
|
rootCmd.PersistentFlags().StringVar(&dbPath, "db", defaultDB, "索引数据库路径")
|
|
// 自定义 help 模板：只输出 Long
|
helpTemplate := `{{.Long}}`
|
rootCmd.SetUsageTemplate(helpTemplate)
|
rootCmd.SetHelpTemplate(helpTemplate)
|
|
rootCmd.SilenceUsage = true
|
rootCmd.SilenceErrors = true
|
}
|
