feat: support extra headers

feat: try simplify headers setting

Author: clChenLiang

src/McpContext.ts

@@ -58,6 +58,8 @@ interface McpContextOptions {
   experimentalDevToolsDebugging: boolean;
   // Whether all page-like targets are exposed as pages.
   experimentalIncludeAllPages?: boolean;
+  // Custom headers to add to all network requests made by the browser.
+  headers?: Record<string, string>;
 }
 
 const DEFAULT_TIMEOUT = 5_000;
@@ -104,6 +106,9 @@ export class McpContext implements Context {
   #textSnapshot: TextSnapshot | null = null;
   #networkCollector: NetworkCollector;
   #consoleCollector: ConsoleCollector;
+  
+  // Custom headers to add to all network requests made by the browser.
+  #headers?: Record<string, string>;
 
   #isRunningTrace = false;
   #networkConditionsMap = new WeakMap<Page, string>();
@@ -127,8 +132,11 @@ export class McpContext implements Context {
     this.logger = logger;
     this.#locatorClass = locatorClass;
     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 {
@@ -675,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

@@ -349,6 +349,8 @@ class PageIssueSubscriber {
 }
 
 export class NetworkCollector extends PageCollector<HTTPRequest> {
+  #headers?: Record<string, string>;
+
   constructor(
     browser: Browser,
     listeners: (
@@ -360,8 +362,36 @@ export class NetworkCollector extends PageCollector<HTTPRequest> {
         },
       } 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[]): Promise<void> {
+    for (const page of pages) {
+      await this.#applyHeadersToPage(page);
+    }
+    await super.init(pages);
+  }
+
+  override addPage(page: Page): void {
+    super.addPage(page);
+    void this.#applyHeadersToPage(page);
+  }
+
+  async #applyHeadersToPage(page: Page): Promise<void> {
+    if (this.#headers) {
+      try {
+        await page.setExtraHTTPHeaders(this.#headers);
+      } catch (error) {
+        logger('Error applying headers to page:', error);
+      }
+    }
   }
   override splitAfterNavigation(page: Page) {
     const navigations = this.storage.get(page) ?? [];

src/cli.ts

@@ -87,6 +87,27 @@ export const cliOptions = {
       }
     },
   },
+  headers: {
+    type: 'string',
+    description:
+      'Custom headers to add to all network requests made by the browser in JSON format (e.g., \'{"Authorization":"Bearer token","User-Agent":"Custom"}\').',
+    coerce: (val: string | undefined) => {
+      if (!val) {
+        return;
+      }
+      try {
+        const parsed = JSON.parse(val);
+        if (typeof parsed !== 'object' || Array.isArray(parsed)) {
+          throw new Error('Headers must be a JSON object');
+        }
+        return parsed as Record<string, string>;
+      } catch (error) {
+        throw new Error(
+          `Invalid JSON for headers: ${(error as Error).message}`,
+        );
+      }
+    },
+  },
   headless: {
     type: 'boolean',
     description: 'Whether to run in headless (no UI) mode.',

src/main.ts

@@ -85,9 +85,10 @@ async function getContext(): Promise<McpContext> {
 
   if (context?.browser !== browser) {
     context = await McpContext.from(browser, logger, {
-      experimentalDevToolsDebugging: devtools,
-      experimentalIncludeAllPages: args.experimentalIncludeAllPages,
-    });
+          experimentalDevToolsDebugging: devtools,
+          experimentalIncludeAllPages: args.experimentalIncludeAllPages,
+          headers: args.headers,
+        });
   }
   return context;
 }