@unifan/pi-commit-zh 1.0.9 → 1.0.11

package/index.ts

@@ -3,37 +3,33 @@ import {
 	commitWithMsg,
 	getChangedFiles,
 	getStagedDiff,
+	getUnpushedCommits,
 	getUnstagedDiff,
 	isGitRepo,
 	pushCurrentBranch,
 	stageAll,
 } from "./src/git.js";
-import { COMMIT_SYSTEM_PROMPT } from "./src/prompt.ts";
+import { COMMIT_SYSTEM_PROMPT } from "./src/prompt.js";
 
 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") {
+		if (t === "-a" || t === "--all") {
 			shouldStageAll = true;
-		} else {
+		} else if (t !== "-y" && t !== "--yes") {
 			hintParts.push(t);
 		}
 	}
 
 	return {
-		yes,
 		stageAll: shouldStageAll,
 		hint: hintParts.join(" ").trim() || undefined,
 	};
@@ -63,6 +59,7 @@ async function generateCommitMessage(
 		try {
 			const provider = ctx.modelRegistry.getProvider(ctx.model.provider);
 			if (provider) {
+				const auth = await ctx.modelRegistry.getApiKeyAndHeaders(ctx.model);
 				const response = await provider
 					.streamSimple(
 						ctx.model,
@@ -76,7 +73,7 @@ async function generateCommitMessage(
 								},
 							],
 						},
-						{ maxTokens: 800 },
+						{ apiKey: auth?.apiKey, headers: auth?.headers, maxTokens: 800 },
 					)
 					.result();
 
@@ -86,16 +83,15 @@ async function generateCommitMessage(
 					.trim();
 
 				if (text) {
-					// Clean any unexpected code blocks
 					return text.replace(/^```[a-zA-Z]*\n?/, "").replace(/\n?```$/, "").trim();
 				}
 			}
-		} catch {
-			/* fallback below */
+		} catch (err) {
+			console.error("pi-commit streamSimple error:", err);
 		}
 	}
 
-	// Simple heuristic fallback
+	// 智能保底推断
 	const firstFile = changedFiles[0] ?? "core";
 	const scope = firstFile.split(/[/\\]/)[0] || "core";
 	return `chore(${scope}): 更新代码与相关配置\n\n- 更新了 ${changedFiles.length} 个文件`;
