sessionsnap 0.0.1 → 0.0.2

package/bin/sessionsnap.js

@@ -19,16 +19,17 @@ program
   .requiredOption('--profile <name>', 'Profile name to save the session under')
   .option('--out <file>', 'Custom output path for the snapshot JSON')
   .option('--wait <minutes>', 'Max minutes to wait for login', parseFloat)
+  .option('--target <pattern>', 'Regex pattern to wait for in URL (e.g., "/dashboard" or "^https://.*/app")')
   .option('--record', 'Record user actions (clicks, inputs, navigations)')
   .action(async (url, opts) => {
     validateRunner(opts.runner);
     try {
       if (opts.runner === 'puppeteer') {
         const { capture } = await import('../src/puppeteer/capture.js');
-        await capture(url, { profile: opts.profile, out: opts.out, waitMinutes: opts.wait, record: opts.record });
+        await capture(url, { profile: opts.profile, out: opts.out, waitMinutes: opts.wait, targetPattern: opts.target, record: opts.record });
       } else {
         const { capture } = await import('../src/playwright/capture.js');
-        await capture(url, { profile: opts.profile, out: opts.out, waitMinutes: opts.wait, record: opts.record });
+        await capture(url, { profile: opts.profile, out: opts.out, waitMinutes: opts.wait, targetPattern: opts.target, record: opts.record });
       }
     } catch (err) {
       console.error(`[sessionsnap] Error: ${err.message}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sessionsnap",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "Capture and reuse authenticated browser sessions (Puppeteer + Playwright)",
   "type": "module",
   "bin": {

package/src/core/heuristics.js

@@ -18,10 +18,22 @@ export async function getCookieCount(page, runner) {
   return cookies.length;
 }
 
-export async function waitForLoginComplete(page, { runner, waitMinutes } = {}) {
+export async function waitForLoginComplete(page, { runner, waitMinutes, targetPattern } = {}) {
   const timeout = (waitMinutes ?? DEFAULT_TIMEOUT_MIN) * 60 * 1000;
   const start = Date.now();
   const initialCookieCount = await getCookieCount(page, runner);
+  
+  // Compile target pattern if provided
+  let targetRegex = null;
+  if (targetPattern) {
+    try {
+      targetRegex = new RegExp(targetPattern);
+      console.log(`[sessionsnap] Waiting for URL matching pattern: ${targetPattern}`);
+    } catch (err) {
+      console.error(`[sessionsnap] Invalid regex pattern: ${targetPattern}. Error: ${err.message}`);
+      throw new Error(`Invalid target pattern: ${targetPattern}`);
+    }
+  }
 
   return new Promise((resolve) => {
     const timer = setInterval(async () => {
@@ -36,6 +48,20 @@ export async function waitForLoginComplete(page, { runner, waitMinutes } = {}) {
       }
 
       const currentUrl = page.url();
+      
+      // If target pattern is provided, check if URL matches
+      if (targetRegex) {
+        if (targetRegex.test(currentUrl)) {
+          clearInterval(timer);
+          console.log(`[sessionsnap] Target URL pattern matched: ${currentUrl}`);
+          resolve(true);
+          return;
+        }
+        // Continue waiting if pattern doesn't match yet
+        return;
+      }
+
+      // Default behavior: check URL and cookie heuristics
       const urlClear = !isLoginUrl(currentUrl);
 
       let cookieIncreased = false;

package/src/playwright/capture.js

@@ -3,7 +3,7 @@ import { waitForLoginComplete } from '../core/heuristics.js';
 import { createSnapshot, normalizePlaywrightState } from '../core/snapshot.js';
 import { setupRecorder } from '../core/recorder.js';
 
-export async function capture(url, { profile, out, waitMinutes, record }) {
+export async function capture(url, { profile, out, waitMinutes, targetPattern, record }) {
   const { chromium } = await import('playwright');
   const userDataDir = await ensureProfileDir(profile);
 
@@ -24,12 +24,16 @@ export async function capture(url, { profile, out, waitMinutes, record }) {
   console.log(`[sessionsnap] Navigating to: ${url}`);
   await page.goto(url, { waitUntil: 'domcontentloaded' });
 
-  console.log('[sessionsnap] Please log in manually. Login completion will be detected automatically.');
+  if (targetPattern) {
+    console.log(`[sessionsnap] Please log in manually. Waiting for URL matching pattern: ${targetPattern}`);
+  } else {
+    console.log('[sessionsnap] Please log in manually. Login completion will be detected automatically.');
+  }
   if (record) {
     console.log('[sessionsnap] Recording actions... (press Ctrl+C to stop early)');
   }
 
-  const detected = await waitForLoginComplete(page, { runner: 'playwright', waitMinutes });
+  const detected = await waitForLoginComplete(page, { runner: 'playwright', waitMinutes, targetPattern });
 
   if (detected) {
     console.log('[sessionsnap] Login detected! Capturing session...');

package/src/puppeteer/capture.js

@@ -3,7 +3,7 @@ import { waitForLoginComplete } from '../core/heuristics.js';
 import { createSnapshot, normalizePuppeteerCookies } from '../core/snapshot.js';
 import { setupRecorder } from '../core/recorder.js';
 
-export async function capture(url, { profile, out, waitMinutes, record }) {
+export async function capture(url, { profile, out, waitMinutes, targetPattern, record }) {
   const puppeteer = await import('puppeteer');
   const userDataDir = await ensureProfileDir(profile);
 
@@ -25,12 +25,16 @@ export async function capture(url, { profile, out, waitMinutes, record }) {
   console.log(`[sessionsnap] Navigating to: ${url}`);
   await page.goto(url, { waitUntil: 'domcontentloaded' });
 
-  console.log('[sessionsnap] Please log in manually. Login completion will be detected automatically.');
+  if (targetPattern) {
+    console.log(`[sessionsnap] Please log in manually. Waiting for URL matching pattern: ${targetPattern}`);
+  } else {
+    console.log('[sessionsnap] Please log in manually. Login completion will be detected automatically.');
+  }
   if (record) {
     console.log('[sessionsnap] Recording actions... (press Ctrl+C to stop early)');
   }
 
-  const detected = await waitForLoginComplete(page, { runner: 'puppeteer', waitMinutes });
+  const detected = await waitForLoginComplete(page, { runner: 'puppeteer', waitMinutes, targetPattern });
 
   if (detected) {
     console.log('[sessionsnap] Login detected! Capturing session...');