102 lines
2.7 KiB
Go
102 lines
2.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"git-air/internal/git"
|
|
"git-air/internal/llm"
|
|
"git-air/internal/reviewer"
|
|
"git-air/internal/ui"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "git-air [git diff 参数，如 HEAD~1, main..feature, 文件名等]",
|
|
Short: "git-air: 极轻量、极速的 Git 原生 AI 代码评审工具",
|
|
Long: `git-air 是一个纯 Go 编写的 Git AI 评审利器。
|
|
支持一键评审暂存区、指定 Commit、分支差异或单文件，无缝兼容 Gemini、DeepSeek、Ollama 和 OpenAI。`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
ctx := cmd.Context()
|
|
|
|
// 1. 检查并获取 Diff
|
|
diff, err := git.GetDiff(ctx, args...)
|
|
if err != nil {
|
|
ui.PrintError("%v", err)
|
|
return
|
|
}
|
|
|
|
// 2. 加载配置
|
|
cfg, err := LoadConfig()
|
|
if err != nil {
|
|
ui.PrintError("加载配置失败: %v", err)
|
|
return
|
|
}
|
|
|
|
// 允许命令行临时覆盖模型/Key
|
|
if flagKey, _ := cmd.Flags().GetString("key"); flagKey != "" {
|
|
cfg.APIKey = flagKey
|
|
}
|
|
if flagModel, _ := cmd.Flags().GetString("model"); flagModel != "" {
|
|
cfg.Model = flagModel
|
|
}
|
|
if flagProvider, _ := cmd.Flags().GetString("provider"); flagProvider != "" {
|
|
cfg.Provider = flagProvider
|
|
}
|
|
|
|
// 3. 初始化大模型客户端
|
|
llmClient, err := llm.NewClient(llm.Config{
|
|
Provider: cfg.Provider,
|
|
APIKey: cfg.APIKey,
|
|
BaseURL: cfg.BaseURL,
|
|
Model: cfg.Model,
|
|
})
|
|
if err != nil {
|
|
ui.PrintError("初始化模型失败: %v (提示: 请先执行 'git air config set --key YOUR_KEY')", err)
|
|
return
|
|
}
|
|
|
|
// 4. 获取仓库根目录 (用于加载 .airules)
|
|
repoRoot, _ := git.GetRepoRoot(ctx)
|
|
|
|
// 5. 初始化评审引擎
|
|
customPrompt, _ := cmd.Flags().GetString("prompt")
|
|
if customPrompt == "" {
|
|
customPrompt = cfg.CustomPrompt
|
|
}
|
|
rev := reviewer.NewReviewer(llmClient, repoRoot, customPrompt)
|
|
|
|
// 6. 打印开场横幅并开始流式评审
|
|
ui.PrintBanner(cfg.Provider, cfg.Model)
|
|
printer := ui.NewStreamPrinter()
|
|
_, err = rev.Execute(ctx, diff, true, func(chunk string) {
|
|
printer.PrintChunk(chunk)
|
|
})
|
|
if err != nil {
|
|
ui.PrintError("\n评审过程中发生错误: %v", err)
|
|
return
|
|
}
|
|
|
|
ui.PrintFooter()
|
|
},
|
|
}
|
|
|
|
func Execute() {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
if err := rootCmd.ExecuteContext(ctx); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(configCmd)
|
|
rootCmd.AddCommand(hookCmd)
|
|
|
|
rootCmd.Flags().StringP("key", "k", "", "临时指定 API Key")
|
|
rootCmd.Flags().StringP("model", "m", "", "临时指定模型名称")
|
|
rootCmd.Flags().StringP("provider", "p", "", "临时指定 Provider (gemini, deepseek, ollama)")
|
|
rootCmd.Flags().String("prompt", "", "临时指定 System Prompt")
|
|
}
|