@binary1702/site-packet 0.1.1 → 0.1.2

package/bin/cli.js

@@ -17,6 +17,7 @@ Options:
   --output, -o <dir>     Output directory (default: ./site-packet)
   --limit, -l <n>        Maximum pages to capture (default: 100)
   --password, -p <pass>  Site password (for password-protected sites)
+  --screenshots, -s      Also capture full-page screenshots (PNG)
   --help, -h             Show this help message
 
 Examples:
@@ -33,6 +34,7 @@ let url = null;
 let outputDir = './site-packet';
 let limit = 100;
 let password = null;
+let screenshots = false;
 
 for (let i = 0; i < args.length; i++) {
   const arg = args[i];
@@ -47,6 +49,8 @@ for (let i = 0; i < args.length; i++) {
     }
   } else if (arg === '--password' || arg === '-p') {
     password = args[++i];
+  } else if (arg === '--screenshots' || arg === '-s') {
+    screenshots = true;
   } else if (!arg.startsWith('-')) {
     url = arg;
   }
@@ -68,7 +72,7 @@ try {
 
 // Run
 try {
-  await createPacket(url, { outputDir, limit, password });
+  await createPacket(url, { outputDir, limit, password, screenshots });
 } catch (error) {
   console.error(`\nFatal error: ${error.message}`);
   process.exit(1);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@binary1702/site-packet",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Turn a public website into a printable website review packet",
   "type": "module",
   "bin": {

package/src/index.js

@@ -11,7 +11,7 @@ import { ensureOutputDirs, writeCapture, writeIndex } from './output.js';
  * Create a website review packet
  */
 export async function createPacket(seedUrl, options = {}) {
-  const { outputDir = './site-packet', limit = 100, password = null } = options;
+  const { outputDir = './site-packet', limit = 100, password = null, screenshots = false } = options;
 
   console.log('\nsite-packet');
   console.log(`Crawling ${seedUrl}...`);
@@ -24,7 +24,7 @@ export async function createPacket(seedUrl, options = {}) {
   }
 
   // Setup
-  await ensureOutputDirs(outputDir);
+  await ensureOutputDirs(outputDir, { screenshots });
   const browser = await createBrowser();
 
   // Create a shared context (for cookie persistence across pages)
@@ -100,7 +100,9 @@ export async function createPacket(seedUrl, options = {}) {
         if (authSessionData) {
           captureOptions.sessionData = authSessionData;
         }
-        const { title, screenshot, pdf, finalUrl } = await capturePage(page, url, captureOptions);
+        const capture = await capturePage(page, url, { ...captureOptions, screenshots });
+        const { title, pdf, finalUrl } = capture;
+        const screenshot = capture.screenshot;
 
         // After first page, lock to the resolved origin
         if (!effectiveOrigin) {
@@ -126,8 +128,8 @@ export async function createPacket(seedUrl, options = {}) {
         const { pdfPath, screenshotPath } = await writeCapture(
           outputDir,
           filename,
-          screenshot,
-          pdf
+          pdf,
+          screenshot
         );
 
         results.push({

package/src/output.js

@@ -8,27 +8,35 @@ import { join } from 'node:path';
 /**
  * Ensure output directories exist
  */
-export async function ensureOutputDirs(outputDir) {
+export async function ensureOutputDirs(outputDir, options = {}) {
+  const { screenshots = false } = options;
   await mkdir(join(outputDir, 'pdf'), { recursive: true });
-  await mkdir(join(outputDir, 'screenshots'), { recursive: true });
+  if (screenshots) {
+    await mkdir(join(outputDir, 'screenshots'), { recursive: true });
+  }
 }
 
 /**
- * Write screenshot and PDF files
+ * Write PDF and optionally screenshot files
  * Returns { pdfPath, screenshotPath }
  */
-export async function writeCapture(outputDir, filename, screenshot, pdf) {
+export async function writeCapture(outputDir, filename, pdf, screenshot = null) {
   const pdfPath = join(outputDir, 'pdf', `${filename}.pdf`);
-  const screenshotPath = join(outputDir, 'screenshots', `${filename}.png`);
 
-  await Promise.all([
-    writeFile(pdfPath, pdf),
-    writeFile(screenshotPath, screenshot),
-  ]);
+  const writes = [writeFile(pdfPath, pdf)];
+
+  let screenshotPath = null;
+  if (screenshot) {
+    const screenshotFullPath = join(outputDir, 'screenshots', `${filename}.png`);
+    writes.push(writeFile(screenshotFullPath, screenshot));
+    screenshotPath = `screenshots/${filename}.png`;
+  }
+
+  await Promise.all(writes);
 
   return {
     pdfPath: `pdf/${filename}.pdf`,
-    screenshotPath: `screenshots/${filename}.png`,
+    screenshotPath,
   };
 }
 
@@ -146,7 +154,7 @@ export async function writeIndex(outputDir, pages, metadata) {
         <td class="url">${escapeHtml(new URL(p.url).pathname)}</td>
         <td class="links">
           <a href="${escapeHtml(p.pdfPath)}" class="link-pdf">PDF</a>
-          <a href="${escapeHtml(p.screenshotPath)}" class="link-screenshot">PNG</a>
+          ${p.screenshotPath ? `<a href="${escapeHtml(p.screenshotPath)}" class="link-screenshot">PNG</a>` : ''}
         </td>
       </tr>
       `).join('')}

package/src/renderer.js

@@ -123,7 +123,7 @@ export async function enterPassword(page, url, password) {
 }
 
 /**
- * Navigate to a URL and capture screenshot + PDF
+ * Navigate to a URL and capture PDF (and optionally screenshot)
  * Returns { title, screenshot, pdf, finalUrl } buffers or throws on failure
  *
  * @param {object} options
@@ -131,9 +131,10 @@ export async function enterPassword(page, url, password) {
  * @param {string} options.allowedOrigin - If provided, only allow redirects within this origin
  * @param {boolean} options.skipNavigation - If true, skip navigation (page already loaded)
  * @param {object} options.sessionData - sessionStorage data to inject before navigation
+ * @param {boolean} options.screenshots - If true, also capture full-page screenshot
  */
 export async function capturePage(page, url, options = {}) {
-  const { timeout = 30000, allowedOrigin = null, skipNavigation = false, sessionData = null } = options;
+  const { timeout = 30000, allowedOrigin = null, skipNavigation = false, sessionData = null, screenshots = false } = options;
 
   if (!skipNavigation) {
     // If we have sessionData, we need to inject it after navigating to the domain
@@ -181,11 +182,14 @@ export async function capturePage(page, url, options = {}) {
   // Small delay for final renders
   await page.waitForTimeout(500);
 
-  // Full-page screenshot
-  const screenshot = await page.screenshot({
-    fullPage: true,
-    type: 'png',
-  });
+  // Full-page screenshot (only if requested)
+  let screenshot = null;
+  if (screenshots) {
+    screenshot = await page.screenshot({
+      fullPage: true,
+      type: 'png',
+    });
+  }
 
   // Letter-size PDF with backgrounds
   const pdf = await page.pdf({