Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 24ba405d55 | |||
| cd305a62ef | |||
| ad667fa782 |
59
AGENTS.md
59
AGENTS.md
@@ -761,7 +761,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)
|
||||
90
changelog.md
90
changelog.md
@@ -32,6 +32,96 @@
|
||||
|
||||
## 版本历史
|
||||
|
||||
### 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) - 环境变量加载修复
|
||||
**类型**: 修复版本
|
||||
**状态**: 开发中
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 修复环境变量加载问题
|
||||
- ✅ 添加godotenv依赖
|
||||
- ✅ 更新memory.md记录踩坑经验
|
||||
- ✅ 测试CLI基本功能
|
||||
|
||||
**讨论记录**:
|
||||
- [环境变量加载修复](taolun.md#2026-03-29-0000-版本-003-环境变量加载修复)
|
||||
|
||||
**下一步**:
|
||||
- 实现更多厂商
|
||||
- 添加更多测试
|
||||
- 完善错误处理
|
||||
|
||||
### 0.0.2 (2026-03-28) - 核心架构实现
|
||||
**类型**: 功能版本
|
||||
**状态**: 开发中
|
||||
|
||||
**变更内容**:
|
||||
- ✅ 实现Config配置类（internal/config/config.go）
|
||||
- ✅ 实现Provider接口和工厂模式（internal/provider/）
|
||||
- ✅ 实现硅基流动厂商（internal/provider/siliconflow.go）
|
||||
- ✅ 实现Translator核心翻译类（internal/translator/）
|
||||
- ✅ 实现Prompt管理器（internal/translator/prompt.go）
|
||||
- ✅ 创建CLI入口点（cmd/yoyo/main.go）
|
||||
- ✅ 添加配置文件模板（configs/config.yaml）
|
||||
- ✅ 添加单元测试（internal/config/config_test.go）
|
||||
- ✅ 初始化Git仓库和版本标签
|
||||
|
||||
**讨论记录**:
|
||||
- [实现核心架构](taolun.md#2026-03-28-2350-版本-002-实现核心架构)
|
||||
|
||||
**下一步**:
|
||||
- 实现其他厂商（火山引擎、国家超算、Qwen、OpenAI兼容）
|
||||
- 添加更多测试
|
||||
- 实现批量翻译功能
|
||||
- 添加翻译历史记录
|
||||
- 实现配置文件热重载
|
||||
|
||||
### 0.0.1 (2026-03-28) - 项目初始化
|
||||
**类型**: 初始化版本
|
||||
**状态**: 开发中
|
||||
|
||||
@@ -4,6 +4,8 @@
|
||||
default_provider: "siliconflow"
|
||||
default_model: "gpt-3.5-turbo"
|
||||
timeout: 30
|
||||
default_source_lang: "auto" # 默认源语言（auto为自动检测）
|
||||
default_target_lang: "zh-CN" # 默认目标语言（简体中文）
|
||||
|
||||
providers:
|
||||
siliconflow:
|
||||
|
||||
13
go.mod
13
go.mod
@@ -1,3 +1,16 @@
|
||||
module github.com/titor/fanyi
|
||||
|
||||
go 1.26.1
|
||||
|
||||
require (
|
||||
github.com/AlecAivazis/survey/v2 v2.3.7 // indirect
|
||||
github.com/joho/godotenv v1.5.1 // indirect
|
||||
github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51 // indirect
|
||||
github.com/mattn/go-colorable v0.1.2 // indirect
|
||||
github.com/mattn/go-isatty v0.0.8 // indirect
|
||||
github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b // indirect
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f // indirect
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211 // indirect
|
||||
golang.org/x/text v0.4.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
52
go.sum
Normal file
52
go.sum
Normal file
@@ -0,0 +1,52 @@
|
||||
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/Netflix/go-expect v0.0.0-20220104043353-73e0943537d2/go.mod h1:HBCaDeC1lPdgDeDbhX8XFpy1jqjK0IBG8W5K+xYqA0w=
|
||||
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/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
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/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 h1:HLtExJ+uU2HOZ+wI0Tt5DtUDrx8yhUqDcp7fYERX4CE=
|
||||
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
|
||||
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/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
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/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/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-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f h1:v4INt8xihDGvnrfjMDVXGxw9wrfxYyCjk0KbXjhR55s=
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
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/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
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=
|
||||
212
internal/config/config.go
Normal file
212
internal/config/config.go
Normal file
@@ -0,0 +1,212 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
// Config 全局配置结构
|
||||
type Config struct {
|
||||
// 全局设置
|
||||
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"`
|
||||
}
|
||||
|
||||
// ProviderConfig 厂商配置
|
||||
type ProviderConfig struct {
|
||||
APIHost string `yaml:"api_host"`
|
||||
APIKey string `yaml:"api_key"`
|
||||
Model string `yaml:"model"`
|
||||
Enabled bool `yaml:"enabled"`
|
||||
}
|
||||
|
||||
// ConfigLoader 配置加载器接口
|
||||
type ConfigLoader interface {
|
||||
Load(path string) (*Config, error)
|
||||
Save(config *Config, path string) error
|
||||
}
|
||||
|
||||
// YAMLConfigLoader YAML配置加载器实现
|
||||
type YAMLConfigLoader struct{}
|
||||
|
||||
// Load 加载YAML配置文件
|
||||
func (l *YAMLConfigLoader) Load(path string) (*Config, error) {
|
||||
// 读取文件
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("读取配置文件失败: %w", err)
|
||||
}
|
||||
|
||||
// 替换环境变量
|
||||
content := string(data)
|
||||
content = os.ExpandEnv(content)
|
||||
|
||||
// 解析YAML
|
||||
config := &Config{}
|
||||
if err := yaml.Unmarshal([]byte(content), config); err != nil {
|
||||
return nil, fmt.Errorf("解析配置文件失败: %w", err)
|
||||
}
|
||||
|
||||
// 设置默认值
|
||||
config.setDefaults()
|
||||
|
||||
return config, nil
|
||||
}
|
||||
|
||||
// Save 保存配置到文件
|
||||
func (l *YAMLConfigLoader) Save(config *Config, path string) error {
|
||||
// 确保目录存在
|
||||
dir := filepath.Dir(path)
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
return fmt.Errorf("创建配置目录失败: %w", err)
|
||||
}
|
||||
|
||||
// 序列化为YAML
|
||||
data, err := yaml.Marshal(config)
|
||||
if err != nil {
|
||||
return fmt.Errorf("序列化配置失败: %w", err)
|
||||
}
|
||||
|
||||
// 写入文件
|
||||
if err := os.WriteFile(path, data, 0644); err != nil {
|
||||
return fmt.Errorf("写入配置文件失败: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// setDefaults 设置默认值
|
||||
func (c *Config) setDefaults() {
|
||||
if c.DefaultProvider == "" {
|
||||
c.DefaultProvider = "siliconflow"
|
||||
}
|
||||
if c.Timeout <= 0 {
|
||||
c.Timeout = 30
|
||||
}
|
||||
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 {
|
||||
if provider.Model == "" {
|
||||
provider.Model = c.DefaultModel
|
||||
c.Providers[name] = provider
|
||||
}
|
||||
// 替换环境变量
|
||||
provider.APIKey = os.ExpandEnv(provider.APIKey)
|
||||
c.Providers[name] = provider
|
||||
}
|
||||
|
||||
// 确保Prompts映射存在
|
||||
if c.Prompts == nil {
|
||||
c.Prompts = make(map[string]string)
|
||||
}
|
||||
}
|
||||
|
||||
// GetProviderConfig 获取指定厂商的配置
|
||||
func (c *Config) GetProviderConfig(name string) (ProviderConfig, error) {
|
||||
config, exists := c.Providers[name]
|
||||
if !exists {
|
||||
return ProviderConfig{}, fmt.Errorf("未找到厂商配置: %s", name)
|
||||
}
|
||||
if !config.Enabled {
|
||||
return ProviderConfig{}, fmt.Errorf("厂商未启用: %s", name)
|
||||
}
|
||||
return config, nil
|
||||
}
|
||||
|
||||
// GetPrompt 获取指定名称的Prompt
|
||||
func (c *Config) GetPrompt(name string) (string, bool) {
|
||||
prompt, exists := c.Prompts[name]
|
||||
return prompt, exists
|
||||
}
|
||||
|
||||
// ExpandEnv 扩展环境变量（辅助函数）
|
||||
func ExpandEnv(s string) string {
|
||||
return os.ExpandEnv(s)
|
||||
}
|
||||
|
||||
// Validate 验证配置是否有效
|
||||
func (c *Config) Validate() error {
|
||||
if c.DefaultProvider == "" {
|
||||
return fmt.Errorf("默认厂商不能为空")
|
||||
}
|
||||
|
||||
// 检查默认厂商是否在配置中
|
||||
if _, exists := c.Providers[c.DefaultProvider]; !exists {
|
||||
return fmt.Errorf("默认厂商 '%s' 未在配置中定义", c.DefaultProvider)
|
||||
}
|
||||
|
||||
// 检查每个厂商配置
|
||||
for name, provider := range c.Providers {
|
||||
if provider.APIKey == "" {
|
||||
return fmt.Errorf("厂商 '%s' 的API密钥不能为空", name)
|
||||
}
|
||||
if provider.APIHost == "" {
|
||||
return fmt.Errorf("厂商 '%s' 的API主机不能为空", name)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetDefaultProvider 获取默认厂商配置
|
||||
func (c *Config) GetDefaultProvider() (ProviderConfig, error) {
|
||||
return c.GetProviderConfig(c.DefaultProvider)
|
||||
}
|
||||
|
||||
// IsProviderEnabled 检查厂商是否启用
|
||||
func (c *Config) IsProviderEnabled(name string) bool {
|
||||
config, exists := c.Providers[name]
|
||||
return exists && config.Enabled
|
||||
}
|
||||
|
||||
// GetEnabledProviders 获取所有启用的厂商
|
||||
func (c *Config) GetEnabledProviders() []string {
|
||||
var enabled []string
|
||||
for name, config := range c.Providers {
|
||||
if config.Enabled {
|
||||
enabled = append(enabled, name)
|
||||
}
|
||||
}
|
||||
return enabled
|
||||
}
|
||||
|
||||
// String 返回配置的字符串表示（隐藏敏感信息）
|
||||
func (c *Config) String() string {
|
||||
var builder strings.Builder
|
||||
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))
|
||||
}
|
||||
builder.WriteString("Prompts:\n")
|
||||
for name := range c.Prompts {
|
||||
builder.WriteString(fmt.Sprintf(" %s\n", name))
|
||||
}
|
||||
return builder.String()
|
||||
}
|
||||
177
internal/config/config_test.go
Normal file
177
internal/config/config_test.go
Normal file
@@ -0,0 +1,177 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestConfigLoader_Load(t *testing.T) {
|
||||
// 创建临时配置文件
|
||||
tmpDir := t.TempDir()
|
||||
configFile := filepath.Join(tmpDir, "config.yaml")
|
||||
|
||||
configContent := `
|
||||
default_provider: "siliconflow"
|
||||
default_model: "gpt-3.5-turbo"
|
||||
timeout: 30
|
||||
|
||||
providers:
|
||||
siliconflow:
|
||||
api_host: "https://api.siliconflow.cn/v1"
|
||||
api_key: "${TEST_API_KEY}"
|
||||
model: "siliconflow-base"
|
||||
enabled: true
|
||||
|
||||
prompts:
|
||||
simple: "请用简单易懂的语言翻译以下内容。"
|
||||
`
|
||||
|
||||
if err := os.WriteFile(configFile, []byte(configContent), 0644); err != nil {
|
||||
t.Fatalf("创建配置文件失败: %v", err)
|
||||
}
|
||||
|
||||
// 设置环境变量
|
||||
os.Setenv("TEST_API_KEY", "test-key-123")
|
||||
defer os.Unsetenv("TEST_API_KEY")
|
||||
|
||||
// 加载配置
|
||||
loader := &YAMLConfigLoader{}
|
||||
config, err := loader.Load(configFile)
|
||||
if err != nil {
|
||||
t.Fatalf("加载配置失败: %v", err)
|
||||
}
|
||||
|
||||
// 验证配置
|
||||
if config.DefaultProvider != "siliconflow" {
|
||||
t.Errorf("Expected default provider 'siliconflow', got '%s'", config.DefaultProvider)
|
||||
}
|
||||
|
||||
if config.DefaultModel != "gpt-3.5-turbo" {
|
||||
t.Errorf("Expected default model 'gpt-3.5-turbo', got '%s'", config.DefaultModel)
|
||||
}
|
||||
|
||||
if config.Timeout != 30 {
|
||||
t.Errorf("Expected timeout 30, got %d", config.Timeout)
|
||||
}
|
||||
|
||||
// 验证厂商配置
|
||||
providerConfig, err := config.GetProviderConfig("siliconflow")
|
||||
if err != nil {
|
||||
t.Fatalf("获取厂商配置失败: %v", err)
|
||||
}
|
||||
|
||||
if providerConfig.APIKey != "test-key-123" {
|
||||
t.Errorf("Expected API key 'test-key-123', got '%s'", providerConfig.APIKey)
|
||||
}
|
||||
|
||||
if providerConfig.APIHost != "https://api.siliconflow.cn/v1" {
|
||||
t.Errorf("Expected API host 'https://api.siliconflow.cn/v1', got '%s'", providerConfig.APIHost)
|
||||
}
|
||||
|
||||
// 验证Prompt
|
||||
prompt, exists := config.GetPrompt("simple")
|
||||
if !exists {
|
||||
t.Error("Prompt 'simple' should exist")
|
||||
}
|
||||
if prompt != "请用简单易懂的语言翻译以下内容。" {
|
||||
t.Errorf("Unexpected prompt content: %s", prompt)
|
||||
}
|
||||
}
|
||||
|
||||
func TestConfig_Validate(t *testing.T) {
|
||||
config := &Config{
|
||||
DefaultProvider: "siliconflow",
|
||||
DefaultModel: "gpt-3.5-turbo",
|
||||
Timeout: 30,
|
||||
Providers: map[string]ProviderConfig{
|
||||
"siliconflow": {
|
||||
APIHost: "https://api.siliconflow.cn/v1",
|
||||
APIKey: "test-key",
|
||||
Model: "siliconflow-base",
|
||||
Enabled: true,
|
||||
},
|
||||
},
|
||||
Prompts: map[string]string{
|
||||
"simple": "test prompt",
|
||||
},
|
||||
}
|
||||
|
||||
if err := config.Validate(); err != nil {
|
||||
t.Errorf("Config validation failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestConfig_GetProviderConfig(t *testing.T) {
|
||||
config := &Config{
|
||||
Providers: map[string]ProviderConfig{
|
||||
"siliconflow": {
|
||||
APIHost: "https://api.siliconflow.cn/v1",
|
||||
APIKey: "test-key",
|
||||
Model: "siliconflow-base",
|
||||
Enabled: true,
|
||||
},
|
||||
"volcano": {
|
||||
APIHost: "https://api.volcengine.com/v1",
|
||||
APIKey: "test-key",
|
||||
Model: "volcano-chat",
|
||||
Enabled: false,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
// 测试获取启用的厂商
|
||||
_, err := config.GetProviderConfig("siliconflow")
|
||||
if err != nil {
|
||||
t.Errorf("Should get enabled provider: %v", err)
|
||||
}
|
||||
|
||||
// 测试获取禁用的厂商
|
||||
_, err = config.GetProviderConfig("volcano")
|
||||
if err == nil {
|
||||
t.Error("Should return error for disabled provider")
|
||||
}
|
||||
|
||||
// 测试获取不存在的厂商
|
||||
_, err = config.GetProviderConfig("nonexistent")
|
||||
if err == nil {
|
||||
t.Error("Should return error for non-existent provider")
|
||||
}
|
||||
}
|
||||
|
||||
func TestConfig_GetEnabledProviders(t *testing.T) {
|
||||
config := &Config{
|
||||
Providers: map[string]ProviderConfig{
|
||||
"siliconflow": {Enabled: true},
|
||||
"volcano": {Enabled: false},
|
||||
"qwen": {Enabled: true},
|
||||
},
|
||||
}
|
||||
|
||||
enabled := config.GetEnabledProviders()
|
||||
if len(enabled) != 2 {
|
||||
t.Errorf("Expected 2 enabled providers, got %d", len(enabled))
|
||||
}
|
||||
}
|
||||
|
||||
func TestConfig_String(t *testing.T) {
|
||||
config := &Config{
|
||||
DefaultProvider: "siliconflow",
|
||||
DefaultModel: "gpt-3.5-turbo",
|
||||
Timeout: 30,
|
||||
Providers: map[string]ProviderConfig{
|
||||
"siliconflow": {
|
||||
Enabled: true,
|
||||
Model: "siliconflow-base",
|
||||
},
|
||||
},
|
||||
Prompts: map[string]string{
|
||||
"simple": "test prompt",
|
||||
},
|
||||
}
|
||||
|
||||
str := config.String()
|
||||
if str == "" {
|
||||
t.Error("String representation should not be empty")
|
||||
}
|
||||
}
|
||||
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
|
||||
}
|
||||
99
internal/provider/factory.go
Normal file
99
internal/provider/factory.go
Normal file
@@ -0,0 +1,99 @@
|
||||
package provider
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// ProviderFactory 厂商工厂
|
||||
type ProviderFactory struct {
|
||||
providers map[string]func(ProviderConfig) (Provider, error)
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
// NewProviderFactory 创建工厂实例
|
||||
func NewProviderFactory() *ProviderFactory {
|
||||
factory := &ProviderFactory{
|
||||
providers: make(map[string]func(ProviderConfig) (Provider, error)),
|
||||
}
|
||||
|
||||
// 注册所有厂商（延迟注册，避免循环依赖）
|
||||
// 实际注册在init()函数中完成
|
||||
|
||||
return factory
|
||||
}
|
||||
|
||||
// Register 注册厂商构造函数
|
||||
func (f *ProviderFactory) Register(name string, creator func(ProviderConfig) (Provider, error)) {
|
||||
f.mu.Lock()
|
||||
defer f.mu.Unlock()
|
||||
f.providers[name] = creator
|
||||
}
|
||||
|
||||
// Create 创建厂商实例
|
||||
func (f *ProviderFactory) Create(name string, config ProviderConfig) (Provider, error) {
|
||||
f.mu.RLock()
|
||||
creator, exists := f.providers[name]
|
||||
f.mu.RUnlock()
|
||||
|
||||
if !exists {
|
||||
return nil, fmt.Errorf("不支持的厂商: %s", name)
|
||||
}
|
||||
|
||||
provider, err := creator(config)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建厂商实例失败: %w", err)
|
||||
}
|
||||
|
||||
if err := provider.Validate(); err != nil {
|
||||
return nil, fmt.Errorf("厂商配置验证失败: %w", err)
|
||||
}
|
||||
|
||||
return provider, nil
|
||||
}
|
||||
|
||||
// HasProvider 检查是否支持指定厂商
|
||||
func (f *ProviderFactory) HasProvider(name string) bool {
|
||||
f.mu.RLock()
|
||||
defer f.mu.RUnlock()
|
||||
_, exists := f.providers[name]
|
||||
return exists
|
||||
}
|
||||
|
||||
// GetSupportedProviders 获取所有支持的厂商
|
||||
func (f *ProviderFactory) GetSupportedProviders() []string {
|
||||
f.mu.RLock()
|
||||
defer f.mu.RUnlock()
|
||||
|
||||
providers := make([]string, 0, len(f.providers))
|
||||
for name := range f.providers {
|
||||
providers = append(providers, name)
|
||||
}
|
||||
return providers
|
||||
}
|
||||
|
||||
// 全局工厂实例
|
||||
var defaultFactory *ProviderFactory
|
||||
|
||||
// init 初始化默认工厂
|
||||
func init() {
|
||||
defaultFactory = NewProviderFactory()
|
||||
|
||||
// 这里可以注册厂商，但由于循环依赖，实际注册会在各厂商的init()中完成
|
||||
// 或者在main包中手动注册
|
||||
}
|
||||
|
||||
// GetDefaultFactory 获取默认工厂
|
||||
func GetDefaultFactory() *ProviderFactory {
|
||||
return defaultFactory
|
||||
}
|
||||
|
||||
// CreateProvider 使用默认工厂创建厂商
|
||||
func CreateProvider(name string, config ProviderConfig) (Provider, error) {
|
||||
return defaultFactory.Create(name, config)
|
||||
}
|
||||
|
||||
// RegisterProvider 注册厂商到默认工厂
|
||||
func RegisterProvider(name string, creator func(ProviderConfig) (Provider, error)) {
|
||||
defaultFactory.Register(name, creator)
|
||||
}
|
||||
108
internal/provider/provider.go
Normal file
108
internal/provider/provider.go
Normal file
@@ -0,0 +1,108 @@
|
||||
package provider
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// Provider 厂商接口
|
||||
type Provider interface {
|
||||
// Translate 调用厂商API进行翻译
|
||||
Translate(ctx context.Context, req *TranslateRequest) (*TranslateResponse, error)
|
||||
|
||||
// Name 返回厂商名称
|
||||
Name() string
|
||||
|
||||
// Validate 验证配置是否有效
|
||||
Validate() error
|
||||
}
|
||||
|
||||
// TranslateRequest 翻译请求
|
||||
type TranslateRequest struct {
|
||||
Text string `json:"text"`
|
||||
FromLang string `json:"from_lang"`
|
||||
ToLang string `json:"to_lang"`
|
||||
Prompt string `json:"prompt"`
|
||||
Model string `json:"model"`
|
||||
Options map[string]interface{} `json:"options"`
|
||||
}
|
||||
|
||||
// TranslateResponse 翻译响应
|
||||
type TranslateResponse struct {
|
||||
Text string `json:"text"`
|
||||
FromLang string `json:"from_lang"`
|
||||
ToLang string `json:"to_lang"`
|
||||
Model string `json:"model"`
|
||||
Usage *Usage `json:"usage"`
|
||||
RawResponse []byte `json:"raw_response,omitempty"`
|
||||
}
|
||||
|
||||
// Usage 用量统计
|
||||
type Usage struct {
|
||||
PromptTokens int `json:"prompt_tokens"`
|
||||
CompletionTokens int `json:"completion_tokens"`
|
||||
TotalTokens int `json:"total_tokens"`
|
||||
}
|
||||
|
||||
// ProviderConfig 厂商配置
|
||||
type ProviderConfig struct {
|
||||
APIHost string `json:"api_host"`
|
||||
APIKey string `json:"api_key"`
|
||||
Model string `json:"model"`
|
||||
}
|
||||
|
||||
// TranslateError 翻译错误
|
||||
type TranslateError struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"message"`
|
||||
Details string `json:"details,omitempty"`
|
||||
}
|
||||
|
||||
// Error 实现error接口
|
||||
func (e *TranslateError) Error() string {
|
||||
if e.Details != "" {
|
||||
return fmt.Sprintf("%s: %s (%s)", e.Code, e.Message, e.Details)
|
||||
}
|
||||
return fmt.Sprintf("%s: %s", e.Code, e.Message)
|
||||
}
|
||||
|
||||
// NewTranslateError 创建翻译错误
|
||||
func NewTranslateError(code, message string) *TranslateError {
|
||||
return &TranslateError{
|
||||
Code: code,
|
||||
Message: message,
|
||||
}
|
||||
}
|
||||
|
||||
// NewTranslateErrorWithDetails 创建带详情的翻译错误
|
||||
func NewTranslateErrorWithDetails(code, message, details string) *TranslateError {
|
||||
return &TranslateError{
|
||||
Code: code,
|
||||
Message: message,
|
||||
Details: details,
|
||||
}
|
||||
}
|
||||
|
||||
// IsNetworkError 检查是否为网络错误
|
||||
func IsNetworkError(err error) bool {
|
||||
if translateErr, ok := err.(*TranslateError); ok {
|
||||
return translateErr.Code == "NETWORK_ERROR"
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// IsRateLimitError 检查是否为限流错误
|
||||
func IsRateLimitError(err error) bool {
|
||||
if translateErr, ok := err.(*TranslateError); ok {
|
||||
return translateErr.Code == "RATE_LIMIT"
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// IsAuthError 检查是否为认证错误
|
||||
func IsAuthError(err error) bool {
|
||||
if translateErr, ok := err.(*TranslateError); ok {
|
||||
return translateErr.Code == "AUTH_ERROR" || translateErr.Code == "INVALID_API_KEY"
|
||||
}
|
||||
return false
|
||||
}
|
||||
185
internal/provider/siliconflow.go
Normal file
185
internal/provider/siliconflow.go
Normal file
@@ -0,0 +1,185 @@
|
||||
package provider
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
// SiliconFlowProvider 硅基流动厂商实现
|
||||
type SiliconFlowProvider struct {
|
||||
config ProviderConfig
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
// NewSiliconFlowProvider 创建硅基流动厂商实例
|
||||
func NewSiliconFlowProvider(config ProviderConfig) (Provider, error) {
|
||||
return &SiliconFlowProvider{
|
||||
config: config,
|
||||
client: &http.Client{
|
||||
Timeout: 30 * time.Second,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
// Name 返回厂商名称
|
||||
func (p *SiliconFlowProvider) Name() string {
|
||||
return "siliconflow"
|
||||
}
|
||||
|
||||
// Validate 验证配置
|
||||
func (p *SiliconFlowProvider) Validate() error {
|
||||
if p.config.APIKey == "" {
|
||||
return fmt.Errorf("siliconflow: API key 不能为空")
|
||||
}
|
||||
if p.config.APIHost == "" {
|
||||
p.config.APIHost = "https://api.siliconflow.cn/v1"
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Translate 调用硅基流动API
|
||||
func (p *SiliconFlowProvider) Translate(ctx context.Context, req *TranslateRequest) (*TranslateResponse, error) {
|
||||
// 构建请求体
|
||||
requestBody := map[string]interface{}{
|
||||
"model": p.config.Model,
|
||||
"messages": []map[string]string{
|
||||
{
|
||||
"role": "user",
|
||||
"content": req.Text,
|
||||
},
|
||||
},
|
||||
"stream": false,
|
||||
}
|
||||
|
||||
// 如果有Prompt，添加到系统消息
|
||||
if req.Prompt != "" {
|
||||
messages := requestBody["messages"].([]map[string]string)
|
||||
requestBody["messages"] = append([]map[string]string{
|
||||
{
|
||||
"role": "system",
|
||||
"content": req.Prompt,
|
||||
},
|
||||
}, messages...)
|
||||
}
|
||||
|
||||
// 序列化请求体
|
||||
jsonData, err := json.Marshal(requestBody)
|
||||
if err != nil {
|
||||
return nil, NewTranslateErrorWithDetails("SERIALIZATION_ERROR", "请求序列化失败", err.Error())
|
||||
}
|
||||
|
||||
// 创建HTTP请求
|
||||
url := fmt.Sprintf("%s/chat/completions", p.config.APIHost)
|
||||
httpReq, err := http.NewRequestWithContext(ctx, "POST", url, bytes.NewBuffer(jsonData))
|
||||
if err != nil {
|
||||
return nil, NewTranslateErrorWithDetails("REQUEST_ERROR", "创建请求失败", err.Error())
|
||||
}
|
||||
|
||||
// 设置请求头
|
||||
httpReq.Header.Set("Content-Type", "application/json")
|
||||
httpReq.Header.Set("Authorization", fmt.Sprintf("Bearer %s", p.config.APIKey))
|
||||
|
||||
// 发送请求
|
||||
resp, err := p.client.Do(httpReq)
|
||||
if err != nil {
|
||||
return nil, NewTranslateErrorWithDetails("NETWORK_ERROR", "请求失败", err.Error())
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// 读取响应
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, NewTranslateErrorWithDetails("RESPONSE_ERROR", "读取响应失败", err.Error())
|
||||
}
|
||||
|
||||
// 检查HTTP状态码
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, NewTranslateErrorWithDetails(
|
||||
"HTTP_ERROR",
|
||||
fmt.Sprintf("HTTP错误: %d", resp.StatusCode),
|
||||
string(body),
|
||||
)
|
||||
}
|
||||
|
||||
// 解析响应
|
||||
var apiResp SiliconFlowResponse
|
||||
if err := json.Unmarshal(body, &apiResp); err != nil {
|
||||
return nil, NewTranslateErrorWithDetails("PARSE_ERROR", "解析响应失败", err.Error())
|
||||
}
|
||||
|
||||
// 检查API错误
|
||||
if apiResp.Error != nil {
|
||||
return nil, NewTranslateErrorWithDetails(
|
||||
"API_ERROR",
|
||||
apiResp.Error.Message,
|
||||
apiResp.Error.Code,
|
||||
)
|
||||
}
|
||||
|
||||
// 构建响应
|
||||
if len(apiResp.Choices) == 0 {
|
||||
return nil, NewTranslateError("NO_RESPONSE", "API返回空响应")
|
||||
}
|
||||
|
||||
translatedText := apiResp.Choices[0].Message.Content
|
||||
|
||||
return &TranslateResponse{
|
||||
Text: translatedText,
|
||||
FromLang: req.FromLang,
|
||||
ToLang: req.ToLang,
|
||||
Model: apiResp.Model,
|
||||
Usage: &Usage{
|
||||
PromptTokens: apiResp.Usage.PromptTokens,
|
||||
CompletionTokens: apiResp.Usage.CompletionTokens,
|
||||
TotalTokens: apiResp.Usage.TotalTokens,
|
||||
},
|
||||
RawResponse: body,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// SiliconFlowResponse 硅基流动API响应
|
||||
type SiliconFlowResponse struct {
|
||||
ID string `json:"id"`
|
||||
Object string `json:"object"`
|
||||
Created int64 `json:"created"`
|
||||
Model string `json:"model"`
|
||||
Choices []SiliconFlowChoice `json:"choices"`
|
||||
Usage SiliconFlowUsage `json:"usage"`
|
||||
Error *SiliconFlowError `json:"error,omitempty"`
|
||||
}
|
||||
|
||||
// SiliconFlowChoice 选择项
|
||||
type SiliconFlowChoice struct {
|
||||
Index int `json:"index"`
|
||||
Message SiliconFlowMessage `json:"message"`
|
||||
FinishReason string `json:"finish_reason"`
|
||||
}
|
||||
|
||||
// SiliconFlowMessage 消息
|
||||
type SiliconFlowMessage struct {
|
||||
Role string `json:"role"`
|
||||
Content string `json:"content"`
|
||||
}
|
||||
|
||||
// SiliconFlowUsage 用量
|
||||
type SiliconFlowUsage struct {
|
||||
PromptTokens int `json:"prompt_tokens"`
|
||||
CompletionTokens int `json:"completion_tokens"`
|
||||
TotalTokens int `json:"total_tokens"`
|
||||
}
|
||||
|
||||
// SiliconFlowError 错误
|
||||
type SiliconFlowError struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
|
||||
// init 注册硅基流动厂商
|
||||
func init() {
|
||||
RegisterProvider("siliconflow", NewSiliconFlowProvider)
|
||||
}
|
||||
131
internal/translator/prompt.go
Normal file
131
internal/translator/prompt.go
Normal file
@@ -0,0 +1,131 @@
|
||||
package translator
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// PromptManager Prompt管理器
|
||||
type PromptManager struct {
|
||||
prompts map[string]string
|
||||
}
|
||||
|
||||
// NewPromptManager 创建Prompt管理器
|
||||
func NewPromptManager(prompts map[string]string) *PromptManager {
|
||||
if prompts == nil {
|
||||
prompts = make(map[string]string)
|
||||
}
|
||||
return &PromptManager{
|
||||
prompts: prompts,
|
||||
}
|
||||
}
|
||||
|
||||
// GetPrompt 获取指定名称的Prompt
|
||||
func (pm *PromptManager) GetPrompt(name string) string {
|
||||
prompt, exists := pm.prompts[name]
|
||||
if !exists {
|
||||
return ""
|
||||
}
|
||||
return prompt
|
||||
}
|
||||
|
||||
// SetPrompt 设置Prompt
|
||||
func (pm *PromptManager) SetPrompt(name, content string) {
|
||||
pm.prompts[name] = content
|
||||
}
|
||||
|
||||
// DeletePrompt 删除Prompt
|
||||
func (pm *PromptManager) DeletePrompt(name string) {
|
||||
delete(pm.prompts, name)
|
||||
}
|
||||
|
||||
// HasPrompt 检查Prompt是否存在
|
||||
func (pm *PromptManager) HasPrompt(name string) bool {
|
||||
_, exists := pm.prompts[name]
|
||||
return exists
|
||||
}
|
||||
|
||||
// GetAllPrompts 获取所有Prompt名称
|
||||
func (pm *PromptManager) GetAllPrompts() []string {
|
||||
names := make([]string, 0, len(pm.prompts))
|
||||
for name := range pm.prompts {
|
||||
names = append(names, name)
|
||||
}
|
||||
return names
|
||||
}
|
||||
|
||||
// GetPromptCount 获取Prompt数量
|
||||
func (pm *PromptManager) GetPromptCount() int {
|
||||
return len(pm.prompts)
|
||||
}
|
||||
|
||||
// FormatPrompt 格式化Prompt，替换变量
|
||||
func (pm *PromptManager) FormatPrompt(name string, vars map[string]string) string {
|
||||
prompt := pm.GetPrompt(name)
|
||||
if prompt == "" {
|
||||
return ""
|
||||
}
|
||||
|
||||
// 替换变量
|
||||
for key, value := range vars {
|
||||
placeholder := fmt.Sprintf("{{%s}}", key)
|
||||
prompt = strings.ReplaceAll(prompt, placeholder, value)
|
||||
}
|
||||
|
||||
return prompt
|
||||
}
|
||||
|
||||
// ValidatePrompt 验证Prompt是否包含必要变量
|
||||
func (pm *PromptManager) ValidatePrompt(name string, requiredVars []string) error {
|
||||
prompt := pm.GetPrompt(name)
|
||||
if prompt == "" {
|
||||
return fmt.Errorf("prompt '%s' 不存在", name)
|
||||
}
|
||||
|
||||
for _, varName := range requiredVars {
|
||||
placeholder := fmt.Sprintf("{{%s}}", varName)
|
||||
if !strings.Contains(prompt, placeholder) {
|
||||
return fmt.Errorf("prompt '%s' 缺少必要变量: %s", name, varName)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// AddDefaultPrompts 添加默认Prompts
|
||||
func (pm *PromptManager) AddDefaultPrompts() {
|
||||
defaults := map[string]string{
|
||||
"technical": "你是一位专业的技术翻译，请准确翻译以下技术文档，保持专业术语的准确性。",
|
||||
"creative": "你是一位富有创造力的翻译家，请用优美流畅的语言翻译以下内容。",
|
||||
"academic": "你是一位学术翻译专家，请用严谨的学术语言翻译以下内容。",
|
||||
"simple": "请用简单易懂的语言翻译以下内容。",
|
||||
"code": "你是一位专业的代码翻译专家，请准确翻译以下代码注释和文档，保持代码结构和注释格式。",
|
||||
}
|
||||
|
||||
for name, content := range defaults {
|
||||
if !pm.HasPrompt(name) {
|
||||
pm.SetPrompt(name, content)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// GetPromptWithFallback 获取Prompt，如果不存在则返回默认Prompt
|
||||
func (pm *PromptManager) GetPromptWithFallback(name, fallback string) string {
|
||||
prompt := pm.GetPrompt(name)
|
||||
if prompt == "" {
|
||||
return fallback
|
||||
}
|
||||
return prompt
|
||||
}
|
||||
|
||||
// MergePrompts 合并多个Prompt
|
||||
func (pm *PromptManager) MergePrompts(names []string, separator string) string {
|
||||
var prompts []string
|
||||
for _, name := range names {
|
||||
prompt := pm.GetPrompt(name)
|
||||
if prompt != "" {
|
||||
prompts = append(prompts, prompt)
|
||||
}
|
||||
}
|
||||
return strings.Join(prompts, separator)
|
||||
}
|
||||
177
internal/translator/translator.go
Normal file
177
internal/translator/translator.go
Normal file
@@ -0,0 +1,177 @@
|
||||
package translator
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/titor/fanyi/internal/config"
|
||||
"github.com/titor/fanyi/internal/provider"
|
||||
)
|
||||
|
||||
// Translator 核心翻译类
|
||||
type Translator struct {
|
||||
config *config.Config
|
||||
provider provider.Provider
|
||||
prompt *PromptManager
|
||||
}
|
||||
|
||||
// NewTranslator 创建翻译器实例
|
||||
func NewTranslator(config *config.Config, provider provider.Provider) *Translator {
|
||||
return &Translator{
|
||||
config: config,
|
||||
provider: provider,
|
||||
prompt: NewPromptManager(config.Prompts),
|
||||
}
|
||||
}
|
||||
|
||||
// Translate 执行翻译
|
||||
func (t *Translator) Translate(ctx context.Context, text string, options *TranslateOptions) (*TranslateResult, error) {
|
||||
// 设置超时
|
||||
timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(t.config.Timeout)*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// 选择Prompt
|
||||
prompt := ""
|
||||
if options.PromptName != "" {
|
||||
prompt = t.prompt.GetPrompt(options.PromptName)
|
||||
}
|
||||
|
||||
// 构建请求
|
||||
req := &provider.TranslateRequest{
|
||||
Text: text,
|
||||
FromLang: options.FromLang,
|
||||
ToLang: options.ToLang,
|
||||
Prompt: prompt,
|
||||
Model: t.selectModel(options.Model),
|
||||
Options: options.ExtraOptions,
|
||||
}
|
||||
|
||||
// 调用厂商API
|
||||
resp, err := t.provider.Translate(timeoutCtx, req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("翻译失败: %w", err)
|
||||
}
|
||||
|
||||
// 构建结果
|
||||
return &TranslateResult{
|
||||
Original: text,
|
||||
Translated: resp.Text,
|
||||
FromLang: resp.FromLang,
|
||||
ToLang: resp.ToLang,
|
||||
Model: resp.Model,
|
||||
Usage: resp.Usage,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// TranslateWithProvider 使用指定厂商执行翻译
|
||||
func (t *Translator) TranslateWithProvider(ctx context.Context, text string, providerName string, options *TranslateOptions) (*TranslateResult, error) {
|
||||
// 创建指定厂商实例
|
||||
providerConfig, err := t.config.GetProviderConfig(providerName)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取厂商配置失败: %w", err)
|
||||
}
|
||||
|
||||
// 创建厂商实例
|
||||
providerInstance, err := provider.CreateProvider(providerName, provider.ProviderConfig{
|
||||
APIHost: providerConfig.APIHost,
|
||||
APIKey: providerConfig.APIKey,
|
||||
Model: providerConfig.Model,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建厂商实例失败: %w", err)
|
||||
}
|
||||
|
||||
// 临时切换厂商
|
||||
originalProvider := t.provider
|
||||
t.provider = providerInstance
|
||||
defer func() {
|
||||
t.provider = originalProvider
|
||||
}()
|
||||
|
||||
// 执行翻译
|
||||
return t.Translate(ctx, text, options)
|
||||
}
|
||||
|
||||
// selectModel 选择模型
|
||||
func (t *Translator) selectModel(model string) string {
|
||||
if model != "" {
|
||||
return model
|
||||
}
|
||||
return t.config.DefaultModel
|
||||
}
|
||||
|
||||
// GetProvider 获取当前厂商
|
||||
func (t *Translator) GetProvider() provider.Provider {
|
||||
return t.provider
|
||||
}
|
||||
|
||||
// GetConfig 获取配置
|
||||
func (t *Translator) GetConfig() *config.Config {
|
||||
return t.config
|
||||
}
|
||||
|
||||
// GetPromptManager 获取Prompt管理器
|
||||
func (t *Translator) GetPromptManager() *PromptManager {
|
||||
return t.prompt
|
||||
}
|
||||
|
||||
// SetTimeout 设置超时时间
|
||||
func (t *Translator) SetTimeout(seconds int) {
|
||||
t.config.Timeout = seconds
|
||||
}
|
||||
|
||||
// TranslateOptions 翻译选项
|
||||
type TranslateOptions struct {
|
||||
FromLang string
|
||||
ToLang string
|
||||
PromptName string
|
||||
Model string
|
||||
Temperature float64
|
||||
ExtraOptions map[string]interface{}
|
||||
}
|
||||
|
||||
// TranslateResult 翻译结果
|
||||
type TranslateResult struct {
|
||||
Original string
|
||||
Translated string
|
||||
FromLang string
|
||||
ToLang string
|
||||
Model string
|
||||
Usage *provider.Usage
|
||||
}
|
||||
|
||||
// String 返回翻译结果的字符串表示
|
||||
func (r *TranslateResult) String() string {
|
||||
return r.Translated
|
||||
}
|
||||
|
||||
// TranslateResultWithInfo 带详细信息的翻译结果
|
||||
type TranslateResultWithInfo struct {
|
||||
Result *TranslateResult
|
||||
Duration time.Duration
|
||||
Provider string
|
||||
Timestamp time.Time
|
||||
}
|
||||
|
||||
// BatchTranslate 批量翻译结果
|
||||
type BatchTranslateRequest struct {
|
||||
Texts []string
|
||||
Options *TranslateOptions
|
||||
}
|
||||
|
||||
// BatchTranslateResult 批量翻译结果
|
||||
type BatchTranslateResult struct {
|
||||
Results []*TranslateResult
|
||||
Errors []error
|
||||
Summary BatchTranslateSummary
|
||||
}
|
||||
|
||||
// BatchTranslateSummary 批量翻译摘要
|
||||
type BatchTranslateSummary struct {
|
||||
Total int
|
||||
Success int
|
||||
Failed int
|
||||
Duration time.Duration
|
||||
AvgTokens int
|
||||
}
|
||||
92
memory.md
92
memory.md
@@ -92,6 +92,31 @@
|
||||
|
||||
---
|
||||
|
||||
### 环境变量加载问题
|
||||
**问题**: 配置文件中的环境变量没有正确加载
|
||||
**原因**: Go程序不会自动加载.env文件，需要使用第三方库
|
||||
**解决方案**:
|
||||
1. 使用`github.com/joho/godotenv`包
|
||||
2. 在程序启动时调用`godotenv.Load()`
|
||||
3. 将.env文件添加到.gitignore
|
||||
|
||||
**代码示例**:
|
||||
```go
|
||||
import "github.com/joho/godotenv"
|
||||
|
||||
func main() {
|
||||
_ = godotenv.Load() // 加载.env文件
|
||||
// 然后加载配置文件
|
||||
}
|
||||
```
|
||||
|
||||
**注意事项**:
|
||||
- 不要提交真实的.env文件到版本控制
|
||||
- 提供.env.example模板
|
||||
- 在文档中说明环境变量配置方法
|
||||
|
||||
---
|
||||
|
||||
## 配置最佳实践
|
||||
|
||||
### 安全配置
|
||||
@@ -163,4 +188,69 @@
|
||||
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` 库
|
||||
- 注意缩进和格式
|
||||
- 提供配置验证
|
||||
141
taolun.md
141
taolun.md
@@ -85,4 +85,143 @@
|
||||
|
||||
**关联文档**:
|
||||
- [AGENTS.md#文档管理](AGENTS.md#开发规范)
|
||||
- [changelog.md#0.0.1](changelog.md#001)
|
||||
- [changelog.md#0.0.1](changelog.md#001)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-28 23:50] 版本 0.0.2 - 实现核心架构
|
||||
**原因**: 开始实现项目核心功能
|
||||
**分析**:
|
||||
- 根据OOP设计模式实现三个核心类
|
||||
- 需要先实现配置加载和厂商接口
|
||||
- 创建基本的CLI入口点
|
||||
|
||||
**解决方案**:
|
||||
1. **Config类实现**：
|
||||
- 支持YAML配置文件加载
|
||||
- 环境变量替换
|
||||
- 配置验证和默认值
|
||||
|
||||
2. **Provider接口实现**：
|
||||
- 定义统一的翻译接口
|
||||
- 工厂模式创建厂商实例
|
||||
- 实现硅基流动厂商作为示例
|
||||
|
||||
3. **Translator类实现**：
|
||||
- 核心翻译逻辑
|
||||
- Prompt管理
|
||||
- 超时控制
|
||||
|
||||
4. **CLI入口点**：
|
||||
- 命令行参数解析
|
||||
- 配置加载
|
||||
- 翻译执行
|
||||
|
||||
**技术细节**:
|
||||
- 使用`gopkg.in/yaml.v3`处理YAML
|
||||
- 实现工厂模式注册机制
|
||||
- 使用context处理超时和取消
|
||||
- 添加基本单元测试
|
||||
|
||||
**关联文档**:
|
||||
- [AGENTS.md#OOP设计模式](AGENTS.md#oop设计模式)
|
||||
- [changelog.md#0.0.2](changelog.md#002)
|
||||
|
||||
---
|
||||
|
||||
### [2026-03-29 00:00] 版本 0.0.3 - 环境变量加载修复
|
||||
**原因**: 测试CLI时发现环境变量没有正确加载
|
||||
**分析**:
|
||||
- 配置文件中使用`${ENV_VAR}`语法
|
||||
- Go的`os.ExpandEnv`只在加载时替换
|
||||
- 需要先加载.env文件到环境变量
|
||||
|
||||
**解决方案**:
|
||||
1. 添加`github.com/joho/godotenv`依赖
|
||||
2. 在main函数开始时调用`godotenv.Load()`
|
||||
3. 更新memory.md记录踩坑经验
|
||||
|
||||
**技术细节**:
|
||||
- godotenv会自动查找当前目录的.env文件
|
||||
- 如果文件不存在会返回错误，可以忽略
|
||||
- 不影响已有的环境变量
|
||||
|
||||
**关联文档**:
|
||||
- [memory.md#环境变量加载问题](memory.md#环境变量加载问题)
|
||||
- [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)
|
||||
Reference in New Issue
Block a user