20 Commits

Author SHA1 Message Date
349ff56299 chore(release): bump version to 0.5.0 and update changelog 2026-07-24 17:59:22 +08:00
206bde0786 fix: 修复 tag/changelog 生成范围,增加 --auto 模式
- fix(get_tags): 使用 find_object+peel_to_commit 支持 annotated tag
- fix(changelog): insert_changelog_entry 替换覆盖逻辑,保留已有章节
- feat: parse_changelog_versions 按 semver 降序提取版本
- feat: sort_tags_by_semver 取代纯时间排序
- feat(tag --auto): 优先读 Cargo.toml/pyproject.toml,回退 commit 分析
- feat(changelog): 无 --from 时自动检测 changelog 最高已有版本
- refactor: TagInfo::version_name(), GitRepo::find_tag_by_version()
2026-07-24 17:41:18 +08:00
995d263a48 feat(commit): 添加.gitignore文件过滤功能,自动跳过被忽略的文件
在自动暂存和`--all`模式下,检测并跳过被.gitignore规则匹配的文件,暂存完成后显示被移除的被忽略文件列表
2026-07-20 17:33:57 +08:00
19aff8a6c1 feat: 新增 Git credential helper 协议支持,实现 get|store|erase 子命令 2026-07-16 11:38:06 +08:00
b6bc091502 feat(commands): 添加 git credential helper 命令 2026-07-16 11:26:41 +08:00
14ebb6857a feat(commit): 添加提交消息模板支持
- 移除 config 命令中未使用的 List 子命令及相关显示字段
- 统一 ChangelogCommand 和 CommitCommand 的 ContentGenerator 初始化方式
2026-06-03 15:20:50 +08:00
459670f363 fix(llm): 修复 Kimi API temperature 参数配置 2026-06-01 17:48:42 +08:00
7636d0b5a6 feat(llm): 统一思考模式配置,支持显式禁用状态 2026-06-01 17:39:36 +08:00
928ebb61b4 refactor(llm): renumber system prompt rules 2026-05-27 15:37:51 +08:00
7e85cdd8b0 chore(release): 升级版本号至 0.3.0 2026-05-27 15:16:26 +08:00
90074e6e32 style: 格式化代码并优化导入顺序 2026-05-27 15:15:15 +08:00
b8182e7538 修复kimi返回信息的读取错误 2026-05-27 14:50:47 +08:00
4331b9306e LLM支持优化 2026-05-26 17:43:42 +08:00
a08bc809bb 修复bug 2026-05-26 16:30:28 +08:00
1063369d96 feat(deepseek): 添加 DeepSeek reasoning 模式支持 2026-05-26 16:27:49 +08:00
3a57d25a76 docs: 添加 QuiCommit 项目路线图文档 2026-05-14 17:04:13 +08:00
8152edba39 chore: 删除构建输出日志文件 2026-05-13 13:54:50 +08:00
679db5b1db chore: 清理大量未使用的变量、方法及结构体警告 2026-05-13 13:54:20 +08:00
b1ad68c7b5 build: 升级版本号至 0.2.0 2026-05-13 12:08:03 +08:00
280d6ec5c9 feat(generator): 按文件重要性对暂存差异排序 2026-05-13 12:07:01 +08:00
47 changed files with 8488 additions and 2605 deletions

6
.gitignore vendored
View File

@@ -6,6 +6,7 @@ Cargo.lock
# IDE # IDE
.idea/ .idea/
.trae/
.vscode/ .vscode/
*.swp *.swp
*.swo *.swo
@@ -21,3 +22,8 @@ test_output/
# Config (for development) # Config (for development)
config.toml config.toml
.claude/
CLAUDE.md
**/agents/
.scratch/
CONTEXT.md

81
AGENTS.md Normal file
View 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`.

View File

@@ -9,6 +9,70 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
暂无。 暂无。
## [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 ## [0.1.11] - 2026-03-23
### ✨ 新功能 ### ✨ 新功能

View File

@@ -1,9 +1,9 @@
[package] [package]
name = "quicommit" name = "quicommit"
version = "0.1.11" version = "0.5.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"]
@@ -33,7 +33,7 @@ git2 = "0.20.3"
which = "6.0" which = "6.0"
# HTTP client for LLM APIs # HTTP client for LLM APIs
reqwest = { version = "0.12", features = ["json", "rustls-tls"], default-features = false } reqwest = { version = "0.12", features = ["json", "rustls-tls", "stream"], default-features = false }
tokio = { version = "1.35", features = ["full"] } tokio = { version = "1.35", features = ["full"] }
# Error handling # Error handling
@@ -57,6 +57,7 @@ sha2 = "0.10"
hex = "0.4" hex = "0.4"
textwrap = "0.16" textwrap = "0.16"
async-trait = "0.1" async-trait = "0.1"
futures-util = "0.3"
serde_json = "1.0" serde_json = "1.0"
atty = "0.2" atty = "0.2"
@@ -83,11 +84,13 @@ mockall = "0.12"
wiremock = "0.6" wiremock = "0.6"
[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
View 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] **安全增强**
- 支持 PATPersonal 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] 支持 SSEServer-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 提供商和生成策略

132
README.md
View File

@@ -63,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
@@ -72,16 +72,24 @@ 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 Cargo.toml/pyproject.toml, fall back to commit analysis
quicommit tag --auto
# Specify bump type # Specify bump type
quicommit tag --bump minor quicommit tag --bump minor
@@ -98,11 +106,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
@@ -160,12 +171,69 @@ 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-llm ollama --url http://localhost:11434 --model llama2 quicommit config set-llm ollama --base-url http://localhost:11434 --model llama2
# Configure OpenAI # Configure OpenAI
quicommit config set-llm openai quicommit config set-llm openai
@@ -239,6 +307,7 @@ quicommit config reset --force
| `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
@@ -251,13 +320,14 @@ quicommit config reset --force
| `-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 |
| `-t, --think` | Enable LLM thinking/reasoning mode (overrides config) |
| `-y, --yes` | Skip confirmation | | `-y, --yes` | Skip confirmation |
| `--push` | Push after committing | | `--push` | Push after committing |
| `--remote` | Specify remote repository (default: origin) | | `--remote` | Specify remote repository (default: origin) |
@@ -268,6 +338,7 @@ quicommit config reset --force
|--------|-------------| |--------|-------------|
| `-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 Cargo.toml/pyproject.toml, 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 |
@@ -276,6 +347,7 @@ quicommit config reset --force
| `-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 |
| `-t, --think` | Enable LLM thinking/reasoning mode (overrides config) |
| `-y, --yes` | Skip confirmation | | `-y, --yes` | Skip confirmation |
### Changelog Options ### Changelog Options
@@ -283,16 +355,16 @@ quicommit config reset --force
| 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) |
| `--think` | Enable LLM thinking/reasoning mode (overrides config) |
| `-y, --yes` | Skip confirmation | | `-y, --yes` | Skip confirmation |
## Configuration File ## Configuration File
@@ -332,58 +404,25 @@ 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]
colors = true
icons = true
date_format = "%Y-%m-%d"
[repo_profiles] [repo_profiles]
"/path/to/work/project" = "work" "/path/to/work/project" = "work"
@@ -402,9 +441,6 @@ date_format = "%Y-%m-%d"
```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

View File

@@ -1,12 +1,13 @@
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
println!("cargo:rerun-if-changed=build.rs"); println!("cargo:rerun-if-changed=build.rs");
} }

View File

@@ -46,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]

View File

@@ -62,7 +62,7 @@ quicommit commit
# 手动提交 # 手动提交
quicommit commit --manual -t feat -m "添加新功能" quicommit commit --manual -t feat -m "添加新功能"
# 暂存所有文件并提交 # 暂存所有文件并提交(自动跳过 .gitignore 匹配的文件)
quicommit commit -a quicommit commit -a
# 跳过确认直接提交 # 跳过确认直接提交
@@ -71,16 +71,24 @@ 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/pyproject.toml 自动检测版本,回退到基于 commit 的分析
quicommit tag --auto
# 指定版本升级类型 # 指定版本升级类型
quicommit tag --bump minor quicommit tag --bump minor
@@ -97,11 +105,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
@@ -159,12 +170,64 @@ 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 提示中输入了 PATGit 要求
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-llm ollama --url http://localhost:11434 --model llama2 quicommit config set-llm ollama --base-url http://localhost:11434 --model llama2
# 配置OpenAI # 配置OpenAI
quicommit config set-llm openai quicommit config set-llm openai
@@ -238,6 +301,7 @@ quicommit config reset --force
| `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命令选项
@@ -250,13 +314,14 @@ quicommit config reset --force
| `-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` | 不验证提交信息 |
| `-t, --think` | 启用 LLM 思考/推理模式(覆盖配置) |
| `-y, --yes` | 跳过确认提示 | | `-y, --yes` | 跳过确认提示 |
| `--push` | 提交后推送到远程 | | `--push` | 提交后推送到远程 |
| `--remote` | 指定远程仓库默认origin | | `--remote` | 指定远程仓库默认origin |
@@ -267,6 +332,7 @@ quicommit config reset --force
|------|------| |------|------|
| `-n, --name` | 标签名称 | | `-n, --name` | 标签名称 |
| `-b, --bump` | 版本升级类型major/minor/patch | | `-b, --bump` | 版本升级类型major/minor/patch |
| `-A, --auto` | 从 Cargo.toml/pyproject.toml 自动检测版本,回退到基于 commit 的分析(与 `--bump` 互斥) |
| `-m, --message` | 标签信息 | | `-m, --message` | 标签信息 |
| `-g, --generate` | AI生成标签信息 | | `-g, --generate` | AI生成标签信息 |
| `-S, --sign` | GPG签名标签 | | `-S, --sign` | GPG签名标签 |
@@ -275,6 +341,7 @@ quicommit config reset --force
| `-p, --push` | 推送到远程 | | `-p, --push` | 推送到远程 |
| `-r, --remote` | 指定远程仓库默认origin | | `-r, --remote` | 指定远程仓库默认origin |
| `--dry-run` | 试运行 | | `--dry-run` | 试运行 |
| `-t, --think` | 启用 LLM 思考/推理模式(覆盖配置) |
| `-y, --yes` | 跳过确认提示 | | `-y, --yes` | 跳过确认提示 |
### changelog命令选项 ### changelog命令选项
@@ -282,16 +349,16 @@ quicommit config reset --force
| 选项 | 说明 | | 选项 | 说明 |
|------|------| |------|------|
| `-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 |
| `--think` | 启用 LLM 思考/推理模式(覆盖配置) |
| `-y, --yes` | 跳过确认提示 | | `-y, --yes` | 跳过确认提示 |
## 配置文件 ## 配置文件
@@ -331,58 +398,25 @@ 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]
colors = true
icons = true
date_format = "%Y-%m-%d"
[repo_profiles] [repo_profiles]
"/path/to/work/project" = "work" "/path/to/work/project" = "work"
@@ -401,9 +435,6 @@ date_format = "%Y-%m-%d"
```bash ```bash
# 查看当前配置 # 查看当前配置
quicommit config list
# 显示配置详情
quicommit config show quicommit config show
# 编辑配置文件 # 编辑配置文件

View File

@@ -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,8 +7,9 @@ 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};
/// Generate changelog /// Generate changelog
@@ -55,6 +56,10 @@ pub struct ChangelogCommand {
#[arg(long)] #[arg(long)]
dry_run: bool, dry_run: bool,
/// Enable thinking mode for this changelog (override config)
#[arg(long)]
think: bool,
/// Skip interactive prompts /// Skip interactive prompts
#[arg(short = 'y', long)] #[arg(short = 'y', long)]
yes: bool, yes: bool,
@@ -74,18 +79,20 @@ impl ChangelogCommand {
// 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))); println!("{}", messages.initialized_changelog(&format!("{:?}", path)));
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
@@ -94,7 +101,10 @@ 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!(
"Unknown format: {}. Use: keep-a-changelog, github-releases",
f
),
}; };
// Get version // Get version
@@ -111,12 +121,15 @@ impl ChangelogCommand {
// Get commits // Get commits
println!("{}", messages.fetching_commits()); println!("{}", 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())); println!("{}", messages.found_commits(commits.len()));
// Generate changelog // Generate changelog
@@ -148,7 +161,7 @@ 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(&format!("{:?}", output_path)))
.default(true) .default(true)
.interact()?; .interact()?;
@@ -158,33 +171,13 @@ impl ChangelogCommand {
} }
} }
// Write to file (always prepend to preserve history) // Write to file (always prepend new entry before existing versions)
if 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 = if existing.is_empty() { let new_content = if existing.is_empty() {
format!("{}{}", CHANGELOG_HEADER, changelog) format!("{}{}", CHANGELOG_HEADER, changelog)
} else if existing.starts_with(CHANGELOG_HEADER) {
format!("{}{}", CHANGELOG_HEADER, changelog)
} else if existing.starts_with("# Changelog") {
let lines: Vec<&str> = existing.lines().collect();
let mut header_end = 0;
for (i, line) in lines.iter().enumerate() {
if i == 0 && line.starts_with('#') {
header_end = i + 1;
} else if line.trim().is_empty() {
header_end = i + 1;
} else {
break;
}
}
let header = lines[..header_end].join("\n");
let rest = lines[header_end..].join("\n");
format!("{}\n{}\n{}", header, changelog, rest)
} else { } else {
format!("{}{}", CHANGELOG_HEADER, changelog) insert_changelog_entry(&existing, &changelog)
}; };
std::fs::write(&output_path, new_content)?; std::fs::write(&output_path, new_content)?;
} else { } else {
@@ -197,6 +190,28 @@ impl ChangelogCommand {
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,
@@ -208,8 +223,10 @@ impl ChangelogCommand {
println!("{}", messages.ai_generating_changelog()); println!("{}", messages.ai_generating_changelog());
let generator = ContentGenerator::new(&manager).await?; let generator = ContentGenerator::new_with_think(&manager, self.think, None).await?;
generator.generate_changelog_entry(version, commits, language).await generator
.generate_changelog_entry(version, commits, language)
.await
} }
fn generate_with_template( fn generate_with_template(
@@ -220,14 +237,14 @@ impl ChangelogCommand {
language: Language, language: Language,
) -> Result<String> { ) -> Result<String> {
let manager = ConfigManager::new()?; let manager = ConfigManager::new()?;
let generator = ChangelogGenerator::new() let generator = ChangelogGenerator::new()
.format(format) .format(format)
.include_hashes(self.include_hashes) .include_hashes(self.include_hashes)
.include_authors(self.include_authors); .include_authors(self.include_authors);
let changelog = generator.generate(version, Utc::now(), commits)?; let changelog = generator.generate(version, Utc::now(), commits)?;
// Translate changelog categories if configured // Translate changelog categories if configured
if !manager.keep_changelog_types_english() { if !manager.keep_changelog_types_english() {
Ok(self.translate_changelog_categories(&changelog, language)) Ok(self.translate_changelog_categories(&changelog, language))
@@ -235,14 +252,15 @@ impl ChangelogCommand {
Ok(changelog) Ok(changelog)
} }
} }
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 {
@@ -253,7 +271,6 @@ impl ChangelogCommand {
} }
}) })
.collect::<Vec<_>>() .collect::<Vec<_>>()
.join("\n"); .join("\n")
translated
} }
} }

View File

@@ -1,14 +1,14 @@
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 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;
@@ -71,6 +71,10 @@ pub struct CommitCommand {
#[arg(long)] #[arg(long)]
no_verify: bool, no_verify: bool,
/// Enable thinking mode for this commit (override config)
#[arg(short = 't', long)]
think: bool,
/// Skip interactive prompts /// Skip interactive prompts
#[arg(short = 'y', long)] #[arg(short = 'y', long)]
yes: bool, yes: bool,
@@ -88,7 +92,7 @@ impl CommitCommand {
pub async fn execute(&self, config_path: Option<PathBuf>) -> 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 = if let Some(ref path) = config_path { let manager = if let Some(ref path) = config_path {
ConfigManager::with_path(path)? ConfigManager::with_path(path)?
@@ -98,7 +102,7 @@ impl CommitCommand {
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);
// Check for changes // Check for changes
let status = repo.status_summary()?; let status = repo.status_summary()?;
if status.clean && !self.amend { if status.clean && !self.amend {
@@ -117,9 +121,18 @@ 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()?;
println!("{}", messages.staged_all().green()); println!("{}", messages.staged_all().green());
if !removed.is_empty() {
println!(
"{}",
format!("Removed {} ignored files from staging:", removed.len()).yellow()
);
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 {
@@ -129,8 +142,17 @@ impl CommitCommand {
// Stage all if requested // Stage all if requested
if self.all { if self.all {
repo.stage_all()?; let removed = repo.stage_all()?;
println!("{}", messages.staged_all().green()); println!("{}", messages.staged_all().green());
if !removed.is_empty() {
println!(
"{}",
format!("Removed {} ignored files from staging:", removed.len()).yellow()
);
for file in &removed {
println!("{}", file);
}
}
} }
// Generate or build commit message // Generate or build commit message
@@ -179,14 +201,22 @@ impl CommitCommand {
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()
@@ -196,9 +226,13 @@ 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()); println!(
"{} {}",
messages.commit_created().green().bold(),
commit_oid.to_string()[..8].to_string().cyan()
);
} else { } else {
println!("{} {}", messages.commit_amended().green().bold(), "successfully"); println!("{} successfully", messages.commit_amended().green().bold());
} }
// Push after commit if requested or ask user // Push after commit if requested or ask user
@@ -228,8 +262,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 {
@@ -255,10 +290,20 @@ 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 generator = ContentGenerator::new(&manager).await let template = manager
.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()); println!("{}", messages.ai_analyzing());
@@ -266,15 +311,23 @@ impl CommitCommand {
let language = manager.get_language().unwrap_or(Language::English); let language = manager.get_language().unwrap_or(Language::English);
let generated = if self.yes { let generated = if self.yes {
generator.generate_commit_from_repo(repo, format, language).await? generator
.generate_commit_from_repo(repo, format, language)
.await?
} else { } else {
generator.generate_commit_interactive(repo, format, language).await? generator
.generate_commit_interactive(repo, format, language)
.await?
}; };
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
@@ -352,20 +405,21 @@ impl CommitCommand {
if !output.status.success() { if !output.status.success() {
let stdout = String::from_utf8_lossy(&output.stdout); let stdout = String::from_utf8_lossy(&output.stdout);
let stderr = String::from_utf8_lossy(&output.stderr); let stderr = String::from_utf8_lossy(&output.stderr);
let error_msg = if stderr.is_empty() { let error_msg = if stderr.is_empty() {
if stdout.is_empty() { if stdout.is_empty() {
"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()
} }
} else { } else {
stderr.to_string() stderr.to_string()
}; };
bail!("Failed to amend commit: {}", error_msg); bail!("Failed to amend commit: {}", error_msg);
} }

File diff suppressed because it is too large Load Diff

458
src/commands/credential.rs Normal file
View File

@@ -0,0 +1,458 @@
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()
.filter(|name| {
manager
.get_profile(name)
.map(|p| p.tokens.contains_key(&service))
.unwrap_or(false)
})
.cloned()
.collect();
for name in &profile_names {
if let Err(e) = manager.remove_token_from_profile(name, &service) {
eprintln!(
"[quicommit credential] failed to erase PAT for '{}': {}",
name, e
);
}
}
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());
}
}

View File

@@ -4,11 +4,11 @@ use colored::Colorize;
use dialoguer::{Confirm, Input, Select}; use dialoguer::{Confirm, Input, Select};
use std::path::PathBuf; 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_supported_providers, get_default_model, provider_needs_api_key}; 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;
/// Initialize quicommit configuration /// Initialize quicommit configuration
@@ -28,23 +28,25 @@ impl InitCommand {
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 = config_path.unwrap_or_else(|| { let config_path =
crate::config::AppConfig::default_path().unwrap() config_path.unwrap_or_else(|| crate::config::AppConfig::default_path().unwrap());
});
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("Configuration already exists. Overwrite?")
.default(false) .default(false)
.interact()?; .interact()?;
if !overwrite { if !overwrite {
println!("{}", "Initialization cancelled.".yellow()); println!("{}", "Initialization cancelled.".yellow());
return Ok(()); return Ok(());
} }
} else { } else {
println!("{}", "Configuration already exists. Use --reset to overwrite.".yellow()); println!(
"{}",
"Configuration already exists. Use --reset to overwrite.".yellow()
);
return Ok(()); return Ok(());
} }
} }
@@ -63,10 +65,10 @@ impl InitCommand {
} }
manager.save()?; manager.save()?;
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()); println!("{}", messages.init_success().bold().green());
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());
@@ -79,15 +81,15 @@ impl InitCommand {
async fn quick_setup(&self, manager: &mut ConfigManager) -> Result<()> { async fn quick_setup(&self, manager: &mut ConfigManager) -> Result<()> {
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 user_email = git_config.get_string("user.email").unwrap_or_else(|_| "user@example.com".to_string());
let profile = GitProfile::new( let user_name = git_config
"default".to_string(), .get_string("user.name")
user_name, .unwrap_or_else(|_| "User".to_string());
user_email, let user_email = git_config
); .get_string("user.email")
.unwrap_or_else(|_| "user@example.com".to_string());
let profile = GitProfile::new("default".to_string(), user_name, user_email);
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()))?;
@@ -102,7 +104,7 @@ impl InitCommand {
println!("\n{}", messages.setup_profile().bold()); println!("\n{}", messages.setup_profile().bold());
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,11 +113,11 @@ 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());
@@ -128,12 +130,14 @@ impl InitCommand {
.interact_text()?; .interact_text()?;
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();
@@ -145,9 +149,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()
@@ -161,9 +163,11 @@ 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
}; };
@@ -190,11 +194,7 @@ impl InitCommand {
None None
}; };
let mut profile = GitProfile::new( let mut profile = GitProfile::new(profile_name.clone(), user_name, user_email);
profile_name.clone(),
user_name,
user_email,
);
if !description.is_empty() { if !description.is_empty() {
profile.description = Some(description); profile.description = Some(description);
@@ -209,16 +209,16 @@ impl InitCommand {
manager.set_default_profile(Some(profile_name))?; manager.set_default_profile(Some(profile_name))?;
println!("\n{}", messages.select_llm_provider().bold()); println!("\n{}", messages.select_llm_provider().bold());
let provider_display_names = 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(&provider_display_names) .items(&provider_display_names)
.default(0) .default(0)
@@ -229,19 +229,28 @@ impl InitCommand {
let keyring = manager.keyring(); let keyring = manager.keyring();
let keyring_available = keyring.is_available(); let keyring_available = keyring.is_available();
if !keyring_available { if !keyring_available {
println!("\n{}", "⚠ Keyring is not available on this system.".yellow()); println!(
"\n{}",
"⚠ Keyring is not available on this system.".yellow()
);
println!("{}", keyring.get_status_message().yellow()); println!("{}", keyring.get_status_message().yellow());
} }
let api_key = if provider_needs_api_key(&provider) { let api_key = if provider_needs_api_key(&provider) {
let env_key = std::env::var("QUICOMMIT_API_KEY") let env_key = std::env::var("QUICOMMIT_API_KEY")
.or_else(|_| std::env::var(format!("QUICOMMIT_{}_API_KEY", provider.to_uppercase()))) .or_else(|_| {
std::env::var(format!("QUICOMMIT_{}_API_KEY", provider.to_uppercase()))
})
.ok(); .ok();
if let Some(_key) = env_key { if let Some(_key) = env_key {
println!("\n{} {}", "".green(), "Found API key in environment variable.".green()); println!(
"\n{} {}",
"".green(),
"Found API key in environment variable.".green()
);
None None
} else if keyring_available { } else if keyring_available {
let prompt = match provider.as_str() { let prompt = match provider.as_str() {
@@ -252,13 +261,14 @@ impl InitCommand {
"openrouter" => messages.openrouter_api_key(), "openrouter" => messages.openrouter_api_key(),
_ => "API Key", _ => "API Key",
}; };
let key: String = Input::new() let key: String = Input::new().with_prompt(prompt).interact_text()?;
.with_prompt(prompt)
.interact_text()?;
Some(key) Some(key)
} else { } else {
println!("\n{}", "Please set the QUICOMMIT_API_KEY environment variable.".yellow()); println!(
"\n{}",
"Please set the QUICOMMIT_API_KEY environment variable.".yellow()
);
None None
} }
} else { } else {
@@ -282,11 +292,9 @@ impl InitCommand {
.with_prompt("Use custom API base URL?") .with_prompt("Use custom API base URL?")
.default(false) .default(false)
.interact()?; .interact()?;
if use_custom_url { if use_custom_url {
let url: String = Input::new() let url: String = Input::new().with_prompt("Base URL").interact_text()?;
.with_prompt("Base URL")
.interact_text()?;
Some(url) Some(url)
} else { } else {
None None
@@ -297,11 +305,15 @@ impl InitCommand {
manager.set_llm_model(model); manager.set_llm_model(model);
manager.set_llm_base_url(base_url); manager.set_llm_base_url(base_url);
if let Some(key) = api_key { if let Some(key) = api_key
if provider_needs_api_key(&provider) { && provider_needs_api_key(&provider)
manager.set_api_key(&key)?; {
println!("\n{} {}", "".green(), "API key stored securely in system keyring.".green()); manager.set_api_key(&key)?;
} println!(
"\n{} {}",
"".green(),
"API key stored securely in system keyring.".green()
);
} }
Ok(()) Ok(())
@@ -319,6 +331,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("SSH public key path (optional, leave empty to auto-detect)")
.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)
@@ -330,13 +353,38 @@ impl InitCommand {
None None
}; };
let agent_forwarding = Confirm::new()
.with_prompt("Enable SSH agent forwarding (-A)?")
.default(false)
.interact()?;
let known_hosts: String = Input::new()
.with_prompt("Custom known_hosts file path (optional)")
.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("Custom SSH command (optional, overrides all other SSH settings)")
.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,
}) })
} }

View 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

View File

@@ -1,4 +1,4 @@
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};
@@ -6,11 +6,12 @@ use semver::Version;
use std::path::PathBuf; 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, TagBuilder, VersionBump, bump_version, get_latest_version, read_project_version,
suggest_version_bump,
}; };
use crate::git::{GitRepo, find_repo};
use crate::i18n::Messages; use crate::i18n::Messages;
/// Generate and create Git tags /// Generate and create Git tags
@@ -56,9 +57,18 @@ pub struct TagCommand {
#[arg(long)] #[arg(long)]
dry_run: bool, dry_run: bool,
/// Enable thinking mode for this tag (override config)
#[arg(short = 't', long)]
think: bool,
/// Skip interactive prompts /// Skip interactive prompts
#[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 {
@@ -76,33 +86,43 @@ 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);
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); println!("{}: {}", "Warning".yellow(), 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("Proceed with this tag name anyway?")
.default(true) .default(true)
.interact()?; .interact()?;
if !proceed { if !proceed {
bail!("{}", messages.tag_cancelled()); bail!("{}", messages.tag_cancelled());
} }
@@ -116,7 +136,10 @@ impl TagCommand {
} 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 && !self.yes) {
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 {
@@ -184,12 +207,17 @@ impl TagCommand {
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)?;
println!("\n{}", messages.version_selection().bold()); println!("\n{}", messages.version_selection().bold());
if let Some(ref version) = latest { if let Some(ref version) = latest {
println!("{} {}{}", messages.latest_version(), prefix, version); println!("{} {}{}", messages.latest_version(), prefix, version);
} else { } else {
@@ -216,36 +244,46 @@ 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())
.default(true) .default(true)
.interact()?; .interact()?;
if confirm { if confirm {
return Ok(format!("{}{}", prefix, version)); return Ok(format!("{}{}", prefix, version));
} }
// 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));
@@ -268,7 +306,12 @@ 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 language = manager.get_language().unwrap_or(Language::English); let language = manager.get_language().unwrap_or(Language::English);
@@ -285,18 +328,71 @@ impl TagCommand {
println!("{}", messages.ai_generating_tag(commits.len())); println!("{}", messages.ai_generating_tag(commits.len()));
let generator = ContentGenerator::new(&manager).await?; let generator = ContentGenerator::new_with_think(&manager, self.think, None).await?;
generator.generate_tag_message(version, &commits, language).await generator
.generate_tag_message(version, &commits, language)
.await
}
async fn auto_detect_version(
&self,
repo: &GitRepo,
prefix: &str,
messages: &Messages,
) -> Result<String> {
let project_dir = std::env::current_dir()?;
// 1. Try reading from project config files
if let Some(version) = read_project_version(&project_dir) {
let tag_name = format!("{}{}", prefix, version);
println!(
"{} {}",
messages.latest_version(),
tag_name.cyan()
);
return Ok(tag_name);
}
// 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 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 = format!("Release {}", version);
let use_editor = Confirm::new() let use_editor = Confirm::new()
.with_prompt(messages.open_editor()) .with_prompt(messages.open_editor())
.default(false) .default(false)
.interact()?; .interact()?;
if use_editor { if use_editor {
crate::utils::editor::edit_content(&default_msg) crate::utils::editor::edit_content(&default_msg)
} else { } else {

View File

@@ -1,6 +1,8 @@
use super::{AppConfig, GitProfile, TokenConfig}; use super::{AppConfig, GitProfile, TokenConfig};
use crate::utils::keyring::{KeyringManager, get_default_base_url, get_default_model, provider_needs_api_key}; use crate::utils::keyring::{
use anyhow::{bail, Context, Result}; KeyringManager, get_default_base_url, get_default_model, provider_needs_api_key,
};
use anyhow::{Context, Result, bail};
// use std::collections::HashMap; // use std::collections::HashMap;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
@@ -91,13 +93,13 @@ impl ConfigManager {
if !self.config.profiles.contains_key(name) { if !self.config.profiles.contains_key(name) {
bail!("Profile '{}' does not exist", name); bail!("Profile '{}' does not exist", name);
} }
if self.config.default_profile.as_ref() == Some(&name.to_string()) { if self.config.default_profile.as_ref() == Some(&name.to_string()) {
self.config.default_profile = None; self.config.default_profile = None;
} }
self.config.repo_profiles.retain(|_, v| v != name); self.config.repo_profiles.retain(|_, v| v != name);
self.config.profiles.remove(name); self.config.profiles.remove(name);
self.modified = true; self.modified = true;
Ok(()) Ok(())
@@ -136,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;
@@ -178,36 +180,49 @@ impl ConfigManager {
// Token management // Token management
/// Add a token to a profile (stores token in keyring) /// 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) { if !self.config.profiles.contains_key(profile_name) {
bail!("Profile '{}' does not exist", 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 /// 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<()> { pub fn store_pat_for_profile(
let profile = self.get_profile(profile_name) &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))?; .ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?;
let user_email = &profile.user_email; let user_email = &profile.user_email;
self.keyring.store_pat(profile_name, user_email, service, token_value) self.keyring
.store_pat(profile_name, user_email, service, token_value)
} }
/// Get a PAT token from keyring for a profile /// Get a PAT token from keyring for a profile
pub fn get_pat_for_profile(&self, profile_name: &str, service: &str) -> Result<Option<String>> { pub fn get_pat_for_profile(&self, profile_name: &str, service: &str) -> Result<Option<String>> {
let profile = self.get_profile(profile_name) let profile = self
.get_profile(profile_name)
.ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?; .ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?;
let user_email = &profile.user_email; let user_email = &profile.user_email;
self.keyring.get_pat(profile_name, user_email, service) self.keyring.get_pat(profile_name, user_email, service)
} }
@@ -226,21 +241,40 @@ impl ConfigManager {
if !self.config.profiles.contains_key(profile_name) { if !self.config.profiles.contains_key(profile_name) {
bail!("Profile '{}' does not exist", profile_name); bail!("Profile '{}' does not exist", profile_name);
} }
let user_email = self.config.profiles.get(profile_name).unwrap().user_email.clone(); let user_email = self
let services: Vec<String> = self.config.profiles.get(profile_name).unwrap().tokens.keys().cloned().collect(); .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()) { if !services.contains(&service.to_string()) {
bail!("Token for service '{}' not found in profile '{}'", service, profile_name); bail!(
"Token for service '{}' not found in profile '{}'",
service,
profile_name
);
} }
self.keyring.delete_pat(profile_name, &user_email, service)?; 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(())
} }
@@ -249,8 +283,9 @@ impl ConfigManager {
if let Some(profile) = self.get_profile(profile_name) { if let Some(profile) = self.get_profile(profile_name) {
let user_email = &profile.user_email; let user_email = &profile.user_email;
let services: Vec<String> = profile.tokens.keys().cloned().collect(); let services: Vec<String> = profile.tokens.keys().cloned().collect();
self.keyring.delete_all_pats_for_profile(profile_name, user_email, &services)?; self.keyring
.delete_all_pats_for_profile(profile_name, user_email, &services)?;
} }
Ok(()) Ok(())
} }
@@ -302,14 +337,24 @@ impl ConfigManager {
// } // }
/// Check and compare profile with git configuration /// Check and compare profile with git configuration
pub fn check_profile_config(&self, profile_name: &str, repo: &git2::Repository) -> Result<super::ProfileComparison> { pub fn check_profile_config(
let profile = self.get_profile(profile_name) &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))?; .ok_or_else(|| anyhow::anyhow!("Profile '{}' not found", profile_name))?;
profile.compare_with_git_config(repo) profile.compare_with_git_config(repo)
} }
/// Find a profile that matches the given user config (name, email, signing_key) /// Find a profile that matches the given user config (name, email, signing_key)
pub fn find_matching_profile(&self, user_name: &str, user_email: &str, signing_key: Option<&str>) -> Option<&GitProfile> { pub fn find_matching_profile(
&self,
user_name: &str,
user_email: &str,
signing_key: Option<&str>,
) -> Option<&GitProfile> {
for profile in self.config.profiles.values() { for profile in self.config.profiles.values() {
let name_match = profile.user_name == user_name; let name_match = profile.user_name == user_name;
let email_match = profile.user_email == user_email; let email_match = profile.user_email == user_email;
@@ -319,7 +364,7 @@ impl ConfigManager {
(Some(_), None) => false, (Some(_), None) => false,
(None, Some(_)) => false, (None, Some(_)) => false,
}; };
if name_match && email_match && key_match { if name_match && email_match && key_match {
return Some(profile); return Some(profile);
} }
@@ -329,7 +374,9 @@ impl ConfigManager {
/// Find profiles that partially match (same name or same email) /// Find profiles that partially match (same name or same email)
pub fn find_partial_matches(&self, user_name: &str, user_email: &str) -> Vec<&GitProfile> { pub fn find_partial_matches(&self, user_name: &str, user_email: &str) -> Vec<&GitProfile> {
self.config.profiles.values() self.config
.profiles
.values()
.filter(|p| p.user_name == user_name || p.user_email == user_email) .filter(|p| p.user_name == user_name || p.user_email == user_email)
.collect() .collect()
} }
@@ -339,6 +386,22 @@ impl ConfigManager {
self.config.repo_profiles.get(repo_path) 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
/// Get LLM provider /// Get LLM provider
@@ -384,7 +447,11 @@ impl ConfigManager {
/// Get API key from configured storage method /// Get API key from configured storage method
pub fn get_api_key(&self) -> Option<String> { pub fn get_api_key(&self) -> Option<String> {
// First try environment variables (always checked) // First try environment variables (always checked)
if let Some(key) = self.keyring.get_api_key(&self.config.llm.provider).unwrap_or(None) { if let Some(key) = self
.keyring
.get_api_key(&self.config.llm.provider)
.unwrap_or(None)
{
return Some(key); return Some(key);
} }
@@ -401,20 +468,29 @@ impl ConfigManager {
match self.config.llm.api_key_storage.as_str() { match self.config.llm.api_key_storage.as_str() {
"keyring" => { "keyring" => {
if !self.keyring.is_available() { if !self.keyring.is_available() {
bail!("Keyring is not available. Set QUICOMMIT_API_KEY environment variable instead or change api_key_storage to 'config'."); 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) self.keyring
}, .store_api_key(&self.config.llm.provider, api_key)
}
"config" => { "config" => {
// We can't modify self.config here since self is immutable // We can't modify self.config here since self is immutable
// This will be handled by the caller updating the config // This will be handled by the caller updating the config
Ok(()) Ok(())
}, }
"environment" => { "environment" => {
bail!("API key storage set to 'environment'. Please set QUICOMMIT_{}_API_KEY environment variable.", self.config.llm.provider.to_uppercase()); 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); bail!(
"Invalid API key storage method: {}",
self.config.llm.api_key_storage
);
} }
} }
} }
@@ -426,16 +502,19 @@ impl ConfigManager {
if self.keyring.is_available() { if self.keyring.is_available() {
self.keyring.delete_api_key(&self.config.llm.provider)?; self.keyring.delete_api_key(&self.config.llm.provider)?;
} }
}, }
"config" => { "config" => {
// We can't modify self.config here since self is immutable // We can't modify self.config here since self is immutable
// This will be handled by the caller updating the config // This will be handled by the caller updating the config
}, }
"environment" => { "environment" => {
// Environment variables are not managed by the app // Environment variables are not managed by the app
}, }
_ => { _ => {
bail!("Invalid API key storage method: {}", self.config.llm.api_key_storage); bail!(
"Invalid API key storage method: {}",
self.config.llm.api_key_storage
);
} }
} }
Ok(()) Ok(())
@@ -448,7 +527,12 @@ impl ConfigManager {
} }
// Check environment variables // Check environment variables
if self.keyring.get_api_key(&self.config.llm.provider).unwrap_or(None).is_some() { if self
.keyring
.get_api_key(&self.config.llm.provider)
.unwrap_or(None)
.is_some()
{
return true; return true;
} }
@@ -468,19 +552,19 @@ impl ConfigManager {
// /// Configure LLM provider with all settings // /// Configure LLM provider with all settings
// pub fn configure_llm(&mut self, provider: String, model: Option<String>, base_url: Option<String>, api_key: Option<&str>) -> Result<()> { // pub fn configure_llm(&mut self, provider: String, model: Option<String>, base_url: Option<String>, api_key: Option<&str>) -> Result<()> {
// self.set_llm_provider(provider.clone()); // self.set_llm_provider(provider.clone());
// if let Some(m) = model { // if let Some(m) = model {
// self.set_llm_model(m); // self.set_llm_model(m);
// } // }
// self.set_llm_base_url(base_url); // self.set_llm_base_url(base_url);
// if let Some(key) = api_key { // if let Some(key) = api_key {
// if provider_needs_api_key(&provider) { // if provider_needs_api_key(&provider) {
// self.set_api_key(key)?; // self.set_api_key(key)?;
// } // }
// } // }
// Ok(()) // Ok(())
// } // }
@@ -576,14 +660,12 @@ impl ConfigManager {
/// 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(())
} }

View File

@@ -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,
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,17 +41,9 @@ 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,
@@ -70,8 +60,6 @@ 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(),
} }
} }
@@ -84,11 +72,12 @@ pub struct LlmConfig {
#[serde(default = "default_llm_provider")] #[serde(default = "default_llm_provider")]
pub provider: String, pub provider: String,
/// Model to use (stored in config, not in keyring) /// Model to use
#[serde(default = "default_model")] #[serde(default = "default_model")]
pub model: String, pub model: String,
/// API base URL (optional, will use provider default if not set) /// API base URL (optional, will use provider default if not set)
#[serde(skip_serializing_if = "Option::is_none")]
pub base_url: Option<String>, pub base_url: Option<String>,
/// Maximum tokens for generation /// Maximum tokens for generation
@@ -107,9 +96,17 @@ pub struct LlmConfig {
#[serde(default = "default_api_key_storage")] #[serde(default = "default_api_key_storage")]
pub api_key_storage: String, pub api_key_storage: String,
/// API key (stored in config for fallback, encrypted if encrypt_sensitive is true) /// API key (stored in config for fallback)
#[serde(default)] #[serde(default, skip_serializing_if = "Option::is_none")]
pub api_key: Option<String>, 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 { fn default_api_key_storage() -> String {
@@ -127,6 +124,8 @@ impl Default for LlmConfig {
timeout: default_timeout(), timeout: default_timeout(),
api_key_storage: default_api_key_storage(), api_key_storage: default_api_key_storage(),
api_key: None, api_key: None,
thinking_enabled: false,
thinking_budget_tokens: None,
} }
} }
} }
@@ -141,33 +140,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 {
@@ -175,13 +147,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(),
} }
} }
} }
@@ -213,18 +178,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 {
@@ -232,9 +185,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,
} }
} }
} }
@@ -249,26 +199,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 {
@@ -276,56 +206,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(),
} }
} }
} }
@@ -408,6 +288,7 @@ impl Language {
} }
// Default value functions // Default value functions
fn default_version() -> String { fn default_version() -> String {
"1".to_string() "1".to_string()
} }
@@ -440,14 +321,6 @@ 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()
} }
@@ -456,14 +329,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()
} }
@@ -484,41 +349,24 @@ 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)
.with_context(|| format!("Failed to create config directory: {:?}", parent))?; .with_context(|| format!("Failed to create config directory: {:?}", parent))?;
} }
fs::write(path, content) fs::write(path, content)
.with_context(|| format!("Failed to write config file: {:?}", path))?; .with_context(|| format!("Failed to write config file: {:?}", path))?;
Ok(()) Ok(())
} }
/// 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
// pub fn get_profile_for_repo(&self, repo_path: &str) -> Option<&GitProfile> {
// let profile_name = self.repo_profiles.get(repo_path)?;
// self.profiles.get(profile_name)
// }
// /// Set profile for a repository
// pub fn set_profile_for_repo(&mut self, repo_path: String, profile_name: String) -> Result<()> {
// if !self.profiles.contains_key(&profile_name) {
// anyhow::bail!("Profile '{}' does not exist", profile_name);
// }
// self.repo_profiles.insert(repo_path, profile_name);
// Ok(())
// }
} }
/// Encrypted PAT data for export /// Encrypted PAT data for export

