feat(cli): add pull request creation with AI-generated messages #2

Merged
Mplan merged 2 commits from feat/pr into main 2026-06-11 00:39:21 +08:00
2 changed files with 144 additions and 16 deletions
Showing only changes of commit 57af72664d - Show all commits
+124 -6
View File
@@ -25,10 +25,12 @@ import {
getBranchCommits, getBranchCommits,
getBranchDiff, getBranchDiff,
detectPlatform, detectPlatform,
getRemoteHostname,
checkCLI, checkCLI,
checkAuth, checkAuth,
createPR, createPR,
} from "./src/pr"; } from "./src/pr";
import type { Platform } from "./src/pr";
import { PR_SYSTEM_PROMPT, buildPRPrompt } from "./src/prompt"; import { PR_SYSTEM_PROMPT, buildPRPrompt } from "./src/prompt";
import { generatePRMessage } from "./src/ai"; import { generatePRMessage } from "./src/ai";
@@ -413,6 +415,119 @@ async function showMenu(): Promise<void> {
}); });
} }
async function selectPlatform(hostname: string): Promise<Platform | null> {
const options = [
{ platform: "github" as Platform, label: "GitHub", desc: "gh CLI" },
{ platform: "gitea" as Platform, label: "Gitea", desc: "tea CLI" },
];
let cursor = 0;
const headerLines = 4;
process.stdout.write(`\n Remote: ${CYAN}${hostname}${RESET} — could not auto-detect platform.\n`);
process.stdout.write(` ${DIM}↑/↓ navigate, space/enter select${RESET}\n\n`);
const totalLines = headerLines + options.length;
function render() {
for (let i = 0; i < options.length; i++) {
process.stdout.write("\x1b[2K\r");
const opt = options[i]!;
const pointer = i === cursor ? `${CYAN}${RESET} ` : " ";
const dot = i === cursor ? `${GREEN}${RESET}` : `${DIM}${RESET}`;
const name = i === cursor ? `${BOLD}${opt.label}${RESET}` : opt.label;
const desc = i === cursor ? opt.desc : `${DIM}${opt.desc}${RESET}`;
process.stdout.write(`${pointer} ${dot} ${name}${" ".repeat(Math.max(1, 10 - opt.label.length))}${desc}\n`);
}
process.stdout.write(`\x1b[${options.length}A`);
}
function clearMenu() {
process.stdout.write(`\x1b[${headerLines}A`);
for (let i = 0; i < totalLines; i++) {
process.stdout.write("\r\x1b[2K\n");
}
process.stdout.write(`\x1b[${totalLines}A`);
}
if (!process.stdin.isTTY) {
console.error(` ${RED}Error: Platform selection requires a TTY.${RESET}`);
process.exit(1);
}
const savedRaw = process.stdin.isRaw;
process.stdin.setRawMode(true);
process.stdin.resume();
process.stdout.write("\x1b[?25l");
render();
return new Promise((resolve) => {
let escapeBuf = "";
function handleSeq(seq: string) {
if (seq === "\x1b[A" || seq === "\x1bOA") {
if (cursor > 0) {
cursor--;
render();
}
} else if (seq === "\x1b[B" || seq === "\x1bOB") {
if (cursor < options.length - 1) {
cursor++;
render();
}
}
}
process.stdin.on("data", (data: Buffer) => {
const key = data.toString();
if (key === "\x03") {
process.stdin.setRawMode(savedRaw === true);
process.stdin.pause();
process.stdin.removeAllListeners("data");
clearMenu();
process.stdout.write("\x1b[?25h");
resolve(null);
return;
}
if (key === "\x1b" || key.startsWith("\x1b[")) {
escapeBuf = key;
if (key.length >= 3) {
handleSeq(key);
escapeBuf = "";
}
return;
}
if (escapeBuf) {
escapeBuf += key;
if (/^[A-Za-z~]$/.test(key)) {
handleSeq(escapeBuf);
escapeBuf = "";
} else if (escapeBuf.length > 8) {
escapeBuf = "";
}
return;
}
if (key === " " || key === "\r") {
const selected = options[cursor]!;
process.stdin.setRawMode(savedRaw === true);
process.stdin.pause();
process.stdin.removeAllListeners("data");
clearMenu();
process.stdout.write("\x1b[?25h");
resolve(selected.platform);
return;
}
});
});
}
async function handleCommit(autoMode: boolean, dryRun: boolean): Promise<void> { async function handleCommit(autoMode: boolean, dryRun: boolean): Promise<void> {
const config = await loadConfig(); const config = await loadConfig();
@@ -549,16 +664,19 @@ async function handlePR(draft: boolean): Promise<void> {
process.exit(1); process.exit(1);
} }
const platform = await detectPlatform(); let platform = await detectPlatform();
if (!platform) { if (!platform) {
console.error( const hostname = (await getRemoteHostname()) || "unknown";
` ${RED}Error: Could not detect GitHub or Gitea from origin remote URL.${RESET}`, const chosen = await selectPlatform(hostname);
); if (!chosen) {
process.exit(1); console.log(" Aborted.");
process.exit(0);
}
platform = chosen;
} }
const platformLabel = platform === "github" ? "GitHub" : "Gitea"; const platformLabel = platform === "github" ? "GitHub" : "Gitea";
console.log(` Remote platform: ${CYAN}${platformLabel}${RESET}`); console.log(` Using: ${CYAN}${platformLabel}${RESET}`);
const cliError = checkCLI(platform); const cliError = checkCLI(platform);
if (cliError) { if (cliError) {
+20 -10
View File
@@ -51,30 +51,40 @@ export async function getBranchDiff(base: string): Promise<string> {
} }
} }
function parseRemoteHostname(url: string): string | null {
const hostname = url
.trim()
.toLowerCase()
.replace(/^(https?:\/\/|ssh:\/\/|git:\/\/)/, "")
.replace(/^[^@]+@/, "")
.split(/[:/]/)[0];
return hostname || null;
}
export async function detectPlatform(): Promise<Platform | null> { export async function detectPlatform(): Promise<Platform | null> {
try { try {
const url = await Bun.$`git remote get-url origin`.quiet().text(); const url = await Bun.$`git remote get-url origin`.quiet().text();
const trimmed = url.trim().toLowerCase(); const hostname = parseRemoteHostname(url);
const hostname = trimmed
.replace(/^(https?:\/\/|ssh:\/\/|git:\/\/)/, "")
.replace(/^[^@]+@/, "")
.split(/[:/]/)[0];
if (!hostname) return null; if (!hostname) return null;
if (hostname === "github.com") return "github"; if (hostname === "github.com") return "github";
if (hostname.includes("gitea")) return "gitea"; if (hostname.includes("gitea")) return "gitea";
if (Bun.which("tea")) return "gitea";
if (Bun.which("gh")) return "github";
return null; return null;
} catch { } catch {
return null; return null;
} }
} }
export async function getRemoteHostname(): Promise<string | null> {
try {
const url = await Bun.$`git remote get-url origin`.quiet().text();
return parseRemoteHostname(url);
} catch {
return null;
}
}
export function checkCLI(platform: Platform): string | null { export function checkCLI(platform: Platform): string | null {
const bin = platform === "github" ? "gh" : "tea"; const bin = platform === "github" ? "gh" : "tea";
const path = Bun.which(bin); const path = Bun.which(bin);