acm-cli 0.1.0__py3-none-any.whl

acm/ui.py

@@ -0,0 +1,279 @@
+"""终端 UI 模块：使用 rich 库实现彩色输出、spinner、diff 预览、交互菜单。"""
+
+from __future__ import annotations
+
+import queue
+import time
+from dataclasses import dataclass
+from threading import Thread
+from typing import Generator, Union
+
+import click
+from rich.console import Console
+from rich.live import Live
+from rich.panel import Panel
+from rich.text import Text
+from acm.git import DiffResult, FileChange
+from acm.splitter import CommitGroup
+
+console = Console()
+err_console = Console(stderr=True)
+
+
+# ---------------------------------------------------------------------------
+# 错误与信息输出
+# ---------------------------------------------------------------------------
+
+def print_error(message: str) -> None:
+    err_console.print(f"[bold red]x[/bold red] {message}")
+
+
+def print_warning(message: str) -> None:
+    console.print(f"[bold yellow]![/bold yellow] {message}")
+
+
+def print_success(message: str) -> None:
+    console.print(f"[bold green]v[/bold green] {message}")
+
+
+def print_info(message: str) -> None:
+    console.print(f"[bold blue]>[/bold blue] {message}")
+
+
+def print_verbose(message: str) -> None:
+    console.print(f"[dim]{message}[/dim]")
+
+
+# ---------------------------------------------------------------------------
+# Diff 预览
+# ---------------------------------------------------------------------------
+
+def _status_color(status: str) -> str:
+    colors = {"A": "green", "M": "yellow", "D": "red", "R": "cyan"}
+    return colors.get(status, "white")
+
+
+def _status_label(status: str) -> str:
+    labels = {"A": "新增", "M": "修改", "D": "删除", "R": "重命名"}
+    return labels.get(status, status)
+
+
+def show_diff_summary(diff_result: DiffResult) -> None:
+    """展示变更摘要（文件树形式）。"""
+    total_ins = sum(f.insertions for f in diff_result.files)
+    total_del = sum(f.deletions for f in diff_result.files)
+
+    console.print()
+    print_info(f"分析暂存区变更...")
+    console.print()
+
+    header = f"  修改了 {len(diff_result.files)} 个文件 ([green]+{total_ins}[/green], [red]-{total_del}[/red])"
+    console.print(header)
+
+    for i, f in enumerate(diff_result.files):
+        color = _status_color(f.status)
+        is_last = i == len(diff_result.files) - 1
+        prefix = "  └── " if is_last else "  ├── "
+        console.print(f"{prefix}[{color}]{f.path}[/{color}]    ([green]+{f.insertions}[/green], [red]-{f.deletions}[/red])")
+
+    console.print()
+
+
+# ---------------------------------------------------------------------------
+# Commit Message 展示
+# ---------------------------------------------------------------------------
+
+def show_commit_message(message: str) -> None:
+    """以面板形式展示 commit message。"""
+    panel = Panel(
+        Text(message),
+        title="[bold]Commit Message[/bold]",
+        border_style="cyan",
+        padding=(1, 2),
+    )
+    console.print(panel)
+
+
+# ---------------------------------------------------------------------------
+# 交互菜单
+# ---------------------------------------------------------------------------
+
+def prompt_action() -> str:
+    """展示操作菜单并获取用户选择。
+
+    Returns:
+        用户选择: 'y' | 'e' | 'r' | 'q'
+    """
+    console.print(
+        "[bold][green]确认提交(y)[/green]  "
+        "[yellow]编辑(e)[/yellow]  "
+        "[blue]反馈重新生成(r)[/blue]  "
+        "[red]放弃(q)[/red][/bold]"
+    )
+
+    while True:
+        choice = click.prompt("", prompt_suffix="> ", type=str).strip().lower()
+        if choice in ("y", "e", "r", "q"):
+            return choice
+        console.print("[dim]请输入 y/e/r/q[/dim]")
+
+
+def prompt_feedback() -> str:
+    """获取用户的反馈意见。"""
+    console.print("[bold blue]>[/bold blue] 请输入你的反馈（告诉AI如何改进）:")
+    return click.prompt("", prompt_suffix="> ", type=str).strip()
+
+
+def prompt_edit(message: str) -> str:
+    """让用户编辑 commit message。使用 click.edit 打开编辑器。"""
+    edited = click.edit(message)
+    if edited is not None:
+        return edited.strip()
+    return message
+
+
+def prompt_confirm(message: str) -> bool:
+    """简单的确认提示。"""
+    return click.confirm(message, default=True)
+
+
+# ---------------------------------------------------------------------------
+# 智能拆分展示
+# ---------------------------------------------------------------------------
+
+def show_split_suggestion(groups: list[CommitGroup]) -> None:
+    """展示拆分提交建议。"""
+    console.print()
+    print_info(f"建议拆分为 {len(groups)} 个原子提交：")
+    console.print()
+
+    for i, group in enumerate(groups, 1):
+        console.print(f"  [bold cyan]提交 {i}[/bold cyan]: {group.message}")
+        for f in group.files:
+            console.print(f"    - {f}")
+        console.print()
+
+
+# ---------------------------------------------------------------------------
+# Spinner 上下文管理
+# ---------------------------------------------------------------------------
+
+def spinner(message: str = "生成 commit message..."):
+    """返回 rich 的 spinner 上下文管理器。"""
+    return console.status(f"[bold blue]>[/bold blue] {message}", spinner="dots")
+
+
+# ---------------------------------------------------------------------------
+# 大提交警告
+# ---------------------------------------------------------------------------
+
+def warn_large_commit(file_count: int) -> bool:
+    """对大提交发出警告，询问是否继续。返回 True 表示继续。"""
+    print_warning(
+        f"暂存区包含 {file_count} 个文件，变更较多，建议拆分提交以保持每次提交的原子性。"
+    )
+    return prompt_confirm("是否继续？")
+
+
+# ---------------------------------------------------------------------------
+# 流式输出
+# ---------------------------------------------------------------------------
+
+@dataclass(slots=True)
+class _Done:
+    """后台线程完成哨兵，携带 generator 的 return 值。"""
+    value: str
+
+
+@dataclass(slots=True)
+class _Error:
+    """后台线程异常哨兵，携带原始异常。"""
+    exc: BaseException
+
+
+def _consume_stream(
+    chunks: Generator[str, None, str],
+    q: queue.Queue,
+) -> None:
+    """在后台线程中消费 generator，将 chunk 放入队列。"""
+    try:
+        while True:
+            chunk = next(chunks)
+            q.put(chunk)
+    except StopIteration as e:
+        q.put(_Done(value=e.value or ""))
+    except Exception as e:
+        q.put(_Error(exc=e))
+
+
+def stream_commit_message(chunks: Generator[str, None, str]) -> str:
+    """流式展示 LLM 生成的 commit message，后台线程收集数据，主线程平滑渲染。
+
+    通过将网络 I/O（generator 消费）放到后台线程，主线程专注于以固定节奏
+    刷新 UI，避免因等待 LLM 响应而导致的画面冻结。
+
+    Args:
+        chunks: 生成器，逐块 yield 文本片段，最终 return 清理后的完整 message。
+
+    Returns:
+        最终清理后的 commit message。
+    """
+    chunk_queue: queue.Queue = queue.Queue()
+    thread = Thread(target=_consume_stream, args=(chunks, chunk_queue), daemon=True)
+    thread.start()
+
+    parts: list[str] = []
+    final_message = ""
+    error: BaseException | None = None
+    done = False
+
+    with Live(
+        _make_stream_panel(""),
+        console=console,
+        refresh_per_second=30,
+        transient=True,
+    ) as live:
+        while not done:
+            got_new = False
+            # 批量 drain：非阻塞取出所有已到达的 chunk
+            while True:
+                try:
+                    item: Union[str, _Done, _Error] = chunk_queue.get_nowait()
+                except queue.Empty:
+                    break
+                if isinstance(item, _Done):
+                    final_message = item.value
+                    done = True
+                    break
+                elif isinstance(item, _Error):
+                    error = item.exc
+                    done = True
+                    break
+                else:
+                    parts.append(item)
+                    got_new = True
+
+            if got_new:
+                live.update(_make_stream_panel("".join(parts)))
+
+            if not done:
+                time.sleep(1 / 60)
+
+    thread.join(timeout=5)
+
+    if error is not None:
+        raise error
+
+    return final_message if final_message else "".join(parts).strip()
+
+
+def _make_stream_panel(text: str) -> Panel:
+    """构建流式输出用的面板。"""
+    display = text if text else " "
+    return Panel(
+        Text(display),
+        title="[bold]Commit Message[/bold]",
+        subtitle="[dim]生成中...[/dim]" if text else None,
+        border_style="cyan",
+        padding=(1, 2),
+    )