View File

@@ -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;
@@ -80,25 +80,25 @@ impl GitProfile {
if self.user_name.is_empty() { if self.user_name.is_empty() {
bail!("User name cannot be empty"); bail!("User name cannot be empty");
} }
if self.user_email.is_empty() { if self.user_email.is_empty() {
bail!("User email cannot be empty"); bail!("User email cannot be empty");
} }
crate::utils::validators::validate_email(&self.user_email)?; crate::utils::validators::validate_email(&self.user_email)?;
if let Some(ref ssh) = self.ssh { if let Some(ref ssh) = self.ssh {
ssh.validate()?; ssh.validate()?;
} }
if let Some(ref gpg) = self.gpg { if let Some(ref gpg) = self.gpg {
gpg.validate()?; gpg.validate()?;
} }
for token in self.tokens.values() { for token in self.tokens.values() {
token.validate()?; token.validate()?;
} }
Ok(()) Ok(())
} }
@@ -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);
@@ -144,7 +148,7 @@ impl GitProfile {
pub fn record_usage(&mut self, repo_path: Option<String>) { pub fn record_usage(&mut self, repo_path: Option<String>) {
self.usage.last_used = Some(chrono::Utc::now().to_rfc3339()); self.usage.last_used = Some(chrono::Utc::now().to_rfc3339());
self.usage.total_uses += 1; self.usage.total_uses += 1;
if let Some(repo) = repo_path { if let Some(repo) = repo_path {
let count = self.usage.repo_usage.entry(repo).or_insert(0); let count = self.usage.repo_usage.entry(repo).or_insert(0);
*count += 1; *count += 1;
@@ -159,93 +163,137 @@ impl GitProfile {
/// Apply this profile to a git repository (local config) /// Apply this profile to a git repository (local config)
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.settings.auto_sign_commits {
config.set_bool("commit.gpgsign", true)?;
}
if self.settings.auto_sign_tags {
config.set_bool("tag.gpgsign", true)?;
}
} }
if self.signing_key().is_none() {
if let Some(ref ssh) = self.ssh { let _ = config.remove("user.signingkey");
if let Some(ref key_path) = ssh.private_key_path { let _ = config.remove("commit.gpgsign");
let path_str = key_path.display().to_string(); let _ = config.remove("tag.gpgsign");
#[cfg(target_os = "windows")] }
{ if self.gpg.is_none() {
config.set_str("core.sshCommand", let _ = config.remove("gpg.program");
&format!("ssh -i \"{}\"", path_str.replace('\\', "/")))?; }
// 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)?;
} }
#[cfg(not(target_os = "windows"))] if self.settings.auto_sign_tags {
{ config.set_bool("tag.gpgsign", true)?;
config.set_str("core.sshCommand", }
&format!("ssh -i '{}'", path_str))?; 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.settings.auto_sign_commits {
config.set_bool("commit.gpgsign", true)?;
}
if self.settings.auto_sign_tags {
config.set_bool("tag.gpgsign", true)?;
}
} }
if self.signing_key().is_none() {
if let Some(ref ssh) = self.ssh { let _ = config.remove("user.signingkey");
if let Some(ref key_path) = ssh.private_key_path { let _ = config.remove("commit.gpgsign");
let path_str = key_path.display().to_string(); let _ = config.remove("tag.gpgsign");
#[cfg(target_os = "windows")] }
{ if self.gpg.is_none() {
config.set_str("core.sshCommand", let _ = config.remove("gpg.program");
&format!("ssh -i \"{}\"", path_str.replace('\\', "/")))?; }
// 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)?;
} }
#[cfg(not(target_os = "windows"))] if self.settings.auto_sign_tags {
{ config.set_bool("tag.gpgsign", true)?;
config.set_str("core.sshCommand", }
&format!("ssh -i '{}'", path_str))?; 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
} }
/// Compare with current git configuration /// Compare with current git configuration
pub fn compare_with_git_config(&self, repo: &git2::Repository) -> Result<ProfileComparison> { pub fn compare_with_git_config(&self, repo: &git2::Repository) -> Result<ProfileComparison> {
let config = repo.config()?; let config = repo.config()?;
let git_user_name = config.get_string("user.name").ok(); let git_user_name = config.get_string("user.name").ok();
let git_user_email = config.get_string("user.email").ok(); let git_user_email = config.get_string("user.email").ok();
let git_signing_key = config.get_string("user.signingkey").ok(); let git_signing_key = config.get_string("user.signingkey").ok();
let mut comparison = ProfileComparison { let mut comparison = ProfileComparison {
profile_name: self.name.clone(), profile_name: self.name.clone(),
matches: true, matches: true,
differences: vec![], differences: vec![],
}; };
if git_user_name.as_deref() != Some(&self.user_name) { if git_user_name.as_deref() != Some(&self.user_name) {
comparison.matches = false; comparison.matches = false;
comparison.differences.push(ConfigDifference { comparison.differences.push(ConfigDifference {
@@ -254,7 +302,7 @@ impl GitProfile {
git_value: git_user_name.unwrap_or_else(|| "<not set>".to_string()), git_value: git_user_name.unwrap_or_else(|| "<not set>".to_string()),
}); });
} }
if git_user_email.as_deref() != Some(&self.user_email) { if git_user_email.as_deref() != Some(&self.user_email) {
comparison.matches = false; comparison.matches = false;
comparison.differences.push(ConfigDifference { comparison.differences.push(ConfigDifference {
@@ -263,24 +311,24 @@ impl GitProfile {
git_value: git_user_email.unwrap_or_else(|| "<not set>".to_string()), git_value: git_user_email.unwrap_or_else(|| "<not set>".to_string()),
}); });
} }
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)
} }
} }
/// 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)]
@@ -307,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 {
@@ -349,37 +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 { }
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(); let path_str = key_path.display().to_string();
#[cfg(target_os = "windows")] #[cfg(target_os = "windows")]
{ {
Some(format!("ssh -i \"{}\"", path_str.replace('\\', "/"))) parts.push(format!("-i \"{}\"", path_str.replace('\\', "/")));
} }
#[cfg(not(target_os = "windows"))] #[cfg(not(target_os = "windows"))]
{ {
Some(format!("ssh -i '{}'", path_str)) parts.push(format!("-i '{}'", path_str));
} }
} else { }
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(" "))
} }
} }
} }
@@ -495,7 +577,9 @@ impl TokenConfig {
/// 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,
@@ -503,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 {
@@ -638,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,
@@ -686,7 +770,7 @@ mod tests {
"".to_string(), "".to_string(),
"invalid-email".to_string(), "invalid-email".to_string(),
); );
assert!(profile.validate().is_err()); assert!(profile.validate().is_err());
} }

View File

