cli.go

package main

import (
	"fmt"
	"os"
	"os/exec"

	"github.com/spf13/cobra"
)

func BuildCli() *cobra.Command {
	rootCmd := &cobra.Command{
		Use:           "shard",
		Short:         "A lightweight and type-safe MCP server for Obsidian that uses Obsidian CLI.",
		SilenceErrors: true,
		SilenceUsage:  true,
	}

	checkCmd := &cobra.Command{
		Use:   "check",
		Short: "Check if Shard can detect Obsidian CLI",
		RunE: func(cmd *cobra.Command, args []string) error {
			executable := GetDefaultExecutableName()
			localExecutable := "." + string(os.PathSeparator) + executable
			if resolved, err := exec.LookPath(localExecutable); err == nil {
				executable = resolved
			} else if resolved, err := exec.LookPath(executable); err == nil {
				executable = resolved
			} else {
				return fmt.Errorf("obsidian CLI %q not found in current directory or PATH", executable)
			}

			fmt.Printf("Found the CLI by path '%s'.", executable)
			return nil
		},
	}

	startCmd := &cobra.Command{
		Use:   "start",
		Short: "Start an MCP server.",
		RunE: func(cmd *cobra.Command, args []string) error {
			return StartServer()
		},
	}

	versionCmd := &cobra.Command{
		Use:   "version",
		Short: "Print version of Shard",
		Run: func(cmd *cobra.Command, args []string) {
			fmt.Println(VERSION)
		},
	}

	rootCmd.AddCommand(checkCmd, startCmd, versionCmd)

	return rootCmd
}