acm_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,369 @@
+Metadata-Version: 2.4
+Name: acm-cli
+Version: 0.1.0
+Summary: 基于 LLM 的 Git Commit Message 自动生成工具
+Project-URL: Homepage, https://github.com/hz-lxt/auto_commit_cli
+Project-URL: Repository, https://github.com/hz-lxt/auto_commit_cli
+Project-URL: Issues, https://github.com/hz-lxt/auto_commit_cli/issues
+Author-email: huangzhao <572478035@qq.com>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ai,automation,commit,commit-message,git,llm
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Version Control :: Git
+Requires-Python: >=3.10
+Requires-Dist: click>=8.0
+Requires-Dist: openai>=1.0
+Requires-Dist: rich>=13.0
+Requires-Dist: tomli-w>=1.0
+Requires-Dist: tomli>=2.0; python_version < '3.11'
+Description-Content-Type: text/markdown
+
+# ACM-CLI
+
+基于 LLM 的 Git Commit Message 自动生成工具。分析暂存区代码变更，自动生成符合规范的中文 commit message。
+
+## 功能特性
+
+- 自动分析 `git diff --staged` 内容，调用 LLM 生成规范的 commit message
+- 遵循 `type(scope): 描述` 格式，scope 根据文件目录自动推断，描述默认中文
+- 支持确认提交、编辑、根据反馈多次重新生成、放弃等交互操作
+- 智能 diff 截断：大变更自动截断并附加摘要，避免超出 token 限制
+- 智能拆分提交：将一次大变更拆分为多个原子提交
+- 多提供商支持：内置通义千问，兼容所有 OpenAI 接口的提供商
+- 彩色终端界面（基于 rich）
+- 流式输出：LLM 生成过程实时逐字显示，无需等待完整响应
+- 全局配置 + 项目级配置，TOML 格式
+- 反馈学习：自动记录用户反馈，动态优化后续生成的 prompt
+- 自动暂存：暂存区为空时提示自动执行 `git add`
+
+## 环境要求
+
+- Python >= 3.10
+- Git
+
+## 安装
+
+```bash
+# 使用 pip
+pip install acm-cli
+
+# 使用 uv
+uv pip install acm-cli
+
+# 从源码安装（开发模式）
+git clone <repo-url> && cd auto_commit
+uv venv --python 3.10 && source .venv/bin/activate
+uv pip install -e .
+```
+
+安装完成后，终端即可使用 `acm` 命令。
+
+## 快速开始
+
+### 1. 初始化配置
+
+首次使用需要配置 LLM 提供商和 API Key：
+
+```bash
+acm init
+```
+
+按提示依次完成：
+
+```
+ACM 配置初始化
+
+可选提供商：
+  1. qwen
+  2. other
+选择提供商: 1
+输入 API Key: ****
+输入模型名称 (必填): qwen-plus
+自定义 Base URL (留空使用默认):
+输出语言 (zh-CN, en) [zh-CN]: zh-CN
+
+v 配置已保存到: .acm.toml
+```
+
+生成全局配置（所有项目共用）：
+
+```bash
+acm init --global
+```
+
+### 2. 日常使用
+
+```bash
+# 先暂存要提交的文件
+git add .
+
+# 自动生成 commit message
+acm
+```
+
+工具会展示变更摘要，然后**实时流式显示**生成的 message，最后等待你选择操作：
+
+```
+> 分析暂存区变更...
+
+  修改了 3 个文件 (+45, -12)
+  ├── src/auth/login.py    (+30, -5)
+  ├── src/auth/models.py   (+10, -2)
+  └── tests/test_login.py  (+5, -5)
+
+> 生成 commit message...
+
+╭─────── Commit Message ───────╮
+│                              │
+│  feat(auth): 实现用户登录认证功能  │
+│                              │
+│  - 新增基于JWT的登录接口        │
+│  - 添加用户模型字段校验         │
+│  - 更新登录相关单元测试         │
+│                              │
+╰──────────────────────────────╯
+
+[y] 确认提交  [e] 编辑  [r] 反馈重新生成  [q] 放弃
+>
+```
+
+| 操作 | 说明 |
+|------|------|
+| `y` | 确认，执行 `git commit` |
+| `e` | 打开编辑器修改 message 后提交 |
+| `r` | 输入反馈，LLM 根据反馈重新生成（可多次） |
+| `q` | 放弃，不提交 |
+
+### 3. 反馈重新生成
+
+输入 `r` 后可以告诉 AI 如何改进：
+
+```
+> r
+> 请输入你的反馈（告诉AI如何改进）:
+> type应该是refactor不是feat
+
+> 根据反馈重新生成...
+
+╭─────── Commit Message ───────╮
+│  refactor(auth): 重构用户登录认证逻辑  │
+╰──────────────────────────────╯
+```
+
+反馈可以多次进行，直到满意为止。
+
+### 4. 自动暂存
+
+当暂存区为空但工作区存在未暂存的修改或未跟踪文件时，acm 会提示你是否自动执行 `git add -A`：
+
+```
+! 暂存区为空，但检测到未暂存的修改
+是否执行 git add . 将所有变更添加到暂存区并继续？ [Y/n]: y
+v 已执行 git add -A，所有变更已添加到暂存区
+
+> 分析暂存区变更...
+```
+
+选择 `y` 后工具会自动暂存所有变更并继续生成 commit message，无需手动执行 `git add`。
+
+### 5. 同类变更收敛
+
+当多个文件的变更属于同一类操作时，生成的 commit message 会自动合并 scope，避免冗余：
+
+```
+# 多个文件都是优化导入 -> 合并为一条
+refactor(cve_service, cve_serializer): 优化导入和移除未使用模块
+
+# 多个文件都是格式调整 -> 合并为一条
+style(auth, api, utils): 统一代码格式化
+```
+
+### 6. 反馈学习
+
+acm 会自动记录你的偏好到本地（`~/.acm/feedback.json`），包括：
+
+- 使用反馈重新生成（`r`）时：记录原始 message、反馈内容、改进后 message
+- 手动编辑（`e`）时：记录编辑前后的 message 差异
+
+后续生成时，acm 会**智能匹配**与当前 diff 最相关的历史反馈（按 scope/关键词匹配，最多注入 5 条），注入 prompt 让 LLM 学习你的偏好。最多保留最近 20 条记录。
+
+```
+# 第一次使用：LLM 生成了 feat(auth): 添加登录功能
+# 你反馈：type 应该是 fix 不是 feat
+# LLM 改进为：fix(auth): 修复登录功能
+
+# 之后修改 auth 相关代码时，LLM 会自动参考这条反馈
+```
+
+管理反馈记录：
+
+```bash
+acm feedback list    # 查看所有历史反馈
+acm feedback clear   # 清空所有反馈记录
+```
+
+## 命令参考
+
+### `acm` — 生成 commit message（默认命令）
+
+```bash
+acm                # 生成并交互式提交
+acm --dry-run      # 仅生成，不执行 git commit
+acm --verbose      # 输出调试信息（配置、prompt、LLM 响应等）
+acm --split        # 智能拆分为多个原子提交
+acm --version      # 查看版本
+```
+
+### `acm init` — 交互式初始化配置
+
+```bash
+acm init           # 生成项目级配置 .acm.toml
+acm init --global  # 生成全局配置 ~/.acm/config.toml
+```
+
+### `acm config` — 配置管理
+
+```bash
+acm config list                        # 列出当前生效的所有配置
+acm config get llm.model               # 查看某个配置项
+acm config set llm.model qwen-turbo    # 修改项目级配置
+acm config set llm.model qwen-turbo --global  # 修改全局配置
+```
+
+### `acm feedback` — 反馈记录管理
+
+```bash
+acm feedback list    # 列出所有历史反馈记录
+acm feedback clear   # 清空所有反馈记录
+```
+
+## 配置文件
+
+### 配置文件位置
+
+| 优先级 | 路径 | 说明 |
+|--------|------|------|
+| 高 | 项目根目录 `.acm.toml` | 项目级配置，仅当前项目生效 |
+| 低 | `~/.acm/config.toml` | 全局配置，所有项目生效 |
+
+项目级配置会覆盖全局配置中的同名字段。
+
+### 配置文件示例
+
+```toml
+[llm]
+provider = "qwen"                # 提供商名称
+api_key = "sk-xxx"               # API Key（建议改用环境变量）
+model = "qwen-plus"              # 模型名称（必填）
+base_url = ""                    # 自定义 API 地址（留空使用提供商默认值）
+
+[commit]
+language = "zh-CN"               # 描述语言，默认中文
+max_diff_lines = 500             # diff 截断阈值（行数）
+types = ["build", "ci"]          # 自定义类型（追加到内置标准类型之上）
+
+[prompt]
+system = ""                      # 自定义 system prompt（留空使用内置默认）
+extra = ""                       # 追加到默认 prompt 之后的额外指令
+```
+
+### API Key 管理
+
+API Key 按以下优先级读取（高到低）：
+
+1. 环境变量 `ACM_API_KEY`
+2. 提供商专用环境变量（如通义千问的 `DASHSCOPE_API_KEY`）
+3. 配置文件中的 `llm.api_key`
+
+推荐使用环境变量，避免将密钥写入配置文件：
+
+```bash
+export ACM_API_KEY="sk-your-key-here"
+# 或通义千问专用
+export DASHSCOPE_API_KEY="sk-your-key-here"
+```
+
+## Commit Message 规范
+
+生成的 message 格式为 `type(scope): 中文描述`。
+
+### 内置 type 类型
+
+| Type | 说明 |
+|------|------|
+| `feat` | 新功能 |
+| `fix` | 缺陷修复 |
+| `docs` | 文档变更 |
+| `style` | 代码风格（不影响逻辑） |
+| `refactor` | 重构 |
+| `perf` | 性能优化 |
+| `test` | 测试相关 |
+| `chore` | 构建/工具变更 |
+
+可通过 `commit.types` 配置项追加自定义类型。
+
+### Scope 自动推断
+
+根据变更文件所在目录自动推断。例如修改了 `src/auth/login.py`，scope 为 `auth`。
+
+多模块变更时会用逗号分隔并分行描述：
+
+```
+feat(auth,api): 添加用户认证模块
+
+- auth: 实现基于JWT的登录认证逻辑
+- api: 新增 /login 和 /logout 接口
+```
+
+## 提供商配置
+
+### 通义千问（内置）
+
+```toml
+[llm]
+provider = "qwen"
+model = "qwen-plus"    # 或 qwen-turbo, qwen-max 等
+```
+
+### 其他 OpenAI 兼容提供商
+
+任何兼容 OpenAI Chat API 的服务都可以使用，只需配置 `base_url`：
+
+```toml
+# DeepSeek
+[llm]
+provider = "deepseek"
+model = "deepseek-chat"
+base_url = "https://api.deepseek.com/v1"
+
+# Ollama 本地模型
+[llm]
+provider = "ollama"
+model = "qwen2.5:7b"
+base_url = "http://localhost:11434/v1"
+
+# 智谱 AI
+[llm]
+provider = "zhipu"
+model = "glm-4-flash"
+base_url = "https://open.bigmodel.cn/api/paas/v4"
+```
+
+## 运行测试
+
+```bash
+python -m unittest discover -s tests -v
+```
+
+## License
+
+MIT