@@ -1,5 +1,5 @@
use crate::config::{CommitFormat, Language};
use crate::config::manager::ConfigManager; 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::llm::{GeneratedCommit, LlmClient};
use anyhow::{Context, Result}; use anyhow::{Context, Result};
@@ -7,18 +7,54 @@ use anyhow::{Context, Result};
/// 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(manager: &ConfigManager) -> Result<Self> { pub async fn new(manager: &ConfigManager) -> Result<Self> {
let llm_client = LlmClient::from_config(manager).await?; Self::new_with_think(manager, false, None).await
}
/// Create new content generator with thinking override and optional commit template
pub async fn new_with_think(
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) {
eprintln!(
"Warning: Provider '{}' does not support thinking mode. \
Disabling thinking for this invocation.",
provider
);
thinking_enabled = false;
}
}
let llm_client = LlmClient::from_config_with_think(manager, thinking_enabled).await?;
if !llm_client.is_available().await { if !llm_client.is_available().await {
anyhow::bail!("LLM provider '{}' is not available", manager.llm_provider()); anyhow::bail!("LLM provider '{}' is not available", manager.llm_provider());
} }
Ok(Self { llm_client }) 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
@@ -36,8 +72,10 @@ impl ContentGenerator {
} 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,13 +85,14 @@ 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() {
anyhow::bail!("No staged changes to generate commit from"); anyhow::bail!("No staged changes to generate commit from");
} }
self.generate_commit_message(&diff, format, language).await self.generate_commit_message(&diff, format, language).await
} }
@@ -64,12 +103,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)
self.llm_client.generate_tag_message(version, &commit_messages, language).await .await
} }
/// Generate changelog entry /// Generate changelog entry
@@ -86,8 +125,10 @@ impl ContentGenerator {
(commit_type, c.subject().to_string()) (commit_type, c.subject().to_string())
}) })
.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
@@ -103,8 +144,9 @@ impl ContentGenerator {
} else { } else {
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
@@ -115,49 +157,53 @@ impl ContentGenerator {
language: Language, language: Language,
) -> Result<GeneratedCommit> { ) -> Result<GeneratedCommit> {
use dialoguer::Select; use dialoguer::Select;
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");
} }
// Show diff summary // Show diff summary
let files = repo.get_staged_files()?; let files = repo.get_staged_files()?;
println!("\nStaged files ({}):", files.len()); println!("\nStaged 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!("\nGenerating 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!("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", "✓ Accept and commit",
"🔄 Regenerate", "🔄 Regenerate",
"✏️ Edit", "✏️ Edit",
"❌ Cancel", "❌ Cancel",
]; ];
let selection = Select::new() let selection = Select::new()
.with_prompt("What would you like to do?") .with_prompt("What would you like to do?")
.items(&options) .items(&options)
.default(0) .default(0)
.interact()?; .interact()?;
match selection { match selection {
0 => return Ok(generated), 0 => return Ok(generated),
1 => { 1 => {
println!("Regenerating..."); println!("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())?;
@@ -171,7 +217,7 @@ impl ContentGenerator {
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 {
commit_type: parsed.commit_type.unwrap_or_else(|| "chore".to_string()), commit_type: parsed.commit_type.unwrap_or_else(|| "chore".to_string()),
scope: parsed.scope, scope: parsed.scope,
@@ -208,11 +254,15 @@ pub mod fallback {
let has_code = files.iter().any(|f| { let has_code = files.iter().any(|f| {
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()
let has_tests = files.iter().any(|f| f.contains("test") || f.contains("spec")); .any(|f| f.ends_with(".md") || f.contains("README"));
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()
} else if has_docs { } else if has_docs {

View File

@@ -95,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)
}
} }
} }
@@ -110,13 +108,13 @@ impl ChangelogGenerator {
commits: &[CommitInfo], commits: &[CommitInfo],
) -> Result<()> { ) -> Result<()> {
let entry = self.generate(version, date, commits)?; let entry = self.generate(version, date, commits)?;
let existing = if changelog_path.exists() { let existing = if changelog_path.exists() {
fs::read_to_string(changelog_path)? fs::read_to_string(changelog_path)?
} else { } else {
String::new() String::new()
}; };
let new_content = if existing.is_empty() { let new_content = if existing.is_empty() {
format!("{}{}", CHANGELOG_HEADER, entry) format!("{}{}", CHANGELOG_HEADER, entry)
} else if existing.starts_with(CHANGELOG_HEADER) { } else if existing.starts_with(CHANGELOG_HEADER) {
@@ -124,7 +122,7 @@ impl ChangelogGenerator {
} else if existing.starts_with("# Changelog") { } 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;
for (i, line) in lines.iter().enumerate() { for (i, line) in lines.iter().enumerate() {
if i == 0 && line.starts_with('#') { if i == 0 && line.starts_with('#') {
header_end = i + 1; header_end = i + 1;
@@ -134,18 +132,18 @@ impl ChangelogGenerator {
break; break;
} }
} }
let header = lines[..header_end].join("\n"); let header = lines[..header_end].join("\n");
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 { } else {
format!("{}{}", CHANGELOG_HEADER, entry) format!("{}{}", CHANGELOG_HEADER, entry)
}; };
fs::write(changelog_path, new_content) fs::write(changelog_path, new_content)
.with_context(|| format!("Failed to write changelog: {:?}", changelog_path))?; .with_context(|| format!("Failed to write changelog: {:?}", changelog_path))?;
Ok(()) Ok(())
} }
@@ -157,10 +155,10 @@ impl ChangelogGenerator {
) -> Result<String> { ) -> Result<String> {
let date_str = date.format("%Y-%m-%d").to_string(); let date_str = date.format("%Y-%m-%d").to_string();
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![
("Added", vec!["feat"]), ("Added", vec!["feat"]),
@@ -170,7 +168,7 @@ impl ChangelogGenerator {
("Fixed", vec!["fix"]), ("Fixed", vec!["fix"]),
("Security", vec!["security"]), ("Security", vec!["security"]),
]; ];
for (title, types) in &categories { for (title, types) in &categories {
let items: Vec<&CommitInfo> = commits let items: Vec<&CommitInfo> = commits
.iter() .iter()
@@ -182,7 +180,7 @@ impl ChangelogGenerator {
} }
}) })
.collect(); .collect();
if !items.is_empty() { if !items.is_empty() {
output.push_str(&format!("### {}\n\n", title)); output.push_str(&format!("### {}\n\n", title));
for commit in items { for commit in items {
@@ -192,13 +190,13 @@ impl ChangelogGenerator {
output.push('\n'); output.push('\n');
} }
} }
// Other changes // Other changes
let categorized: Vec<String> = categories let categorized: Vec<String> = categories
.iter() .iter()
.flat_map(|(_, types)| types.iter().map(|s| s.to_string())) .flat_map(|(_, types)| types.iter().map(|s| s.to_string()))
.collect(); .collect();
let other: Vec<&CommitInfo> = commits let other: Vec<&CommitInfo> = commits
.iter() .iter()
.filter(|c| { .filter(|c| {
@@ -209,7 +207,7 @@ impl ChangelogGenerator {
} }
}) })
.collect(); .collect();
if !other.is_empty() { if !other.is_empty() {
output.push_str("### Other\n\n"); output.push_str("### Other\n\n");
for commit in other { for commit in other {
@@ -224,30 +222,30 @@ impl ChangelogGenerator {
output.push('\n'); output.push('\n');
} }
} }
Ok(output) Ok(output)
} }
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![];
let mut fixes = vec![]; let mut fixes = vec![];
let mut docs = vec![]; let mut docs = vec![];
let mut other = vec![]; let mut other = vec![];
let mut breaking = vec![]; let mut breaking = vec![];
for commit in commits { for commit in commits {
if commit.message.contains("BREAKING CHANGE") { if commit.message.contains("BREAKING CHANGE") {
breaking.push(commit); breaking.push(commit);
} }
if let Some(ref t) = commit.commit_type() { if let Some(ref t) = commit.commit_type() {
match t.as_str() { match t.as_str() {
"feat" => features.push(commit), "feat" => features.push(commit),
@@ -259,7 +257,7 @@ impl ChangelogGenerator {
other.push(commit); other.push(commit);
} }
} }
if !breaking.is_empty() { if !breaking.is_empty() {
output.push_str("### ⚠ Breaking Changes\n\n"); output.push_str("### ⚠ Breaking Changes\n\n");
for commit in breaking { for commit in breaking {
@@ -267,7 +265,7 @@ impl ChangelogGenerator {
} }
output.push('\n'); output.push('\n');
} }
if !features.is_empty() { if !features.is_empty() {
output.push_str("### 🚀 Features\n\n"); output.push_str("### 🚀 Features\n\n");
for commit in features { for commit in features {
@@ -275,7 +273,7 @@ impl ChangelogGenerator {
} }
output.push('\n'); output.push('\n');
} }
if !fixes.is_empty() { if !fixes.is_empty() {
output.push_str("### 🐛 Bug Fixes\n\n"); output.push_str("### 🐛 Bug Fixes\n\n");
for commit in fixes { for commit in fixes {
@@ -283,7 +281,7 @@ impl ChangelogGenerator {
} }
output.push('\n'); output.push('\n');
} }
if !docs.is_empty() { if !docs.is_empty() {
output.push_str("### 📚 Documentation\n\n"); output.push_str("### 📚 Documentation\n\n");
for commit in docs { for commit in docs {
@@ -291,14 +289,14 @@ impl ChangelogGenerator {
} }
output.push('\n'); output.push('\n');
} }
if !other.is_empty() { if !other.is_empty() {
output.push_str("### Other Changes\n\n"); output.push_str("### Other Changes\n\n");
for commit in other { for commit in other {
output.push_str(&self.format_commit_github(commit)); output.push_str(&self.format_commit_github(commit));
} }
} }
Ok(output) Ok(output)
} }
@@ -312,7 +310,7 @@ impl ChangelogGenerator {
if !self.custom_categories.is_empty() { if !self.custom_categories.is_empty() {
let date_str = date.format("%Y-%m-%d").to_string(); let date_str = date.format("%Y-%m-%d").to_string();
let mut output = format!("## [{}] - {}\n\n", version, date_str); let mut output = format!("## [{}] - {}\n\n", version, date_str);
for category in &self.custom_categories { for category in &self.custom_categories {
let items: Vec<&CommitInfo> = commits let items: Vec<&CommitInfo> = commits
.iter() .iter()
@@ -324,7 +322,7 @@ impl ChangelogGenerator {
} }
}) })
.collect(); .collect();
if !items.is_empty() { if !items.is_empty() {
output.push_str(&format!("### {}\n\n", category.title)); output.push_str(&format!("### {}\n\n", category.title));
for commit in items { for commit in items {
@@ -334,7 +332,7 @@ impl ChangelogGenerator {
output.push('\n'); output.push('\n');
} }
} }
Ok(output) Ok(output)
} else { } else {
// Fall back to keep-a-changelog // Fall back to keep-a-changelog
@@ -344,30 +342,35 @@ impl ChangelogGenerator {
fn format_commit(&self, commit: &CommitInfo) -> String { fn format_commit(&self, commit: &CommitInfo) -> String {
let mut line = format!("- {}", commit.subject()); let mut line = format!("- {}", commit.subject());
if self.include_hashes { if self.include_hashes {
line.push_str(&format!(" ({})", &commit.short_id)); line.push_str(&format!(" ({})", &commit.short_id));
} }
if self.include_authors { if self.include_authors {
line.push_str(&format!(" - @{}", commit.author)); line.push_str(&format!(" - @{}", commit.author));
} }
line line
} }
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>> {
let mut groups: HashMap<String, Vec<&'a CommitInfo>> = HashMap::new(); let mut groups: HashMap<String, Vec<&'a CommitInfo>> = HashMap::new();
for commit in commits { for commit in commits {
let commit_type = commit.commit_type().unwrap_or_else(|| "other".to_string()); let commit_type = commit.commit_type().unwrap_or_else(|| "other".to_string());
groups.entry(commit_type).or_default().push(commit); groups.entry(commit_type).or_default().push(commit);
} }
groups groups
} }
} }
@@ -380,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
@@ -389,10 +391,10 @@ pub fn init_changelog(path: &Path) -> Result<()> {
if path.exists() { if path.exists() {
anyhow::bail!("Changelog already exists at {:?}", path); anyhow::bail!("Changelog already exists at {:?}", path);
} }
fs::write(path, CHANGELOG_HEADER) fs::write(path, CHANGELOG_HEADER)
.with_context(|| format!("Failed to create changelog: {:?}", path))?; .with_context(|| format!("Failed to create changelog: {:?}", path))?;
Ok(()) Ok(())
} }
@@ -403,21 +405,17 @@ pub fn generate_from_history(
to_ref: Option<&str>, to_ref: Option<&str>,
) -> Result<Vec<CommitInfo>> { ) -> Result<Vec<CommitInfo>> {
let to_ref = to_ref.unwrap_or("HEAD"); let to_ref = to_ref.unwrap_or("HEAD");
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)
} }
@@ -425,30 +423,29 @@ pub fn update_version_links(
/// Parse changelog to extract versions /// Parse changelog to extract versions
pub fn parse_versions(changelog: &str) -> Vec<(String, String)> { 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()));
}
} }
} }
} }
versions versions
} }
/// Get unreleased changes /// Get unreleased changes
pub fn get_unreleased_changes(repo: &GitRepo) -> Result<Vec<CommitInfo>> { pub fn get_unreleased_changes(repo: &GitRepo) -> Result<Vec<CommitInfo>> {
let tags = repo.get_tags()?; let tags = repo.get_tags()?;
if let Some(latest_tag) = tags.first() { if let Some(latest_tag) = tags.first() {
repo.get_commits_between(&latest_tag.name, "HEAD") repo.get_commits_between(&latest_tag.name, "HEAD")
} else { } else {
@@ -479,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");
}
}

View File

@@ -1,5 +1,5 @@
use super::GitRepo; use super::GitRepo;
use anyhow::{bail, Result}; use anyhow::{Result, bail};
use chrono::Local; use chrono::Local;
/// Commit builder for creating commits /// Commit builder for creating commits
@@ -119,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 {
@@ -166,45 +170,46 @@ impl CommitBuilder {
fn amend_commit(&self, repo: &GitRepo, message: &str) -> Result<()> { fn amend_commit(&self, repo: &GitRepo, message: &str) -> Result<()> {
use std::process::Command; use std::process::Command;
let mut args = vec!["commit", "--amend"]; let mut args = vec!["commit", "--amend"];
if self.no_verify { if self.no_verify {
args.push("--no-verify"); args.push("--no-verify");
} }
args.push("-m"); args.push("-m");
args.push(message); args.push(message);
if self.sign { if self.sign {
args.push("-S"); args.push("-S");
} }
let output = Command::new("git") let output = Command::new("git")
.args(&args) .args(&args)
.current_dir(repo.path()) .current_dir(repo.path())
.output()?; .output()?;
if !output.status.success() { if !output.status.success() {
let stdout = String::from_utf8_lossy(&output.stdout); let stdout = String::from_utf8_lossy(&output.stdout);
let stderr = String::from_utf8_lossy(&output.stderr); let stderr = String::from_utf8_lossy(&output.stderr);
let error_msg = if stderr.is_empty() { let error_msg = if stderr.is_empty() {
if stdout.is_empty() { if stdout.is_empty() {
"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()
} }
} else { } else {
stderr.to_string() stderr.to_string()
}; };
bail!("Failed to amend commit: {}", error_msg); bail!("Failed to amend commit: {}", error_msg);
} }
Ok(()) Ok(())
} }
} }
@@ -219,7 +224,7 @@ impl Default for CommitBuilder {
pub fn create_date_commit_message(prefix: Option<&str>) -> String { pub fn create_date_commit_message(prefix: Option<&str>) -> String {
let now = Local::now(); let now = Local::now();
let date_str = now.format("%Y-%m-%d").to_string(); let date_str = now.format("%Y-%m-%d").to_string();
match prefix { match prefix {
Some(p) => format!("{}: {}", p, date_str), Some(p) => format!("{}: {}", p, date_str),
None => format!("chore: update {}", date_str), None => format!("chore: update {}", date_str),
@@ -229,58 +234,65 @@ pub fn create_date_commit_message(prefix: Option<&str>) -> String {
/// Commit type suggestions based on diff /// Commit type suggestions based on diff
pub fn suggest_commit_type(diff: &str) -> Vec<&'static str> { pub fn suggest_commit_type(diff: &str) -> Vec<&'static str> {
let mut suggestions = vec![]; let mut suggestions = vec![];
// Check for test files // Check for test files
if diff.contains("test") || diff.contains("spec") || diff.contains("__tests__") { if diff.contains("test") || diff.contains("spec") || diff.contains("__tests__") {
suggestions.push("test"); suggestions.push("test");
} }
// Check for documentation // Check for documentation
if diff.contains("README") || diff.contains(".md") || diff.contains("docs/") { if diff.contains("README") || diff.contains(".md") || diff.contains("docs/") {
suggestions.push("docs"); suggestions.push("docs");
} }
// 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");
} }
// Check for CI // Check for CI
if diff.contains(".github/") || diff.contains(".gitlab-") || diff.contains("Jenkinsfile") { if diff.contains(".github/") || diff.contains(".gitlab-") || diff.contains("Jenkinsfile") {
suggestions.push("ci"); suggestions.push("ci");
} }
// Default suggestions // Default suggestions
if suggestions.is_empty() { if suggestions.is_empty() {
suggestions.extend(&["feat", "fix", "refactor"]); suggestions.extend(&["feat", "fix", "refactor"]);
} }
suggestions suggestions
} }
/// Parse existing commit message /// Parse existing commit message
pub fn parse_commit_message(message: &str) -> ParsedCommit { pub fn parse_commit_message(message: &str) -> ParsedCommit {
let lines: Vec<&str> = message.lines().collect(); let lines: Vec<&str> = message.lines().collect();
if lines.is_empty() { if lines.is_empty() {
return ParsedCommit::default(); return ParsedCommit::default();
} }
let first_line = lines[0]; let first_line = lines[0];
// Try to parse as conventional commit // Try to parse as conventional commit
if let Some(colon_pos) = first_line.find(':') { if let Some(colon_pos) = first_line.find(':') {
let type_part = &first_line[..colon_pos]; let type_part = &first_line[..colon_pos];
let description = first_line[colon_pos + 1..].trim(); let description = first_line[colon_pos + 1..].trim();
let breaking = type_part.ends_with('!'); let breaking = type_part.ends_with('!');
let type_part = type_part.trim_end_matches('!'); let type_part = type_part.trim_end_matches('!');
let (commit_type, scope) = if let Some(open) = type_part.find('(') { let (commit_type, scope) = if let Some(open) = type_part.find('(') {
if let Some(close) = type_part.find(')') { if let Some(close) = type_part.find(')') {
let t = &type_part[..open]; let t = &type_part[..open];
@@ -292,42 +304,51 @@ pub fn parse_commit_message(message: &str) -> ParsedCommit {
} else { } else {
(Some(type_part.to_string()), None) (Some(type_part.to_string()), None)
}; };
// Extract body and footer // Extract body and footer
let mut body_lines = vec![]; let mut body_lines = vec![];
let mut footer_lines = vec![]; let mut footer_lines = vec![];
let mut in_footer = false; let mut in_footer = false;
for line in &lines[1..] { for line in &lines[1..] {
if line.trim().is_empty() { if line.trim().is_empty() {
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;
} }
if in_footer { if in_footer {
footer_lines.push(line.to_string()); footer_lines.push(line.to_string());
} else { } else {
body_lines.push(line.to_string()); body_lines.push(line.to_string());
} }
} }
return ParsedCommit { return 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,
}; };
} }
// Non-conventional commit // Non-conventional commit
ParsedCommit { ParsedCommit {
description: Some(first_line.to_string()), description: Some(first_line.to_string()),
@@ -351,7 +372,7 @@ impl ParsedCommit {
pub fn to_message(&self, format: crate::config::CommitFormat) -> String { pub fn to_message(&self, format: crate::config::CommitFormat) -> String {
let commit_type = self.commit_type.as_deref().unwrap_or("chore"); let commit_type = self.commit_type.as_deref().unwrap_or("chore");
let description = self.description.as_deref().unwrap_or("update"); let description = self.description.as_deref().unwrap_or("update");
match format { match format {
crate::config::CommitFormat::Conventional => { crate::config::CommitFormat::Conventional => {
crate::utils::formatter::format_conventional_commit( crate::utils::formatter::format_conventional_commit(

File diff suppressed because it is too large Load Diff

View File

@@ -1,6 +1,7 @@
use super::GitRepo; use super::GitRepo;
use anyhow::{bail, 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 {
@@ -105,10 +106,10 @@ impl TagBuilder {
/// Execute and push tag /// Execute and push tag
pub fn execute_and_push(&self, repo: &GitRepo, remote: &str) -> Result<()> { pub fn execute_and_push(&self, repo: &GitRepo, remote: &str) -> Result<()> {
self.execute(repo)?; self.execute(repo)?;
let name = self.name.as_ref().unwrap(); let name = self.name.as_ref().unwrap();
repo.push(remote, &format!("refs/tags/{}", name))?; repo.push(remote, &format!("refs/tags/{}", name))?;
Ok(()) Ok(())
} }
} }
@@ -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
),
} }
} }
@@ -149,7 +153,7 @@ impl VersionBump {
/// Get latest version tag from repository /// Get latest version tag from repository
pub fn get_latest_version(repo: &GitRepo, prefix: &str) -> Result<Option<Version>> { pub fn get_latest_version(repo: &GitRepo, prefix: &str) -> Result<Option<Version>> {
let tags = repo.get_tags()?; let tags = repo.get_tags()?;
let mut versions: Vec<Version> = tags let mut versions: Vec<Version> = tags
.iter() .iter()
.filter_map(|t| { .filter_map(|t| {
@@ -158,9 +162,9 @@ pub fn get_latest_version(repo: &GitRepo, prefix: &str) -> Result<Option<Version
Version::parse(version_str).ok() Version::parse(version_str).ok()
}) })
.collect(); .collect();
versions.sort_by(|a, b| b.cmp(a)); // Descending order versions.sort_by(|a, b| b.cmp(a)); // Descending order
Ok(versions.into_iter().next()) Ok(versions.into_iter().next())
} }
@@ -183,14 +187,17 @@ pub fn suggest_version_bump(commits: &[super::CommitInfo]) -> VersionBump {
let mut has_breaking = false; let mut has_breaking = false;
let mut has_feature = false; let mut has_feature = false;
let mut has_fix = false; let mut has_fix = false;
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;
} }
if let Some(commit_type) = commit.commit_type() { if let Some(commit_type) = commit.commit_type() {
match commit_type.as_str() { match commit_type.as_str() {
"feat" => has_feature = true, "feat" => has_feature = true,
@@ -199,7 +206,7 @@ pub fn suggest_version_bump(commits: &[super::CommitInfo]) -> VersionBump {
} }
} }
} }
if has_breaking { if has_breaking {
VersionBump::Major VersionBump::Major
} else if has_feature { } else if has_feature {
@@ -214,20 +221,20 @@ pub fn suggest_version_bump(commits: &[super::CommitInfo]) -> VersionBump {
/// Generate tag message from commits /// Generate tag message from commits
pub fn generate_tag_message(version: &str, commits: &[super::CommitInfo]) -> String { pub fn generate_tag_message(version: &str, commits: &[super::CommitInfo]) -> String {
let mut message = format!("Release {}\n\n", version); let mut message = format!("Release {}\n\n", version);
// Group commits by type // Group commits by type
let mut features = vec![]; let mut features = vec![];
let mut fixes = vec![]; let mut fixes = vec![];
let mut other = vec![]; let mut other = vec![];
let mut breaking = vec![]; let mut breaking = vec![];
for commit in commits { for commit in commits {
let subject = commit.subject(); let subject = commit.subject();
if commit.message.contains("BREAKING CHANGE") { if commit.message.contains("BREAKING CHANGE") {
breaking.push(subject.to_string()); breaking.push(subject.to_string());
} }
if let Some(commit_type) = commit.commit_type() { if let Some(commit_type) = commit.commit_type() {
match commit_type.as_str() { match commit_type.as_str() {
"feat" => features.push(subject.to_string()), "feat" => features.push(subject.to_string()),
@@ -238,7 +245,7 @@ pub fn generate_tag_message(version: &str, commits: &[super::CommitInfo]) -> Str
other.push(subject.to_string()); other.push(subject.to_string());
} }
} }
// Build message // Build message
if !breaking.is_empty() { if !breaking.is_empty() {
message.push_str("## Breaking Changes\n\n"); message.push_str("## Breaking Changes\n\n");
@@ -247,7 +254,7 @@ pub fn generate_tag_message(version: &str, commits: &[super::CommitInfo]) -> Str
} }
message.push('\n'); message.push('\n');
} }
if !features.is_empty() { if !features.is_empty() {
message.push_str("## Features\n\n"); message.push_str("## Features\n\n");
for item in &features { for item in &features {
@@ -255,7 +262,7 @@ pub fn generate_tag_message(version: &str, commits: &[super::CommitInfo]) -> Str
} }
message.push('\n'); message.push('\n');
} }
if !fixes.is_empty() { if !fixes.is_empty() {
message.push_str("## Bug Fixes\n\n"); message.push_str("## Bug Fixes\n\n");
for item in &fixes { for item in &fixes {
@@ -263,36 +270,36 @@ pub fn generate_tag_message(version: &str, commits: &[super::CommitInfo]) -> Str
} }
message.push('\n'); message.push('\n');
} }
if !other.is_empty() { if !other.is_empty() {
message.push_str("## Other Changes\n\n"); message.push_str("## Other Changes\n\n");
for item in &other { for item in &other {
message.push_str(&format!("- {}\n", item)); message.push_str(&format!("- {}\n", item));
} }
} }
message message
} }
/// Tag deletion helper /// Tag deletion helper
pub fn delete_tag(repo: &GitRepo, name: &str, remote: Option<&str>) -> Result<()> { pub fn delete_tag(repo: &GitRepo, name: &str, remote: Option<&str>) -> Result<()> {
repo.delete_tag(name)?; repo.delete_tag(name)?;
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 refspec = format!(":refs/tags/{}", name);
let output = Command::new("git") let output = Command::new("git")
.args(&["push", remote, &refspec]) .args(["push", remote, &refspec])
.current_dir(repo.path()) .current_dir(repo.path())
.output()?; .output()?;
if !output.status.success() { if !output.status.success() {
let stderr = String::from_utf8_lossy(&output.stderr); let stderr = String::from_utf8_lossy(&output.stderr);
bail!("Failed to delete remote tag: {}", stderr); bail!("Failed to delete remote tag: {}", stderr);
} }
} }
Ok(()) Ok(())
} }
@@ -303,7 +310,7 @@ pub fn list_tags(
limit: Option<usize>, limit: Option<usize>,
) -> Result<Vec<super::TagInfo>> { ) -> Result<Vec<super::TagInfo>> {
let tags = repo.get_tags()?; let tags = repo.get_tags()?;
let filtered: Vec<_> = tags let filtered: Vec<_> = tags
.into_iter() .into_iter()
.filter(|t| { .filter(|t| {
@@ -314,10 +321,200 @@ pub fn list_tags(
} }
}) })
.collect(); .collect();
if let Some(limit) = limit { if let Some(limit) = limit {
Ok(filtered.into_iter().take(limit).collect()) Ok(filtered.into_iter().take(limit).collect())
} else { } else {
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
}
});
}
/// Read the project version from Cargo.toml or pyproject.toml.
/// Returns None if no project config file is found or version cannot be read.
pub fn read_project_version(project_dir: &Path) -> Option<Version> {
// Try Cargo.toml first
let cargo_path = project_dir.join("Cargo.toml");
if cargo_path.exists() {
if let Ok(content) = std::fs::read_to_string(&cargo_path) {
if let Ok(value) = content.parse::<toml::Value>() {
if let Some(version) = value
.get("package")
.and_then(|p| p.get("version"))
.and_then(|v| v.as_str())
{
if let Ok(ver) = Version::parse(version) {
return Some(ver);
}
}
}
}
}
// Try pyproject.toml
let pyproject_path = project_dir.join("pyproject.toml");
if pyproject_path.exists() {
if let Ok(content) = std::fs::read_to_string(&pyproject_path) {
if let Ok(value) = content.parse::<toml::Value>() {
if let Some(version) = value
.get("project")
.and_then(|p| p.get("version"))
.and_then(|v| v.as_str())
{
if let Ok(ver) = Version::parse(version) {
return Some(ver);
}
}
}
}
}
None
}
#[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_project_version_from_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 version = read_project_version(dir.path());
assert_eq!(version, Some(Version::new(0, 3, 0)));
}
#[test]
fn test_read_project_version_from_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 version = read_project_version(dir.path());
assert_eq!(version, Some(Version::new(0, 2, 1)));
}
#[test]
fn test_read_project_version_no_config_files() {
let dir = tempfile::TempDir::new().unwrap();
let version = read_project_version(dir.path());
assert_eq!(version, None);
}
#[test]
fn test_read_project_version_cargo_takes_priority() {
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 version = read_project_version(dir.path());
assert_eq!(version, Some(Version::new(0, 3, 0)));
}
}

View File

@@ -267,7 +267,9 @@ impl Messages {
Language::Chinese => "没有可提交的更改。工作树是干净的。", Language::Chinese => "没有可提交的更改。工作树是干净的。",
Language::Japanese => "コミットする変更がありません。作業ツリーはクリーンです。", Language::Japanese => "コミットする変更がありません。作業ツリーはクリーンです。",
Language::Korean => "커밋할 변경 사항이 없습니다. 작업 트리가 깨끗합니다.", Language::Korean => "커밋할 변경 사항이 없습니다. 작업 트리가 깨끗합니다.",
Language::Spanish => "No hay cambios para hacer commit. El árbol de trabajo está limpio.", Language::Spanish => {
"No hay cambios para hacer commit. El árbol de trabajo está limpio."
}
Language::French => "Aucun changement à commiter. L'arbre de travail est propre.", Language::French => "Aucun changement à commiter. L'arbre de travail est propre.",
Language::German => "Keine Änderungen zum Committen. Arbeitsbaum ist sauber.", Language::German => "Keine Änderungen zum Committen. Arbeitsbaum ist sauber.",
} }
@@ -289,11 +291,19 @@ impl Messages {
match self.language { match self.language {
Language::English => "No files staged. Auto-staging all changes...", Language::English => "No files staged. Auto-staging all changes...",
Language::Chinese => "没有暂存文件。自动暂存所有更改...", Language::Chinese => "没有暂存文件。自动暂存所有更改...",
Language::Japanese => "ステージされたファイルがありません。すべての変更を自動ステージ中...", Language::Japanese => {
"ステージされたファイルがありません。すべての変更を自動ステージ中..."
}
Language::Korean => "스테이징된 파일이 없습니다. 모든 변경 사항을 자동 스테이징 중...", Language::Korean => "스테이징된 파일이 없습니다. 모든 변경 사항을 자동 스테이징 중...",
Language::Spanish => "No hay archivos preparados. Preparando automáticamente todos los cambios...", Language::Spanish => {
Language::French => "Aucun fichier indexé. Indexation automatique de tous les changements...", "No hay archivos preparados. Preparando automáticamente todos los cambios..."
Language::German => "Keine Dateien bereitgestellt. Alle Änderungen werden automatisch bereitgestellt...", }
Language::French => {
"Aucun fichier indexé. Indexation automatique de tous les changements..."
}
Language::German => {
"Keine Dateien bereitgestellt. Alle Änderungen werden automatisch bereitgestellt..."
}
} }
} }
@@ -359,12 +369,23 @@ impl Messages {
pub fn ai_generating_tag(&self, count: usize) -> String { pub fn ai_generating_tag(&self, count: usize) -> String {
match self.language { match self.language {
Language::English => format!("🤖 AI is generating tag message from {} commits...", count), Language::English => {
format!("🤖 AI is generating tag message from {} commits...", count)
}
Language::Chinese => format!("🤖 AI 正在从 {} 个提交生成标签消息...", count), Language::Chinese => format!("🤖 AI 正在从 {} 个提交生成标签消息...", count),
Language::Japanese => format!("🤖 AIが{}個のコミットからタグメッセージを生成しています...", count), Language::Japanese => format!(
"🤖 AIが{}個のコミットからタグメッセージを生成しています...",
count
),
Language::Korean => format!("🤖 AI가 {}개의 커밋에서 태그 메시지를 생성 중...", count), Language::Korean => format!("🤖 AI가 {}개의 커밋에서 태그 메시지를 생성 중...", count),
Language::Spanish => format!("🤖 La IA está generando mensaje de etiqueta desde {} commits...", count), Language::Spanish => format!(
Language::French => format!("🤖 L'IA génère le message dtiquette à partir de {} commits...", count), "🤖 La IA está generando mensaje de etiqueta desde {} commits...",
count
),
Language::French => format!(
"🤖 L'IA génère le message d'étiquette à partir de {} commits...",
count
),
Language::German => format!("🤖 KI generiert Tag-Nachricht aus {} Commits...", count), Language::German => format!("🤖 KI generiert Tag-Nachricht aus {} Commits...", count),
} }
} }

View File

@@ -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)
} }

9
src/lib.rs Normal file
View File

@@ -0,0 +1,9 @@
#![allow(dead_code)]
pub mod commands;
pub mod config;
pub mod generator;
pub mod git;
pub mod i18n;
pub mod llm;
pub mod utils;

View File

@@ -1,7 +1,9 @@
use super::{create_http_client, LlmProvider}; use super::thinking::ThinkingStateManager;
use anyhow::{bail, Context, Result}; use super::{LlmProvider, create_http_client};
use anyhow::{Context, Result, bail};
use async_trait::async_trait; use async_trait::async_trait;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
/// Anthropic Claude API client /// Anthropic Claude API client
@@ -9,6 +11,12 @@ pub struct AnthropicClient {
api_key: String, api_key: String,
model: String, model: String,
client: reqwest::Client, client: reqwest::Client,
thinking_enabled: bool,
thinking_budget_tokens: u32,
max_tokens: u32,
temperature: f32,
top_p: Option<f32>,
thinking_state: Option<Arc<ThinkingStateManager>>,
} }
#[derive(Debug, Serialize)] #[derive(Debug, Serialize)]
@@ -17,24 +25,59 @@ struct MessagesRequest {
max_tokens: u32, max_tokens: u32,
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
temperature: Option<f32>, temperature: Option<f32>,
#[serde(skip_serializing_if = "Option::is_none")]
top_p: Option<f32>,
messages: Vec<AnthropicMessage>, messages: Vec<AnthropicMessage>,
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
system: Option<String>, system: Option<Vec<SystemContent>>,
#[serde(skip_serializing_if = "Option::is_none")]
thinking: Option<ThinkingConfig>,
stream: bool,
} }
#[derive(Debug, Serialize, Deserialize)] #[derive(Debug, Serialize, Clone)]
struct SystemContent {
#[serde(rename = "type")]
content_type: String,
text: String,
}
#[derive(Debug, Serialize)]
struct ThinkingConfig {
#[serde(rename = "type")]
thinking_type: String,
#[serde(skip_serializing_if = "Option::is_none")]
budget_tokens: Option<u32>,
}
#[derive(Debug, Serialize, Deserialize, Clone)]
struct AnthropicMessage { struct AnthropicMessage {
role: String, role: String,
content: String, content: AnthropicContent,
}
#[derive(Debug, Serialize, Deserialize, Clone)]
#[serde(untagged)]
enum AnthropicContent {
Text(String),
Blocks(Vec<ContentBlock>),
}
#[derive(Debug, Serialize, Deserialize, Clone)]
struct ContentBlock {
#[serde(rename = "type")]
content_type: String,
#[serde(skip_serializing_if = "Option::is_none")]
text: Option<String>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct MessagesResponse { struct MessagesResponse {
content: Vec<ContentBlock>, content: Vec<ResponseContentBlock>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct ContentBlock { struct ResponseContentBlock {
#[serde(rename = "type")] #[serde(rename = "type")]
content_type: String, content_type: String,
text: String, text: String,
@@ -52,31 +95,112 @@ struct AnthropicError {
message: String, message: String,
} }
// --- Streaming SSE event structures ---
#[derive(Debug, Deserialize)]
struct SseEvent {
#[serde(rename = "type")]
event_type: String,
#[serde(default)]
message: Option<SseMessage>,
#[serde(default)]
index: Option<u32>,
#[serde(default)]
content_block: Option<SseContentBlock>,
#[serde(default)]
delta: Option<SseDelta>,
#[serde(default)]
usage: Option<SseUsage>,
}
#[derive(Debug, Deserialize)]
struct SseMessage {
#[serde(default)]
content: Option<Vec<SseContentBlock>>,
}
#[derive(Debug, Deserialize)]
struct SseContentBlock {
#[serde(rename = "type")]
content_type: String,
#[serde(default)]
thinking: Option<String>,
#[serde(default)]
text: Option<String>,
}
#[derive(Debug, Deserialize)]
struct SseDelta {
#[serde(rename = "type")]
delta_type: Option<String>,
#[serde(default)]
thinking: Option<String>,
#[serde(default)]
text: Option<String>,
}
#[derive(Debug, Deserialize)]
struct SseUsage {
#[serde(default)]
output_tokens: Option<u32>,
}
impl AnthropicClient { impl AnthropicClient {
/// Create new Anthropic client
pub fn new(api_key: &str, model: &str) -> Result<Self> { pub fn new(api_key: &str, model: &str) -> Result<Self> {
let client = create_http_client(Duration::from_secs(60))?; let client = create_http_client(Duration::from_secs(60))?;
Ok(Self { Ok(Self {
api_key: api_key.to_string(), api_key: api_key.to_string(),
model: model.to_string(), model: model.to_string(),
client, client,
thinking_enabled: false,
thinking_budget_tokens: 1024,
max_tokens: 500,
temperature: 0.7,
top_p: None,
thinking_state: None,
}) })
} }
/// Set timeout
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> { pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
self.client = create_http_client(timeout)?; self.client = create_http_client(timeout)?;
Ok(self) Ok(self)
} }
/// List available models pub fn with_thinking(mut self, enabled: bool) -> Self {
self.thinking_enabled = enabled;
self
}
pub fn with_thinking_budget_tokens(mut self, budget_tokens: u32) -> Self {
self.thinking_budget_tokens = budget_tokens;
self
}
pub fn with_max_tokens(mut self, max_tokens: u32) -> Self {
self.max_tokens = max_tokens;
self
}
pub fn with_temperature(mut self, temperature: f32) -> Self {
self.temperature = temperature;
self
}
pub fn with_top_p(mut self, top_p: f32) -> Self {
self.top_p = Some(top_p);
self
}
pub fn with_thinking_state(mut self, state: Arc<ThinkingStateManager>) -> Self {
self.thinking_state = Some(state);
self
}
pub async fn list_models(&self) -> Result<Vec<String>> { 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()) Ok(ANTHROPIC_MODELS.iter().map(|&m| m.to_string()).collect())
} }
/// Validate API key
pub async fn validate_key(&self) -> Result<bool> { pub async fn validate_key(&self) -> Result<bool> {
let url = "https://api.anthropic.com/v1/messages"; let url = "https://api.anthropic.com/v1/messages";
@@ -84,14 +208,18 @@ impl AnthropicClient {
model: self.model.clone(), model: self.model.clone(),
max_tokens: 5, max_tokens: 5,
temperature: Some(0.0), temperature: Some(0.0),
top_p: None,
messages: vec![AnthropicMessage { messages: vec![AnthropicMessage {
role: "user".to_string(), role: "user".to_string(),
content: "Hi".to_string(), content: AnthropicContent::Text("Hi".to_string()),
}], }],
system: None, system: None,
thinking: None,
stream: false,
}; };
let response = self.client let response = self
.client
.post(url) .post(url)
.header("x-api-key", &self.api_key) .header("x-api-key", &self.api_key)
.header("anthropic-version", "2023-06-01") .header("anthropic-version", "2023-06-01")
@@ -124,25 +252,28 @@ impl LlmProvider for AnthropicClient {
async fn generate(&self, prompt: &str) -> Result<String> { async fn generate(&self, prompt: &str) -> Result<String> {
let messages = vec![AnthropicMessage { let messages = vec![AnthropicMessage {
role: "user".to_string(), role: "user".to_string(),
content: prompt.to_string(), content: AnthropicContent::Text(prompt.to_string()),
}]; }];
self.messages_request(messages, None).await self.messages_request_with_retry(messages, None).await
} }
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> { async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
let messages = vec![AnthropicMessage { let messages = vec![AnthropicMessage {
role: "user".to_string(), role: "user".to_string(),
content: user.to_string(), content: AnthropicContent::Text(user.to_string()),
}]; }];
let system = if system.is_empty() { let system = if system.is_empty() {
None None
} else { } else {
Some(system.to_string()) Some(vec![SystemContent {
content_type: "text".to_string(),
text: system.to_string(),
}])
}; };
self.messages_request(messages, system).await self.messages_request_with_retry(messages, system).await
} }
async fn is_available(&self) -> bool { async fn is_available(&self) -> bool {
@@ -155,22 +286,84 @@ impl LlmProvider for AnthropicClient {
} }
impl AnthropicClient { impl AnthropicClient {
async fn messages_request_with_retry(
&self,
messages: Vec<AnthropicMessage>,
system: Option<Vec<SystemContent>>,
) -> Result<String> {
let mut last_error = None;
for attempt in 1..=3 {
match self
.messages_request(messages.clone(), system.clone())
.await
{
Ok(result) => return Ok(result),
Err(e) => {
let err_msg = e.to_string();
let is_retryable = err_msg.contains("timeout")
|| err_msg.contains("connection")
|| err_msg.contains("temporary")
|| err_msg.contains("5")
&& (err_msg.contains("500")
|| err_msg.contains("502")
|| err_msg.contains("503")
|| err_msg.contains("504"));
if !is_retryable || attempt == 3 {
last_error = Some(e);
break;
}
tokio::time::sleep(Duration::from_millis(500 * 2u64.pow(attempt - 1))).await;
}
}
}
Err(last_error.unwrap_or_else(|| anyhow::anyhow!("Request failed after retries")))
}
async fn messages_request( async fn messages_request(
&self, &self,
messages: Vec<AnthropicMessage>, messages: Vec<AnthropicMessage>,
system: Option<String>, system: Option<Vec<SystemContent>>,
) -> Result<String> {
if self.thinking_enabled {
self.streaming_messages_request(messages, system).await
} else {
self.non_streaming_messages_request(messages, system).await
}
}
async fn non_streaming_messages_request(
&self,
messages: Vec<AnthropicMessage>,
system: Option<Vec<SystemContent>>,
) -> Result<String> { ) -> Result<String> {
let url = "https://api.anthropic.com/v1/messages"; let url = "https://api.anthropic.com/v1/messages";
let temperature = if self.temperature == 0.0 {
None
} else {
Some(self.temperature)
};
let request = MessagesRequest { let request = MessagesRequest {
model: self.model.clone(), model: self.model.clone(),
max_tokens: 500, max_tokens: self.max_tokens,
temperature: Some(0.7), temperature,
top_p: self.top_p,
messages, messages,
system, system,
thinking: Some(ThinkingConfig {
thinking_type: "disabled".to_string(),
budget_tokens: None,
}),
stream: false,
}; };
let response = self.client let response = self
.client
.post(url) .post(url)
.header("x-api-key", &self.api_key) .header("x-api-key", &self.api_key)
.header("anthropic-version", "2023-06-01") .header("anthropic-version", "2023-06-01")
@@ -179,35 +372,205 @@ impl AnthropicClient {
.send() .send()
.await .await
.context("Failed to send request to Anthropic")?; .context("Failed to send request to Anthropic")?;
let status = response.status(); let status = response.status();
if !status.is_success() { if !status.is_success() {
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
// Try to parse error
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) { 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: {} ({})",
error.error.message,
error.error.error_type
);
} }
bail!("Anthropic API error: {} - {}", status, text); bail!("Anthropic API error: {} - {}", status, text);
} }
let result: MessagesResponse = response let result: MessagesResponse = response
.json() .json()
.await .await
.context("Failed to parse Anthropic response")?; .context("Failed to parse Anthropic response")?;
result.content result
.content
.into_iter() .into_iter()
.find(|c| c.content_type == "text") .find(|c| c.content_type == "text")
.map(|c| c.text.trim().to_string()) .map(|c| c.text.trim().to_string())
.filter(|s| !s.is_empty())
.ok_or_else(|| anyhow::anyhow!("No text response from Anthropic")) .ok_or_else(|| anyhow::anyhow!("No text response from Anthropic"))
} }
/// Streaming request for thinking mode, filters thinking content blocks
async fn streaming_messages_request(
&self,
messages: Vec<AnthropicMessage>,
system: Option<Vec<SystemContent>>,
) -> Result<String> {
let url = "https://api.anthropic.com/v1/messages";
let thinking = ThinkingConfig {
thinking_type: "enabled".to_string(),
budget_tokens: Some(self.thinking_budget_tokens),
};
// max_tokens must exceed budget_tokens
let max_tokens = (self.max_tokens).max(self.thinking_budget_tokens + 100);
let request = MessagesRequest {
model: self.model.clone(),
max_tokens,
temperature: None, // must be omitted for thinking mode
top_p: None,
messages,
system,
thinking: Some(thinking),
stream: true,
};
let response = self
.client
.post(url)
.header("x-api-key", &self.api_key)
.header("anthropic-version", "2023-06-01")
.header("Content-Type", "application/json")
.header("Accept", "text/event-stream")
.json(&request)
.send()
.await
.context("Failed to send streaming request to Anthropic")?;
let status = response.status();
if !status.is_success() {
let text = response.text().await.unwrap_or_default();
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 mut content_buffer = String::new();
let mut in_thinking = false;
let mut has_reasoning = false;
let mut has_content = false;
let thinking_state = self.thinking_state.as_ref();
let mut byte_stream = response.bytes_stream();
let mut line_buffer = String::new();
use futures_util::StreamExt;
while let Some(chunk) = byte_stream.next().await {
let chunk = chunk.context("Failed to read streaming response chunk")?;
let chunk_str =
String::from_utf8(chunk.to_vec()).context("Invalid UTF-8 in stream chunk")?;
line_buffer.push_str(&chunk_str);
while let Some(line_end) = line_buffer.find('\n') {
let line = line_buffer[..line_end].trim().to_string();
line_buffer = line_buffer[line_end + 1..].to_string();
if line.is_empty() {
continue;
}
// Parse SSE event line
if let Some(data) = line.strip_prefix("data: ") {
if let Ok(event) = serde_json::from_str::<SseEvent>(data) {
match event.event_type.as_str() {
"content_block_start" => {
if let Some(ref block) = event.content_block {
if block.content_type == "thinking" {
in_thinking = true;
if !has_reasoning {
has_reasoning = true;
if let Some(state) = thinking_state {
state.start_thinking();
}
}
}
}
}
"content_block_delta" => {
if let Some(ref delta) = event.delta {
// Thinking delta - ignore content but track state
if delta.thinking.is_some() {
continue;
}
// Text delta - collect
if in_thinking && delta.text.is_some() {
// Transition from thinking to text
if let Some(state) = thinking_state {
state.end_thinking();
}
in_thinking = false;
}
if let Some(ref text) = delta.text
&& !text.is_empty()
{
has_content = true;
content_buffer.push_str(text);
}
}
}
"content_block_stop" => {
if in_thinking {
if let Some(state) = thinking_state {
state.end_thinking();
}
in_thinking = false;
}
}
_ => {}
}
}
}
}
}
// Ensure thinking state is ended
if let Some(state) = thinking_state {
state.end_thinking();
}
let result = content_buffer.trim().to_string();
if result.is_empty() {
if has_reasoning && !has_content {
bail!(
"Anthropic returned thinking content but no final answer. \
The model may have entered an incomplete thinking state. \
Please try again or disable thinking mode."
);
}
bail!(
"No response from Anthropic. \
If thinking mode is enabled, try disabling it or ensure the model supports it."
);
}
Ok(result)
}
} }
/// Available Anthropic models /// Available Anthropic models (Claude 4 series with extended thinking)
pub const ANTHROPIC_MODELS: &[&str] = &[ pub const ANTHROPIC_MODELS: &[&str] = &[
"claude-opus-4-7",
"claude-sonnet-4-6",
"claude-haiku-4-5",
// Legacy models
"claude-3-opus-20240229", "claude-3-opus-20240229",
"claude-3-sonnet-20240229", "claude-3-sonnet-20240229",
"claude-3-haiku-20240307", "claude-3-haiku-20240307",
@@ -216,7 +579,6 @@ pub const ANTHROPIC_MODELS: &[&str] = &[
"claude-instant-1.2", "claude-instant-1.2",
]; ];
/// Check if a model name is valid
pub fn is_valid_model(model: &str) -> bool { pub fn is_valid_model(model: &str) -> bool {
ANTHROPIC_MODELS.contains(&model) ANTHROPIC_MODELS.contains(&model)
} }
@@ -226,8 +588,68 @@ mod tests {
use super::*; use super::*;
#[test] #[test]
fn test_model_validation() { fn test_model_validation_claude4() {
assert!(is_valid_model("claude-opus-4-7"));
assert!(is_valid_model("claude-sonnet-4-6"));
assert!(is_valid_model("claude-haiku-4-5"));
assert!(is_valid_model("claude-3-sonnet-20240229")); assert!(is_valid_model("claude-3-sonnet-20240229"));
assert!(!is_valid_model("invalid-model")); assert!(!is_valid_model("invalid-model"));
} }
#[test]
fn test_thinking_config_serialization() {
let config = ThinkingConfig {
thinking_type: "enabled".to_string(),
budget_tokens: Some(2048),
};
let json = serde_json::to_string(&config).unwrap();
assert!(json.contains(r#""type":"enabled""#));
assert!(json.contains(r#""budget_tokens":2048"#));
}
#[test]
fn test_thinking_config_disabled_serialization() {
let config = ThinkingConfig {
thinking_type: "disabled".to_string(),
budget_tokens: None,
};
let json = serde_json::to_string(&config).unwrap();
assert_eq!(json, r#"{"type":"disabled"}"#);
}
#[test]
fn test_system_content_serialization() {
let content = SystemContent {
content_type: "text".to_string(),
text: "You are helpful.".to_string(),
};
let json = serde_json::to_string(&content).unwrap();
assert!(json.contains(r#""type":"text""#));
}
#[test]
fn test_sse_event_parsing_content_block_start() {
let json = r#"{"type":"content_block_start","index":0,"content_block":{"type":"thinking","thinking":""}}"#;
let event: SseEvent = serde_json::from_str(json).unwrap();
assert_eq!(event.event_type, "content_block_start");
assert_eq!(event.content_block.unwrap().content_type, "thinking");
}
#[test]
fn test_sse_event_parsing_text_delta() {
let json = r#"{"type":"content_block_delta","index":0,"delta":{"type":"text_delta","text":"Hello"}}"#;
let event: SseEvent = serde_json::from_str(json).unwrap();
assert_eq!(event.event_type, "content_block_delta");
assert_eq!(event.delta.unwrap().text, Some("Hello".to_string()));
}
#[test]
fn test_anthropic_content_text() {
let msg = AnthropicMessage {
role: "user".to_string(),
content: AnthropicContent::Text("Hello".to_string()),
};
let json = serde_json::to_string(&msg).unwrap();
assert!(json.contains(r#""content":"Hello""#));
}
} }

View File

@@ -1,7 +1,9 @@
use super::{create_http_client, LlmProvider}; use super::thinking::ThinkingStateManager;
use anyhow::{bail, Context, Result}; use super::{LlmProvider, create_http_client};
use anyhow::{Context, Result, bail};
use async_trait::async_trait; use async_trait::async_trait;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
/// DeepSeek API client /// DeepSeek API client
@@ -10,6 +12,11 @@ pub struct DeepSeekClient {
api_key: String, api_key: String,
model: String, model: String,
client: reqwest::Client, client: reqwest::Client,
thinking_enabled: bool,
reasoning_effort: Option<String>,
max_tokens: u32,
temperature: f32,
thinking_state: Option<Arc<ThinkingStateManager>>,
} }
#[derive(Debug, Serialize)] #[derive(Debug, Serialize)]
@@ -20,13 +27,31 @@ struct ChatCompletionRequest {
max_tokens: Option<u32>, max_tokens: Option<u32>,
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
temperature: Option<f32>, temperature: Option<f32>,
#[serde(skip_serializing_if = "Option::is_none")]
top_p: Option<f32>,
#[serde(skip_serializing_if = "Option::is_none")]
presence_penalty: Option<f32>,
#[serde(skip_serializing_if = "Option::is_none")]
frequency_penalty: Option<f32>,
stream: bool, stream: bool,
#[serde(skip_serializing_if = "Option::is_none")]
thinking: Option<ThinkingConfig>,
#[serde(skip_serializing_if = "Option::is_none")]
reasoning_effort: Option<String>,
} }
#[derive(Debug, Serialize, Deserialize)] #[derive(Debug, Serialize)]
struct ThinkingConfig {
#[serde(rename = "type")]
thinking_type: String,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
struct Message { struct Message {
role: String, role: String,
content: String, content: String,
#[serde(skip_serializing_if = "Option::is_none")]
reasoning_content: Option<String>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
@@ -37,6 +62,31 @@ struct ChatCompletionResponse {
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct Choice { struct Choice {
message: Message, message: Message,
#[serde(default)]
reasoning_content: Option<String>,
}
// --- Streaming response structures ---
#[derive(Debug, Deserialize)]
struct StreamChunk {
choices: Vec<StreamChoice>,
}
#[derive(Debug, Deserialize)]
struct StreamChoice {
delta: StreamDelta,
#[serde(default)]
finish_reason: Option<String>,
index: Option<u32>,
}
#[derive(Debug, Deserialize, Default)]
struct StreamDelta {
#[serde(default)]
content: Option<String>,
#[serde(default)]
reasoning_content: Option<String>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
@@ -52,41 +102,73 @@ struct ApiError {
} }
impl DeepSeekClient { impl DeepSeekClient {
/// Create new DeepSeek client
pub fn new(api_key: &str, model: &str) -> Result<Self> { pub fn new(api_key: &str, model: &str) -> Result<Self> {
let client = create_http_client(Duration::from_secs(60))?; let client = create_http_client(Duration::from_secs(300))?;
Ok(Self { Ok(Self {
base_url: "https://api.deepseek.com/v1".to_string(), base_url: "https://api.deepseek.com".to_string(),
api_key: api_key.to_string(), api_key: api_key.to_string(),
model: model.to_string(), model: model.to_string(),
client, client,
thinking_enabled: false,
reasoning_effort: None,
max_tokens: 500,
temperature: 0.7,
thinking_state: None,
}) })
} }
/// Create with custom base URL
pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> { pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> {
let client = create_http_client(Duration::from_secs(60))?; let client = create_http_client(Duration::from_secs(300))?;
Ok(Self { Ok(Self {
base_url: base_url.trim_end_matches('/').to_string(), base_url: base_url.trim_end_matches('/').to_string(),
api_key: api_key.to_string(), api_key: api_key.to_string(),
model: model.to_string(), model: model.to_string(),
client, client,
thinking_enabled: false,
reasoning_effort: None,
max_tokens: 500,
temperature: 0.7,
thinking_state: None,
}) })
} }
/// Set timeout
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> { pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
self.client = create_http_client(timeout)?; self.client = create_http_client(timeout)?;
Ok(self) Ok(self)
} }
/// List available models pub fn with_thinking(mut self, enabled: bool) -> Self {
self.thinking_enabled = enabled;
self
}
pub fn with_reasoning_effort(mut self, effort: Option<String>) -> Self {
self.reasoning_effort = effort;
self
}
pub fn with_max_tokens(mut self, max_tokens: u32) -> Self {
self.max_tokens = max_tokens;
self
}
pub fn with_temperature(mut self, temperature: f32) -> Self {
self.temperature = temperature;
self
}
pub fn with_thinking_state(mut self, state: Arc<ThinkingStateManager>) -> Self {
self.thinking_state = Some(state);
self
}
pub async fn list_models(&self) -> Result<Vec<String>> { pub async fn list_models(&self) -> Result<Vec<String>> {
let url = format!("{}/models", self.base_url); let url = format!("{}/models", self.base_url);
let response = self.client let response = self
.client
.get(&url) .get(&url)
.header("Authorization", format!("Bearer {}", self.api_key)) .header("Authorization", format!("Bearer {}", self.api_key))
.send() .send()
@@ -101,11 +183,11 @@ impl DeepSeekClient {
#[derive(Deserialize)] #[derive(Deserialize)]
struct ModelsResponse { struct ModelsResponse {
data: Vec<Model>, data: Vec<ModelId>,
} }
#[derive(Deserialize)] #[derive(Deserialize)]
struct Model { struct ModelId {
id: String, id: String,
} }
@@ -117,7 +199,6 @@ impl DeepSeekClient {
Ok(result.data.into_iter().map(|m| m.id).collect()) Ok(result.data.into_iter().map(|m| m.id).collect())
} }
/// Validate API key
pub async fn validate_key(&self) -> Result<bool> { pub async fn validate_key(&self) -> Result<bool> {
match self.list_models().await { match self.list_models().await {
Ok(_) => Ok(true), Ok(_) => Ok(true),
@@ -136,32 +217,33 @@ impl DeepSeekClient {
#[async_trait] #[async_trait]
impl LlmProvider for DeepSeekClient { impl LlmProvider for DeepSeekClient {
async fn generate(&self, prompt: &str) -> Result<String> { async fn generate(&self, prompt: &str) -> Result<String> {
let messages = vec![ let messages = vec![Message {
Message { role: "user".to_string(),
role: "user".to_string(), content: prompt.to_string(),
content: prompt.to_string(), reasoning_content: None,
}, }];
];
self.chat_completion_with_retry(messages).await
self.chat_completion(messages).await
} }
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> { async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
let mut messages = vec![]; let mut messages = vec![];
if !system.is_empty() { if !system.is_empty() {
messages.push(Message { messages.push(Message {
role: "system".to_string(), role: "system".to_string(),
content: system.to_string(), content: system.to_string(),
reasoning_content: None,
}); });
} }
messages.push(Message { messages.push(Message {
role: "user".to_string(), role: "user".to_string(),
content: user.to_string(), content: user.to_string(),
reasoning_content: None,
}); });
self.chat_completion(messages).await self.chat_completion_with_retry(messages).await
} }
async fn is_available(&self) -> bool { async fn is_available(&self) -> bool {
@@ -174,59 +256,291 @@ impl LlmProvider for DeepSeekClient {
} }
impl DeepSeekClient { impl DeepSeekClient {
async fn chat_completion_with_retry(&self, messages: Vec<Message>) -> Result<String> {
let mut last_error = None;
for attempt in 1..=3 {
match self.chat_completion(messages.clone()).await {
Ok(result) => return Ok(result),
Err(e) => {
let err_msg = e.to_string();
// 网络临时错误才重试
let is_retryable = err_msg.contains("timeout")
|| err_msg.contains("connection")
|| err_msg.contains("temporary")
|| err_msg.contains("5")
&& (err_msg.contains("500")
|| err_msg.contains("502")
|| err_msg.contains("503")
|| err_msg.contains("504"));
if !is_retryable || attempt == 3 {
last_error = Some(e);
break;
}
// 指数退避
tokio::time::sleep(Duration::from_millis(500 * 2u64.pow(attempt - 1))).await;
}
}
}
Err(last_error.unwrap_or_else(|| anyhow::anyhow!("Request failed after retries")))
}
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> { async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
let url = format!("{}/chat/completions", self.base_url); let url = format!("{}/chat/completions", self.base_url);
let thinking = Some(ThinkingConfig {
thinking_type: if self.thinking_enabled {
"enabled".to_string()
} else {
"disabled".to_string()
},
});
// 思考模式下temperature/top_p 等参数不应传递
// 非思考模式下可以正常传递
let (temperature, max_tokens, top_p, presence_penalty, frequency_penalty) =
if self.thinking_enabled {
(None, Some(self.max_tokens), None, None, None)
} else {
(
Some(self.temperature),
Some(self.max_tokens),
None,
None,
None,
)
};
let reasoning_effort = if self.thinking_enabled {
self.reasoning_effort.clone()
} else {
None
};
let request = ChatCompletionRequest { let request = ChatCompletionRequest {
model: self.model.clone(), model: self.model.clone(),
messages, messages: messages.clone(),
max_tokens: Some(500), max_tokens,
temperature: Some(0.7), temperature,
stream: false, top_p,
presence_penalty,
frequency_penalty,
stream: self.thinking_enabled,
thinking,
reasoning_effort,
}; };
let response = self.client if self.thinking_enabled {
.post(&url) self.streaming_chat_completion(&url, &request).await
} else {
self.non_streaming_chat_completion(&url, &request).await
}
}
/// 非流式请求(非思考模式)
async fn non_streaming_chat_completion(
&self,
url: &str,
request: &ChatCompletionRequest,
) -> Result<String> {
let response = self
.client
.post(url)
.header("Authorization", format!("Bearer {}", self.api_key)) .header("Authorization", format!("Bearer {}", self.api_key))
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
.json(&request) .json(request)
.send() .send()
.await .await
.context("Failed to send request to DeepSeek")?; .context("Failed to send request to DeepSeek")?;
let status = response.status(); let status = response.status();
if !status.is_success() { if !status.is_success() {
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
// Try to parse error
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) { 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: {} ({})",
error.error.message,
error.error.error_type
);
} }
bail!("DeepSeek API error: {} - {}", status, text); bail!("DeepSeek API error: {} - {}", status, text);
} }
let result: ChatCompletionResponse = response let result: ChatCompletionResponse = response
.json() .json()
.await .await
.context("Failed to parse DeepSeek response")?; .context("Failed to parse DeepSeek response")?;
result.choices result
.choices
.into_iter() .into_iter()
.next() .next()
.map(|c| c.message.content.trim().to_string()) .map(|c| c.message.content.trim().to_string())
.filter(|s| !s.is_empty())
.ok_or_else(|| anyhow::anyhow!("No response from DeepSeek")) .ok_or_else(|| anyhow::anyhow!("No response from DeepSeek"))
} }
/// 流式请求(思考模式),处理 reasoning_content 和 content
async fn streaming_chat_completion(
&self,
url: &str,
request: &ChatCompletionRequest,
) -> Result<String> {
let response = self
.client
.post(url)
.header("Authorization", format!("Bearer {}", self.api_key))
.header("Content-Type", "application/json")
.header("Accept", "text/event-stream")
.json(request)
.send()
.await
.context("Failed to send streaming request to DeepSeek")?;
let status = response.status();
if !status.is_success() {
let text = response.text().await.unwrap_or_default();
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 mut content_buffer = String::new();
let mut has_reasoning = false;
let mut has_content = false;
let mut stream_ended = false;
let thinking_state = self.thinking_state.as_ref();
let mut byte_stream = response.bytes_stream();
let mut line_buffer = String::new();
use futures_util::StreamExt;
while let Some(chunk) = byte_stream.next().await {
let chunk = chunk.context("Failed to read streaming response chunk")?;
let chunk_str =
String::from_utf8(chunk.to_vec()).context("Invalid UTF-8 in stream chunk")?;
line_buffer.push_str(&chunk_str);
// 处理完整行
while let Some(line_end) = line_buffer.find('\n') {
let line = line_buffer[..line_end].trim().to_string();
line_buffer = line_buffer[line_end + 1..].to_string();
if line.is_empty() {
continue;
}
// SSE 格式data: {...} 或 data: [DONE]
if line == "data: [DONE]" {
stream_ended = true;
break;
}
if let Some(json_str) = line.strip_prefix("data: ") {
match serde_json::from_str::<StreamChunk>(json_str) {
Ok(chunk) => {
for choice in &chunk.choices {
// 处理 reasoning_content
if let Some(ref reasoning) = choice.delta.reasoning_content
&& !reasoning.is_empty()
{
if !has_reasoning {
has_reasoning = true;
if let Some(state) = thinking_state {
state.start_thinking();
}
}
// reasoning_content 不对外输出,仅用于内部状态判断
continue;
}
// 处理 content
if let Some(ref content) = choice.delta.content
&& !content.is_empty()
{
// reasoning 结束content 开始出现时移除 thinking 标识
if has_reasoning
&& !has_content
&& let Some(state) = thinking_state
{
state.end_thinking();
}
has_content = true;
content_buffer.push_str(content);
}
// 检查 finish_reason
if let Some(ref reason) = choice.finish_reason
&& reason == "stop"
{
stream_ended = true;
}
}
}
Err(_) => {
// 忽略无法解析的行(可能是心跳或注释)
}
}
}
}
if stream_ended {
break;
}
}
// 确保思考状态已结束
if let Some(state) = thinking_state {
state.end_thinking();
}
let result = content_buffer.trim().to_string();
if result.is_empty() {
if has_reasoning && !has_content {
bail!(
"DeepSeek returned reasoning content but no final answer. \
The model may have entered an incomplete thinking state. \
Please try again or disable thinking mode."
);
}
bail!(
"No response from DeepSeek. \
If thinking mode is enabled, try disabling it or ensure the model supports it."
);
}
Ok(result)
}
} }
/// Available DeepSeek models /// 可用 DeepSeek 模型列表
/// deepseek-chat / deepseek-reasoner 将于 2026-07-24 停用,推荐使用 V4 系列
pub const DEEPSEEK_MODELS: &[&str] = &[ pub const DEEPSEEK_MODELS: &[&str] = &[
"deepseek-v4-flash",
"deepseek-v4-pro",
// 兼容旧版模型 ID将于 2026-07-24 停用)
"deepseek-chat", "deepseek-chat",
"deepseek-coder", "deepseek-reasoner",
]; ];
/// Check if a model name is valid
pub fn is_valid_model(model: &str) -> bool { pub fn is_valid_model(model: &str) -> bool {
DEEPSEEK_MODELS.contains(&model) DEEPSEEK_MODELS.contains(&model)
} }
@@ -236,8 +550,73 @@ mod tests {
use super::*; use super::*;
#[test] #[test]
fn test_model_validation() { fn test_model_validation_v4() {
assert!(is_valid_model("deepseek-v4-flash"));
assert!(is_valid_model("deepseek-v4-pro"));
assert!(is_valid_model("deepseek-chat")); assert!(is_valid_model("deepseek-chat"));
assert!(is_valid_model("deepseek-reasoner"));
assert!(!is_valid_model("invalid-model")); assert!(!is_valid_model("invalid-model"));
assert!(!is_valid_model("deepseek-v3"));
} }
}
#[test]
fn test_client_builder_defaults() {
let client = DeepSeekClient::new("test-key", "deepseek-v4-flash").unwrap();
assert!(!client.thinking_enabled);
assert_eq!(client.max_tokens, 500);
assert_eq!(client.temperature, 0.7);
assert!(client.reasoning_effort.is_none());
assert!(client.thinking_state.is_none());
}
#[test]
fn test_client_builder_with_thinking() {
let client = DeepSeekClient::new("test-key", "deepseek-v4-flash")
.unwrap()
.with_thinking(true)
.with_reasoning_effort(Some("high".to_string()))
.with_max_tokens(1000)
.with_temperature(0.5);
assert!(client.thinking_enabled);
assert_eq!(client.reasoning_effort, Some("high".to_string()));
assert_eq!(client.max_tokens, 1000);
assert_eq!(client.temperature, 0.5);
}
#[test]
fn test_thinking_config_serialization() {
let config = ThinkingConfig {
thinking_type: "enabled".to_string(),
};
let json = serde_json::to_string(&config).unwrap();
assert_eq!(json, r#"{"type":"enabled"}"#);
}
#[test]
fn test_message_serialization_without_reasoning() {
let msg = Message {
role: "user".to_string(),
content: "Hello".to_string(),
reasoning_content: None,
};
let json = serde_json::to_string(&msg).unwrap();
assert!(!json.contains("reasoning_content"));
}
#[test]
fn test_stream_delta_parsing() {
let json = r#"{"content":"Hello","reasoning_content":null}"#;
let delta: StreamDelta = serde_json::from_str(json).unwrap();
assert_eq!(delta.content, Some("Hello".to_string()));
assert!(delta.reasoning_content.is_none());
}
#[test]
fn test_stream_delta_reasoning_only() {
let json = r#"{"content":null,"reasoning_content":"Let me think..."}"#;
let delta: StreamDelta = serde_json::from_str(json).unwrap();
assert!(delta.content.is_none());
assert_eq!(delta.reasoning_content, Some("Let me think...".to_string()));
}
}

View File

@@ -1,244 +1,587 @@
use super::{create_http_client, LlmProvider}; use super::thinking::ThinkingStateManager;
use anyhow::{bail, Context, Result}; use super::{LlmProvider, create_http_client};
use async_trait::async_trait; use anyhow::{Context, Result, bail};
use serde::{Deserialize, Serialize}; use async_trait::async_trait;
use std::time::Duration; use serde::{Deserialize, Serialize};
use std::sync::Arc;
/// Kimi API client (Moonshot AI) use std::time::Duration;
pub struct KimiClient {
base_url: String, /// Kimi API client (Moonshot AI)
api_key: String, pub struct KimiClient {
model: String, base_url: String,
client: reqwest::Client, api_key: String,
} model: String,
client: reqwest::Client,
#[derive(Debug, Serialize)] thinking_enabled: bool,
struct ChatCompletionRequest { max_tokens: u32,
model: String, temperature: f32,
messages: Vec<Message>, thinking_state: Option<Arc<ThinkingStateManager>>,
#[serde(skip_serializing_if = "Option::is_none")] }
max_tokens: Option<u32>,
#[serde(skip_serializing_if = "Option::is_none")] #[derive(Debug, Serialize)]
temperature: Option<f32>, struct ChatCompletionRequest {
stream: bool, model: String,
} messages: Vec<Message>,
#[serde(skip_serializing_if = "Option::is_none")]
#[derive(Debug, Serialize, Deserialize)] max_tokens: Option<u32>,
struct Message { #[serde(skip_serializing_if = "Option::is_none")]
role: String, temperature: Option<f32>,
content: String, stream: bool,
} #[serde(skip_serializing_if = "Option::is_none")]
thinking: Option<ThinkingConfig>,
#[derive(Debug, Deserialize)] }
struct ChatCompletionResponse {
choices: Vec<Choice>, #[derive(Debug, Serialize)]
} struct ThinkingConfig {
#[serde(rename = "type")]
#[derive(Debug, Deserialize)] thinking_type: String,
struct Choice { }
message: Message,
} #[derive(Debug, Clone, Serialize, Deserialize)]
struct Message {
#[derive(Debug, Deserialize)] role: String,
struct ErrorResponse { content: String,
error: ApiError, #[serde(skip_serializing_if = "Option::is_none")]
} reasoning_content: Option<String>,
}
#[derive(Debug, Deserialize)]
struct ApiError { #[derive(Debug, Deserialize)]
message: String, struct ChatCompletionResponse {
#[serde(rename = "type")] choices: Vec<Choice>,
error_type: String, }
}
#[derive(Debug, Deserialize)]
impl KimiClient { struct Choice {
/// Create new Kimi client message: Message,
pub fn new(api_key: &str, model: &str) -> Result<Self> { #[serde(default)]
let client = create_http_client(Duration::from_secs(60))?; reasoning_content: Option<String>,
}
Ok(Self {
base_url: "https://api.moonshot.cn/v1".to_string(), // --- Streaming response structures ---
api_key: api_key.to_string(),
model: model.to_string(), #[derive(Debug, Deserialize)]
client, struct StreamChunk {
}) choices: Vec<StreamChoice>,
} }
/// Create with custom base URL #[derive(Debug, Deserialize)]
pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> { struct StreamChoice {
let client = create_http_client(Duration::from_secs(60))?; delta: StreamDelta,
#[serde(default)]
Ok(Self { finish_reason: Option<String>,
base_url: base_url.trim_end_matches('/').to_string(), index: Option<u32>,
api_key: api_key.to_string(), }
model: model.to_string(),
client, #[derive(Debug, Deserialize, Default)]
}) struct StreamDelta {
} #[serde(default)]
content: Option<String>,
/// Set timeout #[serde(default)]
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> { reasoning_content: Option<String>,
self.client = create_http_client(timeout)?; }
Ok(self)
} #[derive(Debug, Deserialize)]
struct ErrorResponse {
/// List available models error: ApiError,
pub async fn list_models(&self) -> Result<Vec<String>> { }
let url = format!("{}/models", self.base_url);
#[derive(Debug, Deserialize)]
let response = self.client struct ApiError {
.get(&url) message: String,
.header("Authorization", format!("Bearer {}", self.api_key)) #[serde(rename = "type")]
.send() error_type: String,
.await }
.context("Failed to list Kimi models")?;
impl KimiClient {
if !response.status().is_success() { pub fn new(api_key: &str, model: &str) -> Result<Self> {
let status = response.status(); let client = create_http_client(Duration::from_secs(300))?;
let text = response.text().await.unwrap_or_default();
bail!("Kimi API error: {} - {}", status, text); Ok(Self {
} base_url: "https://api.moonshot.cn/v1".to_string(),
api_key: api_key.to_string(),
#[derive(Deserialize)] model: model.to_string(),
struct ModelsResponse { client,
data: Vec<Model>, thinking_enabled: false,
} max_tokens: 500,
temperature: 1.0,
#[derive(Deserialize)] thinking_state: None,
struct Model { })
id: String, }
}
pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> {
let result: ModelsResponse = response let client = create_http_client(Duration::from_secs(300))?;
.json()
.await Ok(Self {
.context("Failed to parse Kimi response")?; base_url: base_url.trim_end_matches('/').to_string(),
api_key: api_key.to_string(),
Ok(result.data.into_iter().map(|m| m.id).collect()) model: model.to_string(),
} client,
thinking_enabled: false,
/// Validate API key max_tokens: 500,
pub async fn validate_key(&self) -> Result<bool> { temperature: 1.0,
match self.list_models().await { thinking_state: None,
Ok(_) => Ok(true), })
Err(e) => { }
let err_str = e.to_string();
if err_str.contains("401") || err_str.contains("Unauthorized") { pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
Ok(false) self.client = create_http_client(timeout)?;
} else { Ok(self)
Err(e) }
}
} pub fn with_thinking(mut self, enabled: bool) -> Self {
} self.thinking_enabled = enabled;
} self
} }
#[async_trait] pub fn with_max_tokens(mut self, max_tokens: u32) -> Self {
impl LlmProvider for KimiClient { self.max_tokens = max_tokens;
async fn generate(&self, prompt: &str) -> Result<String> { self
let messages = vec![ }
Message {
role: "user".to_string(), pub fn with_temperature(mut self, temperature: f32) -> Self {
content: prompt.to_string(), self.temperature = temperature;
}, self
]; }
self.chat_completion(messages).await pub fn with_thinking_state(mut self, state: Arc<ThinkingStateManager>) -> Self {
} self.thinking_state = Some(state);
self
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> { }
let mut messages = vec![];
pub async fn list_models(&self) -> Result<Vec<String>> {
if !system.is_empty() { let url = format!("{}/models", self.base_url);
messages.push(Message {
role: "system".to_string(), let response = self
content: system.to_string(), .client
}); .get(&url)
} .header("Authorization", format!("Bearer {}", self.api_key))
.send()
messages.push(Message { .await
role: "user".to_string(), .context("Failed to list Kimi models")?;
content: user.to_string(),
}); if !response.status().is_success() {
let status = response.status();
self.chat_completion(messages).await let text = response.text().await.unwrap_or_default();
} bail!("Kimi API error: {} - {}", status, text);
}
async fn is_available(&self) -> bool {
self.validate_key().await.unwrap_or(false) #[derive(Deserialize)]
} struct ModelsResponse {
data: Vec<ModelId>,
fn name(&self) -> &str { }
"kimi"
} #[derive(Deserialize)]
} struct ModelId {
id: String,
impl KimiClient { }
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
let url = format!("{}/chat/completions", self.base_url); let result: ModelsResponse = response
.json()
let request = ChatCompletionRequest { .await
model: self.model.clone(), .context("Failed to parse Kimi response")?;
messages,
max_tokens: Some(500), Ok(result.data.into_iter().map(|m| m.id).collect())
temperature: Some(0.7), }
stream: false,
}; pub async fn validate_key(&self) -> Result<bool> {
match self.list_models().await {
let response = self.client Ok(_) => Ok(true),
.post(&url) Err(e) => {
.header("Authorization", format!("Bearer {}", self.api_key)) let err_str = e.to_string();
.header("Content-Type", "application/json") if err_str.contains("401") || err_str.contains("Unauthorized") {
.json(&request) Ok(false)
.send() } else {
.await Err(e)
.context("Failed to send request to Kimi")?; }
}
let status = response.status(); }
}
if !status.is_success() { }
let text = response.text().await.unwrap_or_default();
#[async_trait]
// Try to parse error impl LlmProvider for KimiClient {
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) { async fn generate(&self, prompt: &str) -> Result<String> {
bail!("Kimi API error: {} ({})", error.error.message, error.error.error_type); let messages = vec![Message {
} role: "user".to_string(),
content: prompt.to_string(),
bail!("Kimi API error: {} - {}", status, text); reasoning_content: None,
} }];
let result: ChatCompletionResponse = response self.chat_completion_with_retry(messages).await
.json() }
.await
.context("Failed to parse Kimi response")?; async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
let mut messages = vec![];
result.choices
.into_iter() if !system.is_empty() {
.next() messages.push(Message {
.map(|c| c.message.content.trim().to_string()) role: "system".to_string(),
.ok_or_else(|| anyhow::anyhow!("No response from Kimi")) content: system.to_string(),
} reasoning_content: None,
} });
}
/// Available Kimi models
pub const KIMI_MODELS: &[&str] = &[ messages.push(Message {
"moonshot-v1-8k", role: "user".to_string(),
"moonshot-v1-32k", content: user.to_string(),
"moonshot-v1-128k", reasoning_content: None,
]; });
/// Check if a model name is valid self.chat_completion_with_retry(messages).await
pub fn is_valid_model(model: &str) -> bool { }
KIMI_MODELS.contains(&model)
} async fn is_available(&self) -> bool {
self.validate_key().await.unwrap_or(false)
#[cfg(test)] }
mod tests {
use super::*; fn name(&self) -> &str {
"kimi"
#[test] }
fn test_model_validation() { }
assert!(is_valid_model("moonshot-v1-8k"));
assert!(!is_valid_model("invalid-model")); impl KimiClient {
} async fn chat_completion_with_retry(&self, messages: Vec<Message>) -> Result<String> {
} let mut last_error = None;
for attempt in 1..=3 {
match self.chat_completion(messages.clone()).await {
Ok(result) => return Ok(result),
Err(e) => {
let err_msg = e.to_string();
let is_retryable = err_msg.contains("timeout")
|| err_msg.contains("connection")
|| err_msg.contains("temporary")
|| err_msg.contains("5")
&& (err_msg.contains("500")
|| err_msg.contains("502")
|| err_msg.contains("503")
|| err_msg.contains("504"));
if !is_retryable || attempt == 3 {
last_error = Some(e);
break;
}
tokio::time::sleep(Duration::from_millis(500 * 2u64.pow(attempt - 1))).await;
}
}
}
Err(last_error.unwrap_or_else(|| anyhow::anyhow!("Request failed after retries")))
}
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
let url = format!("{}/chat/completions", self.base_url);
let thinking = Some(ThinkingConfig {
thinking_type: if self.thinking_enabled {
"enabled".to_string()
} else {
"disabled".to_string()
},
});
// Kimi API temperature 要求:
// - 思考模式: temperature 必须为 1.0
// - 非思考模式: temperature 必须为 0.6
let temperature = if self.thinking_enabled {
Some(1.0)
} else {
Some(0.6)
};
let request = ChatCompletionRequest {
model: self.model.clone(),
messages: messages.clone(),
max_tokens: Some(self.max_tokens),
temperature,
stream: self.thinking_enabled,
thinking,
};
if self.thinking_enabled {
self.streaming_chat_completion(&url, &request).await
} else {
self.non_streaming_chat_completion(&url, &request).await
}
}
/// 非流式请求(非思考模式)
async fn non_streaming_chat_completion(
&self,
url: &str,
request: &ChatCompletionRequest,
) -> Result<String> {
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();
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| {
let content = c.message.content.trim().to_string();
if content.is_empty() {
c.reasoning_content
.or(c.message.reasoning_content)
.map(|r| r.trim().to_string())
.unwrap_or_default()
} else {
content
}
})
.filter(|s| !s.is_empty())
.ok_or_else(|| anyhow::anyhow!("No response from Kimi"))
}
/// 流式请求(思考模式),处理 reasoning_content 和 content
async fn streaming_chat_completion(
&self,
url: &str,
request: &ChatCompletionRequest,
) -> Result<String> {
let response = self
.client
.post(url)
.header("Authorization", format!("Bearer {}", self.api_key))
.header("Content-Type", "application/json")
.header("Accept", "text/event-stream")
.json(request)
.send()
.await
.context("Failed to send streaming request to Kimi")?;
let status = response.status();
if !status.is_success() {
let text = response.text().await.unwrap_or_default();
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 mut content_buffer = String::new();
let mut has_reasoning = false;
let mut has_content = false;
let mut stream_ended = false;
let thinking_state = self.thinking_state.as_ref();
let mut byte_stream = response.bytes_stream();
let mut line_buffer = String::new();
use futures_util::StreamExt;
while let Some(chunk) = byte_stream.next().await {
let chunk = chunk.context("Failed to read streaming response chunk")?;
let chunk_str =
String::from_utf8(chunk.to_vec()).context("Invalid UTF-8 in stream chunk")?;
line_buffer.push_str(&chunk_str);
while let Some(line_end) = line_buffer.find('\n') {
let line = line_buffer[..line_end].trim().to_string();
line_buffer = line_buffer[line_end + 1..].to_string();
if line.is_empty() {
continue;
}
if line == "data: [DONE]" {
stream_ended = true;
break;
}
if let Some(json_str) = line.strip_prefix("data: ") {
match serde_json::from_str::<StreamChunk>(json_str) {
Ok(chunk) => {
for choice in &chunk.choices {
if let Some(ref reasoning) = choice.delta.reasoning_content
&& !reasoning.is_empty()
{
if !has_reasoning {
has_reasoning = true;
if let Some(state) = thinking_state {
state.start_thinking();
}
}
continue;
}
if let Some(ref content) = choice.delta.content
&& !content.is_empty()
{
if has_reasoning
&& !has_content
&& let Some(state) = thinking_state
{
state.end_thinking();
}
has_content = true;
content_buffer.push_str(content);
}
if let Some(ref reason) = choice.finish_reason
&& reason == "stop"
{
stream_ended = true;
}
}
}
Err(_) => {
// 忽略无法解析的行
}
}
}
}
if stream_ended {
break;
}
}
// 确保思考状态已结束
if let Some(state) = thinking_state {
state.end_thinking();
}
let result = content_buffer.trim().to_string();
if result.is_empty() {
if has_reasoning && !has_content {
bail!(
"Kimi returned reasoning content but no final answer. \
The model may have entered an incomplete thinking state. \
Please try again or disable thinking mode."
);
}
bail!(
"No response from Kimi. \
If thinking mode is enabled, try disabling it or ensure the model supports it."
);
}
Ok(result)
}
}
/// 可用 Kimi 模型列表
pub const KIMI_MODELS: &[&str] = &[
// K2 系列(推荐)
"kimi-k2.6",
"kimi-k2.5",
"kimi-k2-thinking",
"kimi-k2-thinking-turbo",
"kimi-k2-instruct",
"kimi-k2-instruct-0905",
// 兼容旧版模型 ID
"moonshot-v1-8k",
"moonshot-v1-32k",
"moonshot-v1-128k",
];
pub fn is_valid_model(model: &str) -> bool {
KIMI_MODELS.contains(&model)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_model_validation_k2() {
assert!(is_valid_model("kimi-k2.6"));
assert!(is_valid_model("kimi-k2.5"));
assert!(is_valid_model("kimi-k2-thinking"));
assert!(is_valid_model("kimi-k2-thinking-turbo"));
assert!(is_valid_model("moonshot-v1-8k"));
assert!(is_valid_model("moonshot-v1-32k"));
assert!(is_valid_model("moonshot-v1-128k"));
assert!(!is_valid_model("invalid-model"));
assert!(!is_valid_model("kimi-k1.5"));
}
#[test]
fn test_client_builder_defaults() {
let client = KimiClient::new("test-key", "kimi-k2.6").unwrap();
assert!(!client.thinking_enabled);
assert_eq!(client.max_tokens, 500);
assert_eq!(client.temperature, 1.0);
assert!(client.thinking_state.is_none());
}
#[test]
fn test_client_builder_with_thinking() {
let client = KimiClient::new("test-key", "kimi-k2.6")
.unwrap()
.with_thinking(true)
.with_max_tokens(1000)
.with_temperature(0.5);
assert!(client.thinking_enabled);
assert_eq!(client.max_tokens, 1000);
assert_eq!(client.temperature, 0.5);
}
#[test]
fn test_thinking_config_serialization() {
let config = ThinkingConfig {
thinking_type: "enabled".to_string(),
};
let json = serde_json::to_string(&config).unwrap();
assert_eq!(json, r#"{"type":"enabled"}"#);
}
#[test]
fn test_client_new_defaults() {
let client = KimiClient::new("test-key", "kimi-k2.6").unwrap();
assert_eq!(client.name(), "kimi");
assert!(!client.thinking_enabled);
}
#[test]
fn test_message_serialization() {
let msg = Message {
role: "user".to_string(),
content: "Hello".to_string(),
reasoning_content: None,
};
let json = serde_json::to_string(&msg).unwrap();
assert!(!json.contains("reasoning_content"));
}
}

View File

@@ -1,20 +1,21 @@
use anyhow::{bail, Context, Result}; use crate::config::Language;
use anyhow::{Context, Result, bail};
use async_trait::async_trait; use async_trait::async_trait;
use std::time::Duration; use std::time::Duration;
use crate::config::Language;
pub mod anthropic;
pub mod deepseek;
pub mod kimi;
pub mod ollama; pub mod ollama;
pub mod openai; pub mod openai;
pub mod anthropic;
pub mod kimi;
pub mod deepseek;
pub mod openrouter; pub mod openrouter;
pub mod thinking;
pub use anthropic::AnthropicClient;
pub use deepseek::DeepSeekClient;
pub use kimi::KimiClient;
pub use ollama::OllamaClient; pub use ollama::OllamaClient;
pub use openai::OpenAiClient; pub use openai::OpenAiClient;
pub use anthropic::AnthropicClient;
pub use kimi::KimiClient;
pub use deepseek::DeepSeekClient;
pub use openrouter::OpenRouterClient; pub use openrouter::OpenRouterClient;
/// LLM provider trait /// LLM provider trait
@@ -22,13 +23,13 @@ pub use openrouter::OpenRouterClient;
pub trait LlmProvider: Send + Sync { pub trait LlmProvider: Send + Sync {
/// Generate text from prompt /// Generate text from prompt
async fn generate(&self, prompt: &str) -> Result<String>; async fn generate(&self, prompt: &str) -> Result<String>;
/// Generate with system prompt /// Generate with system prompt
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String>; async fn generate_with_system(&self, system: &str, user: &str) -> Result<String>;
/// Check if provider is available /// Check if provider is available
async fn is_available(&self) -> bool; async fn is_available(&self) -> bool;
/// Get provider name /// Get provider name
fn name(&self) -> &str; fn name(&self) -> &str;
} }
@@ -44,6 +45,7 @@ pub struct LlmClientConfig {
pub max_tokens: u32, pub max_tokens: u32,
pub temperature: f32, pub temperature: f32,
pub timeout: Duration, pub timeout: Duration,
pub thinking_enabled: bool,
} }
impl Default for LlmClientConfig { impl Default for LlmClientConfig {
@@ -52,6 +54,7 @@ impl Default for LlmClientConfig {
max_tokens: 500, max_tokens: 500,
temperature: 0.7, temperature: 0.7,
timeout: Duration::from_secs(30), timeout: Duration::from_secs(30),
thinking_enabled: false,
} }
} }
} }
@@ -59,11 +62,20 @@ impl Default for LlmClientConfig {
impl LlmClient { impl LlmClient {
/// Create LLM client from configuration manager /// Create LLM client from configuration manager
pub async fn from_config(manager: &crate::config::manager::ConfigManager) -> Result<Self> { pub async fn from_config(manager: &crate::config::manager::ConfigManager) -> Result<Self> {
Self::from_config_with_think(manager, manager.config().llm.thinking_enabled).await
}
/// Create LLM client from configuration with explicit thinking override
pub async fn from_config_with_think(
manager: &crate::config::manager::ConfigManager,
thinking_enabled: bool,
) -> Result<Self> {
let config = manager.config(); let config = manager.config();
let client_config = LlmClientConfig { let client_config = LlmClientConfig {
max_tokens: config.llm.max_tokens, max_tokens: config.llm.max_tokens,
temperature: config.llm.temperature, temperature: config.llm.temperature,
timeout: Duration::from_secs(config.llm.timeout), timeout: Duration::from_secs(config.llm.timeout),
thinking_enabled,
}; };
let provider = config.llm.provider.as_str(); let provider = config.llm.provider.as_str();
@@ -72,33 +84,99 @@ impl LlmClient {
let api_key = manager.get_api_key(); let api_key = manager.get_api_key();
let provider: Box<dyn LlmProvider> = match provider { let provider: Box<dyn LlmProvider> = match provider {
"ollama" => { "ollama" => Box::new(
Box::new(OllamaClient::new(&base_url, model)) OllamaClient::new(&base_url, model)
} .with_max_tokens(client_config.max_tokens)
.with_temperature(client_config.temperature),
),
"openai" => { "openai" => {
let key = api_key.as_ref() let key = api_key
.as_ref()
.ok_or_else(|| anyhow::anyhow!("OpenAI API key not configured"))?; .ok_or_else(|| anyhow::anyhow!("OpenAI API key not configured"))?;
Box::new(OpenAiClient::new(&base_url, key, model)?) let thinking_state = if thinking_enabled {
Some(thinking::create_console_thinking_state())
} else {
None
};
let mut client = OpenAiClient::new(&base_url, key, model)?
.with_thinking(thinking_enabled)
.with_max_tokens(client_config.max_tokens)
.with_temperature(client_config.temperature)
.with_timeout(client_config.timeout)?;
if let Some(state) = thinking_state {
client = client.with_thinking_state(state);
}
Box::new(client)
} }
"anthropic" => { "anthropic" => {
let key = api_key.as_ref() let key = api_key
.as_ref()
.ok_or_else(|| anyhow::anyhow!("Anthropic API key not configured"))?; .ok_or_else(|| anyhow::anyhow!("Anthropic API key not configured"))?;
Box::new(AnthropicClient::new(key, model)?) let thinking_state = if thinking_enabled {
Some(thinking::create_console_thinking_state())
} else {
None
};
let budget = config.llm.thinking_budget_tokens.unwrap_or(1024);
let mut client = AnthropicClient::new(key, model)?
.with_thinking(thinking_enabled)
.with_thinking_budget_tokens(budget)
.with_max_tokens(client_config.max_tokens)
.with_temperature(client_config.temperature)
.with_timeout(client_config.timeout)?;
if let Some(state) = thinking_state {
client = client.with_thinking_state(state);
}
Box::new(client)
} }
"kimi" => { "kimi" => {
let key = api_key.as_ref() let key = api_key
.as_ref()
.ok_or_else(|| anyhow::anyhow!("Kimi API key not configured"))?; .ok_or_else(|| anyhow::anyhow!("Kimi API key not configured"))?;
Box::new(KimiClient::with_base_url(key, model, &base_url)?) let thinking_state = if thinking_enabled {
Some(thinking::create_console_thinking_state())
} else {
None
};
let mut client = KimiClient::with_base_url(key, model, &base_url)?
.with_thinking(thinking_enabled)
.with_max_tokens(client_config.max_tokens)
.with_temperature(client_config.temperature)
.with_timeout(client_config.timeout)?;
if let Some(state) = thinking_state {
client = client.with_thinking_state(state);
}
Box::new(client)
} }
"deepseek" => { "deepseek" => {
let key = api_key.as_ref() let key = api_key
.as_ref()
.ok_or_else(|| anyhow::anyhow!("DeepSeek API key not configured"))?; .ok_or_else(|| anyhow::anyhow!("DeepSeek API key not configured"))?;
Box::new(DeepSeekClient::with_base_url(key, model, &base_url)?) let thinking_state = if thinking_enabled {
Some(thinking::create_console_thinking_state())
} else {
None
};
let mut client = DeepSeekClient::with_base_url(key, model, &base_url)?
.with_thinking(thinking_enabled)
.with_max_tokens(client_config.max_tokens)
.with_temperature(client_config.temperature)
.with_timeout(client_config.timeout)?;
if let Some(state) = thinking_state {
client = client.with_thinking_state(state);
}
Box::new(client)
} }
"openrouter" => { "openrouter" => {
let key = api_key.as_ref() let key = api_key
.as_ref()
.ok_or_else(|| anyhow::anyhow!("OpenRouter API key not configured"))?; .ok_or_else(|| anyhow::anyhow!("OpenRouter API key not configured"))?;
Box::new(OpenRouterClient::with_base_url(key, model, &base_url)?) Box::new(
OpenRouterClient::with_base_url(key, model, &base_url)?
.with_max_tokens(client_config.max_tokens)
.with_temperature(client_config.temperature)
.with_timeout(client_config.timeout)?,
)
} }
_ => bail!("Unknown LLM provider: {}", provider), _ => bail!("Unknown LLM provider: {}", provider),
}; };
@@ -123,9 +201,17 @@ impl LlmClient {
diff: &str, diff: &str,
format: crate::config::CommitFormat, format: crate::config::CommitFormat,
language: Language, language: Language,
template: Option<&str>,
) -> Result<GeneratedCommit> { ) -> Result<GeneratedCommit> {
let system_prompt = get_commit_system_prompt(format, language); let mut system_prompt = get_commit_system_prompt(format, language).to_string();
if let Some(tmpl) = template {
system_prompt.push_str(&format!(
"\n\n## Commit Message Template\nFollow this template structure:\n{}",
tmpl
));
}
// Add language instruction to the prompt // Add language instruction to the prompt
let language_instruction = match language { let language_instruction = match language {
Language::Chinese => "\n\n请用中文生成提交消息。", Language::Chinese => "\n\n请用中文生成提交消息。",
@@ -136,10 +222,13 @@ impl LlmClient {
Language::German => "\n\nBitte generieren Sie die Commit-Nachricht auf Deutsch.", Language::German => "\n\nBitte generieren Sie die Commit-Nachricht auf Deutsch.",
Language::English => "", Language::English => "",
}; };
let prompt = format!("{}{}", diff, language_instruction); let prompt = format!("{}{}", diff, language_instruction);
let response = self.provider.generate_with_system(system_prompt, &prompt).await?; let response = self
.provider
.generate_with_system(&system_prompt, &prompt)
.await?;
self.parse_commit_response(&response, format) self.parse_commit_response(&response, format)
} }
@@ -152,7 +241,7 @@ impl LlmClient {
) -> Result<String> { ) -> Result<String> {
let system_prompt = get_tag_system_prompt(language); let system_prompt = get_tag_system_prompt(language);
let commits_text = commits.join("\n"); let commits_text = commits.join("\n");
// Add language instruction to the prompt // Add language instruction to the prompt
let language_instruction = match language { let language_instruction = match language {
Language::Chinese => "\n\n请用中文生成标签消息。", Language::Chinese => "\n\n请用中文生成标签消息。",
@@ -163,10 +252,15 @@ impl LlmClient {
Language::German => "\n\nBitte generieren Sie die Tag-Nachricht auf Deutsch.", Language::German => "\n\nBitte generieren Sie die Tag-Nachricht auf Deutsch.",
Language::English => "", Language::English => "",
}; };
let prompt = format!("Version: {}\n\nCommits:\n{}{}", version, commits_text, language_instruction); let prompt = format!(
"Version: {}\n\nCommits:\n{}{}",
self.provider.generate_with_system(system_prompt, &prompt).await version, commits_text, language_instruction
);
self.provider
.generate_with_system(system_prompt, &prompt)
.await
} }
/// Generate changelog entry /// Generate changelog entry
@@ -177,13 +271,13 @@ impl LlmClient {
language: Language, language: Language,
) -> Result<String> { ) -> Result<String> {
let system_prompt = get_changelog_system_prompt(language); let system_prompt = get_changelog_system_prompt(language);
let commits_text = commits let commits_text = commits
.iter() .iter()
.map(|(t, m)| format!("- [{}] {}", t, m)) .map(|(t, m)| format!("- [{}] {}", t, m))
.collect::<Vec<_>>() .collect::<Vec<_>>()
.join("\n"); .join("\n");
// Add language instruction to the prompt // Add language instruction to the prompt
let language_instruction = match language { let language_instruction = match language {
Language::Chinese => "\n\n请用中文生成变更日志。", Language::Chinese => "\n\n请用中文生成变更日志。",
@@ -194,10 +288,15 @@ impl LlmClient {
Language::German => "\n\nBitte generieren Sie das Changelog auf Deutsch.", Language::German => "\n\nBitte generieren Sie das Changelog auf Deutsch.",
Language::English => "", Language::English => "",
}; };
let prompt = format!("Version: {}\n\nCommits:\n{}{}", version, commits_text, language_instruction); let prompt = format!(
"Version: {}\n\nCommits:\n{}{}",
self.provider.generate_with_system(system_prompt, &prompt).await version, commits_text, language_instruction
);
self.provider
.generate_with_system(system_prompt, &prompt)
.await
} }
/// Check if provider is available /// Check if provider is available
@@ -206,35 +305,115 @@ impl LlmClient {
} }
/// Parse commit response from LLM /// Parse commit response from LLM
fn parse_commit_response(&self, response: &str, format: crate::config::CommitFormat) -> Result<GeneratedCommit> { fn parse_commit_response(
let lines: Vec<&str> = response.lines().collect(); &self,
response: &str,
format: crate::config::CommitFormat,
) -> Result<GeneratedCommit> {
// Clean markdown code fences from the response
let cleaned = Self::strip_code_fences(response);
let lines: Vec<&str> = cleaned
.lines()
.map(|l| l.trim())
.filter(|l| !l.is_empty())
.collect();
if lines.is_empty() { if lines.is_empty() {
bail!("Empty response from LLM"); 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
);
} }
let first_line = lines[0]; // Find the line most likely to be the commit subject
let first_line = Self::find_commit_subject_line(&lines, format);
// Parse based on format // Parse based on format
match format { match format {
crate::config::CommitFormat::Conventional => { crate::config::CommitFormat::Conventional => {
self.parse_conventional_commit(first_line, lines) self.parse_conventional_commit(first_line, &lines, response)
} }
crate::config::CommitFormat::Commitlint => { crate::config::CommitFormat::Commitlint => {
self.parse_commitlint_commit(first_line, lines) self.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( fn parse_conventional_commit(
&self, &self,
first_line: &str, first_line: &str,
lines: Vec<&str>, lines: &[&str],
raw_response: &str,
) -> Result<GeneratedCommit> { ) -> Result<GeneratedCommit> {
// Parse: type(scope)!: description // Parse: type(scope)!: description
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 conventional commit format: missing colon"); 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 type_part = parts[0];
@@ -257,7 +436,7 @@ impl LlmClient {
}; };
// Extract body and footer // Extract body and footer
let (body, footer) = self.extract_body_footer(&lines); let (body, footer) = self.extract_body_footer(lines);
Ok(GeneratedCommit { Ok(GeneratedCommit {
commit_type, commit_type,
@@ -272,12 +451,21 @@ impl LlmClient {
fn parse_commitlint_commit( fn parse_commitlint_commit(
&self, &self,
first_line: &str, first_line: &str,
lines: Vec<&str>, lines: &[&str],
raw_response: &str,
) -> Result<GeneratedCommit> { ) -> Result<GeneratedCommit> {
// Similar parsing but with commitlint rules // Similar parsing but with commitlint rules
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: missing colon"); 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 type_part = parts[0];
@@ -323,8 +511,14 @@ impl LlmClient {
} }
// Look for footer markers // Look for footer markers
let footer_markers = ["BREAKING CHANGE:", "Closes", "Fixes", "Refs", "Co-authored-by:"]; let footer_markers = [
"BREAKING CHANGE:",
"Closes",
"Fixes",
"Refs",
"Co-authored-by:",
];
let mut body_lines = vec![]; let mut body_lines = vec![];
let mut footer_lines = vec![]; let mut footer_lines = vec![];
let mut in_footer = false; let mut in_footer = false;
@@ -333,7 +527,7 @@ impl LlmClient {
if footer_markers.iter().any(|m| line.starts_with(m)) { if footer_markers.iter().any(|m| line.starts_with(m)) {
in_footer = true; in_footer = true;
} }
if in_footer { if in_footer {
footer_lines.push(*line); footer_lines.push(*line);
} else { } else {
@@ -403,17 +597,34 @@ pub(crate) fn create_http_client(timeout: Duration) -> Result<reqwest::Client> {
} }
/// Get commit system prompt based on format and language /// Get commit system prompt based on format and language
fn get_commit_system_prompt(format: crate::config::CommitFormat, language: Language) -> &'static str { fn get_commit_system_prompt(
format: crate::config::CommitFormat,
language: Language,
) -> &'static str {
match (format, language) { match (format, language) {
(crate::config::CommitFormat::Conventional, Language::Chinese) => CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ZH, (crate::config::CommitFormat::Conventional, Language::Chinese) => {
(crate::config::CommitFormat::Conventional, Language::Japanese) => CONVENTIONAL_COMMIT_SYSTEM_PROMPT_JA, CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ZH
(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::Japanese) => {
(crate::config::CommitFormat::Conventional, Language::French) => CONVENTIONAL_COMMIT_SYSTEM_PROMPT_FR, CONVENTIONAL_COMMIT_SYSTEM_PROMPT_JA
(crate::config::CommitFormat::Conventional, Language::German) => CONVENTIONAL_COMMIT_SYSTEM_PROMPT_DE, }
(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::Conventional, _) => CONVENTIONAL_COMMIT_SYSTEM_PROMPT,
(crate::config::CommitFormat::Commitlint, Language::Chinese) => COMMITLINT_SYSTEM_PROMPT_ZH, (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::Japanese) => {
COMMITLINT_SYSTEM_PROMPT_JA
}
(crate::config::CommitFormat::Commitlint, Language::Korean) => COMMITLINT_SYSTEM_PROMPT_KO, (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::Spanish) => COMMITLINT_SYSTEM_PROMPT_ES,
(crate::config::CommitFormat::Commitlint, Language::French) => COMMITLINT_SYSTEM_PROMPT_FR, (crate::config::CommitFormat::Commitlint, Language::French) => COMMITLINT_SYSTEM_PROMPT_FR,
@@ -504,8 +715,7 @@ const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成符合 C
4. 不要大写首字母 4. 不要大写首字母
5. 结尾不要句号 5. 结尾不要句号
6. 如果更改特定于模块/组件,请包含作用域 6. 如果更改特定于模块/组件,请包含作用域
7. 仅输出提交消息,不要输出其他内容。
仅输出提交消息,不要输出其他内容。
"#; "#;
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_JA: &str = r#"あなたはConventional Commits仕様に従ったコミットメッセージを生成するアシスタントです。 const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_JA: &str = r#"あなたはConventional Commits仕様に従ったコミットメッセージを生成するアシスタントです。
@@ -534,8 +744,7 @@ const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_JA: &str = r#"あなたはConventional C
4. 先頭を大文字にしない 4. 先頭を大文字にしない
5. 最後にピリオドを付けない 5. 最後にピリオドを付けない
6. 変更がモジュール/コンポーネントに固有の場合はスコープを含める 6. 変更がモジュール/コンポーネントに固有の場合はスコープを含める
7. コミットメッセージのみを出力し、それ以外は出力しないでください。
コミットメッセージのみを出力し、それ以外は出力しないでください。
"#; "#;
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_KO: &str = r#"당신은 Conventional Commits 사양에 따른 커밋 메시지를 생성하는 도우미입니다. const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_KO: &str = r#"당신은 Conventional Commits 사양에 따른 커밋 메시지를 생성하는 도우미입니다.
@@ -564,8 +773,7 @@ const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_KO: &str = r#"당신은 Conventional Com
4. 첫 글자 대문자화하지 않음 4. 첫 글자 대문자화하지 않음
5. 끝에 마침표 사용하지 않음 5. 끝에 마침표 사용하지 않음
6. 변경 사항이 모듈/구성 요소에 특정한 경우 범위 포함 6. 변경 사항이 모듈/구성 요소에 특정한 경우 범위 포함
7. 커밋 메시지만 출력하고 다른 내용은 출력하지 마세요.
커밋 메시지만 출력하고 다른 내용은 출력하지 마세요.
"#; "#;
const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera mensajes de commit siguiendo la especificación Conventional Commits. const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera mensajes de commit siguiendo la especificación Conventional Commits.
@@ -594,8 +802,7 @@ Reglas:
4. No capitalices la primera letra 4. No capitalices la primera letra
5. Sin punto al final 5. Sin punto al final
6. Incluye alcance si el cambio es específico de un módulo/componente 6. Incluye alcance si el cambio es específico de un módulo/componente
7. Genera SOLO el mensaje de commit, nada más.
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. 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.
@@ -624,8 +831,7 @@ Règles:
4. Ne capitalisez pas la première lettre 4. Ne capitalisez pas la première lettre
5. Pas de point à la fin 5. Pas de point à la fin
6. Incluez la portée si le changement est spécifique à un module/composant 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.
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. const CONVENTIONAL_COMMIT_SYSTEM_PROMPT_DE: &str = r#"Sie sind ein Assistent, der Commit-Nachrichten gemäß der Conventional Commits-Spezifikation generiert.
@@ -654,8 +860,7 @@ Regeln:
4. Großschreiben Sie den ersten Buchstaben nicht 4. Großschreiben Sie den ersten Buchstaben nicht
5. Kein Punkt am Ende 5. Kein Punkt am Ende
6. Fügen Sie einen Bereich ein, wenn die Änderung spezifisch für ein Modul/Komponente ist 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.
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. const COMMITLINT_SYSTEM_PROMPT: &str = r#"You are a helpful assistant that generates commit messages following @commitlint/config-conventional.
@@ -672,8 +877,7 @@ Rules:
3. Subject should be 4-100 characters 3. Subject should be 4-100 characters
4. Use imperative mood 4. Use imperative mood
5. Be concise but descriptive 5. Be concise but descriptive
6. Output ONLY the commit message, nothing else.
Output ONLY the commit message, nothing else.
"#; "#;
const COMMITLINT_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成符合 @commitlint/config-conventional 规范的提交消息的助手。 const COMMITLINT_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成符合 @commitlint/config-conventional 规范的提交消息的助手。
@@ -690,8 +894,7 @@ const COMMITLINT_SYSTEM_PROMPT_ZH: &str = r#"你是一个生成符合 @commitlin
3. 主题应为 4-100 个字符 3. 主题应为 4-100 个字符
4. 使用祈使语气 4. 使用祈使语气
5. 简洁但描述性强 5. 简洁但描述性强
6. 仅输出提交消息,不要输出其他额外内容。
仅输出提交消息,不要输出其他内容。
"#; "#;
const COMMITLINT_SYSTEM_PROMPT_JA: &str = r#"あなたは@commitlint/config-conventionalに従ったコミットメッセージを生成するアシスタントです。 const COMMITLINT_SYSTEM_PROMPT_JA: &str = r#"あなたは@commitlint/config-conventionalに従ったコミットメッセージを生成するアシスタントです。
@@ -708,8 +911,7 @@ git diffを分析し、コミットメッセージを生成してください。
3. 件名は4-100文字である必要があります 3. 件名は4-100文字である必要があります
4. 命令形を使用してください 4. 命令形を使用してください
5. 簡潔ですが説明的であること 5. 簡潔ですが説明的であること
6. コミットメッセージのみを出力し、それ以外は出力しないでください。
コミットメッセージのみを出力し、それ以外は出力しないでください。
"#; "#;
const COMMITLINT_SYSTEM_PROMPT_KO: &str = r#"당신은 @commitlint/config-conventional에 따른 커밋 메시지를 생성하는 도우미입니다. const COMMITLINT_SYSTEM_PROMPT_KO: &str = r#"당신은 @commitlint/config-conventional에 따른 커밋 메시지를 생성하는 도우미입니다.
@@ -726,8 +928,7 @@ git diff를 분석하고 커밋 메시지를 생성하세요.
3. 제목은 4-100자여야 합니다 3. 제목은 4-100자여야 합니다
4. 명령형을 사용하세요 4. 명령형을 사용하세요
5. 간결하지만 설명적이어야 합니다 5. 간결하지만 설명적이어야 합니다
6. 커밋 메시지만 출력하고 다른 내용은 출력하지 마세요.
커밋 메시지만 출력하고 다른 내용은 출력하지 마세요.
"#; "#;
const COMMITLINT_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera mensajes de commit siguiendo @commitlint/config-conventional. const COMMITLINT_SYSTEM_PROMPT_ES: &str = r#"Eres un asistente que genera mensajes de commit siguiendo @commitlint/config-conventional.
@@ -744,8 +945,7 @@ Reglas:
3. El asunto debe tener 4-100 caracteres 3. El asunto debe tener 4-100 caracteres
4. Usa modo imperativo 4. Usa modo imperativo
5. Sé conciso pero descriptivo 5. Sé conciso pero descriptivo
6. Genera SOLO el mensaje de commit, nada más.
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. const COMMITLINT_SYSTEM_PROMPT_FR: &str = r#"Vous êtes un assistant qui génère des messages de commit suivant @commitlint/config-conventional.
@@ -762,8 +962,7 @@ Règles:
3. Le sujet doit avoir 4-100 caractères 3. Le sujet doit avoir 4-100 caractères
4. Utilisez le mode impératif 4. Utilisez le mode impératif
5. Soyez concis mais descriptif 5. Soyez concis mais descriptif
6. Générez SEULEMENT le message de commit, rien d'autre.
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. const COMMITLINT_SYSTEM_PROMPT_DE: &str = r#"Sie sind ein Assistent, der Commit-Nachrichten gemäß @commitlint/config-conventional generiert.
@@ -780,8 +979,7 @@ Regeln:
3. Der Betreff sollte 4-100 Zeichen haben 3. Der Betreff sollte 4-100 Zeichen haben
4. Verwenden Sie den Imperativ 4. Verwenden Sie den Imperativ
5. Seien Sie prägnant aber beschreibend 5. Seien Sie prägnant aber beschreibend
6. Geben Sie NUR die Commit-Nachricht aus, nichts anderes.
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. const TAG_MESSAGE_SYSTEM_PROMPT: &str = r#"You are a helpful assistant that generates git tag annotation messages.

View File

@@ -1,4 +1,4 @@
use super::{create_http_client, LlmProvider}; use super::{LlmProvider, create_http_client};
use anyhow::{Context, Result}; use anyhow::{Context, Result};
use async_trait::async_trait; use async_trait::async_trait;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@@ -9,6 +9,9 @@ pub struct OllamaClient {
base_url: String, base_url: String,
model: String, model: String,
client: reqwest::Client, client: reqwest::Client,
max_tokens: u32,
temperature: f32,
top_p: Option<f32>,
} }
#[derive(Debug, Serialize)] #[derive(Debug, Serialize)]
@@ -47,69 +50,88 @@ struct ModelInfo {
impl OllamaClient { impl OllamaClient {
/// Create new Ollama client /// Create new Ollama client
pub fn new(base_url: &str, model: &str) -> Self { pub fn new(base_url: &str, model: &str) -> Self {
let client = create_http_client(Duration::from_secs(120)) let client =
.expect("Failed to create HTTP client"); create_http_client(Duration::from_secs(120)).expect("Failed to create HTTP client");
Self { Self {
base_url: base_url.trim_end_matches('/').to_string(), base_url: base_url.trim_end_matches('/').to_string(),
model: model.to_string(), model: model.to_string(),
client, client,
max_tokens: 500,
temperature: 0.7,
top_p: None,
} }
} }
/// Set timeout /// Set timeout
pub fn with_timeout(mut self, timeout: Duration) -> Self { pub fn with_timeout(mut self, timeout: Duration) -> Self {
self.client = create_http_client(timeout) self.client = create_http_client(timeout).expect("Failed to create HTTP client");
.expect("Failed to create HTTP client"); self
}
pub fn with_max_tokens(mut self, max_tokens: u32) -> Self {
self.max_tokens = max_tokens;
self
}
pub fn with_temperature(mut self, temperature: f32) -> Self {
self.temperature = temperature;
self
}
pub fn with_top_p(mut self, top_p: f32) -> Self {
self.top_p = Some(top_p);
self self
} }
/// List available models /// List available models
pub async fn list_models(&self) -> Result<Vec<String>> { pub async fn list_models(&self) -> Result<Vec<String>> {
let url = format!("{}/api/tags", self.base_url); let url = format!("{}/api/tags", self.base_url);
let response = self.client let response = self
.client
.get(&url) .get(&url)
.send() .send()
.await .await
.context("Failed to list Ollama models")?; .context("Failed to list Ollama models")?;
if !response.status().is_success() { if !response.status().is_success() {
let status = response.status(); let status = response.status();
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
anyhow::bail!("Ollama API error: {} - {}", status, text); anyhow::bail!("Ollama API error: {} - {}", status, text);
} }
let result: ListModelsResponse = response let result: ListModelsResponse = response
.json() .json()
.await .await
.context("Failed to parse Ollama response")?; .context("Failed to parse Ollama response")?;
Ok(result.models.into_iter().map(|m| m.name).collect()) Ok(result.models.into_iter().map(|m| m.name).collect())
} }
/// Pull a model /// Pull a model
pub async fn pull_model(&self, model: &str) -> Result<()> { pub async fn pull_model(&self, model: &str) -> Result<()> {
let url = format!("{}/api/pull", self.base_url); let url = format!("{}/api/pull", self.base_url);
let request = serde_json::json!({ let request = serde_json::json!({
"name": model, "name": model,
"stream": false, "stream": false,
}); });
let response = self.client let response = self
.client
.post(&url) .post(&url)
.json(&request) .json(&request)
.send() .send()
.await .await
.context("Failed to pull Ollama model")?; .context("Failed to pull Ollama model")?;
if !response.status().is_success() { if !response.status().is_success() {
let status = response.status(); let status = response.status();
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
anyhow::bail!("Ollama pull error: {} - {}", status, text); anyhow::bail!("Ollama pull error: {} - {}", status, text);
} }
Ok(()) Ok(())
} }
@@ -130,48 +152,49 @@ impl LlmProvider for OllamaClient {
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> { async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
let url = format!("{}/api/generate", self.base_url); let url = format!("{}/api/generate", self.base_url);
let system = if system.is_empty() { let system = if system.is_empty() {
None None
} else { } else {
Some(system.to_string()) Some(system.to_string())
}; };
let request = GenerateRequest { let request = GenerateRequest {
model: self.model.clone(), model: self.model.clone(),
prompt: user.to_string(), prompt: user.to_string(),
system, system,
stream: false, stream: false,
options: GenerationOptions { options: GenerationOptions {
temperature: Some(0.7), temperature: Some(self.temperature),
num_predict: Some(500), num_predict: Some(self.max_tokens),
}, },
}; };
let response = self.client let response = self
.client
.post(&url) .post(&url)
.json(&request) .json(&request)
.send() .send()
.await .await
.context("Failed to send request to Ollama")?; .context("Failed to send request to Ollama")?;
if !response.status().is_success() { if !response.status().is_success() {
let status = response.status(); let status = response.status();
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
anyhow::bail!("Ollama API error: {} - {}", status, text); anyhow::bail!("Ollama API error: {} - {}", status, text);
} }
let result: GenerateResponse = response let result: GenerateResponse = response
.json() .json()
.await .await
.context("Failed to parse Ollama response")?; .context("Failed to parse Ollama response")?;
Ok(result.response.trim().to_string()) Ok(result.response.trim().to_string())
} }
async fn is_available(&self) -> bool { async fn is_available(&self) -> bool {
let url = format!("{}/api/tags", self.base_url); let url = format!("{}/api/tags", self.base_url);
match self.client.get(&url).send().await { match self.client.get(&url).send().await {
Ok(response) => response.status().is_success(), Ok(response) => response.status().is_success(),
Err(_) => false, Err(_) => false,

View File

@@ -1,15 +1,23 @@
use super::{create_http_client, LlmProvider}; use super::thinking::ThinkingStateManager;
use anyhow::{bail, Context, Result}; use super::{LlmProvider, create_http_client};
use anyhow::{Context, Result, bail};
use async_trait::async_trait; use async_trait::async_trait;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
/// OpenAI API client /// OpenAI API client with o-series reasoning support
pub struct OpenAiClient { pub struct OpenAiClient {
base_url: String, base_url: String,
api_key: String, api_key: String,
model: String, model: String,
client: reqwest::Client, client: reqwest::Client,
thinking_enabled: bool,
reasoning_effort: Option<String>,
max_tokens: u32,
temperature: f32,
top_p: Option<f32>,
thinking_state: Option<Arc<ThinkingStateManager>>,
} }
#[derive(Debug, Serialize)] #[derive(Debug, Serialize)]
@@ -20,10 +28,14 @@ struct ChatCompletionRequest {
max_tokens: Option<u32>, max_tokens: Option<u32>,
#[serde(skip_serializing_if = "Option::is_none")] #[serde(skip_serializing_if = "Option::is_none")]
temperature: Option<f32>, temperature: Option<f32>,
#[serde(skip_serializing_if = "Option::is_none")]
top_p: Option<f32>,
#[serde(skip_serializing_if = "Option::is_none")]
reasoning_effort: Option<String>,
stream: bool, stream: bool,
} }
#[derive(Debug, Serialize, Deserialize)] #[derive(Debug, Serialize, Deserialize, Clone)]
struct Message { struct Message {
role: String, role: String,
content: String, content: String,
@@ -39,6 +51,28 @@ struct Choice {
message: Message, message: Message,
} }
// --- Streaming response structures ---
#[derive(Debug, Deserialize)]
struct StreamChunk {
choices: Vec<StreamChoice>,
}
#[derive(Debug, Deserialize)]
struct StreamChoice {
delta: StreamDelta,
#[serde(default)]
finish_reason: Option<String>,
}
#[derive(Debug, Deserialize, Default)]
struct StreamDelta {
#[serde(default)]
content: Option<String>,
#[serde(default)]
reasoning_content: Option<String>,
}
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct ErrorResponse { struct ErrorResponse {
error: ApiError, error: ApiError,
@@ -55,57 +89,91 @@ impl OpenAiClient {
/// Create new OpenAI client /// Create new OpenAI client
pub fn new(base_url: &str, api_key: &str, model: &str) -> Result<Self> { pub fn new(base_url: &str, api_key: &str, model: &str) -> Result<Self> {
let client = create_http_client(Duration::from_secs(60))?; let client = create_http_client(Duration::from_secs(60))?;
Ok(Self { Ok(Self {
base_url: base_url.trim_end_matches('/').to_string(), base_url: base_url.trim_end_matches('/').to_string(),
api_key: api_key.to_string(), api_key: api_key.to_string(),
model: model.to_string(), model: model.to_string(),
client, client,
thinking_enabled: false,
reasoning_effort: None,
max_tokens: 500,
temperature: 0.7,
top_p: None,
thinking_state: None,
}) })
} }
/// Set timeout
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> { pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
self.client = create_http_client(timeout)?; self.client = create_http_client(timeout)?;
Ok(self) Ok(self)
} }
/// List available models pub fn with_thinking(mut self, enabled: bool) -> Self {
self.thinking_enabled = enabled;
self
}
pub fn with_reasoning_effort(mut self, effort: Option<String>) -> Self {
self.reasoning_effort = effort;
self
}
pub fn with_max_tokens(mut self, max_tokens: u32) -> Self {
self.max_tokens = max_tokens;
self
}
pub fn with_temperature(mut self, temperature: f32) -> Self {
self.temperature = temperature;
self
}
pub fn with_top_p(mut self, top_p: f32) -> Self {
self.top_p = Some(top_p);
self
}
pub fn with_thinking_state(mut self, state: Arc<ThinkingStateManager>) -> Self {
self.thinking_state = Some(state);
self
}
pub async fn list_models(&self) -> Result<Vec<String>> { pub async fn list_models(&self) -> Result<Vec<String>> {
let url = format!("{}/models", self.base_url); let url = format!("{}/models", self.base_url);
let response = self.client let response = self
.client
.get(&url) .get(&url)
.header("Authorization", format!("Bearer {}", self.api_key)) .header("Authorization", format!("Bearer {}", self.api_key))
.send() .send()
.await .await
.context("Failed to list OpenAI models")?; .context("Failed to list OpenAI models")?;
if !response.status().is_success() { if !response.status().is_success() {
let status = response.status(); let status = response.status();
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
bail!("OpenAI API error: {} - {}", status, text); bail!("OpenAI API error: {} - {}", status, text);
} }
#[derive(Deserialize)] #[derive(Deserialize)]
struct ModelsResponse { struct ModelsResponse {
data: Vec<Model>, data: Vec<Model>,
} }
#[derive(Deserialize)] #[derive(Deserialize)]
struct Model { struct Model {
id: String, id: String,
} }
let result: ModelsResponse = response let result: ModelsResponse = response
.json() .json()
.await .await
.context("Failed to parse OpenAI response")?; .context("Failed to parse OpenAI response")?;
Ok(result.data.into_iter().map(|m| m.id).collect()) Ok(result.data.into_iter().map(|m| m.id).collect())
} }
/// Validate API key
pub async fn validate_key(&self) -> Result<bool> { pub async fn validate_key(&self) -> Result<bool> {
match self.list_models().await { match self.list_models().await {
Ok(_) => Ok(true), Ok(_) => Ok(true),
@@ -124,32 +192,30 @@ impl OpenAiClient {
#[async_trait] #[async_trait]
impl LlmProvider for OpenAiClient { impl LlmProvider for OpenAiClient {
async fn generate(&self, prompt: &str) -> Result<String> { async fn generate(&self, prompt: &str) -> Result<String> {
let messages = vec![ let messages = vec![Message {
Message { role: "user".to_string(),
role: "user".to_string(), content: prompt.to_string(),
content: prompt.to_string(), }];
},
]; self.chat_completion_with_retry(messages).await
self.chat_completion(messages).await
} }
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> { async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
let mut messages = vec![]; let mut messages = vec![];
if !system.is_empty() { if !system.is_empty() {
messages.push(Message { messages.push(Message {
role: "system".to_string(), role: "system".to_string(),
content: system.to_string(), content: system.to_string(),
}); });
} }
messages.push(Message { messages.push(Message {
role: "user".to_string(), role: "user".to_string(),
content: user.to_string(), content: user.to_string(),
}); });
self.chat_completion(messages).await self.chat_completion_with_retry(messages).await
} }
async fn is_available(&self) -> bool { async fn is_available(&self) -> bool {
@@ -162,18 +228,63 @@ impl LlmProvider for OpenAiClient {
} }
impl OpenAiClient { impl OpenAiClient {
async fn chat_completion_with_retry(&self, messages: Vec<Message>) -> Result<String> {
let mut last_error = None;
for attempt in 1..=3 {
match self.chat_completion(messages.clone()).await {
Ok(result) => return Ok(result),
Err(e) => {
let err_msg = e.to_string();
let is_retryable = err_msg.contains("timeout")
|| err_msg.contains("connection")
|| err_msg.contains("temporary")
|| err_msg.contains("5")
&& (err_msg.contains("500")
|| err_msg.contains("502")
|| err_msg.contains("503")
|| err_msg.contains("504"));
if !is_retryable || attempt == 3 {
last_error = Some(e);
break;
}
tokio::time::sleep(Duration::from_millis(500 * 2u64.pow(attempt - 1))).await;
}
}
}
Err(last_error.unwrap_or_else(|| anyhow::anyhow!("Request failed after retries")))
}
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> { async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
if self.thinking_enabled {
self.streaming_chat_completion(messages).await
} else {
self.non_streaming_chat_completion(messages).await
}
}
async fn non_streaming_chat_completion(&self, messages: Vec<Message>) -> Result<String> {
let url = format!("{}/chat/completions", self.base_url); let url = format!("{}/chat/completions", self.base_url);
let request = ChatCompletionRequest { let request = ChatCompletionRequest {
model: self.model.clone(), model: self.model.clone(),
messages, messages,
max_tokens: Some(500), max_tokens: Some(self.max_tokens),
temperature: Some(0.7), temperature: Some(self.temperature),
top_p: self.top_p,
reasoning_effort: if is_reasoning_model(&self.model) {
Some("none".to_string())
} else {
None
},
stream: false, stream: false,
}; };
let response = self.client let response = self
.client
.post(&url) .post(&url)
.header("Authorization", format!("Bearer {}", self.api_key)) .header("Authorization", format!("Bearer {}", self.api_key))
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
@@ -181,31 +292,166 @@ impl OpenAiClient {
.send() .send()
.await .await
.context("Failed to send request to OpenAI")?; .context("Failed to send request to OpenAI")?;
let status = response.status(); let status = response.status();
if !status.is_success() { if !status.is_success() {
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
// Try to parse error
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) { 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: {} ({})",
error.error.message,
error.error.error_type
);
} }
bail!("OpenAI API error: {} - {}", status, text); bail!("OpenAI API error: {} - {}", status, text);
} }
let result: ChatCompletionResponse = response let result: ChatCompletionResponse = response
.json() .json()
.await .await
.context("Failed to parse OpenAI response")?; .context("Failed to parse OpenAI response")?;
result.choices result
.choices
.into_iter() .into_iter()
.next() .next()
.map(|c| c.message.content.trim().to_string()) .map(|c| c.message.content.trim().to_string())
.filter(|s| !s.is_empty())
.ok_or_else(|| anyhow::anyhow!("No response from OpenAI")) .ok_or_else(|| anyhow::anyhow!("No response from OpenAI"))
} }
/// Streaming request for reasoning mode, filters reasoning_content from output
async fn streaming_chat_completion(&self, messages: Vec<Message>) -> Result<String> {
let url = format!("{}/chat/completions", self.base_url);
// For reasoning/thinking mode, omit temperature and top_p
let request = ChatCompletionRequest {
model: self.model.clone(),
messages,
max_tokens: Some(self.max_tokens),
temperature: None,
top_p: None,
reasoning_effort: self.reasoning_effort.clone(),
stream: true,
};
let response = self
.client
.post(&url)
.header("Authorization", format!("Bearer {}", self.api_key))
.header("Content-Type", "application/json")
.header("Accept", "text/event-stream")
.json(&request)
.send()
.await
.context("Failed to send streaming request to OpenAI")?;
let status = response.status();
if !status.is_success() {
let text = response.text().await.unwrap_or_default();
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 mut content_buffer = String::new();
let mut has_reasoning = false;
let mut has_content = false;
let thinking_state = self.thinking_state.as_ref();
let mut byte_stream = response.bytes_stream();
let mut line_buffer = String::new();
use futures_util::StreamExt;
while let Some(chunk) = byte_stream.next().await {
let chunk = chunk.context("Failed to read streaming response chunk")?;
let chunk_str =
String::from_utf8(chunk.to_vec()).context("Invalid UTF-8 in stream chunk")?;
line_buffer.push_str(&chunk_str);
while let Some(line_end) = line_buffer.find('\n') {
let line = line_buffer[..line_end].trim().to_string();
line_buffer = line_buffer[line_end + 1..].to_string();
if line.is_empty() {
continue;
}
if line == "data: [DONE]" {
break;
}
if let Some(json_str) = line.strip_prefix("data: ") {
if let Ok(chunk) = serde_json::from_str::<StreamChunk>(json_str) {
for choice in &chunk.choices {
// Handle reasoning_content (o-series)
if let Some(ref reasoning) = choice.delta.reasoning_content
&& !reasoning.is_empty()
{
if !has_reasoning {
has_reasoning = true;
if let Some(state) = thinking_state {
state.start_thinking();
}
}
continue;
}
// Handle content
if let Some(ref content) = choice.delta.content
&& !content.is_empty()
{
if has_reasoning
&& !has_content
&& let Some(state) = thinking_state
{
state.end_thinking();
}
has_content = true;
content_buffer.push_str(content);
}
}
}
}
}
}
if let Some(state) = thinking_state {
state.end_thinking();
}
let result = content_buffer.trim().to_string();
if result.is_empty() {
if has_reasoning && !has_content {
bail!(
"OpenAI returned reasoning content but no final answer. \
The model may have entered an incomplete reasoning state. \
Please try again or disable thinking mode."
);
}
bail!(
"No response from OpenAI. \
If thinking mode is enabled, try disabling it or ensure the model supports reasoning."
);
}
Ok(result)
}
} }
/// Azure OpenAI client (extends OpenAI with Azure-specific config) /// Azure OpenAI client (extends OpenAI with Azure-specific config)
@@ -215,24 +461,30 @@ pub struct AzureOpenAiClient {
deployment: String, deployment: String,
api_version: String, api_version: String,
client: reqwest::Client, client: reqwest::Client,
thinking_enabled: bool,
reasoning_effort: Option<String>,
max_tokens: u32,
temperature: f32,
top_p: Option<f32>,
thinking_state: Option<Arc<ThinkingStateManager>>,
} }
impl AzureOpenAiClient { impl AzureOpenAiClient {
/// Create new Azure OpenAI client pub fn new(endpoint: &str, api_key: &str, deployment: &str, api_version: &str) -> Result<Self> {
pub fn new(
endpoint: &str,
api_key: &str,
deployment: &str,
api_version: &str,
) -> Result<Self> {
let client = create_http_client(Duration::from_secs(60))?; let client = create_http_client(Duration::from_secs(60))?;
Ok(Self { Ok(Self {
endpoint: endpoint.trim_end_matches('/').to_string(), endpoint: endpoint.trim_end_matches('/').to_string(),
api_key: api_key.to_string(), api_key: api_key.to_string(),
deployment: deployment.to_string(), deployment: deployment.to_string(),
api_version: api_version.to_string(), api_version: api_version.to_string(),
client, client,
thinking_enabled: false,
reasoning_effort: None,
max_tokens: 500,
temperature: 0.7,
top_p: None,
thinking_state: None,
}) })
} }
@@ -241,16 +493,19 @@ impl AzureOpenAiClient {
"{}/openai/deployments/{}/chat/completions?api-version={}", "{}/openai/deployments/{}/chat/completions?api-version={}",
self.endpoint, self.deployment, self.api_version self.endpoint, self.deployment, self.api_version
); );
let request = ChatCompletionRequest { let request = ChatCompletionRequest {
model: self.deployment.clone(), model: self.deployment.clone(),
messages, messages,
max_tokens: Some(500), max_tokens: Some(self.max_tokens),
temperature: Some(0.7), temperature: Some(self.temperature),
top_p: self.top_p,
reasoning_effort: self.reasoning_effort.clone(),
stream: false, stream: false,
}; };
let response = self.client let response = self
.client
.post(&url) .post(&url)
.header("api-key", &self.api_key) .header("api-key", &self.api_key)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
@@ -258,22 +513,24 @@ impl AzureOpenAiClient {
.send() .send()
.await .await
.context("Failed to send request to Azure OpenAI")?; .context("Failed to send request to Azure OpenAI")?;
if !response.status().is_success() { if !response.status().is_success() {
let status = response.status(); let status = response.status();
let text = response.text().await.unwrap_or_default(); let text = response.text().await.unwrap_or_default();
bail!("Azure OpenAI API error: {} - {}", status, text); bail!("Azure OpenAI API error: {} - {}", status, text);
} }
let result: ChatCompletionResponse = response let result: ChatCompletionResponse = response
.json() .json()
.await .await
.context("Failed to parse Azure OpenAI response")?; .context("Failed to parse Azure OpenAI response")?;
result.choices result
.choices
.into_iter() .into_iter()
.next() .next()
.map(|c| c.message.content.trim().to_string()) .map(|c| c.message.content.trim().to_string())
.filter(|s| !s.is_empty())
.ok_or_else(|| anyhow::anyhow!("No response from Azure OpenAI")) .ok_or_else(|| anyhow::anyhow!("No response from Azure OpenAI"))
} }
} }
@@ -281,41 +538,38 @@ impl AzureOpenAiClient {
#[async_trait] #[async_trait]
impl LlmProvider for AzureOpenAiClient { impl LlmProvider for AzureOpenAiClient {
async fn generate(&self, prompt: &str) -> Result<String> { async fn generate(&self, prompt: &str) -> Result<String> {
let messages = vec![ let messages = vec![Message {
Message { role: "user".to_string(),
role: "user".to_string(), content: prompt.to_string(),
content: prompt.to_string(), }];
},
];
self.chat_completion(messages).await self.chat_completion(messages).await
} }
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> { async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
let mut messages = vec![]; let mut messages = vec![];
if !system.is_empty() { if !system.is_empty() {
messages.push(Message { messages.push(Message {
role: "system".to_string(), role: "system".to_string(),
content: system.to_string(), content: system.to_string(),
}); });
} }
messages.push(Message { messages.push(Message {
role: "user".to_string(), role: "user".to_string(),
content: user.to_string(), content: user.to_string(),
}); });
self.chat_completion(messages).await self.chat_completion(messages).await
} }
async fn is_available(&self) -> bool { async fn is_available(&self) -> bool {
// Simple check - try to make a minimal request
let url = format!( let url = format!(
"{}/openai/deployments/{}/chat/completions?api-version={}", "{}/openai/deployments/{}/chat/completions?api-version={}",
self.endpoint, self.deployment, self.api_version self.endpoint, self.deployment, self.api_version
); );
let request = ChatCompletionRequest { let request = ChatCompletionRequest {
model: self.deployment.clone(), model: self.deployment.clone(),
messages: vec![Message { messages: vec![Message {
@@ -324,10 +578,13 @@ impl LlmProvider for AzureOpenAiClient {
}], }],
max_tokens: Some(5), max_tokens: Some(5),
temperature: Some(0.0), temperature: Some(0.0),
top_p: None,
reasoning_effort: None,
stream: false, stream: false,
}; };
match self.client match self
.client
.post(&url) .post(&url)
.header("api-key", &self.api_key) .header("api-key", &self.api_key)
.json(&request) .json(&request)
@@ -343,3 +600,60 @@ impl LlmProvider for AzureOpenAiClient {
"azure-openai" "azure-openai"
} }
} }
/// Available OpenAI models (including o-series with reasoning)
pub const OPENAI_MODELS: &[&str] = &[
"o4-mini",
"o3",
"o3-mini",
"o1",
"o1-mini",
"o1-pro",
"gpt-4.1",
"gpt-4.1-mini",
"gpt-4.1-nano",
"gpt-4o",
"gpt-4o-mini",
"gpt-4-turbo",
"gpt-4",
"gpt-3.5-turbo",
];
pub fn is_valid_model(model: &str) -> bool {
OPENAI_MODELS.contains(&model)
}
fn is_reasoning_model(model: &str) -> bool {
model.starts_with("o")
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_model_validation_o_series() {
assert!(is_valid_model("o4-mini"));
assert!(is_valid_model("o3"));
assert!(is_valid_model("o1"));
assert!(is_valid_model("gpt-4o"));
assert!(is_valid_model("gpt-3.5-turbo"));
assert!(!is_valid_model("invalid-model"));
}
#[test]
fn test_stream_delta_reasoning_parsing() {
let json = r#"{"content":null,"reasoning_content":"Let me think..."}"#;
let delta: StreamDelta = serde_json::from_str(json).unwrap();
assert!(delta.content.is_none());
assert_eq!(delta.reasoning_content, Some("Let me think...".to_string()));
}
#[test]
fn test_stream_delta_content_parsing() {
let json = r#"{"content":"Hello","reasoning_content":null}"#;
let delta: StreamDelta = serde_json::from_str(json).unwrap();
assert_eq!(delta.content, Some("Hello".to_string()));
assert!(delta.reasoning_content.is_none());
}
}

View File

@@ -1,257 +1,286 @@
use super::{create_http_client, LlmProvider}; use super::{LlmProvider, create_http_client};
use anyhow::{bail, Context, Result}; use anyhow::{Context, Result, bail};
use async_trait::async_trait; use async_trait::async_trait;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::time::Duration; use std::time::Duration;
/// OpenRouter API client /// OpenRouter API client
pub struct OpenRouterClient { pub struct OpenRouterClient {
base_url: String, base_url: String,
api_key: String, api_key: String,
model: String, model: String,
client: reqwest::Client, client: reqwest::Client,
} max_tokens: u32,
temperature: f32,
#[derive(Debug, Serialize)] top_p: Option<f32>,
struct ChatCompletionRequest { }
model: String,
messages: Vec<Message>, #[derive(Debug, Serialize)]
#[serde(skip_serializing_if = "Option::is_none")] struct ChatCompletionRequest {
max_tokens: Option<u32>, model: String,
#[serde(skip_serializing_if = "Option::is_none")] messages: Vec<Message>,
temperature: Option<f32>, #[serde(skip_serializing_if = "Option::is_none")]
stream: bool, max_tokens: Option<u32>,
} #[serde(skip_serializing_if = "Option::is_none")]
temperature: Option<f32>,
#[derive(Debug, Serialize, Deserialize)] stream: bool,
struct Message { }
role: String,
content: String, #[derive(Debug, Serialize, Deserialize)]
} struct Message {
role: String,
#[derive(Debug, Deserialize)] content: String,
struct ChatCompletionResponse { }
choices: Vec<Choice>,
} #[derive(Debug, Deserialize)]
struct ChatCompletionResponse {
#[derive(Debug, Deserialize)] choices: Vec<Choice>,
struct Choice { }
message: Message,
} #[derive(Debug, Deserialize)]
struct Choice {
#[derive(Debug, Deserialize)] message: Message,
struct ErrorResponse { }
error: ApiError,
} #[derive(Debug, Deserialize)]
struct ErrorResponse {
#[derive(Debug, Deserialize)] error: ApiError,
struct ApiError { }
message: String,
#[serde(rename = "type")] #[derive(Debug, Deserialize)]
error_type: String, struct ApiError {
} message: String,
#[serde(rename = "type")]
impl OpenRouterClient { error_type: String,
/// Create new OpenRouter client }
pub fn new(api_key: &str, model: &str) -> Result<Self> {
let client = create_http_client(Duration::from_secs(60))?; impl OpenRouterClient {
/// Create new OpenRouter client
Ok(Self { pub fn new(api_key: &str, model: &str) -> Result<Self> {
base_url: "https://openrouter.ai/api/v1".to_string(), let client = create_http_client(Duration::from_secs(60))?;
api_key: api_key.to_string(),
model: model.to_string(), Ok(Self {
client, 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 max_tokens: 500,
pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> { temperature: 0.7,
let client = create_http_client(Duration::from_secs(60))?; top_p: None,
})
Ok(Self { }
base_url: base_url.trim_end_matches('/').to_string(),
api_key: api_key.to_string(), /// Create with custom base URL
model: model.to_string(), pub fn with_base_url(api_key: &str, model: &str, base_url: &str) -> Result<Self> {
client, let client = create_http_client(Duration::from_secs(60))?;
})
} Ok(Self {
base_url: base_url.trim_end_matches('/').to_string(),
/// Set timeout api_key: api_key.to_string(),
pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> { model: model.to_string(),
self.client = create_http_client(timeout)?; client,
Ok(self) max_tokens: 500,
} temperature: 0.7,
top_p: None,
/// List available models })
pub async fn list_models(&self) -> Result<Vec<String>> { }
let url = format!("{}/models", self.base_url);
/// Set timeout
let response = self.client pub fn with_timeout(mut self, timeout: Duration) -> Result<Self> {
.get(&url) self.client = create_http_client(timeout)?;
.header("Authorization", format!("Bearer {}", self.api_key)) Ok(self)
.header("HTTP-Referer", "https://quicommit.dev") }
.header("X-Title", "QuiCommit")
.send() pub fn with_max_tokens(mut self, max_tokens: u32) -> Self {
.await self.max_tokens = max_tokens;
.context("Failed to list OpenRouter models")?; self
}
if !response.status().is_success() {
let status = response.status(); pub fn with_temperature(mut self, temperature: f32) -> Self {
let text = response.text().await.unwrap_or_default(); self.temperature = temperature;
bail!("OpenRouter API error: {} - {}", status, text); self
} }
#[derive(Deserialize)] pub fn with_top_p(mut self, top_p: f32) -> Self {
struct ModelsResponse { self.top_p = Some(top_p);
data: Vec<Model>, self
} }
#[derive(Deserialize)] /// List available models
struct Model { pub async fn list_models(&self) -> Result<Vec<String>> {
id: String, let url = format!("{}/models", self.base_url);
}
let response = self
let result: ModelsResponse = response .client
.json() .get(&url)
.await .header("Authorization", format!("Bearer {}", self.api_key))
.context("Failed to parse OpenRouter response")?; .header("HTTP-Referer", "https://quicommit.dev")
.header("X-Title", "QuiCommit")
Ok(result.data.into_iter().map(|m| m.id).collect()) .send()
} .await
.context("Failed to list OpenRouter models")?;
/// Validate API key
pub async fn validate_key(&self) -> Result<bool> { if !response.status().is_success() {
match self.list_models().await { let status = response.status();
Ok(_) => Ok(true), let text = response.text().await.unwrap_or_default();
Err(e) => { bail!("OpenRouter API error: {} - {}", status, text);
let err_str = e.to_string(); }
if err_str.contains("401") || err_str.contains("Unauthorized") {
Ok(false) #[derive(Deserialize)]
} else { struct ModelsResponse {
Err(e) data: Vec<Model>,
} }
}
} #[derive(Deserialize)]
} struct Model {
} id: String,
}
#[async_trait]
impl LlmProvider for OpenRouterClient { let result: ModelsResponse = response
async fn generate(&self, prompt: &str) -> Result<String> { .json()
let messages = vec![ .await
Message { .context("Failed to parse OpenRouter response")?;
role: "user".to_string(),
content: prompt.to_string(), Ok(result.data.into_iter().map(|m| m.id).collect())
}, }
];
/// Validate API key
self.chat_completion(messages).await pub async fn validate_key(&self) -> Result<bool> {
} match self.list_models().await {
Ok(_) => Ok(true),
async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> { Err(e) => {
let mut messages = vec![]; let err_str = e.to_string();
if err_str.contains("401") || err_str.contains("Unauthorized") {
if !system.is_empty() { Ok(false)
messages.push(Message { } else {
role: "system".to_string(), Err(e)
content: system.to_string(), }
}); }
} }
}
messages.push(Message { }
role: "user".to_string(),
content: user.to_string(), #[async_trait]
}); impl LlmProvider for OpenRouterClient {
async fn generate(&self, prompt: &str) -> Result<String> {
self.chat_completion(messages).await let messages = vec![Message {
} role: "user".to_string(),
content: prompt.to_string(),
async fn is_available(&self) -> bool { }];
self.validate_key().await.unwrap_or(false)
} self.chat_completion(messages).await
}
fn name(&self) -> &str {
"openrouter" async fn generate_with_system(&self, system: &str, user: &str) -> Result<String> {
} let mut messages = vec![];
}
if !system.is_empty() {
impl OpenRouterClient { messages.push(Message {
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> { role: "system".to_string(),
let url = format!("{}/chat/completions", self.base_url); content: system.to_string(),
});
let request = ChatCompletionRequest { }
model: self.model.clone(),
messages, messages.push(Message {
max_tokens: Some(500), role: "user".to_string(),
temperature: Some(0.7), content: user.to_string(),
stream: false, });
};
self.chat_completion(messages).await
let response = self.client }
.post(&url)
.header("Authorization", format!("Bearer {}", self.api_key)) async fn is_available(&self) -> bool {
.header("Content-Type", "application/json") self.validate_key().await.unwrap_or(false)
.header("HTTP-Referer", "https://quicommit.dev") }
.header("X-Title", "QuiCommit")
.json(&request) fn name(&self) -> &str {
.send() "openrouter"
.await }
.context("Failed to send request to OpenRouter")?; }
let status = response.status(); impl OpenRouterClient {
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
if !status.is_success() { let url = format!("{}/chat/completions", self.base_url);
let text = response.text().await.unwrap_or_default();
let request = ChatCompletionRequest {
// Try to parse error model: self.model.clone(),
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) { messages,
bail!("OpenRouter API error: {} ({})", error.error.message, error.error.error_type); max_tokens: Some(self.max_tokens),
} temperature: Some(self.temperature),
stream: false,
bail!("OpenRouter API error: {} - {}", status, text); };
}
let response = self
let result: ChatCompletionResponse = response .client
.json() .post(&url)
.await .header("Authorization", format!("Bearer {}", self.api_key))
.context("Failed to parse OpenRouter response")?; .header("Content-Type", "application/json")
.header("HTTP-Referer", "https://quicommit.dev")
result.choices .header("X-Title", "QuiCommit")
.into_iter() .json(&request)
.next() .send()
.map(|c| c.message.content.trim().to_string()) .await
.ok_or_else(|| anyhow::anyhow!("No response from OpenRouter")) .context("Failed to send request to OpenRouter")?;
}
} let status = response.status();
/// Popular OpenRouter models if !status.is_success() {
pub const OPENROUTER_MODELS: &[&str] = &[ let text = response.text().await.unwrap_or_default();
"openai/gpt-3.5-turbo",
"openai/gpt-4", // Try to parse error
"openai/gpt-4-turbo", if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
"anthropic/claude-3-opus", bail!(
"anthropic/claude-3-sonnet", "OpenRouter API error: {} ({})",
"anthropic/claude-3-haiku", error.error.message,
"google/gemini-pro", error.error.error_type
"meta-llama/llama-2-70b-chat", );
"mistralai/mixtral-8x7b-instruct", }
"01-ai/yi-34b-chat",
]; bail!("OpenRouter API error: {} - {}", status, text);
}
/// Check if a model name is valid
pub fn is_valid_model(_model: &str) -> bool { let result: ChatCompletionResponse = response
// Since OpenRouter supports many models, we'll allow any model name .json()
// but provide some popular ones as suggestions .await
true .context("Failed to parse OpenRouter response")?;
}
result
#[cfg(test)] .choices
mod tests { .into_iter()
use super::*; .next()
.map(|c| c.message.content.trim().to_string())
#[test] .ok_or_else(|| anyhow::anyhow!("No response from OpenRouter"))
fn test_model_validation() { }
assert!(is_valid_model("openai/gpt-4")); }
assert!(is_valid_model("custom/model"));
} /// 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"));
}
}

151
src/llm/thinking.rs Normal file
View File

@@ -0,0 +1,151 @@
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>;
/// 创建带有默认控制台输出的思考状态管理器
/// 在思考开始时打印 "thinking...",在思考结束时清除该标识
pub fn create_console_thinking_state() -> SharedThinkingState {
Arc::new(
ThinkingStateManager::new()
.on_thinking_start(|| {
eprint!("\rthinking...");
})
.on_thinking_end(|| {
eprint!("\r \r");
}),
)
}
#[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());
}
}

View File

@@ -1,23 +1,17 @@
#![allow(dead_code)]
use anyhow::Result; use anyhow::Result;
use clap::{Parser, Subcommand}; use clap::{Parser, Subcommand};
use std::path::PathBuf; 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
/// ///
/// A powerful tool that helps you generate conventional commits, tags, and changelogs /// A powerful tool that helps you generate conventional commits, tags, and changelogs
/// using AI (LLM APIs or local Ollama models). Manage multiple Git profiles for different /// using AI (LLM APIs or local Ollama models). Manage multiple Git profiles for different
/// work contexts seamlessly. /// work contexts seamlessly.
@@ -69,6 +63,10 @@ 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]
@@ -81,7 +79,7 @@ async fn main() -> Result<()> {
2 => "debug", 2 => "debug",
_ => "trace", _ => "trace",
}; };
tracing_subscriber::fmt() tracing_subscriber::fmt()
.with_env_filter(log_level) .with_env_filter(log_level)
.with_target(false) .with_target(false)
@@ -98,5 +96,6 @@ async fn main() -> Result<()> {
Commands::Changelog(cmd) => cmd.execute(config_path).await, Commands::Changelog(cmd) => cmd.execute(config_path).await,
Commands::Profile(cmd) => cmd.execute(config_path).await, Commands::Profile(cmd) => cmd.execute(config_path).await,
Commands::Config(cmd) => cmd.execute(config_path).await, Commands::Config(cmd) => cmd.execute(config_path).await,
Commands::Credential(cmd) => cmd.execute(config_path).await,
} }
} }

View File

@@ -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;
@@ -18,63 +18,62 @@ pub fn encrypt(data: &[u8], password: &str) -> Result<String> {
rand::thread_rng().fill(&mut salt); rand::thread_rng().fill(&mut salt);
let mut nonce_bytes = [0u8; NONCE_LEN]; let mut nonce_bytes = [0u8; NONCE_LEN];
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
.encrypt(nonce, data) .encrypt(nonce, data)
.map_err(|e| anyhow::anyhow!("Encryption failed: {:?}", e))?; .map_err(|e| anyhow::anyhow!("Encryption failed: {:?}", e))?;
// Combine salt + nonce + encrypted data // Combine salt + nonce + encrypted data
let mut result = Vec::with_capacity(SALT_LEN + NONCE_LEN + encrypted.len()); let mut result = Vec::with_capacity(SALT_LEN + NONCE_LEN + encrypted.len());
result.extend_from_slice(&salt); result.extend_from_slice(&salt);
result.extend_from_slice(&nonce_bytes); result.extend_from_slice(&nonce_bytes);
result.extend_from_slice(&encrypted); result.extend_from_slice(&encrypted);
Ok(BASE64.encode(&result)) Ok(BASE64.encode(&result))
} }
/// 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 {
anyhow::bail!("Invalid encrypted data format"); anyhow::bail!("Invalid encrypted data format");
} }
let salt = &data[..SALT_LEN]; let salt = &data[..SALT_LEN];
let nonce_bytes = &data[SALT_LEN..SALT_LEN + NONCE_LEN]; let nonce_bytes = &data[SALT_LEN..SALT_LEN + NONCE_LEN];
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
.decrypt(nonce, encrypted) .decrypt(nonce, encrypted)
.map_err(|e| anyhow::anyhow!("Decryption failed: {:?}", e))?; .map_err(|e| anyhow::anyhow!("Decryption failed: {:?}", e))?;
Ok(decrypted) Ok(decrypted)
} }
/// 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);
hasher.update(password.as_bytes()); hasher.update(password.as_bytes());
hasher.update(b"quicommit_key_derivation_v1"); hasher.update(b"quicommit_key_derivation_v1");
let hash = hasher.finalize(); let hash = hasher.finalize();
let mut key = [0u8; KEY_LEN]; let mut key = [0u8; KEY_LEN];
key.copy_from_slice(&hash[..KEY_LEN]); key.copy_from_slice(&hash[..KEY_LEN]);
Ok(key) Ok(key)
} }
@@ -97,7 +96,7 @@ pub fn decrypt_from_file(path: &Path, password: &str) -> Result<Vec<u8>> {
pub fn generate_token(length: usize) -> String { pub fn generate_token(length: usize) -> String {
const CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"; const CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
let mut rng = rand::thread_rng(); let mut rng = rand::thread_rng();
(0..length) (0..length)
.map(|_| { .map(|_| {
let idx = rng.gen_range(0..CHARSET.len()); let idx = rng.gen_range(0..CHARSET.len());
@@ -122,10 +121,10 @@ mod tests {
fn test_encrypt_decrypt() { fn test_encrypt_decrypt() {
let data = b"Hello, World!"; let data = b"Hello, World!";
let password = "my_secret_password"; let password = "my_secret_password";
let encrypted = encrypt(data, password).unwrap(); let encrypted = encrypt(data, password).unwrap();
let decrypted = decrypt(&encrypted, password).unwrap(); let decrypted = decrypt(&encrypted, password).unwrap();
assert_eq!(data.to_vec(), decrypted); assert_eq!(data.to_vec(), decrypted);
} }
@@ -133,7 +132,7 @@ mod tests {
fn test_wrong_password() { fn test_wrong_password() {
let data = b"Hello, World!"; let data = b"Hello, World!";
let encrypted = encrypt(data, "correct_password").unwrap(); let encrypted = encrypt(data, "correct_password").unwrap();
assert!(decrypt(&encrypted, "wrong_password").is_err()); assert!(decrypt(&encrypted, "wrong_password").is_err());
} }
} }

View File

@@ -9,15 +9,12 @@ 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).context("Failed to open editor")?;
let edited = edit::edit(&content)
.context("Failed to open editor")?; fs::write(path, &edited).with_context(|| format!("Failed to write file: {:?}", path))?;
fs::write(path, &edited)
.with_context(|| format!("Failed to write file: {:?}", path))?;
Ok(edited) Ok(edited)
} }
@@ -27,11 +24,10 @@ pub fn edit_temp(initial_content: &str, extension: &str) -> Result<String> {
.suffix(&format!(".{}", extension)) .suffix(&format!(".{}", extension))
.tempfile() .tempfile()
.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,10 +37,10 @@ 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") { if let Ok(_code) = which::which("code") {
return "code --wait".to_string(); return "code --wait".to_string();
} }
if let Ok(notepad) = which::which("notepad") { if let Ok(_notepad) = which::which("notepad") {
return "notepad".to_string(); return "notepad".to_string();
} }
"notepad".to_string() "notepad".to_string()
@@ -65,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(())
} }

View File

@@ -10,7 +10,7 @@ pub fn format_conventional_commit(
breaking: bool, breaking: bool,
) -> String { ) -> String {
let mut message = String::new(); let mut message = String::new();
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));
@@ -19,15 +19,15 @@ pub fn format_conventional_commit(
message.push('!'); message.push('!');
} }
message.push_str(&format!(": {}", description)); message.push_str(&format!(": {}", description));
if let Some(b) = body { if let Some(b) = body {
message.push_str(&format!("\n\n{}", b)); message.push_str(&format!("\n\n{}", b));
} }
if let Some(f) = footer { if let Some(f) = footer {
message.push_str(&format!("\n\n{}", f)); message.push_str(&format!("\n\n{}", f));
} }
message message
} }
@@ -41,27 +41,27 @@ pub fn format_commitlint_commit(
references: Option<&[&str]>, references: Option<&[&str]>,
) -> String { ) -> String {
let mut message = String::new(); let mut message = String::new();
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));
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));
} }
} }
if let Some(b) = body { if let Some(b) = body {
message.push_str(&format!("\n\n{}", b)); message.push_str(&format!("\n\n{}", b));
} }
if let Some(f) = footer { if let Some(f) = footer {
message.push_str(&format!("\n\n{}", f)); message.push_str(&format!("\n\n{}", f));
} }
message message
} }
@@ -73,7 +73,7 @@ pub fn wrap_text(text: &str, width: usize) -> String {
/// 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();
message message
.lines() .lines()
.filter(|line| !comment_regex.is_match(line.trim())) .filter(|line| !comment_regex.is_match(line.trim()))
@@ -97,7 +97,7 @@ mod tests {
Some("Closes #123"), Some("Closes #123"),
false, false,
); );
assert!(msg.contains("feat(auth): add login functionality")); assert!(msg.contains("feat(auth): add login functionality"));
assert!(msg.contains("This adds OAuth2 login support.")); assert!(msg.contains("This adds OAuth2 login support."));
assert!(msg.contains("Closes #123")); assert!(msg.contains("Closes #123"));
@@ -113,7 +113,7 @@ mod tests {
Some("BREAKING CHANGE: response format changed"), Some("BREAKING CHANGE: response format changed"),
true, true,
); );
assert!(msg.starts_with("feat!: change API response format")); assert!(msg.starts_with("feat!: change API response format"));
} }
} }

View File

@@ -1,4 +1,4 @@
use anyhow::{bail, Context, Result}; use anyhow::{Context, Result, bail};
use std::env; use std::env;
const SERVICE_NAME: &str = "quicommit"; const SERVICE_NAME: &str = "quicommit";
@@ -78,17 +78,18 @@ impl KeyringManager {
let entry = keyring::Entry::new(SERVICE_NAME, provider) let entry = keyring::Entry::new(SERVICE_NAME, provider)
.context("Failed to create keyring entry")?; .context("Failed to create keyring entry")?;
entry.set_password(api_key) entry
.set_password(api_key)
.context("Failed to store API key")?; .context("Failed to store API key")?;
Ok(()) Ok(())
} }
pub fn get_api_key(&self, provider: &str) -> Result<Option<String>> { pub fn get_api_key(&self, provider: &str) -> Result<Option<String>> {
if let Ok(key) = env::var(ENV_API_KEY) { if let Ok(key) = env::var(ENV_API_KEY)
if !key.is_empty() { && !key.is_empty()
return Ok(Some(key)); {
} return Ok(Some(key));
} }
if !self.is_available() { if !self.is_available() {
@@ -113,7 +114,8 @@ impl KeyringManager {
let entry = keyring::Entry::new(SERVICE_NAME, provider) let entry = keyring::Entry::new(SERVICE_NAME, provider)
.context("Failed to create keyring entry")?; .context("Failed to create keyring entry")?;
entry.delete_credential() entry
.delete_credential()
.context("Failed to delete API key")?; .context("Failed to delete API key")?;
Ok(()) Ok(())
@@ -127,7 +129,13 @@ impl KeyringManager {
format!("{}/{}", PAT_SERVICE_PREFIX, profile_name) format!("{}/{}", PAT_SERVICE_PREFIX, profile_name)
} }
pub fn store_pat(&self, profile_name: &str, user_email: &str, service: &str, token: &str) -> Result<()> { pub fn store_pat(
&self,
profile_name: &str,
user_email: &str,
service: &str,
token: &str,
) -> Result<()> {
if !self.is_available() { if !self.is_available() {
bail!("Keyring is not available on this system"); bail!("Keyring is not available on this system");
} }
@@ -138,15 +146,19 @@ impl KeyringManager {
let entry = keyring::Entry::new(&keyring_service, &keyring_user) let entry = keyring::Entry::new(&keyring_service, &keyring_user)
.context("Failed to create keyring entry for PAT")?; .context("Failed to create keyring entry for PAT")?;
entry.set_password(token) entry
.set_password(token)
.context("Failed to store PAT in keyring")?; .context("Failed to store PAT in keyring")?;
eprintln!("[DEBUG] PAT stored in keyring: service={}, user={}", keyring_service, keyring_user);
Ok(()) Ok(())
} }
pub fn get_pat(&self, profile_name: &str, user_email: &str, service: &str) -> Result<Option<String>> { pub fn get_pat(
&self,
profile_name: &str,
user_email: &str,
service: &str,
) -> Result<Option<String>> {
if !self.is_available() { if !self.is_available() {
return Ok(None); return Ok(None);
} }
@@ -158,14 +170,8 @@ impl KeyringManager {
.context("Failed to create keyring entry for PAT")?; .context("Failed to create keyring entry for PAT")?;
match entry.get_password() { match entry.get_password() {
Ok(token) => { Ok(token) => Ok(Some(token)),
eprintln!("[DEBUG] PAT retrieved from keyring: service={}, user={}", keyring_service, keyring_user); Err(keyring::Error::NoEntry) => Ok(None),
Ok(Some(token))
}
Err(keyring::Error::NoEntry) => {
eprintln!("[DEBUG] PAT not found in keyring: service={}, user={}", keyring_service, keyring_user);
Ok(None)
}
Err(e) => Err(e.into()), Err(e) => Err(e.into()),
} }
} }
@@ -181,23 +187,27 @@ impl KeyringManager {
let entry = keyring::Entry::new(&keyring_service, &keyring_user) let entry = keyring::Entry::new(&keyring_service, &keyring_user)
.context("Failed to create keyring entry for PAT")?; .context("Failed to create keyring entry for PAT")?;
entry.delete_credential() entry
.delete_credential()
.context("Failed to delete PAT from keyring")?; .context("Failed to delete PAT from keyring")?;
eprintln!("[DEBUG] PAT deleted from keyring: service={}, user={}", keyring_service, keyring_user);
Ok(()) Ok(())
} }
pub fn has_pat(&self, profile_name: &str, user_email: &str, service: &str) -> bool { 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() 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<()> { pub fn delete_all_pats_for_profile(
&self,
profile_name: &str,
user_email: &str,
services: &[String],
) -> Result<()> {
for service in services { for service in services {
if let Err(e) = self.delete_pat(profile_name, user_email, service) { let _ = self.delete_pat(profile_name, user_email, service);
eprintln!("[DEBUG] Failed to delete PAT for service '{}': {}", service, e);
}
} }
Ok(()) Ok(())
} }
@@ -251,8 +261,8 @@ pub fn get_default_model(provider: &str) -> &'static str {
match provider { match provider {
"openai" => "gpt-4", "openai" => "gpt-4",
"anthropic" => "claude-3-sonnet-20240229", "anthropic" => "claude-3-sonnet-20240229",
"kimi" => "moonshot-v1-8k", "kimi" => "kimi-k2.6",
"deepseek" => "deepseek-chat", "deepseek" => "deepseek-v4-flash",
"openrouter" => "openai/gpt-3.5-turbo", "openrouter" => "openai/gpt-3.5-turbo",
"ollama" => "llama2", "ollama" => "llama2",
_ => "", _ => "",
@@ -260,7 +270,14 @@ pub fn get_default_model(provider: &str) -> &'static str {
} }
pub fn get_supported_providers() -> &'static [&'static str] { pub fn get_supported_providers() -> &'static [&'static str] {
&["ollama", "openai", "anthropic", "kimi", "deepseek", "openrouter"] &[
"ollama",
"openai",
"anthropic",
"kimi",
"deepseek",
"openrouter",
]
} }
pub fn provider_needs_api_key(provider: &str) -> bool { pub fn provider_needs_api_key(provider: &str) -> bool {
@@ -274,10 +291,19 @@ mod tests {
#[test] #[test]
fn test_get_default_base_url() { fn test_get_default_base_url() {
assert_eq!(get_default_base_url("openai"), "https://api.openai.com/v1"); 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("anthropic"),
"https://api.anthropic.com/v1"
);
assert_eq!(get_default_base_url("kimi"), "https://api.moonshot.cn/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!(
assert_eq!(get_default_base_url("openrouter"), "https://openrouter.ai/api/v1"); 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"); assert_eq!(get_default_base_url("ollama"), "http://localhost:11434");
} }

View File

@@ -32,10 +32,10 @@ pub fn print_info(msg: &str) {
pub fn confirm(prompt: &str) -> Result<bool> { pub fn confirm(prompt: &str) -> Result<bool> {
print!("{} [y/N] ", prompt); print!("{} [y/N] ", prompt);
io::stdout().flush()?; io::stdout().flush()?;
let mut input = String::new(); let mut input = String::new();
io::stdin().read_line(&mut input)?; io::stdin().read_line(&mut input)?;
Ok(input.trim().to_lowercase().starts_with('y')) Ok(input.trim().to_lowercase().starts_with('y'))
} }
@@ -43,17 +43,17 @@ pub fn confirm(prompt: &str) -> Result<bool> {
pub fn input(prompt: &str) -> Result<String> { pub fn input(prompt: &str) -> Result<String> {
print!("{}: ", prompt); print!("{}: ", prompt);
io::stdout().flush()?; io::stdout().flush()?;
let mut input = String::new(); let mut input = String::new();
io::stdin().read_line(&mut input)?; io::stdin().read_line(&mut input)?;
Ok(input.trim().to_string()) Ok(input.trim().to_string())
} }
/// Get password input (hidden) /// Get password input (hidden)
pub fn password_input(prompt: &str) -> Result<String> { pub fn password_input(prompt: &str) -> Result<String> {
use dialoguer::Password; use dialoguer::Password;
Password::new() Password::new()
.with_prompt(prompt) .with_prompt(prompt)
.interact() .interact()

View File

@@ -1,4 +1,4 @@
use anyhow::{bail, Result}; use anyhow::{Result, bail};
use lazy_static::lazy_static; use lazy_static::lazy_static;
use regex::Regex; use regex::Regex;
@@ -67,7 +67,7 @@ lazy_static! {
/// Validate conventional commit message /// Validate conventional commit message
pub fn validate_conventional_commit(message: &str) -> Result<()> { pub fn validate_conventional_commit(message: &str) -> Result<()> {
let first_line = message.lines().next().unwrap_or(""); let first_line = message.lines().next().unwrap_or("");
if !CONVENTIONAL_COMMIT_REGEX.is_match(first_line) { if !CONVENTIONAL_COMMIT_REGEX.is_match(first_line) {
bail!( bail!(
"Invalid conventional commit format. Expected: <type>[optional scope]: <description>\n\ "Invalid conventional commit format. Expected: <type>[optional scope]: <description>\n\
@@ -75,32 +75,32 @@ pub fn validate_conventional_commit(message: &str) -> Result<()> {
CONVENTIONAL_TYPES.join(", ") CONVENTIONAL_TYPES.join(", ")
); );
} }
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)");
} }
Ok(()) Ok(())
} }
/// Validate @commitlint commit message /// Validate @commitlint commit message
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("");
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>");
} }
let type_part = parts[0]; let type_part = parts[0];
let subject = parts[1].trim(); let subject = parts[1].trim();
let commit_type = type_part let commit_type = type_part
.split('(') .split('(')
.next() .next()
.unwrap_or("") .unwrap_or("")
.trim_end_matches('!'); .trim_end_matches('!');
if !COMMITLINT_TYPES.contains(&commit_type) { if !COMMITLINT_TYPES.contains(&commit_type) {
bail!( bail!(
"Invalid commit type: '{}'. Valid types: {}", "Invalid commit type: '{}'. Valid types: {}",
@@ -108,27 +108,32 @@ pub fn validate_commitlint_commit(message: &str) -> Result<()> {
COMMITLINT_TYPES.join(", ") COMMITLINT_TYPES.join(", ")
); );
} }
if subject.is_empty() { if subject.is_empty() {
bail!("Commit subject cannot be empty"); bail!("Commit subject cannot be empty");
} }
if subject.len() < 4 { if subject.len() < 4 {
bail!("Commit subject too short (min 4 characters)"); bail!("Commit subject too short (min 4 characters)");
} }
if subject.len() > 100 { if subject.len() > 100 {
bail!("Commit subject too long (max 100 characters)"); bail!("Commit subject too long (max 100 characters)");
} }
if subject.chars().next().map(|c| c.is_uppercase()).unwrap_or(false) { if subject
.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");
} }
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");
} }
Ok(()) Ok(())
} }
@@ -137,25 +142,25 @@ pub fn validate_scope(scope: &str) -> Result<()> {
if scope.is_empty() { if scope.is_empty() {
bail!("Scope cannot be empty"); bail!("Scope cannot be empty");
} }
if !SCOPE_REGEX.is_match(scope) { if !SCOPE_REGEX.is_match(scope) {
bail!("Invalid scope format. Use lowercase letters, numbers, and hyphens only"); bail!("Invalid scope format. Use lowercase letters, numbers, and hyphens only");
} }
Ok(()) Ok(())
} }
/// Validate semantic version tag /// Validate semantic version tag
pub fn validate_semver(version: &str) -> Result<()> { pub fn validate_semver(version: &str) -> Result<()> {
let version = version.trim_start_matches('v'); let version = version.trim_start_matches('v');
if !SEMVER_REGEX.is_match(version) { if !SEMVER_REGEX.is_match(version) {
bail!( bail!(
"Invalid semantic version format. Expected: MAJOR.MINOR.PATCH[-prerelease][+build]\n\ "Invalid semantic version format. Expected: MAJOR.MINOR.PATCH[-prerelease][+build]\n\
Examples: 1.0.0, 1.2.3-beta, v2.0.0+build123" Examples: 1.0.0, 1.2.3-beta, v2.0.0+build123"
); );
} }
Ok(()) Ok(())
} }
@@ -164,7 +169,7 @@ pub fn validate_email(email: &str) -> Result<()> {
if !EMAIL_REGEX.is_match(email) { if !EMAIL_REGEX.is_match(email) {
bail!("Invalid email address format"); bail!("Invalid email address format");
} }
Ok(()) Ok(())
} }
@@ -173,7 +178,7 @@ 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) {
bail!("Invalid GPG key ID format. Expected 16-40 hexadecimal characters"); bail!("Invalid GPG key ID format. Expected 16-40 hexadecimal characters");
} }
Ok(()) Ok(())
} }
@@ -182,15 +187,18 @@ pub fn validate_profile_name(name: &str) -> Result<()> {
if name.is_empty() { if name.is_empty() {
bail!("Profile name cannot be empty"); bail!("Profile name cannot be empty");
} }
if name.len() > 50 { if name.len() > 50 {
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");
} }
Ok(()) Ok(())
} }
@@ -201,7 +209,7 @@ pub fn is_valid_commit_type(commit_type: &str, use_commitlint: bool) -> bool {
} else { } else {
CONVENTIONAL_TYPES CONVENTIONAL_TYPES
}; };
types.contains(&commit_type) types.contains(&commit_type)
} }

View File

@@ -1,38 +1,11 @@
use assert_cmd::Command; use assert_cmd::cargo::cargo_bin_cmd;
use predicates::prelude::*; use predicates::prelude::*;
use std::fs; use std::fs;
use std::path::PathBuf; use std::path::PathBuf;
use tempfile::TempDir; use tempfile::TempDir;
fn create_git_repo(dir: &PathBuf) -> std::process::Output {
std::process::Command::new("git")
.args(&["init"])
.current_dir(dir)
.output()
.expect("Failed to init git repo")
}
fn configure_git_user(dir: &PathBuf) {
std::process::Command::new("git")
.args(&["config", "user.name", "Test User"])
.current_dir(dir)
.output()
.expect("Failed to configure git user name");
std::process::Command::new("git")
.args(&["config", "user.email", "test@example.com"])
.current_dir(dir)
.output()
.expect("Failed to configure git user email");
}
fn setup_git_repo(dir: &PathBuf) {
create_git_repo(dir);
configure_git_user(dir);
}
fn init_quicommit(config_path: &PathBuf) { fn init_quicommit(config_path: &PathBuf) {
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
} }
@@ -46,8 +19,13 @@ mod config_export {
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
init_quicommit(&config_path); init_quicommit(&config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "export", "--config", config_path.to_str().unwrap()]); cmd.args(&[
"config",
"export",
"--config",
config_path.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
@@ -62,12 +40,16 @@ mod config_export {
let export_path = temp_dir.path().join("exported.toml"); let export_path = temp_dir.path().join("exported.toml");
init_quicommit(&config_path); init_quicommit(&config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "export", "config",
"--config", config_path.to_str().unwrap(), "export",
"--output", export_path.to_str().unwrap(), "--config",
"--password", "" config_path.to_str().unwrap(),
"--output",
export_path.to_str().unwrap(),
"--password",
"",
]); ]);
cmd.assert() cmd.assert()
@@ -75,10 +57,13 @@ mod config_export {
.stdout(predicate::str::contains("Configuration exported")); .stdout(predicate::str::contains("Configuration exported"));
assert!(export_path.exists(), "Export file should be created"); assert!(export_path.exists(), "Export file should be created");
let content = fs::read_to_string(&export_path).unwrap(); let content = fs::read_to_string(&export_path).unwrap();
assert!(content.contains("version"), "Export should contain version"); assert!(content.contains("version"), "Export should contain version");
assert!(content.contains("[llm]"), "Export should contain LLM config"); assert!(
content.contains("[llm]"),
"Export should contain LLM config"
);
} }
#[test] #[test]
@@ -88,12 +73,16 @@ mod config_export {
let export_path = temp_dir.path().join("encrypted.toml"); let export_path = temp_dir.path().join("encrypted.toml");
init_quicommit(&config_path); init_quicommit(&config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "export", "config",
"--config", config_path.to_str().unwrap(), "export",
"--output", export_path.to_str().unwrap(), "--config",
"--password", "test_password_123" config_path.to_str().unwrap(),
"--output",
export_path.to_str().unwrap(),
"--password",
"test_password_123",
]); ]);
cmd.assert() cmd.assert()
@@ -101,10 +90,16 @@ mod config_export {
.stdout(predicate::str::contains("encrypted and exported")); .stdout(predicate::str::contains("encrypted and exported"));
assert!(export_path.exists(), "Export file should be created"); assert!(export_path.exists(), "Export file should be created");
let content = fs::read_to_string(&export_path).unwrap(); let content = fs::read_to_string(&export_path).unwrap();
assert!(content.starts_with("ENCRYPTED:"), "Encrypted file should start with ENCRYPTED:"); assert!(
assert!(!content.contains("[llm]"), "Encrypted content should not be readable"); content.starts_with("ENCRYPTED:"),
"Encrypted file should start with ENCRYPTED:"
);
assert!(
!content.contains("[llm]"),
"Encrypted content should not be readable"
);
} }
} }
@@ -116,7 +111,7 @@ mod config_import {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let import_path = temp_dir.path().join("import.toml"); let import_path = temp_dir.path().join("import.toml");
let plain_config = r#" let plain_config = r#"
version = "1" version = "1"
@@ -131,31 +126,14 @@ api_key_storage = "keyring"
[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]
colors = true
icons = true
date_format = "%Y-%m-%d"
[language] [language]
output_language = "en" output_language = "en"
@@ -164,19 +142,28 @@ keep_changelog_types_english = true
"#; "#;
fs::write(&import_path, plain_config).unwrap(); fs::write(&import_path, plain_config).unwrap();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "import", "config",
"--config", config_path.to_str().unwrap(), "import",
"--file", import_path.to_str().unwrap() "--config",
config_path.to_str().unwrap(),
"--file",
import_path.to_str().unwrap(),
]); ]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("Configuration imported")); .stdout(predicate::str::contains("Configuration imported"));
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "get", "llm.provider", "--config", config_path.to_str().unwrap()]); cmd.args(&[
"config",
"get",
"llm.provider",
"--config",
config_path.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("openai")); .stdout(predicate::str::contains("openai"));
@@ -188,38 +175,56 @@ keep_changelog_types_english = true
let config_path1 = temp_dir.path().join("config1.toml"); let config_path1 = temp_dir.path().join("config1.toml");
let config_path2 = temp_dir.path().join("config2.toml"); let config_path2 = temp_dir.path().join("config2.toml");
let export_path = temp_dir.path().join("encrypted.toml"); let export_path = temp_dir.path().join("encrypted.toml");
init_quicommit(&config_path1); init_quicommit(&config_path1);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "set", "llm.provider", "anthropic", "config",
"--config", config_path1.to_str().unwrap() "set",
"llm.provider",
"anthropic",
"--config",
config_path1.to_str().unwrap(),
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "export", "config",
"--config", config_path1.to_str().unwrap(), "export",
"--output", export_path.to_str().unwrap(), "--config",
"--password", "secure_password" config_path1.to_str().unwrap(),
"--output",
export_path.to_str().unwrap(),
"--password",
"secure_password",
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "import", "config",
"--config", config_path2.to_str().unwrap(), "import",
"--file", export_path.to_str().unwrap(), "--config",
"--password", "secure_password" config_path2.to_str().unwrap(),
"--file",
export_path.to_str().unwrap(),
"--password",
"secure_password",
]); ]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("Configuration imported")); .stdout(predicate::str::contains("Configuration imported"));
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "get", "llm.provider", "--config", config_path2.to_str().unwrap()]); cmd.args(&[
"config",
"get",
"llm.provider",
"--config",
config_path2.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("anthropic")); .stdout(predicate::str::contains("anthropic"));
@@ -230,24 +235,32 @@ keep_changelog_types_english = true
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let export_path = temp_dir.path().join("encrypted.toml"); let export_path = temp_dir.path().join("encrypted.toml");
init_quicommit(&config_path); init_quicommit(&config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "export", "config",
"--config", config_path.to_str().unwrap(), "export",
"--output", export_path.to_str().unwrap(), "--config",
"--password", "correct_password" config_path.to_str().unwrap(),
"--output",
export_path.to_str().unwrap(),
"--password",
"correct_password",
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "import", "config",
"--config", config_path.to_str().unwrap(), "import",
"--file", export_path.to_str().unwrap(), "--config",
"--password", "wrong_password" config_path.to_str().unwrap(),
"--file",
export_path.to_str().unwrap(),
"--password",
"wrong_password",
]); ]);
cmd.assert() cmd.assert()
.failure() .failure()
@@ -264,35 +277,52 @@ mod config_export_import_roundtrip {
let config_path1 = temp_dir.path().join("config1.toml"); let config_path1 = temp_dir.path().join("config1.toml");
let config_path2 = temp_dir.path().join("config2.toml"); let config_path2 = temp_dir.path().join("config2.toml");
let export_path = temp_dir.path().join("export.toml"); let export_path = temp_dir.path().join("export.toml");
init_quicommit(&config_path1); init_quicommit(&config_path1);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "set", "llm.model", "gpt-4-turbo", "config",
"--config", config_path1.to_str().unwrap() "set",
"llm.model",
"gpt-4-turbo",
"--config",
config_path1.to_str().unwrap(),
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "export", "config",
"--config", config_path1.to_str().unwrap(), "export",
"--output", export_path.to_str().unwrap(), "--config",
"--password", "" config_path1.to_str().unwrap(),
"--output",
export_path.to_str().unwrap(),
"--password",
"",
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "import", "config",
"--config", config_path2.to_str().unwrap(), "import",
"--file", export_path.to_str().unwrap() "--config",
config_path2.to_str().unwrap(),
"--file",
export_path.to_str().unwrap(),
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "get", "llm.model", "--config", config_path2.to_str().unwrap()]); cmd.args(&[
"config",
"get",
"llm.model",
"--config",
config_path2.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("gpt-4-turbo")); .stdout(predicate::str::contains("gpt-4-turbo"));
@@ -305,29 +335,41 @@ mod config_export_import_roundtrip {
let config_path2 = temp_dir.path().join("config2.toml"); let config_path2 = temp_dir.path().join("config2.toml");
let export_path = temp_dir.path().join("encrypted.toml"); let export_path = temp_dir.path().join("encrypted.toml");
let password = "my_secure_password_123"; let password = "my_secure_password_123";
init_quicommit(&config_path1); init_quicommit(&config_path1);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "set", "llm.provider", "deepseek", "config",
"--config", config_path1.to_str().unwrap() "set",
"llm.provider",
"deepseek",
"--config",
config_path1.to_str().unwrap(),
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "set", "llm.model", "deepseek-chat", "config",
"--config", config_path1.to_str().unwrap() "set",
"llm.model",
"deepseek-chat",
"--config",
config_path1.to_str().unwrap(),
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "export", "config",
"--config", config_path1.to_str().unwrap(), "export",
"--output", export_path.to_str().unwrap(), "--config",
"--password", password config_path1.to_str().unwrap(),
"--output",
export_path.to_str().unwrap(),
"--password",
password,
]); ]);
cmd.assert().success(); cmd.assert().success();
@@ -335,23 +377,39 @@ mod config_export_import_roundtrip {
assert!(exported_content.starts_with("ENCRYPTED:")); assert!(exported_content.starts_with("ENCRYPTED:"));
assert!(!exported_content.contains("deepseek")); assert!(!exported_content.contains("deepseek"));
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[ cmd.args(&[
"config", "import", "config",
"--config", config_path2.to_str().unwrap(), "import",
"--file", export_path.to_str().unwrap(), "--config",
"--password", password config_path2.to_str().unwrap(),
"--file",
export_path.to_str().unwrap(),
"--password",
password,
]); ]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "get", "llm.provider", "--config", config_path2.to_str().unwrap()]); cmd.args(&[
"config",
"get",
"llm.provider",
"--config",
config_path2.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("deepseek")); .stdout(predicate::str::contains("deepseek"));
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "get", "llm.model", "--config", config_path2.to_str().unwrap()]); cmd.args(&[
"config",
"get",
"llm.model",
"--config",
config_path2.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("deepseek-chat")); .stdout(predicate::str::contains("deepseek-chat"));

1124
tests/credential_tests.rs Normal file

File diff suppressed because it is too large Load Diff

266
tests/gitignore_tests.rs Normal file
View File

@@ -0,0 +1,266 @@
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 removed = repo.stage_all().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 removed = repo.stage_all().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
);
}