@@ -119,7 +115,28 @@ export default function (pi: ExtensionAPI) {
 		const changedFiles = await getChangedFiles(ctx.cwd);
 
 		if (!stagedDiff && !unstagedDiff) {
-			notify("当前工作区没有任何修改，无需提交。", "info");
+			if (andPush) {
+				const unpushed = await getUnpushedCommits(ctx.cwd);
+				if (unpushed.length > 0) {
+					notify(`当前工作区干净，但检测到有 ${unpushed.length} 个未推送提交，正在推送到远端...`, "info");
+					const pushRes = await pushCurrentBranch(ctx.cwd, (msg) => notify(msg, "info"));
+					if (pushRes.ok) {
+						const successMsg = pushRes.autoRebased
+							? `🚀 已自动完成变基 (pull --rebase)，并成功将 ${unpushed.length} 个本地提交推送至远端！`
+							: `🚀 成功将本地未推送的 ${unpushed.length} 个提交推送至远端！`;
+						notify(successMsg, "info");
+						pi.sendMessage({
+							customType: "pi-commit-result",
+							content: `### 🚀 远端推送完成\n\n当前工作区无未提交的修改，已成功将本地 **${unpushed.length} 个历史提交** 推送至远端分支：\n\n\`\`\`text\n${unpushed.join("\n")}\n\`\`\`\n\n${successMsg}`,
+							display: true,
+						});
+					} else {
+						notify(`⚠️ 推送失败: ${pushRes.output}`, "error");
+					}
+					return;
+				}
+			}
+			notify("当前工作区干净，且所有本地提交均已同步至远端，无需提交和推送。", "info");
 			return;
 		}
 
@@ -135,40 +152,13 @@ export default function (pi: ExtensionAPI) {
 		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);
+		const commitRes = await commitWithMsg(ctx.cwd, commitMessage);
 		if (!commitRes.ok) {
 			notify(`Git 提交失败: ${commitRes.output}`, "error");
 			return;
 		}
 
-		const firstLine = finalMessage.split("\n")[0];
+		const firstLine = commitMessage.split("\n")[0];
 		notify(`✅ 成功提交: ${firstLine}`, "info");
 
 		let pushText = "";
@@ -189,16 +179,15 @@ export default function (pi: ExtensionAPI) {
 
 		pi.sendMessage({
 			customType: "pi-commit-result",
-			content: `### 📦 Git 提交完成\n\n\`\`\`text\n${finalMessage}\n\`\`\`${pushText}`,
+			content: `### 📦 Git 提交完成\n\n\`\`\`text\n${commitMessage}\n\`\`\`${pushText}`,
 			display: true,
 		});
 	};
 
 	pi.registerCommand("commit", {
-		description: "智能 Git 提交助手：自动分析 diff 生成标准中文 Conventional Commit 并提交 (-y 直接提交, -a 暂存全部)",
+		description: "智能 Git 提交助手：自动分析 diff 生成标准中文 Commit 并直接提交 (-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();
@@ -211,7 +200,7 @@ export default function (pi: ExtensionAPI) {
 	});
 
 	pi.registerCommand("commit-push", {
-		description: "智能 Git 提交并推流：生成标准中文 Commit 后自动执行 git push",
+		description: "智能 Git 提交并推流：生成标准中文 Commit 后自动提交并执行 git push (支持自动变基与未推送提交自动推流)",
 		handler: async (args, ctx) => {
 			await handleCommitCommand(args, ctx, true);
 		},

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@unifan/pi-commit-zh",
-  "version": "1.0.9",
+  "version": "1.0.11",
   "description": "Pi 智能 Git 提交助手（规范化 Conventional Commits 纯中文版，支持一键提审与推流）",
   "type": "module",
   "main": "index.ts",

package/src/git.ts

@@ -45,6 +45,27 @@ export async function getChangedFiles(cwd: string): Promise<string[]> {
 		.filter(Boolean);
 }
 
+export async function getUnpushedCommits(cwd: string): Promise<string[]> {
+	const res = await runGit(cwd, ["log", "@{u}..HEAD", "--oneline"]);
+	if (res.ok && res.stdout.trim()) {
+		return res.stdout.trim().split("\n").filter(Boolean);
+	}
+	const statusRes = await runGit(cwd, ["status", "--porcelain=v1", "-b"]);
+	if (statusRes.ok) {
+		const match = statusRes.stdout.match(/\[ahead\s+(\d+)\]/);
+		if (match && match[1]) {
+			const count = parseInt(match[1], 10);
+			if (count > 0) {
+				const logRes = await runGit(cwd, ["log", "-n", String(count), "--oneline"]);
+				if (logRes.ok && logRes.stdout.trim()) {
+					return logRes.stdout.trim().split("\n").filter(Boolean);
+				}
+			}
+		}
+	}
+	return [];
+}
+
 export async function stageAll(cwd: string): Promise<boolean> {
 	const res = await runGit(cwd, ["add", "-A"]);
 	return res.ok;
@@ -59,12 +80,22 @@ export async function pushCurrentBranch(
 	cwd: string,
 	onLog?: (msg: string) => void,
 ): Promise<{ ok: boolean; output: string; autoRebased?: boolean }> {
-	const initialPush = await runGit(cwd, ["push"]);
+	let initialPush = await runGit(cwd, ["push"]);
 	if (initialPush.ok) {
 		return { ok: true, output: initialPush.stdout || "推送成功" };
 	}
 
-	const pushError = `${initialPush.stderr} ${initialPush.stdout}`.toLowerCase();
+	let pushError = `${initialPush.stderr} ${initialPush.stdout}`.toLowerCase();
+
+	if (pushError.includes("no upstream branch") || pushError.includes("set-upstream")) {
+		if (onLog) onLog("未关联远端分支，正在自动关联并推送 (git push -u origin HEAD)...");
+		const setUpstream = await runGit(cwd, ["push", "-u", "origin", "HEAD"]);
+		if (setUpstream.ok) {
+			return { ok: true, output: setUpstream.stdout || "推送成功" };
+		}
+		pushError = `${setUpstream.stderr} ${setUpstream.stdout}`.toLowerCase();
+	}
+
 	const needsPull =
 		pushError.includes("fetch first") ||
 		pushError.includes("non-fast-forward") ||