@skrillex1224/playwright-toolkit 2.0.75 → 2.0.77

package/dist/index.js

@@ -1068,7 +1068,7 @@ var Sse = {
       onEnd,
       onTimeout,
       initialTimeout = 9e4,
-      timeout = 18e4
+      overallTimeout = 18e4
     } = options;
     let initialTimer = null;
     let overallTimer = null;
@@ -1080,82 +1080,87 @@ var Sse = {
       overallTimer = null;
     };
     const workPromise = new Promise(async (resolve, reject) => {
-      await page.route(urlPattern, async (route) => {
-        const request = route.request();
-        const requestUrl = request.url();
-        logger8.info(`[MITM] \u5DF2\u62E6\u622A\u8BF7\u6C42: ${requestUrl}`);
-        try {
-          const headers = await request.allHeaders();
-          const postData = request.postData();
-          const urlObj = new URL(requestUrl);
-          delete headers["accept-encoding"];
-          delete headers["content-length"];
-          const reqOptions = {
-            hostname: urlObj.hostname,
-            port: 443,
-            path: urlObj.pathname + urlObj.search,
-            method: request.method(),
-            headers,
-            timeout
-          };
-          const req = https.request(reqOptions, (res) => {
-            const chunks = [];
-            let accumulatedText = "";
-            res.on("data", (chunk) => {
-              if (!hasReceivedInitialData) {
-                hasReceivedInitialData = true;
-                if (initialTimer) {
-                  clearTimeout(initialTimer);
-                  initialTimer = null;
+      try {
+        await page.route(urlPattern, async (route) => {
+          const request = route.request();
+          logger8.info(`[MITM] \u5DF2\u62E6\u622A\u8BF7\u6C42: ${request.url()}`);
+          try {
+            const headers = await request.allHeaders();
+            const postData = request.postData();
+            const urlObj = new URL(request.url());
+            delete headers["accept-encoding"];
+            delete headers["content-length"];
+            const reqOptions = {
+              hostname: urlObj.hostname,
+              port: 443,
+              path: urlObj.pathname + urlObj.search,
+              method: request.method(),
+              headers,
+              timeout: overallTimeout
+            };
+            const req = https.request(reqOptions, (res) => {
+              const chunks = [];
+              let accumulatedText = "";
+              res.on("data", (chunk) => {
+                if (!hasReceivedInitialData) {
+                  hasReceivedInitialData = true;
+                  if (initialTimer) {
+                    clearTimeout(initialTimer);
+                    initialTimer = null;
+                  }
+                  logger8.debug("[Intercept] \u5DF2\u63A5\u6536\u521D\u59CB\u6570\u636E");
+                }
+                chunks.push(chunk);
+                const textChunk = chunk.toString("utf-8");
+                accumulatedText += textChunk;
+                if (onData) {
+                  try {
+                    onData(textChunk, resolve, accumulatedText);
+                  } catch (e) {
+                    logger8.fail(`onData \u9519\u8BEF`, e);
+                  }
                 }
-                logger8.debug("[Intercept] \u5DF2\u63A5\u6536\u521D\u59CB\u6570\u636E");
-              }
-              chunks.push(chunk);
-              const textChunk = chunk.toString("utf-8");
-              accumulatedText += textChunk;
-              logger8.debug(`[CHUNK] ${textChunk.length} bytes`);
-              if (onData) {
-                try {
-                  onData(textChunk, resolve, accumulatedText);
-                } catch (e) {
-                  logger8.fail(`onData callback error: ${e.message}`, e);
+              });
+              res.on("end", () => {
+                logger8.info("[MITM] \u4E0A\u6E38\u54CD\u5E94\u7ED3\u675F");
+                clearAllTimers();
+                if (onEnd) {
+                  try {
+                    onEnd(accumulatedText, resolve);
+                  } catch (e) {
+                    logger8.fail(`onEnd \u9519\u8BEF`, e);
+                  }
+                } else if (!onData) {
+                  resolve(accumulatedText);
                 }
-              }
+                route.fulfill({
+                  status: res.statusCode,
+                  headers: res.headers,
+                  body: Buffer.concat(chunks)
+                }).catch(() => {
+                });
+              });
             });
-            res.on("end", () => {
-              logger8.info("[MITM] \u4E0A\u6E38\u54CD\u5E94\u7ED3\u675F (Stream End)");
+            req.on("error", (e) => {
               clearAllTimers();
-              const fullBody = Buffer.concat(chunks);
-              if (onEnd) {
-                try {
-                  onEnd(accumulatedText, resolve);
-                } catch (e) {
-                  logger8.fail(`onEnd callback error: ${e.message}`, e);
-                }
-              } else if (!onData) {
-                resolve(accumulatedText);
-              }
-              route.fulfill({
-                status: res.statusCode,
-                headers: res.headers,
-                body: fullBody
-              }).catch((e) => logger8.warn(`Route fulfill failed: ${e.message}`));
+              route.abort().catch(() => {
+              });
+              reject(e);
             });
-          });
-          req.on("error", (e) => {
-            logger8.fail(`[MITM] Upstream request error: ${e.message}`, e);
+            if (postData) req.write(postData);
+            req.end();
+          } catch (e) {
             clearAllTimers();
-            route.abort();
+            route.continue().catch(() => {
+            });
             reject(e);
-          });
-          if (postData) req.write(postData);
-          req.end();
-        } catch (e) {
-          logger8.fail(`[MITM] Setup error: ${e.message}`, e);
-          clearAllTimers();
-          route.continue();
-        }
-      });
+          }
+        });
+      } catch (e) {
+        reject(e);
+      }
+    });
+    workPromise.catch(() => {
     });
     const timeoutPromise = new Promise((_, reject) => {
       initialTimer = setTimeout(() => {
@@ -1163,26 +1168,33 @@ var Sse = {
           const error = new InitialTimeoutError(initialTimeout);
           clearAllTimers();
           if (onTimeout) {
-            onTimeout(error, reject);
+            try {
+              onTimeout(error, reject);
+            } catch (e) {
+              reject(e);
+            }
           } else {
-            logger8.fail(`[Intercept] \u521D\u59CB\u6570\u636E\u63A5\u6536\u8D85\u65F6`, error);
             reject(error);
           }
         }
       }, initialTimeout);
       overallTimer = setTimeout(() => {
-        const error = new OverallTimeoutError(timeout);
+        const error = new OverallTimeoutError(overallTimeout);
         clearAllTimers();
         if (onTimeout) {
-          onTimeout(error, reject);
+          try {
+            onTimeout(error, reject);
+          } catch (e) {
+            reject(e);
+          }
         } else {
-          logger8.fail(`[Intercept] \u6574\u4F53\u8BF7\u6C42\u8D85\u65F6`, error);
           reject(error);
         }
-      }, timeout);
+      }, overallTimeout);
+    });
+    timeoutPromise.catch(() => {
     });
-    const racePromise = Promise.race([workPromise, timeoutPromise]);
-    return racePromise;
+    return Promise.race([workPromise, timeoutPromise]);
   }
 };