Allow for commit prefix

package.json

@@ -64,6 +64,7 @@
     "@dqbd/tiktoken": "^1.0.2",
     "axios": "^1.3.4",
     "chalk": "^5.2.0",
+    "child_process": "^1.0.2",
     "cleye": "^1.3.2",
     "execa": "^7.0.0",
     "ignore": "^5.2.4",

src/api.ts

@@ -1,4 +1,5 @@
 import { intro, outro } from '@clack/prompts';
+import { execSync } from 'child_process';
 import axios from 'axios';
 import chalk from 'chalk';
 import {
@@ -59,7 +60,11 @@ class OpenAi {
 
       const message = data.choices[0].message;
 
-      return message?.content;
+      const prefix = generatePrefix();
+
+      const finalMessage = (prefix != "undefined" ? prefix + ' ' : '') + (message?.content || '')
+
+      return finalMessage;
     } catch (error: unknown) {
       outro(`${chalk.red('✖')} ${error}`);
 
@@ -94,4 +99,57 @@ export const getOpenCommitLatestVersion = async (): Promise<
   }
 };
 
+function generatePrefix(): string | undefined {
+  const prefix = config?.prefix
+
+  if (prefix === undefined) {
+    return undefined;
+  }
+
+  const prefixIsRegexString = prefix.startsWith('/') && prefix.endsWith('/');
+
+  if (prefixIsRegexString) {
+    try {
+      return generatePrefixFromRegex(prefix);
+    } catch (error) {
+      console.error(`Failed to generate prefix from regex: ${error}`);
+      return undefined;
+    }
+  }
+
+  return prefix;
+}
+
 export const api = new OpenAi();
+
+
+function generatePrefixFromRegex(regex: string): string | undefined {
+
+  // We currently only support regex input from git branch name
+
+  const branch = getCurrentGitBranch();
+
+  if (branch === undefined) {
+    return undefined;
+  }
+
+  const regexWithoutSlashes = regex.slice(1, -1);
+  const regexObject = new RegExp(regexWithoutSlashes);
+  const match = branch.match(regexObject);
+
+  if (match === null) {
+    return undefined;
+  }
+
+  return match[0];
+}
+
+function getCurrentGitBranch(): string | undefined {
+  try {
+    const branchName = execSync('git symbolic-ref --short HEAD', { encoding: 'utf8' }).trim();
+    return branchName;
+  } catch (error) {
+    console.error(`Failed to get current git branch: ${error}`);
+    return undefined;
+  }
+}

src/commands/config.ts

@@ -15,7 +15,8 @@ export enum CONFIG_KEYS {
   description = 'description',
   emoji = 'emoji',
   model = 'model',
-  language = 'language'
+  language = 'language',
+  prefix = 'prefix'
 }
 
 export enum CONFIG_MODES {
@@ -109,7 +110,17 @@ export const configValidators = {
       `${value} is not supported yet, use 'gpt-4' or 'gpt-3.5-turbo' (default)`
     );
     return value;
+  },
+
+  [CONFIG_KEYS.prefix](value: any) {
+    validateConfig(
+      CONFIG_KEYS.prefix,
+      true,
+      'Cannot be empty'
+    );
+    return value;
   }
+
 };
 
 export type ConfigType = {