Compare commits
22 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 98f2c69151 | |||
| 217db90cfa | |||
| 18b191d10d | |||
| 1996e60567 | |||
| 3c45730751 | |||
| b35508e623 | |||
| 7539987877 | |||
| 5fb0d5c58b | |||
| 59f9c6de18 | |||
| 8c6b08cec8 | |||
| aefa0e8799 | |||
| 7321951d05 | |||
| 6f872ff285 | |||
| 1787464f52 | |||
| b71f76c8b3 | |||
| ceed482444 | |||
| b752779d18 | |||
| 6807371c5e | |||
| 1bce2d9c7a | |||
| 24ba405d55 | |||
| e18fbad839 | |||
| 7c8a924984 |
120
AGENTS.md
120
AGENTS.md
@@ -3,6 +3,17 @@
|
||||
## 项目概述
|
||||
YOYO是一个命令行翻译工具,使用Go语言编写,采用面向对象设计模式。它通过调用在线大模型API,结合不同的Prompt配置,实现多样化的翻译特色。
|
||||
|
||||
## 开发流程
|
||||
1. AI接手项目后,先阅读 `AGENTS.md` `changelog.md` `taolun.md` `memory.md`,无需读整个项目的所有文件;
|
||||
2. 每次开始执行操作前,先更新 `taolun.md`;
|
||||
3. 执行完开发后,更新 `changelog.md` `memory.md` `AGENTS.md`;
|
||||
4. 最后,执行 git 操作,发布本地的 版本号,而先不提交远程;
|
||||
|
||||
## 文档规范
|
||||
- 讨论的内容:只应该保存在 `taolun.md` 或 `changelod.md`中,根据 文档该写什么该链接什么,分门别类的放置
|
||||
- 不应该再创建其他任何 md 类说明文件。
|
||||
|
||||
|
||||
## OOP设计模式
|
||||
|
||||
### 核心类设计
|
||||
@@ -24,6 +35,32 @@ YOYO是一个命令行翻译工具,使用Go语言编写,采用面向对象
|
||||
### 1. 全局配置类 (Config)
|
||||
负责读取YAML配置文件,提供默认值。
|
||||
|
||||
### 2. 缓存类 (Cache)
|
||||
负责本地缓存管理,减少API调用。
|
||||
|
||||
```go
|
||||
// internal/cache/cache.go
|
||||
package cache
|
||||
|
||||
// Cache 缓存接口
|
||||
type Cache interface {
|
||||
Get(ctx context.Context, key string) (*CacheEntry, error)
|
||||
Set(ctx context.Context, entry *CacheEntry) error
|
||||
Delete(ctx context.Context, key string) error
|
||||
Clear(ctx context.Context) error
|
||||
Stats(ctx context.Context) (*CacheStats, error)
|
||||
Cleanup(ctx context.Context) error
|
||||
Close() error
|
||||
}
|
||||
```
|
||||
|
||||
**缓存功能特点**:
|
||||
- **存储层**: 使用SQLite数据库
|
||||
- **缓存键**: 基于原文+语言对的SHA256哈希
|
||||
- **清理策略**: 组合策略(数量限制+时间过期)
|
||||
- **异步保存**: 不阻塞翻译结果返回
|
||||
- **自动清理**: 定时清理过期缓存
|
||||
|
||||
```go
|
||||
// internal/config/config.go
|
||||
package config
|
||||
@@ -405,6 +442,12 @@ yoyo/
|
||||
│ ├── translator/ # 核心翻译
|
||||
│ │ ├── translator.go
|
||||
│ │ └── prompt.go
|
||||
│ ├── cache/ # 本地缓存
|
||||
│ │ ├── cache.go # 缓存接口
|
||||
│ │ ├── sqlite.go # SQLite实现
|
||||
│ │ ├── key.go # 缓存键生成
|
||||
│ │ ├── cleanup.go # 缓存清理
|
||||
│ │ └── cache_test.go # 单元测试
|
||||
│ └── prompt/ # Prompt管理
|
||||
├── pkg/ # 公共工具
|
||||
├── configs/ # 配置文件目录
|
||||
@@ -456,6 +499,13 @@ prompts:
|
||||
creative: "你是一位富有创造力的翻译家,请用优美流畅的语言翻译以下内容。"
|
||||
academic: "你是一位学术翻译专家,请用严谨的学术语言翻译以下内容。"
|
||||
simple: "请用简单易懂的语言翻译以下内容。"
|
||||
|
||||
# 缓存配置
|
||||
cache:
|
||||
enabled: true # 是否启用缓存
|
||||
max_records: 10000 # 最大缓存记录数
|
||||
expire_days: 30 # 缓存过期天数
|
||||
db_path: "~/.config/yoyo/cache.db" # 缓存数据库文件路径
|
||||
```
|
||||
|
||||
## 开发顺序建议
|
||||
@@ -611,6 +661,17 @@ go run ./cmd/yoyo "This is translation content..."
|
||||
|
||||
# 指定翻译模式
|
||||
./yoyo --mode=technical "API documentation text"
|
||||
|
||||
# 管道功能(与其他命令行工具联合使用)
|
||||
cat file.txt | ./yoyo # 翻译文件内容
|
||||
echo "Hello" | ./yoyo --lang=cn # 翻译命令输出
|
||||
./yoyo "Hello" | grep "你好" # 与其他命令组合
|
||||
cat file.txt | ./yoyo -q # 静默模式,只输出翻译结果
|
||||
|
||||
# 缓存管理命令
|
||||
./yoyo cache clear # 清空翻译缓存
|
||||
./yoyo cache stats # 查看缓存统计信息
|
||||
./yoyo cache cleanup # 清理过期缓存
|
||||
```
|
||||
|
||||
## 代码风格指南
|
||||
@@ -761,7 +822,64 @@ A: 使用指数退避重试,并在`internal/api/`中实现限流器。
|
||||
### Q: 如何支持更多语言?
|
||||
A: 在配置文件中添加语言映射,并更新翻译逻辑。
|
||||
|
||||
## 语言代码处理
|
||||
|
||||
### 支持的语言代码格式
|
||||
项目支持多种语言代码格式,通过 `internal/lang` 模块处理:
|
||||
|
||||
1. **标准BCP47格式**: `zh-CN`, `zh-TW`, `en-US`, `en-GB`, `ja`, `ko` 等
|
||||
2. **简短别名**: `cn`(中文), `en`(英文), `jp`(日文), `kr`(韩文) 等
|
||||
3. **中文名称**: `chinese`(中文), `english`(英文), `japanese`(日文) 等
|
||||
|
||||
### 语言解析函数
|
||||
```go
|
||||
// 解析语言代码
|
||||
lang.ParseLanguageCode("cn") // 返回 "zh-CN"
|
||||
lang.ParseLanguageCode("en") // 返回 "en-US"
|
||||
lang.ParseLanguageCode("zh-TW") // 返回 "zh-TW"
|
||||
|
||||
// 获取语言名称(用于显示)
|
||||
lang.GetLanguageName("zh-CN") // 返回 "中文(简体)"
|
||||
lang.GetLanguageName("en-US") // 返回 "English (US)"
|
||||
```
|
||||
|
||||
## Onboard配置向导
|
||||
|
||||
### 配置流程
|
||||
1. 选择主要翻译厂商
|
||||
2. 配置厂商API密钥、HOST、模型
|
||||
3. 设置全局配置(默认语言、超时)
|
||||
4. 保存配置到 `configs/config.yaml`
|
||||
|
||||
### 使用方法
|
||||
```bash
|
||||
yoyo onboard # 启动配置向导
|
||||
yoyo onboard --force # 强制重新配置
|
||||
```
|
||||
|
||||
### 配置向导实现
|
||||
- 使用 `github.com/AlecAivazis/survey/v2` 实现交互式界面
|
||||
- 支持厂商选择、API配置、语言设置
|
||||
- 生成标准YAML配置文件
|
||||
|
||||
## 分阶段迁移策略
|
||||
|
||||
### 第一阶段:开发阶段(当前)
|
||||
- API密钥存储在 `.env` 文件
|
||||
- 复杂配置存储在 `configs/config.yaml`
|
||||
- 支持环境变量替换
|
||||
|
||||
### 第二阶段:上线前
|
||||
- 实现配置文件路径查找机制
|
||||
- 支持用户配置目录 `~/.config/yoo/yoo.yml`
|
||||
- 提供配置迁移工具
|
||||
|
||||
### 第三阶段:最终优化
|
||||
- 移除对 `.env` 文件依赖
|
||||
- 完全使用配置文件
|
||||
|
||||
## 参考资源
|
||||
- [Effective Go](https://go.dev/doc/effective_go)
|
||||
- [Go Code Review Comments](https://github.com/golang/go/wiki/CodeReviewComments)
|
||||
- [Go Style Guide](https://google.github.io/styleguide/go/)
|
||||
- [Go Style Guide](https://google.github.io/styleguide/go/)
|
||||
- [Survey库文档](https://github.com/AlecAivazis/survey)
|
||||
|
||||
324
changelog.md
324
changelog.md
@@ -9,7 +9,7 @@
|
||||
|
||||
## 未来架构想法
|
||||
- [ ] 支持流式翻译输出
|
||||
- [ ] 添加本地缓存减少API调用
|
||||
- [x] 添加本地缓存减少API调用 ✅ 已完成
|
||||
- [ ] 实现插件系统支持自定义厂商
|
||||
- [ ] 支持批量翻译文件
|
||||
- [ ] 添加Web界面(可选)
|
||||
@@ -24,14 +24,307 @@
|
||||
- [ ] 翻译历史记录
|
||||
|
||||
## 当前正实现
|
||||
- [ ] 项目基础架构搭建
|
||||
- [ ] 核心类设计实现
|
||||
- [x] TUI界面模块拆分计划 ✅ 已制定
|
||||
- [x] 模块1: TUI框架搭建 ✅ 已完成
|
||||
- [x] 模块2: 输入组件 ✅ 已完成
|
||||
- [x] 模块3: 翻译显示区 ✅ 已完成
|
||||
- [x] 模块4: 状态栏/主题 ✅ 已完成
|
||||
- [x] 模块5: 快捷键系统 ✅ 已完成
|
||||
- [x] 模块6: 集成翻译 ✅ 已完成
|
||||
|
||||
## TUI界面实现计划 (v0.6.0) - 全部完成!
|
||||
| 步骤 | 模块 | 内容 | 状态 |
|
||||
|------|------|------|------|
|
||||
| 1 | TUI框架搭建 | bubbletea基础App结构、运行循环 | ✅ 已完成 |
|
||||
| 2 | 输入组件 | 文本输入框,光标、基础编辑 | ✅ 已完成 |
|
||||
| 3 | 翻译显示区 | 结果展示、格式化、滚动 | ✅ 已完成 |
|
||||
| 4 | 状态栏/主题 | 底部状态栏、语言选择、主题配色 | ✅ 已完成 |
|
||||
| 5 | 快捷键系统 | 退出、清空、切换语言等 | ✅ 已完成 |
|
||||
| 6 | 集成翻译 | 对接现有Translator、加载动画 | ✅ 已完成 |
|
||||
|
||||
## TUI界面重构计划 (v0.8.0) - 聊天风格
|
||||
| 步骤 | 模块 | 内容 | 状态 |
|
||||
|------|------|------|------|
|
||||
| 1 | 模块结构拆分 | 创建 view/components/styles/keys 子目录 | ✅ 已完成 |
|
||||
| 2 | 消息数据结构 | ChatMessage, ChatGroup 结构定义 | ✅ 已完成 |
|
||||
| 3 | 消息列表组件 | 可滚动的消息历史展示 (viewport) | ✅ 已完成 |
|
||||
| 4 | 原文+译文样式 | 区分显示用户输入和翻译结果 | ✅ 已完成 |
|
||||
| 5 | 固定底部输入框 | textarea + Ctrl+J 换行 + 固定高度5行 | ✅ 已完成 |
|
||||
| 6 | 状态栏 | 完整信息显示 | ✅ 已完成 |
|
||||
| 7 | 翻译逻辑集成 | 对接 Translator | ✅ 已完成 |
|
||||
| 8 | 输入框背景色 | 使用 FocusedStyle/BlurredStyle 设置背景 | ✅ 已完成 |
|
||||
| 9 | 输入框修复 | 修复Ctrl+J换行后第一行被遮住的问题 | ✅ 已完成 |
|
||||
|
||||
## TUI界面改进计划 (v0.7.0)
|
||||
| 步骤 | 模块 | 内容 | 状态 |
|
||||
|------|------|------|------|
|
||||
| 7 | 多行输入 | textarea组件替换textinput | ✅ 已完成 |
|
||||
| 8 | 弹出框组件 | 通用modal组件 | ⏳ 待实现 |
|
||||
| 9 | 斜杠命令菜单 | / 触发命令选择器,模糊匹配 | ⏳ 待实现 |
|
||||
| 10 | 翻译结果滚动 | viewport组件支持长文本 | ⏳ 待实现 |
|
||||
| 11 | 复制功能 | clipboard集成 | ⏳ 待实现 |
|
||||
| 12 | 状态栏扩展 | 显示耗时、token用量 | ⏳ 待实现 |
|
||||
|
||||
## 待修复BUG
|
||||
- 无
|
||||
|
||||
## 版本历史
|
||||
|
||||
### 0.7.0 (2026-04-06) - TUI界面改进
|
||||
**类型**: 功能版本
|
||||
**状态**: 开发中
|
||||
|
||||
**改进内容**:
|
||||
- ✅ 模块7: 多行输入 - textarea组件替换textinput
|
||||
- ✅ 模块7补充: 布局和样式优化 - 全宽/自适应高度/深色背景
|
||||
- ⏳ 模块8: 弹出框组件 - 通用modal
|
||||
- ⏳ 模块9: 斜杠命令菜单 - / 命令选择器
|
||||
- ⏳ 模块10: 翻译结果滚动 - viewport
|
||||
- ⏳ 模块11: 复制功能 - clipboard
|
||||
- ⏳ 模块12: 状态栏扩展 - 耗时/token
|
||||
|
||||
**讨论记录**:
|
||||
- [TUI界面改进计划](taolun.md#2026-04-06-1400-版本-070---tui界面改进计划)
|
||||
|
||||
**下一步**:
|
||||
- 实现模块8: 弹出框组件
|
||||
|
||||
---
|
||||
|
||||
### 0.6.0 (2026-04-06) - TUI交互界面
|
||||
**类型**: 功能版本
|
||||
**状态**: 已完成
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 模块1: TUI框架搭建 - 添加bubbletea依赖,实现基础App结构
|
||||
- ✅ 模块2: 输入组件 - textinput组件、基础输入处理
|
||||
- ✅ 模块3: 翻译显示区 - 结果显示区域、样式定义
|
||||
- ✅ 模块4: 状态栏/主题 - 底部状态栏、语言显示、配色完善
|
||||
- ✅ 模块5: 快捷键系统 - Ctrl+L清空、Ctrl+T切换语言
|
||||
- ✅ 模块6: 集成翻译 - Enter触发翻译、异步执行、加载状态、错误处理
|
||||
|
||||
**技术实现**:
|
||||
- 使用 `github.com/charmbracelet/bubbletea` v1.3.10
|
||||
- 使用 `github.com/charmbracelet/bubbles` (textinput组件)
|
||||
- 使用 `github.com/charmbracelet/lipgloss` v1.1.0
|
||||
- 基础model结构: config、translator字段
|
||||
|
||||
**讨论记录**:
|
||||
- [TUI界面模块拆分计划](taolun.md#2026-04-06-1000-版本-060---tui界面模块拆分计划)
|
||||
|
||||
**下一步**:
|
||||
- 实现模块7: 多行输入
|
||||
|
||||
---
|
||||
|
||||
### 0.5.0 (2026-03-29) - 本地缓存功能
|
||||
**类型**: 功能版本
|
||||
**状态**: 已发布
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 添加本地缓存模块 (internal/cache/)
|
||||
- ✅ 实现SQLite缓存存储层,支持高效查询和存储
|
||||
- ✅ 实现缓存键生成策略(基于原文+语言对的SHA256哈希)
|
||||
- ✅ 修改Translator类集成缓存功能,先查缓存再调用API
|
||||
- ✅ 添加缓存配置到Config结构,支持自定义缓存参数
|
||||
- ✅ 实现缓存管理命令:cache clear, cache stats, cache cleanup
|
||||
- ✅ 添加组合缓存清理策略(数量限制+时间过期)
|
||||
- ✅ 更新配置文件模板,添加缓存配置示例
|
||||
- ✅ 更新帮助文档,添加缓存相关命令说明
|
||||
|
||||
**技术实现**:
|
||||
- 使用 `github.com/mattn/go-sqlite3` 作为SQLite驱动
|
||||
- 实现缓存接口 (Cache interface),支持多种存储后端
|
||||
- 缓存表包含完整字段:原文、译文、语言对、模型、Prompt、用量统计
|
||||
- 自动清理过期缓存和超出数量限制的缓存
|
||||
- 异步保存缓存,不阻塞翻译结果返回
|
||||
|
||||
**使用示例**:
|
||||
```bash
|
||||
# 基本翻译(自动使用缓存)
|
||||
yoyo "Hello world"
|
||||
yoyo "Hello world" # 第二次调用将从缓存返回
|
||||
|
||||
# 缓存管理命令
|
||||
yoyo cache clear # 清空翻译缓存
|
||||
yoyo cache stats # 查看缓存统计信息
|
||||
yoyo cache cleanup # 清理过期缓存
|
||||
```
|
||||
|
||||
**配置示例**:
|
||||
```yaml
|
||||
cache:
|
||||
enabled: true # 是否启用缓存
|
||||
max_records: 10000 # 最大缓存记录数
|
||||
expire_days: 30 # 缓存过期天数
|
||||
db_path: "~/.config/yoyo/cache.db" # 缓存数据库文件路径
|
||||
```
|
||||
|
||||
**讨论记录**:
|
||||
- [本地缓存功能设计](taolun.md#2026-03-29-1500-版本-050-本地缓存功能设计)
|
||||
|
||||
**下一步**:
|
||||
- 完善缓存功能测试
|
||||
- 添加缓存命中率统计
|
||||
- 实现按语言清理缓存功能
|
||||
- 优化缓存性能
|
||||
|
||||
---
|
||||
|
||||
### 0.5.1 (2026-03-29) - 缓存功能修复
|
||||
**类型**: 修复版本
|
||||
**状态**: 已发布
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 修复缓存清空命令中的VACUUM事务错误
|
||||
- ✅ 修复缓存统计中的NULL值处理错误
|
||||
- ✅ 修复缓存过期清理策略,支持expire_days=0时清理所有记录
|
||||
- ✅ 添加缓存模块单元测试
|
||||
- ✅ 更新版本号到v0.5.1
|
||||
|
||||
**修复问题**:
|
||||
- 缓存清空命令执行时出现"cannot VACUUM from within a transaction"错误
|
||||
- 缓存统计查询在空表时出现NULL值转换错误
|
||||
- 缓存过期清理策略在expire_days=0时不工作
|
||||
|
||||
**测试结果**:
|
||||
- 所有缓存模块测试通过
|
||||
- 缓存命令功能正常
|
||||
- 缓存集成功能正常
|
||||
|
||||
**下一步**:
|
||||
- 完善缓存功能测试
|
||||
- 添加缓存命中率统计
|
||||
- 实现按语言清理缓存功能
|
||||
- 优化缓存性能
|
||||
|
||||
---
|
||||
|
||||
### 0.4.0 (2026-03-29) - 管道符功能
|
||||
**类型**: 功能版本
|
||||
**状态**: 已发布
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 添加管道符支持,允许与其他命令行工具联合使用
|
||||
- ✅ 实现管道输入检测 (isPipeInput函数)
|
||||
- ✅ 实现从标准输入读取 (readFromStdin函数)
|
||||
- ✅ 添加 --quiet 和 -q 参数控制统计信息输出
|
||||
- ✅ 更新帮助文档,添加管道使用示例
|
||||
- ✅ 修复 content/filter.go 中的正则表达式错误
|
||||
- ✅ 更新版本号到 v0.3.0
|
||||
|
||||
**使用示例**:
|
||||
```bash
|
||||
# 管道翻译文件
|
||||
cat file.txt | yoyo
|
||||
cat file.txt | yoyo --lang=en
|
||||
|
||||
# 管道翻译命令输出
|
||||
echo "Hello world" | yoyo --lang=cn
|
||||
|
||||
# 静默模式,只输出翻译结果
|
||||
echo "Hello world" | yoyo -q
|
||||
|
||||
# 与其他命令组合使用
|
||||
cat file.txt | yoyo | grep "关键词"
|
||||
yoyo "Hello" | wc -l
|
||||
```
|
||||
|
||||
**讨论记录**:
|
||||
- [管道符功能设计](taolun.md#管道符功能设计)
|
||||
|
||||
**下一步**:
|
||||
- 实现更多厂商(火山引擎、国家超算、Qwen、OpenAI兼容)
|
||||
- 添加配置文件路径查找机制
|
||||
- 实现配置文件迁移工具
|
||||
- 完善错误处理和用户体验
|
||||
|
||||
### 0.3.0 (2026-03-29) - 内容过滤与代码处理
|
||||
**类型**: 功能版本
|
||||
**状态**: 已发布
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 添加内容过滤模块 (internal/content/)
|
||||
- ✅ 实现基础字符过滤(移除控制字符、规范化换行符、截断超长符号)
|
||||
- ✅ 实现代码块和行内代码识别
|
||||
- ✅ 实现代码注释智能识别(支持 JS/TS/Java/Python/Go/HTML 等 30+ 语言)
|
||||
- ✅ 添加 go-enry 依赖实现编程语言智能检测
|
||||
- ✅ 添加 SkipKeywords 配置项,默认保留 TODO/FIXME/HACK 等关键词不翻译
|
||||
- ✅ 集成内容处理到 Translator 模块
|
||||
|
||||
**新增文件**:
|
||||
- `internal/content/content.go` - 模块入口
|
||||
- `internal/content/filter.go` - 基础字符过滤
|
||||
- `internal/content/parser.go` - 内容解析器和语言检测
|
||||
|
||||
**配置更新**:
|
||||
- `configs/config.yaml` 新增 `skip_keywords` 配置项
|
||||
- 支持用户自定义不翻译的关键词列表
|
||||
|
||||
**使用示例**:
|
||||
```bash
|
||||
# 翻译包含代码的文档,自动识别代码和注释
|
||||
yoyo "这是一个文档 ```js // TODO: fix this ```"
|
||||
# 代码块保持不变,只翻译注释中的词汇
|
||||
# TODO: 修复这个
|
||||
```
|
||||
|
||||
**讨论记录**:
|
||||
- [内容过滤与代码处理设计](taolun.md#内容过滤与代码处理设计)
|
||||
|
||||
**下一步**:
|
||||
- 实现更多厂商(火山引擎、国家超算、Qwen、OpenAI兼容)
|
||||
- 添加配置文件路径查找机制
|
||||
- 实现配置文件迁移工具
|
||||
- 完善错误处理和用户体验
|
||||
|
||||
### 0.2.0 (2026-03-29) - 语言支持和配置向导
|
||||
**类型**: 功能版本
|
||||
**状态**: 开发中
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 添加语言代码智能解析模块 (internal/lang)
|
||||
- ✅ 支持 `--lang` 参数指定目标语言
|
||||
- ✅ 支持多种语言代码格式(标准BCP47、别名、中文名称)
|
||||
- ✅ 实现 onboard 交互式配置向导
|
||||
- ✅ 更新配置结构添加语言字段
|
||||
- ✅ 添加 survey 库依赖用于交互式界面
|
||||
- ✅ 改进CLI命令行接口
|
||||
- ✅ 添加语言模块单元测试
|
||||
|
||||
**新增文件**:
|
||||
- `internal/lang/lang.go` - 语言代码解析模块
|
||||
- `internal/lang/lang_test.go` - 语言模块测试
|
||||
- `internal/onboard/onboard.go` - 配置向导实现
|
||||
|
||||
**支持的语言代码**:
|
||||
- 标准格式: zh-CN, zh-TW, en-US, en-GB, ja, ko, es, fr, de 等
|
||||
- 简短别名: cn(中文), en(英文), jp(日文), kr(韩文) 等
|
||||
- 中文名称: chinese(中文), english(英文), japanese(日文) 等
|
||||
|
||||
**使用示例**:
|
||||
```bash
|
||||
# 基本翻译
|
||||
yoyo "Hello world"
|
||||
yoyo --lang=cn "Hello world"
|
||||
yoyo --lang=en "你好世界"
|
||||
yoyo --lang=zh-TW "Hello world"
|
||||
|
||||
# 配置向导
|
||||
yoyo onboard
|
||||
yoyo onboard --force
|
||||
```
|
||||
|
||||
**讨论记录**:
|
||||
- [语言代码解析设计](taolun.md#语言代码解析设计)
|
||||
- [onboard配置向导](taolun.md#onboard配置向导)
|
||||
|
||||
**下一步**:
|
||||
- 实现更多厂商(火山引擎、国家超算、Qwen、OpenAI兼容)
|
||||
- 添加配置文件路径查找机制
|
||||
- 实现配置文件迁移工具
|
||||
- 完善错误处理和用户体验
|
||||
|
||||
### 0.0.3 (2026-03-29) - 环境变量加载修复
|
||||
**类型**: 修复版本
|
||||
**状态**: 开发中
|
||||
@@ -125,4 +418,27 @@
|
||||
### 示例版本递增
|
||||
- `0.0.1` → `0.0.2`:小修复
|
||||
- `0.0.99` → `0.1.0`:新功能(修订版本溢出)
|
||||
- `1.2.3` → `2.0.0`:重大架构变更
|
||||
- `1.2.3` → `2.0.0`:重大架构变更
|
||||
|
||||
---
|
||||
|
||||
## v0.8.1 (2026-04-07)
|
||||
|
||||
### 新功能
|
||||
- 使用 lipgloss v2 设计翻译结果卡片组件
|
||||
- 卡片包含三部分:元信息行(Tokens/耗时/模型)、用户输入(碳黑背景)、翻译结果
|
||||
|
||||
### 升级
|
||||
- 升级 `charm.land/lipgloss/v2` v1.1.0 → v2.0.2
|
||||
- 升级 `charm.land/bubbletea/v2` v1.3.10 → v2.0.2
|
||||
- 升级 `charm.land/bubbles/v2` → v2.1.0
|
||||
- 更新所有模块路径为 `charm.land/xxx/v2` 格式
|
||||
|
||||
### 技术细节
|
||||
- 背景色: #1A1A1A (碳黑色)
|
||||
- 用户输入区域无边框,纯背景色
|
||||
- 组件间距: 5px marginBottom
|
||||
- View() 方法返回 `tea.View` 类型
|
||||
- KeyMsg 改为 KeyPressMsg,使用 `msg.String()` 判断键位
|
||||
|
||||
**讨论记录**: [taolun.md#版本-0.8.1-翻译结果卡片组件设计](taolun.md#版本-081---翻译结果卡片组件设计)
|
||||
@@ -4,32 +4,34 @@
|
||||
default_provider: "siliconflow"
|
||||
default_model: "gpt-3.5-turbo"
|
||||
timeout: 30
|
||||
default_source_lang: "auto" # 默认源语言(auto为自动检测)
|
||||
default_target_lang: "zh-CN" # 默认目标语言(简体中文)
|
||||
|
||||
providers:
|
||||
siliconflow:
|
||||
api_host: "https://api.siliconflow.cn/v1"
|
||||
api_key: "${SILICONFLOW_API_KEY}"
|
||||
model: "siliconflow-base"
|
||||
model: "${SILICONFLOW_MODEL}"
|
||||
enabled: true
|
||||
|
||||
|
||||
volcano:
|
||||
api_host: "https://api.volcengine.com/v1"
|
||||
api_key: "${VOLCANO_API_KEY}"
|
||||
model: "volcano-chat"
|
||||
enabled: true
|
||||
|
||||
|
||||
national:
|
||||
api_host: "https://api.nsc.gov.cn/v1"
|
||||
api_key: "${NATIONAL_API_KEY}"
|
||||
model: "nsc-base"
|
||||
enabled: false
|
||||
|
||||
|
||||
qwen:
|
||||
api_host: "https://dashscope.aliyuncs.com/compatible-mode/v1"
|
||||
api_key: "${QWEN_API_KEY}"
|
||||
model: "qwen-turbo"
|
||||
enabled: true
|
||||
|
||||
|
||||
openai:
|
||||
api_host: "https://api.openai.com/v1"
|
||||
api_key: "${OPENAI_API_KEY}"
|
||||
@@ -40,4 +42,11 @@ prompts:
|
||||
technical: "你是一位专业的技术翻译,请准确翻译以下技术文档,保持专业术语的准确性。"
|
||||
creative: "你是一位富有创造力的翻译家,请用优美流畅的语言翻译以下内容。"
|
||||
academic: "你是一位学术翻译专家,请用严谨的学术语言翻译以下内容。"
|
||||
simple: "请用简单易懂的语言翻译以下内容。"
|
||||
simple: "请用简单易懂的语言翻译以下内容。"
|
||||
|
||||
# 缓存配置
|
||||
cache:
|
||||
enabled: true # 是否启用缓存
|
||||
max_records: 10000 # 最大缓存记录数
|
||||
expire_days: 30 # 缓存过期天数
|
||||
db_path: "~/.config/yoyo/cache.db" # 缓存数据库文件路径
|
||||
|
||||
45
go.mod
45
go.mod
@@ -3,6 +3,47 @@ module github.com/titor/fanyi
|
||||
go 1.26.1
|
||||
|
||||
require (
|
||||
github.com/joho/godotenv v1.5.1 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
charm.land/bubbles/v2 v2.1.0
|
||||
charm.land/bubbletea/v2 v2.0.2
|
||||
charm.land/lipgloss/v2 v2.0.2
|
||||
github.com/AlecAivazis/survey/v2 v2.3.7
|
||||
github.com/charmbracelet/bubbles v1.0.0
|
||||
github.com/go-enry/go-enry/v2 v2.9.5
|
||||
github.com/joho/godotenv v1.5.1
|
||||
github.com/mattn/go-sqlite3 v1.14.37
|
||||
gopkg.in/yaml.v3 v3.0.1
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/atotto/clipboard v0.1.4 // indirect
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
|
||||
github.com/charmbracelet/bubbletea v1.3.10 // indirect
|
||||
github.com/charmbracelet/colorprofile v0.4.2 // indirect
|
||||
github.com/charmbracelet/lipgloss v1.1.0 // indirect
|
||||
github.com/charmbracelet/ultraviolet v0.0.0-20260205113103-524a6607adb8 // indirect
|
||||
github.com/charmbracelet/x/ansi v0.11.6 // indirect
|
||||
github.com/charmbracelet/x/cellbuf v0.0.15 // indirect
|
||||
github.com/charmbracelet/x/term v0.2.2 // indirect
|
||||
github.com/charmbracelet/x/termios v0.1.1 // indirect
|
||||
github.com/charmbracelet/x/windows v0.2.2 // indirect
|
||||
github.com/clipperhouse/displaywidth v0.11.0 // indirect
|
||||
github.com/clipperhouse/uax29/v2 v2.7.0 // indirect
|
||||
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
|
||||
github.com/go-enry/go-oniguruma v1.2.1 // indirect
|
||||
github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51 // indirect
|
||||
github.com/lucasb-eyer/go-colorful v1.3.0 // indirect
|
||||
github.com/mattn/go-colorable v0.1.2 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/mattn/go-localereader v0.0.1 // indirect
|
||||
github.com/mattn/go-runewidth v0.0.21 // indirect
|
||||
github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b // indirect
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
|
||||
github.com/muesli/cancelreader v0.2.2 // indirect
|
||||
github.com/muesli/termenv v0.16.0 // indirect
|
||||
github.com/rivo/uniseg v0.4.7 // indirect
|
||||
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
|
||||
golang.org/x/sync v0.19.0 // indirect
|
||||
golang.org/x/sys v0.42.0 // indirect
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211 // indirect
|
||||
golang.org/x/text v0.4.0 // indirect
|
||||
)
|
||||
|
||||
132
go.sum
132
go.sum
@@ -1,5 +1,137 @@
|
||||
charm.land/bubbles/v2 v2.1.0 h1:YSnNh5cPYlYjPxRrzs5VEn3vwhtEn3jVGRBT3M7/I0g=
|
||||
charm.land/bubbles/v2 v2.1.0/go.mod h1:l97h4hym2hvWBVfmJDtrEHHCtkIKeTEb3TTJ4ZOB3wY=
|
||||
charm.land/bubbletea/v2 v2.0.2 h1:4CRtRnuZOdFDTWSff9r8QFt/9+z6Emubz3aDMnf/dx0=
|
||||
charm.land/bubbletea/v2 v2.0.2/go.mod h1:3LRff2U4WIYXy7MTxfbAQ+AdfM3D8Xuvz2wbsOD9OHQ=
|
||||
charm.land/lipgloss/v2 v2.0.2 h1:xFolbF8JdpNkM2cEPTfXEcW1p6NRzOWTSamRfYEw8cs=
|
||||
charm.land/lipgloss/v2 v2.0.2/go.mod h1:KjPle2Qd3YmvP1KL5OMHiHysGcNwq6u83MUjYkFvEkM=
|
||||
github.com/AlecAivazis/survey/v2 v2.3.7 h1:6I/u8FvytdGsgonrYsVn2t8t4QiRnh6QSTqkkhIiSjQ=
|
||||
github.com/AlecAivazis/survey/v2 v2.3.7/go.mod h1:xUTIdE4KCOIjsBAE1JYsUPoCqYdZ1reCfTwbto0Fduo=
|
||||
github.com/MakeNowJust/heredoc v1.0.0 h1:cXCdzVdstXyiTqTvfqk9SDHpKNjxuom+DOlyEeQ4pzQ=
|
||||
github.com/MakeNowJust/heredoc v1.0.0/go.mod h1:mG5amYoWBHf8vpLOuehzbGGw0EHxpZZ6lCpQ4fNJ8LE=
|
||||
github.com/Netflix/go-expect v0.0.0-20220104043353-73e0943537d2 h1:+vx7roKuyA63nhn5WAunQHLTznkw5W8b1Xc0dNjp83s=
|
||||
github.com/Netflix/go-expect v0.0.0-20220104043353-73e0943537d2/go.mod h1:HBCaDeC1lPdgDeDbhX8XFpy1jqjK0IBG8W5K+xYqA0w=
|
||||
github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4=
|
||||
github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI=
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
|
||||
github.com/aymanbagabas/go-udiff v0.4.1 h1:OEIrQ8maEeDBXQDoGCbbTTXYJMYRCRO1fnodZ12Gv5o=
|
||||
github.com/aymanbagabas/go-udiff v0.4.1/go.mod h1:0L9PGwj20lrtmEMeyw4WKJ/TMyDtvAoK9bf2u/mNo3w=
|
||||
github.com/charmbracelet/bubbles v1.0.0 h1:12J8/ak/uCZEMQ6KU7pcfwceyjLlWsDLAxB5fXonfvc=
|
||||
github.com/charmbracelet/bubbles v1.0.0/go.mod h1:9d/Zd5GdnauMI5ivUIVisuEm3ave1XwXtD1ckyV6r3E=
|
||||
github.com/charmbracelet/bubbletea v1.3.10 h1:otUDHWMMzQSB0Pkc87rm691KZ3SWa4KUlvF9nRvCICw=
|
||||
github.com/charmbracelet/bubbletea v1.3.10/go.mod h1:ORQfo0fk8U+po9VaNvnV95UPWA1BitP1E0N6xJPlHr4=
|
||||
github.com/charmbracelet/colorprofile v0.4.2 h1:BdSNuMjRbotnxHSfxy+PCSa4xAmz7szw70ktAtWRYrY=
|
||||
github.com/charmbracelet/colorprofile v0.4.2/go.mod h1:0rTi81QpwDElInthtrQ6Ni7cG0sDtwAd4C4le060fT8=
|
||||
github.com/charmbracelet/lipgloss v1.1.0 h1:vYXsiLHVkK7fp74RkV7b2kq9+zDLoEU4MZoFqR/noCY=
|
||||
github.com/charmbracelet/lipgloss v1.1.0/go.mod h1:/6Q8FR2o+kj8rz4Dq0zQc3vYf7X+B0binUUBwA0aL30=
|
||||
github.com/charmbracelet/ultraviolet v0.0.0-20260205113103-524a6607adb8 h1:eyFRbAmexyt43hVfeyBofiGSEmJ7krjLOYt/9CF5NKA=
|
||||
github.com/charmbracelet/ultraviolet v0.0.0-20260205113103-524a6607adb8/go.mod h1:SQpCTRNBtzJkwku5ye4S3HEuthAlGy2n9VXZnWkEW98=
|
||||
github.com/charmbracelet/x/ansi v0.11.6 h1:GhV21SiDz/45W9AnV2R61xZMRri5NlLnl6CVF7ihZW8=
|
||||
github.com/charmbracelet/x/ansi v0.11.6/go.mod h1:2JNYLgQUsyqaiLovhU2Rv/pb8r6ydXKS3NIttu3VGZQ=
|
||||
github.com/charmbracelet/x/cellbuf v0.0.15 h1:ur3pZy0o6z/R7EylET877CBxaiE1Sp1GMxoFPAIztPI=
|
||||
github.com/charmbracelet/x/cellbuf v0.0.15/go.mod h1:J1YVbR7MUuEGIFPCaaZ96KDl5NoS0DAWkskup+mOY+Q=
|
||||
github.com/charmbracelet/x/exp/golden v0.0.0-20250806222409-83e3a29d542f h1:pk6gmGpCE7F3FcjaOEKYriCvpmIN4+6OS/RD0vm4uIA=
|
||||
github.com/charmbracelet/x/exp/golden v0.0.0-20250806222409-83e3a29d542f/go.mod h1:IfZAMTHB6XkZSeXUqriemErjAWCCzT0LwjKFYCZyw0I=
|
||||
github.com/charmbracelet/x/term v0.2.2 h1:xVRT/S2ZcKdhhOuSP4t5cLi5o+JxklsoEObBSgfgZRk=
|
||||
github.com/charmbracelet/x/term v0.2.2/go.mod h1:kF8CY5RddLWrsgVwpw4kAa6TESp6EB5y3uxGLeCqzAI=
|
||||
github.com/charmbracelet/x/termios v0.1.1 h1:o3Q2bT8eqzGnGPOYheoYS8eEleT5ZVNYNy8JawjaNZY=
|
||||
github.com/charmbracelet/x/termios v0.1.1/go.mod h1:rB7fnv1TgOPOyyKRJ9o+AsTU/vK5WHJ2ivHeut/Pcwo=
|
||||
github.com/charmbracelet/x/windows v0.2.2 h1:IofanmuvaxnKHuV04sC0eBy/smG6kIKrWG2/jYn2GuM=
|
||||
github.com/charmbracelet/x/windows v0.2.2/go.mod h1:/8XtdKZzedat74NQFn0NGlGL4soHB0YQZrETF96h75k=
|
||||
github.com/clipperhouse/displaywidth v0.11.0 h1:lBc6kY44VFw+TDx4I8opi/EtL9m20WSEFgwIwO+UVM8=
|
||||
github.com/clipperhouse/displaywidth v0.11.0/go.mod h1:bkrFNkf81G8HyVqmKGxsPufD3JhNl3dSqnGhOoSD/o0=
|
||||
github.com/clipperhouse/uax29/v2 v2.7.0 h1:+gs4oBZ2gPfVrKPthwbMzWZDaAFPGYK72F0NJv2v7Vk=
|
||||
github.com/clipperhouse/uax29/v2 v2.7.0/go.mod h1:EFJ2TJMRUaplDxHKj1qAEhCtQPW2tJSwu5BF98AuoVM=
|
||||
github.com/creack/pty v1.1.17 h1:QeVUsEDNrLBW4tMgZHvxy18sKtr6VI492kBhUfhDJNI=
|
||||
github.com/creack/pty v1.1.17/go.mod h1:MOBLtS5ELjhRRrroQr9kyvTxUAFNvYEK993ew/Vr4O4=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f h1:Y/CXytFA4m6baUTXGLOoWe4PQhGxaX0KpnayAqC48p4=
|
||||
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f/go.mod h1:vw97MGsxSvLiUE2X8qFplwetxpGLQrlU1Q9AUEIzCaM=
|
||||
github.com/go-enry/go-enry/v2 v2.9.5 h1:HPhAQQHYwJgihL2PxBZiUMFWiROsGwOBdB6/D8zCUhY=
|
||||
github.com/go-enry/go-enry/v2 v2.9.5/go.mod h1:9yrj4ES1YrbNb1Wb7/PWYr2bpaCXUGRt0uafN0ISyG8=
|
||||
github.com/go-enry/go-oniguruma v1.2.1 h1:k8aAMuJfMrqm/56SG2lV9Cfti6tC4x8673aHCcBk+eo=
|
||||
github.com/go-enry/go-oniguruma v1.2.1/go.mod h1:bWDhYP+S6xZQgiRL7wlTScFYBe023B6ilRZbCAD5Hf4=
|
||||
github.com/hinshun/vt10x v0.0.0-20220119200601-820417d04eec h1:qv2VnGeEQHchGaZ/u7lxST/RaJw+cv273q79D81Xbog=
|
||||
github.com/hinshun/vt10x v0.0.0-20220119200601-820417d04eec/go.mod h1:Q48J4R4DvxnHolD5P8pOtXigYlRuPLGl6moFx3ulM68=
|
||||
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
||||
github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
|
||||
github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51 h1:Z9n2FFNUXsshfwJMBgNA0RU6/i7WVaAegv3PtuIHPMs=
|
||||
github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51/go.mod h1:CzGEWj7cYgsdH8dAjBGEr58BoE7ScuLd+fwFZ44+/x8=
|
||||
github.com/lucasb-eyer/go-colorful v1.3.0 h1:2/yBRLdWBZKrf7gB40FoiKfAWYQ0lqNcbuQwVHXptag=
|
||||
github.com/lucasb-eyer/go-colorful v1.3.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
|
||||
github.com/mattn/go-colorable v0.1.2 h1:/bC9yWikZXAL9uJdulbSfyVNIR3n3trXl+v8+1sx8mU=
|
||||
github.com/mattn/go-colorable v0.1.2/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
|
||||
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4=
|
||||
github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
|
||||
github.com/mattn/go-runewidth v0.0.21 h1:jJKAZiQH+2mIinzCJIaIG9Be1+0NR+5sz/lYEEjdM8w=
|
||||
github.com/mattn/go-runewidth v0.0.21/go.mod h1:XBkDxAl56ILZc9knddidhrOlY5R/pDhgLpndooCuJAs=
|
||||
github.com/mattn/go-sqlite3 v1.14.37 h1:3DOZp4cXis1cUIpCfXLtmlGolNLp2VEqhiB/PARNBIg=
|
||||
github.com/mattn/go-sqlite3 v1.14.37/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
||||
github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b h1:j7+1HpAFS1zy5+Q4qx1fWh90gTKwiN4QCGoY9TWyyO4=
|
||||
github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b/go.mod h1:01TrycV0kFyexm33Z7vhZRXopbI8J3TDReVlkTgMUxE=
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
|
||||
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
|
||||
github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
|
||||
github.com/muesli/termenv v0.16.0 h1:S5AlUN9dENB57rsbnkPyfdGuWIlkmzJjbFf0Tf5FWUc=
|
||||
github.com/muesli/termenv v0.16.0/go.mod h1:ZRfOIKPFDYQoDFF4Olj7/QJbW60Ol/kL1pU3VfY/Cnk=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
|
||||
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
||||
github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
|
||||
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
|
||||
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
|
||||
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||
golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
|
||||
golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
|
||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
||||
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo=
|
||||
golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211 h1:JGgROgKl9N8DuW20oFS5gxc+lE67/N3FcwmBPMe7ArY=
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||
golang.org/x/text v0.4.0 h1:BrVqGRd7+k1DiOgtnFvAkoQEWQvBc25ouMJM6429SFg=
|
||||
golang.org/x/text v0.4.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
||||
75
internal/cache/cache.go
vendored
Normal file
75
internal/cache/cache.go
vendored
Normal file
@@ -0,0 +1,75 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Cache 缓存接口
|
||||
type Cache interface {
|
||||
// Get 获取缓存
|
||||
Get(ctx context.Context, key string) (*CacheEntry, error)
|
||||
|
||||
// Set 设置缓存
|
||||
Set(ctx context.Context, entry *CacheEntry) error
|
||||
|
||||
// Delete 删除缓存
|
||||
Delete(ctx context.Context, key string) error
|
||||
|
||||
// Clear 清空缓存
|
||||
Clear(ctx context.Context) error
|
||||
|
||||
// Stats 获取缓存统计信息
|
||||
Stats(ctx context.Context) (*CacheStats, error)
|
||||
|
||||
// Cleanup 清理过期缓存
|
||||
Cleanup(ctx context.Context) error
|
||||
|
||||
// Close 关闭缓存
|
||||
Close() error
|
||||
}
|
||||
|
||||
// CacheEntry 缓存条目
|
||||
type CacheEntry struct {
|
||||
ID int64 `json:"id"`
|
||||
CacheKey string `json:"cache_key"`
|
||||
OriginalText string `json:"original_text"`
|
||||
TranslatedText string `json:"translated_text"`
|
||||
FromLang string `json:"from_lang"`
|
||||
ToLang string `json:"to_lang"`
|
||||
Model string `json:"model"`
|
||||
PromptName string `json:"prompt_name,omitempty"`
|
||||
PromptContent string `json:"prompt_content,omitempty"`
|
||||
PromptTokens int `json:"prompt_tokens"`
|
||||
CompletionTokens int `json:"completion_tokens"`
|
||||
TotalTokens int `json:"total_tokens"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
LastUsedAt time.Time `json:"last_used_at"`
|
||||
}
|
||||
|
||||
// CacheStats 缓存统计信息
|
||||
type CacheStats struct {
|
||||
TotalRecords int `json:"total_records"`
|
||||
TotalSizeBytes int64 `json:"total_size_bytes"`
|
||||
OldestRecord time.Time `json:"oldest_record"`
|
||||
NewestRecord time.Time `json:"newest_record"`
|
||||
AvgTokensPerRecord float64 `json:"avg_tokens_per_record"`
|
||||
}
|
||||
|
||||
// CacheConfig 缓存配置
|
||||
type CacheConfig struct {
|
||||
Enabled bool `yaml:"enabled"`
|
||||
MaxRecords int `yaml:"max_records"`
|
||||
ExpireDays int `yaml:"expire_days"`
|
||||
DBPath string `yaml:"db_path"`
|
||||
}
|
||||
|
||||
// NewCacheConfig 创建默认缓存配置
|
||||
func NewCacheConfig() *CacheConfig {
|
||||
return &CacheConfig{
|
||||
Enabled: true,
|
||||
MaxRecords: 10000,
|
||||
ExpireDays: 30,
|
||||
DBPath: "~/.config/yoyo/cache.db",
|
||||
}
|
||||
}
|
||||
325
internal/cache/cache_test.go
vendored
Normal file
325
internal/cache/cache_test.go
vendored
Normal file
@@ -0,0 +1,325 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestGenerateCacheKey(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
text string
|
||||
fromLang string
|
||||
toLang string
|
||||
wantSame bool
|
||||
}{
|
||||
{
|
||||
name: "相同输入应生成相同键",
|
||||
text: "Hello world",
|
||||
fromLang: "en",
|
||||
toLang: "zh-CN",
|
||||
wantSame: true,
|
||||
},
|
||||
{
|
||||
name: "不同文本应生成不同键",
|
||||
text: "Hello universe",
|
||||
fromLang: "en",
|
||||
toLang: "zh-CN",
|
||||
wantSame: false,
|
||||
},
|
||||
{
|
||||
name: "不同语言对应生成不同键",
|
||||
text: "Hello world",
|
||||
fromLang: "en",
|
||||
toLang: "zh-TW",
|
||||
wantSame: false,
|
||||
},
|
||||
{
|
||||
name: "大小写不敏感的语言代码",
|
||||
text: "Hello world",
|
||||
fromLang: "EN",
|
||||
toLang: "zh-cn",
|
||||
wantSame: true,
|
||||
},
|
||||
{
|
||||
name: "多余空白字符应规范化",
|
||||
text: " Hello world ",
|
||||
fromLang: "en",
|
||||
toLang: "zh-CN",
|
||||
wantSame: true,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
key1 := GenerateCacheKey(tt.text, tt.fromLang, tt.toLang)
|
||||
key2 := GenerateCacheKey(tt.text, tt.fromLang, tt.toLang)
|
||||
|
||||
if key1 != key2 {
|
||||
t.Errorf("相同输入生成了不同的键: %s != %s", key1, key2)
|
||||
}
|
||||
|
||||
// 检查键的长度(SHA256哈希应为64个字符)
|
||||
if len(key1) != 64 {
|
||||
t.Errorf("缓存键长度不正确: got %d, want 64", len(key1))
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSQLiteCache(t *testing.T) {
|
||||
// 创建临时目录
|
||||
tmpDir, err := os.MkdirTemp("", "cache_test")
|
||||
if err != nil {
|
||||
t.Fatalf("创建临时目录失败: %v", err)
|
||||
}
|
||||
defer os.RemoveAll(tmpDir)
|
||||
|
||||
dbPath := filepath.Join(tmpDir, "test_cache.db")
|
||||
|
||||
// 创建缓存配置
|
||||
config := &CacheConfig{
|
||||
Enabled: true,
|
||||
MaxRecords: 100,
|
||||
ExpireDays: 1,
|
||||
DBPath: dbPath,
|
||||
}
|
||||
|
||||
// 创建缓存实例
|
||||
cache, err := NewSQLiteCache(config)
|
||||
if err != nil {
|
||||
t.Fatalf("创建缓存实例失败: %v", err)
|
||||
}
|
||||
defer cache.Close()
|
||||
|
||||
ctx := context.Background()
|
||||
|
||||
// 测试设置缓存
|
||||
entry := &CacheEntry{
|
||||
CacheKey: "test_key_1",
|
||||
OriginalText: "Hello world",
|
||||
TranslatedText: "你好世界",
|
||||
FromLang: "en",
|
||||
ToLang: "zh-CN",
|
||||
Model: "gpt-3.5-turbo",
|
||||
PromptName: "simple",
|
||||
PromptContent: "请用简单易懂的语言翻译以下内容。",
|
||||
PromptTokens: 10,
|
||||
CompletionTokens: 5,
|
||||
TotalTokens: 15,
|
||||
}
|
||||
|
||||
err = cache.Set(ctx, entry)
|
||||
if err != nil {
|
||||
t.Fatalf("设置缓存失败: %v", err)
|
||||
}
|
||||
|
||||
// 测试获取缓存
|
||||
cachedEntry, err := cache.Get(ctx, "test_key_1")
|
||||
if err != nil {
|
||||
t.Fatalf("获取缓存失败: %v", err)
|
||||
}
|
||||
if cachedEntry == nil {
|
||||
t.Fatal("缓存未命中")
|
||||
}
|
||||
|
||||
if cachedEntry.TranslatedText != "你好世界" {
|
||||
t.Errorf("缓存翻译结果不正确: got %s, want 你好世界", cachedEntry.TranslatedText)
|
||||
}
|
||||
|
||||
// 测试缓存未命中
|
||||
missingEntry, err := cache.Get(ctx, "non_existent_key")
|
||||
if err != nil {
|
||||
t.Fatalf("查询不存在的缓存失败: %v", err)
|
||||
}
|
||||
if missingEntry != nil {
|
||||
t.Error("不存在的缓存应该返回nil")
|
||||
}
|
||||
|
||||
// 测试统计信息
|
||||
stats, err := cache.Stats(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("获取统计信息失败: %v", err)
|
||||
}
|
||||
if stats.TotalRecords != 1 {
|
||||
t.Errorf("统计记录数不正确: got %d, want 1", stats.TotalRecords)
|
||||
}
|
||||
|
||||
// 测试删除缓存
|
||||
err = cache.Delete(ctx, "test_key_1")
|
||||
if err != nil {
|
||||
t.Fatalf("删除缓存失败: %v", err)
|
||||
}
|
||||
|
||||
// 验证删除
|
||||
deletedEntry, err := cache.Get(ctx, "test_key_1")
|
||||
if err != nil {
|
||||
t.Fatalf("查询已删除的缓存失败: %v", err)
|
||||
}
|
||||
if deletedEntry != nil {
|
||||
t.Error("已删除的缓存应该返回nil")
|
||||
}
|
||||
|
||||
// 测试清空缓存
|
||||
err = cache.Set(ctx, entry)
|
||||
if err != nil {
|
||||
t.Fatalf("设置缓存失败: %v", err)
|
||||
}
|
||||
|
||||
err = cache.Clear(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("清空缓存失败: %v", err)
|
||||
}
|
||||
|
||||
stats, err = cache.Stats(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("获取统计信息失败: %v", err)
|
||||
}
|
||||
if stats.TotalRecords != 0 {
|
||||
t.Errorf("清空后记录数不正确: got %d, want 0", stats.TotalRecords)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCacheExpiration(t *testing.T) {
|
||||
// 创建临时目录
|
||||
tmpDir, err := os.MkdirTemp("", "cache_test")
|
||||
if err != nil {
|
||||
t.Fatalf("创建临时目录失败: %v", err)
|
||||
}
|
||||
defer os.RemoveAll(tmpDir)
|
||||
|
||||
dbPath := filepath.Join(tmpDir, "test_cache.db")
|
||||
|
||||
// 创建缓存配置,设置很短的过期时间
|
||||
config := &CacheConfig{
|
||||
Enabled: true,
|
||||
MaxRecords: 100,
|
||||
ExpireDays: 0, // 0天表示立即过期
|
||||
DBPath: dbPath,
|
||||
}
|
||||
|
||||
// 创建缓存实例
|
||||
cache, err := NewSQLiteCache(config)
|
||||
if err != nil {
|
||||
t.Fatalf("创建缓存实例失败: %v", err)
|
||||
}
|
||||
defer cache.Close()
|
||||
|
||||
ctx := context.Background()
|
||||
|
||||
// 设置缓存
|
||||
entry := &CacheEntry{
|
||||
CacheKey: "test_key_1",
|
||||
OriginalText: "Hello world",
|
||||
TranslatedText: "你好世界",
|
||||
FromLang: "en",
|
||||
ToLang: "zh-CN",
|
||||
Model: "gpt-3.5-turbo",
|
||||
PromptTokens: 10,
|
||||
CompletionTokens: 5,
|
||||
TotalTokens: 15,
|
||||
}
|
||||
|
||||
err = cache.Set(ctx, entry)
|
||||
if err != nil {
|
||||
t.Fatalf("设置缓存失败: %v", err)
|
||||
}
|
||||
|
||||
// 立即清理(应该删除所有记录,因为过期时间为0)
|
||||
err = cache.Cleanup(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("清理缓存失败: %v", err)
|
||||
}
|
||||
|
||||
// 检查统计信息
|
||||
stats, err := cache.Stats(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("获取统计信息失败: %v", err)
|
||||
}
|
||||
if stats.TotalRecords != 0 {
|
||||
t.Errorf("清理后记录数不正确: got %d, want 0", stats.TotalRecords)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNormalizeText(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
input string
|
||||
expected string
|
||||
}{
|
||||
{
|
||||
name: "正常文本",
|
||||
input: "Hello world",
|
||||
expected: "Hello world",
|
||||
},
|
||||
{
|
||||
name: "多余空白字符",
|
||||
input: " Hello world ",
|
||||
expected: "Hello world",
|
||||
},
|
||||
{
|
||||
name: "制表符和换行符",
|
||||
input: "Hello\tworld\n",
|
||||
expected: "Hello world",
|
||||
},
|
||||
{
|
||||
name: "空字符串",
|
||||
input: "",
|
||||
expected: "",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := normalizeText(tt.input)
|
||||
if result != tt.expected {
|
||||
t.Errorf("normalizeText(%q) = %q, want %q", tt.input, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestNormalizeLanguageCode(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
input string
|
||||
expected string
|
||||
}{
|
||||
{
|
||||
name: "正常语言代码",
|
||||
input: "zh-CN",
|
||||
expected: "zh-cn",
|
||||
},
|
||||
{
|
||||
name: "大写语言代码",
|
||||
input: "EN-US",
|
||||
expected: "en-us",
|
||||
},
|
||||
{
|
||||
name: "空字符串",
|
||||
input: "",
|
||||
expected: "auto",
|
||||
},
|
||||
{
|
||||
name: "auto",
|
||||
input: "auto",
|
||||
expected: "auto",
|
||||
},
|
||||
{
|
||||
name: "前后空白",
|
||||
input: " zh-CN ",
|
||||
expected: "zh-cn",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := normalizeLanguageCode(tt.input)
|
||||
if result != tt.expected {
|
||||
t.Errorf("normalizeLanguageCode(%q) = %q, want %q", tt.input, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
53
internal/cache/cleanup.go
vendored
Normal file
53
internal/cache/cleanup.go
vendored
Normal file
@@ -0,0 +1,53 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
)
|
||||
|
||||
// CleanupManager 缓存清理管理器
|
||||
type CleanupManager struct {
|
||||
cache Cache
|
||||
}
|
||||
|
||||
// NewCleanupManager 创建清理管理器
|
||||
func NewCleanupManager(cache Cache) *CleanupManager {
|
||||
return &CleanupManager{
|
||||
cache: cache,
|
||||
}
|
||||
}
|
||||
|
||||
// ClearAll 清空所有缓存
|
||||
func (m *CleanupManager) ClearAll(ctx context.Context) error {
|
||||
if err := m.cache.Clear(ctx); err != nil {
|
||||
return fmt.Errorf("清空缓存失败: %w", err)
|
||||
}
|
||||
log.Println("缓存已清空")
|
||||
return nil
|
||||
}
|
||||
|
||||
// ClearByLanguage 清空指定语言对的缓存
|
||||
func (m *CleanupManager) ClearByLanguage(ctx context.Context, fromLang, toLang string) error {
|
||||
// 这个功能需要在SQLite实现中添加查询功能
|
||||
// 目前先返回一个提示信息
|
||||
return fmt.Errorf("按语言清理功能尚未实现")
|
||||
}
|
||||
|
||||
// GetStats 获取缓存统计信息
|
||||
func (m *CleanupManager) GetStats(ctx context.Context) (*CacheStats, error) {
|
||||
stats, err := m.cache.Stats(ctx)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取缓存统计失败: %w", err)
|
||||
}
|
||||
return stats, nil
|
||||
}
|
||||
|
||||
// CleanupManual 手动清理
|
||||
func (m *CleanupManager) CleanupManual(ctx context.Context) error {
|
||||
if err := m.cache.Cleanup(ctx); err != nil {
|
||||
return fmt.Errorf("手动清理失败: %w", err)
|
||||
}
|
||||
log.Println("缓存清理完成")
|
||||
return nil
|
||||
}
|
||||
63
internal/cache/key.go
vendored
Normal file
63
internal/cache/key.go
vendored
Normal file
@@ -0,0 +1,63 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"crypto/sha256"
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// GenerateCacheKey 生成缓存键
|
||||
// 使用原文+语言对进行SHA256哈希
|
||||
func GenerateCacheKey(originalText, fromLang, toLang string) string {
|
||||
// 规范化语言代码
|
||||
fromLang = normalizeLanguageCode(fromLang)
|
||||
toLang = normalizeLanguageCode(toLang)
|
||||
|
||||
// 规范化原文
|
||||
normalizedText := normalizeText(originalText)
|
||||
|
||||
// 生成缓存键
|
||||
data := fmt.Sprintf("%s|%s|%s", normalizedText, fromLang, toLang)
|
||||
hash := sha256.Sum256([]byte(data))
|
||||
return hex.EncodeToString(hash[:])
|
||||
}
|
||||
|
||||
// normalizeLanguageCode 规范化语言代码
|
||||
func normalizeLanguageCode(lang string) string {
|
||||
if lang == "" || lang == "auto" {
|
||||
return "auto"
|
||||
}
|
||||
return strings.ToLower(strings.TrimSpace(lang))
|
||||
}
|
||||
|
||||
// normalizeText 规范化文本
|
||||
// 移除多余的空白字符,确保相同的文本生成相同的哈希
|
||||
func normalizeText(text string) string {
|
||||
// 移除首尾空白
|
||||
text = strings.TrimSpace(text)
|
||||
|
||||
// 将多个连续空白字符替换为单个空格
|
||||
text = strings.Join(strings.Fields(text), " ")
|
||||
|
||||
return text
|
||||
}
|
||||
|
||||
// GenerateCacheKeyWithModel 生成包含模型信息的缓存键
|
||||
// 如果需要更精确的缓存,可以使用这个函数
|
||||
func GenerateCacheKeyWithModel(originalText, fromLang, toLang, model string) string {
|
||||
// 规范化语言代码
|
||||
fromLang = normalizeLanguageCode(fromLang)
|
||||
toLang = normalizeLanguageCode(toLang)
|
||||
|
||||
// 规范化原文
|
||||
normalizedText := normalizeText(originalText)
|
||||
|
||||
// 规范化模型名称
|
||||
model = strings.ToLower(strings.TrimSpace(model))
|
||||
|
||||
// 生成缓存键
|
||||
data := fmt.Sprintf("%s|%s|%s|%s", normalizedText, fromLang, toLang, model)
|
||||
hash := sha256.Sum256([]byte(data))
|
||||
return hex.EncodeToString(hash[:])
|
||||
}
|
||||
337
internal/cache/sqlite.go
vendored
Normal file
337
internal/cache/sqlite.go
vendored
Normal file
@@ -0,0 +1,337 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
// SQLiteCache SQLite缓存实现
|
||||
type SQLiteCache struct {
|
||||
db *sql.DB
|
||||
config *CacheConfig
|
||||
cleanupTTL time.Duration
|
||||
}
|
||||
|
||||
// NewSQLiteCache 创建SQLite缓存实例
|
||||
func NewSQLiteCache(config *CacheConfig) (*SQLiteCache, error) {
|
||||
if config == nil {
|
||||
config = NewCacheConfig()
|
||||
}
|
||||
|
||||
// 展开路径中的~符号
|
||||
dbPath, err := expandPath(config.DBPath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("无效的数据库路径: %w", err)
|
||||
}
|
||||
|
||||
// 确保目录存在
|
||||
dir := filepath.Dir(dbPath)
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
return nil, fmt.Errorf("创建缓存目录失败: %w", err)
|
||||
}
|
||||
|
||||
// 打开数据库连接
|
||||
db, err := sql.Open("sqlite3", dbPath+"?_journal_mode=WAL&_synchronous=NORMAL")
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("打开数据库失败: %w", err)
|
||||
}
|
||||
|
||||
// 设置连接池参数
|
||||
db.SetMaxOpenConns(1) // SQLite只支持单个写入连接
|
||||
db.SetMaxIdleConns(1)
|
||||
|
||||
cache := &SQLiteCache{
|
||||
db: db,
|
||||
config: config,
|
||||
cleanupTTL: time.Duration(config.ExpireDays) * 24 * time.Hour,
|
||||
}
|
||||
|
||||
// 初始化数据库表
|
||||
if err := cache.initTable(); err != nil {
|
||||
db.Close()
|
||||
return nil, fmt.Errorf("初始化缓存表失败: %w", err)
|
||||
}
|
||||
|
||||
// 设置清理定时器
|
||||
go cache.startCleanupTimer()
|
||||
|
||||
return cache, nil
|
||||
}
|
||||
|
||||
// initTable 初始化缓存表
|
||||
func (c *SQLiteCache) initTable() error {
|
||||
query := `
|
||||
CREATE TABLE IF NOT EXISTS translation_cache (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
cache_key TEXT NOT NULL UNIQUE,
|
||||
original_text TEXT NOT NULL,
|
||||
translated_text TEXT NOT NULL,
|
||||
from_lang TEXT NOT NULL,
|
||||
to_lang TEXT NOT NULL,
|
||||
model TEXT NOT NULL,
|
||||
prompt_name TEXT,
|
||||
prompt_content TEXT,
|
||||
prompt_tokens INTEGER DEFAULT 0,
|
||||
completion_tokens INTEGER DEFAULT 0,
|
||||
total_tokens INTEGER DEFAULT 0,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
last_used_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
|
||||
CREATE INDEX IF NOT EXISTS idx_cache_key ON translation_cache(cache_key);
|
||||
CREATE INDEX IF NOT EXISTS idx_original_text ON translation_cache(original_text);
|
||||
CREATE INDEX IF NOT EXISTS idx_created_at ON translation_cache(created_at);
|
||||
CREATE INDEX IF NOT EXISTS idx_last_used_at ON translation_cache(last_used_at);
|
||||
`
|
||||
|
||||
_, err := c.db.Exec(query)
|
||||
return err
|
||||
}
|
||||
|
||||
// Get 获取缓存
|
||||
func (c *SQLiteCache) Get(ctx context.Context, key string) (*CacheEntry, error) {
|
||||
query := `
|
||||
SELECT
|
||||
id, cache_key, original_text, translated_text, from_lang, to_lang,
|
||||
model, prompt_name, prompt_content, prompt_tokens, completion_tokens,
|
||||
total_tokens, created_at, last_used_at
|
||||
FROM translation_cache
|
||||
WHERE cache_key = ?
|
||||
`
|
||||
|
||||
entry := &CacheEntry{}
|
||||
var promptName, promptContent sql.NullString
|
||||
var createdAt, lastUsedAt string
|
||||
|
||||
err := c.db.QueryRowContext(ctx, query, key).Scan(
|
||||
&entry.ID, &entry.CacheKey, &entry.OriginalText, &entry.TranslatedText,
|
||||
&entry.FromLang, &entry.ToLang, &entry.Model, &promptName, &promptContent,
|
||||
&entry.PromptTokens, &entry.CompletionTokens, &entry.TotalTokens,
|
||||
&createdAt, &lastUsedAt,
|
||||
)
|
||||
|
||||
if err == sql.ErrNoRows {
|
||||
return nil, nil // 缓存未命中
|
||||
}
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("查询缓存失败: %w", err)
|
||||
}
|
||||
|
||||
// 处理可空字段
|
||||
if promptName.Valid {
|
||||
entry.PromptName = promptName.String
|
||||
}
|
||||
if promptContent.Valid {
|
||||
entry.PromptContent = promptContent.String
|
||||
}
|
||||
|
||||
// 解析时间
|
||||
entry.CreatedAt, _ = time.Parse("2006-01-02 15:04:05", createdAt)
|
||||
entry.LastUsedAt, _ = time.Parse("2006-01-02 15:04:05", lastUsedAt)
|
||||
|
||||
// 更新最后使用时间
|
||||
go c.updateLastUsed(context.Background(), key)
|
||||
|
||||
return entry, nil
|
||||
}
|
||||
|
||||
// Set 设置缓存
|
||||
func (c *SQLiteCache) Set(ctx context.Context, entry *CacheEntry) error {
|
||||
// 开始事务
|
||||
tx, err := c.db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("开始事务失败: %w", err)
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
// 插入或替换缓存
|
||||
query := `
|
||||
INSERT OR REPLACE INTO translation_cache
|
||||
(cache_key, original_text, translated_text, from_lang, to_lang,
|
||||
model, prompt_name, prompt_content, prompt_tokens, completion_tokens,
|
||||
total_tokens, created_at, last_used_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
`
|
||||
|
||||
now := time.Now().Format("2006-01-02 15:04:05")
|
||||
_, err = tx.ExecContext(ctx, query,
|
||||
entry.CacheKey, entry.OriginalText, entry.TranslatedText,
|
||||
entry.FromLang, entry.ToLang, entry.Model, entry.PromptName,
|
||||
entry.PromptContent, entry.PromptTokens, entry.CompletionTokens,
|
||||
entry.TotalTokens, now, now,
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("插入缓存失败: %w", err)
|
||||
}
|
||||
|
||||
// 提交事务
|
||||
if err := tx.Commit(); err != nil {
|
||||
return fmt.Errorf("提交事务失败: %w", err)
|
||||
}
|
||||
|
||||
// 触发清理(异步)
|
||||
go c.Cleanup(context.Background())
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Delete 删除缓存
|
||||
func (c *SQLiteCache) Delete(ctx context.Context, key string) error {
|
||||
query := `DELETE FROM translation_cache WHERE cache_key = ?`
|
||||
_, err := c.db.ExecContext(ctx, query, key)
|
||||
if err != nil {
|
||||
return fmt.Errorf("删除缓存失败: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Clear 清空缓存
|
||||
func (c *SQLiteCache) Clear(ctx context.Context) error {
|
||||
// 先删除所有记录
|
||||
_, err := c.db.ExecContext(ctx, `DELETE FROM translation_cache`)
|
||||
if err != nil {
|
||||
return fmt.Errorf("清空缓存失败: %w", err)
|
||||
}
|
||||
|
||||
// 然后执行VACUUM(不能在事务中执行)
|
||||
_, err = c.db.ExecContext(ctx, `VACUUM`)
|
||||
if err != nil {
|
||||
return fmt.Errorf("清理数据库失败: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Stats 获取缓存统计信息
|
||||
func (c *SQLiteCache) Stats(ctx context.Context) (*CacheStats, error) {
|
||||
stats := &CacheStats{}
|
||||
|
||||
// 获取总记录数
|
||||
err := c.db.QueryRowContext(ctx, `SELECT COUNT(*) FROM translation_cache`).Scan(&stats.TotalRecords)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("查询缓存统计失败: %w", err)
|
||||
}
|
||||
|
||||
// 如果没有记录,直接返回
|
||||
if stats.TotalRecords == 0 {
|
||||
return stats, nil
|
||||
}
|
||||
|
||||
// 获取时间范围和平均tokens
|
||||
var oldestStr, newestStr sql.NullString
|
||||
var avgTokens sql.NullFloat64
|
||||
err = c.db.QueryRowContext(ctx, `
|
||||
SELECT
|
||||
MIN(created_at),
|
||||
MAX(created_at),
|
||||
AVG(total_tokens)
|
||||
FROM translation_cache
|
||||
`).Scan(&oldestStr, &newestStr, &avgTokens)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("查询缓存时间范围失败: %w", err)
|
||||
}
|
||||
|
||||
// 解析时间字符串
|
||||
if oldestStr.Valid {
|
||||
stats.OldestRecord, _ = time.Parse("2006-01-02 15:04:05", oldestStr.String)
|
||||
}
|
||||
if newestStr.Valid {
|
||||
stats.NewestRecord, _ = time.Parse("2006-01-02 15:04:05", newestStr.String)
|
||||
}
|
||||
if avgTokens.Valid {
|
||||
stats.AvgTokensPerRecord = avgTokens.Float64
|
||||
}
|
||||
|
||||
// 计算数据库文件大小
|
||||
dbPath, _ := expandPath(c.config.DBPath)
|
||||
if info, err := os.Stat(dbPath); err == nil {
|
||||
stats.TotalSizeBytes = info.Size()
|
||||
}
|
||||
|
||||
return stats, nil
|
||||
}
|
||||
|
||||
// Cleanup 清理过期缓存
|
||||
func (c *SQLiteCache) Cleanup(ctx context.Context) error {
|
||||
tx, err := c.db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("开始事务失败: %w", err)
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
// 清理过期缓存
|
||||
if c.cleanupTTL > 0 {
|
||||
expiredTime := time.Now().Add(-c.cleanupTTL).Format("2006-01-02 15:04:05")
|
||||
_, err = tx.ExecContext(ctx, `DELETE FROM translation_cache WHERE last_used_at < ?`, expiredTime)
|
||||
if err != nil {
|
||||
return fmt.Errorf("清理过期缓存失败: %w", err)
|
||||
}
|
||||
} else if c.cleanupTTL == 0 {
|
||||
// 如果过期时间为0,清理所有记录
|
||||
_, err = tx.ExecContext(ctx, `DELETE FROM translation_cache`)
|
||||
if err != nil {
|
||||
return fmt.Errorf("清理所有缓存失败: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
// 清理超出数量限制的缓存
|
||||
if c.config.MaxRecords > 0 {
|
||||
_, err = tx.ExecContext(ctx, `
|
||||
DELETE FROM translation_cache
|
||||
WHERE id NOT IN (
|
||||
SELECT id FROM translation_cache
|
||||
ORDER BY last_used_at DESC
|
||||
LIMIT ?
|
||||
)
|
||||
`, c.config.MaxRecords)
|
||||
if err != nil {
|
||||
return fmt.Errorf("清理超出数量限制的缓存失败: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
// Close 关闭缓存
|
||||
func (c *SQLiteCache) Close() error {
|
||||
if c.db != nil {
|
||||
return c.db.Close()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// updateLastUsed 更新最后使用时间
|
||||
func (c *SQLiteCache) updateLastUsed(ctx context.Context, key string) {
|
||||
query := `UPDATE translation_cache SET last_used_at = ? WHERE cache_key = ?`
|
||||
now := time.Now().Format("2006-01-02 15:04:05")
|
||||
c.db.ExecContext(ctx, query, now, key)
|
||||
}
|
||||
|
||||
// startCleanupTimer 启动清理定时器
|
||||
func (c *SQLiteCache) startCleanupTimer() {
|
||||
ticker := time.NewTicker(1 * time.Hour) // 每小时清理一次
|
||||
defer ticker.Stop()
|
||||
|
||||
for range ticker.C {
|
||||
c.Cleanup(context.Background())
|
||||
}
|
||||
}
|
||||
|
||||
// expandPath 展开路径中的~符号
|
||||
func expandPath(path string) (string, error) {
|
||||
if strings.HasPrefix(path, "~") {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
path = filepath.Join(home, path[1:])
|
||||
}
|
||||
return path, nil
|
||||
}
|
||||
@@ -12,15 +12,23 @@ import (
|
||||
// Config 全局配置结构
|
||||
type Config struct {
|
||||
// 全局设置
|
||||
DefaultProvider string `yaml:"default_provider"`
|
||||
DefaultModel string `yaml:"default_model"`
|
||||
Timeout int `yaml:"timeout"` // 秒
|
||||
DefaultProvider string `yaml:"default_provider"`
|
||||
DefaultModel string `yaml:"default_model"`
|
||||
Timeout int `yaml:"timeout"` // 秒
|
||||
DefaultSourceLang string `yaml:"default_source_lang"` // 默认源语言(auto为自动检测)
|
||||
DefaultTargetLang string `yaml:"default_target_lang"` // 默认目标语言
|
||||
|
||||
// 厂商配置
|
||||
Providers map[string]ProviderConfig `yaml:"providers"`
|
||||
|
||||
// Prompt配置
|
||||
Prompts map[string]string `yaml:"prompts"`
|
||||
|
||||
// 内容过滤配置
|
||||
SkipKeywords []string `yaml:"skip_keywords"` // 不翻译的关键词
|
||||
|
||||
// 缓存配置
|
||||
Cache CacheConfig `yaml:"cache"`
|
||||
}
|
||||
|
||||
// ProviderConfig 厂商配置
|
||||
@@ -31,6 +39,14 @@ type ProviderConfig struct {
|
||||
Enabled bool `yaml:"enabled"`
|
||||
}
|
||||
|
||||
// CacheConfig 缓存配置
|
||||
type CacheConfig struct {
|
||||
Enabled bool `yaml:"enabled"`
|
||||
MaxRecords int `yaml:"max_records"`
|
||||
ExpireDays int `yaml:"expire_days"`
|
||||
DBPath string `yaml:"db_path"`
|
||||
}
|
||||
|
||||
// ConfigLoader 配置加载器接口
|
||||
type ConfigLoader interface {
|
||||
Load(path string) (*Config, error)
|
||||
@@ -97,6 +113,12 @@ func (c *Config) setDefaults() {
|
||||
if c.DefaultModel == "" {
|
||||
c.DefaultModel = "gpt-3.5-turbo"
|
||||
}
|
||||
if c.DefaultSourceLang == "" {
|
||||
c.DefaultSourceLang = "auto" // 自动检测
|
||||
}
|
||||
if c.DefaultTargetLang == "" {
|
||||
c.DefaultTargetLang = "zh-CN" // 默认翻译为简体中文
|
||||
}
|
||||
|
||||
// 为每个厂商设置默认值
|
||||
for name, provider := range c.Providers {
|
||||
@@ -113,6 +135,27 @@ func (c *Config) setDefaults() {
|
||||
if c.Prompts == nil {
|
||||
c.Prompts = make(map[string]string)
|
||||
}
|
||||
|
||||
// 设置默认关键词
|
||||
if c.SkipKeywords == nil {
|
||||
c.SkipKeywords = []string{
|
||||
"TODO", "FIXME", "HACK", "XXX", "NOTE",
|
||||
"BUG", "WARN", "IMPORTANT", "TODO:",
|
||||
"FIXME:", "HACK:", "XXX:", "NOTE:",
|
||||
"BUG:", "WARN:", "IMPORTANT:",
|
||||
}
|
||||
}
|
||||
|
||||
// 设置缓存配置默认值
|
||||
if c.Cache.MaxRecords <= 0 {
|
||||
c.Cache.MaxRecords = 10000
|
||||
}
|
||||
if c.Cache.ExpireDays <= 0 {
|
||||
c.Cache.ExpireDays = 30
|
||||
}
|
||||
if c.Cache.DBPath == "" {
|
||||
c.Cache.DBPath = "~/.config/yoyo/cache.db"
|
||||
}
|
||||
}
|
||||
|
||||
// GetProviderConfig 获取指定厂商的配置
|
||||
@@ -190,6 +233,8 @@ func (c *Config) String() string {
|
||||
builder.WriteString(fmt.Sprintf("DefaultProvider: %s\n", c.DefaultProvider))
|
||||
builder.WriteString(fmt.Sprintf("DefaultModel: %s\n", c.DefaultModel))
|
||||
builder.WriteString(fmt.Sprintf("Timeout: %d seconds\n", c.Timeout))
|
||||
builder.WriteString(fmt.Sprintf("DefaultSourceLang: %s\n", c.DefaultSourceLang))
|
||||
builder.WriteString(fmt.Sprintf("DefaultTargetLang: %s\n", c.DefaultTargetLang))
|
||||
builder.WriteString("Providers:\n")
|
||||
for name, provider := range c.Providers {
|
||||
builder.WriteString(fmt.Sprintf(" %s: enabled=%v, model=%s\n", name, provider.Enabled, provider.Model))
|
||||
@@ -198,5 +243,10 @@ func (c *Config) String() string {
|
||||
for name := range c.Prompts {
|
||||
builder.WriteString(fmt.Sprintf(" %s\n", name))
|
||||
}
|
||||
builder.WriteString("Cache:\n")
|
||||
builder.WriteString(fmt.Sprintf(" enabled: %v\n", c.Cache.Enabled))
|
||||
builder.WriteString(fmt.Sprintf(" max_records: %d\n", c.Cache.MaxRecords))
|
||||
builder.WriteString(fmt.Sprintf(" expire_days: %d\n", c.Cache.ExpireDays))
|
||||
builder.WriteString(fmt.Sprintf(" db_path: %s\n", c.Cache.DBPath))
|
||||
return builder.String()
|
||||
}
|
||||
|
||||
17
internal/content/content.go
Normal file
17
internal/content/content.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package content
|
||||
|
||||
import (
|
||||
"github.com/go-enry/go-enry/v2"
|
||||
)
|
||||
|
||||
const (
|
||||
Version = "1.0.0"
|
||||
)
|
||||
|
||||
func DetectLanguage(text string) string {
|
||||
return enry.GetLanguage("", []byte(text))
|
||||
}
|
||||
|
||||
func Filter(text string) string {
|
||||
return FilterBasic(text, nil)
|
||||
}
|
||||
56
internal/content/filter.go
Normal file
56
internal/content/filter.go
Normal file
@@ -0,0 +1,56 @@
|
||||
package content
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type FilterOptions struct {
|
||||
RemoveControlChars bool
|
||||
NormalizeLineBreaks bool
|
||||
MaxConsecutiveSymbols int
|
||||
}
|
||||
|
||||
var defaultFilterOptions = &FilterOptions{
|
||||
RemoveControlChars: true,
|
||||
NormalizeLineBreaks: true,
|
||||
MaxConsecutiveSymbols: 20,
|
||||
}
|
||||
|
||||
var controlCharsRegex = regexp.MustCompile(`[\x00-\x08\x0b\x0c\x0e-\x1f\x7f]`)
|
||||
|
||||
func FilterBasic(text string, opts *FilterOptions) string {
|
||||
if opts == nil {
|
||||
opts = defaultFilterOptions
|
||||
}
|
||||
|
||||
result := text
|
||||
|
||||
if opts.RemoveControlChars {
|
||||
result = controlCharsRegex.ReplaceAllString(result, "")
|
||||
}
|
||||
|
||||
if opts.NormalizeLineBreaks {
|
||||
result = strings.ReplaceAll(result, "\r\n", "\n")
|
||||
result = strings.ReplaceAll(result, "\r", "\n")
|
||||
}
|
||||
|
||||
if opts.MaxConsecutiveSymbols > 0 {
|
||||
result = truncateConsecutiveSymbols(result, opts.MaxConsecutiveSymbols)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func truncateConsecutiveSymbols(text string, maxCount int) string {
|
||||
symbols := []string{"=", "-", "_", "*", "#", "~", "`", "."}
|
||||
|
||||
for _, symbol := range symbols {
|
||||
pattern := regexp.MustCompile(regexp.QuoteMeta(symbol) + `{` + fmt.Sprintf("%d", maxCount+1) + `,}`)
|
||||
replacement := strings.Repeat(symbol, maxCount)
|
||||
text = pattern.ReplaceAllString(text, replacement)
|
||||
}
|
||||
|
||||
return text
|
||||
}
|
||||
453
internal/content/parser.go
Normal file
453
internal/content/parser.go
Normal file
@@ -0,0 +1,453 @@
|
||||
package content
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/go-enry/go-enry/v2"
|
||||
)
|
||||
|
||||
type SegmentType int
|
||||
|
||||
const (
|
||||
SegmentTypeText SegmentType = iota
|
||||
SegmentTypeCodeBlock
|
||||
SegmentTypeInlineCode
|
||||
SegmentTypeComment
|
||||
)
|
||||
|
||||
func (t SegmentType) String() string {
|
||||
switch t {
|
||||
case SegmentTypeText:
|
||||
return "text"
|
||||
case SegmentTypeCodeBlock:
|
||||
return "code_block"
|
||||
case SegmentTypeInlineCode:
|
||||
return "inline_code"
|
||||
case SegmentTypeComment:
|
||||
return "comment"
|
||||
default:
|
||||
return "unknown"
|
||||
}
|
||||
}
|
||||
|
||||
type ContentSegment struct {
|
||||
Type SegmentType
|
||||
Content string
|
||||
Translated string
|
||||
Language string
|
||||
IsComment bool
|
||||
StartPos int
|
||||
EndPos int
|
||||
}
|
||||
|
||||
type ParseResult struct {
|
||||
Segments []ContentSegment
|
||||
SourceLang string
|
||||
HasCode bool
|
||||
}
|
||||
|
||||
type languageCommentPatterns struct {
|
||||
LineComment string
|
||||
BlockComment []string
|
||||
}
|
||||
|
||||
var languagePatterns = map[string]languageCommentPatterns{
|
||||
"javascript": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"typescript": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"java": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"kotlin": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"scala": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"c": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"cpp": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"c#": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"go": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"rust": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"php": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"swift": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"objective-c": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"scss": {LineComment: `//`, BlockComment: []string{`/*`, `*/`}},
|
||||
"css": {LineComment: ``, BlockComment: []string{`/*`, `*/`}},
|
||||
"less": {LineComment: ``, BlockComment: []string{`/*`, `*/`}},
|
||||
"html": {LineComment: ``, BlockComment: []string{`<!--`, `-->`}},
|
||||
"xml": {LineComment: ``, BlockComment: []string{`<!--`, `-->`}},
|
||||
"sql": {LineComment: `--`, BlockComment: []string{`/*`, `*/`}},
|
||||
"python": {LineComment: `#`, BlockComment: []string{`"""`, `"""`}},
|
||||
"ruby": {LineComment: `#`, BlockComment: []string{`=begin`, `=end`}},
|
||||
"shell": {LineComment: `#`, BlockComment: []string{}},
|
||||
"bash": {LineComment: `#`, BlockComment: []string{}},
|
||||
"powershell": {LineComment: `#`, BlockComment: []string{`<#`, `#>`}},
|
||||
"yaml": {LineComment: `#()`, BlockComment: []string{}},
|
||||
"json": {LineComment: ``, BlockComment: []string{}},
|
||||
"markdown": {LineComment: ``, BlockComment: []string{}},
|
||||
"vue": {LineComment: `//()`, BlockComment: []string{`/*`, `*/`, `<!--`, `-->`}},
|
||||
"svelte": {LineComment: `//()`, BlockComment: []string{`/*`, `*/`}},
|
||||
"jsx": {LineComment: `//()`, BlockComment: []string{`/*`, `*/`}},
|
||||
"tsx": {LineComment: `//()`, BlockComment: []string{`/*`, `*/`}},
|
||||
}
|
||||
|
||||
var defaultPatterns = languageCommentPatterns{
|
||||
LineComment: `//`,
|
||||
BlockComment: []string{`/*`, `*/`},
|
||||
}
|
||||
|
||||
type Parser struct {
|
||||
skipKeywords []string
|
||||
fallbackLang string
|
||||
}
|
||||
|
||||
func NewParser(skipKeywords []string) *Parser {
|
||||
if skipKeywords == nil {
|
||||
skipKeywords = []string{
|
||||
"TODO", "FIXME", "HACK", "XXX", "NOTE",
|
||||
"BUG", "WARN", "IMPORTANT", "TODO:",
|
||||
"FIXME:", "HACK:", "XXX:", "NOTE:",
|
||||
"BUG:", "WARN:", "IMPORTANT:",
|
||||
}
|
||||
}
|
||||
return &Parser{
|
||||
skipKeywords: skipKeywords,
|
||||
fallbackLang: "javascript",
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Parser) Parse(text string) (*ParseResult, error) {
|
||||
result := &ParseResult{
|
||||
Segments: []ContentSegment{},
|
||||
}
|
||||
|
||||
detectedLang := p.detectLanguage(text)
|
||||
result.SourceLang = detectedLang
|
||||
|
||||
segments := p.splitIntoSegments(text, result.SourceLang)
|
||||
|
||||
for _, seg := range segments {
|
||||
if seg.Type == SegmentTypeCodeBlock || seg.Type == SegmentTypeInlineCode {
|
||||
result.HasCode = true
|
||||
}
|
||||
result.Segments = append(result.Segments, seg)
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (p *Parser) detectLanguage(text string) string {
|
||||
lines := strings.Split(text, "\n")
|
||||
var codeLines []string
|
||||
inCodeBlock := false
|
||||
|
||||
for _, line := range lines {
|
||||
trimmed := strings.TrimSpace(line)
|
||||
if strings.HasPrefix(trimmed, "```") {
|
||||
inCodeBlock = !inCodeBlock
|
||||
continue
|
||||
}
|
||||
if inCodeBlock && trimmed != "" {
|
||||
codeLines = append(codeLines, trimmed)
|
||||
}
|
||||
}
|
||||
|
||||
if len(codeLines) == 0 {
|
||||
for _, line := range lines {
|
||||
if strings.TrimSpace(line) != "" {
|
||||
codeLines = append(codeLines, line)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if len(codeLines) == 0 {
|
||||
return p.fallbackLang
|
||||
}
|
||||
|
||||
sample := strings.Join(codeLines[:min(len(codeLines), 10)], "\n")
|
||||
lang := enry.GetLanguage("", []byte(sample))
|
||||
|
||||
if lang == "" {
|
||||
return p.fallbackLang
|
||||
}
|
||||
|
||||
return strings.ToLower(lang)
|
||||
}
|
||||
|
||||
func (p *Parser) splitIntoSegments(text string, lang string) []ContentSegment {
|
||||
segments := []ContentSegment{}
|
||||
|
||||
codeBlockPattern := regexp.MustCompile("(?s)```[\\s\\S]*?^```|`[^`]+`")
|
||||
matches := codeBlockPattern.FindAllStringIndex(text, -1)
|
||||
|
||||
if len(matches) == 0 {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeText,
|
||||
Content: text,
|
||||
StartPos: 0,
|
||||
EndPos: len(text),
|
||||
})
|
||||
return segments
|
||||
}
|
||||
|
||||
lastEnd := 0
|
||||
for _, match := range matches {
|
||||
start, end := match[0], match[1]
|
||||
|
||||
if start > lastEnd {
|
||||
textPart := text[lastEnd:start]
|
||||
textSegments := p.parseTextContent(textPart, lang)
|
||||
segments = append(segments, textSegments...)
|
||||
}
|
||||
|
||||
content := text[start:end]
|
||||
isInline := len(content) > 0 && content[0] == '`' && (len(content) == 1 || content[len(content)-1] == '`')
|
||||
|
||||
if strings.HasPrefix(content, "```") {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeCodeBlock,
|
||||
Content: content,
|
||||
Language: p.detectCodeBlockLang(content),
|
||||
StartPos: start,
|
||||
EndPos: end,
|
||||
})
|
||||
} else if isInline {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeInlineCode,
|
||||
Content: content,
|
||||
Language: lang,
|
||||
StartPos: start,
|
||||
EndPos: end,
|
||||
})
|
||||
}
|
||||
|
||||
lastEnd = end
|
||||
}
|
||||
|
||||
if lastEnd < len(text) {
|
||||
textPart := text[lastEnd:]
|
||||
textSegments := p.parseTextContent(textPart, lang)
|
||||
segments = append(segments, textSegments...)
|
||||
}
|
||||
|
||||
return segments
|
||||
}
|
||||
|
||||
func (p *Parser) parseTextContent(text string, lang string) []ContentSegment {
|
||||
segments := []ContentSegment{}
|
||||
langPatterns := getLanguagePatterns(lang)
|
||||
|
||||
if langPatterns.SingleLine == "" && len(langPatterns.MultiLine) == 0 {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeText,
|
||||
Content: text,
|
||||
Language: lang,
|
||||
StartPos: 0,
|
||||
EndPos: len(text),
|
||||
})
|
||||
return segments
|
||||
}
|
||||
|
||||
commentPatterns := p.buildCommentRegex(langPatterns)
|
||||
if commentPatterns == nil {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeText,
|
||||
Content: text,
|
||||
Language: lang,
|
||||
StartPos: 0,
|
||||
EndPos: len(text),
|
||||
})
|
||||
return segments
|
||||
}
|
||||
|
||||
matches := commentPatterns.FindAllStringIndex(text, -1)
|
||||
if len(matches) == 0 {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeText,
|
||||
Content: text,
|
||||
Language: lang,
|
||||
StartPos: 0,
|
||||
EndPos: len(text),
|
||||
})
|
||||
return segments
|
||||
}
|
||||
|
||||
lastEnd := 0
|
||||
for _, match := range matches {
|
||||
start, end := match[0], match[1]
|
||||
|
||||
if start > lastEnd {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeText,
|
||||
Content: text[lastEnd:start],
|
||||
Language: lang,
|
||||
StartPos: lastEnd,
|
||||
EndPos: start,
|
||||
})
|
||||
}
|
||||
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeComment,
|
||||
Content: text[start:end],
|
||||
IsComment: true,
|
||||
Language: lang,
|
||||
StartPos: start,
|
||||
EndPos: end,
|
||||
})
|
||||
|
||||
lastEnd = end
|
||||
}
|
||||
|
||||
if lastEnd < len(text) {
|
||||
segments = append(segments, ContentSegment{
|
||||
Type: SegmentTypeText,
|
||||
Content: text[lastEnd:],
|
||||
Language: lang,
|
||||
StartPos: lastEnd,
|
||||
EndPos: len(text),
|
||||
})
|
||||
}
|
||||
|
||||
return segments
|
||||
}
|
||||
|
||||
type languageCommentRegex struct {
|
||||
SingleLine string
|
||||
MultiLine []struct {
|
||||
Start string
|
||||
End string
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Parser) buildCommentRegex(patterns languageCommentRegex) *regexp.Regexp {
|
||||
var parts []string
|
||||
|
||||
if patterns.SingleLine != "" {
|
||||
parts = append(parts, patterns.SingleLine+`.*$`)
|
||||
}
|
||||
|
||||
for _, multi := range patterns.MultiLine {
|
||||
if multi.Start != "" && multi.End != "" {
|
||||
escapedStart := regexp.QuoteMeta(multi.Start)
|
||||
escapedEnd := regexp.QuoteMeta(multi.End)
|
||||
parts = append(parts, escapedStart+`[\s\S]*?`+escapedEnd)
|
||||
}
|
||||
}
|
||||
|
||||
if len(parts) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
pattern := `(?m)` + strings.Join(parts, "|")
|
||||
return regexp.MustCompile(pattern)
|
||||
}
|
||||
|
||||
func getLanguagePatterns(lang string) languageCommentRegex {
|
||||
patterns, ok := languagePatterns[lang]
|
||||
if !ok {
|
||||
patterns = defaultPatterns
|
||||
}
|
||||
|
||||
result := languageCommentRegex{
|
||||
SingleLine: patterns.LineComment,
|
||||
}
|
||||
|
||||
for _, bc := range patterns.BlockComment {
|
||||
if len(bc) >= 2 {
|
||||
result.MultiLine = append(result.MultiLine, struct {
|
||||
Start string
|
||||
End string
|
||||
}{Start: bc[:len(bc)/2], End: bc[len(bc)/2:]})
|
||||
}
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (p *Parser) detectCodeBlockLang(codeBlock string) string {
|
||||
lines := strings.Split(codeBlock, "\n")
|
||||
if len(lines) < 2 {
|
||||
return ""
|
||||
}
|
||||
|
||||
firstLine := strings.TrimSpace(lines[0])
|
||||
firstLine = strings.TrimPrefix(firstLine, "```")
|
||||
firstLine = strings.TrimSpace(firstLine)
|
||||
|
||||
if firstLine != "" {
|
||||
lang := strings.ToLower(firstLine)
|
||||
if _, ok := languagePatterns[lang]; ok {
|
||||
return lang
|
||||
}
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (p *Parser) BuildPrompt(result *ParseResult) string {
|
||||
var prompt strings.Builder
|
||||
|
||||
prompt.WriteString("你是一位专业的技术翻译。请翻译以下内容,遵守以下规则:\n\n")
|
||||
|
||||
prompt.WriteString("需要翻译的部分:\n")
|
||||
prompt.WriteString("- 普通文本:翻译成目标语言\n")
|
||||
prompt.WriteString("- 代码注释:只翻译注释中有意义的词汇,技术术语保留原语言\n\n")
|
||||
|
||||
prompt.WriteString("需要保持不变的部分:\n")
|
||||
prompt.WriteString("- 代码块(如 ```javascript ... ```)保持原样\n")
|
||||
prompt.WriteString("- 行内代码(如 `const count = 10`)保持原样\n")
|
||||
|
||||
if len(p.skipKeywords) > 0 {
|
||||
prompt.WriteString(fmt.Sprintf("- 以下关键词不翻译:%s\n", strings.Join(p.skipKeywords, "、")))
|
||||
}
|
||||
|
||||
prompt.WriteString("\n请将需要翻译的部分翻译成中文,其他部分保持不变。\n\n")
|
||||
prompt.WriteString("原文:\n---\n")
|
||||
|
||||
textToTranslate := p.extractTextForTranslation(result)
|
||||
prompt.WriteString(textToTranslate)
|
||||
|
||||
prompt.WriteString("\n---")
|
||||
|
||||
return prompt.String()
|
||||
}
|
||||
|
||||
func (p *Parser) extractTextForTranslation(result *ParseResult) string {
|
||||
var text strings.Builder
|
||||
|
||||
for _, seg := range result.Segments {
|
||||
switch seg.Type {
|
||||
case SegmentTypeText:
|
||||
text.WriteString(seg.Content)
|
||||
case SegmentTypeComment:
|
||||
text.WriteString(seg.Content)
|
||||
case SegmentTypeCodeBlock, SegmentTypeInlineCode:
|
||||
}
|
||||
}
|
||||
|
||||
return text.String()
|
||||
}
|
||||
|
||||
func (p *Parser) Reconstruct(result *ParseResult, translatedText string) string {
|
||||
translatedLines := strings.Split(translatedText, "\n")
|
||||
var output strings.Builder
|
||||
|
||||
textIndex := 0
|
||||
|
||||
for _, seg := range result.Segments {
|
||||
switch seg.Type {
|
||||
case SegmentTypeText, SegmentTypeComment:
|
||||
if textIndex < len(translatedLines) {
|
||||
output.WriteString(translatedLines[textIndex])
|
||||
textIndex++
|
||||
}
|
||||
case SegmentTypeCodeBlock, SegmentTypeInlineCode:
|
||||
output.WriteString(seg.Content)
|
||||
}
|
||||
}
|
||||
|
||||
return output.String()
|
||||
}
|
||||
|
||||
func min(a, b int) int {
|
||||
if a < b {
|
||||
return a
|
||||
}
|
||||
return b
|
||||
}
|
||||
317
internal/lang/lang.go
Normal file
317
internal/lang/lang.go
Normal file
@@ -0,0 +1,317 @@
|
||||
package lang
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sort"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// 语言代码映射表
|
||||
var languageMap = map[string]string{
|
||||
// 中文变体
|
||||
"cn": "zh-CN",
|
||||
"zh": "zh-CN", // 默认简体中文
|
||||
"zhcn": "zh-CN",
|
||||
"zhtw": "zh-TW",
|
||||
"zhhk": "zh-HK",
|
||||
"zh-hans": "zh-CN",
|
||||
"zh-hant": "zh-TW",
|
||||
"chinese": "zh-CN",
|
||||
"简体中文": "zh-CN",
|
||||
"繁体中文": "zh-TW",
|
||||
|
||||
// 英语变体
|
||||
"en": "en-US", // 默认美式英语
|
||||
"us": "en-US",
|
||||
"uk": "en-GB",
|
||||
"gb": "en-GB",
|
||||
"english": "en-US",
|
||||
"美式英语": "en-US",
|
||||
"英式英语": "en-GB",
|
||||
|
||||
// 日语
|
||||
"jp": "ja",
|
||||
"ja": "ja",
|
||||
"japanese": "ja",
|
||||
"日语": "ja",
|
||||
|
||||
// 韩语
|
||||
"kr": "ko",
|
||||
"ko": "ko",
|
||||
"korean": "ko",
|
||||
"韩语": "ko",
|
||||
|
||||
// 西班牙语
|
||||
"es": "es-ES",
|
||||
"spanish": "es-ES",
|
||||
"西班牙语": "es-ES",
|
||||
|
||||
// 法语
|
||||
"fr": "fr-FR",
|
||||
"french": "fr-FR",
|
||||
"法语": "fr-FR",
|
||||
|
||||
// 德语
|
||||
"de": "de-DE",
|
||||
"german": "de-DE",
|
||||
"德语": "de-DE",
|
||||
|
||||
// 俄语
|
||||
"ru": "ru-RU",
|
||||
"russian": "ru-RU",
|
||||
"俄语": "ru-RU",
|
||||
|
||||
// 葡萄牙语
|
||||
"pt": "pt-PT",
|
||||
"portuguese": "pt-PT",
|
||||
"葡萄牙语": "pt-PT",
|
||||
"br": "pt-BR", // 巴西葡萄牙语
|
||||
|
||||
// 意大利语
|
||||
"it": "it-IT",
|
||||
"italian": "it-IT",
|
||||
"意大利语": "it-IT",
|
||||
|
||||
// 阿拉伯语
|
||||
"ar": "ar-SA",
|
||||
"arabic": "ar-SA",
|
||||
"阿拉伯语": "ar-SA",
|
||||
|
||||
// 印地语
|
||||
"hi": "hi-IN",
|
||||
"hindi": "hi-IN",
|
||||
"印地语": "hi-IN",
|
||||
|
||||
// 其他语言
|
||||
"nl": "nl-NL", // 荷兰语
|
||||
"dutch": "nl-NL",
|
||||
"sv": "sv-SE", // 瑞典语
|
||||
"swedish": "sv-SE",
|
||||
"no": "nb-NO", // 挪威语
|
||||
"norwegian": "nb-NO",
|
||||
"da": "da-DK", // 丹麦语
|
||||
"danish": "da-DK",
|
||||
"fi": "fi-FI", // 芬兰语
|
||||
"finnish": "fi-FI",
|
||||
"pl": "pl-PL", // 波兰语
|
||||
"polish": "pl-PL",
|
||||
"tr": "tr-TR", // 土耳其语
|
||||
"turkish": "tr-TR",
|
||||
"th": "th-TH", // 泰语
|
||||
"thai": "th-TH",
|
||||
"vi": "vi-VN", // 越南语
|
||||
"vietnamese": "vi-VN",
|
||||
"id": "id-ID", // 印尼语
|
||||
"indonesian": "id-ID",
|
||||
"ms": "ms-MY", // 马来语
|
||||
"malay": "ms-MY",
|
||||
}
|
||||
|
||||
// 语言名称到代码的映射(用于显示)
|
||||
var languageNames = map[string]string{
|
||||
"zh-CN": "中文(简体)",
|
||||
"zh-TW": "中文(繁体)",
|
||||
"zh-HK": "中文(香港)",
|
||||
"en-US": "English (US)",
|
||||
"en-GB": "English (UK)",
|
||||
"ja": "日本語",
|
||||
"ko": "한국어",
|
||||
"es-ES": "Español",
|
||||
"fr-FR": "Français",
|
||||
"de-DE": "Deutsch",
|
||||
"ru-RU": "Русский",
|
||||
"pt-PT": "Português",
|
||||
"pt-BR": "Português (Brasil)",
|
||||
"it-IT": "Italiano",
|
||||
"ar-SA": "العربية",
|
||||
"hi-IN": "हिन्दी",
|
||||
"nl-NL": "Nederlands",
|
||||
"sv-SE": "Svenska",
|
||||
"nb-NO": "Norsk",
|
||||
"da-DK": "Dansk",
|
||||
"fi-FI": "Suomi",
|
||||
"pl-PL": "Polski",
|
||||
"tr-TR": "Türkçe",
|
||||
"th-TH": "ไทย",
|
||||
"vi-VN": "Tiếng Việt",
|
||||
"id-ID": "Bahasa Indonesia",
|
||||
"ms-MY": "Bahasa Melayu",
|
||||
}
|
||||
|
||||
// ParseLanguageCode 解析语言代码
|
||||
// 支持多种格式:标准BCP47格式、别名、中文名称等
|
||||
func ParseLanguageCode(input string) string {
|
||||
if input == "" {
|
||||
return ""
|
||||
}
|
||||
|
||||
// 转换为小写进行匹配
|
||||
lower := strings.ToLower(strings.TrimSpace(input))
|
||||
|
||||
// 直接匹配
|
||||
if code, exists := languageMap[lower]; exists {
|
||||
return code
|
||||
}
|
||||
|
||||
// 尝试解析BCP47格式(如 zh-CN, en-US)
|
||||
if isValidLanguageTag(input) {
|
||||
return normalizeLanguageTag(input)
|
||||
}
|
||||
|
||||
// 如果无法解析,返回原始输入
|
||||
return input
|
||||
}
|
||||
|
||||
// isValidLanguageTag 检查是否是有效的语言标签格式
|
||||
func isValidLanguageTag(tag string) bool {
|
||||
// 简单的格式检查:语言代码-地区代码
|
||||
parts := strings.Split(tag, "-")
|
||||
if len(parts) == 1 {
|
||||
// 只有语言代码,如 "zh", "en"
|
||||
return len(parts[0]) >= 2 && len(parts[0]) <= 3
|
||||
}
|
||||
if len(parts) == 2 {
|
||||
// 语言代码-地区代码,如 "zh-CN", "en-US"
|
||||
return len(parts[0]) >= 2 && len(parts[0]) <= 3 && len(parts[1]) == 2
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// normalizeLanguageTag 标准化语言标签
|
||||
func normalizeLanguageTag(tag string) string {
|
||||
parts := strings.Split(tag, "-")
|
||||
if len(parts) == 1 {
|
||||
// 只有语言代码,使用默认地区
|
||||
defaultRegions := map[string]string{
|
||||
"zh": "CN",
|
||||
"en": "US",
|
||||
"ja": "JP",
|
||||
"ko": "KR",
|
||||
"es": "ES",
|
||||
"fr": "FR",
|
||||
"de": "DE",
|
||||
"ru": "RU",
|
||||
"pt": "PT",
|
||||
"it": "IT",
|
||||
"ar": "SA",
|
||||
"hi": "IN",
|
||||
"nl": "NL",
|
||||
"sv": "SE",
|
||||
"no": "NO",
|
||||
"da": "DK",
|
||||
"fi": "FI",
|
||||
"pl": "PL",
|
||||
"tr": "TR",
|
||||
"th": "TH",
|
||||
"vi": "VN",
|
||||
"id": "ID",
|
||||
"ms": "MY",
|
||||
}
|
||||
if region, exists := defaultRegions[parts[0]]; exists {
|
||||
return fmt.Sprintf("%s-%s", strings.ToLower(parts[0]), strings.ToUpper(region))
|
||||
}
|
||||
return tag
|
||||
}
|
||||
if len(parts) == 2 {
|
||||
// 标准化格式:语言小写,地区大写
|
||||
return fmt.Sprintf("%s-%s", strings.ToLower(parts[0]), strings.ToUpper(parts[1]))
|
||||
}
|
||||
return tag
|
||||
}
|
||||
|
||||
// GetLanguageName 获取语言名称(用于显示)
|
||||
func GetLanguageName(code string) string {
|
||||
if name, exists := languageNames[code]; exists {
|
||||
return name
|
||||
}
|
||||
return code
|
||||
}
|
||||
|
||||
// GetLanguageNameOrDefault 获取语言名称,如果不存在则返回代码
|
||||
func GetLanguageNameOrDefault(code string, defaultName string) string {
|
||||
if name, exists := languageNames[code]; exists {
|
||||
return name
|
||||
}
|
||||
return defaultName
|
||||
}
|
||||
|
||||
// SupportedLanguages 获取支持的语言列表
|
||||
func SupportedLanguages() []string {
|
||||
codes := make([]string, 0, len(languageNames))
|
||||
for code := range languageNames {
|
||||
codes = append(codes, code)
|
||||
}
|
||||
sort.Strings(codes)
|
||||
return codes
|
||||
}
|
||||
|
||||
// GetLanguageSuggestions 获取语言建议(用于模糊匹配)
|
||||
func GetLanguageSuggestions(input string, limit int) []string {
|
||||
if input == "" {
|
||||
return []string{}
|
||||
}
|
||||
|
||||
lower := strings.ToLower(input)
|
||||
suggestions := make([]string, 0)
|
||||
|
||||
for alias, code := range languageMap {
|
||||
if strings.Contains(alias, lower) || strings.Contains(strings.ToLower(code), lower) {
|
||||
// 避免重复
|
||||
found := false
|
||||
for _, s := range suggestions {
|
||||
if s == code {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
suggestions = append(suggestions, code)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 限制数量
|
||||
if len(suggestions) > limit {
|
||||
suggestions = suggestions[:limit]
|
||||
}
|
||||
|
||||
return suggestions
|
||||
}
|
||||
|
||||
// IsLanguageSupported 检查语言是否支持
|
||||
func IsLanguageSupported(code string) bool {
|
||||
normalized := ParseLanguageCode(code)
|
||||
_, exists := languageNames[normalized]
|
||||
return exists
|
||||
}
|
||||
|
||||
// GetCommonLanguages 获取常用语言列表
|
||||
func GetCommonLanguages() []string {
|
||||
return []string{
|
||||
"zh-CN", // 中文(简体)
|
||||
"en-US", // 英语(美国)
|
||||
"ja", // 日语
|
||||
"ko", // 韩语
|
||||
"es-ES", // 西班牙语
|
||||
"fr-FR", // 法语
|
||||
"de-DE", // 德语
|
||||
"ru-RU", // 俄语
|
||||
"pt-PT", // 葡萄牙语
|
||||
"it-IT", // 意大利语
|
||||
}
|
||||
}
|
||||
|
||||
// GetLanguageDirection 获取语言方向(从左到右或从右到左)
|
||||
func GetLanguageDirection(code string) string {
|
||||
rtlLanguages := map[string]bool{
|
||||
"ar-SA": true, // 阿拉伯语
|
||||
"he-IL": true, // 希伯来语
|
||||
"fa-IR": true, // 波斯语
|
||||
"ur-PK": true, // 乌尔都语
|
||||
}
|
||||
|
||||
if rtlLanguages[code] {
|
||||
return "rtl"
|
||||
}
|
||||
return "ltr"
|
||||
}
|
||||
224
internal/lang/lang_test.go
Normal file
224
internal/lang/lang_test.go
Normal file
@@ -0,0 +1,224 @@
|
||||
package lang
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestParseLanguageCode(t *testing.T) {
|
||||
tests := []struct {
|
||||
input string
|
||||
expected string
|
||||
}{
|
||||
// 中文变体
|
||||
{"cn", "zh-CN"},
|
||||
{"zh", "zh-CN"},
|
||||
{"zh-CN", "zh-CN"},
|
||||
{"zh-TW", "zh-TW"},
|
||||
{"zh-HK", "zh-HK"},
|
||||
{"chinese", "zh-CN"},
|
||||
{"简体中文", "zh-CN"},
|
||||
|
||||
// 英语变体
|
||||
{"en", "en-US"},
|
||||
{"en-US", "en-US"},
|
||||
{"en-GB", "en-GB"},
|
||||
{"us", "en-US"},
|
||||
{"uk", "en-GB"},
|
||||
{"english", "en-US"},
|
||||
|
||||
// 其他语言
|
||||
{"jp", "ja"},
|
||||
{"ja", "ja"},
|
||||
{"japanese", "ja"},
|
||||
{"kr", "ko"},
|
||||
{"ko", "ko"},
|
||||
{"korean", "ko"},
|
||||
{"es", "es-ES"},
|
||||
{"spanish", "es-ES"},
|
||||
{"fr", "fr-FR"},
|
||||
{"french", "fr-FR"},
|
||||
{"de", "de-DE"},
|
||||
{"german", "de-DE"},
|
||||
|
||||
// 空值
|
||||
{"", ""},
|
||||
|
||||
// 未知语言(应返回原始输入)
|
||||
{"unknown", "unknown"},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.input, func(t *testing.T) {
|
||||
result := ParseLanguageCode(tt.input)
|
||||
if result != tt.expected {
|
||||
t.Errorf("ParseLanguageCode(%q) = %q, 期望 %q", tt.input, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetLanguageName(t *testing.T) {
|
||||
tests := []struct {
|
||||
code string
|
||||
expected string
|
||||
}{
|
||||
{"zh-CN", "中文(简体)"},
|
||||
{"zh-TW", "中文(繁体)"},
|
||||
{"en-US", "English (US)"},
|
||||
{"en-GB", "English (UK)"},
|
||||
{"ja", "日本語"},
|
||||
{"ko", "한국어"},
|
||||
{"es-ES", "Español"},
|
||||
{"fr-FR", "Français"},
|
||||
{"de-DE", "Deutsch"},
|
||||
{"unknown", "unknown"}, // 未知代码返回原值
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.code, func(t *testing.T) {
|
||||
result := GetLanguageName(tt.code)
|
||||
if result != tt.expected {
|
||||
t.Errorf("GetLanguageName(%q) = %q, 期望 %q", tt.code, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSupportedLanguages(t *testing.T) {
|
||||
languages := SupportedLanguages()
|
||||
if len(languages) == 0 {
|
||||
t.Error("SupportedLanguages() 不应返回空列表")
|
||||
}
|
||||
|
||||
// 检查一些关键语言是否在列表中
|
||||
expectedLanguages := []string{"zh-CN", "en-US", "ja", "ko"}
|
||||
for _, expected := range expectedLanguages {
|
||||
found := false
|
||||
for _, lang := range languages {
|
||||
if lang == expected {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
t.Errorf("Expected language %q not found in supported languages", expected)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetLanguageSuggestions(t *testing.T) {
|
||||
tests := []struct {
|
||||
input string
|
||||
limit int
|
||||
minCount int
|
||||
}{
|
||||
{"zh", 5, 1},
|
||||
{"en", 5, 1},
|
||||
{"chinese", 5, 1},
|
||||
{"", 5, 0},
|
||||
{"unknown", 5, 0},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.input, func(t *testing.T) {
|
||||
suggestions := GetLanguageSuggestions(tt.input, tt.limit)
|
||||
if len(suggestions) < tt.minCount {
|
||||
t.Errorf("GetLanguageSuggestions(%q, %d) 返回 %d 个建议,至少需要 %d 个",
|
||||
tt.input, tt.limit, len(suggestions), tt.minCount)
|
||||
}
|
||||
if len(suggestions) > tt.limit {
|
||||
t.Errorf("GetLanguageSuggestions(%q, %d) 返回 %d 个建议,超过限制 %d 个",
|
||||
tt.input, tt.limit, len(suggestions), tt.limit)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestIsLanguageSupported(t *testing.T) {
|
||||
tests := []struct {
|
||||
code string
|
||||
expected bool
|
||||
}{
|
||||
{"zh-CN", true},
|
||||
{"en-US", true},
|
||||
{"ja", true},
|
||||
{"unknown", false},
|
||||
{"", false},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.code, func(t *testing.T) {
|
||||
result := IsLanguageSupported(tt.code)
|
||||
if result != tt.expected {
|
||||
t.Errorf("IsLanguageSupported(%q) = %v, 期望 %v", tt.code, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetCommonLanguages(t *testing.T) {
|
||||
languages := GetCommonLanguages()
|
||||
if len(languages) == 0 {
|
||||
t.Error("GetCommonLanguages() 不应返回空列表")
|
||||
}
|
||||
|
||||
// 检查一些关键语言是否在列表中
|
||||
expectedLanguages := []string{"zh-CN", "en-US", "ja"}
|
||||
for _, expected := range expectedLanguages {
|
||||
found := false
|
||||
for _, lang := range languages {
|
||||
if lang == expected {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
t.Errorf("Expected language %q not found in common languages", expected)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetLanguageDirection(t *testing.T) {
|
||||
tests := []struct {
|
||||
code string
|
||||
expected string
|
||||
}{
|
||||
{"zh-CN", "ltr"},
|
||||
{"en-US", "ltr"},
|
||||
{"ja", "ltr"},
|
||||
{"ar-SA", "rtl"},
|
||||
{"he-IL", "rtl"},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.code, func(t *testing.T) {
|
||||
result := GetLanguageDirection(tt.code)
|
||||
if result != tt.expected {
|
||||
t.Errorf("GetLanguageDirection(%q) = %q, 期望 %q", tt.code, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestNormalizeLanguageTag(t *testing.T) {
|
||||
tests := []struct {
|
||||
input string
|
||||
expected string
|
||||
}{
|
||||
{"zh", "zh-CN"},
|
||||
{"en", "en-US"},
|
||||
{"ja", "ja-JP"},
|
||||
{"zh-CN", "zh-CN"},
|
||||
{"zh-tw", "zh-TW"},
|
||||
{"EN-us", "en-US"},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.input, func(t *testing.T) {
|
||||
result := normalizeLanguageTag(tt.input)
|
||||
if result != tt.expected {
|
||||
t.Errorf("normalizeLanguageTag(%q) = %q, 期望 %q", tt.input, result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
305
internal/onboard/onboard.go
Normal file
305
internal/onboard/onboard.go
Normal file
@@ -0,0 +1,305 @@
|
||||
package onboard
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/AlecAivazis/survey/v2"
|
||||
"github.com/titor/fanyi/internal/config"
|
||||
"github.com/titor/fanyi/internal/lang"
|
||||
)
|
||||
|
||||
// RunOnboard 启动配置向导
|
||||
func RunOnboard(force bool) error {
|
||||
fmt.Println("欢迎使用YOYO翻译工具配置向导!")
|
||||
fmt.Println("这个向导将帮助您配置翻译工具。")
|
||||
fmt.Println()
|
||||
|
||||
// 检查配置文件是否存在
|
||||
configPath := "configs/config.yaml"
|
||||
if _, err := os.Stat(configPath); err == nil && !force {
|
||||
overwrite := false
|
||||
prompt := &survey.Confirm{
|
||||
Message: "检测到配置文件已存在,是否要重新配置?",
|
||||
Default: false,
|
||||
}
|
||||
if err := survey.AskOne(prompt, &overwrite); err != nil {
|
||||
return fmt.Errorf("用户输入错误: %w", err)
|
||||
}
|
||||
if !overwrite {
|
||||
fmt.Println("配置已取消。")
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
// 步骤1: 选择主要厂商
|
||||
fmt.Println("步骤1: 选择主要翻译服务提供商")
|
||||
providerName, err := SelectProvider()
|
||||
if err != nil {
|
||||
return fmt.Errorf("选择厂商失败: %w", err)
|
||||
}
|
||||
|
||||
// 步骤2: 配置主要厂商
|
||||
fmt.Println("\n步骤2: 配置主要厂商")
|
||||
providerConfig, err := ConfigureProvider(providerName)
|
||||
if err != nil {
|
||||
return fmt.Errorf("配置厂商失败: %w", err)
|
||||
}
|
||||
|
||||
// 步骤3: 全局设置
|
||||
fmt.Println("\n步骤3: 全局设置")
|
||||
globalConfig, err := GlobalSettings()
|
||||
if err != nil {
|
||||
return fmt.Errorf("全局设置失败: %w", err)
|
||||
}
|
||||
|
||||
// 步骤4: 确认并保存配置
|
||||
fmt.Println("\n步骤4: 保存配置")
|
||||
configData := BuildConfig(providerName, providerConfig, globalConfig)
|
||||
|
||||
if err := SaveConfig(configData, configPath); err != nil {
|
||||
return fmt.Errorf("保存配置失败: %w", err)
|
||||
}
|
||||
|
||||
fmt.Printf("\n配置完成! 配置文件已保存到: %s\n", configPath)
|
||||
fmt.Println("\n您现在可以使用以下命令进行翻译:")
|
||||
fmt.Println(" yoyo \"Hello world\"")
|
||||
fmt.Println(" yoyo --lang=cn \"Hello world\"")
|
||||
fmt.Println("\n更多帮助请运行: yoyo --help")
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// SelectProvider 选择主要厂商
|
||||
func SelectProvider() (string, error) {
|
||||
providers := []string{
|
||||
"siliconflow",
|
||||
"volcano",
|
||||
"national",
|
||||
"qwen",
|
||||
"openai",
|
||||
}
|
||||
|
||||
providerNames := map[string]string{
|
||||
"siliconflow": "硅基流动 (推荐,免费额度)",
|
||||
"volcano": "火山引擎",
|
||||
"national": "国家超算",
|
||||
"qwen": "Qwen (通义千问)",
|
||||
"openai": "OpenAI兼容格式",
|
||||
}
|
||||
|
||||
var selected string
|
||||
prompt := &survey.Select{
|
||||
Message: "请选择要使用的翻译服务提供商:",
|
||||
Options: func() []string {
|
||||
var opts []string
|
||||
for _, p := range providers {
|
||||
opts = append(opts, providerNames[p])
|
||||
}
|
||||
return opts
|
||||
}(),
|
||||
Default: providerNames["siliconflow"],
|
||||
}
|
||||
|
||||
if err := survey.AskOne(prompt, &selected); err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// 返回对应的厂商名称
|
||||
for name, displayName := range providerNames {
|
||||
if displayName == selected {
|
||||
return name, nil
|
||||
}
|
||||
}
|
||||
|
||||
return "siliconflow", nil
|
||||
}
|
||||
|
||||
// ConfigureProvider 配置厂商
|
||||
func ConfigureProvider(providerName string) (config.ProviderConfig, error) {
|
||||
// 厂商默认配置
|
||||
defaults := map[string]config.ProviderConfig{
|
||||
"siliconflow": {
|
||||
APIHost: "https://api.siliconflow.cn/v1",
|
||||
Model: "siliconflow-base",
|
||||
Enabled: true,
|
||||
},
|
||||
"volcano": {
|
||||
APIHost: "https://api.volcengine.com/v1",
|
||||
Model: "volcano-chat",
|
||||
Enabled: true,
|
||||
},
|
||||
"national": {
|
||||
APIHost: "https://api.nsc.gov.cn/v1",
|
||||
Model: "nsc-base",
|
||||
Enabled: true,
|
||||
},
|
||||
"qwen": {
|
||||
APIHost: "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
||||
Model: "qwen-turbo",
|
||||
Enabled: true,
|
||||
},
|
||||
"openai": {
|
||||
APIHost: "https://api.openai.com/v1",
|
||||
Model: "gpt-3.5-turbo",
|
||||
Enabled: true,
|
||||
},
|
||||
}
|
||||
|
||||
defaultConfig := defaults[providerName]
|
||||
cfg := config.ProviderConfig{
|
||||
APIHost: defaultConfig.APIHost,
|
||||
Model: defaultConfig.Model,
|
||||
Enabled: defaultConfig.Enabled,
|
||||
}
|
||||
|
||||
// 输入API密钥
|
||||
apiKeyPrompt := &survey.Input{
|
||||
Message: fmt.Sprintf("请输入 %s 的API密钥:", providerName),
|
||||
Help: "API密钥用于身份验证,将存储在配置文件中",
|
||||
}
|
||||
if err := survey.AskOne(apiKeyPrompt, &cfg.APIKey, survey.WithValidator(survey.Required)); err != nil {
|
||||
return config.ProviderConfig{}, err
|
||||
}
|
||||
|
||||
// 确认API HOST
|
||||
apiHostPrompt := &survey.Input{
|
||||
Message: "API HOST (直接回车使用默认值):",
|
||||
Default: cfg.APIHost,
|
||||
}
|
||||
if err := survey.AskOne(apiHostPrompt, &cfg.APIHost); err != nil {
|
||||
return config.ProviderConfig{}, err
|
||||
}
|
||||
|
||||
// 确认默认模型
|
||||
modelPrompt := &survey.Input{
|
||||
Message: "默认模型 (直接回车使用默认值):",
|
||||
Default: cfg.Model,
|
||||
}
|
||||
if err := survey.AskOne(modelPrompt, &cfg.Model); err != nil {
|
||||
return config.ProviderConfig{}, err
|
||||
}
|
||||
|
||||
return cfg, nil
|
||||
}
|
||||
|
||||
// GlobalSettings 全局设置
|
||||
type GlobalConfig struct {
|
||||
DefaultProvider string
|
||||
DefaultModel string
|
||||
Timeout int
|
||||
DefaultSourceLang string
|
||||
DefaultTargetLang string
|
||||
}
|
||||
|
||||
// GlobalSettings 全局设置
|
||||
func GlobalSettings() (*GlobalConfig, error) {
|
||||
cfg := &GlobalConfig{
|
||||
DefaultProvider: "siliconflow",
|
||||
DefaultModel: "siliconflow-base",
|
||||
Timeout: 30,
|
||||
DefaultSourceLang: "auto",
|
||||
DefaultTargetLang: "zh-CN",
|
||||
}
|
||||
|
||||
// 选择默认语言
|
||||
targetLangOptions := lang.GetCommonLanguages()
|
||||
var targetLangDisplay []string
|
||||
for _, code := range targetLangOptions {
|
||||
targetLangDisplay = append(targetLangDisplay, fmt.Sprintf("%s (%s)", code, lang.GetLanguageName(code)))
|
||||
}
|
||||
|
||||
targetLangPrompt := &survey.Select{
|
||||
Message: "请选择默认目标语言:",
|
||||
Options: targetLangDisplay,
|
||||
Default: fmt.Sprintf("%s (%s)", "zh-CN", lang.GetLanguageName("zh-CN")),
|
||||
}
|
||||
|
||||
var selectedTarget string
|
||||
if err := survey.AskOne(targetLangPrompt, &selectedTarget); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 从选择中提取语言代码
|
||||
for i, display := range targetLangDisplay {
|
||||
if display == selectedTarget {
|
||||
cfg.DefaultTargetLang = targetLangOptions[i]
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
// 设置超时时间
|
||||
timeoutPrompt := &survey.Input{
|
||||
Message: "API超时时间(秒):",
|
||||
Default: fmt.Sprintf("%d", cfg.Timeout),
|
||||
}
|
||||
var timeoutStr string
|
||||
if err := survey.AskOne(timeoutPrompt, &timeoutStr); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 解析超时时间
|
||||
if timeout := parseIntOrDefault(timeoutStr, 30); timeout > 0 {
|
||||
cfg.Timeout = timeout
|
||||
}
|
||||
|
||||
return cfg, nil
|
||||
}
|
||||
|
||||
// BuildConfig 构建配置对象
|
||||
func BuildConfig(providerName string, providerConfig config.ProviderConfig, globalConfig *GlobalConfig) *config.Config {
|
||||
// 创建厂商配置
|
||||
providers := map[string]config.ProviderConfig{
|
||||
providerName: providerConfig,
|
||||
}
|
||||
|
||||
// 创建Prompt配置
|
||||
prompts := map[string]string{
|
||||
"technical": "你是一位专业的技术翻译,请准确翻译以下技术文档,保持专业术语的准确性。",
|
||||
"creative": "你是一位富有创造力的翻译家,请用优美流畅的语言翻译以下内容。",
|
||||
"academic": "你是一位学术翻译专家,请用严谨的学术语言翻译以下内容。",
|
||||
"simple": "请用简单易懂的语言翻译以下内容。",
|
||||
}
|
||||
|
||||
return &config.Config{
|
||||
DefaultProvider: providerName,
|
||||
DefaultModel: providerConfig.Model,
|
||||
Timeout: globalConfig.Timeout,
|
||||
DefaultSourceLang: globalConfig.DefaultSourceLang,
|
||||
DefaultTargetLang: globalConfig.DefaultTargetLang,
|
||||
Providers: providers,
|
||||
Prompts: prompts,
|
||||
}
|
||||
}
|
||||
|
||||
// SaveConfig 保存配置文件
|
||||
func SaveConfig(cfg *config.Config, path string) error {
|
||||
// 确保目录存在
|
||||
dir := filepath.Dir(path)
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
return fmt.Errorf("创建配置目录失败: %w", err)
|
||||
}
|
||||
|
||||
// 使用config包的Save方法
|
||||
loader := &config.YAMLConfigLoader{}
|
||||
return loader.Save(cfg, path)
|
||||
}
|
||||
|
||||
// parseIntOrDefault 解析整数,失败时返回默认值
|
||||
func parseIntOrDefault(s string, defaultValue int) int {
|
||||
if s == "" {
|
||||
return defaultValue
|
||||
}
|
||||
|
||||
var result int
|
||||
if _, err := fmt.Sscanf(s, "%d", &result); err != nil {
|
||||
return defaultValue
|
||||
}
|
||||
|
||||
if result <= 0 {
|
||||
return defaultValue
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
@@ -5,24 +5,44 @@ import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/titor/fanyi/internal/cache"
|
||||
"github.com/titor/fanyi/internal/config"
|
||||
"github.com/titor/fanyi/internal/content"
|
||||
"github.com/titor/fanyi/internal/provider"
|
||||
)
|
||||
|
||||
// Translator 核心翻译类
|
||||
type Translator struct {
|
||||
config *config.Config
|
||||
provider provider.Provider
|
||||
prompt *PromptManager
|
||||
config *config.Config
|
||||
provider provider.Provider
|
||||
prompt *PromptManager
|
||||
contentParser *content.Parser
|
||||
cache cache.Cache
|
||||
}
|
||||
|
||||
// NewTranslator 创建翻译器实例
|
||||
func NewTranslator(config *config.Config, provider provider.Provider) *Translator {
|
||||
return &Translator{
|
||||
config: config,
|
||||
provider: provider,
|
||||
prompt: NewPromptManager(config.Prompts),
|
||||
translator := &Translator{
|
||||
config: config,
|
||||
provider: provider,
|
||||
prompt: NewPromptManager(config.Prompts),
|
||||
contentParser: content.NewParser(config.SkipKeywords),
|
||||
}
|
||||
|
||||
// 初始化缓存(如果启用)
|
||||
if config.Cache.Enabled {
|
||||
cacheConfig := &cache.CacheConfig{
|
||||
Enabled: config.Cache.Enabled,
|
||||
MaxRecords: config.Cache.MaxRecords,
|
||||
ExpireDays: config.Cache.ExpireDays,
|
||||
DBPath: config.Cache.DBPath,
|
||||
}
|
||||
if cacheInstance, err := cache.NewSQLiteCache(cacheConfig); err == nil {
|
||||
translator.cache = cacheInstance
|
||||
}
|
||||
}
|
||||
|
||||
return translator
|
||||
}
|
||||
|
||||
// Translate 执行翻译
|
||||
@@ -31,15 +51,33 @@ func (t *Translator) Translate(ctx context.Context, text string, options *Transl
|
||||
timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(t.config.Timeout)*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// 基础字符过滤
|
||||
filteredText := content.FilterBasic(text, nil)
|
||||
|
||||
// 内容解析(包含代码检测)
|
||||
parseResult, parseErr := t.contentParser.Parse(filteredText)
|
||||
|
||||
// 选择Prompt
|
||||
prompt := ""
|
||||
if options.PromptName != "" {
|
||||
prompt = t.prompt.GetPrompt(options.PromptName)
|
||||
}
|
||||
|
||||
// 如果包含代码且解析成功,使用增强的Prompt
|
||||
if parseErr == nil && parseResult.HasCode {
|
||||
enhancedPrompt := t.contentParser.BuildPrompt(parseResult)
|
||||
if enhancedPrompt != "" {
|
||||
if prompt != "" {
|
||||
prompt = prompt + "\n\n" + enhancedPrompt
|
||||
} else {
|
||||
prompt = enhancedPrompt
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 构建请求
|
||||
req := &provider.TranslateRequest{
|
||||
Text: text,
|
||||
Text: filteredText,
|
||||
FromLang: options.FromLang,
|
||||
ToLang: options.ToLang,
|
||||
Prompt: prompt,
|
||||
@@ -47,16 +85,63 @@ func (t *Translator) Translate(ctx context.Context, text string, options *Transl
|
||||
Options: options.ExtraOptions,
|
||||
}
|
||||
|
||||
// 检查缓存
|
||||
if t.cache != nil {
|
||||
cacheKey := cache.GenerateCacheKey(filteredText, options.FromLang, options.ToLang)
|
||||
if cachedEntry, err := t.cache.Get(ctx, cacheKey); err == nil && cachedEntry != nil {
|
||||
// 缓存命中
|
||||
return &TranslateResult{
|
||||
Original: text,
|
||||
Translated: cachedEntry.TranslatedText,
|
||||
FromLang: cachedEntry.FromLang,
|
||||
ToLang: cachedEntry.ToLang,
|
||||
Model: cachedEntry.Model,
|
||||
Usage: &provider.Usage{
|
||||
PromptTokens: cachedEntry.PromptTokens,
|
||||
CompletionTokens: cachedEntry.CompletionTokens,
|
||||
TotalTokens: cachedEntry.TotalTokens,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
}
|
||||
|
||||
// 调用厂商API
|
||||
resp, err := t.provider.Translate(timeoutCtx, req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("翻译失败: %w", err)
|
||||
}
|
||||
|
||||
translatedText := resp.Text
|
||||
|
||||
// 如果包含代码且解析成功,重构结果
|
||||
if parseErr == nil && parseResult.HasCode {
|
||||
translatedText = t.contentParser.Reconstruct(parseResult, resp.Text)
|
||||
}
|
||||
|
||||
// 保存到缓存
|
||||
if t.cache != nil {
|
||||
cacheKey := cache.GenerateCacheKey(filteredText, options.FromLang, options.ToLang)
|
||||
cacheEntry := &cache.CacheEntry{
|
||||
CacheKey: cacheKey,
|
||||
OriginalText: filteredText,
|
||||
TranslatedText: translatedText,
|
||||
FromLang: resp.FromLang,
|
||||
ToLang: resp.ToLang,
|
||||
Model: resp.Model,
|
||||
PromptName: options.PromptName,
|
||||
PromptContent: prompt,
|
||||
PromptTokens: resp.Usage.PromptTokens,
|
||||
CompletionTokens: resp.Usage.CompletionTokens,
|
||||
TotalTokens: resp.Usage.TotalTokens,
|
||||
}
|
||||
// 异步保存缓存,不阻塞翻译结果返回
|
||||
go t.cache.Set(context.Background(), cacheEntry)
|
||||
}
|
||||
|
||||
// 构建结果
|
||||
return &TranslateResult{
|
||||
Original: text,
|
||||
Translated: resp.Text,
|
||||
Translated: translatedText,
|
||||
FromLang: resp.FromLang,
|
||||
ToLang: resp.ToLang,
|
||||
Model: resp.Model,
|
||||
|
||||
48
internal/tui/keys.go
Normal file
48
internal/tui/keys.go
Normal file
@@ -0,0 +1,48 @@
|
||||
package tui
|
||||
|
||||
import (
|
||||
"charm.land/bubbles/v2/key"
|
||||
)
|
||||
|
||||
type KeyMap struct {
|
||||
Quit key.Binding
|
||||
Clear key.Binding
|
||||
SwitchLang key.Binding
|
||||
ScrollUp key.Binding
|
||||
ScrollDown key.Binding
|
||||
ScrollTop key.Binding
|
||||
ScrollBottom key.Binding
|
||||
}
|
||||
|
||||
func NewKeyMap() KeyMap {
|
||||
return KeyMap{
|
||||
Quit: key.NewBinding(
|
||||
key.WithKeys("ctrl+c", "esc"),
|
||||
key.WithHelp("Ctrl+C", "退出"),
|
||||
),
|
||||
Clear: key.NewBinding(
|
||||
key.WithKeys("ctrl+l"),
|
||||
key.WithHelp("Ctrl+L", "清空输入"),
|
||||
),
|
||||
SwitchLang: key.NewBinding(
|
||||
key.WithKeys("ctrl+t"),
|
||||
key.WithHelp("Ctrl+T", "切换语言"),
|
||||
),
|
||||
ScrollUp: key.NewBinding(
|
||||
key.WithKeys("up", "ctrl+up"),
|
||||
key.WithHelp("↑/Ctrl+↑", "上滚"),
|
||||
),
|
||||
ScrollDown: key.NewBinding(
|
||||
key.WithKeys("down", "ctrl+down"),
|
||||
key.WithHelp("↓/Ctrl+↓", "下滚"),
|
||||
),
|
||||
ScrollTop: key.NewBinding(
|
||||
key.WithKeys("home"),
|
||||
key.WithHelp("Home", "顶部"),
|
||||
),
|
||||
ScrollBottom: key.NewBinding(
|
||||
key.WithKeys("end"),
|
||||
key.WithHelp("End", "底部"),
|
||||
),
|
||||
}
|
||||
}
|
||||
21
internal/tui/messages.go
Normal file
21
internal/tui/messages.go
Normal file
@@ -0,0 +1,21 @@
|
||||
package tui
|
||||
|
||||
import "time"
|
||||
|
||||
type ChatMessage struct {
|
||||
ID string
|
||||
Input string
|
||||
Output string
|
||||
FromLang string
|
||||
ToLang string
|
||||
Model string
|
||||
Tokens int
|
||||
Timestamp time.Time
|
||||
Error string
|
||||
}
|
||||
|
||||
type ChatHistory struct {
|
||||
Messages []ChatMessage
|
||||
scrollPos int
|
||||
totalLines int
|
||||
}
|
||||
371
internal/tui/model.go
Normal file
371
internal/tui/model.go
Normal file
@@ -0,0 +1,371 @@
|
||||
package tui
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"charm.land/bubbles/v2/textarea"
|
||||
"charm.land/bubbles/v2/viewport"
|
||||
"charm.land/bubbletea/v2"
|
||||
"charm.land/lipgloss/v2"
|
||||
"github.com/titor/fanyi/internal/config"
|
||||
"github.com/titor/fanyi/internal/translator"
|
||||
)
|
||||
|
||||
var supportedLangs = []string{"zh-CN", "en-US", "ja", "ko", "zh-TW", "es", "fr", "de"}
|
||||
|
||||
type translateMsg struct {
|
||||
result string
|
||||
tokens int
|
||||
err error
|
||||
}
|
||||
|
||||
type model struct {
|
||||
config *config.Config
|
||||
translator *translator.Translator
|
||||
|
||||
messages []ChatMessage
|
||||
input textarea.Model
|
||||
viewport viewport.Model
|
||||
keys KeyMap
|
||||
|
||||
targetLang string
|
||||
langIndex int
|
||||
loading bool
|
||||
lastInput string
|
||||
|
||||
width int
|
||||
height int
|
||||
inputHeight int
|
||||
}
|
||||
|
||||
func NewApp(cfg *config.Config, t *translator.Translator) *tea.Program {
|
||||
keys := NewKeyMap()
|
||||
|
||||
ta := textarea.New()
|
||||
ta.Placeholder = "输入要翻译的文本... (Ctrl+J 换行)"
|
||||
ta.Focus()
|
||||
ta.Prompt = ""
|
||||
ta.ShowLineNumbers = false
|
||||
ta.SetWidth(60)
|
||||
ta.SetHeight(5)
|
||||
ta.SetStyles(textarea.DefaultStyles(false))
|
||||
|
||||
ta.KeyMap.InsertNewline.SetEnabled(true)
|
||||
|
||||
vp := viewport.New(viewport.WithWidth(50), viewport.WithHeight(20))
|
||||
vp.SetContent("")
|
||||
|
||||
return tea.NewProgram(model{
|
||||
config: cfg,
|
||||
translator: t,
|
||||
messages: make([]ChatMessage, 0),
|
||||
input: ta,
|
||||
viewport: vp,
|
||||
keys: keys,
|
||||
targetLang: getDefaultLang(cfg),
|
||||
})
|
||||
}
|
||||
|
||||
func getDefaultLang(cfg *config.Config) string {
|
||||
if cfg != nil && cfg.DefaultTargetLang != "" {
|
||||
return cfg.DefaultTargetLang
|
||||
}
|
||||
return "zh-CN"
|
||||
}
|
||||
|
||||
func (m model) Init() tea.Cmd {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
var (
|
||||
cmd tea.Cmd
|
||||
cmds []tea.Cmd
|
||||
)
|
||||
|
||||
switch msg := msg.(type) {
|
||||
case tea.WindowSizeMsg:
|
||||
m.width = msg.Width
|
||||
m.height = msg.Height
|
||||
m.updateLayout()
|
||||
|
||||
case translateMsg:
|
||||
m.loading = false
|
||||
if msg.err != nil {
|
||||
m.addErrorMessage(msg.err.Error())
|
||||
} else {
|
||||
m.addSuccessMessage(msg.result, msg.tokens)
|
||||
}
|
||||
m.updateViewportContent()
|
||||
|
||||
case tea.KeyPressMsg:
|
||||
switch msg.String() {
|
||||
case "ctrl+c", "esc":
|
||||
return m, tea.Quit
|
||||
case "ctrl+l":
|
||||
m.input.Reset()
|
||||
case "ctrl+t":
|
||||
m.langIndex = (m.langIndex + 1) % len(supportedLangs)
|
||||
m.targetLang = supportedLangs[m.langIndex]
|
||||
case "enter":
|
||||
text := strings.TrimSpace(m.input.Value())
|
||||
if text != "" && !m.loading {
|
||||
m.input.Reset()
|
||||
m.lastInput = text
|
||||
m.loading = true
|
||||
cmds = append(cmds, m.doTranslate(text, m.targetLang))
|
||||
}
|
||||
case "alt+up":
|
||||
m.viewport.ScrollUp(3)
|
||||
case "alt+down":
|
||||
m.viewport.ScrollDown(3)
|
||||
case "home":
|
||||
m.viewport.GotoTop()
|
||||
case "end":
|
||||
m.viewport.GotoBottom()
|
||||
}
|
||||
}
|
||||
|
||||
m.input, cmd = m.input.Update(msg)
|
||||
cmds = append(cmds, cmd)
|
||||
|
||||
m.viewport, cmd = m.viewport.Update(msg)
|
||||
cmds = append(cmds, cmd)
|
||||
|
||||
return m, tea.Batch(cmds...)
|
||||
}
|
||||
|
||||
func (m *model) addSuccessMessage(result string, tokens int) {
|
||||
msg := ChatMessage{
|
||||
Input: m.lastInput,
|
||||
Output: result,
|
||||
ToLang: m.targetLang,
|
||||
Model: getModelName(m.config),
|
||||
Tokens: tokens,
|
||||
Timestamp: time.Now(),
|
||||
}
|
||||
m.messages = append(m.messages, msg)
|
||||
}
|
||||
|
||||
func (m *model) addErrorMessage(err string) {
|
||||
msg := ChatMessage{
|
||||
Input: m.lastInput,
|
||||
Error: err,
|
||||
ToLang: m.targetLang,
|
||||
Timestamp: time.Now(),
|
||||
}
|
||||
m.messages = append(m.messages, msg)
|
||||
}
|
||||
|
||||
func (m *model) doTranslate(text, toLang string) tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
result, err := m.translator.Translate(
|
||||
context.Background(),
|
||||
text,
|
||||
&translator.TranslateOptions{
|
||||
ToLang: toLang,
|
||||
PromptName: "simple",
|
||||
},
|
||||
)
|
||||
if err != nil {
|
||||
return translateMsg{err: err}
|
||||
}
|
||||
tokens := 0
|
||||
if result.Usage != nil {
|
||||
tokens = result.Usage.TotalTokens
|
||||
}
|
||||
return translateMsg{result: result.Translated, tokens: tokens}
|
||||
}
|
||||
}
|
||||
|
||||
func (m *model) updateLayout() {
|
||||
if m.width <= 0 || m.height <= 0 {
|
||||
return
|
||||
}
|
||||
|
||||
contentWidth := m.width - 4
|
||||
if contentWidth < 20 {
|
||||
contentWidth = 60
|
||||
}
|
||||
|
||||
m.input.SetWidth(contentWidth)
|
||||
m.viewport.SetWidth(contentWidth)
|
||||
m.viewport.SetHeight(m.height - 12)
|
||||
if m.viewport.Height() < 5 {
|
||||
m.viewport.SetHeight(10)
|
||||
}
|
||||
|
||||
m.updateViewportContent()
|
||||
}
|
||||
|
||||
func (m *model) updateViewportContent() {
|
||||
var b strings.Builder
|
||||
|
||||
for _, msg := range m.messages {
|
||||
b.WriteString(m.renderTranslationCard(msg))
|
||||
}
|
||||
|
||||
m.viewport.SetContent(b.String())
|
||||
m.viewport.GotoBottom()
|
||||
}
|
||||
|
||||
func (m *model) renderTranslationCard(msg ChatMessage) string {
|
||||
metaContent := lipgloss.JoinHorizontal(
|
||||
lipgloss.Left,
|
||||
CardMetaStyle.Render(fmt.Sprintf("Tokens: %d", msg.Tokens)),
|
||||
CardMetaSeparatorStyle,
|
||||
CardMetaStyle.Render(fmt.Sprintf("耗时: %s", msg.Timestamp.Format("15:04:05"))),
|
||||
CardMetaSeparatorStyle,
|
||||
CardMetaStyle.Render(fmt.Sprintf("模型: %s", msg.Model)),
|
||||
)
|
||||
|
||||
metaBlock := lipgloss.NewStyle().
|
||||
BorderStyle(lipgloss.Border{
|
||||
Top: "─",
|
||||
Bottom: "─",
|
||||
Left: "│",
|
||||
Right: "│",
|
||||
}).
|
||||
BorderForeground(lipgloss.Color("#374151")).
|
||||
Width(m.viewport.Width() - 2).
|
||||
Render(metaContent)
|
||||
|
||||
inputContent := lipgloss.NewStyle().
|
||||
Width(m.viewport.Width() - 2).
|
||||
Render(msg.Input)
|
||||
|
||||
inputBlock := lipgloss.NewStyle().
|
||||
Background(lipgloss.Color("#1A1A1A")).
|
||||
Width(m.viewport.Width()).
|
||||
Render(inputContent)
|
||||
|
||||
var outputBlock string
|
||||
if msg.Error != "" {
|
||||
outputContent := lipgloss.NewStyle().
|
||||
Width(m.viewport.Width() - 2).
|
||||
Render(msg.Error)
|
||||
outputBlock = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#F87171")).
|
||||
Width(m.viewport.Width()).
|
||||
Render(outputContent)
|
||||
} else {
|
||||
outputContent := lipgloss.NewStyle().
|
||||
Width(m.viewport.Width() - 2).
|
||||
Render(msg.Output)
|
||||
outputBlock = lipgloss.NewStyle().
|
||||
Width(m.viewport.Width()).
|
||||
Render(outputContent)
|
||||
}
|
||||
|
||||
return CardStyle.Render(
|
||||
lipgloss.JoinVertical(
|
||||
lipgloss.Top,
|
||||
metaBlock,
|
||||
inputBlock,
|
||||
outputBlock,
|
||||
),
|
||||
) + "\n"
|
||||
}
|
||||
|
||||
func getModelName(cfg *config.Config) string {
|
||||
if cfg != nil && cfg.DefaultModel != "" {
|
||||
return cfg.DefaultModel
|
||||
}
|
||||
return "gpt-3.5-turbo"
|
||||
}
|
||||
|
||||
func (m model) View() tea.View {
|
||||
if m.width == 0 {
|
||||
return tea.NewView("正在加载...")
|
||||
}
|
||||
|
||||
header := m.renderHeader()
|
||||
messages := m.viewport.View()
|
||||
inputArea := m.renderInputArea()
|
||||
statusBar := m.renderStatusBar()
|
||||
|
||||
content := header + "\n" + messages + inputArea + statusBar
|
||||
v := tea.NewView(content)
|
||||
v.AltScreen = true
|
||||
return v
|
||||
}
|
||||
|
||||
func (m model) renderHeader() string {
|
||||
title := lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#8B5CF6")).
|
||||
Bold(true).
|
||||
Render("✦ YOYO 翻译")
|
||||
|
||||
width := m.width - 4
|
||||
if width < 20 {
|
||||
width = 60
|
||||
}
|
||||
|
||||
right := lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#6B7280")).
|
||||
Render("[Ctrl+C 退出]")
|
||||
|
||||
return lipgloss.NewStyle().
|
||||
Width(width).
|
||||
Render(title + strings.Repeat(" ", width-len(title)-len(right)-1) + right)
|
||||
}
|
||||
|
||||
func (m model) renderInputArea() string {
|
||||
inputView := m.input.View()
|
||||
|
||||
container := lipgloss.NewStyle().
|
||||
Width(m.input.Width() + 1).
|
||||
BorderStyle(lipgloss.Border{
|
||||
Top: "─",
|
||||
Bottom: "─",
|
||||
Left: "│",
|
||||
Right: "│",
|
||||
}).
|
||||
BorderForeground(lipgloss.Color("#60A5FA"))
|
||||
|
||||
return "\n" + container.Render(inputView) + "\n"
|
||||
}
|
||||
|
||||
func (m model) renderStatusBar() string {
|
||||
langInfo := "目标: " + m.targetLang
|
||||
modelInfo := "模型: " + getModelName(m.config)
|
||||
tokensInfo := "Tokens: -"
|
||||
if len(m.messages) > 0 {
|
||||
lastMsg := m.messages[len(m.messages)-1]
|
||||
if lastMsg.Tokens > 0 {
|
||||
tokensInfo = fmt.Sprintf("Tokens: %d", lastMsg.Tokens)
|
||||
}
|
||||
}
|
||||
|
||||
statusDot := StatusDotStyle.Render("●")
|
||||
if m.loading {
|
||||
statusDot = LoadingStyle.Render("○")
|
||||
}
|
||||
|
||||
sep := StatusItemStyle.Render(" │ ")
|
||||
|
||||
width := m.width - 4
|
||||
if width < 30 {
|
||||
width = 60
|
||||
}
|
||||
|
||||
status := StatusItemStyle.Render(langInfo) +
|
||||
sep + StatusItemStyle.Render(modelInfo) +
|
||||
sep + StatusItemStyle.Render(tokensInfo) +
|
||||
sep + statusDot + " " + StatusValueStyle.Render(m.getStatusText())
|
||||
|
||||
return lipgloss.NewStyle().
|
||||
Width(width).
|
||||
Background(lipgloss.Color("#1F2937")).
|
||||
Render(" " + status)
|
||||
}
|
||||
|
||||
func (m model) getStatusText() string {
|
||||
if m.loading {
|
||||
return "翻译中..."
|
||||
}
|
||||
return "就绪"
|
||||
}
|
||||
71
internal/tui/styles.go
Normal file
71
internal/tui/styles.go
Normal file
@@ -0,0 +1,71 @@
|
||||
package tui
|
||||
|
||||
import (
|
||||
"charm.land/lipgloss/v2"
|
||||
)
|
||||
|
||||
var (
|
||||
HeaderStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#FAFAFA")).
|
||||
Bold(true).
|
||||
Padding(0, 1)
|
||||
|
||||
InputLabelStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#60A5FA")).
|
||||
Bold(true)
|
||||
|
||||
OutputLabelStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#34D399")).
|
||||
Bold(true)
|
||||
|
||||
InputTextStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#E5E7EB"))
|
||||
|
||||
OutputTextStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#FFFFFF"))
|
||||
|
||||
ErrorTextStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#F87171")).
|
||||
Bold(true)
|
||||
|
||||
TimestampStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#6B7280"))
|
||||
|
||||
DividerStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#374151"))
|
||||
|
||||
StatusBarStyle = lipgloss.NewStyle().
|
||||
Background(lipgloss.Color("#1F2937")).
|
||||
Foreground(lipgloss.Color("#9CA3AF"))
|
||||
|
||||
StatusItemStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#9CA3AF"))
|
||||
|
||||
StatusValueStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#FAFAFA"))
|
||||
|
||||
LoadingStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#60A5FA"))
|
||||
|
||||
StatusDotStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#34D399"))
|
||||
|
||||
CardStyle = lipgloss.NewStyle().
|
||||
MarginBottom(5)
|
||||
|
||||
CardMetaStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#6B7280")).
|
||||
Padding(0, 1)
|
||||
|
||||
CardMetaSeparatorStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#374151")).
|
||||
Render(" │ ")
|
||||
|
||||
CardInputStyle = lipgloss.NewStyle().
|
||||
Background(lipgloss.Color("#1A1A1A")).
|
||||
Foreground(lipgloss.Color("#E5E7EB")).
|
||||
Padding(0, 1)
|
||||
|
||||
CardOutputStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#FFFFFF"))
|
||||
)
|
||||
708
memory.md
708
memory.md
@@ -188,4 +188,710 @@ func main() {
|
||||
1. 用户编辑why.md记录初衷
|
||||
2. AI编辑taolun.md记录讨论
|
||||
3. AI更新changelog.md记录版本
|
||||
4. AI更新memory.md记录经验
|
||||
4. AI更新memory.md记录经验
|
||||
|
||||
---
|
||||
|
||||
## 语言代码处理经验
|
||||
|
||||
### 语言代码标准化
|
||||
**问题**: 需要支持多种语言代码格式,但内部应使用标准格式
|
||||
**解决方案**:
|
||||
1. 使用BCP 47语言标签作为标准格式(如 `zh-CN`、`en-US`)
|
||||
2. 实现智能解析函数 `ParseLanguageCode()`
|
||||
3. 支持别名映射(如 `cn` → `zh-CN`、`en` → `en-US`)
|
||||
|
||||
**最佳实践**:
|
||||
- 语言代码小写,地区代码大写(如 `zh-CN`,不是 `zh-cn`)
|
||||
- 提供语言名称映射用于显示(如 `zh-CN` → "中文(简体)")
|
||||
- 支持模糊匹配和建议功能
|
||||
|
||||
### 交互式配置经验
|
||||
**问题**: 命令行工具需要友好的配置界面
|
||||
**解决方案**:
|
||||
1. 使用 `github.com/AlecAivazis/survey/v2` 库
|
||||
2. 实现分步配置流程
|
||||
3. 提供默认值和确认选项
|
||||
|
||||
**注意事项**:
|
||||
- 交互式库需要终端支持
|
||||
- 提供非交互式模式(如配置文件模板)
|
||||
- 错误处理要友好,避免程序崩溃
|
||||
|
||||
### 命令行参数解析经验
|
||||
**问题**: Go标准库 `flag` 包功能有限,需要支持子命令
|
||||
**解决方案**:
|
||||
1. 使用 `flag` 包解析选项参数
|
||||
2. 手动处理子命令(如 `onboard`)
|
||||
3. 提供清晰的帮助信息
|
||||
|
||||
**命名冲突处理**:
|
||||
- 避免变量名与包名冲突(如 `onboard` 变量与 `onboard` 包)
|
||||
- 使用后缀区分(如 `onboardFlag`)
|
||||
|
||||
## 配置文件管理经验
|
||||
|
||||
### 开发阶段配置策略
|
||||
**决策**: 开发阶段使用 `.env` + `configs/config.yaml`
|
||||
**原因**:
|
||||
1. 简化开发环境配置
|
||||
2. 符合12-factor应用原则
|
||||
3. 避免过早优化
|
||||
|
||||
**实施**:
|
||||
- `.env` 文件存储API密钥等敏感信息
|
||||
- `configs/config.yaml` 存储复杂配置结构
|
||||
- 使用环境变量替换 `${VAR}`
|
||||
|
||||
### 配置文件格式选择
|
||||
**决策**: 使用YAML格式
|
||||
**原因**:
|
||||
1. 人类可读性好
|
||||
2. 支持复杂数据结构
|
||||
3. Go生态支持良好
|
||||
|
||||
**注意事项**:
|
||||
- 使用 `gopkg.in/yaml.v3` 库
|
||||
- 注意缩进和格式
|
||||
- 提供配置验证
|
||||
|
||||
---
|
||||
|
||||
## 管道功能实现经验
|
||||
|
||||
### 管道输入检测
|
||||
**问题**: 如何检测是否有管道输入?
|
||||
**解决方案**:
|
||||
1. 使用 `os.Stdin.Stat()` 获取标准输入的状态
|
||||
2. 检查 `fileInfo.Mode() & os.ModeCharDevice` 是否为0
|
||||
3. 如果为0,表示是管道或文件重定向
|
||||
|
||||
**代码示例**:
|
||||
```go
|
||||
func isPipeInput() bool {
|
||||
fileInfo, err := os.Stdin.Stat()
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
return (fileInfo.Mode()&os.ModeCharDevice) == 0
|
||||
}
|
||||
```
|
||||
|
||||
**注意事项**:
|
||||
- 在管道模式下,即使没有命令行参数也应继续执行
|
||||
- 管道输入可能为空,需要处理空输入情况
|
||||
- 错误信息应输出到stderr,避免污染管道输出
|
||||
|
||||
### 管道输入读取
|
||||
**问题**: 如何从标准输入读取所有内容?
|
||||
**解决方案**:
|
||||
1. 使用 `bufio.Scanner` 逐行读取
|
||||
2. 使用 `strings.Join()` 合并多行
|
||||
3. 处理读取错误
|
||||
|
||||
**代码示例**:
|
||||
```go
|
||||
func readFromStdin() (string, error) {
|
||||
scanner := bufio.NewScanner(os.Stdin)
|
||||
var lines []string
|
||||
for scanner.Scan() {
|
||||
lines = append(lines, scanner.Text())
|
||||
}
|
||||
if err := scanner.Err(); err != nil {
|
||||
return "", fmt.Errorf("读取标准输入失败: %w", err)
|
||||
}
|
||||
return strings.Join(lines, "\n"), nil
|
||||
}
|
||||
```
|
||||
|
||||
### 静默模式设计
|
||||
**问题**: 管道使用时,统计信息会污染输出
|
||||
**解决方案**:
|
||||
1. 添加 `--quiet` 和 `-q` 参数
|
||||
2. 将统计信息输出到stderr
|
||||
3. 静默模式下只输出翻译结果
|
||||
|
||||
**设计原则**:
|
||||
- 翻译结果始终输出到stdout
|
||||
- 统计信息输出到stderr
|
||||
- 静默模式下抑制统计信息输出
|
||||
- 错误信息始终输出到stderr
|
||||
|
||||
### 正则表达式转义
|
||||
**问题**: 特殊字符在正则表达式中需要转义
|
||||
**解决方案**:
|
||||
1. 使用 `regexp.QuoteMeta()` 转义特殊字符
|
||||
2. 避免手动拼接正则表达式
|
||||
|
||||
**代码示例**:
|
||||
```go
|
||||
// 错误的方式
|
||||
pattern := regexp.MustCompile(symbol + `{21,}`) // 如果symbol是*会出错
|
||||
|
||||
// 正确的方式
|
||||
pattern := regexp.MustCompile(regexp.QuoteMeta(symbol) + `{21,}`)
|
||||
```
|
||||
|
||||
**特殊字符列表**:
|
||||
- `*`, `+`, `?`, `.`, `^`, `$`, `|`, `(`, `)`, `[`, `]`, `{`, `}`
|
||||
|
||||
### 输出重定向
|
||||
**问题**: 如何确保管道输出不被其他信息污染?
|
||||
**解决方案**:
|
||||
1. 翻译结果使用 `fmt.Println()` 输出到stdout
|
||||
2. 统计信息使用 `fmt.Fprintf(os.Stderr, ...)` 输出到stderr
|
||||
3. 错误信息也输出到stderr
|
||||
|
||||
**最佳实践**:
|
||||
- 始终区分stdout和stderr的使用场景
|
||||
- 为管道功能提供静默模式选项
|
||||
- 确保错误信息不影响管道输出
|
||||
|
||||
---
|
||||
|
||||
## 本地缓存实现经验
|
||||
|
||||
### SQLite数据库设计
|
||||
**问题**: 如何设计缓存表结构以支持高效的查询和存储?
|
||||
**解决方案**:
|
||||
1. 使用缓存键(cache_key)作为唯一索引
|
||||
2. 包含完整字段:原文、译文、语言对、模型、Prompt、用量统计
|
||||
3. 添加created_at和last_used_at时间戳字段
|
||||
4. 创建适当的索引提高查询性能
|
||||
|
||||
**表结构**:
|
||||
```sql
|
||||
CREATE TABLE translation_cache (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
cache_key TEXT NOT NULL UNIQUE,
|
||||
original_text TEXT NOT NULL,
|
||||
translated_text TEXT NOT NULL,
|
||||
from_lang TEXT NOT NULL,
|
||||
to_lang TEXT NOT NULL,
|
||||
model TEXT NOT NULL,
|
||||
prompt_name TEXT,
|
||||
prompt_content TEXT,
|
||||
prompt_tokens INTEGER DEFAULT 0,
|
||||
completion_tokens INTEGER DEFAULT 0,
|
||||
total_tokens INTEGER DEFAULT 0,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
last_used_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
```
|
||||
|
||||
**索引设计**:
|
||||
```sql
|
||||
CREATE INDEX idx_cache_key ON translation_cache(cache_key);
|
||||
CREATE INDEX idx_original_text ON translation_cache(original_text);
|
||||
CREATE INDEX idx_created_at ON translation_cache(created_at);
|
||||
CREATE INDEX idx_last_used_at ON translation_cache(last_used_at);
|
||||
```
|
||||
|
||||
### 缓存键生成策略
|
||||
**问题**: 如何生成唯一的缓存键,确保相同内容返回相同结果?
|
||||
**解决方案**:
|
||||
1. 使用SHA256哈希算法
|
||||
2. 哈希输入:原文+语言对(from_lang+to_lang)
|
||||
3. 规范化输入:移除多余空白字符,统一语言代码格式
|
||||
|
||||
**代码示例**:
|
||||
```go
|
||||
func GenerateCacheKey(originalText, fromLang, toLang string) string {
|
||||
// 规范化语言代码
|
||||
fromLang = normalizeLanguageCode(fromLang)
|
||||
toLang = normalizeLanguageCode(toLang)
|
||||
|
||||
// 规范化原文
|
||||
normalizedText := normalizeText(originalText)
|
||||
|
||||
// 生成缓存键
|
||||
data := fmt.Sprintf("%s|%s|%s", normalizedText, fromLang, toLang)
|
||||
hash := sha256.Sum256([]byte(data))
|
||||
return hex.EncodeToString(hash[:])
|
||||
}
|
||||
```
|
||||
|
||||
### SQLite NULL值处理
|
||||
**问题**: 当缓存表为空时,MIN(created_at)返回NULL,导致时间转换错误
|
||||
**解决方案**:
|
||||
1. 使用sql.NullString和sql.NullFloat64类型
|
||||
2. 检查Valid字段判断是否为NULL
|
||||
3. 只在有记录时才查询时间范围
|
||||
|
||||
**代码示例**:
|
||||
```go
|
||||
var oldestStr, newestStr sql.NullString
|
||||
var avgTokens sql.NullFloat64
|
||||
err = c.db.QueryRowContext(ctx, `
|
||||
SELECT MIN(created_at), MAX(created_at), AVG(total_tokens)
|
||||
FROM translation_cache
|
||||
`).Scan(&oldestStr, &newestStr, &avgTokens)
|
||||
|
||||
if oldestStr.Valid {
|
||||
stats.OldestRecord, _ = time.Parse("2006-01-02 15:04:05", oldestStr.String)
|
||||
}
|
||||
```
|
||||
|
||||
### 缓存集成策略
|
||||
**问题**: 如何将缓存功能集成到现有的翻译流程中?
|
||||
**解决方案**:
|
||||
1. 在Translator结构中添加缓存字段
|
||||
2. 修改NewTranslator函数,初始化缓存实例
|
||||
3. 在Translate方法中实现"先查缓存再调用API"策略
|
||||
4. 翻译成功后异步保存到缓存
|
||||
|
||||
**集成流程**:
|
||||
```
|
||||
翻译请求 → 生成缓存键 → 查询缓存 → 缓存命中? → 直接返回
|
||||
↓ 否
|
||||
调用API翻译 → 保存到缓存 → 返回结果
|
||||
```
|
||||
|
||||
### 异步缓存保存
|
||||
**问题**: 如何避免缓存保存阻塞翻译结果返回?
|
||||
**解决方案**:
|
||||
1. 使用goroutine异步保存缓存
|
||||
2. 使用context.Background()创建新的上下文
|
||||
3. 错误日志记录到标准错误输出
|
||||
|
||||
**代码示例**:
|
||||
```go
|
||||
// 异步保存缓存,不阻塞翻译结果返回
|
||||
go t.cache.Set(context.Background(), cacheEntry)
|
||||
```
|
||||
|
||||
### 组合清理策略
|
||||
**问题**: 如何平衡缓存大小和缓存时效性?
|
||||
**解决方案**:
|
||||
1. 数量限制:限制最大记录数
|
||||
2. 时间过期:设置缓存过期时间
|
||||
3. 组合策略:同时应用两种限制
|
||||
|
||||
**清理逻辑**:
|
||||
1. 清理过期记录:`DELETE FROM cache WHERE last_used_at < ?`
|
||||
2. 清理超出数量限制:保留最近使用的N条记录
|
||||
3. 定时清理:每小时自动清理一次
|
||||
|
||||
### 性能优化
|
||||
**SQLite性能优化**:
|
||||
1. 使用WAL模式(Write-Ahead Logging)
|
||||
2. 设置适当的连接池参数
|
||||
3. 使用索引提高查询性能
|
||||
4. 批量操作使用事务
|
||||
|
||||
**Go SQLite配置**:
|
||||
```go
|
||||
db, err := sql.Open("sqlite3", dbPath+"?_journal_mode=WAL&_synchronous=NORMAL")
|
||||
db.SetMaxOpenConns(1) // SQLite只支持单个写入连接
|
||||
db.SetMaxIdleConns(1)
|
||||
```
|
||||
|
||||
### 错误处理经验
|
||||
**常见错误及解决方案**:
|
||||
1. **NULL值转换错误**: 使用sql.NullString类型
|
||||
2. **时间解析错误**: 使用正确的时间格式字符串
|
||||
3. **数据库锁定错误**: 设置适当的连接池参数
|
||||
4. **权限错误**: 确保目录存在且有写入权限
|
||||
|
||||
### 测试策略
|
||||
**单元测试要点**:
|
||||
1. 测试缓存键生成的一致性
|
||||
2. 测试缓存命中和未命中情况
|
||||
3. 测试清理策略的正确性
|
||||
4. 测试NULL值处理
|
||||
5. 测试并发访问安全性
|
||||
|
||||
**集成测试要点**:
|
||||
1. 测试缓存与Translator的集成
|
||||
2. 测试配置文件的正确加载
|
||||
3. 测试缓存命令的正确执行
|
||||
4. 测试错误情况的正确处理
|
||||
|
||||
---
|
||||
|
||||
## TUI界面开发策略
|
||||
|
||||
### 分模块实现策略
|
||||
**决策**: TUI界面分6个小模块逐步实现,每次只做一个模块
|
||||
**原因**:
|
||||
1. 减少Token消耗和上下文负担
|
||||
2. 便于逐步测试和验证
|
||||
3. 适应free模式的限流约束
|
||||
|
||||
**模块列表**:
|
||||
1. TUI框架搭建 - bubbletea基础App结构
|
||||
2. 输入组件 - 文本输入框
|
||||
3. 翻译显示区 - 结果展示、滚动
|
||||
4. 状态栏/主题 - 底部状态栏、配色
|
||||
5. 快捷键系统 - 操作快捷键
|
||||
6. 集成翻译 - 对接Translator、加载动画
|
||||
|
||||
### 技术选型
|
||||
**决策**: 使用charmbracelet生态
|
||||
- `bubbletea` - Elm架构的TUI框架
|
||||
- `lipgloss` - 样式和主题
|
||||
- `bubbles/textinput` - 输入框组件
|
||||
|
||||
**原因**:
|
||||
1. Go生态最流行的TUI框架
|
||||
2. Elm架构清晰,易于分模块实现
|
||||
3. 组件化设计,便于复用
|
||||
|
||||
### 文档管理规范
|
||||
**决策**: 不再创建新的md文件
|
||||
**原因**: 保持项目文档简洁,避免碎片化
|
||||
**规则**:
|
||||
- 讨论内容 → `taolun.md`
|
||||
- 版本更新 → `changelog.md`
|
||||
- 经验教训 → `memory.md`
|
||||
- 项目初衷 → `why.md` (仅用户编辑)
|
||||
|
||||
---
|
||||
|
||||
## Bubble Tea TUI框架经验
|
||||
|
||||
### 版本信息
|
||||
**当前版本**: v1.3.10
|
||||
**API风格**: v1版本风格(Init返回tea.Cmd)
|
||||
|
||||
### 基础结构
|
||||
```go
|
||||
type model struct {
|
||||
// 状态字段
|
||||
}
|
||||
|
||||
func (m model) Init() tea.Cmd {
|
||||
return nil // 初始化命令
|
||||
}
|
||||
|
||||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
return m, nil // 更新逻辑
|
||||
}
|
||||
|
||||
func (m model) View() string {
|
||||
return "视图内容" // 返回渲染内容
|
||||
}
|
||||
|
||||
func NewApp(cfg, translator) *tea.Program {
|
||||
return tea.NewProgram(model{...})
|
||||
}
|
||||
```
|
||||
|
||||
### 关键点
|
||||
- `tea.NewProgram()` 创建程序实例
|
||||
- `program.Run()` 返回 `(model, error)`
|
||||
- model的字段可以是config、translator等依赖
|
||||
- View方法返回string,使用lipgloss样式
|
||||
|
||||
### main.go集成注意
|
||||
- 版本检查(--version)需要在interactive模式检查之前
|
||||
- 避免interactive模式在非TTY环境启动
|
||||
- Run()需要两个返回值: `_, err := app.Run()`
|
||||
|
||||
### TextInput组件使用
|
||||
```go
|
||||
import "github.com/charmbracelet/bubbles/textinput"
|
||||
|
||||
// model中添加字段
|
||||
type model struct {
|
||||
textInput textinput.Model
|
||||
}
|
||||
|
||||
// 初始化
|
||||
ti := textinput.New()
|
||||
ti.Placeholder = "输入文本..."
|
||||
ti.Focus() // 获取焦点
|
||||
ti.Prompt = "> " // 提示符
|
||||
|
||||
// Update中处理
|
||||
m.textInput, cmd = m.textInput.Update(msg)
|
||||
|
||||
// View中显示
|
||||
m.textInput.View()
|
||||
```
|
||||
|
||||
### Lipgloss样式定义
|
||||
```go
|
||||
var (
|
||||
headerStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#00D9FF")).
|
||||
Bold(true)
|
||||
resultStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#98FB98")).
|
||||
Background(lipgloss.Color("#0D1B2A"))
|
||||
helpStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#888888"))
|
||||
keyStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("#60A5FA"))
|
||||
)
|
||||
```
|
||||
|
||||
### 多区域View渲染
|
||||
```go
|
||||
func (m model) View() string {
|
||||
return "\n" +
|
||||
" " + headerStyle.Render("YOYO翻译") + "\n" +
|
||||
" " + divider + "\n\n" +
|
||||
" " + m.textInput.View() + "\n\n" +
|
||||
m.renderResult() +
|
||||
helpText
|
||||
}
|
||||
|
||||
func (m model) renderResult() string {
|
||||
if m.result == "" {
|
||||
return " " + helpStyle.Render("翻译结果将显示在这里...") + "\n"
|
||||
}
|
||||
return " " + resultStyle.Render(m.result) + "\n"
|
||||
}
|
||||
|
||||
### 异步命令和消息模式
|
||||
```go
|
||||
// 定义自定义消息类型
|
||||
type translateMsg struct {
|
||||
result string
|
||||
err error
|
||||
}
|
||||
|
||||
// 异步执行函数
|
||||
func (m model) doTranslate(text string) tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
result, err := translate(text)
|
||||
if err != nil {
|
||||
return translateMsg{err: err}
|
||||
}
|
||||
return translateMsg{result: result}
|
||||
}
|
||||
}
|
||||
|
||||
// Update中处理消息
|
||||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
case translateMsg:
|
||||
if msg.err != nil {
|
||||
m.errMsg = msg.err.Error()
|
||||
} else {
|
||||
m.result = msg.result
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
```
|
||||
|
||||
### 加载状态处理
|
||||
```go
|
||||
type model struct {
|
||||
loading bool
|
||||
errMsg string
|
||||
}
|
||||
|
||||
// View中显示loading
|
||||
func (m model) View() string {
|
||||
if m.loading {
|
||||
return "正在翻译..."
|
||||
}
|
||||
if m.errMsg != "" {
|
||||
return "错误: " + m.errMsg
|
||||
}
|
||||
return m.result
|
||||
}
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## TUI界面改进知识
|
||||
|
||||
### TextInput vs Textarea
|
||||
- **textinput**: 单行输入,适合短文本
|
||||
- **textarea**: 多行输入,适合长段落
|
||||
- 切换时需要调整布局和样式
|
||||
|
||||
### Modal/弹出框设计
|
||||
```go
|
||||
type model struct {
|
||||
showModal bool
|
||||
modalType string // "help", "command", "info"
|
||||
}
|
||||
|
||||
// View中渲染modal
|
||||
func (m model) View() string {
|
||||
s := "主界面..."
|
||||
if m.showModal {
|
||||
s += m.renderModal()
|
||||
}
|
||||
return s
|
||||
}
|
||||
```
|
||||
|
||||
### 斜杠命令菜单
|
||||
```go
|
||||
type command struct {
|
||||
name string
|
||||
desc string
|
||||
handler func()
|
||||
}
|
||||
|
||||
var commands = []command{
|
||||
{"help", "显示帮助", handleHelp},
|
||||
{"clear", "清空内容", handleClear},
|
||||
{"copy", "复制结果", handleCopy},
|
||||
}
|
||||
|
||||
// 模糊匹配
|
||||
func matchCommand(input string) []command {
|
||||
// 过滤匹配的命令
|
||||
}
|
||||
```
|
||||
|
||||
### Viewport组件
|
||||
用于长文本滚动显示,配合scrollbar展示滚动位置。
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## TUI输入框踩坑记录 (v0.8.0)
|
||||
|
||||
### 问题1:Ctrl+J换行后第一行被遮住
|
||||
|
||||
**现象**:
|
||||
- 按Ctrl+J换行后,第一行内容往上滚动被遮住
|
||||
- 光标在新行,但下方显示一个空行
|
||||
- 实际渲染了3行,但只显示2行
|
||||
|
||||
**尝试过的方案**:
|
||||
1. 移除 lipgloss Width() 限制 - 无效
|
||||
2. 设置 SetWidth() 后再 SetHeight() - 无效
|
||||
3. 动态计算行数后调用 SetHeight() - 无效
|
||||
4. 移除 updateInputHeight() 调用 - 无效
|
||||
|
||||
**根因分析**:
|
||||
- textarea 内部使用 viewport 组件管理滚动
|
||||
- 每次按键后动态调用 `m.input.SetHeight(lines)` 调整高度
|
||||
- 导致 textarea 内部 viewport 滚动位置与渲染不同步
|
||||
|
||||
**最终解决方案**:
|
||||
- 放弃动态调整高度的方案
|
||||
- 固定 textarea 高度为5行
|
||||
- 超过5行时,textarea 内部自动滚动,光标始终可见
|
||||
|
||||
**关键代码** (`internal/tui/model.go`):
|
||||
```go
|
||||
ta.SetWidth(60)
|
||||
ta.SetHeight(5) // 固定高度,不动态调整
|
||||
```
|
||||
|
||||
### 问题2:输入框背景颜色
|
||||
|
||||
**解决方案**:
|
||||
- 使用 `textarea.DefaultStyles()` 获取默认样式
|
||||
- 修改 Style.Base 设置背景色
|
||||
|
||||
**关键代码**:
|
||||
```go
|
||||
focusedStyle, blurredStyle := textarea.DefaultStyles()
|
||||
bgStyle := lipgloss.NewStyle().
|
||||
Background(lipgloss.Color("#1F2937")).
|
||||
Foreground(lipgloss.Color("#FAFAFA"))
|
||||
focusedStyle.Base = bgStyle
|
||||
blurredStyle.Base = bgStyle
|
||||
ta.FocusedStyle = focusedStyle
|
||||
ta.BlurredStyle = blurredStyle
|
||||
```
|
||||
|
||||
### 经验总结
|
||||
|
||||
1. **Bubble Tea的textarea组件**:内部包含viewport,不适合频繁动态调整高度
|
||||
2. **固定高度方案**:更稳定,让组件内部控制滚动
|
||||
3. **样式设置**:使用 FocusedStyle/BlurredStyle + Style.Base 而非直接设置 Style
|
||||
|
||||
---
|
||||
|
||||
## Bubble Tea/Lipgloss v2 升级经验 (v0.8.1)
|
||||
|
||||
### 模块路径变更
|
||||
|
||||
v2 版本全部迁移到 `charm.land` 域名:
|
||||
|
||||
| v1 | v2 |
|
||||
|----|-----|
|
||||
| `github.com/charmbracelet/lipgloss` | `charm.land/lipgloss/v2` |
|
||||
| `github.com/charmbracelet/bubbletea` | `charm.land/bubbletea/v2` |
|
||||
| `github.com/charmbracelet/bubbles` | `charm.land/bubbles/v2` |
|
||||
|
||||
### View() 方法变更
|
||||
|
||||
v1:
|
||||
```go
|
||||
func (m model) View() string {
|
||||
return "内容"
|
||||
}
|
||||
```
|
||||
|
||||
v2:
|
||||
```go
|
||||
func (m model) View() tea.View {
|
||||
v := tea.NewView("内容")
|
||||
v.AltScreen = true // 进入备用屏幕
|
||||
return v
|
||||
}
|
||||
```
|
||||
|
||||
### KeyMsg 变更
|
||||
|
||||
v1:
|
||||
```go
|
||||
case tea.KeyMsg:
|
||||
switch msg.Type {
|
||||
case tea.KeyCtrlC:
|
||||
return m, tea.Quit
|
||||
}
|
||||
```
|
||||
|
||||
v2:
|
||||
```go
|
||||
case tea.KeyPressMsg:
|
||||
switch msg.String() {
|
||||
case "ctrl+c":
|
||||
return m, tea.Quit
|
||||
case "space": // 注意:空格键改为"space"
|
||||
}
|
||||
```
|
||||
|
||||
### 快捷键对比
|
||||
|
||||
| 功能 | v1 | v2 |
|
||||
|-----|-----|-----|
|
||||
| Ctrl+C | `tea.KeyCtrlC` | `"ctrl+c"` |
|
||||
| Alt修饰键 | `msg.Alt` | `msg.Mod.Contains(tea.ModAlt)` |
|
||||
| 空格键 | `" "` | `"space"` |
|
||||
|
||||
### viewport API变更
|
||||
|
||||
| 功能 | v1 | v2 |
|
||||
|-----|-----|-----|
|
||||
| 滚动上 | `m.viewport.LineUp(n)` | `m.viewport.ScrollUp(n)` |
|
||||
| 滚动下 | `m.viewport.LineDown(n)` | `m.viewport.ScrollDown(n)` |
|
||||
| 宽度 | `m.viewport.Width` | `m.viewport.Width()` |
|
||||
| 高度 | `m.viewport.Height` | `m.viewport.Height()` |
|
||||
| 创建 | `viewport.New(w, h)` | `viewport.New(viewport.WithWidth(w), viewport.WithHeight(h))` |
|
||||
|
||||
### textarea API变更
|
||||
|
||||
| 功能 | v1 | v2 |
|
||||
|-----|-----|-----|
|
||||
| 默认样式 | `textarea.DefaultStyles()` | `textarea.DefaultStyles(isDark bool)` |
|
||||
| 重置内容 | `m.input.SetValue("")` | `m.input.Reset()` |
|
||||
|
||||
### Program启动
|
||||
|
||||
v1:
|
||||
```go
|
||||
p := tea.NewProgram(model{})
|
||||
p.Start()
|
||||
```
|
||||
|
||||
v2:
|
||||
```go
|
||||
p := tea.NewProgram(model{})
|
||||
p.Run()
|
||||
```
|
||||
612
taolun.md
612
taolun.md
@@ -148,4 +148,614 @@
|
||||
|
||||
**关联文档**:
|
||||
- [memory.md#环境变量加载问题](memory.md#环境变量加载问题)
|
||||
- [changelog.md#0.0.3](changelog.md#003)
|
||||
- [changelog.md#0.0.3](changelog.md#003)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-29 10:00] 版本 0.2.0 - 语言代码解析设计
|
||||
**原因**: 用户需要通过 `--lang` 参数指定目标语言,支持多种语言代码格式
|
||||
**分析**:
|
||||
- 需要支持标准BCP47格式(如 `zh-CN`、`en-US`)
|
||||
- 需要支持简短别名(如 `cn`、`en`)
|
||||
- 需要支持中文名称(如 `chinese`、`english`)
|
||||
- 需要智能解析和错误提示
|
||||
|
||||
**解决方案**:
|
||||
1. 创建 `internal/lang/lang.go` 模块
|
||||
2. 实现语言代码映射表和解析函数
|
||||
3. 支持大小写不敏感和模糊匹配
|
||||
4. 提供语言名称获取和建议功能
|
||||
|
||||
**技术细节**:
|
||||
- 使用 `map[string]string` 存储语言代码映射
|
||||
- 实现 `ParseLanguageCode()` 函数进行智能解析
|
||||
- 支持30+种语言和变体
|
||||
- 添加完整的单元测试
|
||||
|
||||
**关联文档**:
|
||||
- [AGENTS.md#语言代码处理](AGENTS.md#语言代码处理)
|
||||
- [changelog.md#0.2.0](changelog.md#020)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-29 10:30] 版本 0.2.0 - onboard配置向导
|
||||
**原因**: 用户需要友好的配置界面,特别是第一次使用时
|
||||
**分析**:
|
||||
- 需要交互式配置向导
|
||||
- 需要支持选择厂商、输入API密钥、设置默认值
|
||||
- 需要生成标准的YAML配置文件
|
||||
- 需要支持强制重新配置
|
||||
|
||||
**解决方案**:
|
||||
1. 使用 `github.com/AlecAivazis/survey/v2` 库
|
||||
2. 实现分步配置流程:选择厂商 → 配置厂商 → 全局设置 → 保存
|
||||
3. 提供友好的错误处理和用户提示
|
||||
4. 支持 `--force` 参数强制重新配置
|
||||
|
||||
**技术细节**:
|
||||
- 使用 `survey.Select`、`survey.Input`、`survey.Confirm` 组件
|
||||
- 实现厂商默认配置和自定义选项
|
||||
- 生成完整的配置文件包含所有必要字段
|
||||
- 支持配置文件存在性检查
|
||||
|
||||
**关联文档**:
|
||||
- [AGENTS.md#Onboard配置向导](AGENTS.md#onboard配置向导)
|
||||
- [changelog.md#0.2.0](changelog.md#020)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-29 11:00] 版本 0.2.0 - 分阶段迁移策略
|
||||
**原因**: 需要平衡开发便利性和最终上线需求
|
||||
**分析**:
|
||||
- 开发阶段需要简单配置方式(`.env` + `configs/config.yaml`)
|
||||
- 上线前需要迁移到用户配置目录(`~/.config/yoo/yoo.yml`)
|
||||
- 需要平滑的迁移路径和向后兼容性
|
||||
|
||||
**解决方案**:
|
||||
1. **第一阶段(当前)**: 继续使用 `.env` + `configs/config.yaml`
|
||||
2. **第二阶段(上线前)**: 实现配置文件路径查找和迁移工具
|
||||
3. **第三阶段(最终)**: 移除 `.env` 依赖,完全使用配置文件
|
||||
|
||||
**技术细节**:
|
||||
- 配置文件路径优先级:命令行 > 环境变量 > 用户目录 > 当前目录
|
||||
- 保持向后兼容性,支持旧配置格式
|
||||
- 提供配置验证和错误提示
|
||||
- 实现配置迁移工具(计划)
|
||||
|
||||
**关联文档**:
|
||||
- [AGENTS.md#分阶段迁移策略](AGENTS.md#分阶段迁移策略)
|
||||
- [changelog.md#0.2.0](changelog.md#020)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-29 12:00] 版本 0.4.0 - 管道符功能
|
||||
**原因**: 用户需要与其他命令行工具联合使用
|
||||
**分析**:
|
||||
- 用户希望支持管道符功能,如 `cat a.txt | yoyo | grep "who are you"`
|
||||
- 需要检测管道输入并从stdin读取内容
|
||||
- 需要控制统计信息输出,避免污染管道输出
|
||||
- 需要保持向后兼容性
|
||||
|
||||
**解决方案**:
|
||||
1. **管道输入检测**: 实现 `isPipeInput()` 函数,使用 `os.Stdin.Stat()` 检测管道
|
||||
2. **stdin读取**: 实现 `readFromStdin()` 函数,使用 `bufio.Scanner` 读取所有输入
|
||||
3. **静默模式**: 添加 `--quiet` 和 `-q` 参数,控制统计信息输出
|
||||
4. **输出重定向**: 将统计信息输出到stderr,避免污染管道输出
|
||||
|
||||
**技术细节**:
|
||||
- 使用 `os.ModeCharDevice` 检测是否为管道设备
|
||||
- 使用 `strings.Join()` 合并多行输入为单个字符串
|
||||
- 统计信息输出到 `os.Stderr` 而不是 `os.Stdout`
|
||||
- 修复 content/filter.go 中的正则表达式转义问题
|
||||
|
||||
**使用示例**:
|
||||
```bash
|
||||
# 基本管道功能
|
||||
echo "Hello world" | yoyo
|
||||
cat file.txt | yoyo --lang=en
|
||||
|
||||
# 静默模式
|
||||
echo "Hello world" | yoyo -q
|
||||
echo "Hello world" | yoyo --quiet
|
||||
|
||||
# 与其他命令组合
|
||||
cat file.txt | yoyo | grep "你好"
|
||||
yoyo "Hello" | wc -l
|
||||
```
|
||||
|
||||
**关联文档**:
|
||||
- [AGENTS.md#管道符功能](AGENTS.md#管道符功能)
|
||||
- [changelog.md#0.4.0](changelog.md#040)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-29 15:00] 版本 0.5.0 - 本地缓存功能设计
|
||||
**原因**: 用户希望减少API调用,添加本地缓存功能
|
||||
**分析**:
|
||||
- 需要存储翻译结果,避免重复调用API
|
||||
- 需要设计缓存键策略,确保缓存准确性
|
||||
- 需要考虑数据库选择、事务处理、性能优化
|
||||
- 需要设计缓存管理策略
|
||||
|
||||
**解决方案**:
|
||||
1. **数据库选择**: 使用SQLite
|
||||
- 轻量级,无需服务器
|
||||
- 支持ACID事务
|
||||
- Go生态支持良好 (`github.com/mattn/go-sqlite3`)
|
||||
- 适合嵌入式应用
|
||||
|
||||
2. **缓存键设计**:
|
||||
- 使用文本内容 + 源语言 + 目标语言
|
||||
- 生成SHA256哈希作为缓存键
|
||||
- 规范化输入:移除多余空白字符,统一语言代码格式
|
||||
|
||||
3. **事务处理**:
|
||||
- 使用事务保证数据一致性
|
||||
- 插入操作在事务中执行
|
||||
- 查询操作不需要显式事务
|
||||
|
||||
4. **保存时机**:
|
||||
- 在输出结果之前保存到数据库
|
||||
- 确保数据持久化
|
||||
- 异步保存,不阻塞翻译结果返回
|
||||
|
||||
5. **性能优化**:
|
||||
- 为缓存键创建索引
|
||||
- 使用哈希键减少存储空间
|
||||
- 限制缓存表大小(可配置)
|
||||
- 使用WAL模式提高并发性能
|
||||
|
||||
6. **缓存策略**:
|
||||
- 采用组合策略:数量限制+时间过期
|
||||
- 默认启用缓存功能
|
||||
- 提供手动清理命令
|
||||
|
||||
7. **存储位置**:
|
||||
- 数据库文件存储在用户配置目录 `~/.config/yoyo/cache.db`
|
||||
- 符合XDG规范
|
||||
- 支持自定义路径配置
|
||||
|
||||
**技术细节**:
|
||||
- 使用 `github.com/mattn/go-sqlite3` 驱动
|
||||
- 实现 `internal/cache/cache.go` 模块
|
||||
- 缓存表结构:`id`, `cache_key`, `original_text`, `translated_text`, `from_lang`, `to_lang`, `model`, `prompt`, `created_at`
|
||||
- 缓存键生成:`sha256(text + "|" + fromLang + "|" + toLang)`
|
||||
- 查询缓存时使用 `SELECT translated_text FROM cache WHERE cache_key = ?`
|
||||
- 插入缓存时使用 `INSERT OR IGNORE INTO cache (...) VALUES (...)`
|
||||
|
||||
**关联文档**:
|
||||
- [AGENTS.md#本地缓存功能设计](AGENTS.md#本地缓存功能设计)
|
||||
- [changelog.md#0.5.0](changelog.md#050)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-29 20:00] 版本 0.5.1 - 缓存功能修复
|
||||
**原因**: 缓存功能测试中发现的问题
|
||||
**分析**:
|
||||
1. **VACUUM事务错误**: 缓存清空命令中,VACUUM不能在事务中执行
|
||||
2. **NULL值转换错误**: 缓存统计查询在空表时,MIN(created_at)返回NULL导致转换错误
|
||||
3. **过期清理策略**: 当expire_days=0时,清理逻辑不工作
|
||||
|
||||
**解决方案**:
|
||||
1. **修复VACUUM事务错误**:
|
||||
- 将VACUUM移到事务之外执行
|
||||
- 先删除记录,再执行VACUUM
|
||||
|
||||
2. **修复NULL值转换错误**:
|
||||
- 使用 `sql.NullString` 和 `sql.NullFloat64` 类型
|
||||
- 检查 `Valid` 字段判断是否为NULL
|
||||
- 只在有记录时才查询时间范围
|
||||
|
||||
3. **修复过期清理策略**:
|
||||
- 当cleanupTTL为0时,清理所有记录
|
||||
- 添加条件判断:`if c.cleanupTTL == 0 { ... }`
|
||||
|
||||
**技术细节**:
|
||||
```go
|
||||
// 修复VACUUM事务错误
|
||||
func (c *SQLiteCache) Clear(ctx context.Context) error {
|
||||
// 先删除所有记录
|
||||
_, err := c.db.ExecContext(ctx, `DELETE FROM translation_cache`)
|
||||
if err != nil {
|
||||
return fmt.Errorf("清空缓存失败: %w", err)
|
||||
}
|
||||
// 然后执行VACUUM(不能在事务中执行)
|
||||
_, err = c.db.ExecContext(ctx, `VACUUM`)
|
||||
if err != nil {
|
||||
return fmt.Errorf("清理数据库失败: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 修复NULL值转换错误
|
||||
var oldestStr, newestStr sql.NullString
|
||||
var avgTokens sql.NullFloat64
|
||||
err = c.db.QueryRowContext(ctx, `SELECT MIN(created_at), MAX(created_at), AVG(total_tokens) FROM translation_cache`).Scan(&oldestStr, &newestStr, &avgTokens)
|
||||
|
||||
if oldestStr.Valid {
|
||||
stats.OldestRecord, _ = time.Parse("2006-01-02 15:04:05", oldestStr.String)
|
||||
}
|
||||
```
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.5.1](changelog.md#051)
|
||||
- [memory.md#本地缓存实现经验](memory.md#本地缓存实现经验)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 10:00] 版本 0.6.0 - TUI界面模块拆分计划
|
||||
**原因**: 当前TUI目录(tui/components、tui/theme)已创建但完全为空,需要从零实现终端交互界面
|
||||
**分析**:
|
||||
- 采用分模块逐步实现策略,减少Token消耗和上下文负担
|
||||
- 每次只实现一个模块,完成后再进入下一个
|
||||
- 讨论内容仅保存到taolun.md/changelog.md/memory.md,不新增md文件
|
||||
|
||||
**解决方案 - TUI模块拆分**:
|
||||
|
||||
| 步骤 | 模块 | 内容 | 预计工作量 |
|
||||
|------|------|------|-----------|
|
||||
| 1 | TUI框架搭建 | 选择库(bubbletea)、基础App结构、运行循环 | 小 |
|
||||
| 2 | 输入组件 | 文本输入框、光标、基础编辑 | 中 |
|
||||
| 3 | 翻译显示区 | 结果展示、格式化、滚动 | 中 |
|
||||
| 4 | 状态栏/主题 | 底部状态栏、语言选择、主题配色 | 小 |
|
||||
| 5 | 快捷键系统 | 退出、清空、切换语言等 | 小 |
|
||||
| 6 | 集成翻译 | 对接现有Translator、加载动画 | 中 |
|
||||
|
||||
**技术选型**:
|
||||
- 优先使用 `charmbracelet/bubbletea` (Elm架构、Go生态最流行)
|
||||
- 配合 `charmbracelet/lipgloss` 实现样式和主题
|
||||
- 配合 `charmbracelet/bubbles/textinput` 实现输入框
|
||||
|
||||
**当前状态**: ✅ 模块1已完成,模块2待实现
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.6.0](changelog.md#060)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 10:30] 版本 0.6.0 - 模块1: TUI框架搭建 (已完成)
|
||||
**原因**: 实现TUI界面的第一步,建立基础框架结构
|
||||
**分析**:
|
||||
- 需要创建基本的App结构和model
|
||||
- 需要支持config和translator的注入
|
||||
- 需要修复main.go中版本检查顺序问题
|
||||
|
||||
**解决方案**:
|
||||
1. 创建 `internal/tui/app.go` 基础文件
|
||||
2. 定义model结构体,包含config和translator字段
|
||||
3. 实现Init/Update/View三个基本方法
|
||||
4. 添加bubbletea、bubbles、lipgloss依赖
|
||||
5. 修复main.go中--version在interactive之前检查
|
||||
|
||||
**技术实现**:
|
||||
```go
|
||||
type model struct {
|
||||
config *config.Config
|
||||
translator *translator.Translator
|
||||
}
|
||||
|
||||
func NewApp(cfg *config.Config, t *translator.Translator) *tea.Program {
|
||||
return tea.NewProgram(model{...})
|
||||
}
|
||||
```
|
||||
|
||||
**下一步**: 实现模块2: 输入组件
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.6.0](changelog.md#060)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 11:00] 版本 0.6.0 - 模块2: 输入组件 (已完成)
|
||||
**原因**: 实现TUI输入功能
|
||||
**分析**:
|
||||
- 使用bubbletea的textinput组件
|
||||
- 需要焦点管理和键盘事件处理
|
||||
|
||||
**解决方案**:
|
||||
1. 添加textinput字段到model
|
||||
2. 初始化时设置placeholder和prompt
|
||||
3. Update中处理KeyMsg
|
||||
4. View中渲染输入框
|
||||
5. 支持Ctrl+C和Esc退出
|
||||
|
||||
**技术实现**:
|
||||
```go
|
||||
type model struct {
|
||||
textInput textinput.Model
|
||||
}
|
||||
|
||||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
case tea.KeyMsg:
|
||||
if msg.Type == tea.KeyCtrlC || msg.Type == tea.KeyEsc {
|
||||
return m, tea.Quit
|
||||
}
|
||||
}
|
||||
m.textInput, cmd = m.textInput.Update(msg)
|
||||
return m, cmd
|
||||
}
|
||||
```
|
||||
|
||||
**下一步**: 实现模块3: 翻译显示区
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.6.0](changelog.md#060)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 11:30] 版本 0.6.0 - 模块3: 翻译显示区 (已完成)
|
||||
**原因**: 添加翻译结果显示区域
|
||||
**分析**:
|
||||
- 需要定义结果展示区域
|
||||
- 需要为不同区域定义不同样式
|
||||
|
||||
**解决方案**:
|
||||
1. 添加result字段到model
|
||||
2. 定义headerStyle、resultStyle、helpStyle
|
||||
3. 实现renderResult()辅助方法
|
||||
4. View中组合各个区域
|
||||
|
||||
**下一步**: 实现模块4: 状态栏/主题
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.6.0](changelog.md#060)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 12:00] 版本 0.6.0 - 模块4: 状态栏/主题 (已完成)
|
||||
**原因**: 添加底部状态栏和主题配色
|
||||
**分析**:
|
||||
- 需要显示当前目标语言
|
||||
- 需要完善配色方案
|
||||
- 需要定义状态栏样式
|
||||
|
||||
**解决方案**:
|
||||
1. 添加targetLang字段到model
|
||||
2. 定义statusBarStyle、langStyle等新样式
|
||||
3. 实现renderStatusBar()方法
|
||||
4. View中渲染状态栏
|
||||
|
||||
**下一步**: 实现模块5: 快捷键系统
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.6.0](changelog.md#060)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 12:30] 版本 0.6.0 - 模块5: 快捷键系统 (已完成)
|
||||
**原因**: 添加键盘快捷键提升用户体验
|
||||
**分析**:
|
||||
- 需要常用操作快捷键
|
||||
- 需要清晰显示快捷键提示
|
||||
|
||||
**解决方案**:
|
||||
1. 添加Ctrl+L: 清空输入和结果
|
||||
2. 添加Ctrl+T: 循环切换语言
|
||||
3. 添加keyStyle样式高亮快捷键
|
||||
4. 更新帮助提示显示所有快捷键
|
||||
|
||||
**快捷键列表**:
|
||||
- `Ctrl+L`: 清空输入框和翻译结果
|
||||
- `Ctrl+T`: 循环切换目标语言 (zh-CN→en-US→ja→ko→...)
|
||||
- `Ctrl+C`/`Esc`: 退出程序
|
||||
- `Enter`: 翻译 (后续模块实现)
|
||||
|
||||
**下一步**: 实现模块6: 集成翻译
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.6.0](changelog.md#060)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 13:00] 版本 0.6.0 - 模块6: 集成翻译 (已完成)
|
||||
**原因**: 将Translator集成到TUI,实现真正的翻译功能
|
||||
**分析**:
|
||||
- 需要在Enter键时调用翻译API
|
||||
- 需要异步执行避免阻塞UI
|
||||
- 需要显示loading状态和错误处理
|
||||
|
||||
**解决方案**:
|
||||
1. 添加translateMsg消息类型处理异步结果
|
||||
2. 添加loading和errMsg字段
|
||||
3. 实现doTranslate()函数执行异步翻译
|
||||
4. Update中处理translateMsg消息
|
||||
5. View中显示loading状态或错误信息
|
||||
|
||||
**技术实现**:
|
||||
```go
|
||||
type translateMsg struct {
|
||||
result string
|
||||
err error
|
||||
}
|
||||
|
||||
func (m model) doTranslate(text, toLang string) tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
result, err := m.translator.Translate(...)
|
||||
if err != nil {
|
||||
return translateMsg{err: err}
|
||||
}
|
||||
return translateMsg{result: result.Translated}
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
**下一步**: 测试TUI界面、优化体验
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.6.0](changelog.md#060)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 14:00] 版本 0.7.0 - TUI界面改进计划
|
||||
**原因**: TUI基础功能完成后,讨论改进方向和用户体验优化
|
||||
**分析**:
|
||||
- 当前单行textinput无法输入多行文本
|
||||
- 快捷键固定显示在底部不够美观
|
||||
- 缺少命令菜单系统
|
||||
- 长翻译结果无法滚动
|
||||
|
||||
**解决方案 - 新增模块**:
|
||||
|
||||
| 步骤 | 模块 | 内容 |
|
||||
|------|------|------|
|
||||
| 7 | 多行输入 | textarea组件替换textinput |
|
||||
| 8 | 弹出框组件 | 通用modal,支持快捷键帮助 |
|
||||
| 9 | 斜杠命令菜单 | / 触发命令选择器,类似opencode |
|
||||
| 10 | 翻译结果滚动 | viewport组件 |
|
||||
| 11 | 复制功能 | clipboard集成 |
|
||||
| 12 | 状态栏扩展 | 耗时、token用量 |
|
||||
|
||||
**斜杠命令设计**:
|
||||
| 命令 | 功能 |
|
||||
|------|------|
|
||||
| `/help` | 显示快捷键帮助 |
|
||||
| `/clear` | 清空内容 |
|
||||
| `/copy` | 复制翻译结果 |
|
||||
| `/lang` | 切换语言 |
|
||||
| `/history` | 翻译历史 |
|
||||
| `/quit` | 退出 |
|
||||
|
||||
**设计亮点**:
|
||||
1. 隐藏底部快捷键提示,改为按 ? 或 F1 弹出帮助框
|
||||
2. 输入 / 触发命令菜单,上下键选择,回车执行
|
||||
3. 命令菜单支持模糊匹配
|
||||
4. modal组件通用化,可复用于其他弹窗场景
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.7.0](changelog.md#070)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 14:30] 版本 0.7.0 - 模块7: 多行输入 (已完成)
|
||||
**原因**: 当前textinput只支持单行,需要支持多行文本输入
|
||||
**分析**:
|
||||
- 长段落、多行文本无法输入
|
||||
- 需要换用bubbles的textarea组件
|
||||
|
||||
**解决方案**:
|
||||
1. 将textinput替换为textarea
|
||||
2. 调整样式和布局(宽50,高5)
|
||||
3. 底部状态栏提示按 / 显示命令
|
||||
|
||||
**技术实现**:
|
||||
- 使用 `github.com/charmbracelet/bubbles/textarea`
|
||||
- textarea.SetWidth(50)、SetHeight(5) 设置尺寸
|
||||
- 移除底部固定快捷键提示,改为按需显示
|
||||
- 隐藏行号: `ShowLineNumbers = false`
|
||||
- 移除左侧提示符: `Prompt = ""`
|
||||
- Enter执行翻译,Ctrl+J换行
|
||||
|
||||
**下一步**: 实现模块8: 弹出框组件
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.7.0](changelog.md#070)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 15:00] 版本 0.8.0 - TUI重构: 聊天风格界面
|
||||
**原因**: 用户希望使用类似 charmbracelet/crush 的聊天风格界面
|
||||
**分析**:
|
||||
- 当前界面:标题在上 → 输入框在中 → 结果在下
|
||||
- 期望界面:标题在上 → 聊天消息区域(可滚动历史)→ 输入框固定底部 → 状态栏最底部
|
||||
- 原文+译文成对显示,类似聊天软件
|
||||
|
||||
**解决方案**:
|
||||
1. 界面布局重构:
|
||||
```
|
||||
┌─────────────────────────────────────────┐
|
||||
│ ✦ YOYO 翻译 [Ctrl+C退出] │
|
||||
├─────────────────────────────────────────┤
|
||||
│ (聊天消息区域,可滚动查看历史) │
|
||||
│ ── 用户输入 ── │
|
||||
│ Hello world │
|
||||
│ ── 翻译结果 ── │
|
||||
│ 你好世界 │
|
||||
│ ... │
|
||||
├─────────────────────────────────────────┤
|
||||
│ 输入框... [回车] │
|
||||
├─────────────────────────────────────────┤
|
||||
│ 目标:zh-CN │ 模型:gpt-3.5 │ Tokens:125 │
|
||||
└─────────────────────────────────────────┘
|
||||
```
|
||||
|
||||
2. 技术方案:
|
||||
- 消息结构:原文+译文成对显示
|
||||
- 底部固定输入框(textarea)
|
||||
- 状态栏显示完整信息
|
||||
- Ctrl+J 换行,Enter 发送
|
||||
- 自动调整输入框高度
|
||||
|
||||
**下一步**: 模块1: 创建TUI模块结构
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.8.0](changelog.md#080)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-06 16:00] 版本 0.8.0 - 输入框踩坑与修复
|
||||
**原因**: Ctrl+J换行后第一行被遮住,显示错乱
|
||||
**分析**:
|
||||
- 换行后 textarea 内部渲染3行但只显示2行
|
||||
- 第一行内容往上滚动被遮住,光标在第2行,第3行是空行
|
||||
- 尝试多种方案均无效:移除Width限制、设置SetWidth/SetHeight顺序、动态调整高度
|
||||
|
||||
**解决方案**:
|
||||
1. 放弃动态调整高度方案,固定高度为5行
|
||||
2. 超过5行时textarea内部自动滚动,光标始终可见
|
||||
|
||||
**技术细节**:
|
||||
- textarea内部使用viewport组件,频繁SetHeight导致滚动位置错乱
|
||||
- 使用 FocusedStyle/BlurredStyle + Style.Base 设置背景色
|
||||
|
||||
**代码变更**:
|
||||
```go
|
||||
ta.SetWidth(60)
|
||||
ta.SetHeight(5) // 固定高度,不动态调整
|
||||
```
|
||||
|
||||
**下一步**: 完善其他UI功能
|
||||
|
||||
**关联文档**:
|
||||
- [memory.md#TUI输入框踩坑记录](memory.md#tui输入框踩坑记录)
|
||||
- [changelog.md#0.8.0](changelog.md#080)
|
||||
|
||||
---
|
||||
|
||||
### [2026-04-07 10:00] 版本 0.8.1 - 翻译结果卡片组件设计
|
||||
**原因**: 用户希望改进翻译结果显示样式,使用lipgloss构建更美观的组件
|
||||
**分析**:
|
||||
- 当前翻译结果显示比较简单,只有标签和内容
|
||||
- 需要设计一个结构化的翻译卡片组件
|
||||
- 组件需要显示Tokens、翻译时间、模型名称等元信息
|
||||
|
||||
**解决方案**:
|
||||
1. **卡片组件结构**:
|
||||
```
|
||||
┌─ 标题栏 ─────────────────────────────────────────┐
|
||||
│ Tokens: 150 │ 耗时: 1.2s │ 模型: gpt-4 │
|
||||
└─────────────────────────────────────────────────┘
|
||||
┌─ 用户输入 ───────────────────────────────────────┐
|
||||
│ ██████████ 碳黑背景 ████████████████████████████ │
|
||||
└─────────────────────────────────────────────────┘
|
||||
┌─ 翻译结果 ───────────────────────────────────────┐
|
||||
│ AI 翻译的文本内容... │
|
||||
└─────────────────────────────────────────────────┘
|
||||
```
|
||||
|
||||
2. **技术实现**:
|
||||
- 使用 lipgloss.Div() 和 lipgloss.JoinVertical() 构建组件
|
||||
- 背景色: #1A1A1A (碳黑色)
|
||||
- 用户输入区域: 纯背景色,无边框
|
||||
- 组件间距: 5px marginBottom
|
||||
|
||||
3. **样式定义**:
|
||||
- CardStyle: 卡片容器,marginBottom(5)
|
||||
- CardMetaStyle: 元信息行样式,#6B7280 灰色
|
||||
- CardInputStyle: 用户输入,#1A1A1A 背景 + #E5E7EB 文字
|
||||
- CardOutputStyle: 翻译结果,白色文字
|
||||
|
||||
**下一步**: 实现组件代码
|
||||
|
||||
**关联文档**:
|
||||
- [changelog.md#0.8.1](changelog.md#081)
|
||||
Reference in New Issue
Block a user