Compare commits
48 Commits
88324c21c2
...
v0.7.0
| Author | SHA1 | Date | |
|---|---|---|---|
|
d6958dc41e
|
|||
|
a40c88c768
|
|||
|
e6b8b344aa
|
|||
| a0ea03fd90 | |||
| 3c2b96a4d1 | |||
| 16ffc94a06 | |||
| 9f177f7a1f | |||
| 18728a4a2e | |||
| f534ccc698 | |||
| 29c6ff3935 | |||
| bba15501c6 | |||
|
cc80604710
|
|||
|
35fe6e09b7
|
|||
|
349ff56299
|
|||
|
206bde0786
|
|||
|
995d263a48
|
|||
|
19aff8a6c1
|
|||
|
b6bc091502
|
|||
|
14ebb6857a
|
|||
|
459670f363
|
|||
|
7636d0b5a6
|
|||
|
928ebb61b4
|
|||
|
7e85cdd8b0
|
|||
|
90074e6e32
|
|||
|
b8182e7538
|
|||
|
4331b9306e
|
|||
|
a08bc809bb
|
|||
|
1063369d96
|
|||
|
3a57d25a76
|
|||
|
8152edba39
|
|||
|
679db5b1db
|
|||
|
b1ad68c7b5
|
|||
|
280d6ec5c9
|
|||
|
68427c4a11
|
|||
|
8dd9e85b77
|
|||
|
0c7d2ad518
|
|||
|
0289dd4684
|
|||
|
e2d43315e3
|
|||
|
0e1c2c6350
|
|||
|
da85fc94b1
|
|||
|
c66d782eab
|
|||
|
358b44ab81
|
|||
|
5957d67bc3
|
|||
|
04410ea9e7
|
|||
|
a514cdc69f
|
|||
|
e822ba1f54
|
|||
|
3c925d8268
|
|||
|
c9073ff4a7
|
8
.gitignore
vendored
8
.gitignore
vendored
@@ -6,6 +6,7 @@ Cargo.lock
|
|||||||
|
|
||||||
# IDE
|
# IDE
|
||||||
.idea/
|
.idea/
|
||||||
|
.trae/
|
||||||
.vscode/
|
.vscode/
|
||||||
*.swp
|
*.swp
|
||||||
*.swo
|
*.swo
|
||||||
@@ -21,3 +22,10 @@ test_output/
|
|||||||
|
|
||||||
# Config (for development)
|
# Config (for development)
|
||||||
config.toml
|
config.toml
|
||||||
|
.claude/
|
||||||
|
.qoder/
|
||||||
|
CLAUDE.md
|
||||||
|
**/agents/
|
||||||
|
adr/
|
||||||
|
.scratch/
|
||||||
|
CONTEXT.md
|
||||||
81
AGENTS.md
Normal file
81
AGENTS.md
Normal file
@@ -0,0 +1,81 @@
|
|||||||
|
# AGENTS.md
|
||||||
|
|
||||||
|
Behavioral guidelines to reduce common LLM coding mistakes. Merge with project-specific instructions as needed.
|
||||||
|
|
||||||
|
**Tradeoff:** These guidelines bias toward caution over speed. For trivial tasks, use judgment.
|
||||||
|
|
||||||
|
**Important:** Use Chinese for information responses and thinking; use English for searching and querying.
|
||||||
|
|
||||||
|
## 1. Think Before Coding
|
||||||
|
|
||||||
|
**Don't assume. Don't hide confusion. Surface tradeoffs.**
|
||||||
|
|
||||||
|
Before implementing:
|
||||||
|
- State your assumptions explicitly. If uncertain, ask.
|
||||||
|
- If multiple interpretations exist, present them - don't pick silently.
|
||||||
|
- If a simpler approach exists, say so. Push back when warranted.
|
||||||
|
- If something is unclear, stop. Name what's confusing. Ask.
|
||||||
|
|
||||||
|
## 2. Simplicity First
|
||||||
|
|
||||||
|
**Minimum code that solves the problem. Nothing speculative.**
|
||||||
|
|
||||||
|
- No features beyond what was asked.
|
||||||
|
- No abstractions for single-use code.
|
||||||
|
- No "flexibility" or "configurability" that wasn't requested.
|
||||||
|
- No error handling for impossible scenarios.
|
||||||
|
- If you write 200 lines and it could be 50, rewrite it.
|
||||||
|
|
||||||
|
Ask yourself: "Would a senior engineer say this is overcomplicated?" If yes, simplify.
|
||||||
|
|
||||||
|
## 3. Surgical Changes
|
||||||
|
|
||||||
|
**Touch only what you must. Clean up only your own mess.**
|
||||||
|
|
||||||
|
When editing existing code:
|
||||||
|
- Don't "improve" adjacent code, comments, or formatting.
|
||||||
|
- Don't refactor things that aren't broken.
|
||||||
|
- Match existing style, even if you'd do it differently.
|
||||||
|
- If you notice unrelated dead code, mention it - don't delete it.
|
||||||
|
|
||||||
|
When your changes create orphans:
|
||||||
|
- Remove imports/variables/functions that YOUR changes made unused.
|
||||||
|
- Don't remove pre-existing dead code unless asked.
|
||||||
|
|
||||||
|
The test: Every changed line should trace directly to the user's request.
|
||||||
|
|
||||||
|
## 4. Goal-Driven Execution
|
||||||
|
|
||||||
|
**Define success criteria. Loop until verified.**
|
||||||
|
|
||||||
|
Transform tasks into verifiable goals:
|
||||||
|
- "Add validation" → "Write tests for invalid inputs, then make them pass"
|
||||||
|
- "Fix the bug" → "Write a test that reproduces it, then make it pass"
|
||||||
|
- "Refactor X" → "Ensure tests pass before and after"
|
||||||
|
|
||||||
|
For multi-step tasks, state a brief plan:
|
||||||
|
```
|
||||||
|
1. [Step] → verify: [check]
|
||||||
|
2. [Step] → verify: [check]
|
||||||
|
3. [Step] → verify: [check]
|
||||||
|
```
|
||||||
|
|
||||||
|
Strong success criteria let you loop independently. Weak criteria ("make it work") require constant clarification.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
**These guidelines are working if:** fewer unnecessary changes in diffs, fewer rewrites due to overcomplication, and clarifying questions come before implementation rather than after mistakes.
|
||||||
|
|
||||||
|
## Agent skills
|
||||||
|
|
||||||
|
### Issue tracker
|
||||||
|
|
||||||
|
Issues and specs live as local markdown files under `.scratch/`. See `docs/agents/issue-tracker.md`.
|
||||||
|
|
||||||
|
### Triage labels
|
||||||
|
|
||||||
|
The five canonical triage roles use their default label strings. See `docs/agents/triage-labels.md`.
|
||||||
|
|
||||||
|
### Domain docs
|
||||||
|
|
||||||
|
Single-context repo with `CONTEXT.md` and `docs/adr/` at the root. See `docs/agents/domain.md`.
|
||||||
125
CHANGELOG.md
125
CHANGELOG.md
@@ -5,6 +5,131 @@ All notable changes to this project will be documented in this file.
|
|||||||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
||||||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
||||||
|
|
||||||
|
## [Unreleased]
|
||||||
|
|
||||||
|
暂无。
|
||||||
|
|
||||||
|
## [0.6.1] - 2026-08-18
|
||||||
|
|
||||||
|
### 🔧 其他变更
|
||||||
|
- 依赖瘦身:移除未使用的直接依赖 `config`、`handlebars`、`shell-words`、`walkdir`、`console`、`lazy_static`、`atty`,crate 总数由 341 降至 296,release 二进制体积由 7.38 MB 降至 7.00 MB(-5.2%)
|
||||||
|
- `lazy_static!` 宏替换为标准库 `LazyLock`;`atty` 替换为标准库 `std::io::IsTerminal`
|
||||||
|
- `tokio` 特性由 `full` 精简为 `macros` + `rt-multi-thread`
|
||||||
|
- `regex` 关闭默认 unicode 特性(保留 `std` + `perf`),行为变化:`\s`/`\d`/`\w` 字符类仅匹配 ASCII,对提交信息、版本号、邮箱、GPG Key ID 校验无实质影响
|
||||||
|
|
||||||
|
## [0.6.0] - 2026-08-17
|
||||||
|
|
||||||
|
### ✨ 新功能
|
||||||
|
- LLM 层整体迁移至 rig-core 0.41:六家 provider(Ollama / OpenAI / Anthropic / Kimi / DeepSeek / OpenRouter)统一由 rig 原生客户端驱动,删除约 2100 行手写 HTTP/SSE/重试代码
|
||||||
|
- 新 LLM 门面(`src/llm/rig/`):provider 枚举 + 单一生成入口 + 凭据校验(rig VerifyClient)+ 错误映射层;提示词与提交解析拆分至独立模块
|
||||||
|
- Anthropic 支持自定义 `base_url`(此前被静默忽略,本次修复)
|
||||||
|
- 新增基于 rig mock 后端的 LLM 层离线测试(请求形状、流式事件、错误映射),并附 6 家 provider 的 `#[ignore]` 实网冒烟用例
|
||||||
|
|
||||||
|
### 🐞 错误修复
|
||||||
|
- 各 provider 超时统一由 `llm.timeout` 配置控制(此前默认超时 60/120/300 秒不一致)
|
||||||
|
- 消除 Ollama 客户端构造时的 panic(`expect` 式构造)
|
||||||
|
|
||||||
|
### 🔧 其他变更(行为变化)
|
||||||
|
- Ollama 请求端点由 `/api/generate` 改为 `/api/chat`(功能等价;自定义 Ollama 代理需兼容 chat 端点)
|
||||||
|
- 非流式请求不再做应用层重试(此前按错误文案字符串匹配重试 3 次);流式请求由 rig 的 SSE 机制自动重连
|
||||||
|
- 可用性校验端点变化:DeepSeek 改为 `/user/balance`、OpenRouter 改为 `/key`、Anthropic 改为 `/v1/models`(对外表现为 `LLM provider 'xxx' is not available` 语义不变)
|
||||||
|
- 依赖瘦身:移除 `reqwest`(0.12) 与 `async-trait` 直接依赖,不引入 rig agent/fastembed/lancedb 等组件
|
||||||
|
|
||||||
|
## [0.5.0] - 2026-07-24
|
||||||
|
|
||||||
|
### ✨ 新功能
|
||||||
|
- `commit` 命令新增 `.gitignore` 文件过滤:自动暂存与 `--all` 模式下自动跳过被 `.gitignore` 规则匹配的文件,并清理索引中已被忽略的已跟踪文件,暂存完成后列出被移除的文件
|
||||||
|
- `tag` 命令新增 `-A, --auto` 模式:优先从 `Cargo.toml`/`pyproject.toml` 读取项目版本,回退到基于 commit 的语义化升级分析(与 `--bump` 互斥)
|
||||||
|
- `changelog` 命令在未指定 `--from` 时,自动检测现有 changelog 中最高版本对应的 tag 作为起始点
|
||||||
|
- 新增 `parse_changelog_versions()`,按 semver 降序提取 changelog 中已有的版本
|
||||||
|
- 新增 `sort_tags_by_semver()`,取代纯时间排序,按语义版本正确排序标签
|
||||||
|
|
||||||
|
### 🐞 错误修复
|
||||||
|
- 修复 `get_tags()` 对 annotated tag 的解析:改用 `find_object`+`peel_to_commit` 正确取得标签指向的 commit
|
||||||
|
- 修复 `changelog` 写入时 `insert_changelog_entry` 覆盖已有章节的问题,现保留现有内容并按版本插入新条目
|
||||||
|
|
||||||
|
### 📚 文档
|
||||||
|
- 同步更新 README(中/英文):补充 `--think`、`tag --auto` 选项说明,移除已废弃的 `--prepend` 参数,修正 `set-llm` 的 `--base-url` 选项名
|
||||||
|
|
||||||
|
### 🔧 其他变更
|
||||||
|
- 新增 `TagInfo::version_name()` 与 `GitRepo::find_tag_by_version()` 辅助方法
|
||||||
|
- 新增 `tests/gitignore_tests.rs`,包含 7 个测试覆盖 `.gitignore` 过滤逻辑
|
||||||
|
|
||||||
|
## [0.4.0] - 2026-07-16
|
||||||
|
|
||||||
|
### ✨ 新功能
|
||||||
|
- 新增 `quicommit credential get|store|erase` 子命令,实现标准 Git credential helper 协议([gitcredentials](https://git-scm.com/docs/gitcredentials)),可与原生 `git` 命令无缝集成
|
||||||
|
- 新增 `host_to_service()` 主机名映射,将 github.com、gitlab.com、bitbucket.org 等常见托管平台映射为规范化服务名
|
||||||
|
- 新增 `get_pat_for_host()` 公共 API,支持从已保存的凭据中提取 PAT 用于登录验证
|
||||||
|
- 凭据存储复用现有基于系统密钥环的 PAT 与用户绑定逻辑,按 profile 维度管理
|
||||||
|
|
||||||
|
### 🐞 错误修复
|
||||||
|
- 移除 `keyring.rs` 中的调试 `eprintln!` 输出,避免污染 credential helper 的 stderr
|
||||||
|
|
||||||
|
### 📚 文档
|
||||||
|
- README(中/英文)新增 credential 命令使用说明
|
||||||
|
|
||||||
|
### 🔧 其他变更
|
||||||
|
- 新增 `src/lib.rs` 库目标,支持从 `tests/` 目录导入内部模块进行测试
|
||||||
|
- `src/main.rs` 重构为使用 `quicommit::` 库导入
|
||||||
|
- credential 命令及其子命令均使用 `#[command(hide = true)]` 隐藏,不在 `--help` 中显示
|
||||||
|
- 新增 `tests/credential_tests.rs`,包含 52 个测试用例覆盖协议解析、帮助可见性、完整存取周期及边界场景
|
||||||
|
|
||||||
|
## [0.3.1] - 2026-06-01
|
||||||
|
|
||||||
|
### ✨ 新功能
|
||||||
|
- 按文件重要性对暂存差异排序,优先处理核心变更
|
||||||
|
- DeepSeek 新增 reasoning 推理模式支持
|
||||||
|
- LLM 统一思考模式配置,支持显式启用/禁用思考状态
|
||||||
|
- 新增 `thinking.rs` 思考状态管理模块
|
||||||
|
|
||||||
|
### 🐞 错误修复
|
||||||
|
- 修复 Kimi 返回信息的读取错误
|
||||||
|
- 修复 DeepSeek 和 Kimi 流式响应的解析问题
|
||||||
|
|
||||||
|
### 📚 文档
|
||||||
|
- 新增 ROADMAP.md 项目路线图文档
|
||||||
|
|
||||||
|
### 🔧 其他变更
|
||||||
|
- LLM 模块大规模重构,所有提供商(Anthropic、DeepSeek、Kimi、Ollama、OpenAI、OpenRouter)适配流式响应处理
|
||||||
|
- 代码格式化并优化导入顺序
|
||||||
|
- 清理大量未使用的变量、方法及结构体警告
|
||||||
|
- 清理构建输出日志文件
|
||||||
|
- 重新编号 LLM 系统提示规则
|
||||||
|
- i18n 多语言消息格式修复
|
||||||
|
- 各命令模块(commit、tag、changelog、config、profile、init)持续优化
|
||||||
|
|
||||||
|
## [0.1.11] - 2026-03-23
|
||||||
|
|
||||||
|
### ✨ 新功能
|
||||||
|
- 新增配置导出导入功能,支持加密保护
|
||||||
|
- Profile 支持 Token 管理(PAT 等)
|
||||||
|
- 自动生成和维护 Keep a Changelog 格式的变更日志
|
||||||
|
- 交互式命令行界面,支持预览和确认
|
||||||
|
|
||||||
|
### 🔐 安全特性
|
||||||
|
- 敏感数据加密存储(API 密钥等)
|
||||||
|
- 使用系统密钥环安全保存凭证
|
||||||
|
|
||||||
|
### 🔧 其他变更
|
||||||
|
- 优化 diff 截断逻辑,使用字符边界确保多字节字符安全
|
||||||
|
- 改进配置管理器,支持修改追踪
|
||||||
|
|
||||||
|
## [0.1.9] - 2026-03-06
|
||||||
|
|
||||||
|
### 🐞 错误修复
|
||||||
|
- 修复diff截断时的字符边界问题
|
||||||
|
|
||||||
|
## [0.1.7] - 2026-02-14
|
||||||
|
|
||||||
|
### 🐞 错误修复
|
||||||
|
- 修复 `changelog` 命令默认覆盖文件的问题,现改为智能追加新版本条目到头部之后
|
||||||
|
|
||||||
|
### 🔧 其他变更
|
||||||
|
- 清理 `formatter.rs` 中未使用的函数(`format_commit_date`、`format_changelog_date`、`format_tag_name`、`truncate`、`format_markdown_list`、`format_changelog_section`、`format_git_config_key`)
|
||||||
|
- 清理 `validators.rs` 中未使用的函数(`validate_ssh_key`)
|
||||||
|
- 移除 `changelog` 命令的 `--prepend` 参数(默认行为已改为追加)
|
||||||
|
|
||||||
## [0.1.4] - 2026-02-01
|
## [0.1.4] - 2026-02-01
|
||||||
|
|
||||||
### ✨ 新功能
|
### ✨ 新功能
|
||||||
|
|||||||
37
Cargo.toml
37
Cargo.toml
@@ -1,9 +1,9 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "quicommit"
|
name = "quicommit"
|
||||||
version = "0.1.4"
|
version = "0.7.0"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
authors = ["Sidney Zhang <zly@lyzhang.me>"]
|
authors = ["Sidney Zhang <zly@lyzhang.me>"]
|
||||||
description = "A powerful Git assistant tool with AI-powered commit/tag/changelog generation(alpha version)"
|
description = "A powerful Git assistant tool with AI-powered commit/tag/changelog generation"
|
||||||
license = "MIT"
|
license = "MIT"
|
||||||
repository = "https://git.lyz.one/SidneyZhang/QuiCommit"
|
repository = "https://git.lyz.one/SidneyZhang/QuiCommit"
|
||||||
keywords = ["git", "commit", "ai", "cli", "automation"]
|
keywords = ["git", "commit", "ai", "cli", "automation"]
|
||||||
@@ -19,11 +19,9 @@ path = "src/main.rs"
|
|||||||
clap = { version = "4.5", features = ["derive", "env", "wrap_help"] }
|
clap = { version = "4.5", features = ["derive", "env", "wrap_help"] }
|
||||||
clap_complete = "4.5"
|
clap_complete = "4.5"
|
||||||
dialoguer = "0.11"
|
dialoguer = "0.11"
|
||||||
console = "0.15"
|
|
||||||
indicatif = "0.17"
|
indicatif = "0.17"
|
||||||
|
|
||||||
# Configuration management
|
# Configuration management
|
||||||
config = "0.14"
|
|
||||||
serde = { version = "1.0", features = ["derive"] }
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
toml = "0.8"
|
toml = "0.8"
|
||||||
dirs = "5.0"
|
dirs = "5.0"
|
||||||
@@ -32,9 +30,7 @@ dirs = "5.0"
|
|||||||
git2 = "0.20.3"
|
git2 = "0.20.3"
|
||||||
which = "6.0"
|
which = "6.0"
|
||||||
|
|
||||||
# HTTP client for LLM APIs
|
tokio = { version = "1.35", features = ["macros", "rt-multi-thread"] }
|
||||||
reqwest = { version = "0.12", features = ["json", "rustls-tls"], default-features = false }
|
|
||||||
tokio = { version = "1.35", features = ["full"] }
|
|
||||||
|
|
||||||
# Error handling
|
# Error handling
|
||||||
thiserror = "1.0"
|
thiserror = "1.0"
|
||||||
@@ -46,19 +42,16 @@ tracing-subscriber = { version = "0.3", features = ["env-filter", "fmt"] }
|
|||||||
|
|
||||||
# Utilities
|
# Utilities
|
||||||
chrono = { version = "0.4", features = ["serde"] }
|
chrono = { version = "0.4", features = ["serde"] }
|
||||||
regex = "1.10"
|
regex = { version = "1.10", default-features = false, features = ["std", "perf", "unicode-perl"] }
|
||||||
lazy_static = "1.4"
|
roxmltree = "0.20"
|
||||||
colored = "2.1"
|
colored = "2.1"
|
||||||
handlebars = "5.1"
|
|
||||||
semver = "1.0"
|
semver = "1.0"
|
||||||
walkdir = "2.4"
|
|
||||||
tempfile = "3.9"
|
tempfile = "3.9"
|
||||||
sha2 = "0.10"
|
sha2 = "0.10"
|
||||||
hex = "0.4"
|
hex = "0.4"
|
||||||
textwrap = "0.16"
|
textwrap = "0.16"
|
||||||
async-trait = "0.1"
|
futures-util = "0.3"
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
atty = "0.2"
|
|
||||||
|
|
||||||
# Encryption for sensitive data (SSH keys, GPG, etc.)
|
# Encryption for sensitive data (SSH keys, GPG, etc.)
|
||||||
aes-gcm = "0.10"
|
aes-gcm = "0.10"
|
||||||
@@ -66,11 +59,14 @@ argon2 = "0.5"
|
|||||||
rand = "0.8"
|
rand = "0.8"
|
||||||
base64 = "0.22"
|
base64 = "0.22"
|
||||||
|
|
||||||
|
# System keyring for secure API key storage
|
||||||
|
keyring = { version = "3.6.3", features = ["apple-native", "windows-native", "sync-secret-service"] }
|
||||||
|
|
||||||
# Interactive editor
|
# Interactive editor
|
||||||
edit = "0.1"
|
edit = "0.1"
|
||||||
|
|
||||||
# Shell completion generation
|
# LLM integration (rig-core only: HTTP backend + rustls; no agent/derive)
|
||||||
shell-words = "1.1"
|
rig-core = { version = "0.41", default-features = false, features = ["reqwest", "rustls"] }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
assert_cmd = "2.0"
|
assert_cmd = "2.0"
|
||||||
@@ -78,13 +74,18 @@ predicates = "3.1"
|
|||||||
tempfile = "3.9"
|
tempfile = "3.9"
|
||||||
mockall = "0.12"
|
mockall = "0.12"
|
||||||
wiremock = "0.6"
|
wiremock = "0.6"
|
||||||
|
# rig mock HTTP backends for LLM layer tests
|
||||||
|
rig-core = { version = "0.41", default-features = false, features = ["test-utils"] }
|
||||||
|
http = "1"
|
||||||
|
|
||||||
[profile.release]
|
[profile.release]
|
||||||
opt-level = 3
|
opt-level = "s"
|
||||||
lto = true
|
lto = "thin"
|
||||||
codegen-units = 1
|
codegen-units = 1
|
||||||
|
panic = "abort"
|
||||||
strip = true
|
strip = true
|
||||||
|
debug = false
|
||||||
|
|
||||||
[profile.dev]
|
[profile.dev]
|
||||||
opt-level = 0
|
opt-level = 1
|
||||||
debug = true
|
debug = true
|
||||||
|
|||||||
122
RAODMAP.md
Normal file
122
RAODMAP.md
Normal file
@@ -0,0 +1,122 @@
|
|||||||
|
# QuiCommit Roadmap
|
||||||
|
|
||||||
|
## 已完成 ✅
|
||||||
|
|
||||||
|
- [x] 基础 Git 操作(commit、tag、changelog)
|
||||||
|
- [x] AI 驱动提交信息生成(Conventional Commits / commitlint 格式)
|
||||||
|
- [x] 多 LLM 提供商支持:Ollama、OpenAI、Anthropic、Kimi、DeepSeek、OpenRouter
|
||||||
|
- [x] 多 Git Profile 管理(SSH 密钥 + GPG 签名)
|
||||||
|
- [x] 语义化版本自动升级与 AI 发布说明
|
||||||
|
- [x] Keep a Changelog 格式自动生成
|
||||||
|
- [x] 系统密钥环安全存储 API Key
|
||||||
|
- [x] 敏感数据加密存储(AES-GCM + Argon2)
|
||||||
|
- [x] 交互式 CLI 预览与确认
|
||||||
|
- [x] 7 种语言国际化支持
|
||||||
|
- [x] 配置导出/导入(支持加密保护)
|
||||||
|
- [x] Profile Token 管理(PAT 等)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 进行中 🚧
|
||||||
|
|
||||||
|
暂无。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 计划中 📋
|
||||||
|
|
||||||
|
### 1. Git 凭证管理器
|
||||||
|
|
||||||
|
将 Git 凭证管理集成到 QuiCommit 中,统一管理 HTTPS 仓库的身份认证。
|
||||||
|
|
||||||
|
- [x] **Git Credential Helper 集成**
|
||||||
|
- 实现 `git credential-store` / `git-credential-libsecret` 等标准的 credential helper 协议
|
||||||
|
- 支持 `quicommit credential get|store|erase` 子命令
|
||||||
|
- 与系统密钥环无缝对接,复用已有的 `KeyringManager`
|
||||||
|
|
||||||
|
- [x] **跨平台支持**
|
||||||
|
- Windows:集成 Windows Credential Manager
|
||||||
|
- macOS:集成 Keychain
|
||||||
|
- Linux:通过 Secret Service / D-Bus 对接 GNOME Keyring / KWallet
|
||||||
|
|
||||||
|
- [x] **安全增强**
|
||||||
|
- 支持 PAT(Personal Access Token)按 scope / 有效期管理
|
||||||
|
- 支持凭证过期检查和自动提醒
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 2. 新增模型支持
|
||||||
|
|
||||||
|
扩展 LLM 提供商和模型覆盖范围,满足更多场景和偏好。
|
||||||
|
|
||||||
|
- [x] **新增 DeepSeek 最新模型**
|
||||||
|
- 支持 `deepseek-chat`(DeepSeek-V3)
|
||||||
|
- 支持 `deepseek-reasoner`(DeepSeek-R1)
|
||||||
|
- 支持 `deepseek-v4`
|
||||||
|
|
||||||
|
- [ ] **新增国内模型提供商**
|
||||||
|
- 通义千问 (Qwen) — 阿里云 DashScope API
|
||||||
|
- 文心一言 (ERNIE) — 百度千帆 API
|
||||||
|
- 智谱 GLM — ChatGLM API
|
||||||
|
- 百川 (Baichuan) — Baichuan API
|
||||||
|
|
||||||
|
- [ ] **新增国际模型提供商**
|
||||||
|
- Google Gemini API
|
||||||
|
- Mistral AI API
|
||||||
|
- Cohere API
|
||||||
|
- Groq (LPU 推理加速)
|
||||||
|
|
||||||
|
- [ ] **本地模型扩展**
|
||||||
|
- 支持 llama.cpp 服务端(兼容 OpenAI API 格式)
|
||||||
|
- 支持 vLLM 部署的模型
|
||||||
|
- 本地模型推荐列表与一键配置向导
|
||||||
|
|
||||||
|
- [ ] **模型能力适配**
|
||||||
|
- 不同模型的 token 限制自适应
|
||||||
|
- 模型特定的 prompt 模板优化
|
||||||
|
- 支持 function calling / tool use(用于复杂生成场景)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 3. 生成体验优化
|
||||||
|
|
||||||
|
提升 AI 生成提交信息、标签说明和变更日志时的用户体验。
|
||||||
|
|
||||||
|
- [ ] **流式输出与实时反馈**
|
||||||
|
- [x] 支持 SSE(Server-Sent Events)流式生成
|
||||||
|
- [ ]终端打字机效果实时显示生成内容
|
||||||
|
- [ ]流式生成过程中支持 `Ctrl+C` 中断
|
||||||
|
|
||||||
|
- [ ] **生成质量提升**
|
||||||
|
- 基于 commitlint 规则的后校验与自动修正
|
||||||
|
- 支持 Few-shot 示例引导(用户可自定义示例库)
|
||||||
|
- 生成结果的置信度评分与多候选方案
|
||||||
|
|
||||||
|
- [ ] **Diff 上下文增强**
|
||||||
|
- 智能 diff 摘要(大改动时自动压缩关键信息)
|
||||||
|
- 支持 `.gitattributes` 排除/包含规则
|
||||||
|
- 按文件类型分组生成更精准的提交描述
|
||||||
|
|
||||||
|
- [ ] **交互式编辑增强**
|
||||||
|
- 生成后支持内联编辑(类似 `git rebase -i` 体验)
|
||||||
|
- 支持重新生成指定部分(如 scope、description)
|
||||||
|
- 历史提交信息学习与风格适配
|
||||||
|
|
||||||
|
- [ ] **批量操作支持**
|
||||||
|
- 批量生成多个 commit(分组暂存区变更)
|
||||||
|
- `--dry-run` 预览模式(只生成本地查看,不写 Git)
|
||||||
|
|
||||||
|
- [ ] **性能优化**
|
||||||
|
- API 请求并发优化(多个模型同时生成候选)
|
||||||
|
- 本地缓存常用 prompt 模板
|
||||||
|
- 减少不必要的 diff 计算
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 长远规划 🌟
|
||||||
|
|
||||||
|
- [ ] **VS Code 扩展** — 在 IDE 内直接使用 QuiCommit
|
||||||
|
- [ ] **GitHub Action / GitLab CI 集成** — 自动化 PR 标题和描述生成
|
||||||
|
- [ ] **团队协作** — 共享 commit 风格配置、prompt 模板库
|
||||||
|
- [ ] **Web Dashboard** — 可视化管理多仓库的 Git 活动与 AI 生成统计
|
||||||
|
- [ ] **插件系统** — 允许社区贡献自定义 LLM 提供商和生成策略
|
||||||
228
README.md
228
README.md
@@ -6,18 +6,24 @@ A powerful AI-powered Git assistant for generating conventional commits, tags, a
|
|||||||
|
|
||||||
[Still in early development, some features may not be complete. Feedback and contributions are welcome.]
|
[Still in early development, some features may not be complete. Feedback and contributions are welcome.]
|
||||||
|
|
||||||
|
> ⚠️ **Important Notice**: QuiCommit now uses system keyring to store API keys securely. This change may cause breaking changes to your existing configuration. If you encounter issues after updating, please run `quicommit config reset --force` to reset your configuration, then reconfigure your settings.
|
||||||
|
|
||||||

|

|
||||||

|

|
||||||
|

|
||||||
|
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
- **AI-Powered Generation**: Generate commits, tags, and changelogs using LLM APIs (Ollama, OpenAI, Anthropic, Kimi, DeepSeek, OpenRouter) or local models
|
- **AI-Powered Generation**: Generate commits, tags, and changelogs using LLM APIs (Ollama, OpenAI, Anthropic, Kimi, DeepSeek, OpenRouter) or local models
|
||||||
- **Conventional Commits**: Full support for Conventional Commits and commitlint formats
|
- **Conventional Commits**: Full support for Conventional Commits and commitlint formats
|
||||||
- **Profile Management**: Manage multiple Git identities with SSH keys and GPG signing support
|
- **Profile Management**: Manage multiple Git identities with SSH keys and GPG signing support
|
||||||
- **Smart Tagging**: Semantic version bumping with AI-generated release notes
|
- **Smart Tagging**: Semantic version bumping with AI-generated release notes; auto-detects version from 16+ project config files (Cargo.toml, package.json, pom.xml, pubspec.yaml, etc.)
|
||||||
- **Changelog Generation**: Automatic changelog generation in Keep a Changelog format
|
- **Changelog Generation**: Automatic changelog generation in Keep a Changelog format
|
||||||
- **Security**: Encrypt sensitive data
|
- **Security**: Use system keyring to store API keys securely
|
||||||
- **Interactive UI**: Beautiful CLI with previews and confirmations
|
- **Interactive UI**: Beautiful CLI with previews and confirmations
|
||||||
|
- **Multi-language Support**: Output in 7 languages (English, Chinese, Japanese, Korean, Spanish, French, German)
|
||||||
|
- **Config Export/Import**: Backup and restore configuration with optional encryption
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
@@ -57,7 +63,7 @@ quicommit commit
|
|||||||
# Manual commit
|
# Manual commit
|
||||||
quicommit commit --manual -t feat -m "add new feature"
|
quicommit commit --manual -t feat -m "add new feature"
|
||||||
|
|
||||||
# Stage all and commit
|
# Stage all and commit (skips .gitignore-matched files automatically)
|
||||||
quicommit commit -a
|
quicommit commit -a
|
||||||
|
|
||||||
# Skip confirmation
|
# Skip confirmation
|
||||||
@@ -66,16 +72,25 @@ quicommit commit --yes
|
|||||||
# Use date-based commit message
|
# Use date-based commit message
|
||||||
quicommit commit --date
|
quicommit commit --date
|
||||||
|
|
||||||
|
# Enable LLM thinking/reasoning mode for this commit
|
||||||
|
quicommit commit --think
|
||||||
|
|
||||||
# Push after committing
|
# Push after committing
|
||||||
quicommit commit --push
|
quicommit commit --push
|
||||||
```
|
```
|
||||||
|
|
||||||
|
When staging changes (auto-stage or `--all`), files matched by `.gitignore` rules are skipped and any ignored files already in the index are removed; the skipped list is printed for review.
|
||||||
|
|
||||||
### Create Tag
|
### Create Tag
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Auto-detect version bump
|
# Auto-detect version bump
|
||||||
quicommit tag
|
quicommit tag
|
||||||
|
|
||||||
|
# Auto-detect version from project config files (Cargo.toml, package.json, pom.xml, etc.),
|
||||||
|
# fall back to commit analysis
|
||||||
|
quicommit tag --auto
|
||||||
|
|
||||||
# Specify bump type
|
# Specify bump type
|
||||||
quicommit tag --bump minor
|
quicommit tag --bump minor
|
||||||
|
|
||||||
@@ -92,11 +107,14 @@ quicommit tag --push
|
|||||||
### Generate Changelog
|
### Generate Changelog
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Generate for unreleased changes
|
# Generate for unreleased changes (auto-detects --from from the highest version in the existing changelog)
|
||||||
quicommit changelog
|
quicommit changelog
|
||||||
|
|
||||||
# Generate for specific version
|
# Generate for specific version
|
||||||
quicommit changelog -v 1.0.0
|
quicommit changelog --version 1.0.0
|
||||||
|
|
||||||
|
# Generate from a specific tag
|
||||||
|
quicommit changelog --from v0.9.0
|
||||||
|
|
||||||
# AI-generate changelog
|
# AI-generate changelog
|
||||||
quicommit changelog --generate
|
quicommit changelog --generate
|
||||||
@@ -154,35 +172,96 @@ quicommit profile stats
|
|||||||
quicommit profile token
|
quicommit profile token
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Git Credential Helper
|
||||||
|
|
||||||
|
QuiCommit can act as a Git credential helper to securely store and retrieve
|
||||||
|
Personal Access Tokens (PATs) via the system keyring. The `credential` command
|
||||||
|
is hidden from `--help` because it is invoked automatically by Git, not by end
|
||||||
|
users.
|
||||||
|
|
||||||
|
#### Setup
|
||||||
|
|
||||||
|
Register QuiCommit as a credential helper for Git:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Use the default QuiCommit config
|
||||||
|
git config --global credential.helper quicommit
|
||||||
|
|
||||||
|
# Or specify a custom config file
|
||||||
|
git config --global credential.helper "quicommit --config /path/to/config.toml"
|
||||||
|
```
|
||||||
|
|
||||||
|
You can also limit the helper to a specific host:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
git config --global credential.https://github.com.helper quicommit
|
||||||
|
```
|
||||||
|
|
||||||
|
#### How It Works
|
||||||
|
|
||||||
|
When Git needs credentials (e.g. pushing to a remote), it calls the helper
|
||||||
|
following the [gitcredentials protocol](https://git-scm.com/docs/gitcredentials):
|
||||||
|
|
||||||
|
1. **`get`** — Git asks QuiCommit for a stored PAT matching the requested host.
|
||||||
|
QuiCommit searches all configured profiles in the keyring and returns the
|
||||||
|
PAT (plus username) if found.
|
||||||
|
2. **`store`** — After a successful authentication (e.g. you entered a PAT in
|
||||||
|
the Git prompt), Git asks QuiCommit to save it. The PAT is stored in the
|
||||||
|
system keyring, bound to the matching profile.
|
||||||
|
3. **`erase`** — Git asks QuiCommit to delete a stored PAT for the given host.
|
||||||
|
|
||||||
|
Host names are mapped to canonical service names (`github.com` → `github`,
|
||||||
|
`gitlab.com` → `gitlab`, `bitbucket.org` → `bitbucket`, etc.). Unknown hosts
|
||||||
|
are used as-is.
|
||||||
|
|
||||||
|
#### Removing Stored Credentials
|
||||||
|
|
||||||
|
To remove a stored PAT, you can either use Git's built-in mechanism:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
echo "protocol=https
|
||||||
|
host=github.com" | git credential reject
|
||||||
|
```
|
||||||
|
|
||||||
|
Or remove it via the profile token management command:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
quicommit profile token
|
||||||
|
```
|
||||||
|
|
||||||
### Configure LLM
|
### Configure LLM
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Configure Ollama (local)
|
# Configure Ollama (local)
|
||||||
quicommit config set-llm ollama
|
quicommit config set-llm ollama
|
||||||
quicommit config set-ollama --url http://localhost:11434 --model llama2
|
quicommit config set-llm ollama --base-url http://localhost:11434 --model llama2
|
||||||
|
|
||||||
|
> ⚠️ **Security note**: passing your API key as a CLI argument puts it in your shell
|
||||||
|
> history and process list. Prefer running `quicommit config set-api-key` (or
|
||||||
|
> `config set-llm`) without the key to enter it via a hidden prompt instead.
|
||||||
|
|
||||||
# Configure OpenAI
|
# Configure OpenAI
|
||||||
quicommit config set-llm openai
|
quicommit config set-llm openai
|
||||||
quicommit config set-openai-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
|
|
||||||
# Configure Anthropic Claude
|
# Configure Anthropic Claude
|
||||||
quicommit config set-llm anthropic
|
quicommit config set-llm anthropic
|
||||||
quicommit config set-anthropic-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
|
|
||||||
# Configure Kimi (Moonshot AI)
|
# Configure Kimi (Moonshot AI)
|
||||||
quicommit config set-llm kimi
|
quicommit config set-llm kimi
|
||||||
quicommit config set-kimi-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
quicommit config set-kimi --base-url https://api.moonshot.cn/v1 --model moonshot-v1-8k
|
quicommit config set-llm kimi --base-url https://api.moonshot.cn/v1 --model moonshot-v1-8k
|
||||||
|
|
||||||
# Configure DeepSeek
|
# Configure DeepSeek
|
||||||
quicommit config set-llm deepseek
|
quicommit config set-llm deepseek
|
||||||
quicommit config set-deepseek-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
quicommit config set-deepseek --base-url https://api.deepseek.com/v1 --model deepseek-chat
|
quicommit config set-llm deepseek --base-url https://api.deepseek.com/v1 --model deepseek-chat
|
||||||
|
|
||||||
# Configure OpenRouter
|
# Configure OpenRouter
|
||||||
quicommit config set-llm openrouter
|
quicommit config set-llm openrouter
|
||||||
quicommit config set-openrouter-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
quicommit config set-openrouter --base-url https://openrouter.ai/api/v1 --model openai/gpt-4
|
quicommit config set-llm openrouter --base-url https://openrouter.ai/api/v1 --model openai/gpt-4
|
||||||
|
|
||||||
# Set commit format
|
# Set commit format
|
||||||
quicommit config set-commit-format conventional
|
quicommit config set-commit-format conventional
|
||||||
@@ -193,7 +272,7 @@ quicommit config set-version-prefix v
|
|||||||
# Set changelog path
|
# Set changelog path
|
||||||
quicommit config set-changelog-path CHANGELOG.md
|
quicommit config set-changelog-path CHANGELOG.md
|
||||||
|
|
||||||
# Set output language
|
# Set output language (en, zh, ja, ko, es, fr, de)
|
||||||
quicommit config set-language en
|
quicommit config set-language en
|
||||||
|
|
||||||
# Set keep commit types in English
|
# Set keep commit types in English
|
||||||
@@ -205,8 +284,22 @@ quicommit config set-keep-changelog-types-english true
|
|||||||
# Test LLM connection
|
# Test LLM connection
|
||||||
quicommit config test-llm
|
quicommit config test-llm
|
||||||
|
|
||||||
|
# Check keyring availability
|
||||||
|
quicommit config check-keyring
|
||||||
|
|
||||||
|
# Show config file path
|
||||||
|
quicommit config path
|
||||||
|
|
||||||
|
# Export configuration (with optional encryption)
|
||||||
|
quicommit config export -o config-backup.toml
|
||||||
|
quicommit config export -o config-backup.enc --password
|
||||||
|
|
||||||
|
# Import configuration
|
||||||
|
quicommit config import -i config-backup.toml
|
||||||
|
quicommit config import -i config-backup.enc --password
|
||||||
|
|
||||||
# Reset configuration to defaults
|
# Reset configuration to defaults
|
||||||
quicommit config reset
|
quicommit config reset --force
|
||||||
```
|
```
|
||||||
|
|
||||||
## Command Reference
|
## Command Reference
|
||||||
@@ -219,6 +312,7 @@ quicommit config reset
|
|||||||
| `quicommit changelog` | `cl` | Generate changelog |
|
| `quicommit changelog` | `cl` | Generate changelog |
|
||||||
| `quicommit profile` | `p` | Manage Git profiles |
|
| `quicommit profile` | `p` | Manage Git profiles |
|
||||||
| `quicommit config` | `cfg` | Manage settings |
|
| `quicommit config` | `cfg` | Manage settings |
|
||||||
|
| `quicommit credential` | — | Git credential helper (hidden, invoked by Git) |
|
||||||
|
|
||||||
### Commit Options
|
### Commit Options
|
||||||
|
|
||||||
@@ -231,14 +325,15 @@ quicommit config reset
|
|||||||
| `-b, --breaking` | Mark as breaking change |
|
| `-b, --breaking` | Mark as breaking change |
|
||||||
| `-d, --date` | Use date-based commit message |
|
| `-d, --date` | Use date-based commit message |
|
||||||
| `--manual` | Manual input, skip AI |
|
| `--manual` | Manual input, skip AI |
|
||||||
| `-a, --all` | Stage all changes |
|
| `-a, --all` | Stage all changes (skips `.gitignore`-matched files) |
|
||||||
| `-S, --sign` | GPG sign commit |
|
| `-S, --sign` | GPG sign commit |
|
||||||
| `--amend` | Amend previous commit |
|
| `--amend` | Amend previous commit |
|
||||||
| `--dry-run` | Show without committing |
|
| `--dry-run` | Show without committing |
|
||||||
| `--conventional` | Use Conventional Commits format |
|
| `--conventional` | Use Conventional Commits format |
|
||||||
| `--commitlint` | Use commitlint format |
|
| `--commitlint` | Use commitlint format |
|
||||||
| `--no-verify` | Skip commit message verification |
|
| `--no-verify` | Skip commit message verification |
|
||||||
| `-y, --yes` | Skip confirmation |
|
| `-t, --think` | Enable LLM thinking/reasoning mode (overrides config) |
|
||||||
|
| `-y, --yes` | Skip confirmation prompts only (generation behavior unchanged) |
|
||||||
| `--push` | Push after committing |
|
| `--push` | Push after committing |
|
||||||
| `--remote` | Specify remote repository (default: origin) |
|
| `--remote` | Specify remote repository (default: origin) |
|
||||||
|
|
||||||
@@ -248,6 +343,7 @@ quicommit config reset
|
|||||||
|--------|-------------|
|
|--------|-------------|
|
||||||
| `-n, --name` | Tag name |
|
| `-n, --name` | Tag name |
|
||||||
| `-b, --bump` | Version bump (major/minor/patch) |
|
| `-b, --bump` | Version bump (major/minor/patch) |
|
||||||
|
| `-A, --auto` | Auto-detect version from project config files (Cargo.toml, package.json, pom.xml, etc.), fall back to commit analysis (conflicts with `--bump`) |
|
||||||
| `-m, --message` | Tag message |
|
| `-m, --message` | Tag message |
|
||||||
| `-g, --generate` | AI-generate message |
|
| `-g, --generate` | AI-generate message |
|
||||||
| `-S, --sign` | GPG sign tag |
|
| `-S, --sign` | GPG sign tag |
|
||||||
@@ -256,24 +352,26 @@ quicommit config reset
|
|||||||
| `-p, --push` | Push to remote |
|
| `-p, --push` | Push to remote |
|
||||||
| `-r, --remote` | Specify remote repository (default: origin) |
|
| `-r, --remote` | Specify remote repository (default: origin) |
|
||||||
| `--dry-run` | Dry run |
|
| `--dry-run` | Dry run |
|
||||||
| `-y, --yes` | Skip confirmation |
|
| `-t, --think` | Enable LLM thinking/reasoning mode (overrides config) |
|
||||||
|
| `-y, --yes` | Skip confirmation prompts only (generation behavior unchanged) |
|
||||||
|
|
||||||
### Changelog Options
|
### Changelog Options
|
||||||
|
|
||||||
| Option | Description |
|
| Option | Description |
|
||||||
|--------|-------------|
|
|--------|-------------|
|
||||||
| `-o, --output` | Output file path |
|
| `-o, --output` | Output file path |
|
||||||
| `-v, --version` | Generate for specific version |
|
| `--version` | Generate for specific version |
|
||||||
| `-f, --from` | Generate from specific tag |
|
| `-f, --from` | Generate from specific tag (auto-detected from existing changelog if omitted) |
|
||||||
| `-t, --to` | Generate to specific ref (default: HEAD) |
|
| `-t, --to` | Generate to specific ref (default: HEAD) |
|
||||||
| `-i, --init` | Initialize new changelog file |
|
| `-i, --init` | Initialize new changelog file |
|
||||||
| `-g, --generate` | AI-generate changelog |
|
| `-g, --generate` | AI-generate changelog |
|
||||||
| `--prepend` | Prepend to existing changelog |
|
|
||||||
| `--include-hashes` | Include commit hashes |
|
| `--include-hashes` | Include commit hashes |
|
||||||
| `--include-authors` | Include authors |
|
| `--include-authors` | Include authors |
|
||||||
| `--format` | Format (keep-a-changelog, github-releases) |
|
| `--format` | Format (keep-a-changelog, github-releases) |
|
||||||
| `--dry-run` | Dry run (output to stdout) |
|
| `--dry-run` | Dry run (output to stdout) |
|
||||||
| `-y, --yes` | Skip confirmation |
|
| `--think` | Enable LLM thinking/reasoning mode (overrides config) |
|
||||||
|
| `--no-generate` | Generate deterministically from template (no AI call) |
|
||||||
|
| `-y, --yes` | Skip confirmation prompts only (generation behavior unchanged) |
|
||||||
|
|
||||||
## Configuration File
|
## Configuration File
|
||||||
|
|
||||||
@@ -312,58 +410,28 @@ use_agent = true
|
|||||||
|
|
||||||
[llm]
|
[llm]
|
||||||
provider = "ollama"
|
provider = "ollama"
|
||||||
|
model = "llama2"
|
||||||
|
# base_url = "http://localhost:11434"
|
||||||
max_tokens = 500
|
max_tokens = 500
|
||||||
temperature = 0.7
|
temperature = 0.7
|
||||||
timeout = 30
|
timeout = 30
|
||||||
|
api_key_storage = "keyring"
|
||||||
[llm.ollama]
|
thinking_enabled = false
|
||||||
url = "http://localhost:11434"
|
|
||||||
model = "llama2"
|
|
||||||
|
|
||||||
[llm.openai]
|
|
||||||
model = "gpt-4"
|
|
||||||
base_url = "https://api.openai.com/v1"
|
|
||||||
|
|
||||||
[llm.anthropic]
|
|
||||||
model = "claude-3-sonnet-20240229"
|
|
||||||
|
|
||||||
[llm.kimi]
|
|
||||||
model = "moonshot-v1-8k"
|
|
||||||
|
|
||||||
[llm.deepseek]
|
|
||||||
model = "deepseek-chat"
|
|
||||||
|
|
||||||
[llm.openrouter]
|
|
||||||
model = "openai/gpt-4"
|
|
||||||
|
|
||||||
[commit]
|
[commit]
|
||||||
format = "conventional"
|
format = "conventional"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
allow_empty = false
|
|
||||||
gpg_sign = false
|
|
||||||
max_subject_length = 100
|
|
||||||
require_scope = false
|
|
||||||
require_body = false
|
|
||||||
body_required_types = ["feat", "fix"]
|
|
||||||
|
|
||||||
[tag]
|
[tag]
|
||||||
version_prefix = "v"
|
version_prefix = "v"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
gpg_sign = false
|
|
||||||
include_changelog = true
|
|
||||||
|
|
||||||
[changelog]
|
[changelog]
|
||||||
path = "CHANGELOG.md"
|
path = "CHANGELOG.md"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
format = "keep-a-changelog"
|
|
||||||
include_hashes = false
|
|
||||||
include_authors = false
|
|
||||||
group_by_type = true
|
|
||||||
|
|
||||||
[theme]
|
[output]
|
||||||
colors = true
|
emoji = true
|
||||||
icons = true
|
|
||||||
date_format = "%Y-%m-%d"
|
|
||||||
|
|
||||||
[repo_profiles]
|
[repo_profiles]
|
||||||
"/path/to/work/project" = "work"
|
"/path/to/work/project" = "work"
|
||||||
@@ -376,15 +444,20 @@ date_format = "%Y-%m-%d"
|
|||||||
|----------|-------------|
|
|----------|-------------|
|
||||||
| `QUICOMMIT_CONFIG` | Configuration file path |
|
| `QUICOMMIT_CONFIG` | Configuration file path |
|
||||||
| `EDITOR` | Default editor |
|
| `EDITOR` | Default editor |
|
||||||
| `NO_COLOR` | Disable colored output |
|
| `NO_COLOR` | Disable colored output and emoji decorations (any value) |
|
||||||
|
| `RUST_LOG` | Overrides `-v` log filtering (e.g. `debug`) |
|
||||||
|
|
||||||
|
## Global Options
|
||||||
|
|
||||||
|
- `-v, --verbose` (repeatable): `-v` shows info logs, `-vv` debug, `-vvv` trace.
|
||||||
|
- `--no-color`: disable colors **and** emoji decorations (highest priority).
|
||||||
|
- `--emoji` / `--no-emoji`: force emoji decorations on/off, overriding the `output.emoji` config value.
|
||||||
|
- Emoji decorations can also be configured via `config set output.emoji true|false`.
|
||||||
|
|
||||||
## Troubleshooting
|
## Troubleshooting
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# View current configuration
|
# View current configuration
|
||||||
quicommit config list
|
|
||||||
|
|
||||||
# Show configuration details
|
|
||||||
quicommit config show
|
quicommit config show
|
||||||
|
|
||||||
# Edit configuration file
|
# Edit configuration file
|
||||||
@@ -396,17 +469,31 @@ quicommit config set llm.provider ollama
|
|||||||
# Get configuration value
|
# Get configuration value
|
||||||
quicommit config get llm.provider
|
quicommit config get llm.provider
|
||||||
|
|
||||||
|
# Set API key (stored in system keyring; omit the value for hidden prompt input)
|
||||||
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
|
|
||||||
|
# Delete API key from keyring
|
||||||
|
quicommit config delete-api-key
|
||||||
|
|
||||||
# Test LLM connection
|
# Test LLM connection
|
||||||
quicommit config test-llm
|
quicommit config test-llm
|
||||||
|
|
||||||
# List available models
|
# List available models
|
||||||
quicommit config list-models
|
quicommit config list-models
|
||||||
|
|
||||||
# Export configuration
|
# Check keyring availability
|
||||||
|
quicommit config check-keyring
|
||||||
|
|
||||||
|
# Show config file path
|
||||||
|
quicommit config path
|
||||||
|
|
||||||
|
# Export configuration (with optional encryption)
|
||||||
quicommit config export -o config-backup.toml
|
quicommit config export -o config-backup.toml
|
||||||
|
quicommit config export -o config-backup.enc --password
|
||||||
|
|
||||||
# Import configuration
|
# Import configuration
|
||||||
quicommit config import -i config-backup.toml
|
quicommit config import -i config-backup.toml
|
||||||
|
quicommit config import -i config-backup.enc --password
|
||||||
|
|
||||||
# Reset configuration
|
# Reset configuration
|
||||||
quicommit config reset --force
|
quicommit config reset --force
|
||||||
@@ -471,13 +558,12 @@ src/
|
|||||||
│ ├── commit.rs
|
│ ├── commit.rs
|
||||||
│ ├── tag.rs
|
│ ├── tag.rs
|
||||||
│ └── changelog.rs
|
│ └── changelog.rs
|
||||||
├── llm/ # LLM provider implementations
|
├── llm/ # LLM integration (rig-core based)
|
||||||
│ ├── ollama.rs
|
│ ├── mod.rs
|
||||||
│ ├── openai.rs
|
│ ├── prompts.rs
|
||||||
│ ├── anthropic.rs
|
│ ├── parsing.rs
|
||||||
│ ├── kimi.rs
|
│ ├── thinking.rs
|
||||||
│ ├── deepseek.rs
|
│ └── rig/ # rig provider facade
|
||||||
│ └── openrouter.rs
|
|
||||||
├── i18n/ # Internationalization support
|
├── i18n/ # Internationalization support
|
||||||
│ ├── messages.rs
|
│ ├── messages.rs
|
||||||
│ └── translator.rs
|
│ └── translator.rs
|
||||||
|
|||||||
5
build.rs
5
build.rs
@@ -1,10 +1,11 @@
|
|||||||
use std::env;
|
use std::env;
|
||||||
|
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
// Only generate completions when explicitly requested
|
// Only generate completions when explicitly requested
|
||||||
if env::var("GENERATE_COMPLETIONS").is_ok() {
|
if env::var("GENERATE_COMPLETIONS").is_ok() {
|
||||||
println!("cargo:warning=To generate shell completions, run: cargo run --bin quicommit -- completions");
|
println!(
|
||||||
|
"cargo:warning=To generate shell completions, run: cargo run --bin quicommit -- completions"
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Rerun if build.rs changes
|
// Rerun if build.rs changes
|
||||||
|
|||||||
@@ -4,6 +4,13 @@
|
|||||||
# - macOS: ~/Library/Application Support/quicommit/config.toml
|
# - macOS: ~/Library/Application Support/quicommit/config.toml
|
||||||
# - Windows: %APPDATA%\quicommit\config.toml
|
# - Windows: %APPDATA%\quicommit\config.toml
|
||||||
|
|
||||||
|
# ⚠️ IMPORTANT: Keyring Feature Update
|
||||||
|
# QuiCommit now uses system keyring to store API keys securely.
|
||||||
|
# This change may cause breaking changes to your existing configuration.
|
||||||
|
# If you encounter issues after updating, please reset your configuration:
|
||||||
|
# quicommit config reset --force
|
||||||
|
# Then reconfigure your settings using the CLI commands.
|
||||||
|
|
||||||
# Configuration version (for migration)
|
# Configuration version (for migration)
|
||||||
version = "1"
|
version = "1"
|
||||||
|
|
||||||
@@ -39,61 +46,35 @@ use_agent = true
|
|||||||
|
|
||||||
# LLM Configuration
|
# LLM Configuration
|
||||||
[llm]
|
[llm]
|
||||||
# Provider: ollama, openai, or anthropic
|
# Provider: ollama, openai, anthropic, kimi, deepseek, openrouter
|
||||||
provider = "ollama"
|
provider = "ollama"
|
||||||
|
# Model name (provider-appropriate)
|
||||||
|
model = "llama2"
|
||||||
|
# API base URL (optional, provider default will be used if not set)
|
||||||
|
# base_url = "http://localhost:11434"
|
||||||
max_tokens = 500
|
max_tokens = 500
|
||||||
temperature = 0.7
|
temperature = 0.7
|
||||||
timeout = 30
|
timeout = 30
|
||||||
|
# API key storage: keyring, config, environment
|
||||||
# Ollama settings (local LLM)
|
api_key_storage = "keyring"
|
||||||
[llm.ollama]
|
# Enable thinking/reasoning mode (deepseek, kimi, anthropic)
|
||||||
url = "http://localhost:11434"
|
thinking_enabled = false
|
||||||
model = "llama2"
|
|
||||||
|
|
||||||
# OpenAI settings
|
|
||||||
[llm.openai]
|
|
||||||
# api_key = "sk-..." # Set via: quicommit config set-openai-key
|
|
||||||
model = "gpt-4"
|
|
||||||
base_url = "https://api.openai.com/v1"
|
|
||||||
|
|
||||||
# Anthropic settings
|
|
||||||
[llm.anthropic]
|
|
||||||
# api_key = "sk-ant-..." # Set via: quicommit config set-anthropic-key
|
|
||||||
model = "claude-3-sonnet-20240229"
|
|
||||||
|
|
||||||
# Commit settings
|
# Commit settings
|
||||||
[commit]
|
[commit]
|
||||||
# Format: conventional or commitlint
|
# Format: conventional or commitlint
|
||||||
format = "conventional"
|
format = "conventional"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
allow_empty = false
|
|
||||||
gpg_sign = false
|
|
||||||
max_subject_length = 100
|
|
||||||
require_scope = false
|
|
||||||
require_body = false
|
|
||||||
body_required_types = ["feat", "fix"]
|
|
||||||
|
|
||||||
# Tag settings
|
# Tag settings
|
||||||
[tag]
|
[tag]
|
||||||
version_prefix = "v"
|
version_prefix = "v"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
gpg_sign = false
|
|
||||||
include_changelog = true
|
|
||||||
|
|
||||||
# Changelog settings
|
# Changelog settings
|
||||||
[changelog]
|
[changelog]
|
||||||
path = "CHANGELOG.md"
|
path = "CHANGELOG.md"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
format = "keep-a-changelog" # or "github-releases"
|
|
||||||
include_hashes = false
|
|
||||||
include_authors = false
|
|
||||||
group_by_type = true
|
|
||||||
|
|
||||||
# Theme settings
|
|
||||||
[theme]
|
|
||||||
colors = true
|
|
||||||
icons = true
|
|
||||||
date_format = "%Y-%m-%d"
|
|
||||||
|
|
||||||
# Repository-specific profile mappings
|
# Repository-specific profile mappings
|
||||||
# [repo_profiles]
|
# [repo_profiles]
|
||||||
|
|||||||
222
readme_zh.md
222
readme_zh.md
@@ -6,18 +6,23 @@
|
|||||||
|
|
||||||
【目前还处在早期开发阶段,依然有一些功能未完善,欢迎反馈和贡献。】
|
【目前还处在早期开发阶段,依然有一些功能未完善,欢迎反馈和贡献。】
|
||||||
|
|
||||||
|
> ⚠️ **重要提示**:QuiCommit 现在使用系统密钥环(keyring)来安全存储 API 密钥。此更改可能会对现有配置造成破坏性变更。如果在更新后遇到问题,请运行 `quicommit config reset --force` 重置配置,然后重新配置您的设置。
|
||||||
|
|
||||||

|

|
||||||

|

|
||||||
|

|
||||||
|
|
||||||
## 主要功能
|
## 主要功能
|
||||||
|
|
||||||
- **AI智能生成**:使用LLM API(Ollama本地、OpenAI、Anthropic Claude、Kimi、DeepSeek、OpenRouter)生成提交信息、标签和变更日志
|
- **AI智能生成**:使用LLM API(Ollama本地、OpenAI、Anthropic Claude、Kimi、DeepSeek、OpenRouter)生成提交信息、标签和变更日志
|
||||||
- **规范化提交**:支持Conventional Commits和commitlint格式规范
|
- **规范化提交**:支持Conventional Commits和commitlint格式规范
|
||||||
- **多配置管理**:为不同场景管理多个Git身份,支持SSH密钥和GPG签名配置
|
- **多配置管理**:为不同场景管理多个Git身份,支持SSH密钥和GPG签名配置
|
||||||
- **智能标签管理**:基于语义版本自动检测升级,AI生成标签信息
|
- **智能标签管理**:基于语义版本自动检测升级,AI生成标签信息;支持从 16+ 种项目配置文件(Cargo.toml、package.json、pom.xml、pubspec.yaml 等)自动读取版本
|
||||||
- **变更日志生成**:自动生成Keep a Changelog格式的变更日志
|
- **变更日志生成**:自动生成Keep a Changelog格式的变更日志
|
||||||
- **安全保护**:加密存储敏感数据
|
- **安全保护**:使用系统密钥环进行安全存储
|
||||||
- **交互式界面**:美观的CLI界面,支持预览和确认
|
- **交互式界面**:美观的CLI界面,支持预览和确认
|
||||||
|
- **多语言支持**:支持7种语言输出(中文、英语、日语、韩语、西班牙语、法语、德语)
|
||||||
|
- **配置导出导入**:备份和恢复配置,支持加密保护
|
||||||
|
|
||||||
## 安装
|
## 安装
|
||||||
|
|
||||||
@@ -57,7 +62,7 @@ quicommit commit
|
|||||||
# 手动提交
|
# 手动提交
|
||||||
quicommit commit --manual -t feat -m "添加新功能"
|
quicommit commit --manual -t feat -m "添加新功能"
|
||||||
|
|
||||||
# 暂存所有文件并提交
|
# 暂存所有文件并提交(自动跳过 .gitignore 匹配的文件)
|
||||||
quicommit commit -a
|
quicommit commit -a
|
||||||
|
|
||||||
# 跳过确认直接提交
|
# 跳过确认直接提交
|
||||||
@@ -66,16 +71,25 @@ quicommit commit --yes
|
|||||||
# 使用日期格式的提交信息
|
# 使用日期格式的提交信息
|
||||||
quicommit commit --date
|
quicommit commit --date
|
||||||
|
|
||||||
|
# 为本次提交启用 LLM 思考/推理模式
|
||||||
|
quicommit commit --think
|
||||||
|
|
||||||
# 提交后推送到远程
|
# 提交后推送到远程
|
||||||
quicommit commit --push
|
quicommit commit --push
|
||||||
```
|
```
|
||||||
|
|
||||||
|
在暂存更改时(自动暂存或 `--all` 模式),会自动跳过匹配 `.gitignore` 规则的文件,并清理索引中已被忽略的已跟踪文件,最后打印出被跳过的文件列表。
|
||||||
|
|
||||||
### 创建标签
|
### 创建标签
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# 自动检测版本升级
|
# 自动检测版本升级
|
||||||
quicommit tag
|
quicommit tag
|
||||||
|
|
||||||
|
# 从项目配置文件(Cargo.toml、package.json、pom.xml 等)自动检测版本,
|
||||||
|
# 回退到基于 commit 的分析
|
||||||
|
quicommit tag --auto
|
||||||
|
|
||||||
# 指定版本升级类型
|
# 指定版本升级类型
|
||||||
quicommit tag --bump minor
|
quicommit tag --bump minor
|
||||||
|
|
||||||
@@ -92,11 +106,14 @@ quicommit tag --push
|
|||||||
### 生成变更日志
|
### 生成变更日志
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# 生成未发布变更的变更日志
|
# 生成未发布变更的变更日志(未指定 --from 时自动从现有 changelog 的最高版本检测)
|
||||||
quicommit changelog
|
quicommit changelog
|
||||||
|
|
||||||
# 为特定版本生成
|
# 为特定版本生成
|
||||||
quicommit changelog -v 1.0.0
|
quicommit changelog --version 1.0.0
|
||||||
|
|
||||||
|
# 从指定标签生成
|
||||||
|
quicommit changelog --from v0.9.0
|
||||||
|
|
||||||
# AI生成变更日志
|
# AI生成变更日志
|
||||||
quicommit changelog --generate
|
quicommit changelog --generate
|
||||||
@@ -154,35 +171,91 @@ quicommit profile stats
|
|||||||
quicommit profile token
|
quicommit profile token
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Git 凭据助手(Credential Helper)
|
||||||
|
|
||||||
|
QuiCommit 可以作为 Git 凭据助手,通过系统密钥环安全地存储和读取个人访问令牌(PAT)。
|
||||||
|
`credential` 命令在 `--help` 中是隐藏的,因为它由 Git 自动调用,无需用户手动执行。
|
||||||
|
|
||||||
|
#### 配置方法
|
||||||
|
|
||||||
|
将 QuiCommit 注册为 Git 凭据助手:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 使用默认的 QuiCommit 配置
|
||||||
|
git config --global credential.helper quicommit
|
||||||
|
|
||||||
|
# 或指定自定义配置文件路径
|
||||||
|
git config --global credential.helper "quicommit --config /path/to/config.toml"
|
||||||
|
```
|
||||||
|
|
||||||
|
也可以仅对特定主机启用:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
git config --global credential.https://github.com.helper quicommit
|
||||||
|
```
|
||||||
|
|
||||||
|
#### 工作原理
|
||||||
|
|
||||||
|
当 Git 需要凭据(例如推送到远程仓库)时,会按照
|
||||||
|
[gitcredentials 协议](https://git-scm.com/docs/gitcredentials) 调用助手:
|
||||||
|
|
||||||
|
1. **`get`** — Git 向 QuiCommit 请求与目标主机匹配的已存储 PAT。
|
||||||
|
QuiCommit 在所有已配置的 profile 中搜索密钥环,找到则返回 PAT(及用户名)。
|
||||||
|
2. **`store`** — 认证成功后(例如你在 Git 提示中输入了 PAT),Git 要求
|
||||||
|
QuiCommit 保存该凭据。PAT 将存入系统密钥环,并与匹配的 profile 绑定。
|
||||||
|
3. **`erase`** — Git 要求 QuiCommit 删除指定主机的已存储 PAT。
|
||||||
|
|
||||||
|
主机名会被映射为规范化的服务名(`github.com` → `github`、`gitlab.com` →
|
||||||
|
`gitlab`、`bitbucket.org` → `bitbucket` 等),未知主机则原样使用。
|
||||||
|
|
||||||
|
#### 删除已存储的凭据
|
||||||
|
|
||||||
|
可以通过 Git 内置机制删除已存储的 PAT:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
echo "protocol=https
|
||||||
|
host=github.com" | git credential reject
|
||||||
|
```
|
||||||
|
|
||||||
|
也可以通过 profile 令牌管理命令删除:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
quicommit profile token
|
||||||
|
```
|
||||||
|
|
||||||
### LLM配置
|
### LLM配置
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# 配置Ollama(本地)
|
# 配置Ollama(本地)
|
||||||
quicommit config set-llm ollama
|
quicommit config set-llm ollama
|
||||||
quicommit config set-ollama --url http://localhost:11434 --model llama2
|
quicommit config set-llm ollama --base-url http://localhost:11434 --model llama2
|
||||||
|
|
||||||
|
> ⚠️ **安全提示**:把 API 密钥作为命令行参数传入会留在 shell history 与进程列表中。
|
||||||
|
> 建议直接运行 `quicommit config set-api-key`(或 `config set-llm`)不携带密钥,
|
||||||
|
> 以隐藏方式输入。
|
||||||
|
|
||||||
# 配置OpenAI
|
# 配置OpenAI
|
||||||
quicommit config set-llm openai
|
quicommit config set-llm openai
|
||||||
quicommit config set-openai-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
|
|
||||||
# 配置Anthropic Claude
|
# 配置Anthropic Claude
|
||||||
quicommit config set-llm anthropic
|
quicommit config set-llm anthropic
|
||||||
quicommit config set-anthropic-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
|
|
||||||
# 配置Kimi
|
# 配置Kimi
|
||||||
quicommit config set-llm kimi
|
quicommit config set-llm kimi
|
||||||
quicommit config set-kimi-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
quicommit config set-kimi --base-url https://api.moonshot.cn/v1 --model moonshot-v1-8k
|
quicommit config set-llm kimi --base-url https://api.moonshot.cn/v1 --model moonshot-v1-8k
|
||||||
|
|
||||||
# 配置DeepSeek
|
# 配置DeepSeek
|
||||||
quicommit config set-llm deepseek
|
quicommit config set-llm deepseek
|
||||||
quicommit config set-deepseek-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
quicommit config set-deepseek --base-url https://api.deepseek.com/v1 --model deepseek-chat
|
quicommit config set-llm deepseek --base-url https://api.deepseek.com/v1 --model deepseek-chat
|
||||||
|
|
||||||
# 配置OpenRouter
|
# 配置OpenRouter
|
||||||
quicommit config set-llm openrouter
|
quicommit config set-llm openrouter
|
||||||
quicommit config set-openrouter-key YOUR_API_KEY
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
quicommit config set-openrouter --base-url https://openrouter.ai/api/v1 --model openai/gpt-4
|
quicommit config set-llm openrouter --base-url https://openrouter.ai/api/v1 --model openai/gpt-4
|
||||||
|
|
||||||
# 设置提交格式
|
# 设置提交格式
|
||||||
quicommit config set-commit-format conventional
|
quicommit config set-commit-format conventional
|
||||||
@@ -193,7 +266,7 @@ quicommit config set-version-prefix v
|
|||||||
# 设置变更日志路径
|
# 设置变更日志路径
|
||||||
quicommit config set-changelog-path CHANGELOG.md
|
quicommit config set-changelog-path CHANGELOG.md
|
||||||
|
|
||||||
# 设置输出语言
|
# 设置输出语言(zh, en, ja, ko, es, fr, de)
|
||||||
quicommit config set-language zh
|
quicommit config set-language zh
|
||||||
|
|
||||||
# 设置保持提交类型为英文
|
# 设置保持提交类型为英文
|
||||||
@@ -205,8 +278,22 @@ quicommit config set-keep-changelog-types-english true
|
|||||||
# 测试LLM连接
|
# 测试LLM连接
|
||||||
quicommit config test-llm
|
quicommit config test-llm
|
||||||
|
|
||||||
|
# 检查密钥环可用性
|
||||||
|
quicommit config check-keyring
|
||||||
|
|
||||||
|
# 显示配置文件路径
|
||||||
|
quicommit config path
|
||||||
|
|
||||||
|
# 导出配置(支持加密)
|
||||||
|
quicommit config export -o config-backup.toml
|
||||||
|
quicommit config export -o config-backup.enc --password
|
||||||
|
|
||||||
|
# 导入配置
|
||||||
|
quicommit config import -i config-backup.toml
|
||||||
|
quicommit config import -i config-backup.enc --password
|
||||||
|
|
||||||
# 重置配置为默认值
|
# 重置配置为默认值
|
||||||
quicommit config reset
|
quicommit config reset --force
|
||||||
```
|
```
|
||||||
|
|
||||||
## 命令参考
|
## 命令参考
|
||||||
@@ -219,6 +306,7 @@ quicommit config reset
|
|||||||
| `quicommit changelog` | `cl` | 生成变更日志 |
|
| `quicommit changelog` | `cl` | 生成变更日志 |
|
||||||
| `quicommit profile` | `p` | 管理Git配置 |
|
| `quicommit profile` | `p` | 管理Git配置 |
|
||||||
| `quicommit config` | `cfg` | 管理应用配置 |
|
| `quicommit config` | `cfg` | 管理应用配置 |
|
||||||
|
| `quicommit credential` | — | Git凭据助手(隐藏,由Git调用) |
|
||||||
|
|
||||||
### commit命令选项
|
### commit命令选项
|
||||||
|
|
||||||
@@ -231,14 +319,15 @@ quicommit config reset
|
|||||||
| `-b, --breaking` | 标记为破坏性变更 |
|
| `-b, --breaking` | 标记为破坏性变更 |
|
||||||
| `-d, --date` | 使用日期格式的提交信息 |
|
| `-d, --date` | 使用日期格式的提交信息 |
|
||||||
| `--manual` | 手动输入,跳过AI生成 |
|
| `--manual` | 手动输入,跳过AI生成 |
|
||||||
| `-a, --all` | 暂存所有更改 |
|
| `-a, --all` | 暂存所有更改(自动跳过 `.gitignore` 匹配的文件) |
|
||||||
| `-S, --sign` | GPG签名提交 |
|
| `-S, --sign` | GPG签名提交 |
|
||||||
| `--amend` | 修改上一次提交 |
|
| `--amend` | 修改上一次提交 |
|
||||||
| `--dry-run` | 试运行,不实际提交 |
|
| `--dry-run` | 试运行,不实际提交 |
|
||||||
| `--conventional` | 使用Conventional Commits格式 |
|
| `--conventional` | 使用Conventional Commits格式 |
|
||||||
| `--commitlint` | 使用commitlint格式 |
|
| `--commitlint` | 使用commitlint格式 |
|
||||||
| `--no-verify` | 不验证提交信息 |
|
| `--no-verify` | 不验证提交信息 |
|
||||||
| `-y, --yes` | 跳过确认提示 |
|
| `-t, --think` | 启用 LLM 思考/推理模式(覆盖配置) |
|
||||||
|
| `-y, --yes` | 仅跳过确认提示(生成行为不变) |
|
||||||
| `--push` | 提交后推送到远程 |
|
| `--push` | 提交后推送到远程 |
|
||||||
| `--remote` | 指定远程仓库(默认:origin) |
|
| `--remote` | 指定远程仓库(默认:origin) |
|
||||||
|
|
||||||
@@ -248,6 +337,7 @@ quicommit config reset
|
|||||||
|------|------|
|
|------|------|
|
||||||
| `-n, --name` | 标签名称 |
|
| `-n, --name` | 标签名称 |
|
||||||
| `-b, --bump` | 版本升级类型(major/minor/patch) |
|
| `-b, --bump` | 版本升级类型(major/minor/patch) |
|
||||||
|
| `-A, --auto` | 从项目配置文件(Cargo.toml、package.json、pom.xml 等 16+ 种)自动检测版本,回退到基于 commit 的分析(与 `--bump` 互斥) |
|
||||||
| `-m, --message` | 标签信息 |
|
| `-m, --message` | 标签信息 |
|
||||||
| `-g, --generate` | AI生成标签信息 |
|
| `-g, --generate` | AI生成标签信息 |
|
||||||
| `-S, --sign` | GPG签名标签 |
|
| `-S, --sign` | GPG签名标签 |
|
||||||
@@ -256,24 +346,26 @@ quicommit config reset
|
|||||||
| `-p, --push` | 推送到远程 |
|
| `-p, --push` | 推送到远程 |
|
||||||
| `-r, --remote` | 指定远程仓库(默认:origin) |
|
| `-r, --remote` | 指定远程仓库(默认:origin) |
|
||||||
| `--dry-run` | 试运行 |
|
| `--dry-run` | 试运行 |
|
||||||
| `-y, --yes` | 跳过确认提示 |
|
| `-t, --think` | 启用 LLM 思考/推理模式(覆盖配置) |
|
||||||
|
| `-y, --yes` | 仅跳过确认提示(生成行为不变) |
|
||||||
|
|
||||||
### changelog命令选项
|
### changelog命令选项
|
||||||
|
|
||||||
| 选项 | 说明 |
|
| 选项 | 说明 |
|
||||||
|------|------|
|
|------|------|
|
||||||
| `-o, --output` | 输出文件路径 |
|
| `-o, --output` | 输出文件路径 |
|
||||||
| `-v, --version` | 为特定版本生成 |
|
| `--version` | 为特定版本生成 |
|
||||||
| `-f, --from` | 从指定标签生成 |
|
| `-f, --from` | 从指定标签生成(未指定时自动从现有 changelog 的最高版本检测) |
|
||||||
| `-t, --to` | 生成到指定引用(默认:HEAD) |
|
| `-t, --to` | 生成到指定引用(默认:HEAD) |
|
||||||
| `-i, --init` | 初始化新的变更日志文件 |
|
| `-i, --init` | 初始化新的变更日志文件 |
|
||||||
| `-g, --generate` | AI生成变更日志 |
|
| `-g, --generate` | AI生成变更日志 |
|
||||||
| `--prepend` | 添加到现有变更日志开头 |
|
|
||||||
| `--include-hashes` | 包含提交哈希 |
|
| `--include-hashes` | 包含提交哈希 |
|
||||||
| `--include-authors` | 包含作者信息 |
|
| `--include-authors` | 包含作者信息 |
|
||||||
| `--format` | 格式(keep-a-changelog、github-releases) |
|
| `--format` | 格式(keep-a-changelog、github-releases) |
|
||||||
| `--dry-run` | 试运行(输出到stdout) |
|
| `--dry-run` | 试运行(输出到stdout) |
|
||||||
| `-y, --yes` | 跳过确认提示 |
|
| `--think` | 启用 LLM 思考/推理模式(覆盖配置) |
|
||||||
|
| `--no-generate` | 使用模板确定性生成(不调用 AI) |
|
||||||
|
| `-y, --yes` | 仅跳过确认提示(生成行为不变) |
|
||||||
|
|
||||||
## 配置文件
|
## 配置文件
|
||||||
|
|
||||||
@@ -312,58 +404,28 @@ use_agent = true
|
|||||||
|
|
||||||
[llm]
|
[llm]
|
||||||
provider = "ollama"
|
provider = "ollama"
|
||||||
|
model = "llama2"
|
||||||
|
# base_url = "http://localhost:11434"
|
||||||
max_tokens = 500
|
max_tokens = 500
|
||||||
temperature = 0.7
|
temperature = 0.7
|
||||||
timeout = 30
|
timeout = 30
|
||||||
|
api_key_storage = "keyring"
|
||||||
[llm.ollama]
|
thinking_enabled = false
|
||||||
url = "http://localhost:11434"
|
|
||||||
model = "llama2"
|
|
||||||
|
|
||||||
[llm.openai]
|
|
||||||
model = "gpt-4"
|
|
||||||
base_url = "https://api.openai.com/v1"
|
|
||||||
|
|
||||||
[llm.anthropic]
|
|
||||||
model = "claude-3-sonnet-20240229"
|
|
||||||
|
|
||||||
[llm.kimi]
|
|
||||||
model = "moonshot-v1-8k"
|
|
||||||
|
|
||||||
[llm.deepseek]
|
|
||||||
model = "deepseek-chat"
|
|
||||||
|
|
||||||
[llm.openrouter]
|
|
||||||
model = "openai/gpt-4"
|
|
||||||
|
|
||||||
[commit]
|
[commit]
|
||||||
format = "conventional"
|
format = "conventional"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
allow_empty = false
|
|
||||||
gpg_sign = false
|
|
||||||
max_subject_length = 100
|
|
||||||
require_scope = false
|
|
||||||
require_body = false
|
|
||||||
body_required_types = ["feat", "fix"]
|
|
||||||
|
|
||||||
[tag]
|
[tag]
|
||||||
version_prefix = "v"
|
version_prefix = "v"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
gpg_sign = false
|
|
||||||
include_changelog = true
|
|
||||||
|
|
||||||
[changelog]
|
[changelog]
|
||||||
path = "CHANGELOG.md"
|
path = "CHANGELOG.md"
|
||||||
auto_generate = true
|
auto_generate = true
|
||||||
format = "keep-a-changelog"
|
|
||||||
include_hashes = false
|
|
||||||
include_authors = false
|
|
||||||
group_by_type = true
|
|
||||||
|
|
||||||
[theme]
|
[output]
|
||||||
colors = true
|
emoji = true
|
||||||
icons = true
|
|
||||||
date_format = "%Y-%m-%d"
|
|
||||||
|
|
||||||
[repo_profiles]
|
[repo_profiles]
|
||||||
"/path/to/work/project" = "work"
|
"/path/to/work/project" = "work"
|
||||||
@@ -376,15 +438,20 @@ date_format = "%Y-%m-%d"
|
|||||||
|--------|------|
|
|--------|------|
|
||||||
| `QUICOMMIT_CONFIG` | 配置文件路径 |
|
| `QUICOMMIT_CONFIG` | 配置文件路径 |
|
||||||
| `EDITOR` | 默认编辑器 |
|
| `EDITOR` | 默认编辑器 |
|
||||||
| `NO_COLOR` | 禁用彩色输出 |
|
| `NO_COLOR` | 禁用彩色与 Emoji 装饰输出(任意值即生效) |
|
||||||
|
| `RUST_LOG` | 覆盖 `-v` 的日志过滤(例如 `debug`) |
|
||||||
|
|
||||||
|
## 全局选项
|
||||||
|
|
||||||
|
- `-v, --verbose`(可叠加):`-v` 显示 info 日志、`-vv` debug、`-vvv` trace。
|
||||||
|
- `--no-color`:禁用彩色**与** Emoji 装饰(最高优先级)。
|
||||||
|
- `--emoji` / `--no-emoji`:强制开启/关闭 Emoji 装饰,覆盖 `output.emoji` 配置。
|
||||||
|
- Emoji 装饰也可通过 `quicommit config set output.emoji true|false` 配置。
|
||||||
|
|
||||||
## 故障排除
|
## 故障排除
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# 查看当前配置
|
# 查看当前配置
|
||||||
quicommit config list
|
|
||||||
|
|
||||||
# 显示配置详情
|
|
||||||
quicommit config show
|
quicommit config show
|
||||||
|
|
||||||
# 编辑配置文件
|
# 编辑配置文件
|
||||||
@@ -396,17 +463,31 @@ quicommit config set llm.provider ollama
|
|||||||
# 获取配置值
|
# 获取配置值
|
||||||
quicommit config get llm.provider
|
quicommit config get llm.provider
|
||||||
|
|
||||||
|
# 设置API密钥(存储在系统密钥环中;省略密钥值可通过隐藏提示输入)
|
||||||
|
quicommit config set-api-key YOUR_API_KEY
|
||||||
|
|
||||||
|
# 从密钥环删除API密钥
|
||||||
|
quicommit config delete-api-key
|
||||||
|
|
||||||
# 测试LLM连接
|
# 测试LLM连接
|
||||||
quicommit config test-llm
|
quicommit config test-llm
|
||||||
|
|
||||||
# 列出可用模型
|
# 列出可用模型
|
||||||
quicommit config list-models
|
quicommit config list-models
|
||||||
|
|
||||||
# 导出配置
|
# 检查密钥环可用性
|
||||||
|
quicommit config check-keyring
|
||||||
|
|
||||||
|
# 显示配置文件路径
|
||||||
|
quicommit config path
|
||||||
|
|
||||||
|
# 导出配置(支持加密)
|
||||||
quicommit config export -o config-backup.toml
|
quicommit config export -o config-backup.toml
|
||||||
|
quicommit config export -o config-backup.enc --password
|
||||||
|
|
||||||
# 导入配置
|
# 导入配置
|
||||||
quicommit config import -i config-backup.toml
|
quicommit config import -i config-backup.toml
|
||||||
|
quicommit config import -i config-backup.enc --password
|
||||||
|
|
||||||
# 重置配置
|
# 重置配置
|
||||||
quicommit config reset --force
|
quicommit config reset --force
|
||||||
@@ -471,13 +552,12 @@ src/
|
|||||||
│ ├── commit.rs
|
│ ├── commit.rs
|
||||||
│ ├── tag.rs
|
│ ├── tag.rs
|
||||||
│ └── changelog.rs
|
│ └── changelog.rs
|
||||||
├── llm/ # LLM提供商实现
|
├── llm/ # LLM 集成(基于 rig-core)
|
||||||
│ ├── ollama.rs
|
│ ├── mod.rs
|
||||||
│ ├── openai.rs
|
│ ├── prompts.rs
|
||||||
│ ├── anthropic.rs
|
│ ├── parsing.rs
|
||||||
│ ├── kimi.rs
|
│ ├── thinking.rs
|
||||||
│ ├── deepseek.rs
|
│ └── rig/ # rig provider 门面
|
||||||
│ └── openrouter.rs
|
|
||||||
├── i18n/ # 国际化支持
|
├── i18n/ # 国际化支持
|
||||||
│ ├── messages.rs
|
│ ├── messages.rs
|
||||||
│ └── translator.rs
|
│ └── translator.rs
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
use anyhow::{bail, Result};
|
use anyhow::{Result, bail};
|
||||||
use chrono::Utc;
|
use chrono::Utc;
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use colored::Colorize;
|
use colored::Colorize;
|
||||||
@@ -7,19 +7,22 @@ use std::path::PathBuf;
|
|||||||
|
|
||||||
use crate::config::{Language, manager::ConfigManager};
|
use crate::config::{Language, manager::ConfigManager};
|
||||||
use crate::generator::ContentGenerator;
|
use crate::generator::ContentGenerator;
|
||||||
|
use crate::git::GitRepo;
|
||||||
use crate::git::find_repo;
|
use crate::git::find_repo;
|
||||||
use crate::git::{changelog::*, CommitInfo};
|
use crate::git::{CommitInfo, changelog::*};
|
||||||
use crate::i18n::{Messages, translate_changelog_category};
|
use crate::i18n::{Messages, translate_changelog_category};
|
||||||
|
use crate::utils::{print_progress, print_success, print_warning};
|
||||||
|
|
||||||
/// Generate changelog
|
/// Generate changelog
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
|
#[command(disable_version_flag = true, disable_help_flag = false)]
|
||||||
pub struct ChangelogCommand {
|
pub struct ChangelogCommand {
|
||||||
/// Output file path
|
/// Output file path
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
output: Option<PathBuf>,
|
output: Option<PathBuf>,
|
||||||
|
|
||||||
/// Version to generate changelog for
|
/// Version to generate changelog for
|
||||||
#[arg(short, long)]
|
#[arg(long)]
|
||||||
version: Option<String>,
|
version: Option<String>,
|
||||||
|
|
||||||
/// Generate from specific tag
|
/// Generate from specific tag
|
||||||
@@ -38,10 +41,6 @@ pub struct ChangelogCommand {
|
|||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
generate: bool,
|
generate: bool,
|
||||||
|
|
||||||
/// Prepend to existing changelog
|
|
||||||
#[arg(short, long)]
|
|
||||||
prepend: bool,
|
|
||||||
|
|
||||||
/// Include commit hashes
|
/// Include commit hashes
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
include_hashes: bool,
|
include_hashes: bool,
|
||||||
@@ -51,40 +50,54 @@ pub struct ChangelogCommand {
|
|||||||
include_authors: bool,
|
include_authors: bool,
|
||||||
|
|
||||||
/// Format (keep-a-changelog, github-releases)
|
/// Format (keep-a-changelog, github-releases)
|
||||||
#[arg(short, long)]
|
#[arg(long)]
|
||||||
format: Option<String>,
|
format: Option<String>,
|
||||||
|
|
||||||
/// Dry run (output to stdout)
|
/// Dry run (output to stdout)
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
dry_run: bool,
|
dry_run: bool,
|
||||||
|
|
||||||
/// Skip interactive prompts
|
/// Enable thinking mode for this changelog (override config)
|
||||||
|
#[arg(long)]
|
||||||
|
think: bool,
|
||||||
|
|
||||||
|
/// Generate deterministically from template (no AI call)
|
||||||
|
#[arg(long, conflicts_with = "generate")]
|
||||||
|
no_generate: bool,
|
||||||
|
|
||||||
|
/// Skip interactive prompts only (generation behavior unchanged)
|
||||||
#[arg(short = 'y', long)]
|
#[arg(short = 'y', long)]
|
||||||
yes: bool,
|
yes: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ChangelogCommand {
|
impl ChangelogCommand {
|
||||||
pub async fn execute(&self) -> Result<()> {
|
pub async fn execute(&self, config_path: Option<PathBuf>) -> Result<()> {
|
||||||
let repo = find_repo(std::env::current_dir()?.as_path())?;
|
let repo = find_repo(std::env::current_dir()?.as_path())?;
|
||||||
let manager = ConfigManager::new()?;
|
let manager = if let Some(ref path) = config_path {
|
||||||
|
ConfigManager::with_path(path)?
|
||||||
|
} else {
|
||||||
|
ConfigManager::new()?
|
||||||
|
};
|
||||||
let config = manager.config();
|
let config = manager.config();
|
||||||
let language = manager.get_language().unwrap_or(Language::English);
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
let messages = Messages::new(language);
|
let messages = Messages::new(language);
|
||||||
|
|
||||||
// Initialize changelog if requested
|
// Initialize changelog if requested
|
||||||
if self.init {
|
if self.init {
|
||||||
let path = self.output.as_ref()
|
let path = self
|
||||||
.map(|p| p.clone())
|
.output
|
||||||
|
.clone()
|
||||||
.unwrap_or_else(|| PathBuf::from(&config.changelog.path));
|
.unwrap_or_else(|| PathBuf::from(&config.changelog.path));
|
||||||
|
|
||||||
init_changelog(&path)?;
|
init_changelog(&path)?;
|
||||||
println!("{}", messages.initialized_changelog(&format!("{:?}", path)));
|
print_success(&messages.initialized_changelog(&path.display().to_string()));
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
// Determine output path
|
// Determine output path
|
||||||
let output_path = self.output.as_ref()
|
let output_path = self
|
||||||
.map(|p| p.clone())
|
.output
|
||||||
|
.clone()
|
||||||
.unwrap_or_else(|| PathBuf::from(&config.changelog.path));
|
.unwrap_or_else(|| PathBuf::from(&config.changelog.path));
|
||||||
|
|
||||||
// Determine format
|
// Determine format
|
||||||
@@ -93,7 +106,7 @@ impl ChangelogCommand {
|
|||||||
Some("keep") | Some("keep-a-changelog") => ChangelogFormat::KeepAChangelog,
|
Some("keep") | Some("keep-a-changelog") => ChangelogFormat::KeepAChangelog,
|
||||||
Some("custom") => ChangelogFormat::Custom,
|
Some("custom") => ChangelogFormat::Custom,
|
||||||
None => ChangelogFormat::KeepAChangelog,
|
None => ChangelogFormat::KeepAChangelog,
|
||||||
Some(f) => bail!("Unknown format: {}. Use: keep-a-changelog, github-releases", f),
|
Some(f) => bail!("{}", messages.unknown_changelog_format(f)),
|
||||||
};
|
};
|
||||||
|
|
||||||
// Get version
|
// Get version
|
||||||
@@ -109,17 +122,20 @@ impl ChangelogCommand {
|
|||||||
};
|
};
|
||||||
|
|
||||||
// Get commits
|
// Get commits
|
||||||
println!("{}", messages.fetching_commits());
|
print_progress(messages.fetching_commits());
|
||||||
let commits = generate_from_history(&repo, self.from.as_deref(), Some(&self.to))?;
|
|
||||||
|
// Determine from_tag: use explicit --from, or auto-detect from changelog
|
||||||
|
let from_tag = self.resolve_from_tag(&repo, &output_path, &messages);
|
||||||
|
let commits = generate_from_history(&repo, from_tag.as_deref(), Some(&self.to))?;
|
||||||
|
|
||||||
if commits.is_empty() {
|
if commits.is_empty() {
|
||||||
bail!("{}", messages.no_commits_found());
|
bail!("{}", messages.no_commits_found());
|
||||||
}
|
}
|
||||||
|
|
||||||
println!("{}", messages.found_commits(commits.len()));
|
print_success(&messages.found_commits(commits.len()));
|
||||||
|
|
||||||
// Generate changelog
|
// Generate changelog
|
||||||
let changelog = if self.generate || (config.changelog.auto_generate && !self.yes) {
|
let changelog = if self.generate || (config.changelog.auto_generate && !self.no_generate) {
|
||||||
self.generate_with_ai(&version, &commits, &messages).await?
|
self.generate_with_ai(&version, &commits, &messages).await?
|
||||||
} else {
|
} else {
|
||||||
self.generate_with_template(format, &version, &commits, language)?
|
self.generate_with_template(format, &version, &commits, language)?
|
||||||
@@ -147,30 +163,61 @@ impl ChangelogCommand {
|
|||||||
println!("{}", "─".repeat(60));
|
println!("{}", "─".repeat(60));
|
||||||
|
|
||||||
let confirm = Confirm::new()
|
let confirm = Confirm::new()
|
||||||
.with_prompt(&messages.write_to_file(&format!("{:?}", output_path)))
|
.with_prompt(messages.write_to_file(&output_path.display().to_string()))
|
||||||
.default(true)
|
.default(true)
|
||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
if !confirm {
|
if !confirm {
|
||||||
println!("{}", messages.cancelled().yellow());
|
print_warning(messages.cancelled());
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Write to file
|
// Write to file (always prepend new entry before existing versions)
|
||||||
if self.prepend && output_path.exists() {
|
if output_path.exists() {
|
||||||
let existing = std::fs::read_to_string(&output_path)?;
|
let existing = std::fs::read_to_string(&output_path)?;
|
||||||
let new_content = format!("{}\n{}", changelog, existing);
|
let new_content = if existing.is_empty() {
|
||||||
|
format!("{}{}", CHANGELOG_HEADER, changelog)
|
||||||
|
} else {
|
||||||
|
insert_changelog_entry(&existing, &changelog)
|
||||||
|
};
|
||||||
std::fs::write(&output_path, new_content)?;
|
std::fs::write(&output_path, new_content)?;
|
||||||
} else {
|
} else {
|
||||||
std::fs::write(&output_path, changelog)?;
|
let content = format!("{}{}", CHANGELOG_HEADER, changelog);
|
||||||
|
std::fs::write(&output_path, content)?;
|
||||||
}
|
}
|
||||||
|
|
||||||
println!("{} {:?}", messages.changelog_written(), output_path);
|
print_success(&format!(
|
||||||
|
"{} {}",
|
||||||
|
messages.changelog_written(),
|
||||||
|
output_path.display()
|
||||||
|
));
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn resolve_from_tag(
|
||||||
|
&self,
|
||||||
|
repo: &GitRepo,
|
||||||
|
output_path: &PathBuf,
|
||||||
|
messages: &Messages,
|
||||||
|
) -> Option<String> {
|
||||||
|
// Explicit --from always wins
|
||||||
|
if self.from.is_some() {
|
||||||
|
return self.from.clone();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Auto-detect: find highest version already in changelog
|
||||||
|
let existing = std::fs::read_to_string(output_path).ok()?;
|
||||||
|
let versions = parse_changelog_versions(&existing);
|
||||||
|
let highest = versions.first()?;
|
||||||
|
|
||||||
|
// Match highest version to a git tag
|
||||||
|
let tag = repo.find_tag_by_version(highest)?;
|
||||||
|
println!(" {}: {}", messages.version(), tag.name);
|
||||||
|
Some(tag.name)
|
||||||
|
}
|
||||||
|
|
||||||
async fn generate_with_ai(
|
async fn generate_with_ai(
|
||||||
&self,
|
&self,
|
||||||
version: &str,
|
version: &str,
|
||||||
@@ -178,13 +225,15 @@ impl ChangelogCommand {
|
|||||||
messages: &Messages,
|
messages: &Messages,
|
||||||
) -> Result<String> {
|
) -> Result<String> {
|
||||||
let manager = ConfigManager::new()?;
|
let manager = ConfigManager::new()?;
|
||||||
let config = manager.config();
|
|
||||||
let language = manager.get_language().unwrap_or(Language::English);
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
|
|
||||||
println!("{}", messages.ai_generating_changelog());
|
let generator = ContentGenerator::new_with_think(&manager, self.think, None).await?;
|
||||||
|
let spinner = crate::utils::Spinner::start(messages.ai_generating_changelog());
|
||||||
let generator = ContentGenerator::new(&config.llm).await?;
|
let result = generator
|
||||||
generator.generate_changelog_entry(version, commits, language).await
|
.generate_changelog_entry(version, commits, language)
|
||||||
|
.await;
|
||||||
|
spinner.finish_clear();
|
||||||
|
result
|
||||||
}
|
}
|
||||||
|
|
||||||
fn generate_with_template(
|
fn generate_with_template(
|
||||||
@@ -212,12 +261,13 @@ impl ChangelogCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn translate_changelog_categories(&self, changelog: &str, language: Language) -> String {
|
fn translate_changelog_categories(&self, changelog: &str, language: Language) -> String {
|
||||||
let translated = changelog
|
changelog
|
||||||
.lines()
|
.lines()
|
||||||
.map(|line| {
|
.map(|line| {
|
||||||
if line.starts_with("## ") || line.starts_with("### ") {
|
if line.starts_with("## ") || line.starts_with("### ") {
|
||||||
let category = line.trim_start_matches("## ").trim_start_matches("### ");
|
let category = line.trim_start_matches("## ").trim_start_matches("### ");
|
||||||
let translated_category = translate_changelog_category(category, language, false);
|
let translated_category =
|
||||||
|
translate_changelog_category(category, language, false);
|
||||||
if line.starts_with("## ") {
|
if line.starts_with("## ") {
|
||||||
format!("## {}", translated_category)
|
format!("## {}", translated_category)
|
||||||
} else {
|
} else {
|
||||||
@@ -228,7 +278,6 @@ impl ChangelogCommand {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
.collect::<Vec<_>>()
|
.collect::<Vec<_>>()
|
||||||
.join("\n");
|
.join("\n")
|
||||||
translated
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,15 +1,17 @@
|
|||||||
use anyhow::{bail, Context, Result};
|
use anyhow::{Context, Result, bail};
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use colored::Colorize;
|
use colored::Colorize;
|
||||||
use dialoguer::{Confirm, Input, Select};
|
use dialoguer::{Confirm, Input, Select};
|
||||||
|
use std::path::PathBuf;
|
||||||
|
|
||||||
use crate::config::{Language, manager::ConfigManager};
|
|
||||||
use crate::config::CommitFormat;
|
use crate::config::CommitFormat;
|
||||||
|
use crate::config::{Language, manager::ConfigManager};
|
||||||
use crate::generator::ContentGenerator;
|
use crate::generator::ContentGenerator;
|
||||||
use crate::git::{find_repo, GitRepo};
|
|
||||||
use crate::git::commit::{CommitBuilder, create_date_commit_message};
|
use crate::git::commit::{CommitBuilder, create_date_commit_message};
|
||||||
|
use crate::git::{GitRepo, find_repo};
|
||||||
use crate::i18n::Messages;
|
use crate::i18n::Messages;
|
||||||
use crate::utils::validators::get_commit_types;
|
use crate::utils::validators::get_commit_types;
|
||||||
|
use crate::utils::{print_progress, print_success, print_warning};
|
||||||
|
|
||||||
/// Generate and execute conventional commits
|
/// Generate and execute conventional commits
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
@@ -70,7 +72,11 @@ pub struct CommitCommand {
|
|||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
no_verify: bool,
|
no_verify: bool,
|
||||||
|
|
||||||
/// Skip interactive prompts
|
/// Enable thinking mode for this commit (override config)
|
||||||
|
#[arg(short = 't', long)]
|
||||||
|
think: bool,
|
||||||
|
|
||||||
|
/// Skip interactive prompts only (generation behavior unchanged)
|
||||||
#[arg(short = 'y', long)]
|
#[arg(short = 'y', long)]
|
||||||
yes: bool,
|
yes: bool,
|
||||||
|
|
||||||
@@ -84,12 +90,16 @@ pub struct CommitCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl CommitCommand {
|
impl CommitCommand {
|
||||||
pub async fn execute(&self) -> Result<()> {
|
pub async fn execute(&self, config_path: Option<PathBuf>) -> Result<()> {
|
||||||
// Find git repository
|
// Find git repository
|
||||||
let repo = find_repo(std::env::current_dir()?.as_path())?;
|
let repo = find_repo(std::env::current_dir()?.as_path())?;
|
||||||
|
|
||||||
// Load configuration
|
// Load configuration
|
||||||
let manager = ConfigManager::new()?;
|
let manager = if let Some(ref path) = config_path {
|
||||||
|
ConfigManager::with_path(path)?
|
||||||
|
} else {
|
||||||
|
ConfigManager::new()?
|
||||||
|
};
|
||||||
let config = manager.config();
|
let config = manager.config();
|
||||||
let language = manager.get_language().unwrap_or(Language::English);
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
let messages = Messages::new(language);
|
let messages = Messages::new(language);
|
||||||
@@ -112,20 +122,38 @@ impl CommitCommand {
|
|||||||
// Auto-add if no files are staged and there are unstaged/untracked changes
|
// Auto-add if no files are staged and there are unstaged/untracked changes
|
||||||
if status.staged == 0 && (status.unstaged > 0 || status.untracked > 0) && !self.all {
|
if status.staged == 0 && (status.unstaged > 0 || status.untracked > 0) && !self.all {
|
||||||
println!("{}", messages.auto_stage_changes().yellow());
|
println!("{}", messages.auto_stage_changes().yellow());
|
||||||
repo.stage_all()?;
|
let removed = repo.stage_all(&messages)?;
|
||||||
println!("{}", messages.staged_all().green());
|
println!("{}", messages.staged_all().green());
|
||||||
|
if !removed.is_empty() {
|
||||||
|
print_warning(&format!(
|
||||||
|
"Removed {} ignored files from staging:",
|
||||||
|
removed.len()
|
||||||
|
));
|
||||||
|
for file in &removed {
|
||||||
|
println!(" • {}", file);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Re-check status after staging to ensure changes are detected
|
// Re-check status after staging to ensure changes are detected
|
||||||
let new_status = repo.status_summary()?;
|
let new_status = repo.status_summary()?;
|
||||||
if new_status.staged == 0 {
|
if new_status.staged == 0 {
|
||||||
bail!("Failed to stage changes. Please try running 'git add -A' manually.");
|
bail!("{}", messages.failed_to_stage());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Stage all if requested
|
// Stage all if requested
|
||||||
if self.all {
|
if self.all {
|
||||||
repo.stage_all()?;
|
let removed = repo.stage_all(&messages)?;
|
||||||
println!("{}", messages.staged_all().green());
|
println!("{}", messages.staged_all().green());
|
||||||
|
if !removed.is_empty() {
|
||||||
|
print_warning(&format!(
|
||||||
|
"Removed {} ignored files from staging:",
|
||||||
|
removed.len()
|
||||||
|
));
|
||||||
|
for file in &removed {
|
||||||
|
println!(" • {}", file);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Generate or build commit message
|
// Generate or build commit message
|
||||||
@@ -167,21 +195,21 @@ impl CommitCommand {
|
|||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
if !confirm {
|
if !confirm {
|
||||||
println!("{}", messages.commit_cancelled().yellow());
|
print_warning(messages.commit_cancelled());
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let result = if self.amend {
|
let result = if self.amend {
|
||||||
if self.dry_run {
|
if self.dry_run {
|
||||||
println!("\n{} {}", messages.dry_run(), "- commit not amended.".yellow());
|
println!("\n{}", messages.dry_run_commit_not_amended().yellow());
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
self.amend_commit(&repo, &commit_message)?;
|
self.amend_commit(&repo, &commit_message)?;
|
||||||
None
|
None
|
||||||
} else {
|
} else {
|
||||||
if self.dry_run {
|
if self.dry_run {
|
||||||
println!("\n{} {}", messages.dry_run(), "- commit not created.".yellow());
|
println!("\n{}", messages.dry_run_commit_not_created().yellow());
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
CommitBuilder::new()
|
CommitBuilder::new()
|
||||||
@@ -191,26 +219,34 @@ impl CommitCommand {
|
|||||||
};
|
};
|
||||||
|
|
||||||
if let Some(commit_oid) = result {
|
if let Some(commit_oid) = result {
|
||||||
println!("{} {}", messages.commit_created().green().bold(), commit_oid.to_string()[..8].to_string().cyan());
|
print_success(&format!(
|
||||||
|
"{} {}",
|
||||||
|
messages.commit_created().green().bold(),
|
||||||
|
commit_oid.to_string()[..8].to_string().cyan()
|
||||||
|
));
|
||||||
} else {
|
} else {
|
||||||
println!("{} {}", messages.commit_amended().green().bold(), "successfully");
|
print_success(messages.commit_amended_successfully());
|
||||||
}
|
}
|
||||||
|
|
||||||
// Push after commit if requested or ask user
|
// Push after commit if requested or ask user
|
||||||
if self.push {
|
if self.push || (!self.yes && !self.dry_run) {
|
||||||
println!("\n{}", messages.pushing_commit(&self.remote));
|
let branch = repo
|
||||||
repo.push(&self.remote, "HEAD")?;
|
.current_branch()
|
||||||
println!("{}", messages.pushed_commit(&self.remote));
|
.unwrap_or_else(|_| "HEAD (detached)".to_string());
|
||||||
} else if !self.yes && !self.dry_run {
|
|
||||||
let should_push = Confirm::new()
|
let should_push = if self.push {
|
||||||
.with_prompt(messages.push_after_commit())
|
true
|
||||||
.default(false)
|
} else {
|
||||||
.interact()?;
|
Confirm::new()
|
||||||
|
.with_prompt(messages.push_after_commit(&branch))
|
||||||
|
.default(false)
|
||||||
|
.interact()?
|
||||||
|
};
|
||||||
|
|
||||||
if should_push {
|
if should_push {
|
||||||
println!("\n{}", messages.pushing_commit(&self.remote));
|
print_progress(&messages.pushing_commit(&self.remote, &branch));
|
||||||
repo.push(&self.remote, "HEAD")?;
|
repo.push(&self.remote, "HEAD")?;
|
||||||
println!("{}", messages.pushed_commit(&self.remote));
|
print_success(&messages.pushed_commit(&self.remote, &branch));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -223,8 +259,9 @@ impl CommitCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn create_manual_commit(&self, format: CommitFormat) -> Result<String> {
|
fn create_manual_commit(&self, format: CommitFormat) -> Result<String> {
|
||||||
let description = self.message.clone()
|
let description = self.message.clone().ok_or_else(|| {
|
||||||
.ok_or_else(|| anyhow::anyhow!("Description required for manual commit. Use -m <message>"))?;
|
anyhow::anyhow!("Description required for manual commit. Use -m <message>")
|
||||||
|
})?;
|
||||||
|
|
||||||
// Try to extract commit type from message if not provided
|
// Try to extract commit type from message if not provided
|
||||||
let commit_type = if let Some(ref ct) = self.commit_type {
|
let commit_type = if let Some(ref ct) = self.commit_type {
|
||||||
@@ -250,31 +287,45 @@ impl CommitCommand {
|
|||||||
builder.build_message()
|
builder.build_message()
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn generate_commit(&self, repo: &GitRepo, format: CommitFormat, messages: &Messages) -> Result<String> {
|
async fn generate_commit(
|
||||||
|
&self,
|
||||||
|
repo: &GitRepo,
|
||||||
|
format: CommitFormat,
|
||||||
|
messages: &Messages,
|
||||||
|
) -> Result<String> {
|
||||||
let manager = ConfigManager::new()?;
|
let manager = ConfigManager::new()?;
|
||||||
let config = manager.config();
|
|
||||||
|
|
||||||
// Check if LLM is configured
|
let template = manager
|
||||||
let generator = ContentGenerator::new(&config.llm).await
|
.default_profile()
|
||||||
|
.and_then(|p| p.commit_template().map(|t| t.to_string()));
|
||||||
|
|
||||||
|
let generator = ContentGenerator::new_with_think(&manager, self.think, template)
|
||||||
|
.await
|
||||||
.context("Failed to initialize LLM. Use --manual for manual commit.")?;
|
.context("Failed to initialize LLM. Use --manual for manual commit.")?;
|
||||||
|
|
||||||
println!("{}", messages.ai_analyzing());
|
let spinner = crate::utils::Spinner::start(messages.ai_analyzing());
|
||||||
|
|
||||||
let language_str = &config.language.output_language;
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
let language = Language::from_str(language_str).unwrap_or(Language::English);
|
|
||||||
|
|
||||||
let generated = if self.yes {
|
let generated = if self.yes {
|
||||||
// Non-interactive mode: generate directly
|
generator
|
||||||
generator.generate_commit_from_repo(repo, format, language).await?
|
.generate_commit_from_repo(repo, format, language)
|
||||||
|
.await
|
||||||
} else {
|
} else {
|
||||||
// Interactive mode: allow user to review and regenerate
|
generator
|
||||||
generator.generate_commit_interactive(repo, format, language).await?
|
.generate_commit_interactive(repo, format, language, messages)
|
||||||
|
.await
|
||||||
};
|
};
|
||||||
|
spinner.finish_clear();
|
||||||
|
|
||||||
Ok(generated.to_conventional())
|
Ok(generated?.to_conventional())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn create_interactive_commit(&self, format: CommitFormat, messages: &Messages) -> Result<String> {
|
async fn create_interactive_commit(
|
||||||
|
&self,
|
||||||
|
format: CommitFormat,
|
||||||
|
messages: &Messages,
|
||||||
|
) -> Result<String> {
|
||||||
let types = get_commit_types(format == CommitFormat::Commitlint);
|
let types = get_commit_types(format == CommitFormat::Commitlint);
|
||||||
|
|
||||||
// Select type
|
// Select type
|
||||||
@@ -358,7 +409,8 @@ impl CommitCommand {
|
|||||||
"GPG signing failed. Please check:\n\
|
"GPG signing failed. Please check:\n\
|
||||||
1. GPG signing key is configured (git config --get user.signingkey)\n\
|
1. GPG signing key is configured (git config --get user.signingkey)\n\
|
||||||
2. GPG agent is running\n\
|
2. GPG agent is running\n\
|
||||||
3. You can sign commits manually (try: git commit --amend -S)".to_string()
|
3. You can sign commits manually (try: git commit --amend -S)"
|
||||||
|
.to_string()
|
||||||
} else {
|
} else {
|
||||||
stdout.to_string()
|
stdout.to_string()
|
||||||
}
|
}
|
||||||
@@ -373,26 +425,26 @@ impl CommitCommand {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Helper trait for optional builder methods
|
// // Helper trait for optional builder methods
|
||||||
trait CommitBuilderExt {
|
// trait CommitBuilderExt {
|
||||||
fn scope_opt(self, scope: Option<String>) -> Self;
|
// fn scope_opt(self, scope: Option<String>) -> Self;
|
||||||
fn body_opt(self, body: Option<String>) -> Self;
|
// fn body_opt(self, body: Option<String>) -> Self;
|
||||||
}
|
// }
|
||||||
|
|
||||||
impl CommitBuilderExt for CommitBuilder {
|
// impl CommitBuilderExt for CommitBuilder {
|
||||||
fn scope_opt(self, scope: Option<String>) -> Self {
|
// fn scope_opt(self, scope: Option<String>) -> Self {
|
||||||
if let Some(s) = scope {
|
// if let Some(s) = scope {
|
||||||
self.scope(s)
|
// self.scope(s)
|
||||||
} else {
|
// } else {
|
||||||
self
|
// self
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
|
|
||||||
fn body_opt(self, body: Option<String>) -> Self {
|
// fn body_opt(self, body: Option<String>) -> Self {
|
||||||
if let Some(b) = body {
|
// if let Some(b) = body {
|
||||||
self.body(b)
|
// self.body(b)
|
||||||
} else {
|
// } else {
|
||||||
self
|
// self
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
457
src/commands/credential.rs
Normal file
457
src/commands/credential.rs
Normal file
@@ -0,0 +1,457 @@
|
|||||||
|
use anyhow::{Context, Result};
|
||||||
|
use clap::{Parser, Subcommand};
|
||||||
|
use std::io::{self, BufRead, Write};
|
||||||
|
use std::path::PathBuf;
|
||||||
|
|
||||||
|
use crate::config::manager::ConfigManager;
|
||||||
|
use crate::config::{TokenConfig, TokenType};
|
||||||
|
|
||||||
|
/// Git credential helper command.
|
||||||
|
///
|
||||||
|
/// Implements the git credential helper protocol
|
||||||
|
/// (https://git-scm.com/docs/gitcredentials). Intended to be invoked by git
|
||||||
|
/// via `credential.helper` configuration, not by end users. Hidden from the
|
||||||
|
/// main help output.
|
||||||
|
#[derive(Parser)]
|
||||||
|
#[command(hide = true)]
|
||||||
|
pub struct CredentialCommand {
|
||||||
|
#[command(subcommand)]
|
||||||
|
command: CredentialSubcommand,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Subcommand)]
|
||||||
|
enum CredentialSubcommand {
|
||||||
|
/// Read attributes on stdin and output credentials on stdout.
|
||||||
|
#[command(hide = true)]
|
||||||
|
Get,
|
||||||
|
/// Read attributes (including password) on stdin and store them.
|
||||||
|
#[command(hide = true)]
|
||||||
|
Store,
|
||||||
|
/// Read attributes on stdin and erase any matching stored credentials.
|
||||||
|
#[command(hide = true)]
|
||||||
|
Erase,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl CredentialCommand {
|
||||||
|
pub async fn execute(&self, config_path: Option<PathBuf>) -> Result<()> {
|
||||||
|
match &self.command {
|
||||||
|
CredentialSubcommand::Get => Self::get(config_path),
|
||||||
|
CredentialSubcommand::Store => Self::store(config_path),
|
||||||
|
CredentialSubcommand::Erase => Self::erase(config_path),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_manager(config_path: &Option<PathBuf>) -> Result<ConfigManager> {
|
||||||
|
match config_path {
|
||||||
|
Some(path) => ConfigManager::with_path(path),
|
||||||
|
None => ConfigManager::new(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// `git credential get`: look up a PAT for the requested host and emit it
|
||||||
|
/// on stdout following the git credential helper protocol.
|
||||||
|
fn get(config_path: Option<PathBuf>) -> Result<()> {
|
||||||
|
let attrs = CredentialAttributes::from_stdin()?;
|
||||||
|
|
||||||
|
let host = match attrs.host.as_deref() {
|
||||||
|
Some(h) if !h.is_empty() => h,
|
||||||
|
_ => return Ok(()),
|
||||||
|
};
|
||||||
|
let service = host_to_service(host);
|
||||||
|
|
||||||
|
let manager = match Self::get_manager(&config_path) {
|
||||||
|
Ok(m) => m,
|
||||||
|
Err(_) => return Ok(()),
|
||||||
|
};
|
||||||
|
|
||||||
|
let (profile_name, pat) = match find_pat_for_service(&manager, &service) {
|
||||||
|
Some(tuple) => tuple,
|
||||||
|
None => return Ok(()),
|
||||||
|
};
|
||||||
|
|
||||||
|
// Prefer a git-supplied username; fall back to the profile's user_name.
|
||||||
|
let username = attrs.username.clone().or_else(|| {
|
||||||
|
manager
|
||||||
|
.get_profile(&profile_name)
|
||||||
|
.map(|p| p.user_name.clone())
|
||||||
|
});
|
||||||
|
|
||||||
|
let output = CredentialAttributes {
|
||||||
|
protocol: attrs.protocol.clone(),
|
||||||
|
host: attrs.host.clone(),
|
||||||
|
path: attrs.path.clone(),
|
||||||
|
username,
|
||||||
|
password: Some(pat),
|
||||||
|
};
|
||||||
|
output.to_stdout()?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// `git credential store`: persist the PAT provided by git using the
|
||||||
|
/// existing keyring-backed storage, associated with a matching profile.
|
||||||
|
fn store(config_path: Option<PathBuf>) -> Result<()> {
|
||||||
|
let attrs = CredentialAttributes::from_stdin()?;
|
||||||
|
|
||||||
|
let host = match attrs.host.as_deref() {
|
||||||
|
Some(h) if !h.is_empty() => h,
|
||||||
|
_ => return Ok(()),
|
||||||
|
};
|
||||||
|
let service = host_to_service(host);
|
||||||
|
|
||||||
|
let password = match attrs.password.as_deref() {
|
||||||
|
Some(p) if !p.is_empty() => p,
|
||||||
|
_ => return Ok(()),
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut manager = Self::get_manager(&config_path)?;
|
||||||
|
|
||||||
|
let profile_name = match find_profile_for_store(&manager, attrs.username.as_deref()) {
|
||||||
|
Some(name) => name,
|
||||||
|
None => return Ok(()),
|
||||||
|
};
|
||||||
|
|
||||||
|
// Store PAT in keyring using the existing profile-bound logic.
|
||||||
|
if let Err(e) = manager.store_pat_for_profile(&profile_name, &service, password) {
|
||||||
|
eprintln!("[quicommit credential] failed to store PAT: {}", e);
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
// Register the token in the profile config if not already present.
|
||||||
|
let already_has = manager
|
||||||
|
.get_profile(&profile_name)
|
||||||
|
.map(|p| p.tokens.contains_key(&service))
|
||||||
|
.unwrap_or(false);
|
||||||
|
if !already_has {
|
||||||
|
let _ = manager.add_token_to_profile(
|
||||||
|
&profile_name,
|
||||||
|
service.clone(),
|
||||||
|
TokenConfig::new(TokenType::Personal),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
manager.save()?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// `git credential erase`: remove any PAT stored for the requested host
|
||||||
|
/// from the keyring and the associated profile config entries.
|
||||||
|
fn erase(config_path: Option<PathBuf>) -> Result<()> {
|
||||||
|
let attrs = CredentialAttributes::from_stdin()?;
|
||||||
|
|
||||||
|
let host = match attrs.host.as_deref() {
|
||||||
|
Some(h) if !h.is_empty() => h,
|
||||||
|
_ => return Ok(()),
|
||||||
|
};
|
||||||
|
let service = host_to_service(host);
|
||||||
|
|
||||||
|
let mut manager = Self::get_manager(&config_path)?;
|
||||||
|
|
||||||
|
let profile_names: Vec<String> = manager.list_profiles().into_iter().cloned().collect();
|
||||||
|
|
||||||
|
for name in &profile_names {
|
||||||
|
let registered = manager
|
||||||
|
.get_profile(name)
|
||||||
|
.map(|p| p.tokens.contains_key(&service))
|
||||||
|
.unwrap_or(false);
|
||||||
|
|
||||||
|
if registered {
|
||||||
|
if let Err(e) = manager.remove_token_from_profile(name, &service) {
|
||||||
|
eprintln!(
|
||||||
|
"[quicommit credential] failed to erase PAT for '{}': {}",
|
||||||
|
name, e
|
||||||
|
);
|
||||||
|
}
|
||||||
|
} else if let Ok(true) = manager.delete_orphan_pat(name, &service) {
|
||||||
|
// Keyring PAT without a config token entry (issue 24).
|
||||||
|
eprintln!(
|
||||||
|
"[quicommit credential] erased unregistered PAT for '{}' (service {}) from keyring",
|
||||||
|
name, service
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
manager.save()?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Credential attributes exchanged with git via stdin/stdout following the
|
||||||
|
/// git credential helper protocol (`key=value`, one per line, blank line ends).
|
||||||
|
#[derive(Debug, Default, Clone, PartialEq, Eq)]
|
||||||
|
pub struct CredentialAttributes {
|
||||||
|
pub protocol: Option<String>,
|
||||||
|
pub host: Option<String>,
|
||||||
|
pub path: Option<String>,
|
||||||
|
pub username: Option<String>,
|
||||||
|
pub password: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl CredentialAttributes {
|
||||||
|
/// Read attributes from stdin following the git credential helper protocol.
|
||||||
|
pub fn from_stdin() -> Result<Self> {
|
||||||
|
let stdin = io::stdin();
|
||||||
|
let mut attrs = Self::default();
|
||||||
|
for line in stdin.lock().lines() {
|
||||||
|
let line = line.context("Failed to read credential attributes from stdin")?;
|
||||||
|
if line.is_empty() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if let Some((key, value)) = line.split_once('=') {
|
||||||
|
match key {
|
||||||
|
"protocol" => attrs.protocol = Some(value.to_string()),
|
||||||
|
"host" => attrs.host = Some(value.to_string()),
|
||||||
|
"path" => attrs.path = Some(value.to_string()),
|
||||||
|
"username" => attrs.username = Some(value.to_string()),
|
||||||
|
"password" => attrs.password = Some(value.to_string()),
|
||||||
|
_ => {} // ignore unknown keys
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(attrs)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Write attributes to stdout following the git credential helper protocol.
|
||||||
|
pub fn to_stdout(&self) -> Result<()> {
|
||||||
|
let stdout = io::stdout();
|
||||||
|
let mut handle = stdout.lock();
|
||||||
|
if let Some(ref v) = self.protocol {
|
||||||
|
writeln!(handle, "protocol={}", v)?;
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.host {
|
||||||
|
writeln!(handle, "host={}", v)?;
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.path {
|
||||||
|
writeln!(handle, "path={}", v)?;
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.username {
|
||||||
|
writeln!(handle, "username={}", v)?;
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.password {
|
||||||
|
writeln!(handle, "password={}", v)?;
|
||||||
|
}
|
||||||
|
writeln!(handle)?; // blank line terminates the attribute list
|
||||||
|
handle.flush()?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parse attributes from a text block following the git credential helper
|
||||||
|
/// protocol. Intended for testing and non-stdin input handling.
|
||||||
|
pub fn parse_str(input: &str) -> Self {
|
||||||
|
let mut attrs = Self::default();
|
||||||
|
for line in input.lines() {
|
||||||
|
if line.is_empty() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if let Some((key, value)) = line.split_once('=') {
|
||||||
|
match key {
|
||||||
|
"protocol" => attrs.protocol = Some(value.to_string()),
|
||||||
|
"host" => attrs.host = Some(value.to_string()),
|
||||||
|
"path" => attrs.path = Some(value.to_string()),
|
||||||
|
"username" => attrs.username = Some(value.to_string()),
|
||||||
|
"password" => attrs.password = Some(value.to_string()),
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attrs
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Serialize attributes to a string following the git credential helper
|
||||||
|
/// protocol. Intended for testing and non-stdout output handling.
|
||||||
|
pub fn serialize(&self) -> String {
|
||||||
|
let mut out = String::new();
|
||||||
|
if let Some(ref v) = self.protocol {
|
||||||
|
out.push_str(&format!("protocol={}\n", v));
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.host {
|
||||||
|
out.push_str(&format!("host={}\n", v));
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.path {
|
||||||
|
out.push_str(&format!("path={}\n", v));
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.username {
|
||||||
|
out.push_str(&format!("username={}\n", v));
|
||||||
|
}
|
||||||
|
if let Some(ref v) = self.password {
|
||||||
|
out.push_str(&format!("password={}\n", v));
|
||||||
|
}
|
||||||
|
out.push('\n');
|
||||||
|
out
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Map a git host to a service name used by the keyring-backed PAT storage.
|
||||||
|
///
|
||||||
|
/// Common git hosting services are mapped to short canonical names. Unknown
|
||||||
|
/// hosts are used as-is (lowercased, trailing slash trimmed).
|
||||||
|
pub fn host_to_service(host: &str) -> String {
|
||||||
|
let host = host.to_lowercase();
|
||||||
|
let host = host.trim_end_matches('/');
|
||||||
|
match host {
|
||||||
|
"github.com" | "www.github.com" => "github".to_string(),
|
||||||
|
"gitlab.com" | "www.gitlab.com" => "gitlab".to_string(),
|
||||||
|
"bitbucket.org" | "www.bitbucket.org" => "bitbucket".to_string(),
|
||||||
|
"codeberg.org" | "www.codeberg.org" => "codeberg".to_string(),
|
||||||
|
"gitea.com" | "www.gitea.com" => "gitea".to_string(),
|
||||||
|
"gitee.com" | "www.gitee.com" => "gitee".to_string(),
|
||||||
|
other => other.to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Search all profiles for one that has a PAT stored for the given service.
|
||||||
|
/// Returns the profile name and the PAT value.
|
||||||
|
fn find_pat_for_service(manager: &ConfigManager, service: &str) -> Option<(String, String)> {
|
||||||
|
for profile_name in manager.list_profiles() {
|
||||||
|
if manager.has_pat_for_profile(profile_name, service) {
|
||||||
|
if let Ok(Some(pat)) = manager.get_pat_for_profile(profile_name, service) {
|
||||||
|
return Some((profile_name.clone(), pat));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Determine which profile to use when storing a credential.
|
||||||
|
///
|
||||||
|
/// Prefers a profile whose `user_name` or `user_email` matches the
|
||||||
|
/// git-supplied username; otherwise falls back to the default profile.
|
||||||
|
fn find_profile_for_store(manager: &ConfigManager, username: Option<&str>) -> Option<String> {
|
||||||
|
if let Some(username) = username {
|
||||||
|
for name in manager.list_profiles() {
|
||||||
|
if let Some(profile) = manager.get_profile(name) {
|
||||||
|
if profile.user_name == username || profile.user_email == username {
|
||||||
|
return Some(name.clone());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
manager.default_profile_name().cloned()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Extract a PAT for the given host from saved credentials across all profiles.
|
||||||
|
///
|
||||||
|
/// This is intended for use by other parts of the application (e.g. when
|
||||||
|
/// verifying access to a git hosting service) and searches every configured
|
||||||
|
/// profile for a stored PAT matching the host.
|
||||||
|
pub fn get_pat_for_host(host: &str) -> Result<Option<String>> {
|
||||||
|
let manager = ConfigManager::new()?;
|
||||||
|
let service = host_to_service(host);
|
||||||
|
Ok(find_pat_for_service(&manager, &service).map(|(_, pat)| pat))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_host_to_service_known_hosts() {
|
||||||
|
assert_eq!(host_to_service("github.com"), "github");
|
||||||
|
assert_eq!(host_to_service("www.github.com"), "github");
|
||||||
|
assert_eq!(host_to_service("gitlab.com"), "gitlab");
|
||||||
|
assert_eq!(host_to_service("bitbucket.org"), "bitbucket");
|
||||||
|
assert_eq!(host_to_service("codeberg.org"), "codeberg");
|
||||||
|
assert_eq!(host_to_service("gitea.com"), "gitea");
|
||||||
|
assert_eq!(host_to_service("gitee.com"), "gitee");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_host_to_service_case_insensitive() {
|
||||||
|
assert_eq!(host_to_service("GitHub.Com"), "github");
|
||||||
|
assert_eq!(host_to_service("GITLAB.COM"), "gitlab");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_host_to_service_trailing_slash() {
|
||||||
|
assert_eq!(host_to_service("github.com/"), "github");
|
||||||
|
assert_eq!(host_to_service("gitlab.com//"), "gitlab");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_host_to_service_unknown_host() {
|
||||||
|
assert_eq!(host_to_service("example.com"), "example.com");
|
||||||
|
assert_eq!(host_to_service("git.internal.corp"), "git.internal.corp");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_parse_basic() {
|
||||||
|
let input = "protocol=https\nhost=github.com\nusername=octocat\npassword=ghp_token123\n\n";
|
||||||
|
let attrs = CredentialAttributes::parse_str(input);
|
||||||
|
assert_eq!(attrs.protocol.as_deref(), Some("https"));
|
||||||
|
assert_eq!(attrs.host.as_deref(), Some("github.com"));
|
||||||
|
assert_eq!(attrs.username.as_deref(), Some("octocat"));
|
||||||
|
assert_eq!(attrs.password.as_deref(), Some("ghp_token123"));
|
||||||
|
assert!(attrs.path.is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_parse_with_path() {
|
||||||
|
let input = "protocol=https\nhost=github.com\npath=owner/repo.git\n\n";
|
||||||
|
let attrs = CredentialAttributes::parse_str(input);
|
||||||
|
assert_eq!(attrs.path.as_deref(), Some("owner/repo.git"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_parse_empty() {
|
||||||
|
let attrs = CredentialAttributes::parse_str("");
|
||||||
|
assert!(attrs.protocol.is_none());
|
||||||
|
assert!(attrs.host.is_none());
|
||||||
|
assert!(attrs.username.is_none());
|
||||||
|
assert!(attrs.password.is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_parse_ignores_unknown_keys() {
|
||||||
|
let input = "protocol=https\nhost=github.com\nunknown=value\nfoo=bar\n\n";
|
||||||
|
let attrs = CredentialAttributes::parse_str(input);
|
||||||
|
assert_eq!(attrs.protocol.as_deref(), Some("https"));
|
||||||
|
assert_eq!(attrs.host.as_deref(), Some("github.com"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_parse_stops_at_blank_line() {
|
||||||
|
let input = "protocol=https\nhost=github.com\n\npassword=should_be_ignored\n";
|
||||||
|
let attrs = CredentialAttributes::parse_str(input);
|
||||||
|
assert_eq!(attrs.protocol.as_deref(), Some("https"));
|
||||||
|
assert_eq!(attrs.host.as_deref(), Some("github.com"));
|
||||||
|
assert!(attrs.password.is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_serialize_roundtrip() {
|
||||||
|
let attrs = CredentialAttributes {
|
||||||
|
protocol: Some("https".to_string()),
|
||||||
|
host: Some("github.com".to_string()),
|
||||||
|
path: None,
|
||||||
|
username: Some("octocat".to_string()),
|
||||||
|
password: Some("ghp_token".to_string()),
|
||||||
|
};
|
||||||
|
let serialized = attrs.serialize();
|
||||||
|
let reparsed = CredentialAttributes::parse_str(&serialized);
|
||||||
|
assert_eq!(attrs, reparsed);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_serialize_includes_blank_line() {
|
||||||
|
let attrs = CredentialAttributes {
|
||||||
|
protocol: Some("https".to_string()),
|
||||||
|
host: Some("github.com".to_string()),
|
||||||
|
path: None,
|
||||||
|
username: None,
|
||||||
|
password: None,
|
||||||
|
};
|
||||||
|
let serialized = attrs.serialize();
|
||||||
|
assert!(serialized.ends_with("\n\n"));
|
||||||
|
assert!(serialized.contains("protocol=https\n"));
|
||||||
|
assert!(serialized.contains("host=github.com\n"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_credential_attributes_default() {
|
||||||
|
let attrs = CredentialAttributes::default();
|
||||||
|
assert!(attrs.protocol.is_none());
|
||||||
|
assert!(attrs.host.is_none());
|
||||||
|
assert!(attrs.path.is_none());
|
||||||
|
assert!(attrs.username.is_none());
|
||||||
|
assert!(attrs.password.is_none());
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -2,12 +2,15 @@ use anyhow::Result;
|
|||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use colored::Colorize;
|
use colored::Colorize;
|
||||||
use dialoguer::{Confirm, Input, Select};
|
use dialoguer::{Confirm, Input, Select};
|
||||||
|
use std::path::PathBuf;
|
||||||
|
|
||||||
use crate::config::{GitProfile, Language};
|
|
||||||
use crate::config::manager::ConfigManager;
|
use crate::config::manager::ConfigManager;
|
||||||
use crate::config::profile::{GpgConfig, SshConfig};
|
use crate::config::profile::{GpgConfig, SshConfig};
|
||||||
|
use crate::config::{GitProfile, Language};
|
||||||
use crate::i18n::Messages;
|
use crate::i18n::Messages;
|
||||||
|
use crate::utils::keyring::{get_default_model, get_supported_providers, provider_needs_api_key};
|
||||||
use crate::utils::validators::validate_email;
|
use crate::utils::validators::validate_email;
|
||||||
|
use crate::utils::{print_success, print_warning};
|
||||||
|
|
||||||
/// Initialize quicommit configuration
|
/// Initialize quicommit configuration
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
@@ -22,75 +25,98 @@ pub struct InitCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl InitCommand {
|
impl InitCommand {
|
||||||
pub async fn execute(&self) -> Result<()> {
|
pub async fn execute(&self, config_path: Option<PathBuf>) -> Result<()> {
|
||||||
// Start with English messages for initialization
|
|
||||||
let messages = Messages::new(Language::English);
|
let messages = Messages::new(Language::English);
|
||||||
println!("{}", messages.initializing().bold().cyan());
|
println!("{}", messages.initializing().bold().cyan());
|
||||||
|
|
||||||
let config_path = crate::config::AppConfig::default_path()?;
|
let config_path =
|
||||||
|
config_path.unwrap_or_else(|| crate::config::AppConfig::default_path().unwrap());
|
||||||
|
|
||||||
// Check if config already exists
|
|
||||||
if config_path.exists() && !self.reset {
|
if config_path.exists() && !self.reset {
|
||||||
if !self.yes {
|
if !self.yes {
|
||||||
let overwrite = Confirm::new()
|
let overwrite = Confirm::new()
|
||||||
.with_prompt("Configuration already exists. Overwrite?")
|
.with_prompt(messages.config_exists_overwrite())
|
||||||
.default(false)
|
.default(false)
|
||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
if !overwrite {
|
if !overwrite {
|
||||||
println!("{}", "Initialization cancelled.".yellow());
|
print_warning(messages.init_cancelled());
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
print_warning(messages.config_exists_use_reset());
|
||||||
|
return Ok(());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut manager = if self.reset {
|
if let Some(parent) = config_path.parent() {
|
||||||
ConfigManager::new()?
|
std::fs::create_dir_all(parent)
|
||||||
} else {
|
.map_err(|e| anyhow::anyhow!("Failed to create config directory: {}", e))?;
|
||||||
ConfigManager::new().or_else(|_| Ok::<_, anyhow::Error>(ConfigManager::default()))?
|
}
|
||||||
};
|
|
||||||
|
let mut manager = ConfigManager::with_path_fresh(&config_path)?;
|
||||||
|
|
||||||
if self.yes {
|
if self.yes {
|
||||||
// Quick setup with defaults
|
self.quick_setup(&mut manager, &messages).await?;
|
||||||
self.quick_setup(&mut manager).await?;
|
|
||||||
} else {
|
} else {
|
||||||
// Interactive setup
|
|
||||||
self.interactive_setup(&mut manager).await?;
|
self.interactive_setup(&mut manager).await?;
|
||||||
}
|
}
|
||||||
|
|
||||||
manager.save()?;
|
manager.save()?;
|
||||||
|
|
||||||
// Get configured language for final messages
|
|
||||||
let language = manager.get_language().unwrap_or(Language::English);
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
let messages = Messages::new(language);
|
let messages = Messages::new(language);
|
||||||
|
|
||||||
println!("{}", messages.init_success().bold().green());
|
print_success(messages.init_success());
|
||||||
println!("\n{}: {}", messages.config_file(), config_path.display());
|
println!("\n{}: {}", messages.config_file(), config_path.display());
|
||||||
println!("\n{}:", messages.next_steps());
|
println!("\n{}:", messages.next_steps());
|
||||||
println!(" 1. Create a profile: {}", "quicommit profile add".cyan());
|
println!(
|
||||||
println!(" 2. Configure LLM: {}", "quicommit config set-llm".cyan());
|
" 1. {}: {}",
|
||||||
println!(" 3. Start committing: {}", "quicommit commit".cyan());
|
messages.next_steps_create_profile(),
|
||||||
|
"quicommit profile add".cyan()
|
||||||
|
);
|
||||||
|
println!(
|
||||||
|
" 2. {}: {}",
|
||||||
|
messages.next_steps_configure_llm(),
|
||||||
|
"quicommit config set-llm".cyan()
|
||||||
|
);
|
||||||
|
println!(
|
||||||
|
" 3. {}: {}",
|
||||||
|
messages.next_steps_start_committing(),
|
||||||
|
"quicommit commit".cyan()
|
||||||
|
);
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn quick_setup(&self, manager: &mut ConfigManager) -> Result<()> {
|
async fn quick_setup(&self, manager: &mut ConfigManager, messages: &Messages) -> Result<()> {
|
||||||
// Try to get git user info
|
|
||||||
let git_config = git2::Config::open_default()?;
|
let git_config = git2::Config::open_default()?;
|
||||||
|
|
||||||
let user_name = git_config.get_string("user.name").unwrap_or_else(|_| "User".to_string());
|
let name_result = git_config.get_string("user.name");
|
||||||
let user_email = git_config.get_string("user.email").unwrap_or_else(|_| "user@example.com".to_string());
|
let email_result = git_config.get_string("user.email");
|
||||||
|
let name_missing = name_result.is_err();
|
||||||
|
let email_missing = email_result.is_err();
|
||||||
|
let user_name = name_result.unwrap_or_else(|_| "User".to_string());
|
||||||
|
let user_email = email_result.unwrap_or_else(|_| "user@example.com".to_string());
|
||||||
|
|
||||||
let profile = GitProfile::new(
|
let profile = GitProfile::new("default".to_string(), user_name.clone(), user_email.clone());
|
||||||
"default".to_string(),
|
|
||||||
|
println!("\n{}", messages.quick_setup_summary().bold());
|
||||||
|
println!(" {}: {}", messages.profile_label(), "default".cyan());
|
||||||
|
println!(
|
||||||
|
" {}: {} <{}>",
|
||||||
|
messages.identity_label(),
|
||||||
user_name,
|
user_name,
|
||||||
user_email,
|
user_email
|
||||||
);
|
);
|
||||||
|
if name_missing || email_missing {
|
||||||
|
print_warning(messages.identity_placeholder_warning());
|
||||||
|
}
|
||||||
|
println!(" {}: {}", messages.llm_provider_label(), "ollama".cyan());
|
||||||
|
|
||||||
manager.add_profile("default".to_string(), profile)?;
|
manager.add_profile("default".to_string(), profile)?;
|
||||||
manager.set_default_profile(Some("default".to_string()))?;
|
manager.set_default_profile(Some("default".to_string()))?;
|
||||||
|
|
||||||
// Set default LLM to Ollama
|
|
||||||
manager.set_llm_provider("ollama".to_string());
|
manager.set_llm_provider("ollama".to_string());
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -100,9 +126,8 @@ impl InitCommand {
|
|||||||
let messages = Messages::new(Language::English);
|
let messages = Messages::new(Language::English);
|
||||||
println!("\n{}", messages.setup_profile().bold());
|
println!("\n{}", messages.setup_profile().bold());
|
||||||
|
|
||||||
// Language selection
|
|
||||||
println!("\n{}", messages.select_output_language().bold());
|
println!("\n{}", messages.select_output_language().bold());
|
||||||
let languages = vec![
|
let languages = [
|
||||||
Language::English,
|
Language::English,
|
||||||
Language::Chinese,
|
Language::Chinese,
|
||||||
Language::Japanese,
|
Language::Japanese,
|
||||||
@@ -111,32 +136,31 @@ impl InitCommand {
|
|||||||
Language::French,
|
Language::French,
|
||||||
Language::German,
|
Language::German,
|
||||||
];
|
];
|
||||||
let language_names: Vec<String> = languages.iter().map(|l| l.display_name().to_string()).collect();
|
let language_names: Vec<String> = languages
|
||||||
let language_idx = Select::new()
|
.iter()
|
||||||
.items(&language_names)
|
.map(|l| l.display_name().to_string())
|
||||||
.default(0)
|
.collect();
|
||||||
.interact()?;
|
let language_idx = Select::new().items(&language_names).default(0).interact()?;
|
||||||
|
|
||||||
let selected_language = languages[language_idx];
|
let selected_language = languages[language_idx];
|
||||||
manager.set_output_language(selected_language.to_code().to_string());
|
manager.set_output_language(selected_language.to_code().to_string());
|
||||||
|
|
||||||
// Update messages to selected language
|
|
||||||
let messages = Messages::new(selected_language);
|
let messages = Messages::new(selected_language);
|
||||||
|
|
||||||
// Profile name
|
|
||||||
let profile_name: String = Input::new()
|
let profile_name: String = Input::new()
|
||||||
.with_prompt(messages.profile_name())
|
.with_prompt(messages.profile_name())
|
||||||
.default("personal".to_string())
|
.default("personal".to_string())
|
||||||
.interact_text()?;
|
.interact_text()?;
|
||||||
|
|
||||||
// User info
|
|
||||||
let git_config = git2::Config::open_default().ok();
|
let git_config = git2::Config::open_default().ok();
|
||||||
|
|
||||||
let default_name = git_config.as_ref()
|
let default_name = git_config
|
||||||
|
.as_ref()
|
||||||
.and_then(|c| c.get_string("user.name").ok())
|
.and_then(|c| c.get_string("user.name").ok())
|
||||||
.unwrap_or_default();
|
.unwrap_or_default();
|
||||||
|
|
||||||
let default_email = git_config.as_ref()
|
let default_email = git_config
|
||||||
|
.as_ref()
|
||||||
.and_then(|c| c.get_string("user.email").ok())
|
.and_then(|c| c.get_string("user.email").ok())
|
||||||
.unwrap_or_default();
|
.unwrap_or_default();
|
||||||
|
|
||||||
@@ -148,9 +172,7 @@ impl InitCommand {
|
|||||||
let user_email: String = Input::new()
|
let user_email: String = Input::new()
|
||||||
.with_prompt(messages.git_user_email())
|
.with_prompt(messages.git_user_email())
|
||||||
.default(default_email)
|
.default(default_email)
|
||||||
.validate_with(|input: &String| {
|
.validate_with(|input: &String| validate_email(input).map_err(|e| e.to_string()))
|
||||||
validate_email(input).map_err(|e| e.to_string())
|
|
||||||
})
|
|
||||||
.interact_text()?;
|
.interact_text()?;
|
||||||
|
|
||||||
let description: String = Input::new()
|
let description: String = Input::new()
|
||||||
@@ -164,14 +186,15 @@ impl InitCommand {
|
|||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
let organization = if is_work {
|
let organization = if is_work {
|
||||||
Some(Input::new()
|
Some(
|
||||||
.with_prompt(messages.organization_name())
|
Input::new()
|
||||||
.interact_text()?)
|
.with_prompt(messages.organization_name())
|
||||||
|
.interact_text()?,
|
||||||
|
)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
// SSH configuration
|
|
||||||
let setup_ssh = Confirm::new()
|
let setup_ssh = Confirm::new()
|
||||||
.with_prompt(messages.configure_ssh())
|
.with_prompt(messages.configure_ssh())
|
||||||
.default(false)
|
.default(false)
|
||||||
@@ -183,7 +206,6 @@ impl InitCommand {
|
|||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
// GPG configuration
|
|
||||||
let setup_gpg = Confirm::new()
|
let setup_gpg = Confirm::new()
|
||||||
.with_prompt(messages.configure_gpg())
|
.with_prompt(messages.configure_gpg())
|
||||||
.default(false)
|
.default(false)
|
||||||
@@ -195,12 +217,7 @@ impl InitCommand {
|
|||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
// Create profile
|
let mut profile = GitProfile::new(profile_name.clone(), user_name, user_email);
|
||||||
let mut profile = GitProfile::new(
|
|
||||||
profile_name.clone(),
|
|
||||||
user_name,
|
|
||||||
user_email,
|
|
||||||
);
|
|
||||||
|
|
||||||
if !description.is_empty() {
|
if !description.is_empty() {
|
||||||
profile.description = Some(description);
|
profile.description = Some(description);
|
||||||
@@ -214,59 +231,100 @@ impl InitCommand {
|
|||||||
manager.add_profile(profile_name.clone(), profile)?;
|
manager.add_profile(profile_name.clone(), profile)?;
|
||||||
manager.set_default_profile(Some(profile_name))?;
|
manager.set_default_profile(Some(profile_name))?;
|
||||||
|
|
||||||
// LLM provider selection
|
|
||||||
println!("\n{}", messages.select_llm_provider().bold());
|
println!("\n{}", messages.select_llm_provider().bold());
|
||||||
let providers = vec![
|
|
||||||
|
let provider_display_names = vec![
|
||||||
"Ollama (local)",
|
"Ollama (local)",
|
||||||
"OpenAI",
|
"OpenAI",
|
||||||
"Anthropic Claude",
|
"Anthropic Claude",
|
||||||
"Kimi (Moonshot AI)",
|
"Kimi (Moonshot AI)",
|
||||||
"DeepSeek",
|
"DeepSeek",
|
||||||
"OpenRouter"
|
"OpenRouter",
|
||||||
];
|
];
|
||||||
|
|
||||||
let provider_idx = Select::new()
|
let provider_idx = Select::new()
|
||||||
.items(&providers)
|
.items(&provider_display_names)
|
||||||
.default(0)
|
.default(0)
|
||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
let provider = match provider_idx {
|
let providers = get_supported_providers();
|
||||||
0 => "ollama",
|
let provider = providers[provider_idx].to_string();
|
||||||
1 => "openai",
|
|
||||||
2 => "anthropic",
|
let keyring = manager.keyring();
|
||||||
3 => "kimi",
|
let keyring_available = keyring.is_available();
|
||||||
4 => "deepseek",
|
|
||||||
5 => "openrouter",
|
if !keyring_available {
|
||||||
_ => "ollama",
|
print_warning(messages.keyring_unavailable());
|
||||||
|
print_warning(&keyring.get_status_message());
|
||||||
|
}
|
||||||
|
|
||||||
|
let api_key = if provider_needs_api_key(&provider) {
|
||||||
|
let env_key = std::env::var("QUICOMMIT_API_KEY")
|
||||||
|
.or_else(|_| {
|
||||||
|
std::env::var(format!("QUICOMMIT_{}_API_KEY", provider.to_uppercase()))
|
||||||
|
})
|
||||||
|
.ok();
|
||||||
|
|
||||||
|
if let Some(_key) = env_key {
|
||||||
|
print_success(messages.api_key_found_env());
|
||||||
|
None
|
||||||
|
} else if keyring_available {
|
||||||
|
let prompt = match provider.as_str() {
|
||||||
|
"openai" => messages.openai_api_key(),
|
||||||
|
"anthropic" => messages.anthropic_api_key(),
|
||||||
|
"kimi" => messages.kimi_api_key(),
|
||||||
|
"deepseek" => messages.deepseek_api_key(),
|
||||||
|
"openrouter" => messages.openrouter_api_key(),
|
||||||
|
_ => messages.api_key_prompt_other(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let key: String = crate::utils::password_input(prompt)?;
|
||||||
|
Some(key)
|
||||||
|
} else {
|
||||||
|
print_warning(messages.please_set_api_key_env());
|
||||||
|
None
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
manager.set_llm_provider(provider.to_string());
|
let default_model = get_default_model(&provider);
|
||||||
|
let model: String = Input::new()
|
||||||
|
.with_prompt(messages.model_name())
|
||||||
|
.default(default_model.to_string())
|
||||||
|
.interact_text()?;
|
||||||
|
|
||||||
// Configure API key if needed
|
let base_url: Option<String> = if provider == "ollama" {
|
||||||
if provider == "openai" {
|
let url: String = Input::new()
|
||||||
let api_key: String = Input::new()
|
.with_prompt(messages.ollama_server_url())
|
||||||
.with_prompt(messages.openai_api_key())
|
.default("http://localhost:11434".to_string())
|
||||||
.interact_text()?;
|
.interact_text()?;
|
||||||
manager.set_openai_api_key(api_key);
|
Some(url)
|
||||||
} else if provider == "anthropic" {
|
} else {
|
||||||
let api_key: String = Input::new()
|
let use_custom_url = Confirm::new()
|
||||||
.with_prompt(messages.anthropic_api_key())
|
.with_prompt(messages.use_custom_base_url())
|
||||||
.interact_text()?;
|
.default(false)
|
||||||
manager.set_anthropic_api_key(api_key);
|
.interact()?;
|
||||||
} else if provider == "kimi" {
|
|
||||||
let api_key: String = Input::new()
|
if use_custom_url {
|
||||||
.with_prompt(messages.kimi_api_key())
|
let url: String = Input::new()
|
||||||
.interact_text()?;
|
.with_prompt(messages.base_url_plain())
|
||||||
manager.set_kimi_api_key(api_key);
|
.interact_text()?;
|
||||||
} else if provider == "deepseek" {
|
Some(url)
|
||||||
let api_key: String = Input::new()
|
} else {
|
||||||
.with_prompt(messages.deepseek_api_key())
|
None
|
||||||
.interact_text()?;
|
}
|
||||||
manager.set_deepseek_api_key(api_key);
|
};
|
||||||
} else if provider == "openrouter" {
|
|
||||||
let api_key: String = Input::new()
|
manager.set_llm_provider(provider.clone());
|
||||||
.with_prompt(messages.openrouter_api_key())
|
manager.set_llm_model(model);
|
||||||
.interact_text()?;
|
manager.set_llm_base_url(base_url);
|
||||||
manager.set_openrouter_api_key(api_key);
|
|
||||||
|
if let Some(key) = api_key
|
||||||
|
&& provider_needs_api_key(&provider)
|
||||||
|
{
|
||||||
|
manager.set_api_key(&key)?;
|
||||||
|
print_success(messages.api_key_stored_keyring());
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -284,6 +342,17 @@ impl InitCommand {
|
|||||||
.default(ssh_dir.join("id_rsa").display().to_string())
|
.default(ssh_dir.join("id_rsa").display().to_string())
|
||||||
.interact_text()?;
|
.interact_text()?;
|
||||||
|
|
||||||
|
let pub_key_path: String = Input::new()
|
||||||
|
.with_prompt(messages.ssh_public_key_path())
|
||||||
|
.default(ssh_dir.join("id_rsa.pub").display().to_string())
|
||||||
|
.allow_empty(true)
|
||||||
|
.interact_text()?;
|
||||||
|
let public_key_path = if pub_key_path.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(PathBuf::from(pub_key_path))
|
||||||
|
};
|
||||||
|
|
||||||
let has_passphrase = Confirm::new()
|
let has_passphrase = Confirm::new()
|
||||||
.with_prompt(messages.has_passphrase())
|
.with_prompt(messages.has_passphrase())
|
||||||
.default(false)
|
.default(false)
|
||||||
@@ -295,13 +364,38 @@ impl InitCommand {
|
|||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let agent_forwarding = Confirm::new()
|
||||||
|
.with_prompt(messages.ssh_agent_forwarding())
|
||||||
|
.default(false)
|
||||||
|
.interact()?;
|
||||||
|
|
||||||
|
let known_hosts: String = Input::new()
|
||||||
|
.with_prompt(messages.known_hosts_path())
|
||||||
|
.allow_empty(true)
|
||||||
|
.interact_text()?;
|
||||||
|
let known_hosts_file = if known_hosts.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(PathBuf::from(known_hosts))
|
||||||
|
};
|
||||||
|
|
||||||
|
let custom_cmd: String = Input::new()
|
||||||
|
.with_prompt(messages.custom_ssh_command())
|
||||||
|
.allow_empty(true)
|
||||||
|
.interact_text()?;
|
||||||
|
let ssh_command = if custom_cmd.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(custom_cmd)
|
||||||
|
};
|
||||||
|
|
||||||
Ok(SshConfig {
|
Ok(SshConfig {
|
||||||
private_key_path: Some(PathBuf::from(key_path)),
|
private_key_path: Some(PathBuf::from(key_path)),
|
||||||
public_key_path: None,
|
public_key_path,
|
||||||
passphrase,
|
passphrase,
|
||||||
agent_forwarding: false,
|
agent_forwarding,
|
||||||
ssh_command: None,
|
ssh_command,
|
||||||
known_hosts_file: None,
|
known_hosts_file,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
pub mod changelog;
|
pub mod changelog;
|
||||||
pub mod commit;
|
pub mod commit;
|
||||||
pub mod config;
|
pub mod config;
|
||||||
|
pub mod credential;
|
||||||
pub mod init;
|
pub mod init;
|
||||||
pub mod profile;
|
pub mod profile;
|
||||||
pub mod tag;
|
pub mod tag;
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -1,16 +1,19 @@
|
|||||||
use anyhow::{bail, Result};
|
use anyhow::{Result, bail};
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use colored::Colorize;
|
use colored::Colorize;
|
||||||
use dialoguer::{Confirm, Input, Select};
|
use dialoguer::{Confirm, Input, Select};
|
||||||
use semver::Version;
|
use semver::Version;
|
||||||
|
use std::path::PathBuf;
|
||||||
|
|
||||||
use crate::config::{Language, manager::ConfigManager};
|
use crate::config::{Language, manager::ConfigManager};
|
||||||
use crate::git::{find_repo, GitRepo};
|
|
||||||
use crate::generator::ContentGenerator;
|
use crate::generator::ContentGenerator;
|
||||||
use crate::git::tag::{
|
use crate::git::tag::{
|
||||||
bump_version, get_latest_version, suggest_version_bump, TagBuilder, VersionBump,
|
ConfigVersion, TagBuilder, VersionBump, bump_version, get_latest_version,
|
||||||
|
read_project_versions, suggest_version_bump,
|
||||||
};
|
};
|
||||||
|
use crate::git::{GitRepo, find_repo};
|
||||||
use crate::i18n::Messages;
|
use crate::i18n::Messages;
|
||||||
|
use crate::utils::{print_progress, print_success, print_warning};
|
||||||
|
|
||||||
/// Generate and create Git tags
|
/// Generate and create Git tags
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
@@ -55,15 +58,28 @@ pub struct TagCommand {
|
|||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
dry_run: bool,
|
dry_run: bool,
|
||||||
|
|
||||||
/// Skip interactive prompts
|
/// Enable thinking mode for this tag (override config)
|
||||||
|
#[arg(short = 't', long)]
|
||||||
|
think: bool,
|
||||||
|
|
||||||
|
/// Skip interactive prompts only (generation behavior unchanged)
|
||||||
#[arg(short = 'y', long)]
|
#[arg(short = 'y', long)]
|
||||||
yes: bool,
|
yes: bool,
|
||||||
|
|
||||||
|
/// Auto-detect version from project config (Cargo.toml/pyproject.toml),
|
||||||
|
/// falling back to commit analysis with confirmation. Mutually exclusive with --bump.
|
||||||
|
#[arg(short = 'A', long, conflicts_with = "bump")]
|
||||||
|
auto: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl TagCommand {
|
impl TagCommand {
|
||||||
pub async fn execute(&self) -> Result<()> {
|
pub async fn execute(&self, config_path: Option<PathBuf>) -> Result<()> {
|
||||||
let repo = find_repo(std::env::current_dir()?.as_path())?;
|
let repo = find_repo(std::env::current_dir()?.as_path())?;
|
||||||
let manager = ConfigManager::new()?;
|
let manager = if let Some(ref path) = config_path {
|
||||||
|
ConfigManager::with_path(path)?
|
||||||
|
} else {
|
||||||
|
ConfigManager::new()?
|
||||||
|
};
|
||||||
let config = manager.config();
|
let config = manager.config();
|
||||||
let language = manager.get_language().unwrap_or(Language::English);
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
let messages = Messages::new(language);
|
let messages = Messages::new(language);
|
||||||
@@ -71,11 +87,14 @@ impl TagCommand {
|
|||||||
// Determine tag name
|
// Determine tag name
|
||||||
let tag_name = if let Some(name) = &self.name {
|
let tag_name = if let Some(name) = &self.name {
|
||||||
name.clone()
|
name.clone()
|
||||||
|
} else if self.auto {
|
||||||
|
self.auto_detect_version(&repo, &config.tag.version_prefix, &messages)
|
||||||
|
.await?
|
||||||
} else if let Some(bump_str) = &self.bump {
|
} else if let Some(bump_str) = &self.bump {
|
||||||
// Calculate bumped version
|
// Calculate bumped version
|
||||||
let prefix = &config.tag.version_prefix;
|
let prefix = &config.tag.version_prefix;
|
||||||
let latest = get_latest_version(&repo, prefix)?
|
let latest =
|
||||||
.unwrap_or_else(|| Version::new(0, 0, 0));
|
get_latest_version(&repo, prefix)?.unwrap_or_else(|| Version::new(0, 0, 0));
|
||||||
|
|
||||||
let bump = VersionBump::from_str(bump_str)?;
|
let bump = VersionBump::from_str(bump_str)?;
|
||||||
let new_version = bump_version(&latest, bump, None);
|
let new_version = bump_version(&latest, bump, None);
|
||||||
@@ -83,18 +102,25 @@ impl TagCommand {
|
|||||||
format!("{}{}", prefix, new_version)
|
format!("{}{}", prefix, new_version)
|
||||||
} else {
|
} else {
|
||||||
// Interactive mode
|
// Interactive mode
|
||||||
self.select_version_interactive(&repo, &config.tag.version_prefix, &messages).await?
|
self.select_version_interactive(&repo, &config.tag.version_prefix, &messages)
|
||||||
|
.await?
|
||||||
};
|
};
|
||||||
|
|
||||||
// Validate tag name (if it looks like a version)
|
// Validate tag name (if it looks like a version)
|
||||||
if tag_name.starts_with('v') || tag_name.chars().next().map(|c| c.is_ascii_digit()).unwrap_or(false) {
|
if tag_name.starts_with('v')
|
||||||
|
|| tag_name
|
||||||
|
.chars()
|
||||||
|
.next()
|
||||||
|
.map(|c| c.is_ascii_digit())
|
||||||
|
.unwrap_or(false)
|
||||||
|
{
|
||||||
let version_str = tag_name.trim_start_matches('v');
|
let version_str = tag_name.trim_start_matches('v');
|
||||||
if let Err(e) = crate::utils::validators::validate_semver(version_str) {
|
if let Err(e) = crate::utils::validators::validate_semver(version_str) {
|
||||||
println!("{}: {}", "Warning".yellow(), e);
|
print_warning(&format!("{}: {}", messages.warning(), e));
|
||||||
|
|
||||||
if !self.yes {
|
if !self.yes {
|
||||||
let proceed = Confirm::new()
|
let proceed = Confirm::new()
|
||||||
.with_prompt("Proceed with this tag name anyway?")
|
.with_prompt(messages.proceed_invalid_tag_name())
|
||||||
.default(true)
|
.default(true)
|
||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
@@ -110,12 +136,15 @@ impl TagCommand {
|
|||||||
None
|
None
|
||||||
} else if let Some(msg) = &self.message {
|
} else if let Some(msg) = &self.message {
|
||||||
Some(msg.clone())
|
Some(msg.clone())
|
||||||
} else if self.generate || (config.tag.auto_generate && !self.yes) {
|
} else if self.generate || config.tag.auto_generate {
|
||||||
Some(self.generate_tag_message(&repo, &tag_name, &messages).await?)
|
Some(
|
||||||
|
self.generate_tag_message(&repo, &tag_name, &messages)
|
||||||
|
.await?,
|
||||||
|
)
|
||||||
} else if !self.yes {
|
} else if !self.yes {
|
||||||
Some(self.input_message_interactive(&tag_name, &messages)?)
|
Some(self.input_message_interactive(&tag_name, &messages)?)
|
||||||
} else {
|
} else {
|
||||||
Some(format!("Release {}", tag_name))
|
Some(messages.release_default(&tag_name))
|
||||||
};
|
};
|
||||||
|
|
||||||
// Show preview
|
// Show preview
|
||||||
@@ -137,13 +166,13 @@ impl TagCommand {
|
|||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
if !confirm {
|
if !confirm {
|
||||||
println!("{}", messages.tag_cancelled().yellow());
|
print_warning(messages.tag_cancelled());
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if self.dry_run {
|
if self.dry_run {
|
||||||
println!("\n{} {}", messages.dry_run(), "- tag not created.".yellow());
|
println!("\n{}", messages.dry_run_tag_not_created().yellow());
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -156,13 +185,13 @@ impl TagCommand {
|
|||||||
|
|
||||||
builder.execute(&repo)?;
|
builder.execute(&repo)?;
|
||||||
|
|
||||||
println!("{} {}", messages.tag_created().green(), tag_name.cyan());
|
print_success(&format!("{} {}", messages.tag_created(), tag_name.cyan()));
|
||||||
|
|
||||||
// Push if requested or ask user
|
// Push if requested or ask user
|
||||||
if self.push {
|
if self.push {
|
||||||
println!("{}", messages.pushing_tag(&self.remote));
|
print_progress(&messages.pushing_tag(&self.remote));
|
||||||
repo.push(&self.remote, &format!("refs/tags/{}", tag_name))?;
|
repo.push(&self.remote, &format!("refs/tags/{}", tag_name))?;
|
||||||
println!("{}", messages.pushed_tag(&self.remote));
|
print_success(&messages.pushed_tag(&self.remote));
|
||||||
} else if !self.yes && !self.dry_run {
|
} else if !self.yes && !self.dry_run {
|
||||||
let should_push = Confirm::new()
|
let should_push = Confirm::new()
|
||||||
.with_prompt(messages.push_after_tag())
|
.with_prompt(messages.push_after_tag())
|
||||||
@@ -170,16 +199,21 @@ impl TagCommand {
|
|||||||
.interact()?;
|
.interact()?;
|
||||||
|
|
||||||
if should_push {
|
if should_push {
|
||||||
println!("{}", messages.pushing_tag(&self.remote));
|
print_progress(&messages.pushing_tag(&self.remote));
|
||||||
repo.push(&self.remote, &format!("refs/tags/{}", tag_name))?;
|
repo.push(&self.remote, &format!("refs/tags/{}", tag_name))?;
|
||||||
println!("{}", messages.pushed_tag(&self.remote));
|
print_success(&messages.pushed_tag(&self.remote));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn select_version_interactive(&self, repo: &GitRepo, prefix: &str, messages: &Messages) -> Result<String> {
|
async fn select_version_interactive(
|
||||||
|
&self,
|
||||||
|
repo: &GitRepo,
|
||||||
|
prefix: &str,
|
||||||
|
messages: &Messages,
|
||||||
|
) -> Result<String> {
|
||||||
loop {
|
loop {
|
||||||
let latest = get_latest_version(repo, prefix)?;
|
let latest = get_latest_version(repo, prefix)?;
|
||||||
|
|
||||||
@@ -211,11 +245,18 @@ impl TagCommand {
|
|||||||
// Auto-detect
|
// Auto-detect
|
||||||
let commits = repo.get_commits(50)?;
|
let commits = repo.get_commits(50)?;
|
||||||
let bump = suggest_version_bump(&commits);
|
let bump = suggest_version_bump(&commits);
|
||||||
let version = latest.as_ref()
|
let version = latest
|
||||||
|
.as_ref()
|
||||||
.map(|v| bump_version(v, bump, None))
|
.map(|v| bump_version(v, bump, None))
|
||||||
.unwrap_or_else(|| Version::new(0, 1, 0));
|
.unwrap_or_else(|| Version::new(0, 1, 0));
|
||||||
|
|
||||||
println!("{} {:?} → {}{}", messages.suggested_bump(), bump, prefix, version);
|
println!(
|
||||||
|
"{} {:?} → {}{}",
|
||||||
|
messages.suggested_bump(),
|
||||||
|
bump,
|
||||||
|
prefix,
|
||||||
|
version
|
||||||
|
);
|
||||||
|
|
||||||
let confirm = Confirm::new()
|
let confirm = Confirm::new()
|
||||||
.with_prompt(messages.use_this_version())
|
.with_prompt(messages.use_this_version())
|
||||||
@@ -228,19 +269,22 @@ impl TagCommand {
|
|||||||
// User rejected, continue the loop
|
// User rejected, continue the loop
|
||||||
}
|
}
|
||||||
1 => {
|
1 => {
|
||||||
let version = latest.as_ref()
|
let version = latest
|
||||||
|
.as_ref()
|
||||||
.map(|v| bump_version(v, VersionBump::Major, None))
|
.map(|v| bump_version(v, VersionBump::Major, None))
|
||||||
.unwrap_or_else(|| Version::new(1, 0, 0));
|
.unwrap_or_else(|| Version::new(1, 0, 0));
|
||||||
return Ok(format!("{}{}", prefix, version));
|
return Ok(format!("{}{}", prefix, version));
|
||||||
}
|
}
|
||||||
2 => {
|
2 => {
|
||||||
let version = latest.as_ref()
|
let version = latest
|
||||||
|
.as_ref()
|
||||||
.map(|v| bump_version(v, VersionBump::Minor, None))
|
.map(|v| bump_version(v, VersionBump::Minor, None))
|
||||||
.unwrap_or_else(|| Version::new(0, 1, 0));
|
.unwrap_or_else(|| Version::new(0, 1, 0));
|
||||||
return Ok(format!("{}{}", prefix, version));
|
return Ok(format!("{}{}", prefix, version));
|
||||||
}
|
}
|
||||||
3 => {
|
3 => {
|
||||||
let version = latest.as_ref()
|
let version = latest
|
||||||
|
.as_ref()
|
||||||
.map(|v| bump_version(v, VersionBump::Patch, None))
|
.map(|v| bump_version(v, VersionBump::Patch, None))
|
||||||
.unwrap_or_else(|| Version::new(0, 0, 1));
|
.unwrap_or_else(|| Version::new(0, 0, 1));
|
||||||
return Ok(format!("{}{}", prefix, version));
|
return Ok(format!("{}{}", prefix, version));
|
||||||
@@ -263,12 +307,15 @@ impl TagCommand {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn generate_tag_message(&self, repo: &GitRepo, version: &str, messages: &Messages) -> Result<String> {
|
async fn generate_tag_message(
|
||||||
|
&self,
|
||||||
|
repo: &GitRepo,
|
||||||
|
version: &str,
|
||||||
|
messages: &Messages,
|
||||||
|
) -> Result<String> {
|
||||||
let manager = ConfigManager::new()?;
|
let manager = ConfigManager::new()?;
|
||||||
let config = manager.config();
|
|
||||||
let language = manager.get_language().unwrap_or(Language::English);
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
|
|
||||||
// Get commits since last tag
|
|
||||||
let tags = repo.get_tags()?;
|
let tags = repo.get_tags()?;
|
||||||
let commits = if let Some(latest_tag) = tags.first() {
|
let commits = if let Some(latest_tag) = tags.first() {
|
||||||
repo.get_commits_between(&latest_tag.name, "HEAD")?
|
repo.get_commits_between(&latest_tag.name, "HEAD")?
|
||||||
@@ -277,17 +324,103 @@ impl TagCommand {
|
|||||||
};
|
};
|
||||||
|
|
||||||
if commits.is_empty() {
|
if commits.is_empty() {
|
||||||
return Ok(format!("Release {}", version));
|
return Ok(messages.release_default(version));
|
||||||
}
|
}
|
||||||
|
|
||||||
println!("{}", messages.ai_generating_tag(commits.len()));
|
let generator = ContentGenerator::new_with_think(&manager, self.think, None).await?;
|
||||||
|
let spinner = crate::utils::Spinner::start(&messages.ai_generating_tag(commits.len()));
|
||||||
|
let result = generator
|
||||||
|
.generate_tag_message(version, &commits, language)
|
||||||
|
.await;
|
||||||
|
spinner.finish_clear();
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
let generator = ContentGenerator::new(&config.llm).await?;
|
async fn auto_detect_version(
|
||||||
generator.generate_tag_message(version, &commits, language).await
|
&self,
|
||||||
|
repo: &GitRepo,
|
||||||
|
prefix: &str,
|
||||||
|
messages: &Messages,
|
||||||
|
) -> Result<String> {
|
||||||
|
let project_dir = std::env::current_dir()?;
|
||||||
|
|
||||||
|
// 1. Try reading from project config files
|
||||||
|
let config_versions = read_project_versions(&project_dir);
|
||||||
|
|
||||||
|
if !config_versions.is_empty() {
|
||||||
|
if config_versions.len() == 1 {
|
||||||
|
let cv = &config_versions[0];
|
||||||
|
let tag_name = format!("{}{}", prefix, cv.version);
|
||||||
|
println!(
|
||||||
|
"{} ({}): {}",
|
||||||
|
messages.found_version_in(),
|
||||||
|
cv.source,
|
||||||
|
tag_name.cyan()
|
||||||
|
);
|
||||||
|
return Ok(tag_name);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Multiple config files: let user choose
|
||||||
|
return self.select_config_version(&config_versions, prefix, messages);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Fall back to commit analysis
|
||||||
|
println!("{}", messages.auto_detect_bump());
|
||||||
|
let commits = repo.get_commits(50)?;
|
||||||
|
let bump = suggest_version_bump(&commits);
|
||||||
|
let latest = get_latest_version(repo, prefix)?.unwrap_or_else(|| Version::new(0, 0, 0));
|
||||||
|
let version = bump_version(&latest, bump, None);
|
||||||
|
let tag_name = format!("{}{}", prefix, version);
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"{} {:?} → {}",
|
||||||
|
messages.suggested_bump(),
|
||||||
|
bump,
|
||||||
|
tag_name.cyan()
|
||||||
|
);
|
||||||
|
|
||||||
|
if !self.yes {
|
||||||
|
let confirm = Confirm::new()
|
||||||
|
.with_prompt(messages.use_this_version())
|
||||||
|
.default(true)
|
||||||
|
.interact()?;
|
||||||
|
|
||||||
|
if !confirm {
|
||||||
|
// Fall through to interactive version selection
|
||||||
|
return self
|
||||||
|
.select_version_interactive(repo, prefix, messages)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(tag_name)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn select_config_version(
|
||||||
|
&self,
|
||||||
|
config_versions: &[ConfigVersion],
|
||||||
|
prefix: &str,
|
||||||
|
messages: &Messages,
|
||||||
|
) -> Result<String> {
|
||||||
|
println!("\n{}", messages.found_multiple_versions().bold());
|
||||||
|
|
||||||
|
let items: Vec<String> = config_versions
|
||||||
|
.iter()
|
||||||
|
.map(|cv| format!("{} → {}{}", cv.source, prefix, cv.version))
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
let selection = Select::new()
|
||||||
|
.with_prompt(messages.select_version_to_use())
|
||||||
|
.items(&items)
|
||||||
|
.default(0)
|
||||||
|
.interact()?;
|
||||||
|
|
||||||
|
let cv = &config_versions[selection];
|
||||||
|
Ok(format!("{}{}", prefix, cv.version))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn input_message_interactive(&self, version: &str, messages: &Messages) -> Result<String> {
|
fn input_message_interactive(&self, version: &str, messages: &Messages) -> Result<String> {
|
||||||
let default_msg = format!("Release {}", version);
|
let default_msg = messages.release_default(version);
|
||||||
|
|
||||||
let use_editor = Confirm::new()
|
let use_editor = Confirm::new()
|
||||||
.with_prompt(messages.open_editor())
|
.with_prompt(messages.open_editor())
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
use super::{AppConfig, GitProfile, TokenConfig};
|
use super::{AppConfig, GitProfile, TokenConfig};
|
||||||
use anyhow::{bail, Context, Result};
|
use crate::utils::keyring::{
|
||||||
use std::collections::HashMap;
|
KeyringManager, get_default_base_url, get_default_model, provider_needs_api_key,
|
||||||
|
};
|
||||||
|
use anyhow::{Context, Result, bail};
|
||||||
|
// use std::collections::HashMap;
|
||||||
use std::path::{Path, PathBuf};
|
use std::path::{Path, PathBuf};
|
||||||
|
|
||||||
/// Configuration manager
|
/// Configuration manager
|
||||||
@@ -8,6 +11,7 @@ pub struct ConfigManager {
|
|||||||
config: AppConfig,
|
config: AppConfig,
|
||||||
config_path: PathBuf,
|
config_path: PathBuf,
|
||||||
modified: bool,
|
modified: bool,
|
||||||
|
keyring: KeyringManager,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ConfigManager {
|
impl ConfigManager {
|
||||||
@@ -19,11 +23,26 @@ impl ConfigManager {
|
|||||||
|
|
||||||
/// Create config manager with specific path
|
/// Create config manager with specific path
|
||||||
pub fn with_path(path: &Path) -> Result<Self> {
|
pub fn with_path(path: &Path) -> Result<Self> {
|
||||||
let config = AppConfig::load(path)?;
|
let config = if path.exists() {
|
||||||
|
AppConfig::load(path)?
|
||||||
|
} else {
|
||||||
|
AppConfig::default()
|
||||||
|
};
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
config,
|
config,
|
||||||
config_path: path.to_path_buf(),
|
config_path: path.to_path_buf(),
|
||||||
modified: false,
|
modified: false,
|
||||||
|
keyring: KeyringManager::new(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create config manager with fresh config (ignoring existing)
|
||||||
|
pub fn with_path_fresh(path: &Path) -> Result<Self> {
|
||||||
|
Ok(Self {
|
||||||
|
config: AppConfig::default(),
|
||||||
|
config_path: path.to_path_buf(),
|
||||||
|
modified: true,
|
||||||
|
keyring: KeyringManager::new(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -47,10 +66,10 @@ impl ConfigManager {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Force save configuration
|
// /// Force save configuration
|
||||||
pub fn force_save(&self) -> Result<()> {
|
// pub fn force_save(&self) -> Result<()> {
|
||||||
self.config.save(&self.config_path)
|
// self.config.save(&self.config_path)
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// Get configuration file path
|
/// Get configuration file path
|
||||||
pub fn path(&self) -> &Path {
|
pub fn path(&self) -> &Path {
|
||||||
@@ -101,11 +120,11 @@ impl ConfigManager {
|
|||||||
self.config.profiles.get(name)
|
self.config.profiles.get(name)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get mutable profile
|
// /// Get mutable profile
|
||||||
pub fn get_profile_mut(&mut self, name: &str) -> Option<&mut GitProfile> {
|
// pub fn get_profile_mut(&mut self, name: &str) -> Option<&mut GitProfile> {
|
||||||
self.modified = true;
|
// self.modified = true;
|
||||||
self.config.profiles.get_mut(name)
|
// self.config.profiles.get_mut(name)
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// List all profile names
|
/// List all profile names
|
||||||
pub fn list_profiles(&self) -> Vec<&String> {
|
pub fn list_profiles(&self) -> Vec<&String> {
|
||||||
@@ -119,10 +138,10 @@ impl ConfigManager {
|
|||||||
|
|
||||||
/// Set default profile
|
/// Set default profile
|
||||||
pub fn set_default_profile(&mut self, name: Option<String>) -> Result<()> {
|
pub fn set_default_profile(&mut self, name: Option<String>) -> Result<()> {
|
||||||
if let Some(ref n) = name {
|
if let Some(ref n) = name
|
||||||
if !self.config.profiles.contains_key(n) {
|
&& !self.config.profiles.contains_key(n)
|
||||||
bail!("Profile '{}' does not exist", n);
|
{
|
||||||
}
|
bail!("Profile '{}' does not exist", n);
|
||||||
}
|
}
|
||||||
self.config.default_profile = name;
|
self.config.default_profile = name;
|
||||||
self.modified = true;
|
self.modified = true;
|
||||||
@@ -153,54 +172,154 @@ impl ConfigManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get profile usage statistics
|
// /// Get profile usage statistics
|
||||||
pub fn get_profile_usage(&self, name: &str) -> Option<&super::UsageStats> {
|
// pub fn get_profile_usage(&self, name: &str) -> Option<&super::UsageStats> {
|
||||||
self.config.profiles.get(name).map(|p| &p.usage)
|
// self.config.profiles.get(name).map(|p| &p.usage)
|
||||||
}
|
// }
|
||||||
|
|
||||||
// Token management
|
// Token management
|
||||||
|
|
||||||
/// Add a token to a profile
|
/// Add a token to a profile (stores token in keyring)
|
||||||
pub fn add_token_to_profile(&mut self, profile_name: &str, service: String, token: TokenConfig) -> Result<()> {
|
pub fn add_token_to_profile(
|
||||||
|
&mut self,
|
||||||
|
profile_name: &str,
|
||||||
|
service: String,
|
||||||
|
token: TokenConfig,
|
||||||
|
) -> Result<()> {
|
||||||
|
if !self.config.profiles.contains_key(profile_name) {
|
||||||
|
bail!("Profile '{}' does not exist", profile_name);
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(profile) = self.config.profiles.get_mut(profile_name) {
|
if let Some(profile) = self.config.profiles.get_mut(profile_name) {
|
||||||
profile.add_token(service, token);
|
profile.add_token(service, token);
|
||||||
self.modified = true;
|
self.modified = true;
|
||||||
Ok(())
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Store a PAT token in keyring for a profile
|
||||||
|
pub fn store_pat_for_profile(
|
||||||
|
&self,
|
||||||
|
profile_name: &str,
|
||||||
|
service: &str,
|
||||||
|
token_value: &str,
|
||||||
|
) -> Result<()> {
|
||||||
|
let profile = self
|
||||||
|
.get_profile(profile_name)
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?;
|
||||||
|
|
||||||
|
let user_email = &profile.user_email;
|
||||||
|
|
||||||
|
self.keyring
|
||||||
|
.store_pat(profile_name, user_email, service, token_value)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get a PAT token from keyring for a profile
|
||||||
|
pub fn get_pat_for_profile(&self, profile_name: &str, service: &str) -> Result<Option<String>> {
|
||||||
|
let profile = self
|
||||||
|
.get_profile(profile_name)
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?;
|
||||||
|
|
||||||
|
let user_email = &profile.user_email;
|
||||||
|
|
||||||
|
self.keyring.get_pat(profile_name, user_email, service)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Check if a PAT token exists for a profile
|
||||||
|
pub fn has_pat_for_profile(&self, profile_name: &str, service: &str) -> bool {
|
||||||
|
if let Some(profile) = self.get_profile(profile_name) {
|
||||||
|
let user_email = &profile.user_email;
|
||||||
|
self.keyring.has_pat(profile_name, user_email, service)
|
||||||
} else {
|
} else {
|
||||||
bail!("Profile '{}' does not exist", profile_name);
|
false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get a token from a profile
|
/// Remove a token from a profile (deletes from keyring)
|
||||||
pub fn get_token_from_profile(&self, profile_name: &str, service: &str) -> Option<&TokenConfig> {
|
|
||||||
self.config.profiles.get(profile_name)?.get_token(service)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Remove a token from a profile
|
|
||||||
pub fn remove_token_from_profile(&mut self, profile_name: &str, service: &str) -> Result<()> {
|
pub fn remove_token_from_profile(&mut self, profile_name: &str, service: &str) -> Result<()> {
|
||||||
|
if !self.config.profiles.contains_key(profile_name) {
|
||||||
|
bail!("Profile '{}' does not exist", profile_name);
|
||||||
|
}
|
||||||
|
|
||||||
|
let user_email = self
|
||||||
|
.config
|
||||||
|
.profiles
|
||||||
|
.get(profile_name)
|
||||||
|
.unwrap()
|
||||||
|
.user_email
|
||||||
|
.clone();
|
||||||
|
let services: Vec<String> = self
|
||||||
|
.config
|
||||||
|
.profiles
|
||||||
|
.get(profile_name)
|
||||||
|
.unwrap()
|
||||||
|
.tokens
|
||||||
|
.keys()
|
||||||
|
.cloned()
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
if !services.contains(&service.to_string()) {
|
||||||
|
bail!(
|
||||||
|
"Token for service '{}' not found in profile '{}'",
|
||||||
|
service,
|
||||||
|
profile_name
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
self.keyring
|
||||||
|
.delete_pat(profile_name, &user_email, service)?;
|
||||||
|
|
||||||
if let Some(profile) = self.config.profiles.get_mut(profile_name) {
|
if let Some(profile) = self.config.profiles.get_mut(profile_name) {
|
||||||
profile.remove_token(service);
|
profile.remove_token(service);
|
||||||
self.modified = true;
|
self.modified = true;
|
||||||
Ok(())
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Delete a PAT from the keyring even when no config token entry exists.
|
||||||
|
/// Returns true if a keyring entry was deleted (issue 24).
|
||||||
|
pub fn delete_orphan_pat(&self, profile_name: &str, service: &str) -> Result<bool> {
|
||||||
|
if let Some(profile) = self.get_profile(profile_name) {
|
||||||
|
let user_email = &profile.user_email;
|
||||||
|
if self.keyring.has_pat(profile_name, user_email, service) {
|
||||||
|
self.keyring.delete_pat(profile_name, user_email, service)?;
|
||||||
|
Ok(true)
|
||||||
|
} else {
|
||||||
|
Ok(false)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
bail!("Profile '{}' does not exist", profile_name);
|
Ok(false)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// List all tokens in a profile
|
/// Delete all PAT tokens for a profile (used when removing a profile)
|
||||||
pub fn list_profile_tokens(&self, profile_name: &str) -> Option<Vec<&String>> {
|
pub fn delete_all_pats_for_profile(&self, profile_name: &str) -> Result<()> {
|
||||||
self.config.profiles.get(profile_name).map(|p| p.tokens.keys().collect())
|
if let Some(profile) = self.get_profile(profile_name) {
|
||||||
|
let user_email = &profile.user_email;
|
||||||
|
let services: Vec<String> = profile.tokens.keys().cloned().collect();
|
||||||
|
|
||||||
|
self.keyring
|
||||||
|
.delete_all_pats_for_profile(profile_name, user_email, &services)?;
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// /// List all tokens in a profile
|
||||||
|
// pub fn list_profile_tokens(&self, profile_name: &str) -> Option<Vec<&String>> {
|
||||||
|
// self.config.profiles.get(profile_name).map(|p| p.tokens.keys().collect())
|
||||||
|
// }
|
||||||
|
|
||||||
// Repository profile management
|
// Repository profile management
|
||||||
|
|
||||||
/// Get profile for repository
|
// /// Get profile for repository
|
||||||
pub fn get_repo_profile(&self, repo_path: &str) -> Option<&GitProfile> {
|
// pub fn get_repo_profile(&self, repo_path: &str) -> Option<&GitProfile> {
|
||||||
self.config
|
// self.config
|
||||||
.repo_profiles
|
// .repo_profiles
|
||||||
.get(repo_path)
|
// .get(repo_path)
|
||||||
.and_then(|name| self.config.profiles.get(name))
|
// .and_then(|name| self.config.profiles.get(name))
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// Set profile for repository
|
/// Set profile for repository
|
||||||
pub fn set_repo_profile(&mut self, repo_path: String, profile_name: String) -> Result<()> {
|
pub fn set_repo_profile(&mut self, repo_path: String, profile_name: String) -> Result<()> {
|
||||||
@@ -212,32 +331,91 @@ impl ConfigManager {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Remove repository profile mapping
|
// /// Remove repository profile mapping
|
||||||
pub fn remove_repo_profile(&mut self, repo_path: &str) {
|
// pub fn remove_repo_profile(&mut self, repo_path: &str) {
|
||||||
self.config.repo_profiles.remove(repo_path);
|
// self.config.repo_profiles.remove(repo_path);
|
||||||
self.modified = true;
|
// self.modified = true;
|
||||||
|
// }
|
||||||
|
|
||||||
|
// /// List repository profile mappings
|
||||||
|
// pub fn list_repo_profiles(&self) -> &HashMap<String, String> {
|
||||||
|
// &self.config.repo_profiles
|
||||||
|
// }
|
||||||
|
|
||||||
|
// /// Get effective profile for a repository (repo-specific -> default)
|
||||||
|
// pub fn get_effective_profile(&self, repo_path: Option<&str>) -> Option<&GitProfile> {
|
||||||
|
// if let Some(path) = repo_path {
|
||||||
|
// if let Some(profile) = self.get_repo_profile(path) {
|
||||||
|
// return Some(profile);
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// self.default_profile()
|
||||||
|
// }
|
||||||
|
|
||||||
|
/// Check and compare profile with git configuration
|
||||||
|
pub fn check_profile_config(
|
||||||
|
&self,
|
||||||
|
profile_name: &str,
|
||||||
|
repo: &git2::Repository,
|
||||||
|
) -> Result<super::ProfileComparison> {
|
||||||
|
let profile = self
|
||||||
|
.get_profile(profile_name)
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?;
|
||||||
|
profile.compare_with_git_config(repo)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// List repository profile mappings
|
/// Find a profile that matches the given user config (name, email, signing_key)
|
||||||
pub fn list_repo_profiles(&self) -> &HashMap<String, String> {
|
pub fn find_matching_profile(
|
||||||
&self.config.repo_profiles
|
&self,
|
||||||
}
|
user_name: &str,
|
||||||
|
user_email: &str,
|
||||||
|
signing_key: Option<&str>,
|
||||||
|
) -> Option<&GitProfile> {
|
||||||
|
for profile in self.config.profiles.values() {
|
||||||
|
let name_match = profile.user_name == user_name;
|
||||||
|
let email_match = profile.user_email == user_email;
|
||||||
|
let key_match = match (signing_key, profile.signing_key()) {
|
||||||
|
(Some(git_key), Some(profile_key)) => git_key == profile_key,
|
||||||
|
(None, None) => true,
|
||||||
|
(Some(_), None) => false,
|
||||||
|
(None, Some(_)) => false,
|
||||||
|
};
|
||||||
|
|
||||||
/// Get effective profile for a repository (repo-specific -> default)
|
if name_match && email_match && key_match {
|
||||||
pub fn get_effective_profile(&self, repo_path: Option<&str>) -> Option<&GitProfile> {
|
|
||||||
if let Some(path) = repo_path {
|
|
||||||
if let Some(profile) = self.get_repo_profile(path) {
|
|
||||||
return Some(profile);
|
return Some(profile);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
self.default_profile()
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Check and compare profile with git configuration
|
/// Find profiles that partially match (same name or same email)
|
||||||
pub fn check_profile_config(&self, profile_name: &str, repo: &git2::Repository) -> Result<super::ProfileComparison> {
|
pub fn find_partial_matches(&self, user_name: &str, user_email: &str) -> Vec<&GitProfile> {
|
||||||
let profile = self.get_profile(profile_name)
|
self.config
|
||||||
.ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?;
|
.profiles
|
||||||
profile.compare_with_git_config(repo)
|
.values()
|
||||||
|
.filter(|p| p.user_name == user_name || p.user_email == user_email)
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get repo profile mapping
|
||||||
|
pub fn get_repo_profile_name(&self, repo_path: &str) -> Option<&String> {
|
||||||
|
self.config.repo_profiles.get(repo_path)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Find profiles whose repo_patterns match the given repo path
|
||||||
|
pub fn match_profiles_by_repo_pattern(&self, repo_path: &str) -> Vec<&GitProfile> {
|
||||||
|
self.config
|
||||||
|
.profiles
|
||||||
|
.values()
|
||||||
|
.filter(|p| {
|
||||||
|
p.settings.repo_patterns.iter().any(|pattern| {
|
||||||
|
let trimmed = pattern.trim_matches('*');
|
||||||
|
repo_path.ends_with(trimmed)
|
||||||
|
|| repo_path.starts_with(trimmed)
|
||||||
|
|| repo_path == trimmed
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
}
|
}
|
||||||
|
|
||||||
// LLM configuration
|
// LLM configuration
|
||||||
@@ -249,104 +427,169 @@ impl ConfigManager {
|
|||||||
|
|
||||||
/// Set LLM provider
|
/// Set LLM provider
|
||||||
pub fn set_llm_provider(&mut self, provider: String) {
|
pub fn set_llm_provider(&mut self, provider: String) {
|
||||||
self.config.llm.provider = provider;
|
let default_model = get_default_model(&provider);
|
||||||
|
self.config.llm.provider = provider.clone();
|
||||||
|
if self.config.llm.model.is_empty() || self.config.llm.model == "llama2" {
|
||||||
|
self.config.llm.model = default_model.to_string();
|
||||||
|
}
|
||||||
self.modified = true;
|
self.modified = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get OpenAI API key
|
/// Get model
|
||||||
pub fn openai_api_key(&self) -> Option<&String> {
|
pub fn llm_model(&self) -> &str {
|
||||||
self.config.llm.openai.api_key.as_ref()
|
&self.config.llm.model
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set OpenAI API key
|
/// Set model
|
||||||
pub fn set_openai_api_key(&mut self, key: String) {
|
pub fn set_llm_model(&mut self, model: String) {
|
||||||
self.config.llm.openai.api_key = Some(key);
|
self.config.llm.model = model;
|
||||||
self.modified = true;
|
self.modified = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get Anthropic API key
|
/// Get base URL (returns provider default if not set)
|
||||||
pub fn anthropic_api_key(&self) -> Option<&String> {
|
pub fn llm_base_url(&self) -> String {
|
||||||
self.config.llm.anthropic.api_key.as_ref()
|
match &self.config.llm.base_url {
|
||||||
|
Some(url) => url.clone(),
|
||||||
|
None => get_default_base_url(&self.config.llm.provider).to_string(),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set Anthropic API key
|
/// Set base URL
|
||||||
pub fn set_anthropic_api_key(&mut self, key: String) {
|
pub fn set_llm_base_url(&mut self, url: Option<String>) {
|
||||||
self.config.llm.anthropic.api_key = Some(key);
|
self.config.llm.base_url = url;
|
||||||
self.modified = true;
|
self.modified = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get Kimi API key
|
/// Get API key from configured storage method
|
||||||
pub fn kimi_api_key(&self) -> Option<&String> {
|
pub fn get_api_key(&self) -> Option<String> {
|
||||||
self.config.llm.kimi.api_key.as_ref()
|
// First try environment variables (always checked)
|
||||||
|
if let Some(key) = self
|
||||||
|
.keyring
|
||||||
|
.get_api_key(&self.config.llm.provider)
|
||||||
|
.unwrap_or(None)
|
||||||
|
{
|
||||||
|
return Some(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Then try config file if configured
|
||||||
|
if self.config.llm.api_key_storage == "config" {
|
||||||
|
return self.config.llm.api_key.clone();
|
||||||
|
}
|
||||||
|
|
||||||
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set Kimi API key
|
/// Store API key in configured storage method
|
||||||
pub fn set_kimi_api_key(&mut self, key: String) {
|
pub fn set_api_key(&self, api_key: &str) -> Result<()> {
|
||||||
self.config.llm.kimi.api_key = Some(key);
|
match self.config.llm.api_key_storage.as_str() {
|
||||||
self.modified = true;
|
"keyring" => {
|
||||||
|
if !self.keyring.is_available() {
|
||||||
|
bail!(
|
||||||
|
"Keyring is not available. Set QUICOMMIT_API_KEY environment variable instead or change api_key_storage to 'config'."
|
||||||
|
);
|
||||||
|
}
|
||||||
|
self.keyring
|
||||||
|
.store_api_key(&self.config.llm.provider, api_key)
|
||||||
|
}
|
||||||
|
"config" => {
|
||||||
|
// We can't modify self.config here since self is immutable
|
||||||
|
// This will be handled by the caller updating the config
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
"environment" => {
|
||||||
|
bail!(
|
||||||
|
"API key storage set to 'environment'. Please set QUICOMMIT_{}_API_KEY environment variable.",
|
||||||
|
self.config.llm.provider.to_uppercase()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
bail!(
|
||||||
|
"Invalid API key storage method: {}",
|
||||||
|
self.config.llm.api_key_storage
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get Kimi base URL
|
/// Delete API key from configured storage method
|
||||||
pub fn kimi_base_url(&self) -> &str {
|
pub fn delete_api_key(&self) -> Result<()> {
|
||||||
&self.config.llm.kimi.base_url
|
match self.config.llm.api_key_storage.as_str() {
|
||||||
|
"keyring" => {
|
||||||
|
if self.keyring.is_available() {
|
||||||
|
self.keyring.delete_api_key(&self.config.llm.provider)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"config" => {
|
||||||
|
// We can't modify self.config here since self is immutable
|
||||||
|
// This will be handled by the caller updating the config
|
||||||
|
}
|
||||||
|
"environment" => {
|
||||||
|
// Environment variables are not managed by the app
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
bail!(
|
||||||
|
"Invalid API key storage method: {}",
|
||||||
|
self.config.llm.api_key_storage
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set Kimi base URL
|
/// Check if API key is configured
|
||||||
pub fn set_kimi_base_url(&mut self, url: String) {
|
pub fn has_api_key(&self) -> bool {
|
||||||
self.config.llm.kimi.base_url = url;
|
if !provider_needs_api_key(&self.config.llm.provider) {
|
||||||
self.modified = true;
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check environment variables
|
||||||
|
if self
|
||||||
|
.keyring
|
||||||
|
.get_api_key(&self.config.llm.provider)
|
||||||
|
.unwrap_or(None)
|
||||||
|
.is_some()
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check config file if configured
|
||||||
|
if self.config.llm.api_key_storage == "config" {
|
||||||
|
return self.config.llm.api_key.is_some();
|
||||||
|
}
|
||||||
|
|
||||||
|
false
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get DeepSeek API key
|
/// Get keyring manager reference
|
||||||
pub fn deepseek_api_key(&self) -> Option<&String> {
|
pub fn keyring(&self) -> &KeyringManager {
|
||||||
self.config.llm.deepseek.api_key.as_ref()
|
&self.keyring
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set DeepSeek API key
|
// /// Configure LLM provider with all settings
|
||||||
pub fn set_deepseek_api_key(&mut self, key: String) {
|
// pub fn configure_llm(&mut self, provider: String, model: Option<String>, base_url: Option<String>, api_key: Option<&str>) -> Result<()> {
|
||||||
self.config.llm.deepseek.api_key = Some(key);
|
// self.set_llm_provider(provider.clone());
|
||||||
self.modified = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Get DeepSeek base URL
|
// if let Some(m) = model {
|
||||||
pub fn deepseek_base_url(&self) -> &str {
|
// self.set_llm_model(m);
|
||||||
&self.config.llm.deepseek.base_url
|
// }
|
||||||
}
|
|
||||||
|
|
||||||
/// Set DeepSeek base URL
|
// self.set_llm_base_url(base_url);
|
||||||
pub fn set_deepseek_base_url(&mut self, url: String) {
|
|
||||||
self.config.llm.deepseek.base_url = url;
|
|
||||||
self.modified = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Get OpenRouter API key
|
// if let Some(key) = api_key {
|
||||||
pub fn openrouter_api_key(&self) -> Option<&String> {
|
// if provider_needs_api_key(&provider) {
|
||||||
self.config.llm.openrouter.api_key.as_ref()
|
// self.set_api_key(key)?;
|
||||||
}
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
/// Set OpenRouter API key
|
// Ok(())
|
||||||
pub fn set_openrouter_api_key(&mut self, key: String) {
|
// }
|
||||||
self.config.llm.openrouter.api_key = Some(key);
|
|
||||||
self.modified = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Get OpenRouter base URL
|
|
||||||
pub fn openrouter_base_url(&self) -> &str {
|
|
||||||
&self.config.llm.openrouter.base_url
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set OpenRouter base URL
|
|
||||||
pub fn set_openrouter_base_url(&mut self, url: String) {
|
|
||||||
self.config.llm.openrouter.base_url = url;
|
|
||||||
self.modified = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Commit configuration
|
// Commit configuration
|
||||||
|
|
||||||
/// Get commit format
|
// /// Get commit format
|
||||||
pub fn commit_format(&self) -> super::CommitFormat {
|
// pub fn commit_format(&self) -> super::CommitFormat {
|
||||||
self.config.commit.format
|
// self.config.commit.format
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// Set commit format
|
/// Set commit format
|
||||||
pub fn set_commit_format(&mut self, format: super::CommitFormat) {
|
pub fn set_commit_format(&mut self, format: super::CommitFormat) {
|
||||||
@@ -354,10 +597,10 @@ impl ConfigManager {
|
|||||||
self.modified = true;
|
self.modified = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Check if auto-generate is enabled
|
// /// Check if auto-generate is enabled
|
||||||
pub fn auto_generate_commits(&self) -> bool {
|
// pub fn auto_generate_commits(&self) -> bool {
|
||||||
self.config.commit.auto_generate
|
// self.config.commit.auto_generate
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// Set auto-generate commits
|
/// Set auto-generate commits
|
||||||
pub fn set_auto_generate_commits(&mut self, enabled: bool) {
|
pub fn set_auto_generate_commits(&mut self, enabled: bool) {
|
||||||
@@ -367,10 +610,10 @@ impl ConfigManager {
|
|||||||
|
|
||||||
// Tag configuration
|
// Tag configuration
|
||||||
|
|
||||||
/// Get version prefix
|
// /// Get version prefix
|
||||||
pub fn version_prefix(&self) -> &str {
|
// pub fn version_prefix(&self) -> &str {
|
||||||
&self.config.tag.version_prefix
|
// &self.config.tag.version_prefix
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// Set version prefix
|
/// Set version prefix
|
||||||
pub fn set_version_prefix(&mut self, prefix: String) {
|
pub fn set_version_prefix(&mut self, prefix: String) {
|
||||||
@@ -380,10 +623,10 @@ impl ConfigManager {
|
|||||||
|
|
||||||
// Changelog configuration
|
// Changelog configuration
|
||||||
|
|
||||||
/// Get changelog path
|
// /// Get changelog path
|
||||||
pub fn changelog_path(&self) -> &str {
|
// pub fn changelog_path(&self) -> &str {
|
||||||
&self.config.changelog.path
|
// &self.config.changelog.path
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// Set changelog path
|
/// Set changelog path
|
||||||
pub fn set_changelog_path(&mut self, path: String) {
|
pub fn set_changelog_path(&mut self, path: String) {
|
||||||
@@ -393,10 +636,10 @@ impl ConfigManager {
|
|||||||
|
|
||||||
// Language configuration
|
// Language configuration
|
||||||
|
|
||||||
/// Get output language
|
// /// Get output language
|
||||||
pub fn output_language(&self) -> &str {
|
// pub fn output_language(&self) -> &str {
|
||||||
&self.config.language.output_language
|
// &self.config.language.output_language
|
||||||
}
|
// }
|
||||||
|
|
||||||
/// Set output language
|
/// Set output language
|
||||||
pub fn set_output_language(&mut self, language: String) {
|
pub fn set_output_language(&mut self, language: String) {
|
||||||
@@ -431,16 +674,25 @@ impl ConfigManager {
|
|||||||
self.modified = true;
|
self.modified = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Check if emoji decorations are enabled
|
||||||
|
pub fn emoji_enabled(&self) -> bool {
|
||||||
|
self.config.output.emoji
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Set emoji decorations flag
|
||||||
|
pub fn set_emoji_enabled(&mut self, enabled: bool) {
|
||||||
|
self.config.output.emoji = enabled;
|
||||||
|
self.modified = true;
|
||||||
|
}
|
||||||
|
|
||||||
/// Export configuration to TOML string
|
/// Export configuration to TOML string
|
||||||
pub fn export(&self) -> Result<String> {
|
pub fn export(&self) -> Result<String> {
|
||||||
toml::to_string_pretty(&self.config)
|
toml::to_string_pretty(&self.config).context("Failed to serialize config")
|
||||||
.context("Failed to serialize config")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Import configuration from TOML string
|
/// Import configuration from TOML string
|
||||||
pub fn import(&mut self, toml_str: &str) -> Result<()> {
|
pub fn import(&mut self, toml_str: &str) -> Result<()> {
|
||||||
self.config = toml::from_str(toml_str)
|
self.config = toml::from_str(toml_str).context("Failed to parse config")?;
|
||||||
.context("Failed to parse config")?;
|
|
||||||
self.modified = true;
|
self.modified = true;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@@ -458,6 +710,7 @@ impl Default for ConfigManager {
|
|||||||
config: AppConfig::default(),
|
config: AppConfig::default(),
|
||||||
config_path: PathBuf::new(),
|
config_path: PathBuf::new(),
|
||||||
modified: false,
|
modified: false,
|
||||||
|
keyring: KeyringManager::new(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,10 +7,7 @@ use std::path::{Path, PathBuf};
|
|||||||
pub mod manager;
|
pub mod manager;
|
||||||
pub mod profile;
|
pub mod profile;
|
||||||
|
|
||||||
pub use profile::{
|
pub use profile::{GitProfile, ProfileComparison, TokenConfig, TokenType};
|
||||||
GitProfile, TokenConfig, TokenType,
|
|
||||||
UsageStats, ProfileComparison
|
|
||||||
};
|
|
||||||
|
|
||||||
/// Application configuration
|
/// Application configuration
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
@@ -20,10 +17,11 @@ pub struct AppConfig {
|
|||||||
pub version: String,
|
pub version: String,
|
||||||
|
|
||||||
/// Default profile name
|
/// Default profile name
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
pub default_profile: Option<String>,
|
pub default_profile: Option<String>,
|
||||||
|
|
||||||
/// All configured profiles
|
/// All configured profiles
|
||||||
#[serde(default)]
|
#[serde(default, skip_serializing_if = "HashMap::is_empty")]
|
||||||
pub profiles: HashMap<String, GitProfile>,
|
pub profiles: HashMap<String, GitProfile>,
|
||||||
|
|
||||||
/// LLM configuration
|
/// LLM configuration
|
||||||
@@ -43,20 +41,16 @@ pub struct AppConfig {
|
|||||||
pub changelog: ChangelogConfig,
|
pub changelog: ChangelogConfig,
|
||||||
|
|
||||||
/// Repository-specific profile mappings
|
/// Repository-specific profile mappings
|
||||||
#[serde(default)]
|
#[serde(default, skip_serializing_if = "HashMap::is_empty")]
|
||||||
pub repo_profiles: HashMap<String, String>,
|
pub repo_profiles: HashMap<String, String>,
|
||||||
|
|
||||||
/// Whether to encrypt sensitive data
|
|
||||||
#[serde(default = "default_true")]
|
|
||||||
pub encrypt_sensitive: bool,
|
|
||||||
|
|
||||||
/// Theme settings
|
|
||||||
#[serde(default)]
|
|
||||||
pub theme: ThemeConfig,
|
|
||||||
|
|
||||||
/// Language settings
|
/// Language settings
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub language: LanguageConfig,
|
pub language: LanguageConfig,
|
||||||
|
|
||||||
|
/// Output settings
|
||||||
|
#[serde(default)]
|
||||||
|
pub output: OutputConfig,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for AppConfig {
|
impl Default for AppConfig {
|
||||||
@@ -70,9 +64,8 @@ impl Default for AppConfig {
|
|||||||
tag: TagConfig::default(),
|
tag: TagConfig::default(),
|
||||||
changelog: ChangelogConfig::default(),
|
changelog: ChangelogConfig::default(),
|
||||||
repo_profiles: HashMap::new(),
|
repo_profiles: HashMap::new(),
|
||||||
encrypt_sensitive: true,
|
|
||||||
theme: ThemeConfig::default(),
|
|
||||||
language: LanguageConfig::default(),
|
language: LanguageConfig::default(),
|
||||||
|
output: OutputConfig::default(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -80,37 +73,17 @@ impl Default for AppConfig {
|
|||||||
/// LLM configuration
|
/// LLM configuration
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct LlmConfig {
|
pub struct LlmConfig {
|
||||||
/// Default LLM provider
|
/// Current LLM provider (ollama, openai, anthropic, kimi, deepseek, openrouter)
|
||||||
#[serde(default = "default_llm_provider")]
|
#[serde(default = "default_llm_provider")]
|
||||||
pub provider: String,
|
pub provider: String,
|
||||||
|
|
||||||
/// OpenAI configuration
|
/// Model to use
|
||||||
#[serde(default)]
|
#[serde(default = "default_model")]
|
||||||
pub openai: OpenAiConfig,
|
pub model: String,
|
||||||
|
|
||||||
/// Ollama configuration
|
/// API base URL (optional, will use provider default if not set)
|
||||||
#[serde(default)]
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
pub ollama: OllamaConfig,
|
pub base_url: Option<String>,
|
||||||
|
|
||||||
/// Anthropic Claude configuration
|
|
||||||
#[serde(default)]
|
|
||||||
pub anthropic: AnthropicConfig,
|
|
||||||
|
|
||||||
/// Kimi (Moonshot AI) configuration
|
|
||||||
#[serde(default)]
|
|
||||||
pub kimi: KimiConfig,
|
|
||||||
|
|
||||||
/// DeepSeek configuration
|
|
||||||
#[serde(default)]
|
|
||||||
pub deepseek: DeepSeekConfig,
|
|
||||||
|
|
||||||
/// OpenRouter configuration
|
|
||||||
#[serde(default)]
|
|
||||||
pub openrouter: OpenRouterConfig,
|
|
||||||
|
|
||||||
/// Custom API configuration
|
|
||||||
#[serde(default)]
|
|
||||||
pub custom: Option<CustomLlmConfig>,
|
|
||||||
|
|
||||||
/// Maximum tokens for generation
|
/// Maximum tokens for generation
|
||||||
#[serde(default = "default_max_tokens")]
|
#[serde(default = "default_max_tokens")]
|
||||||
@@ -123,186 +96,45 @@ pub struct LlmConfig {
|
|||||||
/// Timeout in seconds
|
/// Timeout in seconds
|
||||||
#[serde(default = "default_timeout")]
|
#[serde(default = "default_timeout")]
|
||||||
pub timeout: u64,
|
pub timeout: u64,
|
||||||
|
|
||||||
|
/// API key storage method (keyring, config, environment)
|
||||||
|
#[serde(default = "default_api_key_storage")]
|
||||||
|
pub api_key_storage: String,
|
||||||
|
|
||||||
|
/// API key (stored in config for fallback)
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
pub api_key: Option<String>,
|
||||||
|
|
||||||
|
/// Enable thinking/reasoning mode (deepseek, kimi, anthropic)
|
||||||
|
#[serde(default)]
|
||||||
|
pub thinking_enabled: bool,
|
||||||
|
|
||||||
|
/// Budget tokens for thinking mode (Anthropic Claude 4)
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
pub thinking_budget_tokens: Option<u32>,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn default_api_key_storage() -> String {
|
||||||
|
"keyring".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for LlmConfig {
|
impl Default for LlmConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
provider: default_llm_provider(),
|
provider: default_llm_provider(),
|
||||||
openai: OpenAiConfig::default(),
|
model: default_model(),
|
||||||
ollama: OllamaConfig::default(),
|
base_url: None,
|
||||||
anthropic: AnthropicConfig::default(),
|
|
||||||
kimi: KimiConfig::default(),
|
|
||||||
deepseek: DeepSeekConfig::default(),
|
|
||||||
openrouter: OpenRouterConfig::default(),
|
|
||||||
custom: None,
|
|
||||||
max_tokens: default_max_tokens(),
|
max_tokens: default_max_tokens(),
|
||||||
temperature: default_temperature(),
|
temperature: default_temperature(),
|
||||||
timeout: default_timeout(),
|
timeout: default_timeout(),
|
||||||
}
|
api_key_storage: default_api_key_storage(),
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// OpenAI API configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct OpenAiConfig {
|
|
||||||
/// API key
|
|
||||||
pub api_key: Option<String>,
|
|
||||||
|
|
||||||
/// Model to use
|
|
||||||
#[serde(default = "default_openai_model")]
|
|
||||||
pub model: String,
|
|
||||||
|
|
||||||
/// API base URL (for custom endpoints)
|
|
||||||
#[serde(default = "default_openai_base_url")]
|
|
||||||
pub base_url: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for OpenAiConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
api_key: None,
|
api_key: None,
|
||||||
model: default_openai_model(),
|
thinking_enabled: false,
|
||||||
base_url: default_openai_base_url(),
|
thinking_budget_tokens: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ollama configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct OllamaConfig {
|
|
||||||
/// Ollama server URL
|
|
||||||
#[serde(default = "default_ollama_url")]
|
|
||||||
pub url: String,
|
|
||||||
|
|
||||||
/// Model to use
|
|
||||||
#[serde(default = "default_ollama_model")]
|
|
||||||
pub model: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for OllamaConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
url: default_ollama_url(),
|
|
||||||
model: default_ollama_model(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Anthropic Claude configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct AnthropicConfig {
|
|
||||||
/// API key
|
|
||||||
pub api_key: Option<String>,
|
|
||||||
|
|
||||||
/// Model to use
|
|
||||||
#[serde(default = "default_anthropic_model")]
|
|
||||||
pub model: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for AnthropicConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
api_key: None,
|
|
||||||
model: default_anthropic_model(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Kimi (Moonshot AI) configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct KimiConfig {
|
|
||||||
/// API key
|
|
||||||
pub api_key: Option<String>,
|
|
||||||
|
|
||||||
/// Model to use
|
|
||||||
#[serde(default = "default_kimi_model")]
|
|
||||||
pub model: String,
|
|
||||||
|
|
||||||
/// API base URL (for custom endpoints)
|
|
||||||
#[serde(default = "default_kimi_base_url")]
|
|
||||||
pub base_url: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for KimiConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
api_key: None,
|
|
||||||
model: default_kimi_model(),
|
|
||||||
base_url: default_kimi_base_url(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// DeepSeek configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct DeepSeekConfig {
|
|
||||||
/// API key
|
|
||||||
pub api_key: Option<String>,
|
|
||||||
|
|
||||||
/// Model to use
|
|
||||||
#[serde(default = "default_deepseek_model")]
|
|
||||||
pub model: String,
|
|
||||||
|
|
||||||
/// API base URL (for custom endpoints)
|
|
||||||
#[serde(default = "default_deepseek_base_url")]
|
|
||||||
pub base_url: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for DeepSeekConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
api_key: None,
|
|
||||||
model: default_deepseek_model(),
|
|
||||||
base_url: default_deepseek_base_url(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// OpenRouter configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct OpenRouterConfig {
|
|
||||||
/// API key
|
|
||||||
pub api_key: Option<String>,
|
|
||||||
|
|
||||||
/// Model to use
|
|
||||||
#[serde(default = "default_openrouter_model")]
|
|
||||||
pub model: String,
|
|
||||||
|
|
||||||
/// API base URL (for custom endpoints)
|
|
||||||
#[serde(default = "default_openrouter_base_url")]
|
|
||||||
pub base_url: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for OpenRouterConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
api_key: None,
|
|
||||||
model: default_openrouter_model(),
|
|
||||||
base_url: default_openrouter_base_url(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Custom LLM API configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct CustomLlmConfig {
|
|
||||||
/// API endpoint URL
|
|
||||||
pub url: String,
|
|
||||||
|
|
||||||
/// API key (optional)
|
|
||||||
pub api_key: Option<String>,
|
|
||||||
|
|
||||||
/// Model name
|
|
||||||
pub model: String,
|
|
||||||
|
|
||||||
/// Request format template (JSON)
|
|
||||||
pub request_template: String,
|
|
||||||
|
|
||||||
/// Response path to extract content (e.g., "choices.0.message.content")
|
|
||||||
pub response_path: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Commit configuration
|
/// Commit configuration
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct CommitConfig {
|
pub struct CommitConfig {
|
||||||
@@ -313,33 +145,6 @@ pub struct CommitConfig {
|
|||||||
/// Enable AI generation by default
|
/// Enable AI generation by default
|
||||||
#[serde(default = "default_true")]
|
#[serde(default = "default_true")]
|
||||||
pub auto_generate: bool,
|
pub auto_generate: bool,
|
||||||
|
|
||||||
/// Allow empty commits
|
|
||||||
#[serde(default)]
|
|
||||||
pub allow_empty: bool,
|
|
||||||
|
|
||||||
/// Sign commits with GPG
|
|
||||||
#[serde(default)]
|
|
||||||
pub gpg_sign: bool,
|
|
||||||
|
|
||||||
/// Default scope (optional)
|
|
||||||
pub default_scope: Option<String>,
|
|
||||||
|
|
||||||
/// Maximum subject length
|
|
||||||
#[serde(default = "default_max_subject_length")]
|
|
||||||
pub max_subject_length: usize,
|
|
||||||
|
|
||||||
/// Require scope
|
|
||||||
#[serde(default)]
|
|
||||||
pub require_scope: bool,
|
|
||||||
|
|
||||||
/// Require body for certain types
|
|
||||||
#[serde(default)]
|
|
||||||
pub require_body: bool,
|
|
||||||
|
|
||||||
/// Types that require body
|
|
||||||
#[serde(default = "default_body_required_types")]
|
|
||||||
pub body_required_types: Vec<String>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for CommitConfig {
|
impl Default for CommitConfig {
|
||||||
@@ -347,13 +152,6 @@ impl Default for CommitConfig {
|
|||||||
Self {
|
Self {
|
||||||
format: default_commit_format(),
|
format: default_commit_format(),
|
||||||
auto_generate: true,
|
auto_generate: true,
|
||||||
allow_empty: false,
|
|
||||||
gpg_sign: false,
|
|
||||||
default_scope: None,
|
|
||||||
max_subject_length: default_max_subject_length(),
|
|
||||||
require_scope: false,
|
|
||||||
require_body: false,
|
|
||||||
body_required_types: default_body_required_types(),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -385,18 +183,6 @@ pub struct TagConfig {
|
|||||||
/// Enable AI generation for tag messages
|
/// Enable AI generation for tag messages
|
||||||
#[serde(default = "default_true")]
|
#[serde(default = "default_true")]
|
||||||
pub auto_generate: bool,
|
pub auto_generate: bool,
|
||||||
|
|
||||||
/// Sign tags with GPG
|
|
||||||
#[serde(default)]
|
|
||||||
pub gpg_sign: bool,
|
|
||||||
|
|
||||||
/// Include changelog in annotated tags
|
|
||||||
#[serde(default = "default_true")]
|
|
||||||
pub include_changelog: bool,
|
|
||||||
|
|
||||||
/// Default annotation template
|
|
||||||
#[serde(default)]
|
|
||||||
pub annotation_template: Option<String>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for TagConfig {
|
impl Default for TagConfig {
|
||||||
@@ -404,9 +190,6 @@ impl Default for TagConfig {
|
|||||||
Self {
|
Self {
|
||||||
version_prefix: default_version_prefix(),
|
version_prefix: default_version_prefix(),
|
||||||
auto_generate: true,
|
auto_generate: true,
|
||||||
gpg_sign: false,
|
|
||||||
include_changelog: true,
|
|
||||||
annotation_template: None,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -421,26 +204,6 @@ pub struct ChangelogConfig {
|
|||||||
/// Enable AI generation for changelog entries
|
/// Enable AI generation for changelog entries
|
||||||
#[serde(default = "default_true")]
|
#[serde(default = "default_true")]
|
||||||
pub auto_generate: bool,
|
pub auto_generate: bool,
|
||||||
|
|
||||||
/// Changelog format
|
|
||||||
#[serde(default = "default_changelog_format")]
|
|
||||||
pub format: ChangelogFormat,
|
|
||||||
|
|
||||||
/// Include commit hashes
|
|
||||||
#[serde(default)]
|
|
||||||
pub include_hashes: bool,
|
|
||||||
|
|
||||||
/// Include authors
|
|
||||||
#[serde(default)]
|
|
||||||
pub include_authors: bool,
|
|
||||||
|
|
||||||
/// Group by type
|
|
||||||
#[serde(default = "default_true")]
|
|
||||||
pub group_by_type: bool,
|
|
||||||
|
|
||||||
/// Custom categories
|
|
||||||
#[serde(default)]
|
|
||||||
pub custom_categories: Vec<ChangelogCategory>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for ChangelogConfig {
|
impl Default for ChangelogConfig {
|
||||||
@@ -448,56 +211,6 @@ impl Default for ChangelogConfig {
|
|||||||
Self {
|
Self {
|
||||||
path: default_changelog_path(),
|
path: default_changelog_path(),
|
||||||
auto_generate: true,
|
auto_generate: true,
|
||||||
format: default_changelog_format(),
|
|
||||||
include_hashes: false,
|
|
||||||
include_authors: false,
|
|
||||||
group_by_type: true,
|
|
||||||
custom_categories: vec![],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Changelog format
|
|
||||||
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
|
|
||||||
#[serde(rename_all = "kebab-case")]
|
|
||||||
pub enum ChangelogFormat {
|
|
||||||
KeepAChangelog,
|
|
||||||
GitHubReleases,
|
|
||||||
Custom,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Changelog category mapping
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct ChangelogCategory {
|
|
||||||
/// Category title
|
|
||||||
pub title: String,
|
|
||||||
|
|
||||||
/// Commit types included in this category
|
|
||||||
pub types: Vec<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Theme configuration
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
pub struct ThemeConfig {
|
|
||||||
/// Enable colors
|
|
||||||
#[serde(default = "default_true")]
|
|
||||||
pub colors: bool,
|
|
||||||
|
|
||||||
/// Enable icons
|
|
||||||
#[serde(default = "default_true")]
|
|
||||||
pub icons: bool,
|
|
||||||
|
|
||||||
/// Preferred date format
|
|
||||||
#[serde(default = "default_date_format")]
|
|
||||||
pub date_format: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for ThemeConfig {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
colors: true,
|
|
||||||
icons: true,
|
|
||||||
date_format: default_date_format(),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -528,6 +241,20 @@ impl Default for LanguageConfig {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Output configuration
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
|
pub struct OutputConfig {
|
||||||
|
/// Show emoji/symbol decorations (✓/✗/⚠/→)
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
pub emoji: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for OutputConfig {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self { emoji: true }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Supported languages
|
/// Supported languages
|
||||||
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
||||||
pub enum Language {
|
pub enum Language {
|
||||||
@@ -580,6 +307,7 @@ impl Language {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Default value functions
|
// Default value functions
|
||||||
|
|
||||||
fn default_version() -> String {
|
fn default_version() -> String {
|
||||||
"1".to_string()
|
"1".to_string()
|
||||||
}
|
}
|
||||||
@@ -592,6 +320,10 @@ fn default_llm_provider() -> String {
|
|||||||
"ollama".to_string()
|
"ollama".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn default_model() -> String {
|
||||||
|
"llama2".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
fn default_max_tokens() -> u32 {
|
fn default_max_tokens() -> u32 {
|
||||||
500
|
500
|
||||||
}
|
}
|
||||||
@@ -604,62 +336,10 @@ fn default_timeout() -> u64 {
|
|||||||
30
|
30
|
||||||
}
|
}
|
||||||
|
|
||||||
fn default_openai_model() -> String {
|
|
||||||
"gpt-4".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_openai_base_url() -> String {
|
|
||||||
"https://api.openai.com/v1".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_ollama_url() -> String {
|
|
||||||
"http://localhost:11434".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_ollama_model() -> String {
|
|
||||||
"llama2".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_anthropic_model() -> String {
|
|
||||||
"claude-3-sonnet-20240229".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_kimi_model() -> String {
|
|
||||||
"moonshot-v1-8k".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_kimi_base_url() -> String {
|
|
||||||
"https://api.moonshot.cn/v1".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_deepseek_model() -> String {
|
|
||||||
"deepseek-chat".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_deepseek_base_url() -> String {
|
|
||||||
"https://api.deepseek.com/v1".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_openrouter_model() -> String {
|
|
||||||
"openai/gpt-3.5-turbo".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_openrouter_base_url() -> String {
|
|
||||||
"https://openrouter.ai/api/v1".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_commit_format() -> CommitFormat {
|
fn default_commit_format() -> CommitFormat {
|
||||||
CommitFormat::Conventional
|
CommitFormat::Conventional
|
||||||
}
|
}
|
||||||
|
|
||||||
fn default_max_subject_length() -> usize {
|
|
||||||
100
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_body_required_types() -> Vec<String> {
|
|
||||||
vec!["feat".to_string(), "fix".to_string()]
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_version_prefix() -> String {
|
fn default_version_prefix() -> String {
|
||||||
"v".to_string()
|
"v".to_string()
|
||||||
}
|
}
|
||||||
@@ -668,14 +348,6 @@ fn default_changelog_path() -> String {
|
|||||||
"CHANGELOG.md".to_string()
|
"CHANGELOG.md".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn default_changelog_format() -> ChangelogFormat {
|
|
||||||
ChangelogFormat::KeepAChangelog
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_date_format() -> String {
|
|
||||||
"%Y-%m-%d".to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_output_language() -> String {
|
fn default_output_language() -> String {
|
||||||
"en".to_string()
|
"en".to_string()
|
||||||
}
|
}
|
||||||
@@ -696,8 +368,7 @@ impl AppConfig {
|
|||||||
|
|
||||||
/// Save configuration to file
|
/// Save configuration to file
|
||||||
pub fn save(&self, path: &Path) -> Result<()> {
|
pub fn save(&self, path: &Path) -> Result<()> {
|
||||||
let content = toml::to_string_pretty(self)
|
let content = toml::to_string_pretty(self).context("Failed to serialize config")?;
|
||||||
.context("Failed to serialize config")?;
|
|
||||||
|
|
||||||
if let Some(parent) = path.parent() {
|
if let Some(parent) = path.parent() {
|
||||||
fs::create_dir_all(parent)
|
fs::create_dir_all(parent)
|
||||||
@@ -712,23 +383,59 @@ impl AppConfig {
|
|||||||
|
|
||||||
/// Get default config path
|
/// Get default config path
|
||||||
pub fn default_path() -> Result<PathBuf> {
|
pub fn default_path() -> Result<PathBuf> {
|
||||||
let config_dir = dirs::config_dir()
|
let config_dir = dirs::config_dir().context("Could not find config directory")?;
|
||||||
.context("Could not find config directory")?;
|
|
||||||
Ok(config_dir.join("quicommit").join("config.toml"))
|
Ok(config_dir.join("quicommit").join("config.toml"))
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Get profile for a repository
|
/// Encrypted PAT data for export
|
||||||
pub fn get_profile_for_repo(&self, repo_path: &str) -> Option<&GitProfile> {
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
let profile_name = self.repo_profiles.get(repo_path)?;
|
pub struct EncryptedPat {
|
||||||
self.profiles.get(profile_name)
|
/// Profile name
|
||||||
|
pub profile_name: String,
|
||||||
|
/// Service name (e.g., github, gitlab)
|
||||||
|
pub service: String,
|
||||||
|
/// User email (for keyring lookup)
|
||||||
|
pub user_email: String,
|
||||||
|
/// Encrypted token value
|
||||||
|
pub encrypted_token: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Export data container with optional encrypted PATs
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
|
pub struct ExportData {
|
||||||
|
/// Configuration content (TOML string)
|
||||||
|
pub config: String,
|
||||||
|
/// Encrypted PATs (only present when exporting with encryption)
|
||||||
|
#[serde(default, skip_serializing_if = "Vec::is_empty")]
|
||||||
|
pub encrypted_pats: Vec<EncryptedPat>,
|
||||||
|
/// Export version for future compatibility
|
||||||
|
#[serde(default = "default_export_version")]
|
||||||
|
pub export_version: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn default_export_version() -> String {
|
||||||
|
"1".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ExportData {
|
||||||
|
pub fn new(config: String) -> Self {
|
||||||
|
Self {
|
||||||
|
config,
|
||||||
|
encrypted_pats: Vec::new(),
|
||||||
|
export_version: default_export_version(),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set profile for a repository
|
pub fn with_encrypted_pats(config: String, pats: Vec<EncryptedPat>) -> Self {
|
||||||
pub fn set_profile_for_repo(&mut self, repo_path: String, profile_name: String) -> Result<()> {
|
Self {
|
||||||
if !self.profiles.contains_key(&profile_name) {
|
config,
|
||||||
anyhow::bail!("Profile '{}' does not exist", profile_name);
|
encrypted_pats: pats,
|
||||||
|
export_version: default_export_version(),
|
||||||
}
|
}
|
||||||
self.repo_profiles.insert(repo_path, profile_name);
|
}
|
||||||
Ok(())
|
|
||||||
|
pub fn has_encrypted_pats(&self) -> bool {
|
||||||
|
!self.encrypted_pats.is_empty()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
use anyhow::{bail, Result};
|
use anyhow::{Result, bail};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
@@ -120,11 +120,15 @@ impl GitProfile {
|
|||||||
/// Get signing key (from GPG config or direct)
|
/// Get signing key (from GPG config or direct)
|
||||||
pub fn signing_key(&self) -> Option<&str> {
|
pub fn signing_key(&self) -> Option<&str> {
|
||||||
self.signing_key
|
self.signing_key
|
||||||
.as_ref()
|
.as_deref()
|
||||||
.map(|s| s.as_str())
|
|
||||||
.or_else(|| self.gpg.as_ref().map(|g| g.key_id.as_str()))
|
.or_else(|| self.gpg.as_ref().map(|g| g.key_id.as_str()))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Get the commit template if set
|
||||||
|
pub fn commit_template(&self) -> Option<&str> {
|
||||||
|
self.settings.commit_template.as_deref()
|
||||||
|
}
|
||||||
|
|
||||||
/// Add a token to the profile
|
/// Add a token to the profile
|
||||||
pub fn add_token(&mut self, service: String, token: TokenConfig) {
|
pub fn add_token(&mut self, service: String, token: TokenConfig) {
|
||||||
self.tokens.insert(service, token);
|
self.tokens.insert(service, token);
|
||||||
@@ -160,43 +164,120 @@ impl GitProfile {
|
|||||||
pub fn apply_to_repo(&self, repo: &git2::Repository) -> Result<()> {
|
pub fn apply_to_repo(&self, repo: &git2::Repository) -> Result<()> {
|
||||||
let mut config = repo.config()?;
|
let mut config = repo.config()?;
|
||||||
|
|
||||||
config.set_str("user.name", &self.user_name)?;
|
// Clean up old managed keys that the new profile won't set
|
||||||
config.set_str("user.email", &self.user_email)?;
|
if self.ssh.as_ref().and_then(|s| s.git_ssh_command()).is_none() {
|
||||||
|
let _ = config.remove("core.sshCommand");
|
||||||
if let Some(key) = self.signing_key() {
|
}
|
||||||
config.set_str("user.signingkey", key)?;
|
if self.signing_key().is_none() {
|
||||||
|
let _ = config.remove("user.signingkey");
|
||||||
if self.settings.auto_sign_commits {
|
let _ = config.remove("commit.gpgsign");
|
||||||
config.set_bool("commit.gpgsign", true)?;
|
let _ = config.remove("tag.gpgsign");
|
||||||
}
|
}
|
||||||
|
if self.gpg.is_none() {
|
||||||
if self.settings.auto_sign_tags {
|
let _ = config.remove("gpg.program");
|
||||||
config.set_bool("tag.gpgsign", true)?;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(ref ssh) = self.ssh {
|
// Apply new values; track whether we've written past name/email for rollback
|
||||||
if let Some(ref key_path) = ssh.private_key_path {
|
let mut wrote_optional = false;
|
||||||
config.set_str("core.sshCommand",
|
let result = (|| -> Result<()> {
|
||||||
&format!("ssh -i {}", key_path.display()))?;
|
config.set_str("user.name", &self.user_name)?;
|
||||||
|
config.set_str("user.email", &self.user_email)?;
|
||||||
|
|
||||||
|
if let Some(ref gpg) = self.gpg {
|
||||||
|
config.set_str("gpg.program", &gpg.program)?;
|
||||||
|
wrote_optional = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if let Some(key) = self.signing_key() {
|
||||||
|
config.set_str("user.signingkey", key)?;
|
||||||
|
if self.settings.auto_sign_commits {
|
||||||
|
config.set_bool("commit.gpgsign", true)?;
|
||||||
|
}
|
||||||
|
if self.settings.auto_sign_tags {
|
||||||
|
config.set_bool("tag.gpgsign", true)?;
|
||||||
|
}
|
||||||
|
wrote_optional = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(ref ssh) = self.ssh {
|
||||||
|
if let Some(ssh_cmd) = ssh.git_ssh_command() {
|
||||||
|
config.set_str("core.sshCommand", &ssh_cmd)?;
|
||||||
|
wrote_optional = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
})();
|
||||||
|
|
||||||
|
if result.is_err() && wrote_optional {
|
||||||
|
let _ = config.remove("core.sshCommand");
|
||||||
|
let _ = config.remove("user.signingkey");
|
||||||
|
let _ = config.remove("commit.gpgsign");
|
||||||
|
let _ = config.remove("tag.gpgsign");
|
||||||
|
let _ = config.remove("gpg.program");
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
result
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Apply this profile globally
|
/// Apply this profile globally
|
||||||
pub fn apply_global(&self) -> Result<()> {
|
pub fn apply_global(&self) -> Result<()> {
|
||||||
let mut config = git2::Config::open_default()?;
|
let mut config = git2::Config::open_default()?;
|
||||||
|
|
||||||
config.set_str("user.name", &self.user_name)?;
|
// Clean up old managed keys that the new profile won't set
|
||||||
config.set_str("user.email", &self.user_email)?;
|
if self.ssh.as_ref().and_then(|s| s.git_ssh_command()).is_none() {
|
||||||
|
let _ = config.remove("core.sshCommand");
|
||||||
if let Some(key) = self.signing_key() {
|
}
|
||||||
config.set_str("user.signingkey", key)?;
|
if self.signing_key().is_none() {
|
||||||
|
let _ = config.remove("user.signingkey");
|
||||||
|
let _ = config.remove("commit.gpgsign");
|
||||||
|
let _ = config.remove("tag.gpgsign");
|
||||||
|
}
|
||||||
|
if self.gpg.is_none() {
|
||||||
|
let _ = config.remove("gpg.program");
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
// Apply new values; track whether we've written past name/email for rollback
|
||||||
|
let mut wrote_optional = false;
|
||||||
|
let result = (|| -> Result<()> {
|
||||||
|
config.set_str("user.name", &self.user_name)?;
|
||||||
|
config.set_str("user.email", &self.user_email)?;
|
||||||
|
|
||||||
|
if let Some(ref gpg) = self.gpg {
|
||||||
|
config.set_str("gpg.program", &gpg.program)?;
|
||||||
|
wrote_optional = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(key) = self.signing_key() {
|
||||||
|
config.set_str("user.signingkey", key)?;
|
||||||
|
if self.settings.auto_sign_commits {
|
||||||
|
config.set_bool("commit.gpgsign", true)?;
|
||||||
|
}
|
||||||
|
if self.settings.auto_sign_tags {
|
||||||
|
config.set_bool("tag.gpgsign", true)?;
|
||||||
|
}
|
||||||
|
wrote_optional = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(ref ssh) = self.ssh {
|
||||||
|
if let Some(ssh_cmd) = ssh.git_ssh_command() {
|
||||||
|
config.set_str("core.sshCommand", &ssh_cmd)?;
|
||||||
|
wrote_optional = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
})();
|
||||||
|
|
||||||
|
if result.is_err() && wrote_optional {
|
||||||
|
let _ = config.remove("core.sshCommand");
|
||||||
|
let _ = config.remove("user.signingkey");
|
||||||
|
let _ = config.remove("commit.gpgsign");
|
||||||
|
let _ = config.remove("tag.gpgsign");
|
||||||
|
let _ = config.remove("gpg.program");
|
||||||
|
}
|
||||||
|
|
||||||
|
result
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Compare with current git configuration
|
/// Compare with current git configuration
|
||||||
@@ -231,15 +312,15 @@ impl GitProfile {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(profile_key) = self.signing_key() {
|
if let Some(profile_key) = self.signing_key()
|
||||||
if git_signing_key.as_deref() != Some(profile_key) {
|
&& git_signing_key.as_deref() != Some(profile_key)
|
||||||
comparison.matches = false;
|
{
|
||||||
comparison.differences.push(ConfigDifference {
|
comparison.matches = false;
|
||||||
key: "user.signingkey".to_string(),
|
comparison.differences.push(ConfigDifference {
|
||||||
profile_value: profile_key.to_string(),
|
key: "user.signingkey".to_string(),
|
||||||
git_value: git_signing_key.unwrap_or_else(|| "<not set>".to_string()),
|
profile_value: profile_key.to_string(),
|
||||||
});
|
git_value: git_signing_key.unwrap_or_else(|| "<not set>".to_string()),
|
||||||
}
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(comparison)
|
Ok(comparison)
|
||||||
@@ -247,7 +328,7 @@ impl GitProfile {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Profile settings
|
/// Profile settings
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
|
||||||
pub struct ProfileSettings {
|
pub struct ProfileSettings {
|
||||||
/// Automatically sign commits
|
/// Automatically sign commits
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
@@ -274,19 +355,6 @@ pub struct ProfileSettings {
|
|||||||
pub commit_template: Option<String>,
|
pub commit_template: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for ProfileSettings {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
auto_sign_commits: false,
|
|
||||||
auto_sign_tags: false,
|
|
||||||
default_commit_format: None,
|
|
||||||
repo_patterns: vec![],
|
|
||||||
llm_provider: None,
|
|
||||||
commit_template: None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// SSH configuration
|
/// SSH configuration
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct SshConfig {
|
pub struct SshConfig {
|
||||||
@@ -316,29 +384,84 @@ pub struct SshConfig {
|
|||||||
impl SshConfig {
|
impl SshConfig {
|
||||||
/// Validate SSH configuration
|
/// Validate SSH configuration
|
||||||
pub fn validate(&self) -> Result<()> {
|
pub fn validate(&self) -> Result<()> {
|
||||||
if let Some(ref path) = self.private_key_path {
|
if let Some(ref path) = self.private_key_path
|
||||||
if !path.exists() {
|
&& !path.exists()
|
||||||
bail!("SSH private key does not exist: {:?}", path);
|
{
|
||||||
}
|
bail!("SSH private key does not exist: {:?}", path);
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(ref path) = self.public_key_path {
|
if let Some(ref path) = self.public_key_path
|
||||||
if !path.exists() {
|
&& !path.exists()
|
||||||
bail!("SSH public key does not exist: {:?}", path);
|
{
|
||||||
}
|
bail!("SSH public key does not exist: {:?}", path);
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(ref path) = self.known_hosts_file
|
||||||
|
&& !path.exists()
|
||||||
|
{
|
||||||
|
bail!("SSH known_hosts file does not exist: {:?}", path);
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get SSH command for git
|
/// Get the effective public key path, deriving from private key if not explicitly set
|
||||||
|
pub fn effective_public_key_path(&self) -> Option<std::path::PathBuf> {
|
||||||
|
self.public_key_path.clone().or_else(|| {
|
||||||
|
self.private_key_path.as_ref().map(|pk| {
|
||||||
|
let mut pub_path = pk.clone();
|
||||||
|
let ext = pk
|
||||||
|
.extension()
|
||||||
|
.map(|e| format!("{}.pub", e.to_string_lossy()))
|
||||||
|
.unwrap_or_else(|| "pub".to_string());
|
||||||
|
pub_path.set_extension(&ext);
|
||||||
|
pub_path
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the effective SSH command for git config
|
||||||
|
///
|
||||||
|
/// Priority: custom `ssh_command` > constructed from key/agent/known_hosts
|
||||||
pub fn git_ssh_command(&self) -> Option<String> {
|
pub fn git_ssh_command(&self) -> Option<String> {
|
||||||
if let Some(ref cmd) = self.ssh_command {
|
if let Some(ref cmd) = self.ssh_command {
|
||||||
Some(cmd.clone())
|
return Some(cmd.clone());
|
||||||
} else if let Some(ref key_path) = self.private_key_path {
|
}
|
||||||
Some(format!("ssh -i '{}'", key_path.display()))
|
|
||||||
} else {
|
let mut parts: Vec<String> = vec!["ssh".to_string()];
|
||||||
|
|
||||||
|
if self.agent_forwarding {
|
||||||
|
parts.push("-A".to_string());
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(ref key_path) = self.private_key_path {
|
||||||
|
let path_str = key_path.display().to_string();
|
||||||
|
#[cfg(target_os = "windows")]
|
||||||
|
{
|
||||||
|
parts.push(format!("-i \"{}\"", path_str.replace('\\', "/")));
|
||||||
|
}
|
||||||
|
#[cfg(not(target_os = "windows"))]
|
||||||
|
{
|
||||||
|
parts.push(format!("-i '{}'", path_str));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(ref known_hosts) = self.known_hosts_file {
|
||||||
|
let kh_str = known_hosts.display().to_string();
|
||||||
|
#[cfg(target_os = "windows")]
|
||||||
|
{
|
||||||
|
parts.push(format!("-o UserKnownHostsFile=\"{}\"", kh_str.replace('\\', "/")));
|
||||||
|
}
|
||||||
|
#[cfg(not(target_os = "windows"))]
|
||||||
|
{
|
||||||
|
parts.push(format!("-o UserKnownHostsFile='{}'", kh_str));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if parts.len() == 1 {
|
||||||
None
|
None
|
||||||
|
} else {
|
||||||
|
Some(parts.join(" "))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -382,10 +505,6 @@ impl GpgConfig {
|
|||||||
/// Token configuration for services (GitHub, GitLab, etc.)
|
/// Token configuration for services (GitHub, GitLab, etc.)
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct TokenConfig {
|
pub struct TokenConfig {
|
||||||
/// Token value (encrypted)
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
pub token: Option<String>,
|
|
||||||
|
|
||||||
/// Token type (personal, oauth, etc.)
|
/// Token type (personal, oauth, etc.)
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub token_type: TokenType,
|
pub token_type: TokenType,
|
||||||
@@ -405,25 +524,41 @@ pub struct TokenConfig {
|
|||||||
/// Description
|
/// Description
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub description: Option<String>,
|
pub description: Option<String>,
|
||||||
|
|
||||||
|
/// Indicates if a token is stored in keyring
|
||||||
|
#[serde(default)]
|
||||||
|
pub has_token: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl TokenConfig {
|
impl TokenConfig {
|
||||||
/// Create a new token config
|
/// Create a new token config (token stored separately in keyring)
|
||||||
pub fn new(token: String, token_type: TokenType) -> Self {
|
pub fn new(token_type: TokenType) -> Self {
|
||||||
Self {
|
Self {
|
||||||
token: Some(token),
|
|
||||||
token_type,
|
token_type,
|
||||||
scopes: vec![],
|
scopes: vec![],
|
||||||
expires_at: None,
|
expires_at: None,
|
||||||
last_used: None,
|
last_used: None,
|
||||||
description: None,
|
description: None,
|
||||||
|
has_token: true,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a new token config without token
|
||||||
|
pub fn without_token(token_type: TokenType) -> Self {
|
||||||
|
Self {
|
||||||
|
token_type,
|
||||||
|
scopes: vec![],
|
||||||
|
expires_at: None,
|
||||||
|
last_used: None,
|
||||||
|
description: None,
|
||||||
|
has_token: false,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Validate token configuration
|
/// Validate token configuration
|
||||||
pub fn validate(&self) -> Result<()> {
|
pub fn validate(&self) -> Result<()> {
|
||||||
if self.token.is_none() && self.token_type != TokenType::None {
|
if !self.has_token && self.token_type != TokenType::None {
|
||||||
bail!("Token value is required for {:?}", self.token_type);
|
bail!("Token is required for {:?}", self.token_type);
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@@ -432,12 +567,19 @@ impl TokenConfig {
|
|||||||
pub fn record_usage(&mut self) {
|
pub fn record_usage(&mut self) {
|
||||||
self.last_used = Some(chrono::Utc::now().to_rfc3339());
|
self.last_used = Some(chrono::Utc::now().to_rfc3339());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Mark that a token is stored
|
||||||
|
pub fn set_has_token(&mut self, has_token: bool) {
|
||||||
|
self.has_token = has_token;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Token type
|
/// Token type
|
||||||
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
|
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
|
||||||
#[serde(rename_all = "lowercase")]
|
#[serde(rename_all = "lowercase")]
|
||||||
|
#[derive(Default)]
|
||||||
pub enum TokenType {
|
pub enum TokenType {
|
||||||
|
#[default]
|
||||||
None,
|
None,
|
||||||
Personal,
|
Personal,
|
||||||
OAuth,
|
OAuth,
|
||||||
@@ -445,12 +587,6 @@ pub enum TokenType {
|
|||||||
App,
|
App,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for TokenType {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self::None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl std::fmt::Display for TokenType {
|
impl std::fmt::Display for TokenType {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
@@ -496,7 +632,11 @@ pub struct ConfigDifference {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn default_gpg_program() -> String {
|
fn default_gpg_program() -> String {
|
||||||
"gpg".to_string()
|
if cfg!(target_os = "windows") {
|
||||||
|
"gpg.exe".to_string()
|
||||||
|
} else {
|
||||||
|
"gpg".to_string()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn default_true() -> bool {
|
fn default_true() -> bool {
|
||||||
@@ -576,9 +716,15 @@ impl GitProfileBuilder {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn build(self) -> Result<GitProfile> {
|
pub fn build(self) -> Result<GitProfile> {
|
||||||
let name = self.name.ok_or_else(|| anyhow::anyhow!("Name is required"))?;
|
let name = self
|
||||||
let user_name = self.user_name.ok_or_else(|| anyhow::anyhow!("User name is required"))?;
|
.name
|
||||||
let user_email = self.user_email.ok_or_else(|| anyhow::anyhow!("User email is required"))?;
|
.ok_or_else(|| anyhow::anyhow!("Name is required"))?;
|
||||||
|
let user_name = self
|
||||||
|
.user_name
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("User name is required"))?;
|
||||||
|
let user_email = self
|
||||||
|
.user_email
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("User email is required"))?;
|
||||||
|
|
||||||
Ok(GitProfile {
|
Ok(GitProfile {
|
||||||
name,
|
name,
|
||||||
@@ -630,7 +776,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_token_config() {
|
fn test_token_config() {
|
||||||
let token = TokenConfig::new("test-token".to_string(), TokenType::Personal);
|
let token = TokenConfig::new(TokenType::Personal);
|
||||||
assert!(token.validate().is_ok());
|
assert!(token.validate().is_ok());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,25 +1,61 @@
|
|||||||
use crate::config::{CommitFormat, LlmConfig, Language};
|
use crate::config::manager::ConfigManager;
|
||||||
|
use crate::config::{CommitFormat, Language};
|
||||||
use crate::git::{CommitInfo, GitRepo};
|
use crate::git::{CommitInfo, GitRepo};
|
||||||
use crate::llm::{GeneratedCommit, LlmClient};
|
use crate::i18n::Messages;
|
||||||
|
use crate::llm::parsing::GeneratedCommit;
|
||||||
|
use crate::llm::rig::LlmClient;
|
||||||
|
use crate::utils::{eprint_warning, success_prefix};
|
||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use chrono::Utc;
|
|
||||||
|
|
||||||
/// Content generator using LLM
|
/// Content generator using LLM
|
||||||
pub struct ContentGenerator {
|
pub struct ContentGenerator {
|
||||||
llm_client: LlmClient,
|
llm_client: LlmClient,
|
||||||
|
template: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ContentGenerator {
|
impl ContentGenerator {
|
||||||
/// Create new content generator
|
/// Create new content generator
|
||||||
pub async fn new(config: &LlmConfig) -> Result<Self> {
|
pub async fn new(manager: &ConfigManager) -> Result<Self> {
|
||||||
let llm_client = LlmClient::from_config(config).await?;
|
Self::new_with_think(manager, false, None).await
|
||||||
|
}
|
||||||
|
|
||||||
// Check if provider is available
|
/// Create new content generator with thinking override and optional commit template
|
||||||
if !llm_client.is_available().await {
|
pub async fn new_with_think(
|
||||||
anyhow::bail!("LLM provider '{}' is not available", config.provider);
|
manager: &ConfigManager,
|
||||||
|
think_override: bool,
|
||||||
|
template: Option<String>,
|
||||||
|
) -> Result<Self> {
|
||||||
|
let mut thinking_enabled = if think_override {
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
manager.config().llm.thinking_enabled
|
||||||
|
};
|
||||||
|
|
||||||
|
// Validate thinking support per provider
|
||||||
|
if thinking_enabled {
|
||||||
|
let provider = manager.llm_provider();
|
||||||
|
if !Self::supports_thinking(provider) {
|
||||||
|
let language = manager.get_language().unwrap_or(Language::English);
|
||||||
|
let messages = Messages::new(language);
|
||||||
|
eprint_warning(&messages.thinking_unsupported(provider));
|
||||||
|
thinking_enabled = false;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(Self { llm_client })
|
let llm_client = LlmClient::from_config_with_think(manager, thinking_enabled).await?;
|
||||||
|
|
||||||
|
if !llm_client.is_available().await {
|
||||||
|
anyhow::bail!("LLM provider '{}' is not available", manager.llm_provider());
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
llm_client,
|
||||||
|
template,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn supports_thinking(provider: &str) -> bool {
|
||||||
|
matches!(provider, "deepseek" | "kimi" | "anthropic" | "openai")
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate commit message from diff
|
/// Generate commit message from diff
|
||||||
@@ -32,12 +68,15 @@ impl ContentGenerator {
|
|||||||
// Truncate diff if too long
|
// Truncate diff if too long
|
||||||
let max_diff_len = 4000;
|
let max_diff_len = 4000;
|
||||||
let truncated_diff = if diff.len() > max_diff_len {
|
let truncated_diff = if diff.len() > max_diff_len {
|
||||||
format!("{}\n... (truncated)", &diff[..max_diff_len])
|
let boundary = diff.floor_char_boundary(max_diff_len);
|
||||||
|
format!("{}\n... (truncated)", &diff[..boundary])
|
||||||
} else {
|
} else {
|
||||||
diff.to_string()
|
diff.to_string()
|
||||||
};
|
};
|
||||||
|
|
||||||
self.llm_client.generate_commit_message(&truncated_diff, format, language).await
|
self.llm_client
|
||||||
|
.generate_commit_message(&truncated_diff, format, language, self.template.as_deref())
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate commit message from repository changes
|
/// Generate commit message from repository changes
|
||||||
@@ -47,7 +86,8 @@ impl ContentGenerator {
|
|||||||
format: CommitFormat,
|
format: CommitFormat,
|
||||||
language: Language,
|
language: Language,
|
||||||
) -> Result<GeneratedCommit> {
|
) -> Result<GeneratedCommit> {
|
||||||
let diff = repo.get_staged_diff()
|
let diff = repo
|
||||||
|
.get_staged_diff_sorted()
|
||||||
.context("Failed to get staged diff")?;
|
.context("Failed to get staged diff")?;
|
||||||
|
|
||||||
if diff.is_empty() {
|
if diff.is_empty() {
|
||||||
@@ -64,12 +104,12 @@ impl ContentGenerator {
|
|||||||
commits: &[CommitInfo],
|
commits: &[CommitInfo],
|
||||||
language: Language,
|
language: Language,
|
||||||
) -> Result<String> {
|
) -> Result<String> {
|
||||||
let commit_messages: Vec<String> = commits
|
let commit_messages: Vec<String> =
|
||||||
.iter()
|
commits.iter().map(|c| c.subject().to_string()).collect();
|
||||||
.map(|c| c.subject().to_string())
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
self.llm_client.generate_tag_message(version, &commit_messages, language).await
|
self.llm_client
|
||||||
|
.generate_tag_message(version, &commit_messages, language)
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate changelog entry
|
/// Generate changelog entry
|
||||||
@@ -87,7 +127,9 @@ impl ContentGenerator {
|
|||||||
})
|
})
|
||||||
.collect();
|
.collect();
|
||||||
|
|
||||||
self.llm_client.generate_changelog_entry(version, &typed_commits, language).await
|
self.llm_client
|
||||||
|
.generate_changelog_entry(version, &typed_commits, language)
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate changelog from repository
|
/// Generate changelog from repository
|
||||||
@@ -104,7 +146,8 @@ impl ContentGenerator {
|
|||||||
repo.get_commits(50)?
|
repo.get_commits(50)?
|
||||||
};
|
};
|
||||||
|
|
||||||
self.generate_changelog_entry(version, &commits, language).await
|
self.generate_changelog_entry(version, &commits, language)
|
||||||
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Interactive commit generation with user feedback
|
/// Interactive commit generation with user feedback
|
||||||
@@ -113,11 +156,11 @@ impl ContentGenerator {
|
|||||||
repo: &GitRepo,
|
repo: &GitRepo,
|
||||||
format: CommitFormat,
|
format: CommitFormat,
|
||||||
language: Language,
|
language: Language,
|
||||||
|
messages: &Messages,
|
||||||
) -> Result<GeneratedCommit> {
|
) -> Result<GeneratedCommit> {
|
||||||
use dialoguer::{Confirm, Select};
|
use dialoguer::Select;
|
||||||
use console::Term;
|
|
||||||
|
|
||||||
let diff = repo.get_staged_diff()?;
|
let diff = repo.get_staged_diff_sorted()?;
|
||||||
|
|
||||||
if diff.is_empty() {
|
if diff.is_empty() {
|
||||||
anyhow::bail!("No staged changes");
|
anyhow::bail!("No staged changes");
|
||||||
@@ -125,32 +168,33 @@ impl ContentGenerator {
|
|||||||
|
|
||||||
// Show diff summary
|
// Show diff summary
|
||||||
let files = repo.get_staged_files()?;
|
let files = repo.get_staged_files()?;
|
||||||
println!("\nStaged files ({}):", files.len());
|
println!("\n{}", messages.staged_files(files.len()));
|
||||||
for file in &files {
|
for file in &files {
|
||||||
println!(" • {}", file);
|
println!(" • {}", file);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Generate initial commit
|
// Generate initial commit
|
||||||
println!("\nGenerating commit message...");
|
println!("\n{}", messages.generating_commit_message());
|
||||||
let mut generated = self.generate_commit_message(&diff, format, language).await?;
|
let mut generated = self
|
||||||
|
.generate_commit_message(&diff, format, language)
|
||||||
|
.await?;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
println!("\n{}", "─".repeat(60));
|
println!("\n{}", "─".repeat(60));
|
||||||
println!("Generated commit message:");
|
println!("{}", messages.generated_commit_message());
|
||||||
println!("{}", "─".repeat(60));
|
println!("{}", "─".repeat(60));
|
||||||
println!("{}", generated.to_conventional());
|
println!("{}", generated.to_conventional());
|
||||||
println!("{}", "─".repeat(60));
|
println!("{}", "─".repeat(60));
|
||||||
|
|
||||||
let options = vec![
|
let options = vec![
|
||||||
"✓ Accept and commit",
|
format!("{} {}", success_prefix(), messages.accept_and_commit()),
|
||||||
"🔄 Regenerate",
|
messages.regenerate().to_string(),
|
||||||
"✏️ Edit",
|
messages.edit().to_string(),
|
||||||
"📋 Copy to clipboard",
|
messages.cancel().to_string(),
|
||||||
"❌ Cancel",
|
|
||||||
];
|
];
|
||||||
|
|
||||||
let selection = Select::new()
|
let selection = Select::new()
|
||||||
.with_prompt("What would you like to do?")
|
.with_prompt(messages.what_would_you_like_to_do())
|
||||||
.items(&options)
|
.items(&options)
|
||||||
.default(0)
|
.default(0)
|
||||||
.interact()?;
|
.interact()?;
|
||||||
@@ -158,31 +202,22 @@ impl ContentGenerator {
|
|||||||
match selection {
|
match selection {
|
||||||
0 => return Ok(generated),
|
0 => return Ok(generated),
|
||||||
1 => {
|
1 => {
|
||||||
println!("Regenerating...");
|
println!("{}", messages.regenerating());
|
||||||
generated = self.generate_commit_message(&diff, format, language).await?;
|
generated = self
|
||||||
|
.generate_commit_message(&diff, format, language)
|
||||||
|
.await?;
|
||||||
}
|
}
|
||||||
2 => {
|
2 => {
|
||||||
let edited = crate::utils::editor::edit_content(&generated.to_conventional())?;
|
let edited = crate::utils::editor::edit_content(&generated.to_conventional())?;
|
||||||
generated = self.parse_edited_commit(&edited, format)?;
|
generated = self.parse_edited_commit(&edited, format)?;
|
||||||
}
|
}
|
||||||
3 => {
|
3 => anyhow::bail!("{}", messages.cancelled_by_user()),
|
||||||
#[cfg(feature = "clipboard")]
|
|
||||||
{
|
|
||||||
arboard::Clipboard::new()?.set_text(generated.to_conventional())?;
|
|
||||||
println!("Copied to clipboard!");
|
|
||||||
}
|
|
||||||
#[cfg(not(feature = "clipboard"))]
|
|
||||||
{
|
|
||||||
println!("Clipboard feature not enabled");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
4 => anyhow::bail!("Cancelled by user"),
|
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse_edited_commit(&self, edited: &str, format: CommitFormat) -> Result<GeneratedCommit> {
|
fn parse_edited_commit(&self, edited: &str, _format: CommitFormat) -> Result<GeneratedCommit> {
|
||||||
let parsed = crate::git::commit::parse_commit_message(edited);
|
let parsed = crate::git::commit::parse_commit_message(edited);
|
||||||
|
|
||||||
Ok(GeneratedCommit {
|
Ok(GeneratedCommit {
|
||||||
@@ -222,9 +257,13 @@ pub mod fallback {
|
|||||||
f.ends_with(".rs") || f.ends_with(".py") || f.ends_with(".js") || f.ends_with(".ts")
|
f.ends_with(".rs") || f.ends_with(".py") || f.ends_with(".js") || f.ends_with(".ts")
|
||||||
});
|
});
|
||||||
|
|
||||||
let has_docs = files.iter().any(|f| f.ends_with(".md") || f.contains("README"));
|
let has_docs = files
|
||||||
|
.iter()
|
||||||
|
.any(|f| f.ends_with(".md") || f.contains("README"));
|
||||||
|
|
||||||
let has_tests = files.iter().any(|f| f.contains("test") || f.contains("spec"));
|
let has_tests = files
|
||||||
|
.iter()
|
||||||
|
.any(|f| f.contains("test") || f.contains("spec"));
|
||||||
|
|
||||||
if has_tests {
|
if has_tests {
|
||||||
"test: update tests".to_string()
|
"test: update tests".to_string()
|
||||||
|
|||||||
@@ -1,10 +1,19 @@
|
|||||||
use super::{CommitInfo, GitRepo};
|
use super::{CommitInfo, GitRepo};
|
||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use chrono::{DateTime, TimeZone, Utc};
|
use chrono::{DateTime, Utc};
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::fs;
|
use std::fs;
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
|
|
||||||
|
pub const CHANGELOG_HEADER: &str = r#"# Changelog
|
||||||
|
|
||||||
|
All notable changes to this project will be documented in this file.
|
||||||
|
|
||||||
|
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
||||||
|
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
||||||
|
|
||||||
|
"#;
|
||||||
|
|
||||||
/// Changelog generator
|
/// Changelog generator
|
||||||
pub struct ChangelogGenerator {
|
pub struct ChangelogGenerator {
|
||||||
format: ChangelogFormat,
|
format: ChangelogFormat,
|
||||||
@@ -86,9 +95,7 @@ impl ChangelogGenerator {
|
|||||||
ChangelogFormat::GitHubReleases => {
|
ChangelogFormat::GitHubReleases => {
|
||||||
self.generate_github_releases(version, date, commits)
|
self.generate_github_releases(version, date, commits)
|
||||||
}
|
}
|
||||||
ChangelogFormat::Custom => {
|
ChangelogFormat::Custom => self.generate_custom(version, date, commits),
|
||||||
self.generate_custom(version, date, commits)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -109,9 +116,10 @@ impl ChangelogGenerator {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let new_content = if existing.is_empty() {
|
let new_content = if existing.is_empty() {
|
||||||
format!("# Changelog\n\n{}", entry)
|
format!("{}{}", CHANGELOG_HEADER, entry)
|
||||||
} else {
|
} else if existing.starts_with(CHANGELOG_HEADER) {
|
||||||
// Find position after header
|
format!("{}{}", CHANGELOG_HEADER, entry)
|
||||||
|
} else if existing.starts_with("# Changelog") {
|
||||||
let lines: Vec<&str> = existing.lines().collect();
|
let lines: Vec<&str> = existing.lines().collect();
|
||||||
let mut header_end = 0;
|
let mut header_end = 0;
|
||||||
|
|
||||||
@@ -129,6 +137,8 @@ impl ChangelogGenerator {
|
|||||||
let rest = lines[header_end..].join("\n");
|
let rest = lines[header_end..].join("\n");
|
||||||
|
|
||||||
format!("{}\n{}\n{}", header, entry, rest)
|
format!("{}\n{}\n{}", header, entry, rest)
|
||||||
|
} else {
|
||||||
|
format!("{}{}", CHANGELOG_HEADER, entry)
|
||||||
};
|
};
|
||||||
|
|
||||||
fs::write(changelog_path, new_content)
|
fs::write(changelog_path, new_content)
|
||||||
@@ -147,7 +157,7 @@ impl ChangelogGenerator {
|
|||||||
let mut output = format!("## [{}] - {}\n\n", version, date_str);
|
let mut output = format!("## [{}] - {}\n\n", version, date_str);
|
||||||
|
|
||||||
if self.group_by_type {
|
if self.group_by_type {
|
||||||
let grouped = self.group_commits(commits);
|
let _grouped = self.group_commits(commits);
|
||||||
|
|
||||||
// Standard categories
|
// Standard categories
|
||||||
let categories = vec![
|
let categories = vec![
|
||||||
@@ -218,11 +228,11 @@ impl ChangelogGenerator {
|
|||||||
|
|
||||||
fn generate_github_releases(
|
fn generate_github_releases(
|
||||||
&self,
|
&self,
|
||||||
version: &str,
|
_version: &str,
|
||||||
_date: DateTime<Utc>,
|
_date: DateTime<Utc>,
|
||||||
commits: &[CommitInfo],
|
commits: &[CommitInfo],
|
||||||
) -> Result<String> {
|
) -> Result<String> {
|
||||||
let mut output = format!("## What's Changed\n\n");
|
let mut output = "## What's Changed\n\n".to_string();
|
||||||
|
|
||||||
// Group by type
|
// Group by type
|
||||||
let mut features = vec![];
|
let mut features = vec![];
|
||||||
@@ -232,8 +242,6 @@ impl ChangelogGenerator {
|
|||||||
let mut breaking = vec![];
|
let mut breaking = vec![];
|
||||||
|
|
||||||
for commit in commits {
|
for commit in commits {
|
||||||
let msg = commit.subject();
|
|
||||||
|
|
||||||
if commit.message.contains("BREAKING CHANGE") {
|
if commit.message.contains("BREAKING CHANGE") {
|
||||||
breaking.push(commit);
|
breaking.push(commit);
|
||||||
}
|
}
|
||||||
@@ -347,7 +355,12 @@ impl ChangelogGenerator {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn format_commit_github(&self, commit: &CommitInfo) -> String {
|
fn format_commit_github(&self, commit: &CommitInfo) -> String {
|
||||||
format!("- {} by @{} in {}\n", commit.subject(), commit.author, &commit.short_id)
|
format!(
|
||||||
|
"- {} by @{} in {}\n",
|
||||||
|
commit.subject(),
|
||||||
|
commit.author,
|
||||||
|
&commit.short_id
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn group_commits<'a>(&self, commits: &'a [CommitInfo]) -> HashMap<String, Vec<&'a CommitInfo>> {
|
fn group_commits<'a>(&self, commits: &'a [CommitInfo]) -> HashMap<String, Vec<&'a CommitInfo>> {
|
||||||
@@ -370,8 +383,7 @@ impl Default for ChangelogGenerator {
|
|||||||
|
|
||||||
/// Read existing changelog
|
/// Read existing changelog
|
||||||
pub fn read_changelog(path: &Path) -> Result<String> {
|
pub fn read_changelog(path: &Path) -> Result<String> {
|
||||||
fs::read_to_string(path)
|
fs::read_to_string(path).with_context(|| format!("Failed to read changelog: {:?}", path))
|
||||||
.with_context(|| format!("Failed to read changelog: {:?}", path))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Initialize new changelog file
|
/// Initialize new changelog file
|
||||||
@@ -380,16 +392,7 @@ pub fn init_changelog(path: &Path) -> Result<()> {
|
|||||||
anyhow::bail!("Changelog already exists at {:?}", path);
|
anyhow::bail!("Changelog already exists at {:?}", path);
|
||||||
}
|
}
|
||||||
|
|
||||||
let content = r#"# Changelog
|
fs::write(path, CHANGELOG_HEADER)
|
||||||
|
|
||||||
All notable changes to this project will be documented in this file.
|
|
||||||
|
|
||||||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
|
||||||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
|
||||||
|
|
||||||
"#;
|
|
||||||
|
|
||||||
fs::write(path, content)
|
|
||||||
.with_context(|| format!("Failed to create changelog: {:?}", path))?;
|
.with_context(|| format!("Failed to create changelog: {:?}", path))?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -406,17 +409,13 @@ pub fn generate_from_history(
|
|||||||
if let Some(from) = from_tag {
|
if let Some(from) = from_tag {
|
||||||
repo.get_commits_between(from, to_ref)
|
repo.get_commits_between(from, to_ref)
|
||||||
} else {
|
} else {
|
||||||
// Get last 50 commits if no tag specified
|
// Get all commits from the beginning (no from_tag = initial changelog)
|
||||||
repo.get_commits(50)
|
repo.get_commits(usize::MAX)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Update version links in changelog
|
/// Update version links in changelog
|
||||||
pub fn update_version_links(
|
pub fn update_version_links(changelog: &str, version: &str, compare_url: &str) -> String {
|
||||||
changelog: &str,
|
|
||||||
version: &str,
|
|
||||||
compare_url: &str,
|
|
||||||
) -> String {
|
|
||||||
// Add version link at the end of changelog
|
// Add version link at the end of changelog
|
||||||
format!("{}\n[{}]: {}\n", changelog, version, compare_url)
|
format!("{}\n[{}]: {}\n", changelog, version, compare_url)
|
||||||
}
|
}
|
||||||
@@ -426,17 +425,16 @@ pub fn parse_versions(changelog: &str) -> Vec<(String, String)> {
|
|||||||
let mut versions = vec![];
|
let mut versions = vec![];
|
||||||
|
|
||||||
for line in changelog.lines() {
|
for line in changelog.lines() {
|
||||||
if line.starts_with("## [") {
|
if line.starts_with("## [")
|
||||||
if let Some(start) = line.find('[') {
|
&& let Some(start) = line.find('[')
|
||||||
if let Some(end) = line.find(']') {
|
&& let Some(end) = line.find(']')
|
||||||
let version = &line[start + 1..end];
|
{
|
||||||
if version != "Unreleased" {
|
let version = &line[start + 1..end];
|
||||||
if let Some(date_start) = line.find(" - ") {
|
if version != "Unreleased"
|
||||||
let date = &line[date_start + 3..].trim();
|
&& let Some(date_start) = line.find(" - ")
|
||||||
versions.push((version.to_string(), date.to_string()));
|
{
|
||||||
}
|
let date = &line[date_start + 3..].trim();
|
||||||
}
|
versions.push((version.to_string(), date.to_string()));
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -478,3 +476,137 @@ impl ChangelogEntry {
|
|||||||
self
|
self
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Extract version strings from Keep a Changelog content.
|
||||||
|
/// Returns versions sorted by semver descending (highest first).
|
||||||
|
/// Excludes "Unreleased".
|
||||||
|
pub fn parse_changelog_versions(content: &str) -> Vec<String> {
|
||||||
|
use semver::Version;
|
||||||
|
|
||||||
|
let mut versions: Vec<String> = vec![];
|
||||||
|
for line in content.lines() {
|
||||||
|
let trimmed = line.trim();
|
||||||
|
if trimmed.starts_with("## [") {
|
||||||
|
if let Some(end) = trimmed.find(']') {
|
||||||
|
let version = &trimmed[4..end];
|
||||||
|
if version != "Unreleased" && !version.is_empty() {
|
||||||
|
versions.push(version.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Sort by semver descending so highest version is first
|
||||||
|
versions.sort_by(|a, b| {
|
||||||
|
match (Version::parse(a), Version::parse(b)) {
|
||||||
|
(Ok(va), Ok(vb)) => vb.cmp(&va),
|
||||||
|
(Ok(_), Err(_)) => std::cmp::Ordering::Less,
|
||||||
|
(Err(_), Ok(_)) => std::cmp::Ordering::Greater,
|
||||||
|
(Err(_), Err(_)) => std::cmp::Ordering::Equal,
|
||||||
|
}
|
||||||
|
});
|
||||||
|
versions
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Insert a new changelog entry into existing changelog content.
|
||||||
|
/// The new entry is placed after the header and before the first existing version section.
|
||||||
|
/// If no existing version sections are found, the new entry is appended after the header.
|
||||||
|
pub fn insert_changelog_entry(existing: &str, new_entry: &str) -> String {
|
||||||
|
// Find the first version section (## [x.y.z])
|
||||||
|
if let Some(first_ver_pos) = existing.find("\n## [") {
|
||||||
|
let (header, rest) = existing.split_at(first_ver_pos);
|
||||||
|
// Ensure new_entry ends with a blank line before the next section
|
||||||
|
let entry = new_entry.trim_end();
|
||||||
|
format!("{}\n{}\n{}", header.trim_end(), entry, rest)
|
||||||
|
} else {
|
||||||
|
// No existing version sections — append after header
|
||||||
|
format!("{}\n{}", existing.trim_end(), new_entry.trim_end())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_parse_changelog_versions_extracts_versions() {
|
||||||
|
let content = "# Changelog\n\n## [0.2.0] - 2024-03-01\n### Added\n- feature B\n\n## [0.1.0] - 2024-02-01\n### Added\n- feature A\n";
|
||||||
|
let versions = parse_changelog_versions(content);
|
||||||
|
assert_eq!(versions, vec!["0.2.0", "0.1.0"]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_parse_changelog_versions_sorts_highest_first() {
|
||||||
|
// Versions in reversed order — should still return highest first
|
||||||
|
let content = "# Changelog\n\n## [0.1.0] - 2024-02-01\n### Added\n- feature A\n\n## [0.3.0] - 2024-04-01\n### Added\n- feature C\n\n## [0.2.0] - 2024-03-01\n### Added\n- feature B\n";
|
||||||
|
let versions = parse_changelog_versions(content);
|
||||||
|
assert_eq!(versions, vec!["0.3.0", "0.2.0", "0.1.0"]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_parse_changelog_versions_excludes_unreleased() {
|
||||||
|
let content = "# Changelog\n\n## [Unreleased]\n### Added\n- wip\n\n## [0.1.0] - 2024-02-01\n### Added\n- feature A\n";
|
||||||
|
let versions = parse_changelog_versions(content);
|
||||||
|
assert_eq!(versions, vec!["0.1.0"]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_parse_changelog_versions_empty_content() {
|
||||||
|
let versions = parse_changelog_versions("");
|
||||||
|
assert!(versions.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_parse_changelog_versions_no_versions() {
|
||||||
|
let content = "# Changelog\n\nSome description text.\n";
|
||||||
|
let versions = parse_changelog_versions(content);
|
||||||
|
assert!(versions.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_insert_changelog_entry_between_header_and_existing() {
|
||||||
|
let existing = "# Changelog\n\nAll notable changes...\n\n## [0.1.0] - 2024-02-01\n### Added\n- feature A\n";
|
||||||
|
let new_entry = "## [0.2.0] - 2024-03-01\n### Added\n- feature B\n";
|
||||||
|
let result = insert_changelog_entry(existing, new_entry);
|
||||||
|
|
||||||
|
// New entry should appear after header, before 0.1.0
|
||||||
|
assert!(result.contains("## [0.2.0]"));
|
||||||
|
assert!(result.contains("## [0.1.0]"));
|
||||||
|
let pos_new = result.find("## [0.2.0]").unwrap();
|
||||||
|
let pos_old = result.find("## [0.1.0]").unwrap();
|
||||||
|
assert!(pos_new < pos_old, "new version should be before old version");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_insert_changelog_entry_preserves_all_existing_content() {
|
||||||
|
let existing = "# Changelog\n\nAll notable changes...\n\n## [0.2.0] - 2024-03-01\n### Added\n- feature B\n\n## [0.1.0] - 2024-02-01\n### Added\n- feature A\n";
|
||||||
|
let new_entry = "## [0.3.0] - 2024-04-01\n### Added\n- feature C\n";
|
||||||
|
let result = insert_changelog_entry(existing, new_entry);
|
||||||
|
|
||||||
|
assert!(result.contains("## [0.3.0]"));
|
||||||
|
assert!(result.contains("## [0.2.0]"));
|
||||||
|
assert!(result.contains("## [0.1.0]"));
|
||||||
|
assert!(result.contains("feature A"), "oldest content preserved");
|
||||||
|
assert!(result.contains("feature B"), "middle content preserved");
|
||||||
|
assert!(result.contains("feature C"), "new content present");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_insert_changelog_entry_empty_existing() {
|
||||||
|
let existing = "# Changelog\n\nAll notable changes...\n\n";
|
||||||
|
let new_entry = "## [0.1.0] - 2024-02-01\n### Added\n- feature A\n";
|
||||||
|
let result = insert_changelog_entry(existing, new_entry);
|
||||||
|
assert!(result.contains("## [0.1.0]"));
|
||||||
|
assert!(result.contains("# Changelog"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_insert_changelog_entry_extra_blank_lines_in_header() {
|
||||||
|
let existing = "# Changelog\n\n\n\n## [0.1.0] - 2024-02-01\n### Added\n- feature A\n";
|
||||||
|
let new_entry = "## [0.2.0] - 2024-03-01\n### Added\n- feature B\n";
|
||||||
|
let result = insert_changelog_entry(existing, new_entry);
|
||||||
|
|
||||||
|
let pos_new = result.find("## [0.2.0]").unwrap();
|
||||||
|
let pos_old = result.find("## [0.1.0]").unwrap();
|
||||||
|
assert!(pos_new < pos_old, "new version should be before old version");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
use super::GitRepo;
|
use super::GitRepo;
|
||||||
use anyhow::{bail, Context, Result};
|
use anyhow::{Result, bail};
|
||||||
use chrono::Local;
|
use chrono::Local;
|
||||||
|
|
||||||
/// Commit builder for creating commits
|
/// Commit builder for creating commits
|
||||||
@@ -47,6 +47,12 @@ impl CommitBuilder {
|
|||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Set scope (optional)
|
||||||
|
pub fn scope_opt(mut self, scope: Option<String>) -> Self {
|
||||||
|
self.scope = scope;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
/// Set description
|
/// Set description
|
||||||
pub fn description(mut self, description: impl Into<String>) -> Self {
|
pub fn description(mut self, description: impl Into<String>) -> Self {
|
||||||
self.description = Some(description.into());
|
self.description = Some(description.into());
|
||||||
@@ -59,6 +65,12 @@ impl CommitBuilder {
|
|||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Set body (optional)
|
||||||
|
pub fn body_opt(mut self, body: Option<String>) -> Self {
|
||||||
|
self.body = body;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
/// Set footer
|
/// Set footer
|
||||||
pub fn footer(mut self, footer: impl Into<String>) -> Self {
|
pub fn footer(mut self, footer: impl Into<String>) -> Self {
|
||||||
self.footer = Some(footer.into());
|
self.footer = Some(footer.into());
|
||||||
@@ -107,10 +119,14 @@ impl CommitBuilder {
|
|||||||
return Ok(msg.clone());
|
return Ok(msg.clone());
|
||||||
}
|
}
|
||||||
|
|
||||||
let commit_type = self.commit_type.as_ref()
|
let commit_type = self
|
||||||
|
.commit_type
|
||||||
|
.as_ref()
|
||||||
.ok_or_else(|| anyhow::anyhow!("Commit type is required"))?;
|
.ok_or_else(|| anyhow::anyhow!("Commit type is required"))?;
|
||||||
|
|
||||||
let description = self.description.as_ref()
|
let description = self
|
||||||
|
.description
|
||||||
|
.as_ref()
|
||||||
.ok_or_else(|| anyhow::anyhow!("Description is required"))?;
|
.ok_or_else(|| anyhow::anyhow!("Description is required"))?;
|
||||||
|
|
||||||
let message = match self.format {
|
let message = match self.format {
|
||||||
@@ -182,7 +198,8 @@ impl CommitBuilder {
|
|||||||
"GPG signing failed. Please check:\n\
|
"GPG signing failed. Please check:\n\
|
||||||
1. GPG signing key is configured (git config --get user.signingkey)\n\
|
1. GPG signing key is configured (git config --get user.signingkey)\n\
|
||||||
2. GPG agent is running\n\
|
2. GPG agent is running\n\
|
||||||
3. You can sign commits manually (try: git commit --amend -S)".to_string()
|
3. You can sign commits manually (try: git commit --amend -S)"
|
||||||
|
.to_string()
|
||||||
} else {
|
} else {
|
||||||
stdout.to_string()
|
stdout.to_string()
|
||||||
}
|
}
|
||||||
@@ -229,12 +246,19 @@ pub fn suggest_commit_type(diff: &str) -> Vec<&'static str> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Check for configuration files
|
// Check for configuration files
|
||||||
if diff.contains("config") || diff.contains(".json") || diff.contains(".yaml") || diff.contains(".toml") {
|
if diff.contains("config")
|
||||||
|
|| diff.contains(".json")
|
||||||
|
|| diff.contains(".yaml")
|
||||||
|
|| diff.contains(".toml")
|
||||||
|
{
|
||||||
suggestions.push("chore");
|
suggestions.push("chore");
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check for dependencies
|
// Check for dependencies
|
||||||
if diff.contains("Cargo.toml") || diff.contains("package.json") || diff.contains("requirements.txt") {
|
if diff.contains("Cargo.toml")
|
||||||
|
|| diff.contains("package.json")
|
||||||
|
|| diff.contains("requirements.txt")
|
||||||
|
{
|
||||||
suggestions.push("build");
|
suggestions.push("build");
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -291,11 +315,12 @@ pub fn parse_commit_message(message: &str) -> ParsedCommit {
|
|||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if line.starts_with("BREAKING CHANGE:") ||
|
if line.starts_with("BREAKING CHANGE:")
|
||||||
line.starts_with("Closes") ||
|
|| line.starts_with("Closes")
|
||||||
line.starts_with("Fixes") ||
|
|| line.starts_with("Fixes")
|
||||||
line.starts_with("Refs") ||
|
|| line.starts_with("Refs")
|
||||||
line.starts_with("Co-authored-by:") {
|
|| line.starts_with("Co-authored-by:")
|
||||||
|
{
|
||||||
in_footer = true;
|
in_footer = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -310,8 +335,16 @@ pub fn parse_commit_message(message: &str) -> ParsedCommit {
|
|||||||
commit_type,
|
commit_type,
|
||||||
scope,
|
scope,
|
||||||
description: Some(description.to_string()),
|
description: Some(description.to_string()),
|
||||||
body: if body_lines.is_empty() { None } else { Some(body_lines.join("\n")) },
|
body: if body_lines.is_empty() {
|
||||||
footer: if footer_lines.is_empty() { None } else { Some(footer_lines.join("\n")) },
|
None
|
||||||
|
} else {
|
||||||
|
Some(body_lines.join("\n"))
|
||||||
|
},
|
||||||
|
footer: if footer_lines.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(footer_lines.join("\n"))
|
||||||
|
},
|
||||||
breaking,
|
breaking,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
933
src/git/mod.rs
933
src/git/mod.rs
File diff suppressed because it is too large
Load Diff
483
src/git/tag.rs
483
src/git/tag.rs
@@ -1,6 +1,7 @@
|
|||||||
use super::GitRepo;
|
use super::GitRepo;
|
||||||
use anyhow::{bail, Context, Result};
|
use anyhow::{Result, bail};
|
||||||
use semver::Version;
|
use semver::Version;
|
||||||
|
use std::path::Path;
|
||||||
|
|
||||||
/// Tag builder for creating tags
|
/// Tag builder for creating tags
|
||||||
pub struct TagBuilder {
|
pub struct TagBuilder {
|
||||||
@@ -69,19 +70,19 @@ impl TagBuilder {
|
|||||||
|
|
||||||
/// Build tag message
|
/// Build tag message
|
||||||
pub fn build_message(&self) -> Result<String> {
|
pub fn build_message(&self) -> Result<String> {
|
||||||
let message = self.message.as_ref()
|
let message = self.message.as_ref().cloned().unwrap_or_else(|| {
|
||||||
.cloned()
|
let name = self.name.as_deref().unwrap_or("unknown");
|
||||||
.unwrap_or_else(|| {
|
format!("Release {}", name)
|
||||||
let name = self.name.as_deref().unwrap_or("unknown");
|
});
|
||||||
format!("Release {}", name)
|
|
||||||
});
|
|
||||||
|
|
||||||
Ok(message)
|
Ok(message)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Execute tag creation
|
/// Execute tag creation
|
||||||
pub fn execute(&self, repo: &GitRepo) -> Result<()> {
|
pub fn execute(&self, repo: &GitRepo) -> Result<()> {
|
||||||
let name = self.name.as_ref()
|
let name = self
|
||||||
|
.name
|
||||||
|
.as_ref()
|
||||||
.ok_or_else(|| anyhow::anyhow!("Tag name is required"))?;
|
.ok_or_else(|| anyhow::anyhow!("Tag name is required"))?;
|
||||||
|
|
||||||
if !self.force {
|
if !self.force {
|
||||||
@@ -136,7 +137,10 @@ impl VersionBump {
|
|||||||
"minor" => Ok(Self::Minor),
|
"minor" => Ok(Self::Minor),
|
||||||
"patch" => Ok(Self::Patch),
|
"patch" => Ok(Self::Patch),
|
||||||
"prerelease" | "pre" => Ok(Self::Prerelease),
|
"prerelease" | "pre" => Ok(Self::Prerelease),
|
||||||
_ => bail!("Invalid version bump: {}. Use: major, minor, patch, prerelease", s),
|
_ => bail!(
|
||||||
|
"Invalid version bump: {}. Use: major, minor, patch, prerelease",
|
||||||
|
s
|
||||||
|
),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -187,7 +191,10 @@ pub fn suggest_version_bump(commits: &[super::CommitInfo]) -> VersionBump {
|
|||||||
for commit in commits {
|
for commit in commits {
|
||||||
let msg = commit.message.to_lowercase();
|
let msg = commit.message.to_lowercase();
|
||||||
|
|
||||||
if msg.contains("breaking change") || msg.contains("breaking-change") || msg.contains("breaking_change") {
|
if msg.contains("breaking change")
|
||||||
|
|| msg.contains("breaking-change")
|
||||||
|
|| msg.contains("breaking_change")
|
||||||
|
{
|
||||||
has_breaking = true;
|
has_breaking = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -281,8 +288,9 @@ pub fn delete_tag(repo: &GitRepo, name: &str, remote: Option<&str>) -> Result<()
|
|||||||
if let Some(remote) = remote {
|
if let Some(remote) = remote {
|
||||||
use std::process::Command;
|
use std::process::Command;
|
||||||
|
|
||||||
|
let refspec = format!(":refs/tags/{}", name);
|
||||||
let output = Command::new("git")
|
let output = Command::new("git")
|
||||||
.args(&["push", remote, ":refs/tags/{}"])
|
.args(["push", remote, &refspec])
|
||||||
.current_dir(repo.path())
|
.current_dir(repo.path())
|
||||||
.output()?;
|
.output()?;
|
||||||
|
|
||||||
@@ -320,3 +328,456 @@ pub fn list_tags(
|
|||||||
Ok(filtered)
|
Ok(filtered)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sort tags by semver version descending, then by time descending.
|
||||||
|
/// Non-semver tags are placed after semver tags, sorted by time descending.
|
||||||
|
pub fn sort_tags_by_semver(tags: &mut [super::TagInfo]) {
|
||||||
|
use semver::Version;
|
||||||
|
|
||||||
|
tags.sort_by(|a, b| {
|
||||||
|
match (
|
||||||
|
Version::parse(a.version_name()),
|
||||||
|
Version::parse(b.version_name()),
|
||||||
|
) {
|
||||||
|
(Ok(va), Ok(vb)) => {
|
||||||
|
// Both semver: version descending, then time descending as tiebreaker
|
||||||
|
vb.cmp(&va).then_with(|| b.time.cmp(&a.time))
|
||||||
|
}
|
||||||
|
(Ok(_), Err(_)) => std::cmp::Ordering::Less, // semver tags first
|
||||||
|
(Err(_), Ok(_)) => std::cmp::Ordering::Greater, // semver tags first
|
||||||
|
(Err(_), Err(_)) => b.time.cmp(&a.time), // both non-semver: time descending
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A version found in a project config file, tagged with its source.
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub struct ConfigVersion {
|
||||||
|
/// Source file name, e.g. "Cargo.toml", "package.json"
|
||||||
|
pub source: String,
|
||||||
|
/// Parsed semantic version
|
||||||
|
pub version: Version,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Read project versions from all supported config files in the project root.
|
||||||
|
///
|
||||||
|
/// If `go.mod` exists the function returns an empty `Vec` immediately — Go projects
|
||||||
|
/// use git tags as the version authority and store nothing in go.mod.
|
||||||
|
///
|
||||||
|
/// Otherwise every supported format is tried. Results are returned in a stable
|
||||||
|
/// order (see `PARSER_ORDER`). Files that don't exist or whose version can't be
|
||||||
|
/// parsed are silently skipped.
|
||||||
|
pub fn read_project_versions(project_dir: &Path) -> Vec<ConfigVersion> {
|
||||||
|
// Go projects: go.mod has no version field; skip directly to commit analysis
|
||||||
|
if project_dir.join("go.mod").exists() {
|
||||||
|
return vec![];
|
||||||
|
}
|
||||||
|
|
||||||
|
let parsers: &[(&str, fn(&Path) -> Option<Version>)] = PARSER_ORDER;
|
||||||
|
let mut results = Vec::new();
|
||||||
|
|
||||||
|
for &(source, parser) in parsers {
|
||||||
|
if let Some(v) = parser(project_dir) {
|
||||||
|
results.push(ConfigVersion {
|
||||||
|
source: source.to_string(),
|
||||||
|
version: v,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
results
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Stable order for parser dispatch. No priority implied — all results are
|
||||||
|
/// surfaced to the user when more than one is found.
|
||||||
|
type Parser = fn(&Path) -> Option<Version>;
|
||||||
|
const PARSER_ORDER: &[(&str, Parser)] = &[
|
||||||
|
("Cargo.toml", try_cargo_toml),
|
||||||
|
("package.json", try_package_json),
|
||||||
|
("pyproject.toml", try_pyproject_toml),
|
||||||
|
("pom.xml", try_pom_xml),
|
||||||
|
("pubspec.yaml", try_pubspec_yaml),
|
||||||
|
("build.sbt", try_build_sbt),
|
||||||
|
("project.clj", try_project_clj),
|
||||||
|
("mix.exs", try_mix_exs),
|
||||||
|
("package.yaml", try_package_yaml),
|
||||||
|
("Project.toml", try_project_toml_julia),
|
||||||
|
("composer.json", try_composer_json),
|
||||||
|
(".csproj", try_csproj),
|
||||||
|
("CMakeLists.txt", try_cmake_lists),
|
||||||
|
("meson.build", try_meson_build),
|
||||||
|
(".gemspec", try_gemspec),
|
||||||
|
];
|
||||||
|
|
||||||
|
// ── Structured parsers ──────────────────────────────────────────────
|
||||||
|
|
||||||
|
fn try_cargo_toml(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("Cargo.toml")).ok()?;
|
||||||
|
let value = content.parse::<toml::Value>().ok()?;
|
||||||
|
let version_str = value.get("package")?.get("version")?.as_str()?;
|
||||||
|
Version::parse(version_str).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_pyproject_toml(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("pyproject.toml")).ok()?;
|
||||||
|
let value = content.parse::<toml::Value>().ok()?;
|
||||||
|
let version_str = value.get("project")?.get("version")?.as_str()?;
|
||||||
|
Version::parse(version_str).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_project_toml_julia(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("Project.toml")).ok()?;
|
||||||
|
let value = content.parse::<toml::Value>().ok()?;
|
||||||
|
let version_str = value.get("version")?.as_str()?;
|
||||||
|
Version::parse(version_str).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_package_json(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("package.json")).ok()?;
|
||||||
|
let value: serde_json::Value = serde_json::from_str(&content).ok()?;
|
||||||
|
let version_str = value.get("version")?.as_str()?;
|
||||||
|
Version::parse(version_str).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_composer_json(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("composer.json")).ok()?;
|
||||||
|
let value: serde_json::Value = serde_json::from_str(&content).ok()?;
|
||||||
|
let version_str = value.get("version")?.as_str()?;
|
||||||
|
Version::parse(version_str).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_pom_xml(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("pom.xml")).ok()?;
|
||||||
|
let doc = roxmltree::Document::parse(&content).ok()?;
|
||||||
|
let project = doc.root_element();
|
||||||
|
// Read <version> that is a direct child of <project>, not inside <parent>
|
||||||
|
for child in project.children() {
|
||||||
|
if child.is_element() && child.tag_name().name() == "version" {
|
||||||
|
if let Some(t) = child.text() {
|
||||||
|
return Version::parse(t.trim()).ok();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_csproj(dir: &Path) -> Option<Version> {
|
||||||
|
// Find any .csproj file in the root (take the first)
|
||||||
|
let csproj = std::fs::read_dir(dir)
|
||||||
|
.ok()?
|
||||||
|
.filter_map(|e| e.ok())
|
||||||
|
.map(|e| e.path())
|
||||||
|
.find(|p| p.extension().map_or(false, |ext| ext == "csproj"))?;
|
||||||
|
let content = std::fs::read_to_string(&csproj).ok()?;
|
||||||
|
let doc = roxmltree::Document::parse(&content).ok()?;
|
||||||
|
// Look for <Version> (preferred) or <VersionPrefix> anywhere in the document
|
||||||
|
for node in doc.descendants() {
|
||||||
|
if node.is_element() {
|
||||||
|
match node.tag_name().name() {
|
||||||
|
"Version" => {
|
||||||
|
if let Some(t) = node.text() {
|
||||||
|
return Version::parse(t.trim()).ok();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"VersionPrefix" => {
|
||||||
|
if let Some(t) = node.text() {
|
||||||
|
return Version::parse(t.trim()).ok();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
// ── Regex-based parsers (code-as-config or simple data formats) ─────
|
||||||
|
|
||||||
|
fn try_pubspec_yaml(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("pubspec.yaml")).ok()?;
|
||||||
|
let re =
|
||||||
|
regex::Regex::new(r##"(?m)^version\s*:\s*['"]?(\d+\.\d+\.\d+[^'"#\s]*)['"]?"##).ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_package_yaml(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("package.yaml")).ok()?;
|
||||||
|
let re =
|
||||||
|
regex::Regex::new(r##"(?m)^version\s*:\s*['"]?(\d+\.\d+\.\d+[^'"#\s]*)['"]?"##).ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_gemspec(dir: &Path) -> Option<Version> {
|
||||||
|
// A directory may contain multiple .gemspec files; take the first.
|
||||||
|
let gemspec = std::fs::read_dir(dir)
|
||||||
|
.ok()?
|
||||||
|
.filter_map(|e| e.ok())
|
||||||
|
.map(|e| e.path())
|
||||||
|
.find(|p| p.extension().map_or(false, |ext| ext == "gemspec"))?;
|
||||||
|
let content = std::fs::read_to_string(&gemspec).ok()?;
|
||||||
|
let re =
|
||||||
|
regex::Regex::new(r#"spec\.version\s*=\s*["'](\d+\.\d+\.\d+[^"']*)["']"#).ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_build_sbt(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("build.sbt")).ok()?;
|
||||||
|
let re = regex::Regex::new(r#"version\s*:=\s*["'](\d+\.\d+\.\d+[^"']*)["']"#).ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_project_clj(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("project.clj")).ok()?;
|
||||||
|
let re = regex::Regex::new(r#"\(defproject\s+[^\s]+\s+"(\d+\.\d+\.\d+[^"]*)""#).ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_mix_exs(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("mix.exs")).ok()?;
|
||||||
|
let re = regex::Regex::new(r#"version\s*:\s*["'](\d+\.\d+\.\d+[^"']*)["']"#).ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_cmake_lists(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("CMakeLists.txt")).ok()?;
|
||||||
|
let re =
|
||||||
|
regex::Regex::new(r"project\s*\([^)]*VERSION\s+(\d+\.\d+\.\d+[^\s)]*)").ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn try_meson_build(dir: &Path) -> Option<Version> {
|
||||||
|
let content = std::fs::read_to_string(dir.join("meson.build")).ok()?;
|
||||||
|
let re =
|
||||||
|
regex::Regex::new(r#"version\s*:\s*["'](\d+\.\d+\.\d+[^"']*)["']"#).ok()?;
|
||||||
|
let caps = re.captures(&content)?;
|
||||||
|
Version::parse(caps.get(1)?.as_str()).ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use crate::git::TagInfo;
|
||||||
|
|
||||||
|
fn make_tag(name: &str, time: i64) -> TagInfo {
|
||||||
|
TagInfo {
|
||||||
|
name: name.to_string(),
|
||||||
|
target: "abc123".to_string(),
|
||||||
|
message: String::new(),
|
||||||
|
time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sort_by_semver_descending() {
|
||||||
|
let mut tags = vec![
|
||||||
|
make_tag("v1.0.0", 100),
|
||||||
|
make_tag("v2.0.0", 200),
|
||||||
|
make_tag("v1.5.0", 150),
|
||||||
|
];
|
||||||
|
sort_tags_by_semver(&mut tags);
|
||||||
|
let names: Vec<&str> = tags.iter().map(|t| t.name.as_str()).collect();
|
||||||
|
assert_eq!(names, vec!["v2.0.0", "v1.5.0", "v1.0.0"]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sort_semver_ties_broken_by_time() {
|
||||||
|
let mut tags = vec![
|
||||||
|
make_tag("v1.0.0", 100),
|
||||||
|
make_tag("v1.0.0", 300),
|
||||||
|
make_tag("v1.0.0", 200),
|
||||||
|
];
|
||||||
|
sort_tags_by_semver(&mut tags);
|
||||||
|
let times: Vec<i64> = tags.iter().map(|t| t.time).collect();
|
||||||
|
assert_eq!(times, vec![300, 200, 100]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sort_non_semver_tags_at_end() {
|
||||||
|
let mut tags = vec![
|
||||||
|
make_tag("release-2024", 400),
|
||||||
|
make_tag("v1.0.0", 100),
|
||||||
|
make_tag("staging", 300),
|
||||||
|
make_tag("v0.2.0", 200),
|
||||||
|
];
|
||||||
|
sort_tags_by_semver(&mut tags);
|
||||||
|
let names: Vec<&str> = tags.iter().map(|t| t.name.as_str()).collect();
|
||||||
|
assert_eq!(names[0], "v1.0.0");
|
||||||
|
assert_eq!(names[1], "v0.2.0");
|
||||||
|
assert_eq!(names[2], "release-2024");
|
||||||
|
assert_eq!(names[3], "staging");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sort_no_prefix_semver() {
|
||||||
|
let mut tags = vec![
|
||||||
|
make_tag("1.0.0", 100),
|
||||||
|
make_tag("2.0.0", 200),
|
||||||
|
make_tag("0.1.0", 50),
|
||||||
|
];
|
||||||
|
sort_tags_by_semver(&mut tags);
|
||||||
|
let names: Vec<&str> = tags.iter().map(|t| t.name.as_str()).collect();
|
||||||
|
assert_eq!(names, vec!["2.0.0", "1.0.0", "0.1.0"]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sort_all_non_semver_by_time() {
|
||||||
|
let mut tags = vec![
|
||||||
|
make_tag("release-2024", 100),
|
||||||
|
make_tag("release-2023", 300),
|
||||||
|
make_tag("beta", 200),
|
||||||
|
];
|
||||||
|
sort_tags_by_semver(&mut tags);
|
||||||
|
let times: Vec<i64> = tags.iter().map(|t| t.time).collect();
|
||||||
|
assert_eq!(times, vec![300, 200, 100]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_cargo_toml() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("Cargo.toml"),
|
||||||
|
"[package]\nname = \"test\"\nversion = \"0.3.0\"\n",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 1);
|
||||||
|
assert_eq!(versions[0].source, "Cargo.toml");
|
||||||
|
assert_eq!(versions[0].version, Version::new(0, 3, 0));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_pyproject_toml() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("pyproject.toml"),
|
||||||
|
"[project]\nname = \"test\"\nversion = \"0.2.1\"\n",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 1);
|
||||||
|
assert_eq!(versions[0].source, "pyproject.toml");
|
||||||
|
assert_eq!(versions[0].version, Version::new(0, 2, 1));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_no_config_files() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert!(versions.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_multiple_configs_returns_all() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("Cargo.toml"),
|
||||||
|
"[package]\nname = \"test\"\nversion = \"0.3.0\"\n",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("pyproject.toml"),
|
||||||
|
"[project]\nname = \"test\"\nversion = \"0.2.1\"\n",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 2);
|
||||||
|
// Order is defined by PARSER_ORDER: Cargo.toml before pyproject.toml
|
||||||
|
assert_eq!(versions[0].source, "Cargo.toml");
|
||||||
|
assert_eq!(versions[0].version, Version::new(0, 3, 0));
|
||||||
|
assert_eq!(versions[1].source, "pyproject.toml");
|
||||||
|
assert_eq!(versions[1].version, Version::new(0, 2, 1));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_package_json() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("package.json"),
|
||||||
|
r#"{"name": "test", "version": "1.2.3"}"#,
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 1);
|
||||||
|
assert_eq!(versions[0].source, "package.json");
|
||||||
|
assert_eq!(versions[0].version, Version::new(1, 2, 3));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_pom_xml() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("pom.xml"),
|
||||||
|
"<project><modelVersion>4.0.0</modelVersion><version>2.0.0</version></project>",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 1);
|
||||||
|
assert_eq!(versions[0].source, "pom.xml");
|
||||||
|
assert_eq!(versions[0].version, Version::new(2, 0, 0));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_pom_xml_ignores_parent_version() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("pom.xml"),
|
||||||
|
"<project><parent><version>3.0.0</version></parent><version>2.0.0</version></project>",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 1);
|
||||||
|
assert_eq!(versions[0].version, Version::new(2, 0, 0));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_pubspec_yaml() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("pubspec.yaml"),
|
||||||
|
"name: myapp\nversion: 1.0.0+1\n",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 1);
|
||||||
|
assert_eq!(versions[0].source, "pubspec.yaml");
|
||||||
|
assert_eq!(
|
||||||
|
versions[0].version,
|
||||||
|
Version::parse("1.0.0+1").unwrap()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_go_mod_skips_all() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(dir.path().join("go.mod"), "module example.com/m\n").unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("Cargo.toml"),
|
||||||
|
"[package]\nname = \"test\"\nversion = \"0.3.0\"\n",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
// go.mod triggers immediate return — Cargo.toml is ignored
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert!(versions.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_read_versions_project_toml_julia() {
|
||||||
|
let dir = tempfile::TempDir::new().unwrap();
|
||||||
|
std::fs::write(
|
||||||
|
dir.path().join("Project.toml"),
|
||||||
|
"name = \"MyPkg\"\nversion = \"0.1.0\"\n",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let versions = read_project_versions(dir.path());
|
||||||
|
assert_eq!(versions.len(), 1);
|
||||||
|
assert_eq!(versions[0].source, "Project.toml");
|
||||||
|
assert_eq!(versions[0].version, Version::new(0, 1, 0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
4068
src/i18n/messages.rs
4068
src/i18n/messages.rs
File diff suppressed because it is too large
Load Diff
@@ -2,6 +2,4 @@ pub mod messages;
|
|||||||
pub mod translator;
|
pub mod translator;
|
||||||
|
|
||||||
pub use messages::Messages;
|
pub use messages::Messages;
|
||||||
pub use translator::Translator;
|
|
||||||
pub use translator::translate_commit_type;
|
|
||||||
pub use translator::translate_changelog_category;
|
pub use translator::translate_changelog_category;
|
||||||
|
|||||||
@@ -7,7 +7,11 @@ pub struct Translator {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl Translator {
|
impl Translator {
|
||||||
pub fn new(language: Language, keep_types_english: bool, keep_changelog_types_english: bool) -> Self {
|
pub fn new(
|
||||||
|
language: Language,
|
||||||
|
keep_types_english: bool,
|
||||||
|
keep_changelog_types_english: bool,
|
||||||
|
) -> Self {
|
||||||
Self {
|
Self {
|
||||||
language,
|
language,
|
||||||
keep_types_english,
|
keep_types_english,
|
||||||
@@ -227,7 +231,11 @@ pub fn translate_commit_type(commit_type: &str, language: Language, keep_english
|
|||||||
translator.translate_commit_type(commit_type)
|
translator.translate_commit_type(commit_type)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn translate_changelog_category(category: &str, language: Language, keep_english: bool) -> String {
|
pub fn translate_changelog_category(
|
||||||
|
category: &str,
|
||||||
|
language: Language,
|
||||||
|
keep_english: bool,
|
||||||
|
) -> String {
|
||||||
let translator = Translator::new(language, true, keep_english);
|
let translator = Translator::new(language, true, keep_english);
|
||||||
translator.translate_changelog_category(category)
|
translator.translate_changelog_category(category)
|
||||||
}
|
}
|
||||||
|
|||||||
7
src/lib.rs
Normal file
7
src/lib.rs
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
pub mod commands;
|
||||||
|
pub mod config;
|
||||||
|
pub mod generator;
|
||||||
|
pub mod git;
|
||||||
|
pub mod i18n;
|
||||||
|
pub mod llm;
|
||||||
|
pub mod utils;
|
||||||
@@ -1,233 +0,0 @@
|
|||||||
use super::{create_http_client, LlmProvider};
|
|
||||||
use anyhow::{bail, Context, Result};
|
|
||||||
use async_trait::async_trait;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::time::Duration;
|
|
||||||
|
|
||||||
/// Anthropic Claude API client
|
|
||||||
pub struct AnthropicClient {
|
|
||||||
api_key: String,
|
|
||||||
model: String,
|
|
||||||
client: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
struct MessagesRequest {
|
|
||||||
model: String,
|
|
||||||
max_tokens: u32,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
temperature: Option<f32>,
|
|
||||||
messages: Vec<AnthropicMessage>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
system: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
|
||||||
struct AnthropicMessage {
|
|
||||||
role: String,
|
|
||||||
content: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct MessagesResponse {
|
|
||||||
content: Vec<ContentBlock>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ContentBlock {
|
|
||||||
#[serde(rename = "type")]
|
|
||||||
content_type: String,
|
|
||||||
text: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ErrorResponse {
|
|
||||||
error: AnthropicError,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct AnthropicError {
|
|
||||||
#[serde(rename = "type")]
|
|
||||||
error_type: String,
|
|
||||||
message: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl AnthropicClient {
|
|
||||||
/// Create new Anthropic client
|
|
||||||
pub fn new(api_key: &str, model: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set timeout
|
|
||||||
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
|
|
||||||
self.client = create_http_client(timeout)?;
|
|
||||||
Ok(self)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// List available models
|
|
||||||
pub async fn list_models(&self) -> Result<Vec<String>> {
|
|
||||||
// Anthropic doesn't have a models API endpoint, return predefined list
|
|
||||||
Ok(ANTHROPIC_MODELS.iter().map(|&m| m.to_string()).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Validate API key
|
|
||||||
pub async fn validate_key(&self) -> Result<bool> {
|
|
||||||
let url = "https://api.anthropic.com/v1/messages";
|
|
||||||
|
|
||||||
let request = MessagesRequest {
|
|
||||||
model: self.model.clone(),
|
|
||||||
max_tokens: 5,
|
|
||||||
temperature: Some(0.0),
|
|
||||||
messages: vec![AnthropicMessage {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: "Hi".to_string(),
|
|
||||||
}],
|
|
||||||
system: None,
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(url)
|
|
||||||
.header("x-api-key", &self.api_key)
|
|
||||||
.header("anthropic-version", "2023-06-01")
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await;
|
|
||||||
|
|
||||||
match response {
|
|
||||||
Ok(resp) => {
|
|
||||||
if resp.status().is_success() {
|
|
||||||
Ok(true)
|
|
||||||
} else {
|
|
||||||
let status = resp.status();
|
|
||||||
if status.as_u16() == 401 {
|
|
||||||
Ok(false)
|
|
||||||
} else {
|
|
||||||
let text = resp.text().await.unwrap_or_default();
|
|
||||||
bail!("Anthropic API error: {} - {}", status, text)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(e) => Err(e.into()),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl LlmProvider for AnthropicClient {
|
|
||||||
async fn generate(&self, prompt: &str) -> Result<String> {
|
|
||||||
let messages = vec![AnthropicMessage {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: prompt.to_string(),
|
|
||||||
}];
|
|
||||||
|
|
||||||
self.messages_request(messages, None).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
|
|
||||||
let messages = vec![AnthropicMessage {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: user.to_string(),
|
|
||||||
}];
|
|
||||||
|
|
||||||
let system = if system.is_empty() {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(system.to_string())
|
|
||||||
};
|
|
||||||
|
|
||||||
self.messages_request(messages, system).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_available(&self) -> bool {
|
|
||||||
self.validate_key().await.unwrap_or(false)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"anthropic"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl AnthropicClient {
|
|
||||||
async fn messages_request(
|
|
||||||
&self,
|
|
||||||
messages: Vec<AnthropicMessage>,
|
|
||||||
system: Option<String>,
|
|
||||||
) -> Result<String> {
|
|
||||||
let url = "https://api.anthropic.com/v1/messages";
|
|
||||||
|
|
||||||
let request = MessagesRequest {
|
|
||||||
model: self.model.clone(),
|
|
||||||
max_tokens: 500,
|
|
||||||
temperature: Some(0.7),
|
|
||||||
messages,
|
|
||||||
system,
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(url)
|
|
||||||
.header("x-api-key", &self.api_key)
|
|
||||||
.header("anthropic-version", "2023-06-01")
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to send request to Anthropic")?;
|
|
||||||
|
|
||||||
let status = response.status();
|
|
||||||
|
|
||||||
if !status.is_success() {
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
|
|
||||||
// Try to parse error
|
|
||||||
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
|
|
||||||
bail!("Anthropic API error: {} ({})", error.error.message, error.error.error_type);
|
|
||||||
}
|
|
||||||
|
|
||||||
bail!("Anthropic API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: MessagesResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse Anthropic response")?;
|
|
||||||
|
|
||||||
result.content
|
|
||||||
.into_iter()
|
|
||||||
.find(|c| c.content_type == "text")
|
|
||||||
.map(|c| c.text.trim().to_string())
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("No text response from Anthropic"))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Available Anthropic models
|
|
||||||
pub const ANTHROPIC_MODELS: &[&str] = &[
|
|
||||||
"claude-3-opus-20240229",
|
|
||||||
"claude-3-sonnet-20240229",
|
|
||||||
"claude-3-haiku-20240307",
|
|
||||||
"claude-2.1",
|
|
||||||
"claude-2.0",
|
|
||||||
"claude-instant-1.2",
|
|
||||||
];
|
|
||||||
|
|
||||||
/// Check if a model name is valid
|
|
||||||
pub fn is_valid_model(model: &str) -> bool {
|
|
||||||
ANTHROPIC_MODELS.contains(&model)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_model_validation() {
|
|
||||||
assert!(is_valid_model("claude-3-sonnet-20240229"));
|
|
||||||
assert!(!is_valid_model("invalid-model"));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,243 +0,0 @@
|
|||||||
use super::{create_http_client, LlmProvider};
|
|
||||||
use anyhow::{bail, Context, Result};
|
|
||||||
use async_trait::async_trait;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::time::Duration;
|
|
||||||
|
|
||||||
/// DeepSeek API client
|
|
||||||
pub struct DeepSeekClient {
|
|
||||||
base_url: String,
|
|
||||||
api_key: String,
|
|
||||||
model: String,
|
|
||||||
client: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
struct ChatCompletionRequest {
|
|
||||||
model: String,
|
|
||||||
messages: Vec<Message>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
max_tokens: Option<u32>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
temperature: Option<f32>,
|
|
||||||
stream: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
|
||||||
struct Message {
|
|
||||||
role: String,
|
|
||||||
content: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ChatCompletionResponse {
|
|
||||||
choices: Vec<Choice>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct Choice {
|
|
||||||
message: Message,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ErrorResponse {
|
|
||||||
error: ApiError,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ApiError {
|
|
||||||
message: String,
|
|
||||||
#[serde(rename = "type")]
|
|
||||||
error_type: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DeepSeekClient {
|
|
||||||
/// Create new DeepSeek client
|
|
||||||
pub fn new(api_key: &str, model: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
base_url: "https://api.deepseek.com/v1".to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Create with custom base URL
|
|
||||||
pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
base_url: base_url.trim_end_matches('/').to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set timeout
|
|
||||||
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
|
|
||||||
self.client = create_http_client(timeout)?;
|
|
||||||
Ok(self)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// List available models
|
|
||||||
pub async fn list_models(&self) -> Result<Vec<String>> {
|
|
||||||
let url = format!("{}/models", self.base_url);
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.get(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to list DeepSeek models")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
bail!("DeepSeek API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct ModelsResponse {
|
|
||||||
data: Vec<Model>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct Model {
|
|
||||||
id: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ModelsResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse DeepSeek response")?;
|
|
||||||
|
|
||||||
Ok(result.data.into_iter().map(|m| m.id).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Validate API key
|
|
||||||
pub async fn validate_key(&self) -> Result<bool> {
|
|
||||||
match self.list_models().await {
|
|
||||||
Ok(_) => Ok(true),
|
|
||||||
Err(e) => {
|
|
||||||
let err_str = e.to_string();
|
|
||||||
if err_str.contains("401") || err_str.contains("Unauthorized") {
|
|
||||||
Ok(false)
|
|
||||||
} else {
|
|
||||||
Err(e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl LlmProvider for DeepSeekClient {
|
|
||||||
async fn generate(&self, prompt: &str) -> Result<String> {
|
|
||||||
let messages = vec![
|
|
||||||
Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: prompt.to_string(),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
|
|
||||||
let mut messages = vec![];
|
|
||||||
|
|
||||||
if !system.is_empty() {
|
|
||||||
messages.push(Message {
|
|
||||||
role: "system".to_string(),
|
|
||||||
content: system.to_string(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
messages.push(Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: user.to_string(),
|
|
||||||
});
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_available(&self) -> bool {
|
|
||||||
self.validate_key().await.unwrap_or(false)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"deepseek"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DeepSeekClient {
|
|
||||||
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
|
|
||||||
let url = format!("{}/chat/completions", self.base_url);
|
|
||||||
|
|
||||||
let request = ChatCompletionRequest {
|
|
||||||
model: self.model.clone(),
|
|
||||||
messages,
|
|
||||||
max_tokens: Some(500),
|
|
||||||
temperature: Some(0.7),
|
|
||||||
stream: false,
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to send request to DeepSeek")?;
|
|
||||||
|
|
||||||
let status = response.status();
|
|
||||||
|
|
||||||
if !status.is_success() {
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
|
|
||||||
// Try to parse error
|
|
||||||
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
|
|
||||||
bail!("DeepSeek API error: {} ({})", error.error.message, error.error.error_type);
|
|
||||||
}
|
|
||||||
|
|
||||||
bail!("DeepSeek API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ChatCompletionResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse DeepSeek response")?;
|
|
||||||
|
|
||||||
result.choices
|
|
||||||
.into_iter()
|
|
||||||
.next()
|
|
||||||
.map(|c| c.message.content.trim().to_string())
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("No response from DeepSeek"))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Available DeepSeek models
|
|
||||||
pub const DEEPSEEK_MODELS: &[&str] = &[
|
|
||||||
"deepseek-chat",
|
|
||||||
"deepseek-coder",
|
|
||||||
];
|
|
||||||
|
|
||||||
/// Check if a model name is valid
|
|
||||||
pub fn is_valid_model(model: &str) -> bool {
|
|
||||||
DEEPSEEK_MODELS.contains(&model)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_model_validation() {
|
|
||||||
assert!(is_valid_model("deepseek-chat"));
|
|
||||||
assert!(!is_valid_model("invalid-model"));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
244
src/llm/kimi.rs
244
src/llm/kimi.rs
@@ -1,244 +0,0 @@
|
|||||||
use super::{create_http_client, LlmProvider};
|
|
||||||
use anyhow::{bail, Context, Result};
|
|
||||||
use async_trait::async_trait;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::time::Duration;
|
|
||||||
|
|
||||||
/// Kimi API client (Moonshot AI)
|
|
||||||
pub struct KimiClient {
|
|
||||||
base_url: String,
|
|
||||||
api_key: String,
|
|
||||||
model: String,
|
|
||||||
client: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
struct ChatCompletionRequest {
|
|
||||||
model: String,
|
|
||||||
messages: Vec<Message>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
max_tokens: Option<u32>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
temperature: Option<f32>,
|
|
||||||
stream: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
|
||||||
struct Message {
|
|
||||||
role: String,
|
|
||||||
content: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ChatCompletionResponse {
|
|
||||||
choices: Vec<Choice>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct Choice {
|
|
||||||
message: Message,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ErrorResponse {
|
|
||||||
error: ApiError,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ApiError {
|
|
||||||
message: String,
|
|
||||||
#[serde(rename = "type")]
|
|
||||||
error_type: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl KimiClient {
|
|
||||||
/// Create new Kimi client
|
|
||||||
pub fn new(api_key: &str, model: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
base_url: "https://api.moonshot.cn/v1".to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Create with custom base URL
|
|
||||||
pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
base_url: base_url.trim_end_matches('/').to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set timeout
|
|
||||||
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
|
|
||||||
self.client = create_http_client(timeout)?;
|
|
||||||
Ok(self)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// List available models
|
|
||||||
pub async fn list_models(&self) -> Result<Vec<String>> {
|
|
||||||
let url = format!("{}/models", self.base_url);
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.get(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to list Kimi models")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
bail!("Kimi API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct ModelsResponse {
|
|
||||||
data: Vec<Model>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct Model {
|
|
||||||
id: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ModelsResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse Kimi response")?;
|
|
||||||
|
|
||||||
Ok(result.data.into_iter().map(|m| m.id).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Validate API key
|
|
||||||
pub async fn validate_key(&self) -> Result<bool> {
|
|
||||||
match self.list_models().await {
|
|
||||||
Ok(_) => Ok(true),
|
|
||||||
Err(e) => {
|
|
||||||
let err_str = e.to_string();
|
|
||||||
if err_str.contains("401") || err_str.contains("Unauthorized") {
|
|
||||||
Ok(false)
|
|
||||||
} else {
|
|
||||||
Err(e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl LlmProvider for KimiClient {
|
|
||||||
async fn generate(&self, prompt: &str) -> Result<String> {
|
|
||||||
let messages = vec![
|
|
||||||
Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: prompt.to_string(),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
|
|
||||||
let mut messages = vec![];
|
|
||||||
|
|
||||||
if !system.is_empty() {
|
|
||||||
messages.push(Message {
|
|
||||||
role: "system".to_string(),
|
|
||||||
content: system.to_string(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
messages.push(Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: user.to_string(),
|
|
||||||
});
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_available(&self) -> bool {
|
|
||||||
self.validate_key().await.unwrap_or(false)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"kimi"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl KimiClient {
|
|
||||||
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
|
|
||||||
let url = format!("{}/chat/completions", self.base_url);
|
|
||||||
|
|
||||||
let request = ChatCompletionRequest {
|
|
||||||
model: self.model.clone(),
|
|
||||||
messages,
|
|
||||||
max_tokens: Some(500),
|
|
||||||
temperature: Some(0.7),
|
|
||||||
stream: false,
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to send request to Kimi")?;
|
|
||||||
|
|
||||||
let status = response.status();
|
|
||||||
|
|
||||||
if !status.is_success() {
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
|
|
||||||
// Try to parse error
|
|
||||||
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
|
|
||||||
bail!("Kimi API error: {} ({})", error.error.message, error.error.error_type);
|
|
||||||
}
|
|
||||||
|
|
||||||
bail!("Kimi API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ChatCompletionResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse Kimi response")?;
|
|
||||||
|
|
||||||
result.choices
|
|
||||||
.into_iter()
|
|
||||||
.next()
|
|
||||||
.map(|c| c.message.content.trim().to_string())
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("No response from Kimi"))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Available Kimi models
|
|
||||||
pub const KIMI_MODELS: &[&str] = &[
|
|
||||||
"moonshot-v1-8k",
|
|
||||||
"moonshot-v1-32k",
|
|
||||||
"moonshot-v1-128k",
|
|
||||||
];
|
|
||||||
|
|
||||||
/// Check if a model name is valid
|
|
||||||
pub fn is_valid_model(model: &str) -> bool {
|
|
||||||
KIMI_MODELS.contains(&model)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_model_validation() {
|
|
||||||
assert!(is_valid_model("moonshot-v1-8k"));
|
|
||||||
assert!(!is_valid_model("invalid-model"));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
1021
src/llm/mod.rs
1021
src/llm/mod.rs
File diff suppressed because it is too large
Load Diff
@@ -1,206 +0,0 @@
|
|||||||
use super::{create_http_client, LlmProvider};
|
|
||||||
use anyhow::{Context, Result};
|
|
||||||
use async_trait::async_trait;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::time::Duration;
|
|
||||||
|
|
||||||
/// Ollama API client
|
|
||||||
pub struct OllamaClient {
|
|
||||||
base_url: String,
|
|
||||||
model: String,
|
|
||||||
client: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
struct GenerateRequest {
|
|
||||||
model: String,
|
|
||||||
prompt: String,
|
|
||||||
system: Option<String>,
|
|
||||||
stream: bool,
|
|
||||||
options: GenerationOptions,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Default)]
|
|
||||||
struct GenerationOptions {
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
temperature: Option<f32>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
num_predict: Option<u32>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct GenerateResponse {
|
|
||||||
response: String,
|
|
||||||
done: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ListModelsResponse {
|
|
||||||
models: Vec<ModelInfo>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ModelInfo {
|
|
||||||
name: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl OllamaClient {
|
|
||||||
/// Create new Ollama client
|
|
||||||
pub fn new(base_url: &str, model: &str) -> Self {
|
|
||||||
let client = create_http_client(Duration::from_secs(120))
|
|
||||||
.expect("Failed to create HTTP client");
|
|
||||||
|
|
||||||
Self {
|
|
||||||
base_url: base_url.trim_end_matches('/').to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set timeout
|
|
||||||
pub fn with_timeout(mut self, timeout: Duration) -> Self {
|
|
||||||
self.client = create_http_client(timeout)
|
|
||||||
.expect("Failed to create HTTP client");
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// List available models
|
|
||||||
pub async fn list_models(&self) -> Result<Vec<String>> {
|
|
||||||
let url = format!("{}/api/tags", self.base_url);
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.get(&url)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to list Ollama models")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
anyhow::bail!("Ollama API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ListModelsResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse Ollama response")?;
|
|
||||||
|
|
||||||
Ok(result.models.into_iter().map(|m| m.name).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Pull a model
|
|
||||||
pub async fn pull_model(&self, model: &str) -> Result<()> {
|
|
||||||
let url = format!("{}/api/pull", self.base_url);
|
|
||||||
|
|
||||||
let request = serde_json::json!({
|
|
||||||
"name": model,
|
|
||||||
"stream": false,
|
|
||||||
});
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(&url)
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to pull Ollama model")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
anyhow::bail!("Ollama pull error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Check if model exists
|
|
||||||
pub async fn model_exists(&self, model: &str) -> bool {
|
|
||||||
match self.list_models().await {
|
|
||||||
Ok(models) => models.contains(&model.to_string()),
|
|
||||||
Err(_) => false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl LlmProvider for OllamaClient {
|
|
||||||
async fn generate(&self, prompt: &str) -> Result<String> {
|
|
||||||
self.generate_with_system("", prompt).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
|
|
||||||
let url = format!("{}/api/generate", self.base_url);
|
|
||||||
|
|
||||||
let system = if system.is_empty() {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(system.to_string())
|
|
||||||
};
|
|
||||||
|
|
||||||
let request = GenerateRequest {
|
|
||||||
model: self.model.clone(),
|
|
||||||
prompt: user.to_string(),
|
|
||||||
system,
|
|
||||||
stream: false,
|
|
||||||
options: GenerationOptions {
|
|
||||||
temperature: Some(0.7),
|
|
||||||
num_predict: Some(500),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(&url)
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to send request to Ollama")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
anyhow::bail!("Ollama API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: GenerateResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse Ollama response")?;
|
|
||||||
|
|
||||||
Ok(result.response.trim().to_string())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_available(&self) -> bool {
|
|
||||||
let url = format!("{}/api/tags", self.base_url);
|
|
||||||
|
|
||||||
match self.client.get(&url).send().await {
|
|
||||||
Ok(response) => response.status().is_success(),
|
|
||||||
Err(_) => false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"ollama"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
// These tests require a running Ollama server
|
|
||||||
#[tokio::test]
|
|
||||||
#[ignore]
|
|
||||||
async fn test_ollama_connection() {
|
|
||||||
let client = OllamaClient::new("http://localhost:11434", "llama2");
|
|
||||||
assert!(client.is_available().await);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
#[ignore]
|
|
||||||
async fn test_ollama_generate() {
|
|
||||||
let client = OllamaClient::new("http://localhost:11434", "llama2");
|
|
||||||
let response = client.generate("Hello, how are you?").await;
|
|
||||||
assert!(response.is_ok());
|
|
||||||
println!("Response: {}", response.unwrap());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,345 +0,0 @@
|
|||||||
use super::{create_http_client, LlmProvider};
|
|
||||||
use anyhow::{bail, Context, Result};
|
|
||||||
use async_trait::async_trait;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::time::Duration;
|
|
||||||
|
|
||||||
/// OpenAI API client
|
|
||||||
pub struct OpenAiClient {
|
|
||||||
base_url: String,
|
|
||||||
api_key: String,
|
|
||||||
model: String,
|
|
||||||
client: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
struct ChatCompletionRequest {
|
|
||||||
model: String,
|
|
||||||
messages: Vec<Message>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
max_tokens: Option<u32>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
temperature: Option<f32>,
|
|
||||||
stream: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
|
||||||
struct Message {
|
|
||||||
role: String,
|
|
||||||
content: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ChatCompletionResponse {
|
|
||||||
choices: Vec<Choice>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct Choice {
|
|
||||||
message: Message,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ErrorResponse {
|
|
||||||
error: ApiError,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ApiError {
|
|
||||||
message: String,
|
|
||||||
#[serde(rename = "type")]
|
|
||||||
error_type: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl OpenAiClient {
|
|
||||||
/// Create new OpenAI client
|
|
||||||
pub fn new(base_url: &str, api_key: &str, model: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
base_url: base_url.trim_end_matches('/').to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set timeout
|
|
||||||
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
|
|
||||||
self.client = create_http_client(timeout)?;
|
|
||||||
Ok(self)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// List available models
|
|
||||||
pub async fn list_models(&self) -> Result<Vec<String>> {
|
|
||||||
let url = format!("{}/models", self.base_url);
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.get(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to list OpenAI models")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
bail!("OpenAI API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct ModelsResponse {
|
|
||||||
data: Vec<Model>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct Model {
|
|
||||||
id: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ModelsResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse OpenAI response")?;
|
|
||||||
|
|
||||||
Ok(result.data.into_iter().map(|m| m.id).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Validate API key
|
|
||||||
pub async fn validate_key(&self) -> Result<bool> {
|
|
||||||
match self.list_models().await {
|
|
||||||
Ok(_) => Ok(true),
|
|
||||||
Err(e) => {
|
|
||||||
let err_str = e.to_string();
|
|
||||||
if err_str.contains("401") || err_str.contains("Unauthorized") {
|
|
||||||
Ok(false)
|
|
||||||
} else {
|
|
||||||
Err(e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl LlmProvider for OpenAiClient {
|
|
||||||
async fn generate(&self, prompt: &str) -> Result<String> {
|
|
||||||
let messages = vec![
|
|
||||||
Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: prompt.to_string(),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
|
|
||||||
let mut messages = vec![];
|
|
||||||
|
|
||||||
if !system.is_empty() {
|
|
||||||
messages.push(Message {
|
|
||||||
role: "system".to_string(),
|
|
||||||
content: system.to_string(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
messages.push(Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: user.to_string(),
|
|
||||||
});
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_available(&self) -> bool {
|
|
||||||
self.validate_key().await.unwrap_or(false)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"openai"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl OpenAiClient {
|
|
||||||
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
|
|
||||||
let url = format!("{}/chat/completions", self.base_url);
|
|
||||||
|
|
||||||
let request = ChatCompletionRequest {
|
|
||||||
model: self.model.clone(),
|
|
||||||
messages,
|
|
||||||
max_tokens: Some(500),
|
|
||||||
temperature: Some(0.7),
|
|
||||||
stream: false,
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to send request to OpenAI")?;
|
|
||||||
|
|
||||||
let status = response.status();
|
|
||||||
|
|
||||||
if !status.is_success() {
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
|
|
||||||
// Try to parse error
|
|
||||||
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
|
|
||||||
bail!("OpenAI API error: {} ({})", error.error.message, error.error.error_type);
|
|
||||||
}
|
|
||||||
|
|
||||||
bail!("OpenAI API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ChatCompletionResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse OpenAI response")?;
|
|
||||||
|
|
||||||
result.choices
|
|
||||||
.into_iter()
|
|
||||||
.next()
|
|
||||||
.map(|c| c.message.content.trim().to_string())
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("No response from OpenAI"))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Azure OpenAI client (extends OpenAI with Azure-specific config)
|
|
||||||
pub struct AzureOpenAiClient {
|
|
||||||
endpoint: String,
|
|
||||||
api_key: String,
|
|
||||||
deployment: String,
|
|
||||||
api_version: String,
|
|
||||||
client: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl AzureOpenAiClient {
|
|
||||||
/// Create new Azure OpenAI client
|
|
||||||
pub fn new(
|
|
||||||
endpoint: &str,
|
|
||||||
api_key: &str,
|
|
||||||
deployment: &str,
|
|
||||||
api_version: &str,
|
|
||||||
) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
endpoint: endpoint.trim_end_matches('/').to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
deployment: deployment.to_string(),
|
|
||||||
api_version: api_version.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
|
|
||||||
let url = format!(
|
|
||||||
"{}/openai/deployments/{}/chat/completions?api-version={}",
|
|
||||||
self.endpoint, self.deployment, self.api_version
|
|
||||||
);
|
|
||||||
|
|
||||||
let request = ChatCompletionRequest {
|
|
||||||
model: self.deployment.clone(),
|
|
||||||
messages,
|
|
||||||
max_tokens: Some(500),
|
|
||||||
temperature: Some(0.7),
|
|
||||||
stream: false,
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(&url)
|
|
||||||
.header("api-key", &self.api_key)
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to send request to Azure OpenAI")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
bail!("Azure OpenAI API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ChatCompletionResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse Azure OpenAI response")?;
|
|
||||||
|
|
||||||
result.choices
|
|
||||||
.into_iter()
|
|
||||||
.next()
|
|
||||||
.map(|c| c.message.content.trim().to_string())
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("No response from Azure OpenAI"))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl LlmProvider for AzureOpenAiClient {
|
|
||||||
async fn generate(&self, prompt: &str) -> Result<String> {
|
|
||||||
let messages = vec![
|
|
||||||
Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: prompt.to_string(),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
|
|
||||||
let mut messages = vec![];
|
|
||||||
|
|
||||||
if !system.is_empty() {
|
|
||||||
messages.push(Message {
|
|
||||||
role: "system".to_string(),
|
|
||||||
content: system.to_string(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
messages.push(Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: user.to_string(),
|
|
||||||
});
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_available(&self) -> bool {
|
|
||||||
// Simple check - try to make a minimal request
|
|
||||||
let url = format!(
|
|
||||||
"{}/openai/deployments/{}/chat/completions?api-version={}",
|
|
||||||
self.endpoint, self.deployment, self.api_version
|
|
||||||
);
|
|
||||||
|
|
||||||
let request = ChatCompletionRequest {
|
|
||||||
model: self.deployment.clone(),
|
|
||||||
messages: vec![Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: "Hi".to_string(),
|
|
||||||
}],
|
|
||||||
max_tokens: Some(5),
|
|
||||||
temperature: Some(0.0),
|
|
||||||
stream: false,
|
|
||||||
};
|
|
||||||
|
|
||||||
match self.client
|
|
||||||
.post(&url)
|
|
||||||
.header("api-key", &self.api_key)
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(response) => response.status().is_success(),
|
|
||||||
Err(_) => false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"azure-openai"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,257 +0,0 @@
|
|||||||
use super::{create_http_client, LlmProvider};
|
|
||||||
use anyhow::{bail, Context, Result};
|
|
||||||
use async_trait::async_trait;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::time::Duration;
|
|
||||||
|
|
||||||
/// OpenRouter API client
|
|
||||||
pub struct OpenRouterClient {
|
|
||||||
base_url: String,
|
|
||||||
api_key: String,
|
|
||||||
model: String,
|
|
||||||
client: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
struct ChatCompletionRequest {
|
|
||||||
model: String,
|
|
||||||
messages: Vec<Message>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
max_tokens: Option<u32>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
temperature: Option<f32>,
|
|
||||||
stream: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
|
||||||
struct Message {
|
|
||||||
role: String,
|
|
||||||
content: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ChatCompletionResponse {
|
|
||||||
choices: Vec<Choice>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct Choice {
|
|
||||||
message: Message,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ErrorResponse {
|
|
||||||
error: ApiError,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
struct ApiError {
|
|
||||||
message: String,
|
|
||||||
#[serde(rename = "type")]
|
|
||||||
error_type: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl OpenRouterClient {
|
|
||||||
/// Create new OpenRouter client
|
|
||||||
pub fn new(api_key: &str, model: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
base_url: "https://openrouter.ai/api/v1".to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Create with custom base URL
|
|
||||||
pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> {
|
|
||||||
let client = create_http_client(Duration::from_secs(60))?;
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
base_url: base_url.trim_end_matches('/').to_string(),
|
|
||||||
api_key: api_key.to_string(),
|
|
||||||
model: model.to_string(),
|
|
||||||
client,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set timeout
|
|
||||||
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
|
|
||||||
self.client = create_http_client(timeout)?;
|
|
||||||
Ok(self)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// List available models
|
|
||||||
pub async fn list_models(&self) -> Result<Vec<String>> {
|
|
||||||
let url = format!("{}/models", self.base_url);
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.get(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.header("HTTP-Referer", "https://quicommit.dev")
|
|
||||||
.header("X-Title", "QuiCommit")
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to list OpenRouter models")?;
|
|
||||||
|
|
||||||
if !response.status().is_success() {
|
|
||||||
let status = response.status();
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
bail!("OpenRouter API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct ModelsResponse {
|
|
||||||
data: Vec<Model>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct Model {
|
|
||||||
id: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ModelsResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse OpenRouter response")?;
|
|
||||||
|
|
||||||
Ok(result.data.into_iter().map(|m| m.id).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Validate API key
|
|
||||||
pub async fn validate_key(&self) -> Result<bool> {
|
|
||||||
match self.list_models().await {
|
|
||||||
Ok(_) => Ok(true),
|
|
||||||
Err(e) => {
|
|
||||||
let err_str = e.to_string();
|
|
||||||
if err_str.contains("401") || err_str.contains("Unauthorized") {
|
|
||||||
Ok(false)
|
|
||||||
} else {
|
|
||||||
Err(e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl LlmProvider for OpenRouterClient {
|
|
||||||
async fn generate(&self, prompt: &str) -> Result<String> {
|
|
||||||
let messages = vec![
|
|
||||||
Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: prompt.to_string(),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
|
|
||||||
let mut messages = vec![];
|
|
||||||
|
|
||||||
if !system.is_empty() {
|
|
||||||
messages.push(Message {
|
|
||||||
role: "system".to_string(),
|
|
||||||
content: system.to_string(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
messages.push(Message {
|
|
||||||
role: "user".to_string(),
|
|
||||||
content: user.to_string(),
|
|
||||||
});
|
|
||||||
|
|
||||||
self.chat_completion(messages).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn is_available(&self) -> bool {
|
|
||||||
self.validate_key().await.unwrap_or(false)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn name(&self) -> &str {
|
|
||||||
"openrouter"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl OpenRouterClient {
|
|
||||||
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
|
|
||||||
let url = format!("{}/chat/completions", self.base_url);
|
|
||||||
|
|
||||||
let request = ChatCompletionRequest {
|
|
||||||
model: self.model.clone(),
|
|
||||||
messages,
|
|
||||||
max_tokens: Some(500),
|
|
||||||
temperature: Some(0.7),
|
|
||||||
stream: false,
|
|
||||||
};
|
|
||||||
|
|
||||||
let response = self.client
|
|
||||||
.post(&url)
|
|
||||||
.header("Authorization", format!("Bearer {}", self.api_key))
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.header("HTTP-Referer", "https://quicommit.dev")
|
|
||||||
.header("X-Title", "QuiCommit")
|
|
||||||
.json(&request)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.context("Failed to send request to OpenRouter")?;
|
|
||||||
|
|
||||||
let status = response.status();
|
|
||||||
|
|
||||||
if !status.is_success() {
|
|
||||||
let text = response.text().await.unwrap_or_default();
|
|
||||||
|
|
||||||
// Try to parse error
|
|
||||||
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
|
|
||||||
bail!("OpenRouter API error: {} ({})", error.error.message, error.error.error_type);
|
|
||||||
}
|
|
||||||
|
|
||||||
bail!("OpenRouter API error: {} - {}", status, text);
|
|
||||||
}
|
|
||||||
|
|
||||||
let result: ChatCompletionResponse = response
|
|
||||||
.json()
|
|
||||||
.await
|
|
||||||
.context("Failed to parse OpenRouter response")?;
|
|
||||||
|
|
||||||
result.choices
|
|
||||||
.into_iter()
|
|
||||||
.next()
|
|
||||||
.map(|c| c.message.content.trim().to_string())
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("No response from OpenRouter"))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Popular OpenRouter models
|
|
||||||
pub const OPENROUTER_MODELS: &[&str] = &[
|
|
||||||
"openai/gpt-3.5-turbo",
|
|
||||||
"openai/gpt-4",
|
|
||||||
"openai/gpt-4-turbo",
|
|
||||||
"anthropic/claude-3-opus",
|
|
||||||
"anthropic/claude-3-sonnet",
|
|
||||||
"anthropic/claude-3-haiku",
|
|
||||||
"google/gemini-pro",
|
|
||||||
"meta-llama/llama-2-70b-chat",
|
|
||||||
"mistralai/mixtral-8x7b-instruct",
|
|
||||||
"01-ai/yi-34b-chat",
|
|
||||||
];
|
|
||||||
|
|
||||||
/// Check if a model name is valid
|
|
||||||
pub fn is_valid_model(model: &str) -> bool {
|
|
||||||
// Since OpenRouter supports many models, we'll allow any model name
|
|
||||||
// but provide some popular ones as suggestions
|
|
||||||
true
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_model_validation() {
|
|
||||||
assert!(is_valid_model("openai/gpt-4"));
|
|
||||||
assert!(is_valid_model("custom/model"));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
281
src/llm/parsing.rs
Normal file
281
src/llm/parsing.rs
Normal file
@@ -0,0 +1,281 @@
|
|||||||
|
use anyhow::{Result, bail};
|
||||||
|
|
||||||
|
/// Parse commit response from LLM
|
||||||
|
pub(crate) fn parse_commit_response(
|
||||||
|
response: &str,
|
||||||
|
format: crate::config::CommitFormat,
|
||||||
|
) -> Result<GeneratedCommit> {
|
||||||
|
// Clean markdown code fences from the response
|
||||||
|
let cleaned = strip_code_fences(response);
|
||||||
|
|
||||||
|
let lines: Vec<&str> = cleaned
|
||||||
|
.lines()
|
||||||
|
.map(|l| l.trim())
|
||||||
|
.filter(|l| !l.is_empty())
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
if lines.is_empty() {
|
||||||
|
let preview: String = response.chars().take(200).collect();
|
||||||
|
bail!(
|
||||||
|
"LLM returned empty or whitespace-only response. \
|
||||||
|
Raw response preview: '{}'. \
|
||||||
|
Hint: If using DeepSeek/Kimi with thinking enabled, \
|
||||||
|
the model may have returned reasoning_content only. \
|
||||||
|
Try disabling thinking mode or switching models.",
|
||||||
|
preview
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Find the line most likely to be the commit subject
|
||||||
|
let first_line = find_commit_subject_line(&lines, format);
|
||||||
|
|
||||||
|
// Parse based on format
|
||||||
|
match format {
|
||||||
|
crate::config::CommitFormat::Conventional => {
|
||||||
|
parse_conventional_commit(first_line, &lines, response)
|
||||||
|
}
|
||||||
|
crate::config::CommitFormat::Commitlint => {
|
||||||
|
parse_commitlint_commit(first_line, &lines, response)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Remove surrounding markdown code fences (```) from LLM output
|
||||||
|
fn strip_code_fences(response: &str) -> String {
|
||||||
|
let mut lines: Vec<&str> = response.lines().collect();
|
||||||
|
|
||||||
|
// Strip leading fence lines (``` or ```lang)
|
||||||
|
while lines.first().map_or(false, |l| l.trim().starts_with("```")) {
|
||||||
|
lines.remove(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Strip trailing fence lines
|
||||||
|
while lines.last().map_or(false, |l| l.trim() == "```") {
|
||||||
|
lines.pop();
|
||||||
|
}
|
||||||
|
|
||||||
|
lines.join("\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Find the line that is most likely the commit subject among extracted lines
|
||||||
|
fn find_commit_subject_line<'a>(
|
||||||
|
lines: &[&'a str],
|
||||||
|
format: crate::config::CommitFormat,
|
||||||
|
) -> &'a str {
|
||||||
|
let valid_types = crate::utils::validators::get_commit_types(matches!(
|
||||||
|
format,
|
||||||
|
crate::config::CommitFormat::Commitlint
|
||||||
|
));
|
||||||
|
|
||||||
|
// First pass: line starting with a known type that also has proper syntax
|
||||||
|
// (e.g. "type:", "type(scope):", "type!:")
|
||||||
|
for &line in lines {
|
||||||
|
let trimmed = line.trim();
|
||||||
|
for &t in valid_types {
|
||||||
|
if let Some(rest) = trimmed.strip_prefix(t) {
|
||||||
|
if rest.starts_with(':') || rest.starts_with('(') || rest.starts_with("!:") {
|
||||||
|
return trimmed;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Second pass: any line containing a colon (generic "prefix: description")
|
||||||
|
for &line in lines {
|
||||||
|
if line.contains(':') {
|
||||||
|
return line.trim();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Fallback: return the first line as-is
|
||||||
|
lines[0].trim()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_conventional_commit(
|
||||||
|
first_line: &str,
|
||||||
|
lines: &[&str],
|
||||||
|
raw_response: &str,
|
||||||
|
) -> Result<GeneratedCommit> {
|
||||||
|
// Parse: type(scope)!: description
|
||||||
|
let parts: Vec<&str> = first_line.splitn(2, ':').collect();
|
||||||
|
if parts.len() != 2 {
|
||||||
|
let preview: String = raw_response.chars().take(300).collect();
|
||||||
|
bail!(
|
||||||
|
"Invalid conventional commit format: missing colon.\n\
|
||||||
|
Parsed subject line: '{}'\n\
|
||||||
|
Raw response preview: '{}'\n\
|
||||||
|
Expected: <type>[optional scope]: <description>",
|
||||||
|
first_line,
|
||||||
|
preview
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let type_part = parts[0];
|
||||||
|
let description = parts[1].trim();
|
||||||
|
|
||||||
|
// Extract type, scope, and breaking indicator
|
||||||
|
let breaking = type_part.ends_with('!');
|
||||||
|
let type_part = type_part.trim_end_matches('!');
|
||||||
|
|
||||||
|
let (commit_type, scope) = if let Some(start) = type_part.find('(') {
|
||||||
|
if let Some(end) = type_part.find(')') {
|
||||||
|
let t = &type_part[..start];
|
||||||
|
let s = &type_part[start + 1..end];
|
||||||
|
(t.to_string(), Some(s.to_string()))
|
||||||
|
} else {
|
||||||
|
bail!("Invalid scope format: missing closing parenthesis");
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
(type_part.to_string(), None)
|
||||||
|
};
|
||||||
|
|
||||||
|
// Extract body and footer
|
||||||
|
let (body, footer) = extract_body_footer(lines);
|
||||||
|
|
||||||
|
Ok(GeneratedCommit {
|
||||||
|
commit_type,
|
||||||
|
scope,
|
||||||
|
description: description.to_string(),
|
||||||
|
body,
|
||||||
|
footer,
|
||||||
|
breaking,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_commitlint_commit(
|
||||||
|
first_line: &str,
|
||||||
|
lines: &[&str],
|
||||||
|
raw_response: &str,
|
||||||
|
) -> Result<GeneratedCommit> {
|
||||||
|
// Similar parsing but with commitlint rules
|
||||||
|
let parts: Vec<&str> = first_line.splitn(2, ':').collect();
|
||||||
|
if parts.len() != 2 {
|
||||||
|
let preview: String = raw_response.chars().take(300).collect();
|
||||||
|
bail!(
|
||||||
|
"Invalid commit format: missing colon.\n\
|
||||||
|
Parsed subject line: '{}'\n\
|
||||||
|
Raw response preview: '{}'\n\
|
||||||
|
Expected: <type>[optional scope]: <subject>",
|
||||||
|
first_line,
|
||||||
|
preview
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let type_part = parts[0];
|
||||||
|
let subject = parts[1].trim();
|
||||||
|
|
||||||
|
let (commit_type, scope) = if let Some(start) = type_part.find('(') {
|
||||||
|
if let Some(end) = type_part.find(')') {
|
||||||
|
let t = &type_part[..start];
|
||||||
|
let s = &type_part[start + 1..end];
|
||||||
|
(t.to_string(), Some(s.to_string()))
|
||||||
|
} else {
|
||||||
|
(type_part.to_string(), None)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
(type_part.to_string(), None)
|
||||||
|
};
|
||||||
|
|
||||||
|
let (body, footer) = extract_body_footer(&lines);
|
||||||
|
|
||||||
|
Ok(GeneratedCommit {
|
||||||
|
commit_type,
|
||||||
|
scope,
|
||||||
|
description: subject.to_string(),
|
||||||
|
body,
|
||||||
|
footer,
|
||||||
|
breaking: false,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn extract_body_footer(lines: &[&str]) -> (Option<String>, Option<String>) {
|
||||||
|
if lines.len() <= 1 {
|
||||||
|
return (None, None);
|
||||||
|
}
|
||||||
|
|
||||||
|
let rest: Vec<&str> = lines[1..]
|
||||||
|
.iter()
|
||||||
|
.skip_while(|l| l.trim().is_empty())
|
||||||
|
.copied()
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
if rest.is_empty() {
|
||||||
|
return (None, None);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Look for footer markers
|
||||||
|
let footer_markers = [
|
||||||
|
"BREAKING CHANGE:",
|
||||||
|
"Closes",
|
||||||
|
"Fixes",
|
||||||
|
"Refs",
|
||||||
|
"Co-authored-by:",
|
||||||
|
];
|
||||||
|
|
||||||
|
let mut body_lines = vec![];
|
||||||
|
let mut footer_lines = vec![];
|
||||||
|
let mut in_footer = false;
|
||||||
|
|
||||||
|
for line in &rest {
|
||||||
|
if footer_markers.iter().any(|m| line.starts_with(m)) {
|
||||||
|
in_footer = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if in_footer {
|
||||||
|
footer_lines.push(*line);
|
||||||
|
} else {
|
||||||
|
body_lines.push(*line);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let body = if body_lines.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(body_lines.join("\n"))
|
||||||
|
};
|
||||||
|
|
||||||
|
let footer = if footer_lines.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(footer_lines.join("\n"))
|
||||||
|
};
|
||||||
|
|
||||||
|
(body, footer)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Generated commit structure
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub struct GeneratedCommit {
|
||||||
|
pub commit_type: String,
|
||||||
|
pub scope: Option<String>,
|
||||||
|
pub description: String,
|
||||||
|
pub body: Option<String>,
|
||||||
|
pub footer: Option<String>,
|
||||||
|
pub breaking: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl GeneratedCommit {
|
||||||
|
/// Format as conventional commit
|
||||||
|
pub fn to_conventional(&self) -> String {
|
||||||
|
crate::utils::formatter::format_conventional_commit(
|
||||||
|
&self.commit_type,
|
||||||
|
self.scope.as_deref(),
|
||||||
|
&self.description,
|
||||||
|
self.body.as_deref(),
|
||||||
|
self.footer.as_deref(),
|
||||||
|
self.breaking,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Format as commitlint commit
|
||||||
|
pub fn to_commitlint(&self) -> String {
|
||||||
|
crate::utils::formatter::format_commitlint_commit(
|
||||||
|
&self.commit_type,
|
||||||
|
self.scope.as_deref(),
|
||||||
|
&self.description,
|
||||||
|
self.body.as_deref(),
|
||||||
|
self.footer.as_deref(),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
618
src/llm/prompts.rs
Normal file
618
src/llm/prompts.rs
Normal file
@@ -0,0 +1,618 @@
|
|||||||
|
use crate::config::Language;
|
||||||
|
|
||||||
|
/// Get commit system prompt based on format and language
|
||||||
|
pub(crate) fn get_commit_system_prompt(
|
||||||
|
format: crate::config::CommitFormat,
|
||||||
|
language: Language,
|
||||||
|
) -> &'static str {
|
||||||
|
match (format, language) {
|
||||||
|
(crate::config::CommitFormat::Conventional, Language::Chinese) => {
|
||||||
|
CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ZH
|
||||||
|
}
|
||||||
|
(crate::config::CommitFormat::Conventional, Language::Japanese) => {
|
||||||
|
CONVENTIONAL_COMMIT_SYSTEM_PROMPT_JA
|
||||||
|
}
|
||||||
|
(crate::config::CommitFormat::Conventional, Language::Korean) => {
|
||||||
|
CONVENTIONAL_COMMIT_SYSTEM_PROMPT_KO
|
||||||
|
}
|
||||||
|
(crate::config::CommitFormat::Conventional, Language::Spanish) => {
|
||||||
|
CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ES
|
||||||
|
}
|
||||||
|
(crate::config::CommitFormat::Conventional, Language::French) => {
|
||||||
|
CONVENTIONAL_COMMIT_SYSTEM_PROMPT_FR
|
||||||
|
}
|
||||||
|
(crate::config::CommitFormat::Conventional, Language::German) => {
|
||||||
|
CONVENTIONAL_COMMIT_SYSTEM_PROMPT_DE
|
||||||
|
}
|
||||||
|
(crate::config::CommitFormat::Conventional, _) => CONVENTIONAL_COMMIT_SYSTEM_PROMPT,
|
||||||
|
(crate::config::CommitFormat::Commitlint, Language::Chinese) => COMMITLINT_SYSTEM_PROMPT_ZH,
|
||||||
|
(crate::config::CommitFormat::Commitlint, Language::Japanese) => {
|
||||||
|
COMMITLINT_SYSTEM_PROMPT_JA
|
||||||
|
}
|
||||||
|
(crate::config::CommitFormat::Commitlint, Language::Korean) => COMMITLINT_SYSTEM_PROMPT_KO,
|
||||||
|
(crate::config::CommitFormat::Commitlint, Language::Spanish) => COMMITLINT_SYSTEM_PROMPT_ES,
|
||||||
|
(crate::config::CommitFormat::Commitlint, Language::French) => COMMITLINT_SYSTEM_PROMPT_FR,
|
||||||
|
(crate::config::CommitFormat::Commitlint, Language::German) => COMMITLINT_SYSTEM_PROMPT_DE,
|
||||||
|
(crate::config::CommitFormat::Commitlint, _) => COMMITLINT_SYSTEM_PROMPT,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn get_tag_system_prompt(language: Language) -> &'static str {
|
||||||
|
match language {
|
||||||
|
Language::Chinese => TAG_MESSAGE_SYSTEM_PROMPT_ZH,
|
||||||
|
Language::Japanese => TAG_MESSAGE_SYSTEM_PROMPT_JA,
|
||||||
|
Language::Korean => TAG_MESSAGE_SYSTEM_PROMPT_KO,
|
||||||
|
Language::Spanish => TAG_MESSAGE_SYSTEM_PROMPT_ES,
|
||||||
|
Language::French => TAG_MESSAGE_SYSTEM_PROMPT_FR,
|
||||||
|
Language::German => TAG_MESSAGE_SYSTEM_PROMPT_DE,
|
||||||
|
_ => TAG_MESSAGE_SYSTEM_PROMPT,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn get_changelog_system_prompt(language: Language) -> &'static str {
|
||||||
|
match language {
|
||||||
|
Language::Chinese => CHANGELOG_SYSTEM_PROMPT_ZH,
|
||||||
|
Language::Japanese => CHANGELOG_SYSTEM_PROMPT_JA,
|
||||||
|
Language::Korean => CHANGELOG_SYSTEM_PROMPT_KO,
|
||||||
|
Language::Spanish => CHANGELOG_SYSTEM_PROMPT_ES,
|
||||||
|
Language::French => CHANGELOG_SYSTEM_PROMPT_FR,
|
||||||
|
Language::German => CHANGELOG_SYSTEM_PROMPT_DE,
|
||||||
|
_ => CHANGELOG_SYSTEM_PROMPT,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// System prompts for LLM
|
||||||
|
|
||||||
|
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT: &str = r#"You are a helpful assistant that generates conventional commit messages.
|
||||||
|
|
||||||
|
Analyze the git diff provided and generate a commit message following the Conventional Commits specification.
|
||||||
|
|
||||||
|
Format: <type>[optional scope]: <description>
|
||||||
|
|
||||||
|
Types:
|
||||||
|
- feat: A new feature
|
||||||
|
- fix: A bug fix
|
||||||
|
- docs: Documentation only changes
|
||||||
|
- style: Changes that don't affect code meaning (formatting, semicolons, etc.)
|
||||||
|
- refactor: Code change that neither fixes a bug nor adds a feature
|
||||||
|
- perf: Code change that improves performance
|
||||||
|
- test: Adding or correcting tests
|
||||||
|
- build: Changes to build system or dependencies
|
||||||
|
- ci: Changes to CI configuration
|
||||||
|
- chore: Other changes that don't modify src or test files
|
||||||
|
- revert: Reverts a previous commit
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
1. Use lowercase for type and scope
|
||||||
|
2. Keep description under 100 characters
|
||||||
|
3. Use imperative mood ("add" not "added")
|
||||||
|
4. Don't capitalize first letter
|
||||||
|
5. No period at the end
|
||||||
|
6. Include scope if the change is specific to a module/component
|
||||||
|
|
||||||
|
Output ONLY the commit message, nothing else.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成符合 Conventional Commits 规范的提交消息的助手。
|
||||||
|
|
||||||
|
分析提供的 git diff,并生成符合 Conventional Commits 规范的提交消息。
|
||||||
|
|
||||||
|
格式: <type>[可选作用域]: <描述>
|
||||||
|
|
||||||
|
类型:
|
||||||
|
- feat: 新功能
|
||||||
|
- fix: 修复错误
|
||||||
|
- docs: 仅文档更改
|
||||||
|
- style: 不影响代码含义的更改(格式化、分号等)
|
||||||
|
- refactor: 既不修复错误也不添加功能的代码更改
|
||||||
|
- perf: 提高性能的代码更改
|
||||||
|
- test: 添加或更正测试
|
||||||
|
- build: 更改构建系统或依赖项
|
||||||
|
- ci: 更改 CI 配置
|
||||||
|
- chore: 其他不修改 src 或测试文件的更改
|
||||||
|
- revert: 撤销之前的提交
|
||||||
|
|
||||||
|
规则:
|
||||||
|
1. 类型和小写使用小写
|
||||||
|
2. 描述保持在 100 个字符以内
|
||||||
|
3. 使用祈使语气("添加"而不是"已添加")
|
||||||
|
4. 不要大写首字母
|
||||||
|
5. 结尾不要句号
|
||||||
|
6. 如果更改特定于模块/组件,请包含作用域
|
||||||
|
7. 仅输出提交消息,不要输出其他内容。
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_JA: &str = r#"あなたはConventional Commits仕様に従ったコミットメッセージを生成するアシスタントです。
|
||||||
|
|
||||||
|
提供されたgit diffを分析し、Conventional Commits仕様に従ったコミットメッセージを生成してください。
|
||||||
|
|
||||||
|
形式: <type>[オプションのスコープ]: <説明>
|
||||||
|
|
||||||
|
タイプ:
|
||||||
|
- feat: 新機能
|
||||||
|
- fix: バグ修正
|
||||||
|
- docs: ドキュメントのみの変更
|
||||||
|
- style: コードの意味に影響しない変更(フォーマット、セミコロンなど)
|
||||||
|
- refactor: バグ修正や機能追加を伴わないコード変更
|
||||||
|
- perf: パフォーマンスを向上させるコード変更
|
||||||
|
- test: テストの追加または修正
|
||||||
|
- build: ビルドシステムまたは依存関係の変更
|
||||||
|
- ci: CI設定の変更
|
||||||
|
- chore: srcやテストファイルを変更しないその他の変更
|
||||||
|
- revert: 以前のコミットを取り消す
|
||||||
|
|
||||||
|
ルール:
|
||||||
|
1. タイプとスコープは小文字を使用
|
||||||
|
2. 説明は100文字以内にする
|
||||||
|
3. 命令形を使用する("追加"ではなく"追加する")
|
||||||
|
4. 先頭を大文字にしない
|
||||||
|
5. 最後にピリオドを付けない
|
||||||
|
6. 変更がモジュール/コンポーネントに固有の場合はスコープを含める
|
||||||
|
7. コミットメッセージのみを出力し、それ以外は出力しないでください。
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_KO: &str = r#"당신은 Conventional Commits 사양에 따른 커밋 메시지를 생성하는 도우미입니다.
|
||||||
|
|
||||||
|
제공된 git diff를 분석하고 Conventional Commits 사양에 따른 커밋 메시지를 생성하세요.
|
||||||
|
|
||||||
|
형식: <type>[선택적 범위]: <설명>
|
||||||
|
|
||||||
|
유형:
|
||||||
|
- feat: 새 기능
|
||||||
|
- fix: 버그 수정
|
||||||
|
- docs: 문서 변경만
|
||||||
|
- style: 코드 의미에 영향을 주지 않는 변경(서식, 세미콜론 등)
|
||||||
|
- refactor: 버그를 수정하거나 기능을 추가하지 않는 코드 변경
|
||||||
|
- perf: 성능을 향상시키는 코드 변경
|
||||||
|
- test: 테스트 추가 또는 수정
|
||||||
|
- build: 빌드 시스템 또는 종속성 변경
|
||||||
|
- ci: CI 구성 변경
|
||||||
|
- chore: src 또는 테스트 파일을 수정하지 않는 기타 변경
|
||||||
|
- revert: 이전 커밋 되돌리기
|
||||||
|
|
||||||
|
규칙:
|
||||||
|
1. 유형과 범위는 소문자 사용
|
||||||
|
2. 설명은 100자 이내로 유지
|
||||||
|
3. 명령형 사용("추가"가 아닌 "추가하다")
|
||||||
|
4. 첫 글자 대문자화하지 않음
|
||||||
|
5. 끝에 마침표 사용하지 않음
|
||||||
|
6. 변경 사항이 모듈/구성 요소에 특정한 경우 범위 포함
|
||||||
|
7. 커밋 메시지만 출력하고 다른 내용은 출력하지 마세요.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera mensajes de commit siguiendo la especificación Conventional Commits.
|
||||||
|
|
||||||
|
Analiza el diff de git proporcionado y genera un mensaje de commit siguiendo la especificación Conventional Commits.
|
||||||
|
|
||||||
|
Formato: <tipo>[alcance opcional]: <descripción>
|
||||||
|
|
||||||
|
Tipos:
|
||||||
|
- feat: Una nueva característica
|
||||||
|
- fix: Una corrección de error
|
||||||
|
- docs: Solo cambios en documentación
|
||||||
|
- style: Cambios que no afectan el significado del código (formato, punto y coma, etc.)
|
||||||
|
- refactor: Cambio de código que no corrige un error ni agrega una característica
|
||||||
|
- perf: Cambio de código que mejora el rendimiento
|
||||||
|
- test: Agregar o corregir pruebas
|
||||||
|
- build: Cambios en el sistema de construcción o dependencias
|
||||||
|
- ci: Cambios en la configuración de CI
|
||||||
|
- chore: Otros cambios que no modifican archivos src o de prueba
|
||||||
|
- revert: Revierte un commit anterior
|
||||||
|
|
||||||
|
Reglas:
|
||||||
|
1. Usa minúsculas para tipo y alcance
|
||||||
|
2. Mantén la descripción bajo 100 caracteres
|
||||||
|
3. Usa modo imperativo ("agregar" no "agregado")
|
||||||
|
4. No capitalices la primera letra
|
||||||
|
5. Sin punto al final
|
||||||
|
6. Incluye alcance si el cambio es específico de un módulo/componente
|
||||||
|
7. Genera SOLO el mensaje de commit, nada más.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_FR: &str = r#"Vous êtes un assistant qui génère des messages de commit suivant la spécification Conventional Commits.
|
||||||
|
|
||||||
|
Analysez le diff git fourni et générez un message de commit suivant la spécification Conventional Commits.
|
||||||
|
|
||||||
|
Format: <type>[portée optionnelle]: <description>
|
||||||
|
|
||||||
|
Types:
|
||||||
|
- feat: Une nouvelle fonctionnalité
|
||||||
|
- fix: Une correction de bug
|
||||||
|
- docs: Changements de documentation uniquement
|
||||||
|
- style: Changements qui n'affectent pas la signification du code (formatage, points-virgules, etc.)
|
||||||
|
- refactor: Changement de code qui ne corrige pas un bug ni n'ajoute une fonctionnalité
|
||||||
|
- perf: Changement de code qui améliore les performances
|
||||||
|
- test: Ajout ou correction de tests
|
||||||
|
- build: Changements du système de build ou des dépendances
|
||||||
|
- ci: Changements de la configuration CI
|
||||||
|
- chore: Autres changements qui ne modifient pas les fichiers src ou de test
|
||||||
|
- revert: Révertit un commit précédent
|
||||||
|
|
||||||
|
Règles:
|
||||||
|
1. Utilisez des minuscules pour le type et la portée
|
||||||
|
2. Gardez la description sous 100 caractères
|
||||||
|
3. Utilisez le mode impératif ("ajouter" non "ajouté")
|
||||||
|
4. Ne capitalisez pas la première lettre
|
||||||
|
5. Pas de point à la fin
|
||||||
|
6. Incluez la portée si le changement est spécifique à un module/composant
|
||||||
|
7. Générez SEULEMENT le message de commit, rien d'autre.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_DE: &str = r#"Sie sind ein Assistent, der Commit-Nachrichten gemäß der Conventional Commits-Spezifikation generiert.
|
||||||
|
|
||||||
|
Analysieren Sie den bereitgestellten git diff und generieren Sie eine Commit-Nachricht gemäß der Conventional Commits-Spezifikation.
|
||||||
|
|
||||||
|
Format: <typ>[optionaler Bereich]: <beschreibung>
|
||||||
|
|
||||||
|
Typen:
|
||||||
|
- feat: Eine neue Funktion
|
||||||
|
- fix: Ein Bugfix
|
||||||
|
- docs: Nur Dokumentationsänderungen
|
||||||
|
- style: Änderungen, die die Code-Bedeutung nicht beeinflussen (Formatierung, Semikolons usw.)
|
||||||
|
- refactor: Code-Änderung, die weder einen Bug behebt noch eine Funktion hinzufügt
|
||||||
|
- perf: Code-Änderung, die die Leistung verbessert
|
||||||
|
- test: Hinzufügen oder Korrigieren von Tests
|
||||||
|
- build: Änderungen am Build-System oder Abhängigkeiten
|
||||||
|
- ci: Änderungen an der CI-Konfiguration
|
||||||
|
- chore: Andere Änderungen, die src- oder Testdateien nicht ändern
|
||||||
|
- revert: Setzt einen vorherigen Commit zurück
|
||||||
|
|
||||||
|
Regeln:
|
||||||
|
1. Verwenden Sie Kleinbuchstaben für Typ und Bereich
|
||||||
|
2. Halten Sie die Beschreibung unter 100 Zeichen
|
||||||
|
3. Verwenden Sie den Imperativ ("hinzufügen" nicht "hinzugefügt")
|
||||||
|
4. Großschreiben Sie den ersten Buchstaben nicht
|
||||||
|
5. Kein Punkt am Ende
|
||||||
|
6. Fügen Sie einen Bereich ein, wenn die Änderung spezifisch für ein Modul/Komponente ist
|
||||||
|
7. Geben Sie NUR die Commit-Nachricht aus, nichts anderes.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const COMMITLINT_SYSTEM_PROMPT: &str = r#"You are a helpful assistant that generates commit messages following @commitlint/config-conventional.
|
||||||
|
|
||||||
|
Analyze the git diff and generate a commit message.
|
||||||
|
|
||||||
|
Format: <type>[optional scope]: <subject>
|
||||||
|
|
||||||
|
Types: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
1. Subject should not start with uppercase
|
||||||
|
2. Subject should not end with period
|
||||||
|
3. Subject should be 4-100 characters
|
||||||
|
4. Use imperative mood
|
||||||
|
5. Be concise but descriptive
|
||||||
|
6. Output ONLY the commit message, nothing else.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const COMMITLINT_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成符合 @commitlint/config-conventional 规范的提交消息的助手。
|
||||||
|
|
||||||
|
分析 git diff 并生成提交消息。
|
||||||
|
|
||||||
|
格式: <type>[可选作用域]: <主题>
|
||||||
|
|
||||||
|
类型: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||||
|
|
||||||
|
规则:
|
||||||
|
1. 主题不应以大写字母开头
|
||||||
|
2. 主题不应以句号结尾
|
||||||
|
3. 主题应为 4-100 个字符
|
||||||
|
4. 使用祈使语气
|
||||||
|
5. 简洁但描述性强
|
||||||
|
6. 仅输出提交消息,不要输出其他额外内容。
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const COMMITLINT_SYSTEM_PROMPT_JA: &str = r#"あなたは@commitlint/config-conventionalに従ったコミットメッセージを生成するアシスタントです。
|
||||||
|
|
||||||
|
git diffを分析し、コミットメッセージを生成してください。
|
||||||
|
|
||||||
|
形式: <type>[オプションのスコープ]: <件名>
|
||||||
|
|
||||||
|
タイプ: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||||
|
|
||||||
|
ルール:
|
||||||
|
1. 件名は大文字で始めないでください
|
||||||
|
2. 件名はピリオドで終わらないでください
|
||||||
|
3. 件名は4-100文字である必要があります
|
||||||
|
4. 命令形を使用してください
|
||||||
|
5. 簡潔ですが説明的であること
|
||||||
|
6. コミットメッセージのみを出力し、それ以外は出力しないでください。
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const COMMITLINT_SYSTEM_PROMPT_KO: &str = r#"당신은 @commitlint/config-conventional에 따른 커밋 메시지를 생성하는 도우미입니다.
|
||||||
|
|
||||||
|
git diff를 분석하고 커밋 메시지를 생성하세요.
|
||||||
|
|
||||||
|
형식: <type>[선택적 범위]: <제목>
|
||||||
|
|
||||||
|
유형: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||||
|
|
||||||
|
규칙:
|
||||||
|
1. 제목은 대문자로 시작하지 않아야 합니다
|
||||||
|
2. 제목은 마침표로 끝나지 않아야 합니다
|
||||||
|
3. 제목은 4-100자여야 합니다
|
||||||
|
4. 명령형을 사용하세요
|
||||||
|
5. 간결하지만 설명적이어야 합니다
|
||||||
|
6. 커밋 메시지만 출력하고 다른 내용은 출력하지 마세요.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const COMMITLINT_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera mensajes de commit siguiendo @commitlint/config-conventional.
|
||||||
|
|
||||||
|
Analiza el diff de git y genera un mensaje de commit.
|
||||||
|
|
||||||
|
Formato: <tipo>[alcance opcional]: <asunto>
|
||||||
|
|
||||||
|
Tipos: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||||
|
|
||||||
|
Reglas:
|
||||||
|
1. El asunto no debe comenzar con mayúscula
|
||||||
|
2. El asunto no debe terminar con punto
|
||||||
|
3. El asunto debe tener 4-100 caracteres
|
||||||
|
4. Usa modo imperativo
|
||||||
|
5. Sé conciso pero descriptivo
|
||||||
|
6. Genera SOLO el mensaje de commit, nada más.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const COMMITLINT_SYSTEM_PROMPT_FR: &str = r#"Vous êtes un assistant qui génère des messages de commit suivant @commitlint/config-conventional.
|
||||||
|
|
||||||
|
Analysez le diff git et générez un message de commit.
|
||||||
|
|
||||||
|
Format: <type>[portée optionnelle]: <sujet>
|
||||||
|
|
||||||
|
Types: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||||
|
|
||||||
|
Règles:
|
||||||
|
1. Le sujet ne doit pas commencer par une majuscule
|
||||||
|
2. Le sujet ne doit pas se terminer par un point
|
||||||
|
3. Le sujet doit avoir 4-100 caractères
|
||||||
|
4. Utilisez le mode impératif
|
||||||
|
5. Soyez concis mais descriptif
|
||||||
|
6. Générez SEULEMENT le message de commit, rien d'autre.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const COMMITLINT_SYSTEM_PROMPT_DE: &str = r#"Sie sind ein Assistent, der Commit-Nachrichten gemäß @commitlint/config-conventional generiert.
|
||||||
|
|
||||||
|
Analysieren Sie den git diff und generieren Sie eine Commit-Nachricht.
|
||||||
|
|
||||||
|
Format: <typ>[optionaler Bereich]: <betreff>
|
||||||
|
|
||||||
|
Typen: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||||
|
|
||||||
|
Regeln:
|
||||||
|
1. Der Betreff sollte nicht mit einem Großbuchstaben beginnen
|
||||||
|
2. Der Betreff sollte nicht mit einem Punkt enden
|
||||||
|
3. Der Betreff sollte 4-100 Zeichen haben
|
||||||
|
4. Verwenden Sie den Imperativ
|
||||||
|
5. Seien Sie prägnant aber beschreibend
|
||||||
|
6. Geben Sie NUR die Commit-Nachricht aus, nichts anderes.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const TAG_MESSAGE_SYSTEM_PROMPT: &str = r#"You are a helpful assistant that generates git tag annotation messages.
|
||||||
|
|
||||||
|
Given a version number and a list of commits, generate a concise but informative tag message.
|
||||||
|
|
||||||
|
The message should:
|
||||||
|
1. Start with a brief summary of the release
|
||||||
|
2. Group changes by type (features, fixes, etc.)
|
||||||
|
3. Be suitable for a git annotated tag
|
||||||
|
|
||||||
|
Format:
|
||||||
|
<version> Release
|
||||||
|
|
||||||
|
Summary of changes...
|
||||||
|
|
||||||
|
Changes:
|
||||||
|
- Feature: description
|
||||||
|
- Fix: description
|
||||||
|
...
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const TAG_MESSAGE_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成 git 标签注释消息的助手。
|
||||||
|
|
||||||
|
给定版本号和提交列表,生成简洁但信息丰富的标签消息。
|
||||||
|
|
||||||
|
消息应该:
|
||||||
|
1. 以发布的简要摘要开始
|
||||||
|
2. 按类型分组更改(功能、修复等)
|
||||||
|
3. 适合 git 标注标签
|
||||||
|
|
||||||
|
格式:
|
||||||
|
<version> 发布
|
||||||
|
|
||||||
|
更改摘要...
|
||||||
|
|
||||||
|
更改:
|
||||||
|
- 功能:描述
|
||||||
|
- 修复:描述
|
||||||
|
...
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const TAG_MESSAGE_SYSTEM_PROMPT_JA: &str = r#"あなたはgitタグ注釈メッセージを生成するアシスタントです。
|
||||||
|
|
||||||
|
バージョン番号とコミットのリストを考慮して、簡潔ですが情報豊富なタグメッセージを生成してください。
|
||||||
|
|
||||||
|
メッセージは以下のようであるべきです:
|
||||||
|
1. リリースの簡単な要約から始める
|
||||||
|
2. タイプ別に変更をグループ化する(機能、修正など)
|
||||||
|
3. git注釈タグに適している
|
||||||
|
|
||||||
|
形式:
|
||||||
|
<version> リリース
|
||||||
|
|
||||||
|
変更の概要...
|
||||||
|
|
||||||
|
変更:
|
||||||
|
- 機能:説明
|
||||||
|
- 修正:説明
|
||||||
|
...
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const TAG_MESSAGE_SYSTEM_PROMPT_KO: &str = r#"당신은 git 태그 주석 메시지를 생성하는 도우미입니다.
|
||||||
|
|
||||||
|
버전 번호와 커밋 목록을 고려하여 간결하지만 정보가 풍부한 태그 메시지를 생성하세요.
|
||||||
|
|
||||||
|
메시지는 다음과 같아야 합니다:
|
||||||
|
1. 릴리스의 간단한 요약으로 시작
|
||||||
|
2. 유형별로 변경 사항 그룹화(기능, 수정 등)
|
||||||
|
3. git 주석 태그에 적합
|
||||||
|
|
||||||
|
형식:
|
||||||
|
<version> 릴리스
|
||||||
|
|
||||||
|
변경 사항 요약...
|
||||||
|
|
||||||
|
변경 사항:
|
||||||
|
- 기능: 설명
|
||||||
|
- 수정: 설명
|
||||||
|
...
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const TAG_MESSAGE_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera mensajes de anotación de etiquetas git.
|
||||||
|
|
||||||
|
Dado un número de versión y una lista de commits, genera un mensaje de etiqueta conciso pero informativo.
|
||||||
|
|
||||||
|
El mensaje debe:
|
||||||
|
1. Comenzar con un resumen breve de la versión
|
||||||
|
2. Agrupar cambios por tipo (características, correcciones, etc.)
|
||||||
|
3. Ser adecuado para una etiqueta git anotada
|
||||||
|
|
||||||
|
Formato:
|
||||||
|
<version> Versión
|
||||||
|
|
||||||
|
Resumen de cambios...
|
||||||
|
|
||||||
|
Cambios:
|
||||||
|
- Característica: descripción
|
||||||
|
- Corrección: descripción
|
||||||
|
...
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const TAG_MESSAGE_SYSTEM_PROMPT_FR: &str = r#"Vous êtes un assistant qui génère des messages d'annotation de balises git.
|
||||||
|
|
||||||
|
Étant donné un numéro de version et une liste de commits, générez un message de balise concis mais informatif.
|
||||||
|
|
||||||
|
Le message doit :
|
||||||
|
1. Commencer par un bref résumé de la version
|
||||||
|
2. Grouper les changements par type (fonctionnalités, corrections, etc.)
|
||||||
|
3. Être adapté à une balise git annotée
|
||||||
|
|
||||||
|
Format :
|
||||||
|
<version> Version
|
||||||
|
|
||||||
|
Résumé des changements...
|
||||||
|
|
||||||
|
Changements :
|
||||||
|
- Fonctionnalité : description
|
||||||
|
- Correction : description
|
||||||
|
...
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const TAG_MESSAGE_SYSTEM_PROMPT_DE: &str = r#"Sie sind ein Assistent, der git-Tag-Anmerkungsnachrichten generiert.
|
||||||
|
|
||||||
|
Gegeben eine Versionsnummer und eine Liste von Commits, generieren Sie eine prägnante aber informative Tag-Nachricht.
|
||||||
|
|
||||||
|
Die Nachricht sollte:
|
||||||
|
1. Mit einer kurzen Zusammenfassung der Version beginnen
|
||||||
|
2. Änderungen nach Typ gruppieren (Funktionen, Fixes, etc.)
|
||||||
|
3. Für ein git-annotiertes Tag geeignet sein
|
||||||
|
|
||||||
|
Format:
|
||||||
|
<version> Version
|
||||||
|
|
||||||
|
Zusammenfassung der Änderungen...
|
||||||
|
|
||||||
|
Änderungen:
|
||||||
|
- Funktion: Beschreibung
|
||||||
|
- Fix: Beschreibung
|
||||||
|
...
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CHANGELOG_SYSTEM_PROMPT: &str = r#"You are a helpful assistant that generates changelog entries.
|
||||||
|
|
||||||
|
Given a version and a list of commits, generate a well-formatted changelog section.
|
||||||
|
|
||||||
|
Group commits by:
|
||||||
|
- Features (feat)
|
||||||
|
- Bug Fixes (fix)
|
||||||
|
- Documentation (docs)
|
||||||
|
- Other Changes
|
||||||
|
|
||||||
|
Format in markdown with proper headings and bullet points.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CHANGELOG_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成变更日志条目的助手。
|
||||||
|
|
||||||
|
给定版本和提交列表,生成格式良好的变更日志部分。
|
||||||
|
|
||||||
|
按以下方式分组提交:
|
||||||
|
- 功能 (feat)
|
||||||
|
- 错误修复 (fix)
|
||||||
|
- 文档 (docs)
|
||||||
|
- 其他更改
|
||||||
|
|
||||||
|
使用适当的标题和项目符号以 markdown 格式输出。
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CHANGELOG_SYSTEM_PROMPT_JA: &str = r#"あなたは変更ログエントリを生成するアシスタントです。
|
||||||
|
|
||||||
|
バージョンとコミットのリストを考慮して、適切にフォーマットされた変更ログセクションを生成してください。
|
||||||
|
|
||||||
|
コミットを以下でグループ化してください:
|
||||||
|
- 機能 (feat)
|
||||||
|
- バグ修正 (fix)
|
||||||
|
- ドキュメント (docs)
|
||||||
|
- その他の変更
|
||||||
|
|
||||||
|
適切な見出しと箇条書きを使用してmarkdown形式でフォーマットしてください。
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CHANGELOG_SYSTEM_PROMPT_KO: &str = r#"당신은 변경 로그 항목을 생성하는 도우미입니다.
|
||||||
|
|
||||||
|
버전과 커밋 목록을 고려하여 잘 포맷된 변경 로그 섹션을 생성하세요.
|
||||||
|
|
||||||
|
다음으로 커밋을 그룹화하세요:
|
||||||
|
- 기능 (feat)
|
||||||
|
- 버그 수정 (fix)
|
||||||
|
- 문서 (docs)
|
||||||
|
- 기타 변경 사항
|
||||||
|
|
||||||
|
적절한 제목과 글머리 기호를 사용하여 markdown 형식으로 포맷하세요.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CHANGELOG_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera entradas de registro de cambios.
|
||||||
|
|
||||||
|
Dada una versión y una lista de commits, genera una sección de registro de cambios bien formateada.
|
||||||
|
|
||||||
|
Agrupa los commits por:
|
||||||
|
- Características (feat)
|
||||||
|
- Correcciones de errores (fix)
|
||||||
|
- Documentación (docs)
|
||||||
|
- Otros cambios
|
||||||
|
|
||||||
|
Formatea en markdown con encabezados y viñetas apropiados.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CHANGELOG_SYSTEM_PROMPT_FR: &str = r#"Vous êtes un assistant qui génère des entrées de journal des modifications.
|
||||||
|
|
||||||
|
Étant donné une version et une liste de commits, générez une section de journal des modifications bien formatée.
|
||||||
|
|
||||||
|
Groupez les commits par :
|
||||||
|
- Fonctionnalités (feat)
|
||||||
|
- Corrections de bugs (fix)
|
||||||
|
- Documentation (docs)
|
||||||
|
- Autres modifications
|
||||||
|
|
||||||
|
Formatez en markdown avec des en-têtes et des puces appropriés.
|
||||||
|
"#;
|
||||||
|
|
||||||
|
const CHANGELOG_SYSTEM_PROMPT_DE: &str = r#"Sie sind ein Assistent, der Changelog-Einträge generiert.
|
||||||
|
|
||||||
|
Gegeben eine Version und eine Liste von Commits, generieren Sie einen gut formatierten Changelog-Abschnitt.
|
||||||
|
|
||||||
|
Gruppieren Sie Commits nach:
|
||||||
|
- Funktionen (feat)
|
||||||
|
- Bugfixes (fix)
|
||||||
|
- Dokumentation (docs)
|
||||||
|
- Andere Änderungen
|
||||||
|
|
||||||
|
Formatieren Sie in Markdown mit geeigneten Überschriften und Aufzählungspunkten.
|
||||||
|
"#;
|
||||||
1528
src/llm/rig/mod.rs
Normal file
1528
src/llm/rig/mod.rs
Normal file
File diff suppressed because it is too large
Load Diff
143
src/llm/thinking.rs
Normal file
143
src/llm/thinking.rs
Normal file
@@ -0,0 +1,143 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
use std::sync::atomic::{AtomicBool, Ordering};
|
||||||
|
|
||||||
|
/// 统一的思考状态管理器,用于管理模型思考状态的显示与隐藏
|
||||||
|
pub struct ThinkingStateManager {
|
||||||
|
is_thinking: AtomicBool,
|
||||||
|
on_start: Option<Box<dyn Fn() + Send + Sync>>,
|
||||||
|
on_end: Option<Box<dyn Fn() + Send + Sync>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ThinkingStateManager {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
is_thinking: AtomicBool::new(false),
|
||||||
|
on_start: None,
|
||||||
|
on_end: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 设置思考开始回调
|
||||||
|
pub fn on_thinking_start<F: Fn() + Send + Sync + 'static>(mut self, callback: F) -> Self {
|
||||||
|
self.on_start = Some(Box::new(callback));
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 设置思考结束回调
|
||||||
|
pub fn on_thinking_end<F: Fn() + Send + Sync + 'static>(mut self, callback: F) -> Self {
|
||||||
|
self.on_end = Some(Box::new(callback));
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 开始思考状态
|
||||||
|
pub fn start_thinking(&self) {
|
||||||
|
if !self.is_thinking.load(Ordering::SeqCst) {
|
||||||
|
self.is_thinking.store(true, Ordering::SeqCst);
|
||||||
|
if let Some(ref cb) = self.on_start {
|
||||||
|
cb();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 结束思考状态
|
||||||
|
pub fn end_thinking(&self) {
|
||||||
|
if self.is_thinking.load(Ordering::SeqCst) {
|
||||||
|
self.is_thinking.store(false, Ordering::SeqCst);
|
||||||
|
if let Some(ref cb) = self.on_end {
|
||||||
|
cb();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 当前是否处于思考状态
|
||||||
|
pub fn is_thinking(&self) -> bool {
|
||||||
|
self.is_thinking.load(Ordering::SeqCst)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for ThinkingStateManager {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 线程安全的思考状态管理器引用
|
||||||
|
pub type SharedThinkingState = Arc<ThinkingStateManager>;
|
||||||
|
|
||||||
|
/// 创建 LLM 流式使用的共享思考状态。
|
||||||
|
/// 进度显示由 AI 生成 spinner 负责(issue 21),此处不再附加控制台输出。
|
||||||
|
pub fn create_console_thinking_state() -> SharedThinkingState {
|
||||||
|
Arc::new(ThinkingStateManager::new())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use std::sync::Mutex;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_thinking_state_transitions() {
|
||||||
|
let manager = ThinkingStateManager::new();
|
||||||
|
assert!(!manager.is_thinking());
|
||||||
|
|
||||||
|
manager.start_thinking();
|
||||||
|
assert!(manager.is_thinking());
|
||||||
|
|
||||||
|
manager.end_thinking();
|
||||||
|
assert!(!manager.is_thinking());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_thinking_idempotent_start() {
|
||||||
|
let manager = ThinkingStateManager::new();
|
||||||
|
manager.start_thinking();
|
||||||
|
manager.start_thinking(); // 重复调用不应触发回调两次
|
||||||
|
assert!(manager.is_thinking());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_thinking_idempotent_end() {
|
||||||
|
let manager = ThinkingStateManager::new();
|
||||||
|
manager.end_thinking(); // 未开始时结束不应触发问题
|
||||||
|
assert!(!manager.is_thinking());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_thinking_callbacks() {
|
||||||
|
let events: Arc<Mutex<Vec<String>>> = Arc::new(Mutex::new(Vec::new()));
|
||||||
|
let events_clone = events.clone();
|
||||||
|
|
||||||
|
let manager = ThinkingStateManager::new().on_thinking_start(move || {
|
||||||
|
events_clone.lock().unwrap().push("start".to_string());
|
||||||
|
});
|
||||||
|
|
||||||
|
let events_clone2 = events.clone();
|
||||||
|
let manager = manager.on_thinking_end(move || {
|
||||||
|
events_clone2.lock().unwrap().push("end".to_string());
|
||||||
|
});
|
||||||
|
|
||||||
|
manager.start_thinking();
|
||||||
|
manager.end_thinking();
|
||||||
|
|
||||||
|
let recorded = events.lock().unwrap();
|
||||||
|
assert_eq!(recorded.len(), 2);
|
||||||
|
assert_eq!(recorded[0], "start");
|
||||||
|
assert_eq!(recorded[1], "end");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_create_console_thinking_state() {
|
||||||
|
let state = create_console_thinking_state();
|
||||||
|
assert!(!state.is_thinking());
|
||||||
|
state.start_thinking();
|
||||||
|
assert!(state.is_thinking());
|
||||||
|
state.end_thinking();
|
||||||
|
assert!(!state.is_thinking());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_default() {
|
||||||
|
let manager = ThinkingStateManager::default();
|
||||||
|
assert!(!manager.is_thinking());
|
||||||
|
}
|
||||||
|
}
|
||||||
76
src/main.rs
76
src/main.rs
@@ -1,18 +1,11 @@
|
|||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use clap::{Parser, Subcommand};
|
use clap::{Parser, Subcommand};
|
||||||
|
use std::path::PathBuf;
|
||||||
use tracing::debug;
|
use tracing::debug;
|
||||||
|
|
||||||
mod commands;
|
use quicommit::commands::{
|
||||||
mod config;
|
|
||||||
mod generator;
|
|
||||||
mod git;
|
|
||||||
mod i18n;
|
|
||||||
mod llm;
|
|
||||||
mod utils;
|
|
||||||
|
|
||||||
use commands::{
|
|
||||||
changelog::ChangelogCommand, commit::CommitCommand, config::ConfigCommand,
|
changelog::ChangelogCommand, commit::CommitCommand, config::ConfigCommand,
|
||||||
init::InitCommand, profile::ProfileCommand, tag::TagCommand,
|
credential::CredentialCommand, init::InitCommand, profile::ProfileCommand, tag::TagCommand,
|
||||||
};
|
};
|
||||||
|
|
||||||
/// QuiCommit - AI-powered Git assistant
|
/// QuiCommit - AI-powered Git assistant
|
||||||
@@ -27,7 +20,7 @@ use commands::{
|
|||||||
#[command(propagate_version = true)]
|
#[command(propagate_version = true)]
|
||||||
#[command(arg_required_else_help = true)]
|
#[command(arg_required_else_help = true)]
|
||||||
struct Cli {
|
struct Cli {
|
||||||
/// Enable verbose output
|
/// Increase verbosity (-v: info, -vv: debug, -vvv: trace)
|
||||||
#[arg(short, long, global = true, action = clap::ArgAction::Count)]
|
#[arg(short, long, global = true, action = clap::ArgAction::Count)]
|
||||||
verbose: u8,
|
verbose: u8,
|
||||||
|
|
||||||
@@ -36,9 +29,17 @@ struct Cli {
|
|||||||
config: Option<String>,
|
config: Option<String>,
|
||||||
|
|
||||||
/// Disable colored output
|
/// Disable colored output
|
||||||
#[arg(long, global = true, env = "NO_COLOR")]
|
#[arg(long, global = true)]
|
||||||
no_color: bool,
|
no_color: bool,
|
||||||
|
|
||||||
|
/// Force emoji decorations on (overrides config)
|
||||||
|
#[arg(long, global = true, conflicts_with = "no_emoji")]
|
||||||
|
emoji: bool,
|
||||||
|
|
||||||
|
/// Disable emoji/symbol decorations
|
||||||
|
#[arg(long, global = true)]
|
||||||
|
no_emoji: bool,
|
||||||
|
|
||||||
#[command(subcommand)]
|
#[command(subcommand)]
|
||||||
command: Commands,
|
command: Commands,
|
||||||
}
|
}
|
||||||
@@ -68,13 +69,41 @@ enum Commands {
|
|||||||
/// Manage configuration settings
|
/// Manage configuration settings
|
||||||
#[command(alias = "cfg")]
|
#[command(alias = "cfg")]
|
||||||
Config(ConfigCommand),
|
Config(ConfigCommand),
|
||||||
|
|
||||||
|
/// Git credential helper (hidden, invoked by git)
|
||||||
|
#[command(hide = true)]
|
||||||
|
Credential(CredentialCommand),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> Result<()> {
|
async fn main() -> Result<()> {
|
||||||
let cli = Cli::parse();
|
let cli = Cli::parse();
|
||||||
|
|
||||||
// Initialize logging
|
// Apply the global color decision before any output is produced.
|
||||||
|
// --no-color disables colors; NO_COLOR follows the no-color.org spec
|
||||||
|
// (any presence, regardless of value, disables color).
|
||||||
|
let no_color = cli.no_color || std::env::var_os("NO_COLOR").is_some();
|
||||||
|
colored::control::set_override(!no_color);
|
||||||
|
|
||||||
|
// Resolve the decoration switch (issue 14): explicit flag > config >
|
||||||
|
// default; --no-color disables decorations too (ADR-0003).
|
||||||
|
let emoji_from_config = cli
|
||||||
|
.config
|
||||||
|
.as_deref()
|
||||||
|
.map(PathBuf::from)
|
||||||
|
.or_else(|| quicommit::config::AppConfig::default_path().ok())
|
||||||
|
.and_then(|path| quicommit::config::AppConfig::load(&path).ok())
|
||||||
|
.map(|config| config.output.emoji)
|
||||||
|
.unwrap_or(true);
|
||||||
|
let emoji_enabled = if no_color || cli.no_emoji {
|
||||||
|
false
|
||||||
|
} else if cli.emoji {
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
emoji_from_config
|
||||||
|
};
|
||||||
|
quicommit::utils::set_emoji_enabled(emoji_enabled);
|
||||||
|
|
||||||
let log_level = match cli.verbose {
|
let log_level = match cli.verbose {
|
||||||
0 => "warn",
|
0 => "warn",
|
||||||
1 => "info",
|
1 => "info",
|
||||||
@@ -82,20 +111,25 @@ async fn main() -> Result<()> {
|
|||||||
_ => "trace",
|
_ => "trace",
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// RUST_LOG takes precedence when set; otherwise -v decides (issue 23).
|
||||||
|
let filter = tracing_subscriber::EnvFilter::try_from_default_env()
|
||||||
|
.unwrap_or_else(|_| tracing_subscriber::EnvFilter::new(log_level));
|
||||||
tracing_subscriber::fmt()
|
tracing_subscriber::fmt()
|
||||||
.with_env_filter(log_level)
|
.with_env_filter(filter)
|
||||||
.with_target(false)
|
.with_target(false)
|
||||||
.init();
|
.init();
|
||||||
|
|
||||||
debug!("Starting quicommit v{}", env!("CARGO_PKG_VERSION"));
|
debug!("Starting quicommit v{}", env!("CARGO_PKG_VERSION"));
|
||||||
|
|
||||||
// Execute command
|
let config_path: Option<PathBuf> = cli.config.map(PathBuf::from);
|
||||||
|
|
||||||
match cli.command {
|
match cli.command {
|
||||||
Commands::Init(cmd) => cmd.execute().await,
|
Commands::Init(cmd) => cmd.execute(config_path).await,
|
||||||
Commands::Commit(cmd) => cmd.execute().await,
|
Commands::Commit(cmd) => cmd.execute(config_path).await,
|
||||||
Commands::Tag(cmd) => cmd.execute().await,
|
Commands::Tag(cmd) => cmd.execute(config_path).await,
|
||||||
Commands::Changelog(cmd) => cmd.execute().await,
|
Commands::Changelog(cmd) => cmd.execute(config_path).await,
|
||||||
Commands::Profile(cmd) => cmd.execute().await,
|
Commands::Profile(cmd) => cmd.execute(config_path).await,
|
||||||
Commands::Config(cmd) => cmd.execute().await,
|
Commands::Config(cmd) => cmd.execute(config_path).await,
|
||||||
|
Commands::Credential(cmd) => cmd.execute(config_path).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
use aes_gcm::{
|
use aes_gcm::{
|
||||||
aead::{Aead, KeyInit},
|
|
||||||
Aes256Gcm, Nonce,
|
Aes256Gcm, Nonce,
|
||||||
|
aead::{Aead, KeyInit},
|
||||||
};
|
};
|
||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use base64::{engine::general_purpose::STANDARD as BASE64, Engine};
|
use base64::{Engine, engine::general_purpose::STANDARD as BASE64};
|
||||||
use rand::Rng;
|
use rand::Rng;
|
||||||
use std::fs;
|
use std::fs;
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
@@ -20,8 +20,7 @@ pub fn encrypt(data: &[u8], password: &str) -> Result<String> {
|
|||||||
rand::thread_rng().fill(&mut nonce_bytes);
|
rand::thread_rng().fill(&mut nonce_bytes);
|
||||||
|
|
||||||
let key = derive_key(password, &salt)?;
|
let key = derive_key(password, &salt)?;
|
||||||
let cipher = Aes256Gcm::new_from_slice(&key)
|
let cipher = Aes256Gcm::new_from_slice(&key).context("Failed to create cipher")?;
|
||||||
.context("Failed to create cipher")?;
|
|
||||||
let nonce = Nonce::from_slice(&nonce_bytes);
|
let nonce = Nonce::from_slice(&nonce_bytes);
|
||||||
|
|
||||||
let encrypted = cipher
|
let encrypted = cipher
|
||||||
@@ -39,7 +38,8 @@ pub fn encrypt(data: &[u8], password: &str) -> Result<String> {
|
|||||||
|
|
||||||
/// Decrypt data with password
|
/// Decrypt data with password
|
||||||
pub fn decrypt(encrypted_data: &str, password: &str) -> Result<Vec<u8>> {
|
pub fn decrypt(encrypted_data: &str, password: &str) -> Result<Vec<u8>> {
|
||||||
let data = BASE64.decode(encrypted_data)
|
let data = BASE64
|
||||||
|
.decode(encrypted_data)
|
||||||
.context("Invalid base64 encoding")?;
|
.context("Invalid base64 encoding")?;
|
||||||
|
|
||||||
if data.len() < SALT_LEN + NONCE_LEN {
|
if data.len() < SALT_LEN + NONCE_LEN {
|
||||||
@@ -51,8 +51,7 @@ pub fn decrypt(encrypted_data: &str, password: &str) -> Result<Vec<u8>> {
|
|||||||
let encrypted = &data[SALT_LEN + NONCE_LEN..];
|
let encrypted = &data[SALT_LEN + NONCE_LEN..];
|
||||||
|
|
||||||
let key = derive_key(password, salt)?;
|
let key = derive_key(password, salt)?;
|
||||||
let cipher = Aes256Gcm::new_from_slice(&key)
|
let cipher = Aes256Gcm::new_from_slice(&key).context("Failed to create cipher")?;
|
||||||
.context("Failed to create cipher")?;
|
|
||||||
let nonce = Nonce::from_slice(nonce_bytes);
|
let nonce = Nonce::from_slice(nonce_bytes);
|
||||||
|
|
||||||
let decrypted = cipher
|
let decrypted = cipher
|
||||||
@@ -64,7 +63,7 @@ pub fn decrypt(encrypted_data: &str, password: &str) -> Result<Vec<u8>> {
|
|||||||
|
|
||||||
/// Derive key from password using simple method
|
/// Derive key from password using simple method
|
||||||
fn derive_key(password: &str, salt: &[u8]) -> Result<[u8; KEY_LEN]> {
|
fn derive_key(password: &str, salt: &[u8]) -> Result<[u8; KEY_LEN]> {
|
||||||
use sha2::{Sha256, Digest};
|
use sha2::{Digest, Sha256};
|
||||||
|
|
||||||
let mut hasher = Sha256::new();
|
let mut hasher = Sha256::new();
|
||||||
hasher.update(salt);
|
hasher.update(salt);
|
||||||
|
|||||||
@@ -9,14 +9,11 @@ pub fn edit_content(initial_content: &str) -> Result<String> {
|
|||||||
|
|
||||||
/// Edit file in user's default editor
|
/// Edit file in user's default editor
|
||||||
pub fn edit_file(path: &Path) -> Result<String> {
|
pub fn edit_file(path: &Path) -> Result<String> {
|
||||||
let content = fs::read_to_string(path)
|
let content = fs::read_to_string(path).unwrap_or_default();
|
||||||
.unwrap_or_default();
|
|
||||||
|
|
||||||
let edited = edit::edit(&content)
|
let edited = edit::edit(&content).context("Failed to open editor")?;
|
||||||
.context("Failed to open editor")?;
|
|
||||||
|
|
||||||
fs::write(path, &edited)
|
fs::write(path, &edited).with_context(|| format!("Failed to write file: {:?}", path))?;
|
||||||
.with_context(|| format!("Failed to write file: {:?}", path))?;
|
|
||||||
|
|
||||||
Ok(edited)
|
Ok(edited)
|
||||||
}
|
}
|
||||||
@@ -29,8 +26,7 @@ pub fn edit_temp(initial_content: &str, extension: &str) -> Result<String> {
|
|||||||
.context("Failed to create temp file")?;
|
.context("Failed to create temp file")?;
|
||||||
|
|
||||||
let path = temp_file.path();
|
let path = temp_file.path();
|
||||||
fs::write(path, initial_content)
|
fs::write(path, initial_content).context("Failed to write temp file")?;
|
||||||
.context("Failed to write temp file")?;
|
|
||||||
|
|
||||||
edit_file(path)
|
edit_file(path)
|
||||||
}
|
}
|
||||||
@@ -41,8 +37,22 @@ pub fn get_editor() -> String {
|
|||||||
.or_else(|_| std::env::var("VISUAL"))
|
.or_else(|_| std::env::var("VISUAL"))
|
||||||
.unwrap_or_else(|_| {
|
.unwrap_or_else(|_| {
|
||||||
if cfg!(target_os = "windows") {
|
if cfg!(target_os = "windows") {
|
||||||
|
if let Ok(_code) = which::which("code") {
|
||||||
|
return "code --wait".to_string();
|
||||||
|
}
|
||||||
|
if let Ok(_notepad) = which::which("notepad") {
|
||||||
|
return "notepad".to_string();
|
||||||
|
}
|
||||||
"notepad".to_string()
|
"notepad".to_string()
|
||||||
|
} else if cfg!(target_os = "macos") {
|
||||||
|
if which::which("code").is_ok() {
|
||||||
|
return "code --wait".to_string();
|
||||||
|
}
|
||||||
|
"vi".to_string()
|
||||||
} else {
|
} else {
|
||||||
|
if which::which("nano").is_ok() {
|
||||||
|
return "nano".to_string();
|
||||||
|
}
|
||||||
"vi".to_string()
|
"vi".to_string()
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -51,7 +61,6 @@ pub fn get_editor() -> String {
|
|||||||
/// Check if editor is available
|
/// Check if editor is available
|
||||||
pub fn check_editor() -> Result<()> {
|
pub fn check_editor() -> Result<()> {
|
||||||
let editor = get_editor();
|
let editor = get_editor();
|
||||||
which::which(&editor)
|
which::which(&editor).with_context(|| format!("Editor '{}' not found in PATH", editor))?;
|
||||||
.with_context(|| format!("Editor '{}' not found in PATH", editor))?;
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,3 @@
|
|||||||
use chrono::{DateTime, Local, Utc};
|
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
|
|
||||||
/// Format commit message with conventional commit format
|
/// Format commit message with conventional commit format
|
||||||
@@ -12,7 +11,6 @@ pub fn format_conventional_commit(
|
|||||||
) -> String {
|
) -> String {
|
||||||
let mut message = String::new();
|
let mut message = String::new();
|
||||||
|
|
||||||
// Type and scope
|
|
||||||
message.push_str(commit_type);
|
message.push_str(commit_type);
|
||||||
if let Some(s) = scope {
|
if let Some(s) = scope {
|
||||||
message.push_str(&format!("({})", s));
|
message.push_str(&format!("({})", s));
|
||||||
@@ -22,12 +20,10 @@ pub fn format_conventional_commit(
|
|||||||
}
|
}
|
||||||
message.push_str(&format!(": {}", description));
|
message.push_str(&format!(": {}", description));
|
||||||
|
|
||||||
// Body
|
|
||||||
if let Some(b) = body {
|
if let Some(b) = body {
|
||||||
message.push_str(&format!("\n\n{}", b));
|
message.push_str(&format!("\n\n{}", b));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Footer
|
|
||||||
if let Some(f) = footer {
|
if let Some(f) = footer {
|
||||||
message.push_str(&format!("\n\n{}", f));
|
message.push_str(&format!("\n\n{}", f));
|
||||||
}
|
}
|
||||||
@@ -46,26 +42,22 @@ pub fn format_commitlint_commit(
|
|||||||
) -> String {
|
) -> String {
|
||||||
let mut message = String::new();
|
let mut message = String::new();
|
||||||
|
|
||||||
// Header
|
|
||||||
message.push_str(commit_type);
|
message.push_str(commit_type);
|
||||||
if let Some(s) = scope {
|
if let Some(s) = scope {
|
||||||
message.push_str(&format!("({})", s));
|
message.push_str(&format!("({})", s));
|
||||||
}
|
}
|
||||||
message.push_str(&format!(": {}", subject));
|
message.push_str(&format!(": {}", subject));
|
||||||
|
|
||||||
// References
|
|
||||||
if let Some(refs) = references {
|
if let Some(refs) = references {
|
||||||
for reference in refs {
|
for reference in refs {
|
||||||
message.push_str(&format!(" #{}", reference));
|
message.push_str(&format!(" #{}", reference));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Body
|
|
||||||
if let Some(b) = body {
|
if let Some(b) = body {
|
||||||
message.push_str(&format!("\n\n{}", b));
|
message.push_str(&format!("\n\n{}", b));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Footer
|
|
||||||
if let Some(f) = footer {
|
if let Some(f) = footer {
|
||||||
message.push_str(&format!("\n\n{}", f));
|
message.push_str(&format!("\n\n{}", f));
|
||||||
}
|
}
|
||||||
@@ -73,38 +65,11 @@ pub fn format_commitlint_commit(
|
|||||||
message
|
message
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Format date for commit message
|
|
||||||
pub fn format_commit_date(date: &DateTime<Local>) -> String {
|
|
||||||
date.format("%Y-%m-%d %H:%M:%S").to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Format date for changelog
|
|
||||||
pub fn format_changelog_date(date: &DateTime<Utc>) -> String {
|
|
||||||
date.format("%Y-%m-%d").to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Format tag name with version
|
|
||||||
pub fn format_tag_name(version: &str, prefix: Option<&str>) -> String {
|
|
||||||
match prefix {
|
|
||||||
Some(p) => format!("{}{}", p, version),
|
|
||||||
None => version.to_string(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Wrap text at specified width
|
/// Wrap text at specified width
|
||||||
pub fn wrap_text(text: &str, width: usize) -> String {
|
pub fn wrap_text(text: &str, width: usize) -> String {
|
||||||
textwrap::fill(text, width)
|
textwrap::fill(text, width)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Truncate text with ellipsis
|
|
||||||
pub fn truncate(text: &str, max_len: usize) -> String {
|
|
||||||
if text.len() <= max_len {
|
|
||||||
text.to_string()
|
|
||||||
} else {
|
|
||||||
format!("{}...", &text[..max_len.saturating_sub(3)])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Clean commit message (remove comments, extra whitespace)
|
/// Clean commit message (remove comments, extra whitespace)
|
||||||
pub fn clean_message(message: &str) -> String {
|
pub fn clean_message(message: &str) -> String {
|
||||||
let comment_regex = Regex::new(r"^#.*$").unwrap();
|
let comment_regex = Regex::new(r"^#.*$").unwrap();
|
||||||
@@ -118,44 +83,6 @@ pub fn clean_message(message: &str) -> String {
|
|||||||
.to_string()
|
.to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Format list as markdown bullet points
|
|
||||||
pub fn format_markdown_list(items: &[String]) -> String {
|
|
||||||
items
|
|
||||||
.iter()
|
|
||||||
.map(|item| format!("- {}", item))
|
|
||||||
.collect::<Vec<_>>()
|
|
||||||
.join("\n")
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Format changelog section
|
|
||||||
pub fn format_changelog_section(
|
|
||||||
version: &str,
|
|
||||||
date: &str,
|
|
||||||
changes: &[(String, Vec<String>)],
|
|
||||||
) -> String {
|
|
||||||
let mut section = format!("## [{}] - {}\n\n", version, date);
|
|
||||||
|
|
||||||
for (category, items) in changes {
|
|
||||||
if !items.is_empty() {
|
|
||||||
section.push_str(&format!("### {}\n\n", category));
|
|
||||||
for item in items {
|
|
||||||
section.push_str(&format!("- {}\n", item));
|
|
||||||
}
|
|
||||||
section.push('\n');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
section
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Format git config key
|
|
||||||
pub fn format_git_config_key(section: &str, subsection: Option<&str>, key: &str) -> String {
|
|
||||||
match subsection {
|
|
||||||
Some(sub) => format!("{}.{}.{}", section, sub, key),
|
|
||||||
None => format!("{}.{}", section, key),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
@@ -189,10 +116,4 @@ mod tests {
|
|||||||
|
|
||||||
assert!(msg.starts_with("feat!: change API response format"));
|
assert!(msg.starts_with("feat!: change API response format"));
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_truncate() {
|
|
||||||
assert_eq!(truncate("hello", 10), "hello");
|
|
||||||
assert_eq!(truncate("hello world", 8), "hello...");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
321
src/utils/keyring.rs
Normal file
321
src/utils/keyring.rs
Normal file
@@ -0,0 +1,321 @@
|
|||||||
|
use anyhow::{Context, Result, bail};
|
||||||
|
use std::env;
|
||||||
|
|
||||||
|
const SERVICE_NAME: &str = "quicommit";
|
||||||
|
const ENV_API_KEY: &str = "QUICOMMIT_API_KEY";
|
||||||
|
|
||||||
|
const PAT_SERVICE_PREFIX: &str = "quicommit/pat";
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
||||||
|
pub enum KeyringStatus {
|
||||||
|
Available,
|
||||||
|
Unavailable,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct KeyringManager {
|
||||||
|
status: KeyringStatus,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl KeyringManager {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
let status = Self::check_keyring_availability();
|
||||||
|
Self { status }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn check_keyring_availability() -> KeyringStatus {
|
||||||
|
#[cfg(target_os = "windows")]
|
||||||
|
{
|
||||||
|
KeyringStatus::Available
|
||||||
|
}
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
{
|
||||||
|
KeyringStatus::Available
|
||||||
|
}
|
||||||
|
#[cfg(target_os = "linux")]
|
||||||
|
{
|
||||||
|
Self::check_linux_keyring()
|
||||||
|
}
|
||||||
|
#[cfg(not(any(target_os = "windows", target_os = "macos", target_os = "linux")))]
|
||||||
|
{
|
||||||
|
KeyringStatus::Unavailable
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(target_os = "linux")]
|
||||||
|
fn check_linux_keyring() -> KeyringStatus {
|
||||||
|
use std::path::Path;
|
||||||
|
|
||||||
|
let has_dbus = Path::new("/usr/bin/dbus-daemon").exists()
|
||||||
|
|| Path::new("/bin/dbus-daemon").exists()
|
||||||
|
|| env::var("DBUS_SESSION_BUS_ADDRESS").is_ok();
|
||||||
|
|
||||||
|
let has_keyring = Path::new("/usr/bin/gnome-keyring-daemon").exists()
|
||||||
|
|| Path::new("/usr/bin/gnome-keyring").exists()
|
||||||
|
|| Path::new("/usr/bin/kwalletd5").exists()
|
||||||
|
|| Path::new("/usr/bin/kwalletd6").exists()
|
||||||
|
|| env::var("SECRET_SERVICE").is_ok();
|
||||||
|
|
||||||
|
if has_dbus && has_keyring {
|
||||||
|
KeyringStatus::Available
|
||||||
|
} else {
|
||||||
|
KeyringStatus::Unavailable
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn status(&self) -> KeyringStatus {
|
||||||
|
self.status
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn is_available(&self) -> bool {
|
||||||
|
self.status == KeyringStatus::Available
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn store_api_key(&self, provider: &str, api_key: &str) -> Result<()> {
|
||||||
|
if !self.is_available() {
|
||||||
|
bail!("Keyring is not available on this system");
|
||||||
|
}
|
||||||
|
|
||||||
|
let entry = keyring::Entry::new(SERVICE_NAME, provider)
|
||||||
|
.context("Failed to create keyring entry")?;
|
||||||
|
|
||||||
|
entry
|
||||||
|
.set_password(api_key)
|
||||||
|
.context("Failed to store API key")?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_api_key(&self, provider: &str) -> Result<Option<String>> {
|
||||||
|
if let Ok(key) = env::var(ENV_API_KEY)
|
||||||
|
&& !key.is_empty()
|
||||||
|
{
|
||||||
|
return Ok(Some(key));
|
||||||
|
}
|
||||||
|
|
||||||
|
if !self.is_available() {
|
||||||
|
return Ok(None);
|
||||||
|
}
|
||||||
|
|
||||||
|
let entry = keyring::Entry::new(SERVICE_NAME, provider)
|
||||||
|
.context("Failed to create keyring entry")?;
|
||||||
|
|
||||||
|
match entry.get_password() {
|
||||||
|
Ok(key) => Ok(Some(key)),
|
||||||
|
Err(keyring::Error::NoEntry) => Ok(None),
|
||||||
|
Err(e) => Err(e.into()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn delete_api_key(&self, provider: &str) -> Result<()> {
|
||||||
|
if !self.is_available() {
|
||||||
|
bail!("Keyring is not available on this system");
|
||||||
|
}
|
||||||
|
|
||||||
|
let entry = keyring::Entry::new(SERVICE_NAME, provider)
|
||||||
|
.context("Failed to create keyring entry")?;
|
||||||
|
|
||||||
|
entry
|
||||||
|
.delete_credential()
|
||||||
|
.context("Failed to delete API key")?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn has_api_key(&self, provider: &str) -> bool {
|
||||||
|
self.get_api_key(provider).unwrap_or(None).is_some()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn make_pat_service_name(profile_name: &str) -> String {
|
||||||
|
format!("{}/{}", PAT_SERVICE_PREFIX, profile_name)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn store_pat(
|
||||||
|
&self,
|
||||||
|
profile_name: &str,
|
||||||
|
user_email: &str,
|
||||||
|
service: &str,
|
||||||
|
token: &str,
|
||||||
|
) -> Result<()> {
|
||||||
|
if !self.is_available() {
|
||||||
|
bail!("Keyring is not available on this system");
|
||||||
|
}
|
||||||
|
|
||||||
|
let keyring_service = Self::make_pat_service_name(profile_name);
|
||||||
|
let keyring_user = format!("{}:{}", user_email, service);
|
||||||
|
|
||||||
|
let entry = keyring::Entry::new(&keyring_service, &keyring_user)
|
||||||
|
.context("Failed to create keyring entry for PAT")?;
|
||||||
|
|
||||||
|
entry
|
||||||
|
.set_password(token)
|
||||||
|
.context("Failed to store PAT in keyring")?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_pat(
|
||||||
|
&self,
|
||||||
|
profile_name: &str,
|
||||||
|
user_email: &str,
|
||||||
|
service: &str,
|
||||||
|
) -> Result<Option<String>> {
|
||||||
|
if !self.is_available() {
|
||||||
|
return Ok(None);
|
||||||
|
}
|
||||||
|
|
||||||
|
let keyring_service = Self::make_pat_service_name(profile_name);
|
||||||
|
let keyring_user = format!("{}:{}", user_email, service);
|
||||||
|
|
||||||
|
let entry = keyring::Entry::new(&keyring_service, &keyring_user)
|
||||||
|
.context("Failed to create keyring entry for PAT")?;
|
||||||
|
|
||||||
|
match entry.get_password() {
|
||||||
|
Ok(token) => Ok(Some(token)),
|
||||||
|
Err(keyring::Error::NoEntry) => Ok(None),
|
||||||
|
Err(e) => Err(e.into()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn delete_pat(&self, profile_name: &str, user_email: &str, service: &str) -> Result<()> {
|
||||||
|
if !self.is_available() {
|
||||||
|
bail!("Keyring is not available on this system");
|
||||||
|
}
|
||||||
|
|
||||||
|
let keyring_service = Self::make_pat_service_name(profile_name);
|
||||||
|
let keyring_user = format!("{}:{}", user_email, service);
|
||||||
|
|
||||||
|
let entry = keyring::Entry::new(&keyring_service, &keyring_user)
|
||||||
|
.context("Failed to create keyring entry for PAT")?;
|
||||||
|
|
||||||
|
entry
|
||||||
|
.delete_credential()
|
||||||
|
.context("Failed to delete PAT from keyring")?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn has_pat(&self, profile_name: &str, user_email: &str, service: &str) -> bool {
|
||||||
|
self.get_pat(profile_name, user_email, service)
|
||||||
|
.unwrap_or(None)
|
||||||
|
.is_some()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn delete_all_pats_for_profile(
|
||||||
|
&self,
|
||||||
|
profile_name: &str,
|
||||||
|
user_email: &str,
|
||||||
|
services: &[String],
|
||||||
|
) -> Result<()> {
|
||||||
|
for service in services {
|
||||||
|
let _ = self.delete_pat(profile_name, user_email, service);
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_status_message(&self) -> String {
|
||||||
|
match self.status {
|
||||||
|
KeyringStatus::Available => {
|
||||||
|
#[cfg(target_os = "windows")]
|
||||||
|
{
|
||||||
|
"Windows Credential Manager is available".to_string()
|
||||||
|
}
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
{
|
||||||
|
"macOS Keychain is available".to_string()
|
||||||
|
}
|
||||||
|
#[cfg(target_os = "linux")]
|
||||||
|
{
|
||||||
|
"Linux secret service is available".to_string()
|
||||||
|
}
|
||||||
|
#[cfg(not(any(target_os = "windows", target_os = "macos", target_os = "linux")))]
|
||||||
|
{
|
||||||
|
"Keyring is available".to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
KeyringStatus::Unavailable => "Keyring is not available on this system.".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for KeyringManager {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_default_base_url(provider: &str) -> &'static str {
|
||||||
|
match provider {
|
||||||
|
"openai" => "https://api.openai.com/v1",
|
||||||
|
"anthropic" => "https://api.anthropic.com/v1",
|
||||||
|
"kimi" => "https://api.moonshot.cn/v1",
|
||||||
|
"deepseek" => "https://api.deepseek.com/v1",
|
||||||
|
"openrouter" => "https://openrouter.ai/api/v1",
|
||||||
|
"ollama" => "http://localhost:11434",
|
||||||
|
_ => "",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_default_model(provider: &str) -> &'static str {
|
||||||
|
match provider {
|
||||||
|
"openai" => "gpt-4",
|
||||||
|
"anthropic" => "claude-3-sonnet-20240229",
|
||||||
|
"kimi" => "kimi-k2.6",
|
||||||
|
"deepseek" => "deepseek-v4-flash",
|
||||||
|
"openrouter" => "openai/gpt-3.5-turbo",
|
||||||
|
"ollama" => "llama2",
|
||||||
|
_ => "",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_supported_providers() -> &'static [&'static str] {
|
||||||
|
&[
|
||||||
|
"ollama",
|
||||||
|
"openai",
|
||||||
|
"anthropic",
|
||||||
|
"kimi",
|
||||||
|
"deepseek",
|
||||||
|
"openrouter",
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn provider_needs_api_key(provider: &str) -> bool {
|
||||||
|
provider != "ollama"
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_get_default_base_url() {
|
||||||
|
assert_eq!(get_default_base_url("openai"), "https://api.openai.com/v1");
|
||||||
|
assert_eq!(
|
||||||
|
get_default_base_url("anthropic"),
|
||||||
|
"https://api.anthropic.com/v1"
|
||||||
|
);
|
||||||
|
assert_eq!(get_default_base_url("kimi"), "https://api.moonshot.cn/v1");
|
||||||
|
assert_eq!(
|
||||||
|
get_default_base_url("deepseek"),
|
||||||
|
"https://api.deepseek.com/v1"
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
get_default_base_url("openrouter"),
|
||||||
|
"https://openrouter.ai/api/v1"
|
||||||
|
);
|
||||||
|
assert_eq!(get_default_base_url("ollama"), "http://localhost:11434");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_get_default_model() {
|
||||||
|
assert_eq!(get_default_model("openai"), "gpt-4");
|
||||||
|
assert_eq!(get_default_model("anthropic"), "claude-3-sonnet-20240229");
|
||||||
|
assert_eq!(get_default_model("ollama"), "llama2");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_provider_needs_api_key() {
|
||||||
|
assert!(provider_needs_api_key("openai"));
|
||||||
|
assert!(provider_needs_api_key("anthropic"));
|
||||||
|
assert!(!provider_needs_api_key("ollama"));
|
||||||
|
}
|
||||||
|
}
|
||||||
157
src/utils/mod.rs
157
src/utils/mod.rs
@@ -1,52 +1,149 @@
|
|||||||
pub mod crypto;
|
pub mod crypto;
|
||||||
pub mod editor;
|
pub mod editor;
|
||||||
pub mod formatter;
|
pub mod formatter;
|
||||||
|
pub mod keyring;
|
||||||
pub mod validators;
|
pub mod validators;
|
||||||
|
|
||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use colored::Colorize;
|
use colored::Colorize;
|
||||||
use std::io::{self, Write};
|
use std::io;
|
||||||
|
use std::sync::OnceLock;
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
|
/// Process-global decoration switch (issue 14).
|
||||||
|
static EMOJI_ENABLED: OnceLock<bool> = OnceLock::new();
|
||||||
|
|
||||||
|
/// Set whether emoji/symbol decorations are shown. Called once from `main`
|
||||||
|
/// after resolving config + flags.
|
||||||
|
pub fn set_emoji_enabled(enabled: bool) {
|
||||||
|
let _ = EMOJI_ENABLED.set(enabled);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Whether emoji/symbol decorations are shown.
|
||||||
|
pub fn emoji_enabled() -> bool {
|
||||||
|
*EMOJI_ENABLED.get_or_init(|| true)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Styled decoration prefix for status lines; empty when decorations are off
|
||||||
|
/// (ADR-0003: message text never embeds decorations).
|
||||||
|
pub fn success_prefix() -> colored::ColoredString {
|
||||||
|
if emoji_enabled() {
|
||||||
|
"✓".green().bold()
|
||||||
|
} else {
|
||||||
|
"".normal()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn error_prefix() -> colored::ColoredString {
|
||||||
|
if emoji_enabled() {
|
||||||
|
"✗".red().bold()
|
||||||
|
} else {
|
||||||
|
"".normal()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn warning_prefix() -> colored::ColoredString {
|
||||||
|
if emoji_enabled() {
|
||||||
|
"⚠".yellow().bold()
|
||||||
|
} else {
|
||||||
|
"".normal()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn info_prefix() -> colored::ColoredString {
|
||||||
|
if emoji_enabled() {
|
||||||
|
"ℹ".blue().bold()
|
||||||
|
} else {
|
||||||
|
"".normal()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn progress_prefix() -> colored::ColoredString {
|
||||||
|
if emoji_enabled() {
|
||||||
|
"→".cyan()
|
||||||
|
} else {
|
||||||
|
"".normal()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn print_decorated(prefix: colored::ColoredString, msg: &str, to_stderr: bool) {
|
||||||
|
if emoji_enabled() {
|
||||||
|
if to_stderr {
|
||||||
|
eprintln!("{} {}", prefix, msg);
|
||||||
|
} else {
|
||||||
|
println!("{} {}", prefix, msg);
|
||||||
|
}
|
||||||
|
} else if to_stderr {
|
||||||
|
eprintln!("{}", msg);
|
||||||
|
} else {
|
||||||
|
println!("{}", msg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Print success message
|
/// Print success message
|
||||||
pub fn print_success(msg: &str) {
|
pub fn print_success(msg: &str) {
|
||||||
println!("{} {}", "✓".green().bold(), msg);
|
print_decorated(success_prefix(), msg, false);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Print error message
|
/// Print error message to stderr
|
||||||
pub fn print_error(msg: &str) {
|
pub fn print_error(msg: &str) {
|
||||||
eprintln!("{} {}", "✗".red().bold(), msg);
|
print_decorated(error_prefix(), msg, true);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Print warning message
|
/// Print warning message
|
||||||
pub fn print_warning(msg: &str) {
|
pub fn print_warning(msg: &str) {
|
||||||
println!("{} {}", "⚠".yellow().bold(), msg);
|
print_decorated(warning_prefix(), msg, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Print warning message to stderr (stdout must stay clean, e.g. exports)
|
||||||
|
pub fn eprint_warning(msg: &str) {
|
||||||
|
print_decorated(warning_prefix(), msg, true);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Print info message
|
/// Print info message
|
||||||
pub fn print_info(msg: &str) {
|
pub fn print_info(msg: &str) {
|
||||||
println!("{} {}", "ℹ".blue().bold(), msg);
|
print_decorated(info_prefix(), msg, false);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Confirm action with user
|
/// Print progress/status message (→ prefix when decorations are on)
|
||||||
pub fn confirm(prompt: &str) -> Result<bool> {
|
pub fn print_progress(msg: &str) {
|
||||||
print!("{} [y/N] ", prompt);
|
print_decorated(progress_prefix(), msg, false);
|
||||||
io::stdout().flush()?;
|
|
||||||
|
|
||||||
let mut input = String::new();
|
|
||||||
io::stdin().read_line(&mut input)?;
|
|
||||||
|
|
||||||
Ok(input.trim().to_lowercase().starts_with('y'))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get user input
|
/// Progress spinner for long-running operations (issue 21).
|
||||||
pub fn input(prompt: &str) -> Result<String> {
|
/// Degrades to a static line when stdout is not a terminal or when
|
||||||
print!("{}: ", prompt);
|
/// decorations are disabled.
|
||||||
io::stdout().flush()?;
|
pub struct Spinner {
|
||||||
|
bar: Option<indicatif::ProgressBar>,
|
||||||
|
}
|
||||||
|
|
||||||
let mut input = String::new();
|
impl Spinner {
|
||||||
io::stdin().read_line(&mut input)?;
|
pub fn start(msg: &str) -> Self {
|
||||||
|
if emoji_enabled() && std::io::IsTerminal::is_terminal(&io::stdout()) {
|
||||||
|
let bar = indicatif::ProgressBar::new_spinner();
|
||||||
|
bar.set_message(msg.to_string());
|
||||||
|
bar.enable_steady_tick(Duration::from_millis(100));
|
||||||
|
Self { bar: Some(bar) }
|
||||||
|
} else {
|
||||||
|
println!("{}", msg);
|
||||||
|
Self { bar: None }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Ok(input.trim().to_string())
|
/// Stop the spinner, erasing it from the terminal (no completion line).
|
||||||
|
pub fn finish_clear(&self) {
|
||||||
|
if let Some(bar) = &self.bar {
|
||||||
|
bar.finish_and_clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Stop the spinner and print a completion line.
|
||||||
|
pub fn finish_with(&self, msg: &str) {
|
||||||
|
if let Some(bar) = &self.bar {
|
||||||
|
bar.finish_and_clear();
|
||||||
|
}
|
||||||
|
println!("{}", msg);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get password input (hidden)
|
/// Get password input (hidden)
|
||||||
@@ -58,19 +155,3 @@ pub fn password_input(prompt: &str) -> Result<String> {
|
|||||||
.interact()
|
.interact()
|
||||||
.context("Failed to read password")
|
.context("Failed to read password")
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Check if running in a terminal
|
|
||||||
pub fn is_terminal() -> bool {
|
|
||||||
atty::is(atty::Stream::Stdout)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Format duration in human-readable format
|
|
||||||
pub fn format_duration(secs: u64) -> String {
|
|
||||||
if secs < 60 {
|
|
||||||
format!("{}s", secs)
|
|
||||||
} else if secs < 3600 {
|
|
||||||
format!("{}m {}s", secs / 60, secs % 60)
|
|
||||||
} else {
|
|
||||||
format!("{}h {}m", secs / 3600, (secs % 3600) / 60)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use anyhow::{bail, Result};
|
use anyhow::{Result, bail};
|
||||||
use lazy_static::lazy_static;
|
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
|
use std::sync::LazyLock;
|
||||||
|
|
||||||
/// Conventional commit types
|
/// Conventional commit types
|
||||||
pub const CONVENTIONAL_TYPES: &[&str] = &[
|
pub const CONVENTIONAL_TYPES: &[&str] = &[
|
||||||
@@ -37,37 +37,33 @@ pub const COMMITLINT_TYPES: &[&str] = &[
|
|||||||
"security", // Security-related changes
|
"security", // Security-related changes
|
||||||
];
|
];
|
||||||
|
|
||||||
lazy_static! {
|
/// Regex for conventional commit format
|
||||||
/// Regex for conventional commit format
|
static CONVENTIONAL_COMMIT_REGEX: LazyLock<Regex> = LazyLock::new(|| {
|
||||||
static ref CONVENTIONAL_COMMIT_REGEX: Regex = Regex::new(
|
Regex::new(
|
||||||
r"^(?P<type>feat|fix|docs|style|refactor|perf|test|build|ci|chore|revert)(?:\((?P<scope>[^)]+)\))?(?P<breaking>!)?: (?P<description>.+)$"
|
r"^(?P<type>feat|fix|docs|style|refactor|perf|test|build|ci|chore|revert)(?:\((?P<scope>[^)]+)\))?(?P<breaking>!)?: (?P<description>.+)$",
|
||||||
).unwrap();
|
)
|
||||||
|
.unwrap()
|
||||||
|
});
|
||||||
|
|
||||||
/// Regex for scope validation
|
/// Regex for scope validation
|
||||||
static ref SCOPE_REGEX: Regex = Regex::new(
|
static SCOPE_REGEX: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"^[a-z0-9-]+$").unwrap());
|
||||||
r"^[a-z0-9-]+$"
|
|
||||||
).unwrap();
|
|
||||||
|
|
||||||
/// Regex for version tag validation (semver)
|
/// Regex for version tag validation (semver)
|
||||||
static ref SEMVER_REGEX: Regex = Regex::new(
|
static SEMVER_REGEX: LazyLock<Regex> = LazyLock::new(|| {
|
||||||
r"^v?(?P<major>0|[1-9]\d*)\.(?P<minor>0|[1-9]\d*)\.(?P<patch>0|[1-9]\d*)(?:-(?P<prerelease>(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*)(?:\.(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*))*))?(?:\+(?P<buildmetadata>[0-9a-zA-Z-]+(?:\.[0-9a-zA-Z-]+)*))?$"
|
Regex::new(
|
||||||
).unwrap();
|
r"^v?(?P<major>0|[1-9]\d*)\.(?P<minor>0|[1-9]\d*)\.(?P<patch>0|[1-9]\d*)(?:-(?P<prerelease>(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*)(?:\.(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*))*))?(?:\+(?P<buildmetadata>[0-9a-zA-Z-]+(?:\.[0-9a-zA-Z-]+)*))?$",
|
||||||
|
)
|
||||||
|
.unwrap()
|
||||||
|
});
|
||||||
|
|
||||||
/// Regex for email validation
|
/// Regex for email validation
|
||||||
static ref EMAIL_REGEX: Regex = Regex::new(
|
static EMAIL_REGEX: LazyLock<Regex> = LazyLock::new(|| {
|
||||||
r"^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$"
|
Regex::new(r"^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$").unwrap()
|
||||||
).unwrap();
|
});
|
||||||
|
|
||||||
/// Regex for SSH key validation (basic)
|
/// Regex for GPG key ID validation
|
||||||
static ref SSH_KEY_REGEX: Regex = Regex::new(
|
static GPG_KEY_ID_REGEX: LazyLock<Regex> =
|
||||||
r"^(ssh-rsa|ssh-ed25519|ecdsa-sha2-nistp256|ecdsa-sha2-nistp384|ecdsa-sha2-nistp521)\s+[A-Za-z0-9+/]+={0,2}\s+.*$"
|
LazyLock::new(|| Regex::new(r"^[A-F0-9]{16,40}$").unwrap());
|
||||||
).unwrap();
|
|
||||||
|
|
||||||
/// Regex for GPG key ID validation
|
|
||||||
static ref GPG_KEY_ID_REGEX: Regex = Regex::new(
|
|
||||||
r"^[A-F0-9]{16,40}$"
|
|
||||||
).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Validate conventional commit message
|
/// Validate conventional commit message
|
||||||
pub fn validate_conventional_commit(message: &str) -> Result<()> {
|
pub fn validate_conventional_commit(message: &str) -> Result<()> {
|
||||||
@@ -81,7 +77,6 @@ pub fn validate_conventional_commit(message: &str) -> Result<()> {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check description length (max 100 chars for first line)
|
|
||||||
if first_line.len() > 100 {
|
if first_line.len() > 100 {
|
||||||
bail!("Commit subject too long (max 100 characters)");
|
bail!("Commit subject too long (max 100 characters)");
|
||||||
}
|
}
|
||||||
@@ -93,7 +88,6 @@ pub fn validate_conventional_commit(message: &str) -> Result<()> {
|
|||||||
pub fn validate_commitlint_commit(message: &str) -> Result<()> {
|
pub fn validate_commitlint_commit(message: &str) -> Result<()> {
|
||||||
let first_line = message.lines().next().unwrap_or("");
|
let first_line = message.lines().next().unwrap_or("");
|
||||||
|
|
||||||
// Commitlint is more lenient but still requires type prefix
|
|
||||||
let parts: Vec<&str> = first_line.splitn(2, ':').collect();
|
let parts: Vec<&str> = first_line.splitn(2, ':').collect();
|
||||||
if parts.len() != 2 {
|
if parts.len() != 2 {
|
||||||
bail!("Invalid commit format. Expected: <type>[optional scope]: <subject>");
|
bail!("Invalid commit format. Expected: <type>[optional scope]: <subject>");
|
||||||
@@ -102,7 +96,6 @@ pub fn validate_commitlint_commit(message: &str) -> Result<()> {
|
|||||||
let type_part = parts[0];
|
let type_part = parts[0];
|
||||||
let subject = parts[1].trim();
|
let subject = parts[1].trim();
|
||||||
|
|
||||||
// Extract type (handle scope and breaking indicator)
|
|
||||||
let commit_type = type_part
|
let commit_type = type_part
|
||||||
.split('(')
|
.split('(')
|
||||||
.next()
|
.next()
|
||||||
@@ -117,7 +110,6 @@ pub fn validate_commitlint_commit(message: &str) -> Result<()> {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Validate subject
|
|
||||||
if subject.is_empty() {
|
if subject.is_empty() {
|
||||||
bail!("Commit subject cannot be empty");
|
bail!("Commit subject cannot be empty");
|
||||||
}
|
}
|
||||||
@@ -130,12 +122,15 @@ pub fn validate_commitlint_commit(message: &str) -> Result<()> {
|
|||||||
bail!("Commit subject too long (max 100 characters)");
|
bail!("Commit subject too long (max 100 characters)");
|
||||||
}
|
}
|
||||||
|
|
||||||
// Subject should not start with uppercase
|
if subject
|
||||||
if subject.chars().next().map(|c| c.is_uppercase()).unwrap_or(false) {
|
.chars()
|
||||||
|
.next()
|
||||||
|
.map(|c| c.is_uppercase())
|
||||||
|
.unwrap_or(false)
|
||||||
|
{
|
||||||
bail!("Commit subject should not start with uppercase letter");
|
bail!("Commit subject should not start with uppercase letter");
|
||||||
}
|
}
|
||||||
|
|
||||||
// Subject should not end with period
|
|
||||||
if subject.ends_with('.') {
|
if subject.ends_with('.') {
|
||||||
bail!("Commit subject should not end with a period");
|
bail!("Commit subject should not end with a period");
|
||||||
}
|
}
|
||||||
@@ -179,15 +174,6 @@ pub fn validate_email(email: &str) -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Validate SSH key format
|
|
||||||
pub fn validate_ssh_key(key: &str) -> Result<()> {
|
|
||||||
if !SSH_KEY_REGEX.is_match(key.trim()) {
|
|
||||||
bail!("Invalid SSH public key format");
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Validate GPG key ID
|
/// Validate GPG key ID
|
||||||
pub fn validate_gpg_key_id(key_id: &str) -> Result<()> {
|
pub fn validate_gpg_key_id(key_id: &str) -> Result<()> {
|
||||||
if !GPG_KEY_ID_REGEX.is_match(key_id) {
|
if !GPG_KEY_ID_REGEX.is_match(key_id) {
|
||||||
@@ -207,7 +193,10 @@ pub fn validate_profile_name(name: &str) -> Result<()> {
|
|||||||
bail!("Profile name too long (max 50 characters)");
|
bail!("Profile name too long (max 50 characters)");
|
||||||
}
|
}
|
||||||
|
|
||||||
if !name.chars().all(|c| c.is_alphanumeric() || c == '-' || c == '_') {
|
if !name
|
||||||
|
.chars()
|
||||||
|
.all(|c| c.is_alphanumeric() || c == '-' || c == '_')
|
||||||
|
{
|
||||||
bail!("Profile name can only contain letters, numbers, hyphens, and underscores");
|
bail!("Profile name can only contain letters, numbers, hyphens, and underscores");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
417
tests/config_export_import_tests.rs
Normal file
417
tests/config_export_import_tests.rs
Normal file
@@ -0,0 +1,417 @@
|
|||||||
|
use assert_cmd::cargo::cargo_bin_cmd;
|
||||||
|
use predicates::prelude::*;
|
||||||
|
use std::fs;
|
||||||
|
use std::path::PathBuf;
|
||||||
|
use tempfile::TempDir;
|
||||||
|
|
||||||
|
fn init_quicommit(config_path: &PathBuf) {
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
|
||||||
|
cmd.assert().success();
|
||||||
|
}
|
||||||
|
|
||||||
|
mod config_export {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_export_to_stdout() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path = temp_dir.path().join("config.toml");
|
||||||
|
init_quicommit(&config_path);
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"export",
|
||||||
|
"--config",
|
||||||
|
config_path.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("version"))
|
||||||
|
.stdout(predicate::str::contains("[llm]"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_export_to_file() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path = temp_dir.path().join("config.toml");
|
||||||
|
let export_path = temp_dir.path().join("exported.toml");
|
||||||
|
init_quicommit(&config_path);
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"export",
|
||||||
|
"--config",
|
||||||
|
config_path.to_str().unwrap(),
|
||||||
|
"--output",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
"",
|
||||||
|
]);
|
||||||
|
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("Configuration exported"));
|
||||||
|
|
||||||
|
assert!(export_path.exists(), "Export file should be created");
|
||||||
|
|
||||||
|
let content = fs::read_to_string(&export_path).unwrap();
|
||||||
|
assert!(content.contains("version"), "Export should contain version");
|
||||||
|
assert!(
|
||||||
|
content.contains("[llm]"),
|
||||||
|
"Export should contain LLM config"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_export_encrypted() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path = temp_dir.path().join("config.toml");
|
||||||
|
let export_path = temp_dir.path().join("encrypted.toml");
|
||||||
|
init_quicommit(&config_path);
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"export",
|
||||||
|
"--config",
|
||||||
|
config_path.to_str().unwrap(),
|
||||||
|
"--output",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
"test_password_123",
|
||||||
|
]);
|
||||||
|
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("encrypted and exported"));
|
||||||
|
|
||||||
|
assert!(export_path.exists(), "Export file should be created");
|
||||||
|
|
||||||
|
let content = fs::read_to_string(&export_path).unwrap();
|
||||||
|
assert!(
|
||||||
|
content.starts_with("ENCRYPTED:"),
|
||||||
|
"Encrypted file should start with ENCRYPTED:"
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
!content.contains("[llm]"),
|
||||||
|
"Encrypted content should not be readable"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod config_import {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_import_plain_config() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path = temp_dir.path().join("config.toml");
|
||||||
|
let import_path = temp_dir.path().join("import.toml");
|
||||||
|
|
||||||
|
let plain_config = r#"
|
||||||
|
version = "1"
|
||||||
|
|
||||||
|
[llm]
|
||||||
|
provider = "openai"
|
||||||
|
model = "gpt-4"
|
||||||
|
max_tokens = 1000
|
||||||
|
temperature = 0.7
|
||||||
|
timeout = 60
|
||||||
|
api_key_storage = "keyring"
|
||||||
|
|
||||||
|
[commit]
|
||||||
|
format = "conventional"
|
||||||
|
auto_generate = true
|
||||||
|
|
||||||
|
[tag]
|
||||||
|
version_prefix = "v"
|
||||||
|
auto_generate = true
|
||||||
|
|
||||||
|
[changelog]
|
||||||
|
path = "CHANGELOG.md"
|
||||||
|
auto_generate = true
|
||||||
|
|
||||||
|
[language]
|
||||||
|
output_language = "en"
|
||||||
|
keep_types_english = true
|
||||||
|
keep_changelog_types_english = true
|
||||||
|
"#;
|
||||||
|
fs::write(&import_path, plain_config).unwrap();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"import",
|
||||||
|
"--config",
|
||||||
|
config_path.to_str().unwrap(),
|
||||||
|
"--file",
|
||||||
|
import_path.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("Configuration imported"));
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"get",
|
||||||
|
"llm.provider",
|
||||||
|
"--config",
|
||||||
|
config_path.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("openai"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_import_encrypted_config() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path1 = temp_dir.path().join("config1.toml");
|
||||||
|
let config_path2 = temp_dir.path().join("config2.toml");
|
||||||
|
let export_path = temp_dir.path().join("encrypted.toml");
|
||||||
|
|
||||||
|
init_quicommit(&config_path1);
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"set",
|
||||||
|
"llm.provider",
|
||||||
|
"anthropic",
|
||||||
|
"--config",
|
||||||
|
config_path1.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"export",
|
||||||
|
"--config",
|
||||||
|
config_path1.to_str().unwrap(),
|
||||||
|
"--output",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
"secure_password",
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"import",
|
||||||
|
"--config",
|
||||||
|
config_path2.to_str().unwrap(),
|
||||||
|
"--file",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
"secure_password",
|
||||||
|
]);
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("Configuration imported"));
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"get",
|
||||||
|
"llm.provider",
|
||||||
|
"--config",
|
||||||
|
config_path2.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("anthropic"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_import_encrypted_wrong_password() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path = temp_dir.path().join("config.toml");
|
||||||
|
let export_path = temp_dir.path().join("encrypted.toml");
|
||||||
|
|
||||||
|
init_quicommit(&config_path);
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"export",
|
||||||
|
"--config",
|
||||||
|
config_path.to_str().unwrap(),
|
||||||
|
"--output",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
"correct_password",
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"import",
|
||||||
|
"--config",
|
||||||
|
config_path.to_str().unwrap(),
|
||||||
|
"--file",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
"wrong_password",
|
||||||
|
]);
|
||||||
|
cmd.assert()
|
||||||
|
.failure()
|
||||||
|
.stderr(predicate::str::contains("Failed to decrypt"));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod config_export_import_roundtrip {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_roundtrip_plain() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path1 = temp_dir.path().join("config1.toml");
|
||||||
|
let config_path2 = temp_dir.path().join("config2.toml");
|
||||||
|
let export_path = temp_dir.path().join("export.toml");
|
||||||
|
|
||||||
|
init_quicommit(&config_path1);
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"set",
|
||||||
|
"llm.model",
|
||||||
|
"gpt-4-turbo",
|
||||||
|
"--config",
|
||||||
|
config_path1.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"export",
|
||||||
|
"--config",
|
||||||
|
config_path1.to_str().unwrap(),
|
||||||
|
"--output",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
"",
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"import",
|
||||||
|
"--config",
|
||||||
|
config_path2.to_str().unwrap(),
|
||||||
|
"--file",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"get",
|
||||||
|
"llm.model",
|
||||||
|
"--config",
|
||||||
|
config_path2.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("gpt-4-turbo"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_roundtrip_encrypted() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let config_path1 = temp_dir.path().join("config1.toml");
|
||||||
|
let config_path2 = temp_dir.path().join("config2.toml");
|
||||||
|
let export_path = temp_dir.path().join("encrypted.toml");
|
||||||
|
let password = "my_secure_password_123";
|
||||||
|
|
||||||
|
init_quicommit(&config_path1);
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"set",
|
||||||
|
"llm.provider",
|
||||||
|
"deepseek",
|
||||||
|
"--config",
|
||||||
|
config_path1.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"set",
|
||||||
|
"llm.model",
|
||||||
|
"deepseek-chat",
|
||||||
|
"--config",
|
||||||
|
config_path1.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"export",
|
||||||
|
"--config",
|
||||||
|
config_path1.to_str().unwrap(),
|
||||||
|
"--output",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
password,
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let exported_content = fs::read_to_string(&export_path).unwrap();
|
||||||
|
assert!(exported_content.starts_with("ENCRYPTED:"));
|
||||||
|
assert!(!exported_content.contains("deepseek"));
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"import",
|
||||||
|
"--config",
|
||||||
|
config_path2.to_str().unwrap(),
|
||||||
|
"--file",
|
||||||
|
export_path.to_str().unwrap(),
|
||||||
|
"--password",
|
||||||
|
password,
|
||||||
|
]);
|
||||||
|
cmd.assert().success();
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"get",
|
||||||
|
"llm.provider",
|
||||||
|
"--config",
|
||||||
|
config_path2.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("deepseek"));
|
||||||
|
|
||||||
|
let mut cmd = cargo_bin_cmd!("quicommit");
|
||||||
|
cmd.args(&[
|
||||||
|
"config",
|
||||||
|
"get",
|
||||||
|
"llm.model",
|
||||||
|
"--config",
|
||||||
|
config_path2.to_str().unwrap(),
|
||||||
|
]);
|
||||||
|
cmd.assert()
|
||||||
|
.success()
|
||||||
|
.stdout(predicate::str::contains("deepseek-chat"));
|
||||||
|
}
|
||||||
|
}
|
||||||
1124
tests/credential_tests.rs
Normal file
1124
tests/credential_tests.rs
Normal file
File diff suppressed because it is too large
Load Diff
268
tests/gitignore_tests.rs
Normal file
268
tests/gitignore_tests.rs
Normal file
@@ -0,0 +1,268 @@
|
|||||||
|
use quicommit::git::GitRepo;
|
||||||
|
use std::fs;
|
||||||
|
use std::path::Path;
|
||||||
|
use std::process::Command;
|
||||||
|
use tempfile::TempDir;
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Helpers
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
/// Run a git command in the given directory, returning stdout as a String.
|
||||||
|
/// Panics if the command fails.
|
||||||
|
fn git(dir: &Path, args: &[&str]) -> String {
|
||||||
|
let output = Command::new("git")
|
||||||
|
.args(args)
|
||||||
|
.current_dir(dir)
|
||||||
|
.output()
|
||||||
|
.expect("Failed to execute git command");
|
||||||
|
if !output.status.success() {
|
||||||
|
let stderr = String::from_utf8_lossy(&output.stderr);
|
||||||
|
panic!("git {:?} failed in {:?}: {}", args, dir, stderr);
|
||||||
|
}
|
||||||
|
String::from_utf8_lossy(&output.stdout).to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Initialize a new git repo in the given directory and configure a local
|
||||||
|
/// user identity so commits can be created.
|
||||||
|
fn init_repo(dir: &Path) {
|
||||||
|
git(dir, &["init"]);
|
||||||
|
git(dir, &["config", "user.name", "Test User"]);
|
||||||
|
git(dir, &["config", "user.email", "test@example.com"]);
|
||||||
|
// Disable commit signing in case the global config enables it.
|
||||||
|
git(dir, &["config", "commit.gpgsign", "false"]);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Write a file with the given content, creating parent directories as needed.
|
||||||
|
fn write_file(dir: &Path, rel_path: &str, content: &str) {
|
||||||
|
let file_path = dir.join(rel_path);
|
||||||
|
if let Some(parent) = file_path.parent() {
|
||||||
|
fs::create_dir_all(parent).expect("Failed to create parent directories");
|
||||||
|
}
|
||||||
|
fs::write(&file_path, content).expect("Failed to write file");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the list of files in the index as a String (one path per line).
|
||||||
|
fn ls_files(dir: &Path) -> String {
|
||||||
|
git(dir, &["ls-files"])
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Tests for is_path_ignored
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_is_path_ignored_with_ignored_path() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let repo_path = temp_dir.path();
|
||||||
|
init_repo(repo_path);
|
||||||
|
|
||||||
|
write_file(repo_path, ".gitignore", "__pycache__/\n");
|
||||||
|
write_file(repo_path, "__pycache__/foo.pyc", "bytecode");
|
||||||
|
write_file(repo_path, "subdir/__pycache__/bar.pyc", "more bytecode");
|
||||||
|
|
||||||
|
let repo = GitRepo::open(repo_path).expect("Failed to open repo");
|
||||||
|
|
||||||
|
// Top-level ignored file
|
||||||
|
assert!(
|
||||||
|
repo.is_path_ignored("__pycache__/foo.pyc").unwrap(),
|
||||||
|
"__pycache__/foo.pyc should be ignored"
|
||||||
|
);
|
||||||
|
|
||||||
|
// Nested ignored file under a subdirectory
|
||||||
|
assert!(
|
||||||
|
repo.is_path_ignored("subdir/__pycache__/bar.pyc").unwrap(),
|
||||||
|
"subdir/__pycache__/bar.pyc should be ignored"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_is_path_ignored_with_non_ignored_path() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let repo_path = temp_dir.path();
|
||||||
|
init_repo(repo_path);
|
||||||
|
|
||||||
|
write_file(repo_path, ".gitignore", "__pycache__/\n");
|
||||||
|
write_file(repo_path, "src/main.rs", "fn main() {}");
|
||||||
|
|
||||||
|
let repo = GitRepo::open(repo_path).expect("Failed to open repo");
|
||||||
|
|
||||||
|
assert!(
|
||||||
|
!repo.is_path_ignored("src/main.rs").unwrap(),
|
||||||
|
"src/main.rs should not be ignored"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Tests for stage_all
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_stage_all_removes_ignored_tracked_files() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let repo_path = temp_dir.path();
|
||||||
|
init_repo(repo_path);
|
||||||
|
|
||||||
|
// Create and commit __pycache__/foo.pyc so it becomes a tracked file.
|
||||||
|
write_file(repo_path, "__pycache__/foo.pyc", "original bytecode");
|
||||||
|
git(repo_path, &["add", "__pycache__/foo.pyc"]);
|
||||||
|
git(repo_path, &["commit", "-m", "initial commit"]);
|
||||||
|
|
||||||
|
// Add a .gitignore that now ignores __pycache__/.
|
||||||
|
write_file(repo_path, ".gitignore", "__pycache__/\n");
|
||||||
|
|
||||||
|
// Modify the tracked file so the working tree has unstaged changes.
|
||||||
|
write_file(repo_path, "__pycache__/foo.pyc", "modified bytecode");
|
||||||
|
|
||||||
|
let repo = GitRepo::open(repo_path).expect("Failed to open repo");
|
||||||
|
let messages = quicommit::i18n::Messages::new(quicommit::config::Language::English);
|
||||||
|
let removed = repo.stage_all(&messages).expect("stage_all should succeed");
|
||||||
|
|
||||||
|
assert!(
|
||||||
|
removed.iter().any(|f| f == "__pycache__/foo.pyc"),
|
||||||
|
"stage_all should return __pycache__/foo.pyc in removed list, got: {:?}",
|
||||||
|
removed
|
||||||
|
);
|
||||||
|
|
||||||
|
// Verify the file is no longer in the index.
|
||||||
|
let files = ls_files(repo_path);
|
||||||
|
assert!(
|
||||||
|
!files.lines().any(|l| l == "__pycache__/foo.pyc"),
|
||||||
|
"__pycache__/foo.pyc should no longer be in the index, got: {}",
|
||||||
|
files
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_stage_all_no_ignored_files_returns_empty() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let repo_path = temp_dir.path();
|
||||||
|
init_repo(repo_path);
|
||||||
|
|
||||||
|
write_file(repo_path, ".gitignore", "*.log\n");
|
||||||
|
write_file(repo_path, "src/main.rs", "fn main() {}");
|
||||||
|
|
||||||
|
let repo = GitRepo::open(repo_path).expect("Failed to open repo");
|
||||||
|
let messages = quicommit::i18n::Messages::new(quicommit::config::Language::English);
|
||||||
|
let removed = repo.stage_all(&messages).expect("stage_all should succeed");
|
||||||
|
|
||||||
|
assert!(
|
||||||
|
removed.is_empty(),
|
||||||
|
"stage_all should return empty Vec when no ignored files are tracked, got: {:?}",
|
||||||
|
removed
|
||||||
|
);
|
||||||
|
|
||||||
|
// Verify src/main.rs was staged.
|
||||||
|
let files = ls_files(repo_path);
|
||||||
|
assert!(
|
||||||
|
files.lines().any(|l| l == "src/main.rs"),
|
||||||
|
"src/main.rs should be in the index, got: {}",
|
||||||
|
files
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Tests for stage_files
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_stage_files_skips_ignored_paths() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let repo_path = temp_dir.path();
|
||||||
|
init_repo(repo_path);
|
||||||
|
|
||||||
|
write_file(repo_path, ".gitignore", "__pycache__/\n");
|
||||||
|
write_file(repo_path, "__pycache__/foo.pyc", "bytecode");
|
||||||
|
write_file(repo_path, "src/main.rs", "fn main() {}");
|
||||||
|
|
||||||
|
let repo = GitRepo::open(repo_path).expect("Failed to open repo");
|
||||||
|
let skipped = repo
|
||||||
|
.stage_files(&["__pycache__/foo.pyc", "src/main.rs"])
|
||||||
|
.expect("stage_files should succeed");
|
||||||
|
|
||||||
|
assert!(
|
||||||
|
skipped.iter().any(|f| f == "__pycache__/foo.pyc"),
|
||||||
|
"skipped list should contain __pycache__/foo.pyc, got: {:?}",
|
||||||
|
skipped
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
!skipped.iter().any(|f| f == "src/main.rs"),
|
||||||
|
"skipped list should not contain src/main.rs, got: {:?}",
|
||||||
|
skipped
|
||||||
|
);
|
||||||
|
|
||||||
|
let files = ls_files(repo_path);
|
||||||
|
assert!(
|
||||||
|
files.lines().any(|l| l == "src/main.rs"),
|
||||||
|
"src/main.rs should be staged, got: {}",
|
||||||
|
files
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
!files.lines().any(|l| l == "__pycache__/foo.pyc"),
|
||||||
|
"__pycache__/foo.pyc should not be staged, got: {}",
|
||||||
|
files
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_stage_files_all_paths_ignored() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let repo_path = temp_dir.path();
|
||||||
|
init_repo(repo_path);
|
||||||
|
|
||||||
|
write_file(repo_path, ".gitignore", "__pycache__/\n");
|
||||||
|
write_file(repo_path, "__pycache__/foo.pyc", "bytecode");
|
||||||
|
|
||||||
|
let repo = GitRepo::open(repo_path).expect("Failed to open repo");
|
||||||
|
let skipped = repo
|
||||||
|
.stage_files(&["__pycache__/foo.pyc"])
|
||||||
|
.expect("stage_files should succeed");
|
||||||
|
|
||||||
|
assert!(
|
||||||
|
skipped.iter().any(|f| f == "__pycache__/foo.pyc"),
|
||||||
|
"skipped list should contain __pycache__/foo.pyc, got: {:?}",
|
||||||
|
skipped
|
||||||
|
);
|
||||||
|
|
||||||
|
// Verify the index remains empty (the file was not staged).
|
||||||
|
let files = ls_files(repo_path);
|
||||||
|
assert!(
|
||||||
|
files.trim().is_empty(),
|
||||||
|
"index should be empty, got: {}",
|
||||||
|
files
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_stage_files_normal_paths_unchanged_behavior() {
|
||||||
|
let temp_dir = TempDir::new().unwrap();
|
||||||
|
let repo_path = temp_dir.path();
|
||||||
|
init_repo(repo_path);
|
||||||
|
|
||||||
|
// No .gitignore (or one that does not match these paths).
|
||||||
|
write_file(repo_path, "src/main.rs", "fn main() {}");
|
||||||
|
write_file(repo_path, "src/lib.rs", "pub fn lib() {}");
|
||||||
|
|
||||||
|
let repo = GitRepo::open(repo_path).expect("Failed to open repo");
|
||||||
|
let skipped = repo
|
||||||
|
.stage_files(&["src/main.rs", "src/lib.rs"])
|
||||||
|
.expect("stage_files should succeed");
|
||||||
|
|
||||||
|
assert!(
|
||||||
|
skipped.is_empty(),
|
||||||
|
"skipped list should be empty, got: {:?}",
|
||||||
|
skipped
|
||||||
|
);
|
||||||
|
|
||||||
|
let files = ls_files(repo_path);
|
||||||
|
assert!(
|
||||||
|
files.lines().any(|l| l == "src/main.rs"),
|
||||||
|
"src/main.rs should be staged, got: {}",
|
||||||
|
files
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
files.lines().any(|l| l == "src/lib.rs"),
|
||||||
|
"src/lib.rs should be staged, got: {}",
|
||||||
|
files
|
||||||
|
);
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user