test-proxy-recorder 0.1.2 → 0.1.4

package/dist/index.mjs

@@ -1,8 +1,10 @@
 import fs from 'fs/promises';
 import http from 'http';
+import https from 'https';
 import httpProxy from 'http-proxy';
 import { WebSocket, WebSocketServer } from 'ws';
 import path from 'path';
+import filenamify from 'filenamify';
 
 // src/ProxyServer.ts
 
@@ -38,6 +40,24 @@ async function saveRecordingSession(recordingsDir, session) {
     `Saved ${session.recordings.length} HTTP recordings and ${session.websocketRecordings?.length || 0} WebSocket recordings to ${filePath}`
   );
 }
+var QUERY_HASH_LENGTH = 8;
+function getReqID(req) {
+  const urlParts = req.url.split("?");
+  const pathname = urlParts[0];
+  const query = urlParts[1] || "";
+  const pathPart = pathname === "/" ? "root" : pathname.slice(1);
+  const normalizedPath = filenamify(pathPart, { replacement: "_" });
+  const queryHash = generateQueryHash(query);
+  const filename = `${req.method}_${normalizedPath}${queryHash}.json`;
+  return filenamify(filename, { replacement: "_" });
+}
+function generateQueryHash(query) {
+  if (!query) {
+    return "";
+  }
+  const hash = Buffer.from(query).toString("base64").replaceAll(/[^a-zA-Z0-9]/g, "").slice(0, Math.max(0, QUERY_HASH_LENGTH));
+  return `_${hash}`;
+}
 
 // src/utils/httpHelpers.ts
 var CONTENT_TYPE_JSON = "application/json";
@@ -53,28 +73,6 @@ function sendJsonResponse(res, statusCode, data) {
   res.end(JSON.stringify(data));
 }
 
