@unifan/pi-commit-zh 1.0.7

package/index.ts

@@ -0,0 +1,216 @@
+import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import {
+	commitWithMsg,
+	getChangedFiles,
+	getStagedDiff,
+	getUnstagedDiff,
+	isGitRepo,
+	pushCurrentBranch,
+	stageAll,
+} from "./src/git.js";
+import { COMMIT_SYSTEM_PROMPT } from "./src/prompt.ts";
+
+interface ParsedCommitArgs {
+	yes: boolean;
+	stageAll: boolean;
+	hint?: string;
+}
+
+function parseArgs(raw: string): ParsedCommitArgs {
+	const tokens = raw.trim().split(/\s+/).filter(Boolean);
+	let yes = false;
+	let shouldStageAll = false;
+	const hintParts: string[] = [];
+
+	for (const t of tokens) {
+		if (t === "-y" || t === "--yes") {
+			yes = true;
+		} else if (t === "-a" || t === "--all") {
+			shouldStageAll = true;
+		} else {
+			hintParts.push(t);
+		}
+	}
+
+	return {
+		yes,
+		stageAll: shouldStageAll,
+		hint: hintParts.join(" ").trim() || undefined,
+	};
+}
+
+async function generateCommitMessage(
+	ctx: ExtensionCommandContext,
+	diff: string,
+	changedFiles: string[],
+	userHint?: string,
+): Promise<string> {
+	const truncatedDiff = diff.length > 30000 ? diff.slice(0, 30000) + "\n\n... (diff已截断)" : diff;
+	const userPrompt = [
+		`## 修改的文件清单 (${changedFiles.length} 个文件):`,
+		changedFiles.map((f) => `- ${f}`).join("\n"),
+		"",
+		userHint ? `## 用户额外补充说明:\n${userHint}\n` : "",
+		"## Git Diff 变动详情:",
+		"```diff",
+		truncatedDiff,
+		"```",
+	]
+		.filter(Boolean)
+		.join("\n");
+
+	if (ctx.model && ctx.modelRegistry) {
+		try {
+			const provider = ctx.modelRegistry.getProvider(ctx.model.provider);
+			if (provider) {
+				const response = await provider
+					.streamSimple(
+						ctx.model,
+						{
+							systemPrompt: COMMIT_SYSTEM_PROMPT,
+							messages: [
+								{
+									role: "user",
+									content: [{ type: "text", text: userPrompt }],
+									timestamp: Date.now(),
+								},
+							],
+						},
+						{ maxTokens: 800 },
+					)
+					.result();
+
+				const text = response.content
+					?.map((c) => (c.type === "text" ? c.text : ""))
+					.join("")
+					.trim();
+
+				if (text) {
+					// Clean any unexpected code blocks
+					return text.replace(/^```[a-zA-Z]*\n?/, "").replace(/\n?```$/, "").trim();
+				}
+			}
+		} catch {
+			/* fallback below */
+		}
+	}
+
+	// Simple heuristic fallback
+	const firstFile = changedFiles[0] ?? "core";
+	const scope = firstFile.split(/[/\\]/)[0] || "core";
+	return `chore(${scope}): 更新代码与相关配置\n\n- 更新了 ${changedFiles.length} 个文件`;
+}
+
+export default function (pi: ExtensionAPI) {
+	const handleCommitCommand = async (args: string, ctx: ExtensionCommandContext, andPush = false) => {
+		const notify = (msg: string, level: "info" | "warning" | "error" = "info") => {
+			if (ctx.hasUI) ctx.ui.notify(msg, level);
+			else console.log(`pi-commit: ${msg}`);
+		};
+
+		if (!(await isGitRepo(ctx.cwd))) {
+			notify("当前目录不是 Git 仓库，无法执行 commit。", "error");
+			return;
+		}
+
+		const parsed = parseArgs(args);
+		let stagedDiff = await getStagedDiff(ctx.cwd);
+		const unstagedDiff = await getUnstagedDiff(ctx.cwd);
+		const changedFiles = await getChangedFiles(ctx.cwd);
+
+		if (!stagedDiff && !unstagedDiff) {
+			notify("当前工作区没有任何修改，无需提交。", "info");
+			return;
+		}
+
+		if (!stagedDiff && unstagedDiff) {
+			notify("未发现暂存区文件，已自动暂存全部修改 (git add -A)...", "info");
+			await stageAll(ctx.cwd);
+			stagedDiff = await getStagedDiff(ctx.cwd);
+		} else if (parsed.stageAll) {
+			await stageAll(ctx.cwd);
+			stagedDiff = await getStagedDiff(ctx.cwd);
+		}
+
+		notify("正在深度分析代码改动并生成中文 Commit Message...", "info");
+		const commitMessage = await generateCommitMessage(ctx, stagedDiff || unstagedDiff, changedFiles, parsed.hint);
+
+		let finalMessage = commitMessage;
+
+		if (!parsed.yes && ctx.hasUI) {
+			const choice = await ctx.ui.select(
+				`✨ AI 生成的提交信息:\n\n${commitMessage}\n\n请选择操作:`,
+				[
+					{ label: "✅ 立即以此信息提交 (Commit)", value: "commit" },
+					{ label: "📝 编辑后再提交 (Edit & Commit)", value: "edit" },
+					{ label: "❌ 取消提交 (Cancel)", value: "cancel" },
+				],
+			);
+
+			if (!choice || choice === "cancel") {
+				notify("已取消本次提交。", "info");
+				return;
+			}
+
+			if (choice === "edit") {
+				const edited = await ctx.ui.editor("编辑 Commit Message", commitMessage);
+				if (!edited || !edited.trim()) {
+					notify("提交信息为空，已取消提交。", "warning");
+					return;
+				}
+				finalMessage = edited.trim();
+			}
+		}
+
+		const commitRes = await commitWithMsg(ctx.cwd, finalMessage);
+		if (!commitRes.ok) {
+			notify(`Git 提交失败: ${commitRes.output}`, "error");
+			return;
+		}
+
+		const firstLine = finalMessage.split("\n")[0];
+		notify(`✅ 成功提交: ${firstLine}`, "info");
+
+		let pushText = "";
+		if (andPush) {
+			notify("正在推送到远端仓库 (git push)...", "info");
+			const pushRes = await pushCurrentBranch(ctx.cwd);
+			if (pushRes.ok) {
+				notify("🚀 成功推送到远端仓库！", "info");
+				pushText = "\n\n🚀 **已自动推送到远端分支**";
+			} else {
+				notify(`⚠️ 推送失败: ${pushRes.output}`, "warning");
+				pushText = `\n\n⚠️ **推送到远端失败**: ${pushRes.output}`;
+			}
+		}
+
+		pi.sendMessage({
+			customType: "pi-commit-result",
+			content: `### 📦 Git 提交完成\n\n\`\`\`text\n${finalMessage}\n\`\`\`${pushText}`,
+			display: true,
+		});
+	};
+
+	pi.registerCommand("commit", {
+		description: "智能 Git 提交助手：自动分析 diff 生成标准中文 Conventional Commit 并提交 (-y 直接提交, -a 暂存全部)",
+		getArgumentCompletions: (prefix: string) => {
+			const options = [
+				{ value: "-y", label: "-y / --yes", description: "直接提交无需二次确认" },
+				{ value: "-a", label: "-a / --all", description: "自动暂存全部修改 (git add -A)" },
+			];
+			const trimmed = prefix.trimStart();
+			if (!trimmed) return options;
+			return options.filter((o) => o.value.startsWith(trimmed));
+		},
+		handler: async (args, ctx) => {
+			await handleCommitCommand(args, ctx, false);
+		},
+	});
+
+	pi.registerCommand("commit-push", {
+		description: "智能 Git 提交并推流：生成标准中文 Commit 后自动执行 git push",
+		handler: async (args, ctx) => {
+			await handleCommitCommand(args, ctx, true);
+		},
+	});
+}

