opencommit/src/engine/testAi.ts

import { OpenAI } from 'openai';

import { AiEngine } from './Engine';

export const TEST_MOCK_TYPES = [
  'commit-message',
  'prompt-module-commitlint-config'
] as const;

export type TestMockType = (typeof TEST_MOCK_TYPES)[number];

type TestAiEngine = Partial<AiEngine> & {
  mockType: TestMockType;
};

export class TestAi implements TestAiEngine {
  mockType: TestMockType;

  // those are not used in the test engine
  config: any;
  client: any;
  // ---

  constructor(mockType: TestMockType) {
    this.mockType = mockType;
  }

  async generateCommitMessage(
    _messages: Array<OpenAI.Chat.Completions.ChatCompletionMessageParam>
  ): Promise<string | undefined> {
    switch (this.mockType) {
      case 'commit-message':
        return 'fix(testAi.ts): test commit message';
      case 'prompt-module-commitlint-config':
        return (
          `{\n` +
          `  "localLanguage": "english",\n` +
          `  "commitFix": "fix(server): Change 'port' variable to uppercase 'PORT'",\n` +
          `  "commitFeat": "feat(server): Allow server to listen on a port specified through environment variable",\n` +
          `  "commitDescription": "Change 'port' variable to uppercase 'PORT'. Allow server to listen on a port specified through environment variable."\n` +
          `}`
        );
      default:
        throw Error('unsupported test mock type');
    }
  }
}