Allow for commit prefix

src/api.ts

@@ -8,6 +8,7 @@ import {
 } from 'openai';
 
 import { CONFIG_MODES, getConfig } from './commands/config';
+import { execaCommandSync } from 'execa';
 
 const config = getConfig();
 
@@ -59,7 +60,12 @@ class OpenAi {
 
       const message = data.choices[0].message;
 
-      return message?.content;
+      const prefix = generatePrefix();
+      const usePrefix = prefix != undefined && prefix?.trim() != ""
+
+      const finalMessage = (usePrefix ? prefix + ' ' : '') + (message?.content || '')
+
+      return finalMessage;
     } catch (error: unknown) {
       outro(`${chalk.red('✖')} ${error}`);
 
@@ -94,4 +100,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.length > 1 ? match[1] : match[0];
+}
+
+function getCurrentGitBranch(): string | undefined {
+  try {
+    const branchName = execaCommandSync('git symbolic-ref --short HEAD').stdout.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 = {