wehub-resource-sync
67 lines
1.8 KiB
TypeScript
67 lines
1.8 KiB
TypeScript
import { chromium, type Browser, type Page } from '@playwright/test';
|
|
import type { PPTElement } from '@openmaic/dsl';
|
|
import { mkdirSync } from 'fs';
|
|
import { join } from 'path';
|
|
|
|
const VIEWPORT = { width: 1000, height: 563 };
|
|
|
|
let browser: Browser | null = null;
|
|
let page: Page | null = null;
|
|
|
|
/**
|
|
* Initialize Playwright browser (reused across captures).
|
|
*/
|
|
export async function initCapture(baseUrl: string): Promise<void> {
|
|
browser = await chromium.launch({ headless: true });
|
|
const context = await browser.newContext({ viewport: VIEWPORT });
|
|
page = await context.newPage();
|
|
|
|
await page.goto(`${baseUrl}/eval/whiteboard`);
|
|
// Wait for the page to signal readiness
|
|
await page.waitForFunction(
|
|
() => (window as unknown as Record<string, unknown>).__evalReady === true,
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Capture a screenshot of the whiteboard with the given elements.
|
|
* Returns the path to the saved screenshot.
|
|
*/
|
|
export async function captureWhiteboard(
|
|
elements: PPTElement[],
|
|
outputDir: string,
|
|
filename: string,
|
|
): Promise<string> {
|
|
if (!page) throw new Error('Capture not initialized. Call initCapture() first.');
|
|
|
|
// Inject elements into the page
|
|
await page.evaluate(
|
|
(els: unknown[]) => {
|
|
const setter = (window as unknown as Record<string, (els: unknown[]) => void>).__setElements;
|
|
setter(els);
|
|
},
|
|
elements as unknown as unknown[],
|
|
);
|
|
|
|
// Wait for rendering to stabilize (fonts, KaTeX, images)
|
|
await page.waitForTimeout(1500);
|
|
|
|
mkdirSync(outputDir, { recursive: true });
|
|
const filepath = join(outputDir, filename);
|
|
|
|
await page.screenshot({ path: filepath, clip: { x: 0, y: 0, width: 1000, height: 563 } });
|
|
|
|
return filepath;
|
|
}
|
|
|
|
/**
|
|
* Close the browser.
|
|
*/
|
|
export async function closeCapture(): Promise<void> {
|
|
if (browser) {
|
|
await browser.close();
|
|
browser = null;
|
|
page = null;
|
|
}
|
|
}
|