105 lines
2.8 KiB
JavaScript
105 lines
2.8 KiB
JavaScript
#!/usr/bin/env node
|
|
|
|
import chalk from "chalk";
|
|
import { execSync } from "child_process";
|
|
import inquirer from "inquirer";
|
|
import fetch from "node-fetch";
|
|
|
|
let OPENAI_API_KEY = process.env.OPENAI_API_KEY;
|
|
|
|
export async function main() {
|
|
console.log(chalk.white("▲ ") + chalk.green("Welcome to AICommit!"));
|
|
|
|
if (!OPENAI_API_KEY) {
|
|
console.error(
|
|
chalk.white("▲ ") +
|
|
"Please specify an OpenAI key using export OPEN_AI_KEY='YOUR_API_KEY'"
|
|
);
|
|
process.exit(1);
|
|
}
|
|
try {
|
|
execSync("git rev-parse --is-inside-work-tree", {
|
|
encoding: "utf8",
|
|
stdio: "ignore",
|
|
});
|
|
} catch (e) {
|
|
console.error(chalk.white("▲ ") + "This is not a git repository");
|
|
process.exit(1);
|
|
}
|
|
|
|
const diff = execSync("git diff --cached", { encoding: "utf8" });
|
|
|
|
if (!diff) {
|
|
console.log(
|
|
chalk.white("▲ ") +
|
|
"No staged changes found. Make sure there are changes and run `git add .`"
|
|
);
|
|
process.exit(1);
|
|
}
|
|
|
|
// Accounting for GPT-3's input req of 4k tokens (approx 8k chars)
|
|
if (diff.length > 8000) {
|
|
console.log(
|
|
chalk.white("▲ ") + "The diff is too large to write a commit message."
|
|
);
|
|
process.exit(1);
|
|
}
|
|
|
|
let prompt = `I want you to act like a git commit message writer. I will input a git diff and your job is to convert it into a useful commit message. Do not preface the commit with anything, return a complete sentence, and do not repeat yourself: ${diff}`;
|
|
|
|
console.log(
|
|
chalk.white("▲ ") + chalk.gray("Generating your AI commit message...\n")
|
|
);
|
|
const aiCommitMessage = await generateCommitMessage(prompt);
|
|
|
|
console.log(
|
|
chalk.white("▲ ") + chalk.bold("Commit message: ") + aiCommitMessage + "\n"
|
|
);
|
|
|
|
const confirmationMessage = await inquirer.prompt([
|
|
{
|
|
name: "useCommitMessage",
|
|
message: "Would you like to use this commit message? (Y / n)",
|
|
choices: ["Y", "y", "n"],
|
|
default: "y",
|
|
},
|
|
]);
|
|
|
|
if (confirmationMessage.useCommitMessage === "n") {
|
|
console.log(chalk.white("▲ ") + "Commit message has not been commited.");
|
|
process.exit(1);
|
|
}
|
|
|
|
execSync(`git commit -m "${aiCommitMessage}"`, {
|
|
stdio: "inherit",
|
|
encoding: "utf8",
|
|
});
|
|
}
|
|
|
|
async function generateCommitMessage(prompt: string) {
|
|
const payload = {
|
|
model: "text-davinci-003",
|
|
prompt,
|
|
temperature: 0.7,
|
|
top_p: 1,
|
|
frequency_penalty: 0,
|
|
presence_penalty: 0,
|
|
max_tokens: 200,
|
|
stream: false,
|
|
n: 1,
|
|
};
|
|
const response = await fetch("https://api.openai.com/v1/completions", {
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${OPENAI_API_KEY ?? ""}`,
|
|
},
|
|
method: "POST",
|
|
body: JSON.stringify(payload),
|
|
});
|
|
|
|
const json: any = await response.json();
|
|
const aiCommit = json.choices[0].text;
|
|
|
|
return aiCommit.replace(/(\r\n|\n|\r)/gm, "");
|
|
}
|