mirror of
https://github.com/alvinunreal/tmuxai.git
synced 2026-03-02 18:23:51 +01:00
95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
// cli.go: Command-line interface for TmuxAI, including root command and flags
|
|
|
|
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/alvinunreal/tmuxai/config"
|
|
"github.com/alvinunreal/tmuxai/internal"
|
|
"github.com/alvinunreal/tmuxai/logger"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
initMessage string
|
|
taskFileFlag string
|
|
kbFlag string
|
|
modelFlag string
|
|
)
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "tmuxai [request message]",
|
|
Short: "TmuxAI - AI-Powered Tmux Companion",
|
|
Long: `TmuxAI - AI-Powered Tmux Companion`,
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {
|
|
if v, _ := cmd.Flags().GetBool("version"); v {
|
|
fmt.Printf("tmuxai version: %s\ncommit: %s\nbuild date: %s\n", internal.Version, internal.Commit, internal.Date)
|
|
os.Exit(0)
|
|
}
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
logger.Error("Error loading configuration: %v", err)
|
|
fmt.Fprintf(os.Stderr, "Error loading configuration: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if len(args) > 0 {
|
|
initMessage = strings.Join(args, " ")
|
|
}
|
|
|
|
if taskFileFlag != "" {
|
|
content, err := os.ReadFile(taskFileFlag)
|
|
if err != nil {
|
|
logger.Error("Error reading task file: %v", err)
|
|
fmt.Fprintf(os.Stderr, "Error reading task file: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
initMessage = string(content)
|
|
logger.Info("Read request from file: %s", taskFileFlag)
|
|
}
|
|
|
|
mgr, err := internal.NewManager(cfg)
|
|
if err != nil {
|
|
logger.Error("manager.NewManager failed: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Load knowledge bases from CLI flag
|
|
if kbFlag != "" {
|
|
kbNames := strings.Split(kbFlag, ",")
|
|
mgr.LoadKBsFromCLI(kbNames)
|
|
}
|
|
|
|
// Set model from CLI flag
|
|
if modelFlag != "" {
|
|
mgr.SetModelsDefault(modelFlag)
|
|
logger.Info("Set model from CLI flag: %s", modelFlag)
|
|
}
|
|
|
|
if initMessage != "" {
|
|
logger.Info("Starting with initial subcommand: %s", initMessage)
|
|
}
|
|
|
|
if err := mgr.Start(initMessage); err != nil {
|
|
logger.Error("manager.Start failed: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.Flags().StringVarP(&taskFileFlag, "file", "f", "", "Read request from specified file")
|
|
rootCmd.Flags().StringVar(&kbFlag, "kb", "", "Comma-separated list of knowledge bases to load (e.g., --kb docker,git)")
|
|
rootCmd.Flags().StringVar(&modelFlag, "model", "", "AI model configuration to use (e.g., --model gpt4)")
|
|
rootCmd.Flags().BoolP("version", "v", false, "Print version information")
|
|
}
|
|
|
|
func Execute() error {
|
|
return rootCmd.Execute()
|
|
}
|