@skrillex1224/playwright-toolkit 2.0.92 → 2.0.93

package/dist/index.js

@@ -1308,8 +1308,11 @@ var Blocking = {
     };
   },
   /**
-   * 设置 CDN 直连规则
-   * 对于指定域名的请求，使用 Node.js 原生 fetch 直连获取（绕过浏览器代理）
+   * 设置 CDN 直连规则（与 setupBlockingResources 协同工作）
+   * 
+   * 对于指定域名：
+   * 1. 如果资源在屏蔽列表中 → 直接 abort（不请求）
+   * 2. 如果资源不在屏蔽列表中 → 使用 Node.js 原生 fetch 直连获取（绕过浏览器代理）
    * 
    * 适用场景：
    * - 代理 IP 无法访问某些 CDN 域名（如 statics.moonshot.cn）
@@ -1320,33 +1323,37 @@ var Blocking = {
    * - 如果在海外服务器（如 Apify 云端）运行，直连仍使用海外 IP，可能依然无法访问
    * 
    * @param {import('playwright').Page} page - Playwright Page 对象
-   * @param {Object} [options] - 配置选项
-   * @param {string[]} [options.domains] - 需要直连的域名列表
-   * @param {string[]} [options.extensions] - 需要直连的扩展名列表（可选，默认 CSS/JS/字体）
+   * @param {Object} options - 配置选项
+   * @param {string[]} options.domains - 需要直连的域名列表（必填）
+   * @param {Object} [options.blockingConfig] - 资源屏蔽配置（与 setupBlockingResources 相同）
    * @param {boolean} [options.fallbackToProxy] - 直连失败时是否回退到代理（默认 true）
    * @returns {Promise<void>}
    */
   async setupDirectConnect(page, options = {}) {
     const {
       domains = [],
-      extensions = [".css", ".js", ".woff", ".woff2", ".ttf", ".otf"],
+      blockingConfig = {},
       fallbackToProxy = true
     } = options;
     if (domains.length === 0) {
       logger9.warn("setupDirectConnect", "\u672A\u6307\u5B9A\u57DF\u540D\u5217\u8868\uFF0C\u8DF3\u8FC7\u76F4\u8FDE\u914D\u7F6E");
       return;
     }
-    logger9.start("setupDirectConnect", `\u57DF\u540D: [${domains.join(", ")}]`);
-    let directCount = 0;
-    let fallbackCount = 0;
+    const mergedBlockingConfig = { ...DEFAULT_BLOCKING_CONFIG, ...blockingConfig };
+    const blockedExtensions = this.getBlockedExtensions(mergedBlockingConfig);
+    logger9.start("setupDirectConnect", `\u76F4\u8FDE\u57DF\u540D: [${domains.join(", ")}]`);
     await page.route("**/*", async (route) => {
       const request = route.request();
       const url = request.url();
       const urlLower = url.toLowerCase();
       const urlPath = urlLower.split("?")[0];
       const matchesDomain = domains.some((domain) => url.includes(domain));
-      const matchesExtension = extensions.length === 0 || extensions.some((ext) => urlPath.endsWith(ext));
-      if (matchesDomain && matchesExtension) {
+      if (matchesDomain) {
+        const shouldBlock2 = blockedExtensions.some((ext) => urlPath.endsWith(ext));
+        if (shouldBlock2) {
+          logger9.debug(`blocked: ${url.substring(0, 100)}`);
+          return route.abort();
+        }
         try {
           const response = await fetch(url, {
             method: request.method(),
@@ -1362,7 +1369,6 @@ var Blocking = {
           response.headers.forEach((value, key) => {
             headers[key] = value;
           });
-          directCount++;
           logger9.debug(`direct: ${url.substring(0, 100)}`);
           await route.fulfill({
             status: response.status,
@@ -1372,18 +1378,31 @@ var Blocking = {
           return;
         } catch (e) {
           if (fallbackToProxy) {
-            fallbackCount++;
             logger9.debug(`fallback: ${url.substring(0, 80)} (${e.message})`);
             return route.continue();
           } else {
-            logger9.warn("setupDirectConnect", `\u76F4\u8FDE\u5931\u8D25\u4E14\u4E0D\u56DE\u9000: ${url.substring(0, 80)}`);
+            logger9.warn("setupDirectConnect", `\u76F4\u8FDE\u5931\u8D25: ${url.substring(0, 80)}`);
             return route.abort();
           }
         }
       }
+      const shouldBlock = blockedExtensions.some((ext) => urlPath.endsWith(ext));
+      if (shouldBlock) {
+        logger9.debug(`blocked: ${url.substring(0, 100)}`);
+        return route.abort();
+      }
       return route.continue();
     });
-    logger9.success("setupDirectConnect", `\u6269\u5C55\u540D: [${extensions.join(", ")}]`);
+    const enabledCategories = [];
+    if (mergedBlockingConfig.blockArchive) enabledCategories.push("\u538B\u7F29\u5305");
+    if (mergedBlockingConfig.blockExecutable) enabledCategories.push("\u53EF\u6267\u884C\u6587\u4EF6");
+    if (mergedBlockingConfig.blockDocument) enabledCategories.push("\u529E\u516C\u6587\u6863");
+    if (mergedBlockingConfig.blockImage) enabledCategories.push("\u56FE\u7247");
+    if (mergedBlockingConfig.blockMedia) enabledCategories.push("\u97F3\u89C6\u9891");
+    if (mergedBlockingConfig.blockFont) enabledCategories.push("\u5B57\u4F53");
+    if (mergedBlockingConfig.blockCss) enabledCategories.push("CSS");
+    if (mergedBlockingConfig.blockOther) enabledCategories.push("\u5176\u4ED6");
+    logger9.success("setupDirectConnect", `\u5C4F\u853D\u5206\u7C7B: [${enabledCategories.join(", ")}]`);
   }
 };