aicommits.ts

#!/usr/bin/env node

import chalk from "chalk";
import { execSync } from "child_process";
import inquirer from "inquirer";
import fetch from "node-fetch";

let OPENAI_KEY = process.env.OPENAI_KEY ?? process.env.OPENAI_API_KEY;

export async function main() {
  console.log(chalk.white("▲ ") + chalk.green("Welcome to AICommits!"));

  if (!OPENAI_KEY) {
    console.error(
      chalk.white("▲ ") +
        "Please save your OpenAI API key as an env variable by doing 'export OPENAI_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 . ":(exclude)package-lock.json" ":(exclude)yarn.lock" ":(exclude)pnpm-lock.yaml"`,
    {
      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 in Korean language. Do not preface the commit with anything, use the present tense, return a complete sentence, and do not repeat yourself: ${diff}`;

  console.log(
    chalk.white("▲ ") + chalk.gray("Generating your AI commit message...\n")
  );

  try {
    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",
    });
  } catch (e) {
    console.error(chalk.white("▲ ") + chalk.red(e.message));
    process.exit(1);
  }
}

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_KEY ?? ""}`,
    },
    method: "POST",
    body: JSON.stringify(payload),
  });
  if (response.status !== 200) {
    const errorJson: any = await response.json();
    throw new Error(
      `OpenAI API failed while processing the request '${errorJson?.error?.message}'`,
    );
  }

  const json: any = await response.json();
  const aiCommit = json.choices[0].text;

  return aiCommit.replace(/(\r\n|\n|\r)/gm, "");
}