View File

@@ -1,4 +1,4 @@
use assert_cmd::Command; use assert_cmd::cargo::cargo_bin_cmd;
use predicates::prelude::*; use predicates::prelude::*;
use std::fs; use std::fs;
use std::path::PathBuf; use std::path::PathBuf;
@@ -59,7 +59,7 @@ fn setup_test_repo_with_file(dir: &PathBuf, file_name: &str, file_content: &str)
} }
fn init_quicommit(dir: &PathBuf, config_path: &PathBuf) { fn init_quicommit(dir: &PathBuf, config_path: &PathBuf) {
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()])
.current_dir(dir); .current_dir(dir);
cmd.assert().success(); cmd.assert().success();
@@ -70,7 +70,7 @@ mod cli_basic {
#[test] #[test]
fn test_help() { fn test_help() {
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.arg("--help"); cmd.arg("--help");
cmd.assert() cmd.assert()
.success() .success()
@@ -83,7 +83,7 @@ mod cli_basic {
#[test] #[test]
fn test_version() { fn test_version() {
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.arg("--version"); cmd.arg("--version");
cmd.assert() cmd.assert()
.success() .success()
@@ -92,7 +92,7 @@ mod cli_basic {
#[test] #[test]
fn test_no_args_shows_help() { fn test_no_args_shows_help() {
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.assert() cmd.assert()
.failure() .failure()
.stderr(predicate::str::contains("Usage:")); .stderr(predicate::str::contains("Usage:"));
@@ -106,9 +106,15 @@ mod cli_basic {
create_git_repo(&repo_path); create_git_repo(&repo_path);
configure_git_user(&repo_path); configure_git_user(&repo_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["-vv", "init", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "-vv",
"init",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert().success(); cmd.assert().success();
} }
@@ -122,7 +128,7 @@ mod init_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert() cmd.assert()
@@ -135,7 +141,7 @@ mod init_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
@@ -152,7 +158,7 @@ mod init_command {
let config_path = repo_path.join("test_config.toml"); let config_path = repo_path.join("test_config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()])
.current_dir(&repo_path); .current_dir(&repo_path);
@@ -164,12 +170,18 @@ mod init_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--reset", "--config", config_path.to_str().unwrap()]); cmd.args(&[
"init",
"--yes",
"--reset",
"--config",
config_path.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("initialized successfully")); .stdout(predicate::str::contains("initialized successfully"));
@@ -184,7 +196,7 @@ mod profile_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["profile", "list", "--config", config_path.to_str().unwrap()]); cmd.args(&["profile", "list", "--config", config_path.to_str().unwrap()]);
cmd.assert() cmd.assert()
@@ -197,11 +209,11 @@ mod profile_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["profile", "list", "--config", config_path.to_str().unwrap()]); cmd.args(&["profile", "list", "--config", config_path.to_str().unwrap()]);
cmd.assert() cmd.assert()
@@ -218,11 +230,11 @@ mod config_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "show", "--config", config_path.to_str().unwrap()]); cmd.args(&["config", "show", "--config", config_path.to_str().unwrap()]);
cmd.assert() cmd.assert()
@@ -235,11 +247,11 @@ mod config_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "path", "--config", config_path.to_str().unwrap()]); cmd.args(&["config", "path", "--config", config_path.to_str().unwrap()]);
cmd.assert() cmd.assert()
@@ -256,13 +268,19 @@ mod commit_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(temp_dir.path()); "commit",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(temp_dir.path());
cmd.assert() cmd.assert()
.failure() .failure()
@@ -278,9 +296,18 @@ mod commit_command {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--manual", "-m", "test: empty", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "commit",
"--manual",
"-m",
"test: empty",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.success() .success()
@@ -296,9 +323,18 @@ mod commit_command {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--manual", "-m", "test: add test file", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "commit",
"--manual",
"-m",
"test: add test file",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.success() .success()
@@ -314,14 +350,47 @@ mod commit_command {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--date", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "commit",
"--date",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("Dry run")); .stdout(predicate::str::contains("Dry run"));
} }
#[test]
fn test_commit_with_think_flag() {
let temp_dir = TempDir::new().unwrap();
let repo_path = temp_dir.path().to_path_buf();
setup_test_repo_with_file(&repo_path, "test.txt", "Hello, World!");
let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path);
let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[
"commit",
"--think",
"--manual",
"-m",
"test: think flag",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert().success();
}
} }
mod tag_command { mod tag_command {
@@ -332,13 +401,19 @@ mod tag_command {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["tag", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(temp_dir.path()); "tag",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(temp_dir.path());
cmd.assert() cmd.assert()
.failure() .failure()
@@ -358,14 +433,51 @@ mod tag_command {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["tag", "--name", "v0.1.0", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "tag",
"--name",
"v0.1.0",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.success() .success()
.stdout(predicate::str::contains("v0.1.0")); .stdout(predicate::str::contains("v0.1.0"));
} }
#[test]
fn test_tag_with_think_flag() {
let temp_dir = TempDir::new().unwrap();
let repo_path = temp_dir.path().to_path_buf();
setup_git_repo(&repo_path);
create_test_file(&repo_path, "test.txt", "content");
stage_file(&repo_path, "test.txt");
create_commit(&repo_path, "feat: initial commit");
let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path);
let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&[
"tag",
"--think",
"--name",
"v0.2.0",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert().success();
}
} }
mod changelog_command { mod changelog_command {
@@ -382,9 +494,16 @@ mod changelog_command {
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["changelog", "--init", "--output", changelog_path.to_str().unwrap(), "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "changelog",
"--init",
"--output",
changelog_path.to_str().unwrap(),
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert().success(); cmd.assert().success();
@@ -404,12 +523,17 @@ mod changelog_command {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["changelog", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "changelog",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert().success();
.success();
} }
} }
@@ -421,7 +545,7 @@ mod cross_platform {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("subdir").join("config.toml"); let config_path = temp_dir.path().join("subdir").join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
@@ -435,7 +559,7 @@ mod cross_platform {
fs::create_dir_all(&space_dir).unwrap(); fs::create_dir_all(&space_dir).unwrap();
let config_path = space_dir.join("config.toml"); let config_path = space_dir.join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
@@ -449,7 +573,7 @@ mod cross_platform {
fs::create_dir_all(&unicode_dir).unwrap(); fs::create_dir_all(&unicode_dir).unwrap();
let config_path = unicode_dir.join("config.toml"); let config_path = unicode_dir.join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
@@ -523,9 +647,18 @@ mod validators {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--manual", "-m", "invalid commit message without type", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "commit",
"--manual",
"-m",
"invalid commit message without type",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.failure() .failure()
@@ -541,9 +674,18 @@ mod validators {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--manual", "-m", "feat: add new feature", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "commit",
"--manual",
"-m",
"feat: add new feature",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.success() .success()
@@ -563,9 +705,18 @@ mod subcommands {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["c", "--manual", "-m", "fix: test", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "c",
"--manual",
"-m",
"fix: test",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.success() .success()
@@ -577,7 +728,7 @@ mod subcommands {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["i", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["i", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert() cmd.assert()
@@ -590,11 +741,11 @@ mod subcommands {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let config_path = temp_dir.path().join("config.toml"); let config_path = temp_dir.path().join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["p", "list", "--config", config_path.to_str().unwrap()]); cmd.args(&["p", "list", "--config", config_path.to_str().unwrap()]);
cmd.assert() cmd.assert()
@@ -611,8 +762,13 @@ mod edge_cases {
let temp_dir = TempDir::new().unwrap(); let temp_dir = TempDir::new().unwrap();
let non_existent_config = temp_dir.path().join("non_existent_config.toml"); let non_existent_config = temp_dir.path().join("non_existent_config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["config", "show", "--config", non_existent_config.to_str().unwrap()]); cmd.args(&[
"config",
"show",
"--config",
non_existent_config.to_str().unwrap(),
]);
cmd.assert() cmd.assert()
.success() .success()
@@ -627,13 +783,19 @@ mod edge_cases {
let repo_path = temp_dir.path().to_path_buf(); let repo_path = temp_dir.path().to_path_buf();
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]); cmd.args(&["init", "--yes", "--config", config_path.to_str().unwrap()]);
cmd.assert().success(); cmd.assert().success();
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "commit",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert()
.failure() .failure()
@@ -649,12 +811,21 @@ mod edge_cases {
let config_path = repo_path.join("config.toml"); let config_path = repo_path.join("config.toml");
init_quicommit(&repo_path, &config_path); init_quicommit(&repo_path, &config_path);
let mut cmd = Command::cargo_bin("quicommit").unwrap(); let mut cmd = cargo_bin_cmd!("quicommit");
cmd.args(&["commit", "--manual", "-m", "", "--dry-run", "--yes", "--config", config_path.to_str().unwrap()]) cmd.args(&[
.current_dir(&repo_path); "commit",
"--manual",
"-m",
"",
"--dry-run",
"--yes",
"--config",
config_path.to_str().unwrap(),
])
.current_dir(&repo_path);
cmd.assert() cmd.assert().failure().stderr(predicate::str::contains(
.failure() "Invalid conventional commit format",
.stderr(predicate::str::contains("Invalid conventional commit format")); ));
} }
} }