2 Commits

+175 -38
View File
@@ -27,9 +27,10 @@ function showHelp() {
${BOLD}gai${RESET} — AI-powered git commit message generator ${BOLD}gai${RESET} — AI-powered git commit message generator
${BOLD}Usage:${RESET} ${BOLD}Usage:${RESET}
gai Generate commit message for staged/changed files gai Open interactive menu
gai --auto Auto-stage all changed files gai commit Generate commit message for staged/changed files
gai --dry-run Generate message without committing gai commit --auto Auto-stage all changed files
gai commit -d Generate message without committing
gai config Configure API settings gai config Configure API settings
gai --help Show this help message gai --help Show this help message
gai --version Show version gai --version Show version
@@ -38,7 +39,7 @@ ${BOLD}Configuration:${RESET}
Set via ${CYAN}gai config${RESET} or environment variables: Set via ${CYAN}gai config${RESET} or environment variables:
GAI_API_KEY OpenAI-compatible API key GAI_API_KEY OpenAI-compatible API key
GAI_API_BASE API base URL (default: https://api.deepseek.com/v1) GAI_API_BASE API base URL (default: https://api.deepseek.com/v1)
GAI_MODEL Model name (default: deepseek-v4-flash) GAI_MODEL Model name (default: deepseek-chat)
GAI_MAX_TOKENS Max tokens (default: 500) GAI_MAX_TOKENS Max tokens (default: 500)
GAI_TEMPERATURE Temperature (default: 0.7) GAI_TEMPERATURE Temperature (default: 0.7)
`); `);
@@ -119,11 +120,6 @@ async function editMessage(current: string): Promise<string | null> {
const ENTER = "\r"; const ENTER = "\r";
const CTRL_C = "\x03"; const CTRL_C = "\x03";
const BACKSPACE = "\x7f"; const BACKSPACE = "\x7f";
const DELETE = "\x1b[3~";
const LEFT = "\x1b[D";
const RIGHT = "\x1b[C";
const HOME = "\x1b[H" /* or \x01 */;
const END = "\x1b[F" /* or \x05 */;
function render() { function render() {
process.stdout.write("\x1b[2K\r > " + buffer); process.stdout.write("\x1b[2K\r > " + buffer);
@@ -253,25 +249,151 @@ async function editMessage(current: string): Promise<string | null> {
}); });
} }
async function main() { function printCommitResult(
if (args.includes("--help") || args.includes("-h")) { result: { branch: string; hash: string; files: number; insertions: number; deletions: number },
showHelp(); msg: string,
) {
console.log(`\n ${GREEN}${BOLD}✔ Committed successfully!${RESET}`);
const id = result.branch && result.hash
? `${YELLOW}[${result.branch} ${result.hash}]${RESET}`
: result.hash
? `${YELLOW}${result.hash}${RESET}`
: "";
console.log(` ${id} ${msg}`);
const parts: string[] = [];
if (result.files > 0) parts.push(`${YELLOW}${result.files} file${result.files > 1 ? "s" : ""} changed${RESET}`);
if (result.insertions > 0) parts.push(`${GREEN}${result.insertions} insertion${result.insertions > 1 ? "s" : ""}(+)${RESET}`);
if (result.deletions > 0) parts.push(`${RED}${result.deletions} deletion${result.deletions > 1 ? "s" : ""}(-)${RESET}`);
if (parts.length > 0) console.log(` ${parts.join(", ")}`);
}
interface MenuAction {
key: string;
label: string;
description: string;
}
const MENU_ACTIONS: MenuAction[] = [
{ key: "commit", label: "commit", description: "Generate AI commit message" },
];
async function showMenu(): Promise<void> {
const actions = MENU_ACTIONS;
let cursor = 0;
const headerLines = 4;
process.stdout.write(`\n ${BOLD}gai${RESET}\n`);
process.stdout.write(` ${DIM}↑/↓ navigate, space/enter select${RESET}\n\n`);
const totalLines = headerLines + actions.length;
function render() {
for (let i = 0; i < actions.length; i++) {
process.stdout.write("\x1b[2K\r");
const a = actions[i]!;
const pointer = i === cursor ? `${CYAN}${RESET} ` : " ";
const dot = i === cursor ? `${GREEN}${RESET}` : `${DIM}${RESET}`;
const name = i === cursor ? `${BOLD}${a.label}${RESET}` : a.label;
const desc = i === cursor ? a.description : `${DIM}${a.description}${RESET}`;
process.stdout.write(`${pointer} ${dot} ${name}${" ".repeat(Math.max(1, 14 - a.label.length))}${desc}\n`);
}
process.stdout.write(`\x1b[${actions.length}A`);
}
function clearMenu() {
process.stdout.write(`\x1b[${totalLines}A`);
for (let i = 0; i < totalLines; i++) {
process.stdout.write("\x1b[2K\x1b[1B");
}
process.stdout.write(`\x1b[${totalLines}A`);
}
if (!process.stdin.isTTY) {
console.error(` ${RED}Error: Interactive menu 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 < actions.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();
return; return;
} }
if (args.includes("--version") || args.includes("-v")) { if (key === "\x1b" || key.startsWith("\x1b[")) {
console.log("gai v0.1.0"); escapeBuf = key;
if (key.length >= 3) {
handleSeq(key);
escapeBuf = "";
}
return; return;
} }
if (args[0] === "config") { if (escapeBuf) {
await handleConfig(); escapeBuf += key;
if (/^[A-Za-z~]$/.test(key)) {
handleSeq(escapeBuf);
escapeBuf = "";
} else if (escapeBuf.length > 8) {
escapeBuf = "";
}
return; return;
} }
const autoMode = args.includes("--auto") || args.includes("-a"); if (key === " " || key === "\r") {
const dryRun = args.includes("--dry-run") || args.includes("-d"); const selected = actions[cursor]!;
process.stdin.setRawMode(savedRaw === true);
process.stdin.pause();
process.stdin.removeAllListeners("data");
clearMenu();
process.stdout.write("\x1b[?25h");
if (selected.key === "commit") {
handleCommit(false, false).then(resolve);
} else {
resolve();
}
return;
}
});
});
}
async function handleCommit(autoMode: boolean, dryRun: boolean): Promise<void> {
const config = await loadConfig(); const config = await loadConfig();
if (!config.apiKey) { if (!config.apiKey) {
@@ -355,26 +477,6 @@ async function main() {
return; return;
} }
function printCommitResult(
result: { branch: string; hash: string; files: number; insertions: number; deletions: number },
msg: string,
) {
console.log(`\n ${GREEN}${BOLD}✔ Committed successfully!${RESET}`);
const id = result.branch && result.hash
? `${YELLOW}[${result.branch} ${result.hash}]${RESET}`
: result.hash
? `${YELLOW}${result.hash}${RESET}`
: "";
console.log(` ${id} ${msg}`);
const parts: string[] = [];
if (result.files > 0) parts.push(`${YELLOW}${result.files} file${result.files > 1 ? "s" : ""} changed${RESET}`);
if (result.insertions > 0) parts.push(`${GREEN}${result.insertions} insertion${result.insertions > 1 ? "s" : ""}(+)${RESET}`);
if (result.deletions > 0) parts.push(`${RED}${result.deletions} deletion${result.deletions > 1 ? "s" : ""}(-)${RESET}`);
if (parts.length > 0) console.log(` ${parts.join(", ")}`);
}
const action = await confirmCommit(message); const action = await confirmCommit(message);
if (action === "y") { if (action === "y") {
@@ -412,6 +514,41 @@ function printCommitResult(
} }
} }
async function main() {
if (args.includes("--help") || args.includes("-h")) {
showHelp();
return;
}
if (args.includes("--version") || args.includes("-v")) {
console.log("gai v0.1.0");
return;
}
const subcommand = args[0];
if (subcommand === "config") {
await handleConfig();
return;
}
if (subcommand === "commit") {
const autoMode = args.includes("--auto") || args.includes("-a");
const dryRun = args.includes("--dry-run") || args.includes("-d");
await handleCommit(autoMode, dryRun);
return;
}
if (!subcommand) {
await showMenu();
return;
}
console.error(` ${RED}Unknown command: ${subcommand}${RESET}`);
showHelp();
process.exit(1);
}
main().catch((err) => { main().catch((err) => {
console.error(` ${RED}Unexpected error: ${err}${RESET}`); console.error(` ${RED}Unexpected error: ${err}${RESET}`);
process.exit(1); process.exit(1);