Initial commit

skills/scripts/screenshot.ts

@@ -0,0 +1,91 @@
+#!/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();