@skrillex1224/playwright-toolkit 2.1.9 → 2.1.11

package/dist/index.cjs

@@ -1235,6 +1235,14 @@ var import_got_scraping = require("got-scraping");
 var import_http = require("http");
 var import_https2 = require("https");
 var logger9 = createLogger("Interception");
+var SHARED_HTTP_AGENT = new import_http.Agent({ keepAlive: false });
+var SHARED_HTTPS_AGENT = new import_https2.Agent({ keepAlive: false, rejectUnauthorized: false });
+var DirectConfig = {
+  /** 直连请求超时时间（秒） */
+  directTimeout: 12,
+  /** 静默扩展名：这些扩展名的直连成功日志用 debug 级别 */
+  silentExtensions: [".js"]
+};
 var ARCHIVE_EXTENSIONS = [".7z", ".zip", ".rar", ".gz", ".bz2", ".tar", ".zst"];
 var EXECUTABLE_EXTENSIONS = [".exe", ".apk", ".bin", ".dmg", ".jar", ".class"];
 var DOCUMENT_EXTENSIONS = [".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx", ".pdf", ".csv"];
@@ -1281,6 +1289,14 @@ var DEFAULT_BLOCKING_CONFIG = {
   /** 额外自定义扩展名列表 */
   customExtensions: []
 };
