refactor: consolidated src to one file (#39)

2023-02-15 02:31:16 -05:00
parent e4a9ea85fb
commit 2d1db36b9d
2 changed files with 35 additions and 41 deletions
--- a/src/cli.ts
+++ b/src/cli.ts
@@ -0,0 +1,121 @@
+#!/usr/bin/env node
+
+import { execSync } from 'child_process';
+import chalk from 'chalk';
+import inquirer from 'inquirer';
+import fetch from 'node-fetch';
+
+const OPENAI_KEY = process.env.OPENAI_KEY ?? process.env.OPENAI_API_KEY;
+
+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)/g, '');
+}
+
+(async () => {
+	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 {
+		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);
+	}
+
+	const 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 (error) {
+		console.error(chalk.white('▲ ') + chalk.red(error.message));
+		process.exit(1);
+	}
+})();