1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
| package cmd
|
| import (
| "fmt"
| "os"
|
| "github.com/spf13/cobra"
| )
|
| var (
| vaultPath string
| dbPath string
| )
|
| var rootCmd = &cobra.Command{
| Use: "kb",
| Short: "知识库 CLI 工具",
| Long: "kb-cli: 知识库搜索与管理工具，支持知识图谱搜索",
| }
|
| func Execute() {
| if err := rootCmd.Execute(); err != nil {
| fmt.Fprintln(os.Stderr, err)
| os.Exit(1)
| }
| }
|
| func init() {
| homeDir, _ := os.UserHomeDir()
| defaultVault := os.Getenv("KB_VAULT")
| if defaultVault == "" {
| defaultVault = homeDir + "/aisim/note/001/笔记001"
| }
| defaultDB := homeDir + "/.cache/kb-cli/kb.db"
|
| rootCmd.PersistentFlags().StringVar(&vaultPath, "vault", defaultVault, "知识库根目录")
| rootCmd.PersistentFlags().StringVar(&dbPath, "db", defaultDB, "索引数据库路径")
| }
|
|
