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, "索引数据库路径")
}