Initial commit

skills/play-tight/scripts/take-screenshot.js

@@ -0,0 +1,70 @@
+#!/usr/bin/env node
+
+/**
+ * Take a screenshot of a page or specific element
+ * Usage: node take-screenshot.js <url> <output-path> [selector]
+ * If selector is provided, captures only that element
+ * Returns: Compact JSON with screenshot info
+ */
+
+const { chromium } = require('playwright');
+const path = require('path');
+
+async function takeScreenshot(url, outputPath, selector = null) {
+  const browser = await chromium.launch({ headless: true });
+  const page = await browser.newPage();
+  
+  try {
+    await page.goto(url, { waitUntil: 'domcontentloaded', timeout: 30000 });
+    await page.waitForTimeout(1000);
+    
+    const absolutePath = path.resolve(outputPath);
+    
+    if (selector) {
+      const element = await page.$(selector);
+      if (!element) {
+        await browser.close();
+        return { success: false, error: `Element not found: ${selector}` };
+      }
+      await element.screenshot({ path: absolutePath });
+    } else {
+      await page.screenshot({ path: absolutePath, fullPage: true });
+    }
+    
+    await browser.close();
+    
+    return {
+      success: true,
+      path: absolutePath,
+      url,
+      selector: selector || 'full-page'
+    };
+    
+  } catch (error) {
+    await browser.close();
+    return {
+      success: false,
+      error: error.message.substring(0, 100)
+    };
+  }
+}
+
+// Main execution
+if (require.main === module) {
+  const args = process.argv.slice(2);
+  
+  if (args.length < 2) {
+    console.log(JSON.stringify({
+      error: 'Usage: node take-screenshot.js <url> <output-path> [selector]'
+    }));
+    process.exit(1);
+  }
+  
+  const [url, outputPath, selector] = args;
+  
+  takeScreenshot(url, outputPath, selector)
+    .then(result => console.log(JSON.stringify(result, null, 2)))
+    .catch(error => console.log(JSON.stringify({ error: error.message })));
+}
+
+module.exports = { takeScreenshot };