package/package.json

@@ -0,0 +1,30 @@
+{
+  "name": "@unifan/pi-commit-zh",
+  "version": "1.0.7",
+  "description": "Pi 智能 Git 提交助手（规范化 Conventional Commits 纯中文版，支持一键提审与推流）",
+  "type": "module",
+  "main": "index.ts",
+  "author": "821869798",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/821869798/pi-unifan-zh.git"
+  },
+  "keywords": [
+    "pi-package",
+    "pi-extension",
+    "git",
+    "commit",
+    "conventional-commits",
+    "chinese"
+  ],
+  "peerDependencies": {
+    "@earendil-works/pi-ai": ">=0.74.0 <1.0.0",
+    "@earendil-works/pi-coding-agent": ">=0.74.0 <1.0.0"
+  },
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  }
+}

package/src/git.ts

@@ -0,0 +1,61 @@
+import { execFile } from "node:child_process";
+import { promisify } from "node:util";
+
+const execFileAsync = promisify(execFile);
+
+export async function runGit(cwd: string, args: string[]): Promise<{ stdout: string; stderr: string; ok: boolean }> {
+	try {
+		const res = await execFileAsync("git", args, { cwd, maxBuffer: 10 * 1024 * 1024 });
+		return { stdout: res.stdout, stderr: res.stderr, ok: true };
+	} catch (err: any) {
+		return {
+			stdout: err?.stdout ?? "",
+			stderr: err?.stderr ?? err?.message ?? String(err),
+			ok: false,
+		};
+	}
+}
+
+export async function isGitRepo(cwd: string): Promise<boolean> {
+	const res = await runGit(cwd, ["rev-parse", "--is-inside-work-tree"]);
+	return res.ok && res.stdout.trim() === "true";
+}
+
+export async function getGitStatus(cwd: string): Promise<string> {
+	const res = await runGit(cwd, ["status", "--porcelain"]);
+	return res.ok ? res.stdout.trim() : "";
+}
+
+export async function getStagedDiff(cwd: string): Promise<string> {
+	const res = await runGit(cwd, ["diff", "--cached"]);
+	return res.ok ? res.stdout : "";
+}
+
+export async function getUnstagedDiff(cwd: string): Promise<string> {
+	const res = await runGit(cwd, ["diff"]);
+	return res.ok ? res.stdout : "";
+}
+
+export async function getChangedFiles(cwd: string): Promise<string[]> {
+	const res = await runGit(cwd, ["status", "--porcelain"]);
+	if (!res.ok || !res.stdout.trim()) return [];
+	return res.stdout
+		.split("\n")
+		.map((line) => line.trim().slice(3).trim())
+		.filter(Boolean);
+}
+
+export async function stageAll(cwd: string): Promise<boolean> {
+	const res = await runGit(cwd, ["add", "-A"]);
+	return res.ok;
+}
+
+export async function commitWithMsg(cwd: string, message: string): Promise<{ ok: boolean; output: string }> {
+	const res = await runGit(cwd, ["commit", "-m", message]);
+	return { ok: res.ok, output: res.ok ? res.stdout : res.stderr };
+}
+
+export async function pushCurrentBranch(cwd: string): Promise<{ ok: boolean; output: string }> {
+	const res = await runGit(cwd, ["push"]);
+	return { ok: res.ok, output: res.ok ? res.stdout : res.stderr };
+}

