chore: create src, dist, .github directories (#36)

src/aicommits.ts

@@ -0,0 +1,121 @@
+#!/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. 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, "");
+}

src/index.ts

@@ -0,0 +1,6 @@
+#! /usr/bin/env node
+import { main } from "./aicommits";
+
+(async () => {
+  await main();
+})();