orca/tests/e2e/terminal-ime-byte-reader.ts

import { randomUUID } from 'node:crypto'
import { rmSync, writeFileSync } from 'node:fs'
import path from 'node:path'
import type { Page } from '@stablyai/playwright-test'
import { expect } from '@stablyai/playwright-test'
import { getTerminalContent, sendToTerminal, waitForTerminalOutput } from './helpers/terminal'

export type TerminalImeByteReader = {
  expectedLineCount: number
  readyMarker: string
  resultPrefix: string
  scriptPath: string
}

export function createTerminalImeByteReader(
  testRepoPath: string,
  expectedLineCount: number
): TerminalImeByteReader {
  const runId = randomUUID().replaceAll('-', '')
  const readyMarker = `ORCA_IME_READER_READY_${runId}`
  const resultPrefix = `ORCA_IME_BYTES_${runId}`
  const scriptPath = path.join(testRepoPath, `.orca-ime-byte-reader-${runId}.cjs`)
  const source = `
const expectedLineCount = ${expectedLineCount}
const readyMarker = ${JSON.stringify(readyMarker)}
const resultPrefix = ${JSON.stringify(resultPrefix)}
let pending = Buffer.alloc(0)
let receivedLineCount = 0

process.stdout.write(readyMarker + '\\n')
process.stdin.on('data', (chunk) => {
  pending = Buffer.concat([pending, Buffer.from(chunk)])
  let newlineIndex = pending.indexOf(0x0a)
  while (newlineIndex >= 0) {
    const line = pending.subarray(0, newlineIndex + 1)
    pending = pending.subarray(newlineIndex + 1)
    receivedLineCount += 1
    process.stdout.write(resultPrefix + ':' + receivedLineCount + ':' + line.toString('hex') + '\\n')
    if (receivedLineCount === expectedLineCount) {
      process.exit(0)
    }
    newlineIndex = pending.indexOf(0x0a)
  }
})
`
  writeFileSync(scriptPath, source)
  return { expectedLineCount, readyMarker, resultPrefix, scriptPath }
}

export async function startTerminalImeByteReader(
  page: Page,
  ptyId: string,
  reader: TerminalImeByteReader
): Promise<void> {
  await sendToTerminal(page, ptyId, `node ${JSON.stringify(reader.scriptPath)}\r`)
  await waitForTerminalOutput(page, reader.readyMarker, 10_000, 20_000)
}

export async function waitForTerminalImeBytes(
  page: Page,
  reader: TerminalImeByteReader,
  timeoutMs = 15_000
): Promise<string[]> {
  let results: string[] = []
  await expect
    .poll(
      async () => {
        const terminal = await getTerminalContent(page, 100_000)
        const resultPattern = new RegExp(`${reader.resultPrefix}:(\\d+):([0-9a-f]+)`, 'g')
        const bySequence = new Map<number, string>()
        for (const match of terminal.matchAll(resultPattern)) {
          bySequence.set(Number(match[1]), match[2])
        }
        results = [...bySequence.entries()]
          .sort(([left], [right]) => left - right)
          .map(([, hex]) => hex)
        return results.length
      },
      { timeout: timeoutMs, message: 'IME byte reader did not receive every expected line' }
    )
    .toBe(reader.expectedLineCount)
  return results
}

export function removeTerminalImeByteReader(reader: TerminalImeByteReader): void {
  rmSync(reader.scriptPath, { force: true })
}