package/src/prompt.ts

@@ -0,0 +1,24 @@
+export const COMMIT_SYSTEM_PROMPT = `你是专业的 Git 提交信息生成专家（Conventional Commits 中文版）。
+你的唯一任务是根据提供的 Git Diff 变动和修改文件列表，生成规范、准确、地道的中文 Commit Message。
+
+## 严格遵循的标准（Conventional Commits 1.0.0）：
+1. 格式：<type>(<scope>): <中文描述>
+   - type 必须为标准英文类型之一：
+     • feat: 新增功能/新模块/新特性
+     • fix: 修复Bug/缺陷
+     • perf: 性能优化/降低GC/提高帧率
+     • refactor: 代码重构/结构优化
+     • docs: 文档/注释更新
+     • style: 格式调整/无逻辑影响
+     • test: 单元测试/基准测试
+     • chore: 构建/依赖/配置杂项
+   - scope 为具体修改的业务模块名（如：combat/战斗、session/会话、review/审查、ui/界面、network/网络、core/核心）
+   - 中文描述：50个汉字以内，动词开头，言简意赅，末尾严禁加句号。
+
+2. 若修改较为复杂，可附带简要中文正文列表（Body）：
+   - 核心改动点 1
+   - 核心改动点 2
+
+3. 语言约束：所有标题描述与正文说明必须且只能使用纯正中文，严禁长句英文。
+
+4. 输出格式：直接输出生成的 Commit 信息文本（首行标题，空行后跟正文），不要附加任何额外的 markdown 代码块或寒暄。`;