-// src/utils/requestKeyGenerator.ts
-var QUERY_HASH_LENGTH = 8;
-function generateRequestKey(req) {
-  const urlParts = req.url.split("?");
-  const pathname = urlParts[0];
-  const query = urlParts[1] || "";
-  const normalizedPath = normalizePathname(pathname);
-  const queryHash = generateQueryHash(query);
-  return `${req.method}_${normalizedPath}${queryHash}.json`;
-}
-function normalizePathname(pathname) {
-  const normalized = pathname.replaceAll("/", "_").replace(/^_/, "");
-  return normalized || "root";
-}
-function generateQueryHash(query) {
-  if (!query) {
-    return "";
-  }
-  const hash = Buffer.from(query).toString("base64").replaceAll(/[^a-zA-Z0-9]/g, "").slice(0, Math.max(0, QUERY_HASH_LENGTH));
-  return `_${hash}`;
-}
-
 // src/ProxyServer.ts
 var ProxyServer = class {
   targets;
@@ -86,6 +84,10 @@ var ProxyServer = class {
   proxy;
   currentSession;
   recordingsDir;
+  requestSequenceMap;
+  // Track sequence per request key
+  replaySequenceMap;
+  // Track replay position per request key
   constructor(targets, recordingsDir) {
     this.targets = targets;
     this.currentTargetIndex = 0;
@@ -95,6 +97,8 @@ var ProxyServer = class {
     this.modeTimeout = null;
     this.currentSession = null;
     this.recordingsDir = recordingsDir;
+    this.requestSequenceMap = /* @__PURE__ */ new Map();
+    this.replaySequenceMap = /* @__PURE__ */ new Map();
     this.proxy = httpProxy.createProxyServer({
       secure: false,
       changeOrigin: true
@@ -133,10 +137,19 @@ var ProxyServer = class {
     res.end(JSON.stringify({ error: "Proxy error", message: err.message }));
   }
   handleProxyResponse(proxyRes, req) {
+    this.addCorsHeaders(proxyRes, req);
     if (this.mode === Modes.record && this.recordingId) {
       this.recordResponse(req, proxyRes);
     }
   }
+  addCorsHeaders(proxyRes, req) {
+    const origin = req.headers.origin;
+    proxyRes.headers["access-control-allow-origin"] = origin || "*";
+    proxyRes.headers["access-control-allow-credentials"] = "true";
+    proxyRes.headers["access-control-allow-headers"] = req.headers["access-control-request-headers"] || "Origin, X-Requested-With, Content-Type, Accept, Authorization";
+    proxyRes.headers["access-control-allow-methods"] = "GET, POST, PUT, DELETE, PATCH, OPTIONS";
+    proxyRes.headers["access-control-expose-headers"] = "*";
+  }
   getTarget() {
     const target = this.targets[this.currentTargetIndex];
     this.currentTargetIndex = (this.currentTargetIndex + 1) % this.targets.length;
@@ -200,6 +213,7 @@ var ProxyServer = class {
     this.recordingId = null;
     this.replayId = null;
     this.currentSession = null;
+    clearTimeout(this.modeTimeout || 0);
     console.log("Switched to transparent mode");
   }
   switchToRecordMode(id) {
@@ -210,6 +224,7 @@ var ProxyServer = class {
     this.recordingId = id;
     this.replayId = null;
     this.currentSession = { id, recordings: [], websocketRecordings: [] };
+    this.requestSequenceMap.clear();
     console.log(`Switched to record mode with ID: ${id}`);
   }
   switchToReplayMode(id) {
@@ -220,6 +235,7 @@ var ProxyServer = class {
     this.replayId = id;
     this.recordingId = null;
     this.currentSession = null;
+    this.replaySequenceMap.clear();
     console.log(`Switched to replay mode with ID: ${id}`);
   }
   setupModeTimeout(timeout) {
@@ -250,7 +266,9 @@ var ProxyServer = class {
     if (!this.currentSession) {
       return;
     }
-    const key = generateRequestKey(req);
+    const key = getReqID(req);
+    const currentSequence = this.requestSequenceMap.get(key) || 0;
+    this.requestSequenceMap.set(key, currentSequence + 1);
     const record = {
       request: {
         method: req.method,
@@ -259,7 +277,8 @@ var ProxyServer = class {
         body: body || null
       },
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-      key
+      key,
+      sequence: currentSequence
     };
     this.currentSession.recordings.push(record);
   }
@@ -267,8 +286,10 @@ var ProxyServer = class {
     if (!this.currentSession) {
       return;
     }
-    const key = generateRequestKey(req);
-    const record = this.currentSession.recordings.find((r) => r.key === key);
+    const key = getReqID(req);
+    const record = this.currentSession.recordings.findLast(
+      (r) => r.key === key && !r.response
+    );
     if (!record) {
       console.error("Request record not found for response:", key);
       return;
@@ -289,21 +310,35 @@ var ProxyServer = class {
     });
   }
   async handleReplayRequest(req, res) {
-    const key = generateRequestKey(req);
+    const key = getReqID(req);
     const filePath = getRecordingPath(this.recordingsDir, this.replayId);
     try {
       const session = await loadRecordingSession(filePath);
-      const record = session.recordings.find((r) => r.key === key);
+      const currentSequence = this.replaySequenceMap.get(key) || 0;
+      const record = session.recordings.find(
+        (r) => r.key === key && r.sequence === currentSequence
+      );
       if (!record) {
-        throw new Error(`No recording found for ${key}`);
+        throw new Error(
+          `No recording found for ${key} with sequence ${currentSequence}`
+        );
       }
       if (!record.response) {
         throw new Error("No response recorded for this request");
       }
+      this.replaySequenceMap.set(key, currentSequence + 1);
       const { statusCode, headers, body } = record.response;
-      res.writeHead(statusCode, headers);
+      const origin = req.headers.origin;
+      const responseHeaders = {
+        ...headers,
+        "access-control-allow-origin": origin || "*",
+        "access-control-allow-credentials": "true"
+      };
+      res.writeHead(statusCode, responseHeaders);
       res.end(body);
-      console.log(`Replayed: ${req.method} ${req.url}`);
+      console.log(
+        `Replayed: ${req.method} ${req.url} (sequence: ${currentSequence})`
+      );
     } catch (error) {
       this.handleReplayError(res, error, key, filePath);
     }
@@ -319,6 +354,9 @@ var ProxyServer = class {
     });
   }
   async handleRequest(req, res) {
+    if (req.method === "OPTIONS") {
+      return this.handleCorsPreflightRequest(req, res);
+    }
     if (req.url === CONTROL_ENDPOINT) {
       return this.handleControlRequest(req, res);
     }
@@ -327,23 +365,63 @@ var ProxyServer = class {
     }
     await this.handleProxyRequest(req, res);
   }
+  handleCorsPreflightRequest(req, res) {
+    const origin = req.headers.origin;
+    res.writeHead(HTTP_STATUS_OK, {
+      "Access-Control-Allow-Origin": origin || "*",
+      "Access-Control-Allow-Credentials": "true",
+      "Access-Control-Allow-Headers": req.headers["access-control-request-headers"] || "Origin, X-Requested-With, Content-Type, Accept, Authorization",
+      "Access-Control-Allow-Methods": "GET, POST, PUT, DELETE, PATCH, OPTIONS",
+      "Access-Control-Max-Age": "86400"
+      // 24 hours
+    });
+    res.end();
+  }
   async handleProxyRequest(req, res) {
     const target = this.getTarget();
     console.log(`[${this.mode}] ${req.method} ${req.url} -> ${target}`);
     if (this.mode === Modes.record) {
-      await this.bufferRequestForRecord(req);
+      await this.bufferAndProxyRequest(req, res, target);
+    } else {
+      this.proxy.web(req, res, { target });
     }
-    this.proxy.web(req, res, { target });
   }
-  async bufferRequestForRecord(req) {
+  async bufferAndProxyRequest(req, res, target) {
     const chunks = [];
     req.on("data", (chunk) => {
       chunks.push(chunk);
     });
-    req.on("end", async () => {
-      const body = Buffer.concat(chunks).toString("utf8");
-      await this.saveRequestRecord(req, body);
+    await new Promise((resolve) => {
+      req.on("end", () => resolve());
     });
+    const body = Buffer.concat(chunks).toString("utf8");
+    await this.saveRequestRecord(req, body);
+    const targetUrl = new URL(target);
+    const isHttps = targetUrl.protocol === "https:";
+    const requestModule = isHttps ? https : http;
+    const defaultPort = isHttps ? 443 : 80;
+    const proxyReq = requestModule.request(
+      {
+        hostname: targetUrl.hostname,
+        port: targetUrl.port || defaultPort,
+        path: req.url,
+        method: req.method,
+        headers: req.headers
+      },
+      (proxyRes) => {
+        this.addCorsHeaders(proxyRes, req);
+        this.recordResponse(req, proxyRes);
+        res.writeHead(proxyRes.statusCode || 200, proxyRes.headers);
+        proxyRes.pipe(res);
+      }
+    );
+    proxyReq.on("error", (err) => {
+      this.handleProxyError(err, req, res);
+    });
+    if (chunks.length > 0) {
+      proxyReq.write(Buffer.concat(chunks));
+    }
+    proxyReq.end();
   }
   handleUpgrade(req, socket, head) {
     if (this.mode === Modes.replay) {
@@ -546,15 +624,15 @@ function generateSessionId(testInfo) {
 }
 async function startRecording(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("recording", sessionId);
+  await setProxyMode(Modes.record, sessionId);
 }
 async function startReplay(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("replay", sessionId);
+  await setProxyMode(Modes.replay, sessionId);
 }
 async function stopProxy(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("transparent", sessionId);
+  await setProxyMode(Modes.transparent, sessionId);
 }
 var playwrightProxy = {
   /**
@@ -573,7 +651,7 @@ var playwrightProxy = {
    */
   async after(testInfo) {
     const sessionId = generateSessionId(testInfo);
-    await setProxyMode("transparent", sessionId);
+    await setProxyMode(Modes.transparent, sessionId);
   }
 };
 

package/dist/playwright/index.cjs

@@ -1,5 +1,12 @@
 'use strict';
 
+// src/types.ts
+var Modes = {
+  transparent: "transparent",
+  record: "record",
+  replay: "replay"
+};
+
 // src/playwright/index.ts
 var INTERNAL_API_URL = process.env.INTERNAL_API_URL || "http://localhost:8100";
 async function setProxyMode(mode, sessionId, timeout) {
@@ -31,15 +38,15 @@ function generateSessionId(testInfo) {
 }
 async function startRecording(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("recording", sessionId);
+  await setProxyMode(Modes.record, sessionId);
 }
 async function startReplay(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("replay", sessionId);
+  await setProxyMode(Modes.replay, sessionId);
 }
 async function stopProxy(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("transparent", sessionId);
+  await setProxyMode(Modes.transparent, sessionId);
 }
 var playwrightProxy = {
   /**
@@ -58,7 +65,7 @@ var playwrightProxy = {
    */
   async after(testInfo) {
     const sessionId = generateSessionId(testInfo);
-    await setProxyMode("transparent", sessionId);
+    await setProxyMode(Modes.transparent, sessionId);
   }
 };
 

package/dist/playwright/index.d.cts

@@ -1,50 +1,3 @@
-import { TestInfo } from '@playwright/test';
-
-type ProxyMode = 'recording' | 'replay' | 'transparent';
-type PlaywrightTestInfo = Pick<TestInfo, 'title'>;
-/**
- * Set the proxy mode for a given session
- * @param mode - The proxy mode to set (recording, replay, transparent)
- * @param sessionId - Unique identifier for the session
- * @param timeout - Optional timeout in milliseconds
- */
-declare function setProxyMode(mode: ProxyMode, sessionId: string, timeout?: number): Promise<void>;
-/**
- * Generate a session ID from test info
- * @param testInfo - Playwright test info object
- */
-declare function generateSessionId(testInfo: PlaywrightTestInfo): string;
-/**
- * Start recording for a test
- * @param testInfo - Playwright test info object
- */
-declare function startRecording(testInfo: PlaywrightTestInfo): Promise<void>;
-/**
- * Start replay for a test
- * @param testInfo - Playwright test info object
- */
-declare function startReplay(testInfo: PlaywrightTestInfo): Promise<void>;
-/**
- * Stop recording/replay and return to transparent mode
- * @param testInfo - Playwright test info object
- */
-declare function stopProxy(testInfo: PlaywrightTestInfo): Promise<void>;
-/**
- * Playwright test fixture helper for managing proxy mode
- * Use this in beforeEach/afterEach hooks
- */
-declare const playwrightProxy: {
-    /**
-     * Setup before test - sets the proxy mode
-     * @param testInfo - Playwright test info object
-     * @param mode - The proxy mode to use for this test
-     */
-    before(testInfo: PlaywrightTestInfo, mode: ProxyMode): Promise<void>;
-    /**
-     * Cleanup after test - returns to transparent mode
-     * @param testInfo - Playwright test info object
-     */
-    after(testInfo: PlaywrightTestInfo): Promise<void>;
-};
-
-export { type PlaywrightTestInfo, type ProxyMode, generateSessionId, playwrightProxy, setProxyMode, startRecording, startReplay, stopProxy };
+import '@playwright/test';
+export { P as PlaywrightTestInfo, g as generateSessionId, p as playwrightProxy, s as setProxyMode, b as startRecording, c as startReplay, d as stopProxy } from '../index-De4mgziH.cjs';
+import 'node:http';

package/dist/playwright/index.d.ts

@@ -1,50 +1,3 @@
-import { TestInfo } from '@playwright/test';
-
-type ProxyMode = 'recording' | 'replay' | 'transparent';
-type PlaywrightTestInfo = Pick<TestInfo, 'title'>;
-/**
- * Set the proxy mode for a given session
- * @param mode - The proxy mode to set (recording, replay, transparent)
- * @param sessionId - Unique identifier for the session
- * @param timeout - Optional timeout in milliseconds
- */
-declare function setProxyMode(mode: ProxyMode, sessionId: string, timeout?: number): Promise<void>;
-/**
- * Generate a session ID from test info
- * @param testInfo - Playwright test info object
- */
-declare function generateSessionId(testInfo: PlaywrightTestInfo): string;
-/**
- * Start recording for a test
- * @param testInfo - Playwright test info object
- */
-declare function startRecording(testInfo: PlaywrightTestInfo): Promise<void>;
-/**
- * Start replay for a test
- * @param testInfo - Playwright test info object
- */
-declare function startReplay(testInfo: PlaywrightTestInfo): Promise<void>;
-/**
- * Stop recording/replay and return to transparent mode
- * @param testInfo - Playwright test info object
- */
-declare function stopProxy(testInfo: PlaywrightTestInfo): Promise<void>;
-/**
- * Playwright test fixture helper for managing proxy mode
- * Use this in beforeEach/afterEach hooks
- */
-declare const playwrightProxy: {
-    /**
-     * Setup before test - sets the proxy mode
-     * @param testInfo - Playwright test info object
-     * @param mode - The proxy mode to use for this test
-     */
-    before(testInfo: PlaywrightTestInfo, mode: ProxyMode): Promise<void>;
-    /**
-     * Cleanup after test - returns to transparent mode
-     * @param testInfo - Playwright test info object
-     */
-    after(testInfo: PlaywrightTestInfo): Promise<void>;
-};
-
-export { type PlaywrightTestInfo, type ProxyMode, generateSessionId, playwrightProxy, setProxyMode, startRecording, startReplay, stopProxy };
+import '@playwright/test';
+export { P as PlaywrightTestInfo, g as generateSessionId, p as playwrightProxy, s as setProxyMode, b as startRecording, c as startReplay, d as stopProxy } from '../index-De4mgziH.js';
+import 'node:http';

package/dist/playwright/index.mjs

@@ -1,3 +1,10 @@
+// src/types.ts
+var Modes = {
+  transparent: "transparent",
+  record: "record",
+  replay: "replay"
+};
+
 // src/playwright/index.ts
 var INTERNAL_API_URL = process.env.INTERNAL_API_URL || "http://localhost:8100";
 async function setProxyMode(mode, sessionId, timeout) {
@@ -29,15 +36,15 @@ function generateSessionId(testInfo) {
 }
 async function startRecording(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("recording", sessionId);
+  await setProxyMode(Modes.record, sessionId);
 }
 async function startReplay(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("replay", sessionId);
+  await setProxyMode(Modes.replay, sessionId);
 }
 async function stopProxy(testInfo) {
   const sessionId = generateSessionId(testInfo);
-  await setProxyMode("transparent", sessionId);
+  await setProxyMode(Modes.transparent, sessionId);
 }
 var playwrightProxy = {
   /**
@@ -56,7 +63,7 @@ var playwrightProxy = {
    */
   async after(testInfo) {
     const sessionId = generateSessionId(testInfo);
-    await setProxyMode("transparent", sessionId);
+    await setProxyMode(Modes.transparent, sessionId);
   }
 };