feat: try simplify headers setting

Author: clChenLiang

src/McpContext.ts

@@ -134,7 +134,9 @@ export class McpContext implements Context {
     this.#options = options;
     this.#headers = options.headers;
 
-    this.#networkCollector = new NetworkCollector(this.browser);
+    this.#networkCollector = new NetworkCollector(this.browser, undefined, {
+      headers: this.#headers,
+    });
 
     this.#consoleCollector = new ConsoleCollector(this.browser, collect => {
       return {
@@ -159,42 +161,9 @@ export class McpContext implements Context {
 
   async #init() {
     const pages = await this.createPagesSnapshot();
-    await this.#networkCollector.init(pages, this.#headers);
+    await this.#networkCollector.init(pages);
     await this.#consoleCollector.init(pages);
   }
-  
-  /**
-   * Gets the custom headers to add to all network requests.
-   */
-  getHeaders(): Record<string, string> | undefined {
-    return this.#headers;
-  }
-  
-  /**
-   * Sets custom headers to add to all network requests.
-   */
-  setHeaders(headers: Record<string, string> | undefined): void {
-    this.#headers = headers;
-    // Update all existing pages with the new headers
-    for (const page of this.#pages) {
-      this.#applyHeadersToPage(page, headers);
-    }
-  }
-  
-  /**
-   * Applies custom headers to a specific page.
-   */
-  async #applyHeadersToPage(page: Page, headers: Record<string, string> | undefined): Promise<void> {
-    try {
-      if (headers) {
-        await page.setExtraHTTPHeaders(headers);
-      } else {
-        await page.setExtraHTTPHeaders({});
-      }
-    } catch (error) {
-      this.logger('Error applying headers to page:', error);
-    }
-  }
 
   dispose() {
     this.#networkCollector.dispose();
@@ -714,6 +683,8 @@ export class McpContext implements Context {
           collect(req);
         },
       } as ListenerMap;
+    }, {
+      headers: this.#headers,
     });
     await this.#networkCollector.init(await this.browser.pages());
   }

src/PageCollector.ts

@@ -355,15 +355,24 @@ export class NetworkCollector extends PageCollector<HTTPRequest> {
     browser: Browser,
     listeners: (
       collector: (item: HTTPRequest) => void,
-    ) => ListenerMap<PageEvents> = collect => ({
-      request: req => collect(req),
-    } as ListenerMap),
+    ) => ListenerMap<PageEvents> = collect => {
+      return {
+        request: req => {
+          collect(req);
+        },
+      } as ListenerMap;
+    },
+    options?: Record<string, unknown> & {
+      headers?: Record<string, string>
+    }
   ) {
     super(browser, listeners);
+    if (options?.headers) {
+      this.#headers = options?.headers;
+    }
   }
 
-  override async init(pages: Page[], headers?: Record<string, string>): Promise<void> {
-    this.#headers = headers;
+  override async init(pages: Page[]): Promise<void> {
     for (const page of pages) {
       await this.#applyHeadersToPage(page);
     }