@skrillex1224/playwright-toolkit 2.1.91 → 2.1.93

package/dist/index.cjs

@@ -1350,17 +1350,20 @@ var Sse = {
       initialTimer = null;
       overallTimer = null;
     };
-    let streamIndex = 0;
+    let routeHandler = null;
+    let isCancelled = false;
     const workPromise = new Promise((resolve, reject) => {
-      page.route(urlPattern, async (route) => {
-        streamIndex++;
-        const currentStreamIndex = streamIndex;
+      routeHandler = async (route) => {
+        if (isCancelled) {
+          route.continue().catch(() => {
+          });
+          return;
+        }
         const request = route.request();
-        const postData = request.postData();
-        const postPreview = postData ? postData.substring(0, 1e3) : "(\u65E0POST\u6570\u636E)";
-        logger8.info(`[MITM] \u{1F539} \u6D41#${currentStreamIndex} \u62E6\u622A\u8BF7\u6C42: ${request.url()} POST\u9884\u89C8: ${postPreview}`);
+        logger8.info(`[MITM] \u5DF2\u62E6\u622A\u8BF7\u6C42: ${request.url()}`);
         try {
           const headers = await request.allHeaders();
+          const postData = request.postData();
           const urlObj = new import_url.URL(request.url());
           delete headers["accept-encoding"];
           delete headers["content-length"];
@@ -1396,7 +1399,7 @@ var Sse = {
               }
             });
             res.on("end", () => {
-              logger8.info(`[MITM] \u{1F539} \u6D41#${currentStreamIndex} \u4E0A\u6E38\u54CD\u5E94\u7ED3\u675F | \u7D2F\u8BA1\u5B57\u7B26\u6570: ${accumulatedText.length}`);
+              logger8.info("[MITM] \u4E0A\u6E38\u54CD\u5E94\u7ED3\u675F");
               clearAllTimers();
               if (onEnd) {
                 try {
@@ -1429,7 +1432,8 @@ var Sse = {
           });
           reject(e);
         }
-      }).catch(reject);
+      };
+      page.route(urlPattern, routeHandler).catch(reject);
     });
     const timeoutPromise = new Promise((_, reject) => {
       initialTimer = setTimeout(() => {
@@ -1476,7 +1480,20 @@ var Sse = {
     const racePromise = Promise.race([workPromise, timeoutPromise]);
     racePromise.catch(() => {
     });
-    return racePromise;
+    const cancel = async () => {
+      if (isCancelled) return;
+      isCancelled = true;
+      clearAllTimers();
+      if (routeHandler) {
+        try {
+          await page.unroute(urlPattern, routeHandler);
+          logger8.info("[MITM] \u5DF2\u53D6\u6D88 SSE \u6D41\u76D1\u542C");
+        } catch (e) {
+          logger8.debug("[MITM] \u53D6\u6D88\u76D1\u542C\u65F6\u51FA\u9519\uFF08\u53EF\u80FD\u5DF2\u53D6\u6D88\uFF09");
+        }
+      }
+    };
+    return [racePromise, cancel];
   }
 };