unclesam
7632652b7b
- 新增 `updater` 模块实现 GitHub Release 版本异步检查与本地缓存 - 新增 `git air update` 与 `git air version` 命令支持一键升级及版本检查 - 优化 CLI 终端交互，在首页渲染时提供友好的新版本升级提示卡片
102 lines
3.8 KiB
Go
102 lines
3.8 KiB
Go
package ui
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// ANSI 终端颜色常量
|
|
const (
|
|
colorReset = "\033[0m"
|
|
colorRed = "\033[1;31m"
|
|
colorGreen = "\033[1;32m"
|
|
colorYellow = "\033[1;33m"
|
|
colorBlue = "\033[1;34m"
|
|
colorCyan = "\033[1;36m"
|
|
colorGray = "\033[0;90m"
|
|
colorBold = "\033[1m"
|
|
)
|
|
|
|
// PrintBanner 打印开场横幅与当前评审状态
|
|
func PrintBanner(provider, model string) {
|
|
fmt.Printf("%s[git-air]%s 代码评审中... %s(Engine: %s / %s)%s\n", colorCyan, colorReset, colorGray, provider, model, colorReset)
|
|
fmt.Println(colorGray + strings.Repeat("─", 65) + colorReset)
|
|
}
|
|
|
|
// PrintFooter 打印结束分界线
|
|
func PrintFooter() {
|
|
fmt.Println()
|
|
fmt.Println(colorGray + strings.Repeat("─", 65) + colorReset)
|
|
}
|
|
|
|
// PrintUsage 打印 Token 消耗与费用估算
|
|
func PrintUsage(inputTokens, outputTokens int, costUSD float64) {
|
|
if inputTokens == 0 && outputTokens == 0 {
|
|
return
|
|
}
|
|
costStr := ""
|
|
if costUSD < 0 {
|
|
costStr = "费用未知"
|
|
} else if costUSD == 0 {
|
|
costStr = "本地免费"
|
|
} else {
|
|
costStr = fmt.Sprintf("≈ $%.4f", costUSD)
|
|
}
|
|
|
|
fmt.Printf("\n- %sToken: 输入 %s%d%s / 输出 %s%d%s | %s%s%s\n",
|
|
colorBold,
|
|
colorCyan, inputTokens, colorGray,
|
|
colorCyan, outputTokens, colorGray,
|
|
colorCyan, costStr, colorReset,
|
|
)
|
|
}
|
|
|
|
// PrintUpdateBanner 打印新版本升级提醒卡片
|
|
func PrintUpdateBanner(currentVer, latestVer, releaseURL string) {
|
|
fmt.Println()
|
|
fmt.Println(colorYellow + "┌─────────────────────────────────────────────────────────────┐" + colorReset)
|
|
fmt.Printf(colorYellow+"│"+colorReset+" 🚀 %sgit-air 新版本发布: %s%s%s → %s%s%s\n",
|
|
colorBold, colorGray, currentVer, colorReset,
|
|
colorGreen, latestVer, colorReset,
|
|
)
|
|
fmt.Println(colorYellow + "│" + colorReset + " 运行以下命令一键升级: " + colorYellow + "│" + colorReset)
|
|
fmt.Printf(colorYellow+"│"+colorReset+" %sgo install github.com/unclesam-ly/git-air@latest%s "+colorYellow+"│"+colorReset+"\n", colorCyan, colorReset)
|
|
fmt.Println(colorYellow + "└─────────────────────────────────────────────────────────────┘" + colorReset)
|
|
}
|
|
|
|
// PrintError 打印错误信息
|
|
func PrintError(format string, a ...interface{}) {
|
|
msg := fmt.Sprintf(format, a...)
|
|
_, _ = fmt.Fprintf(os.Stderr, "%s[ERROR]%s %s\n", colorRed, colorReset, msg)
|
|
}
|
|
|
|
// PrintSuccess 打印成功信息
|
|
func PrintSuccess(format string, a ...interface{}) {
|
|
msg := fmt.Sprintf(format, a...)
|
|
fmt.Printf("%s[SUCCESS]%s %s\n", colorGreen, colorReset, msg)
|
|
}
|
|
|
|
// StreamPrinter 带有标签高亮的流式输出打印器
|
|
type StreamPrinter struct{}
|
|
|
|
func NewStreamPrinter() *StreamPrinter {
|
|
return &StreamPrinter{}
|
|
}
|
|
|
|
// PrintChunk 逐字流式输出，遇到关键标签进行颜色增强
|
|
func (p *StreamPrinter) PrintChunk(chunk string) {
|
|
// 对关键标签进行颜色替换
|
|
highlighted := chunk
|
|
highlighted = strings.ReplaceAll(highlighted, "[BLOCKER]", colorRed+"[BLOCKER]"+colorReset)
|
|
highlighted = strings.ReplaceAll(highlighted, "[CRITICAL]", colorRed+"[CRITICAL]"+colorReset)
|
|
highlighted = strings.ReplaceAll(highlighted, "[WARNING]", colorYellow+"[WARNING]"+colorReset)
|
|
highlighted = strings.ReplaceAll(highlighted, "[WARN]", colorYellow+"[WARN]"+colorReset)
|
|
highlighted = strings.ReplaceAll(highlighted, "[SUGGESTION]", colorBlue+"[SUGGESTION]"+colorReset)
|
|
highlighted = strings.ReplaceAll(highlighted, "[PASS]", colorGreen+"[PASS]"+colorReset)
|
|
highlighted = strings.ReplaceAll(highlighted, "[LGTM]", colorGreen+"[LGTM]"+colorReset)
|
|
highlighted = strings.ReplaceAll(highlighted, "[REJECT]", colorRed+"[REJECT]"+colorReset)
|
|
|
|
fmt.Print(highlighted)
|
|
}
|