92 lines
2.3 KiB
JavaScript
92 lines
2.3 KiB
JavaScript
#!/usr/bin/env node
|
|
/**
|
|
* Capture screenshots for visual debugging
|
|
* Usage: node screenshot.js --url https://example.com --output screenshot.png [--full-page true] [--selector .element]
|
|
*/
|
|
|
|
import {
|
|
getBrowserSession,
|
|
closeBrowserSession,
|
|
parseArgs,
|
|
outputJSON,
|
|
outputError,
|
|
createScreenshot,
|
|
getOutputDirectory,
|
|
} from "../lib/browser.js";
|
|
import { writeFile } from "fs/promises";
|
|
import { join } from "path";
|
|
|
|
interface ScreenshotArgs {
|
|
url?: string;
|
|
output?: string;
|
|
"full-page"?: string | boolean;
|
|
selector?: string;
|
|
format?: string;
|
|
quality?: string;
|
|
"wait-until"?: string;
|
|
headless?: string | boolean;
|
|
close?: string | boolean;
|
|
browser?: string;
|
|
}
|
|
|
|
async function takeScreenshot() {
|
|
const args = parseArgs(process.argv.slice(2)) as ScreenshotArgs;
|
|
|
|
if (!args.url && !args.selector) {
|
|
return outputError(new Error("Either --url or --selector is required"));
|
|
}
|
|
|
|
// --output is optional - will auto-generate if not provided
|
|
|
|
try {
|
|
const session = await getBrowserSession({
|
|
browser: args.browser as any,
|
|
headless: args.headless !== "false",
|
|
});
|
|
|
|
// Navigate to URL if provided
|
|
if (args.url) {
|
|
await session.page.goto(args.url, {
|
|
waitUntil: (args["wait-until"] as any) || "load",
|
|
});
|
|
}
|
|
|
|
const outputPath =
|
|
args.output || join(getOutputDirectory(), `screenshot_${Date.now()}.png`);
|
|
const fullPage = args["full-page"] === "true";
|
|
|
|
// Create screenshot
|
|
const result = await createScreenshot(session.page, outputPath, {
|
|
fullPage,
|
|
selector: args.selector,
|
|
});
|
|
|
|
const screenshotData = {
|
|
success: true,
|
|
url: session.page.url(),
|
|
title: await session.page.title(),
|
|
outputPath: typeof result === "string" ? result : outputPath,
|
|
fullPage,
|
|
selector: args.selector || null,
|
|
size: typeof result === "string" ? null : result.length,
|
|
sessionId: session.sessionId,
|
|
timestamp: new Date().toISOString(),
|
|
};
|
|
|
|
outputJSON(screenshotData);
|
|
|
|
if (args.close !== "false") {
|
|
await closeBrowserSession();
|
|
} else {
|
|
// Explicitly exit the process when keeping session open
|
|
process.exit(0);
|
|
}
|
|
} catch (error) {
|
|
return outputError(
|
|
error instanceof Error ? error : new Error(String(error)),
|
|
);
|
|
}
|
|
}
|
|
|
|
takeScreenshot();
|