opencode/packages/simulation/src/frontend/renderer.ts

54 lines
1.8 KiB
TypeScript

import type { CliRenderer, CliRendererConfig } from "@opentui/core"
import { createTestRenderer, type TestRendererSetup } from "@opentui/core/testing"
import { Timeline } from "../recording"
const setups = new WeakMap<CliRenderer, TestRendererSetup>()
const recordings = new WeakMap<CliRenderer, Timeline>()
/**
* Creates a headless renderer with optional recording: a real CliRenderer
* backed by an in-memory screen buffer. The TestRendererSetup is kept
* module-side so the harness can use supported testing APIs without app
* code carrying it around.
*/
export async function create(options: CliRendererConfig, path?: string): Promise<CliRenderer> {
if (!path) {
const setup = await createTestRenderer({
...options,
width: 100,
height: 40,
})
setups.set(setup.renderer, setup)
return setup.renderer
}
const recording = await Timeline.create(path, 100, 40)
const setup = await createTestRenderer({
...options,
width: 100,
height: 40,
stdout: recording as unknown as NodeJS.WriteStream,
bufferedOutput: "stdout",
onDestroy: () => {
void recording.finish().catch((error) => process.stderr.write(`Failed to finish UI recording: ${error}\n`))
options.onDestroy?.()
},
}).catch(async (error) => {
await recording.finish().catch(() => undefined)
throw error
})
setups.set(setup.renderer, setup)
recordings.set(setup.renderer, recording)
return setup.renderer
}
export function setupFor(renderer: CliRenderer): TestRendererSetup | undefined {
return setups.get(renderer)
}
export function finish(renderer: CliRenderer) {
const recording = recordings.get(renderer)
if (!recording) throw new Error("UI recording is not available")
return recording.finish()
}
export * as SimulationRenderer from "./renderer"