acm_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,18 @@
+acm/__init__.py,sha256=o2xtxQu69ewo2c8lCYSwrVOdlp5_IE4-Jt4NpT48r5Q,91
+acm/cli.py,sha256=-ZuZ_vQj-054TKtz_KfpOUsKSzNdFbi8IPRUIXs9jWo,14578
+acm/config.py,sha256=_KSyCCVSYYHDZpO6ZM8XejNVz-c67WDoVBl1cJmWCB4,8368
+acm/feedback.py,sha256=5whdUPyfX52TmoWHQgmKhoGbhtDVaUgEMUO2n4q3zc8,4542
+acm/generator.py,sha256=rcBuqY1P4R4BHE7xhV1wj31YQPEbPh64LVPLlHDB-MU,4561
+acm/git.py,sha256=aGSf1alIKGLC59QJsMq3j3ULUe5NvvYaJrkm0wUXwtM,6224
+acm/prompt.py,sha256=dOoOu8dhjdwuauTASeE1nH-kcvErXo33nDXVDfO39ME,6223
+acm/splitter.py,sha256=u3mWD529N47brzs-zrPEHvGCWxKHHOQLZ3kM9QeMTjc,2718
+acm/ui.py,sha256=pvlCP4qmpO1wS3q7tZYE37ULcUAG8tkzNcRKiCnMb8Y,8787
+acm/llm/__init__.py,sha256=_RCkUDHE8OdW1GDQzc1hn3G0XG51MSJNYHQasR5Ay04,841
+acm/llm/base.py,sha256=3Lby6OimlWdQF53L5AtX4Ky6DiSQiytb075_SZj7jTc,1647
+acm/llm/openai_compat.py,sha256=78QWbJFNGyp9waTC84VR5FBRf3TaLHYYo1AfFEsgO88,2540
+acm/llm/qwen.py,sha256=ARAC_Eu7l6zu-OhmZCTVi0GPEgbrmr0vt77J450OLKY,540
+acm_cli-0.1.0.dist-info/METADATA,sha256=FnT5vLAw54cTOlPE2CRiiN_Xdh2BdQcWZixA5jP1hwg,10246
+acm_cli-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+acm_cli-0.1.0.dist-info/entry_points.txt,sha256=MSdN3GKf5V7mnZExG70tTcMhw8gT62TDL87xnz3j1S4,37
+acm_cli-0.1.0.dist-info/licenses/LICENSE,sha256=R3-HWc6iPbskO4okGD3xiYoyWn8zkJo3qKEA9nP7ye4,1066
+acm_cli-0.1.0.dist-info/RECORD,,

acm_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

acm_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+acm = acm.cli:main

acm_cli-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 huangzhao
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.