+var SHARED_GOT_OPTIONS = {
+  http2: false,
+  // 禁用 HTTP2 避免在拦截场景下的握手兼容性问题
+  retry: { limit: 0 },
+  // 让 Playwright 或外层逻辑处理重试
+  throwHttpErrors: false
+  // 404/500 等错误不抛出异常，直接透传给浏览器
+};
 var Interception = {
   /**
    * 根据配置生成需要屏蔽的扩展名列表
@@ -1324,16 +1340,6 @@ var Interception = {
   /**
    * 设置网络拦截规则（资源屏蔽 + CDN 直连）
    * 
-   * 工作流程：
-   * 1. 检查请求是否在屏蔽列表中 → 如果是，直接 abort
-   * 2. 检查是否匹配直连域名 → 如果是，使用 Node.js fetch 直连
-   * 3. 其他请求正常走代理
-   * 
-   * 适用场景：
-   * - 代理 IP 无法访问某些 CDN 域名
-   * - 需要加速静态资源加载
-   * - 屏蔽不必要的资源请求
-   * 
    * @param {import('playwright').Page} page - Playwright Page 对象
    * @param {Object} [options] - 配置选项
    * @param {string[]} [options.directDomains] - 需要直连的域名列表
@@ -1346,10 +1352,10 @@ var Interception = {
       directDomains = [],
       blockingConfig = {},
       fallbackToProxy = true
-      // 默认回退到代理，保证可用性
     } = options;
     const mergedBlockingConfig = { ...DEFAULT_BLOCKING_CONFIG, ...blockingConfig };
     const blockedExtensions = this.getBlockedExtensions(mergedBlockingConfig);
+    const hasDirectDomains = directDomains.length > 0;
     const enabledCategories = [];
     if (mergedBlockingConfig.blockArchive) enabledCategories.push("\u538B\u7F29\u5305");
     if (mergedBlockingConfig.blockExecutable) enabledCategories.push("\u53EF\u6267\u884C\u6587\u4EF6");
@@ -1359,21 +1365,22 @@ var Interception = {
     if (mergedBlockingConfig.blockFont) enabledCategories.push("\u5B57\u4F53");
     if (mergedBlockingConfig.blockCss) enabledCategories.push("CSS");
     if (mergedBlockingConfig.blockOther) enabledCategories.push("\u5176\u4ED6");
-    const hasDirectDomains = directDomains.length > 0;
-    logger9.start("setup", hasDirectDomains ? `\u76F4\u8FDE\u57DF\u540D: [${directDomains.join(", ")}]` : "\u4EC5\u8D44\u6E90\u5C4F\u853D\u6A21\u5F0F");
+    logger9.start("setup", hasDirectDomains ? `\u76F4\u8FDE\u57DF\u540D: [${directDomains.length} \u4E2A] | \u5C4F\u853D: [${enabledCategories.join(", ")}]` : `\u4EC5\u8D44\u6E90\u5C4F\u853D\u6A21\u5F0F | \u5C4F\u853D: [${enabledCategories.join(", ")}]`);
     await page.route("**/*", async (route) => {
-      const request = route.request();
-      const url = request.url();
-      const urlLower = url.toLowerCase();
-      const urlPath = urlLower.split("?")[0];
-      const shouldBlock = blockedExtensions.some((ext) => urlPath.endsWith(ext));
-      if (shouldBlock) {
-        logger9.debug(`\u5DF2\u5C4F\u853D: ${url.substring(0, 100)}`);
-        return route.abort();
-      }
-      if (hasDirectDomains) {
-        const matchesDomain = directDomains.some((domain) => url.includes(domain));
-        if (matchesDomain) {
+      let handled = false;
+      try {
+        const request = route.request();
+        const url = request.url();
+        const urlLower = url.toLowerCase();
+        const urlPath = urlLower.split("?")[0];
+        const isSilent = DirectConfig.silentExtensions.some((ext) => urlPath.endsWith(ext));
+        const shouldBlock = blockedExtensions.some((ext) => urlPath.endsWith(ext));
+        if (shouldBlock) {
+          await route.abort();
+          handled = true;
+          return;
+        }
+        if (hasDirectDomains && directDomains.some((domain) => url.includes(domain))) {
           try {
             const reqHeaders = await request.allHeaders();
             delete reqHeaders["host"];
@@ -1381,30 +1388,24 @@ var Interception = {
             const method = request.method();
             const postData = method !== "GET" && method !== "HEAD" ? request.postDataBuffer() : void 0;
             const response = await (0, import_got_scraping.gotScraping)({
+              ...SHARED_GOT_OPTIONS,
+              // 应用通用配置
               url,
               method,
               headers: reqHeaders,
               body: postData,
               responseType: "buffer",
-              // 直接获取 Buffer
-              // 【修复 1】禁用 HTTP/2，防止握手挂起
-              http2: false,
-              // 模拟浏览器指纹 (Header 部分)
+              // 强制获取 Buffer
+              // 模拟浏览器 TLS 指纹
               headerGeneratorOptions: Stealth.getTlsFingerprintOptions(userAgent),
-              retry: { limit: 0 },
-              // 【修复 2】显式使用新的 Agent 实例，彻底无视 Apify 的环境变量代理
-              // 注意：这会牺牲一部分 TLS 指纹模拟能力，但能保证网络通畅
+              // 使用共享的 Agent 单例（keepAlive: false，不会池化连接）
               agent: {
-                http: new import_http.Agent({ keepAlive: true }),
-                https: new import_https2.Agent({ keepAlive: true, rejectUnauthorized: false })
+                http: SHARED_HTTP_AGENT,
+                https: SHARED_HTTPS_AGENT
               },
-              // 设置超时，防止永久 Pending
-              timeout: {
-                request: 15 * 1e3
-                // 建议缩短到 15s
-              }
+              // 超时时间
+              timeout: { request: DirectConfig.directTimeout * 1e3 }
             });
-            const body = response.body;
             const resHeaders = {};
             for (const [key, value] of Object.entries(response.headers)) {
               if (Array.isArray(value)) {
@@ -1415,29 +1416,66 @@ var Interception = {
             }
             delete resHeaders["content-encoding"];
             delete resHeaders["content-length"];
-            logger9.debug(`\u76F4\u8FDE\u6210\u529F: ${url.substring(0, 100)}`);
-            await route.fulfill({
+            delete resHeaders["transfer-encoding"];
+            delete resHeaders["connection"];
+            delete resHeaders["keep-alive"];
+            isSilent ? logger9.debug(`\u76F4\u8FDE\u6210\u529F: ${url}`) : logger9.info(`\u76F4\u8FDE\u6210\u529F: ${url}`);
+            await safeFulfill(route, {
               status: response.statusCode,
               headers: resHeaders,
-              body
+              body: response.body
             });
+            handled = true;
             return;
           } catch (e) {
+            const isTimeout = e.code === "ETIMEDOUT" || e.message.toLowerCase().includes("timeout");
+            const action = fallbackToProxy ? "\u56DE\u9000\u4EE3\u7406" : "\u5DF2\u653E\u5F03";
+            const reason = isTimeout ? `\u8D85\u65F6(${DirectConfig.directTimeout}s)` : `\u5F02\u5E38: ${e.message}`;
+            logger9.warn(`\u76F4\u8FDE${reason}\uFF0C${action}: ${url}`);
             if (fallbackToProxy) {
-              logger9.warn(`\u76F4\u8FDE\u5931\u8D25\uFF0C\u56DE\u9000\u4EE3\u7406: ${url.substring(0, 80)} | \u539F\u56E0: ${e.message}`);
-              return route.continue();
+              await safeContinue(route);
             } else {
-              logger9.warn(`\u76F4\u8FDE\u5931\u8D25: ${url.substring(0, 80)} | \u539F\u56E0: ${e.message}`);
-              return route.abort();
+              await route.abort();
             }
+            handled = true;
+            return;
+          }
+        }
+        await safeContinue(route);
+        handled = true;
+      } catch (err) {
+        logger9.warn(`\u8DEF\u7531\u5904\u7406\u5F02\u5E38: ${err.message}`);
+        if (!handled) {
+          try {
+            await route.continue();
+          } catch (_) {
           }
         }
       }
-      return route.continue();
     });
-    logger9.success("setup", `\u5C4F\u853D\u5206\u7C7B: [${enabledCategories.join(", ")}]`);
   }
 };
+async function safeFulfill(route, options) {
+  try {
+    await route.fulfill(options);
+  } catch (error) {
+    if (!isIgnorableError(error)) {
+      console.error(`[Interception] Fulfill Error: ${error.message}`);
+    }
+  }
+}
+async function safeContinue(route) {
+  try {
+    await route.continue();
+  } catch (error) {
+    if (!isIgnorableError(error)) {
+    }
+  }
+}
+function isIgnorableError(error) {
+  const msg = error.message;
+  return msg.includes("already handled") || msg.includes("Target closed") || msg.includes("closed");
+}
 
 // index.js
 var usePlaywrightToolKit = () => {