feat: first can use

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,6 +132,7 @@ export class McpContext implements Context {
     this.logger = logger;
     this.#locatorClass = locatorClass;
     this.#options = options;
+    this.#headers = options.headers;
 
     this.#networkCollector = new NetworkCollector(this.browser);
 
@@ -153,9 +159,42 @@ export class McpContext implements Context {
 
   async #init() {
     const pages = await this.createPagesSnapshot();
-    await this.#networkCollector.init(pages);
+    await this.#networkCollector.init(pages, this.#headers);
     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();

src/PageCollector.ts

@@ -349,20 +349,41 @@ class PageIssueSubscriber {
 }
 
 export class NetworkCollector extends PageCollector<HTTPRequest> {
+  #headers?: Record<string, string>;
+
   constructor(
     browser: Browser,
     listeners: (
       collector: (item: HTTPRequest) => void,
-    ) => ListenerMap<PageEvents> = collect => {
-      return {
-        request: req => {
-          collect(req);
-        },
-      } as ListenerMap;
-    },
+    ) => ListenerMap<PageEvents> = collect => ({
+      request: req => collect(req),
+    } as ListenerMap),
   ) {
     super(browser, listeners);
   }
+
+  override async init(pages: Page[], headers?: Record<string, string>): Promise<void> {
+    this.#headers = headers;
+    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) ?? [];
     if (!navigations) {

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;
 }