orca/tests/e2e/helpers/source-control-ai-generators.ts

import type { Page } from '@stablyai/playwright-test'
import { writeFileSync } from 'node:fs'

async function setCustomGenerator(page: Page, scriptPath: string): Promise<void> {
  await page.evaluate(async (scriptPath) => {
    const store =
      window.__store ??
      (() => {
        throw new Error('window.__store is not available')
      })()
    const currentSettings = store.getState().settings
    if (!currentSettings) {
      throw new Error('Settings were not loaded')
    }
    await store.getState().updateSettings({
      activeRuntimeEnvironmentId: null,
      commitMessageAi: {
        ...currentSettings.commitMessageAi,
        enabled: true,
        agentId: 'custom' as const,
        selectedModelByAgent: {},
        selectedThinkingByModel: {},
        customPrompt: '',
        customAgentCommand: `node ${JSON.stringify(scriptPath)}`
      }
    })
  }, scriptPath)
}

export async function installDelayedPrGenerator(
  page: Page,
  generatorScriptPath: string,
  callLogPath: string,
  base: string
): Promise<void> {
  writeFileSync(
    generatorScriptPath,
    [
      "const fs = require('fs')",
      `fs.appendFileSync(${JSON.stringify(callLogPath)}, 'start\\n')`,
      'setTimeout(() => {',
      '  console.log(JSON.stringify({',
      `    base: ${JSON.stringify(base)},`,
      "    title: 'Generated PR title after switch',",
      "    body: 'Generated PR body after switch',",
      '    draft: false',
      '  }))',
      `  fs.appendFileSync(${JSON.stringify(callLogPath)}, 'finish\\n')`,
      '}, 1500)'
    ].join('\n')
  )
  await setCustomGenerator(page, generatorScriptPath)
}

export async function installDelayedCommitMessageGenerator(
  page: Page,
  generatorScriptPath: string,
  callLogPath: string
): Promise<void> {
  writeFileSync(
    generatorScriptPath,
    [
      "const fs = require('fs')",
      `fs.appendFileSync(${JSON.stringify(callLogPath)}, 'start\\n')`,
      'setTimeout(() => {',
      "  console.log('Generated commit message after switch')",
      "  console.log('')",
      "  console.log('Generated from staged e2e-commit-message-generation.txt after switching worktrees')",
      `  fs.appendFileSync(${JSON.stringify(callLogPath)}, 'finish\\n')`,
      '}, 1500)'
    ].join('\n')
  )
  await setCustomGenerator(page, generatorScriptPath)
}