refactor: bundle puppeteer-core (#417)

Continue speeding up boot time.

---------

Co-authored-by: Piotr Paulski <[email protected]>

Author: zyzyzyryxy

package.json

@@ -40,7 +40,6 @@
   "dependencies": {
     "core-js": "3.46.0",
     "debug": "4.4.3",
-    "puppeteer-core": "^24.24.1",
     "yargs": "18.0.0"
   },
   "devDependencies": {

rollup.config.mjs

@@ -29,11 +29,31 @@ import license from 'rollup-plugin-license';
 
 const isProduction = process.env.NODE_ENV === 'production';
 
-/** @type {import('rollup').RollupOptions} */
-const sdk = {
-  input: './build/src/third_party/modelcontextprotocol-sdk/index.js',
+const allowedLicenses = [
+  'MIT',
+  'Apache 2.0',
+  'Apache-2.0',
+  'BSD-3-Clause',
+  'BSD-2-Clause',
+  'ISC',
+  '0BSD',
+];
+
+/**
+ * @param {string} wrapperIndexPath
+ * @param {import('rollup').OutputOptions} [extraOutputOptions={}]
+ * @param {string[]} [external=[]]
+ * @returns {import('rollup').RollupOptions}
+ */
+const bundleDependency = (
+  wrapperIndexPath,
+  extraOutputOptions = {},
+  external = [],
+) => ({
+  input: wrapperIndexPath,
   output: {
-    file: './build/src/third_party/modelcontextprotocol-sdk/index.js',
+    ...extraOutputOptions,
+    file: wrapperIndexPath,
     sourcemap: !isProduction,
     format: 'esm',
   },
@@ -48,18 +68,14 @@ const sdk = {
       thirdParty: {
         allow: {
           test: dependency => {
-            let allowed_licenses = ['MIT', 'Apache 2.0', 'BSD-2-Clause', 'ISC'];
-            return allowed_licenses.includes(dependency.license);
+            return allowedLicenses.includes(dependency.license);
           },
           failOnUnlicensed: true,
           failOnViolation: true,
         },
         output: {
           file: path.join(
-            'build',
-            'src',
-            'third_party',
-            'modelcontextprotocol-sdk',
+            path.dirname(wrapperIndexPath),
             'THIRD_PARTY_NOTICES',
           ),
           template(dependencies) {
@@ -90,6 +106,16 @@ const sdk = {
     json(),
     nodeResolve(),
   ],
-};
+  external,
+});
 
-export default [sdk];
+export default [
+  bundleDependency('./build/src/third_party/modelcontextprotocol-sdk/index.js'),
+  bundleDependency(
+    './build/src/third_party/puppeteer-core/index.js',
+    {
+      inlineDynamicImports: true,
+    },
+    ['./bidi.js', '../bidi/bidi.js'],
+  ),
+];

src/DevToolsConnectionAdapter.ts

@@ -4,10 +4,10 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import {type ConnectionTransport} from 'puppeteer-core';
-
 import {Connection} from '../node_modules/chrome-devtools-frontend/front_end/core/protocol_client/InspectorBackend.js';
 
+import {type ConnectionTransport} from './third_party/puppeteer-core/index.js';
+
 /**
  * Allows a puppeteer {@link ConnectionTransport} to act like a DevTools {@link Connection}.
  */

src/McpContext.ts

@@ -8,6 +8,10 @@ import os from 'node:os';
 import path from 'node:path';
 
 import type {Debugger} from 'debug';
+
+import type {ListenerMap} from './PageCollector.js';
+import {NetworkCollector, PageCollector} from './PageCollector.js';
+import {Locator} from './third_party/puppeteer-core/index.js';
 import type {
   Browser,
   ConsoleMessage,
@@ -17,10 +21,7 @@ import type {
   Page,
   SerializedAXNode,
   PredefinedNetworkConditions,
-} from 'puppeteer-core';
-
-import type {ListenerMap} from './PageCollector.js';
-import {NetworkCollector, PageCollector} from './PageCollector.js';
+} from './third_party/puppeteer-core/index.js';
 import {listPages} from './tools/pages.js';
 import {takeSnapshot} from './tools/snapshot.js';
 import {CLOSE_PAGE_ERROR} from './tools/ToolDefinition.js';
@@ -91,9 +92,16 @@ export class McpContext implements Context {
   #nextSnapshotId = 1;
   #traceResults: TraceResult[] = [];
 
-  private constructor(browser: Browser, logger: Debugger) {
+  #locatorClass: typeof Locator;
+
+  private constructor(
+    browser: Browser,
+    logger: Debugger,
+    locatorClass: typeof Locator,
+  ) {
     this.browser = browser;
     this.logger = logger;
+    this.#locatorClass = locatorClass;
 
     this.#networkCollector = new NetworkCollector(this.browser);
 
@@ -122,8 +130,13 @@ export class McpContext implements Context {
     await this.#consoleCollector.init();
   }
 
-  static async from(browser: Browser, logger: Debugger) {
-    const context = new McpContext(browser, logger);
+  static async from(
+    browser: Browser,
+    logger: Debugger,
+    /* Let tests use unbundled Locator class to avoid overly strict checks within puppeteer that fail when mixing bundled and unbundled class instances */
+    locatorClass: typeof Locator = Locator,
+  ) {
+    const context = new McpContext(browser, logger, locatorClass);
     await context.#init();
     return context;
   }
@@ -428,4 +441,28 @@ export class McpContext implements Context {
   getNetworkRequestStableId(request: HTTPRequest): number {
     return this.#networkCollector.getIdForResource(request);
   }
+
+  waitForTextOnPage({
+    text,
+    timeout,
+  }: {
+    text: string;
+    timeout?: number | undefined;
+  }): Promise<Element> {
+    const page = this.getSelectedPage();
+    const frames = page.frames();
+
+    const locator = this.#locatorClass.race(
+      frames.flatMap(frame => [
+        frame.locator(`aria/${text}`),
+        frame.locator(`text/${text}`),
+      ]),
+    );
+
+    if (timeout) {
+      locator.setTimeout(timeout);
+    }
+
+    return locator.wait();
+  }
 }

src/McpResponse.ts

@@ -3,8 +3,6 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import type {ConsoleMessage, ResourceType} from 'puppeteer-core';
-
 import type {ConsoleMessageData} from './formatters/consoleFormatter.js';
 import {
   formatConsoleEventShort,
@@ -23,6 +21,10 @@ import type {
   ImageContent,
   TextContent,
 } from './third_party/modelcontextprotocol-sdk/index.js';
+import type {
+  ConsoleMessage,
+  ResourceType,
+} from './third_party/puppeteer-core/index.js';
 import {handleDialog} from './tools/pages.js';
 import type {ImageContentData, Response} from './tools/ToolDefinition.js';
 import {paginate} from './utils/pagination.js';

src/PageCollector.ts

@@ -11,7 +11,7 @@ import {
   type HTTPRequest,
   type Page,
   type PageEvents,
-} from 'puppeteer-core';
+} from './third_party/puppeteer-core/index.js';
 
 export type ListenerMap<EventMap extends PageEvents = PageEvents> = {
   [K in keyof EventMap]?: (event: EventMap[K]) => void;

src/WaitForHelper.ts

@@ -3,10 +3,10 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import type {Page, Protocol} from 'puppeteer-core';
 import type {CdpPage} from 'puppeteer-core/internal/cdp/Page.js';
 
 import {logger} from './logger.js';
+import type {Page, Protocol} from './third_party/puppeteer-core/index.js';
 
 export class WaitForHelper {
   #abortController = new AbortController();

src/browser.ts

@@ -13,8 +13,8 @@ import type {
   ChromeReleaseChannel,
   LaunchOptions,
   Target,
-} from 'puppeteer-core';
-import puppeteer from 'puppeteer-core';
+} from './third_party/puppeteer-core/index.js';
+import {puppeteer} from './third_party/puppeteer-core/index.js';
 
 let browser: Browser | undefined;
 

src/formatters/networkFormatter.ts

@@ -6,7 +6,10 @@
 
 import {isUtf8} from 'node:buffer';
 
-import type {HTTPRequest, HTTPResponse} from 'puppeteer-core';
+import type {
+  HTTPRequest,
+  HTTPResponse,
+} from '../third_party/puppeteer-core/index.js';
 
 const BODY_CONTEXT_SIZE_LIMIT = 10000;
 

src/third_party/puppeteer-core/index.ts

@@ -0,0 +1,9 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+export {Locator, PredefinedNetworkConditions} from 'puppeteer-core';
+export {default as puppeteer} from 'puppeteer-core';
+export type * from 'puppeteer-core';