chrome-devtools-mcp 0.21.0 → 0.23.0

package/README.md

@@ -1,11 +1,12 @@
-# Chrome DevTools MCP
+# Chrome DevTools for Agents
 
 [![npm chrome-devtools-mcp package](https://img.shields.io/npm/v/chrome-devtools-mcp.svg)](https://npmjs.org/package/chrome-devtools-mcp)
 
-`chrome-devtools-mcp` lets your coding agent (such as Gemini, Claude, Cursor or Copilot)
+Chrome DevTools for Agents (`chrome-devtools-mcp`) lets your coding agent (such as Gemini, Claude, Cursor or Copilot)
 control and inspect a live Chrome browser. It acts as a Model-Context-Protocol
 (MCP) server, giving your AI coding assistant access to the full power of
 Chrome DevTools for reliable automation, in-depth debugging, and performance analysis.
+A [CLI](docs/cli.md) is also provided for use without MCP.
 
 ## [Tool reference](./docs/tool-reference.md) | [Changelog](./CHANGELOG.md) | [Contributing](./CONTRIBUTING.md) | [Troubleshooting](./docs/troubleshooting.md) | [Design Principles](./docs/design-principles.md)
 
@@ -28,7 +29,7 @@ Avoid sharing sensitive or personal information that you don't want to share wit
 MCP clients.
 
 `chrome-devtools-mcp` officially supports Google Chrome and [Chrome for Testing](https://developer.chrome.com/blog/chrome-for-testing/) only.
-Other Chromium-based browser may work, but this is not guaranteed, and you may encounter unexpected behavior. Use at your own discretion.
+Other Chromium-based browsers may work, but this is not guaranteed, and you may encounter unexpected behavior. Use at your own discretion.
 We are committed to providing fixes and support for the latest version of [Extended Stable Chrome](https://chromiumdash.appspot.com/schedule).
 
 Performance tools may send trace URLs to the Google CrUX API to fetch real-user
@@ -51,7 +52,12 @@ Google handles this data in accordance with the [Google Privacy Policy](https://
 
 Google's collection of usage statistics for Chrome DevTools MCP is independent from the Chrome browser's usage statistics. Opting out of Chrome metrics does not automatically opt you out of this tool, and vice-versa.
 
-Collection is disabled if CHROME_DEVTOOLS_MCP_NO_USAGE_STATISTICS or CI env variables are set.
+Collection is disabled if `CHROME_DEVTOOLS_MCP_NO_USAGE_STATISTICS` or `CI` env variables are set.
+
+## Update checks
+
+By default, the server periodically checks the npm registry for updates and logs a notification when a newer version is available.
+You can disable these update checks by setting the `CHROME_DEVTOOLS_MCP_NO_UPDATE_CHECKS` environment variable.
 
 ## Requirements
 
@@ -74,7 +80,7 @@ Add the following config to your MCP client:
 }
 ```
 
-> [!NOTE]  
+> [!NOTE]
 > Using `chrome-devtools-mcp@latest` ensures that your MCP client will always use the latest version of the Chrome DevTools MCP server.
 
 If you are interested in doing only basic browser tasks, use the `--slim` mode:
@@ -143,7 +149,7 @@ claude mcp add chrome-devtools --scope user npx chrome-devtools-mcp@latest
 
 **Install as a Plugin (MCP + Skills)**
 
-> [!NOTE]  
+> [!NOTE]
 > If you already had Chrome DevTools MCP installed previously for Claude Code, make sure to remove it first from your installation and configuration files.
 
 To install Chrome DevTools MCP with skills, add the marketplace registry in Claude Code:
@@ -200,7 +206,7 @@ startup_timeout_ms = 20_000
 
 <details>
   <summary>Command Code</summary>
-  
+
 Use the Command Code CLI to add the Chrome DevTools MCP server (<a href="https://commandcode.ai/docs/mcp">MCP guide</a>):
 
 ```bash
@@ -235,6 +241,22 @@ Configure the following fields and press `CTRL+S` to save the configuration:
 <details>
   <summary>Copilot / VS Code</summary>
 
+**Install as a Plugin (Recommended)**
+
+The easiest way to get up and running is to install `chrome-devtools-mcp` as an agent plugin.
+This bundles the **MCP server** and all **skills** together, so your agent gets both the tools
+and the expert guidance it needs to use them effectively.
+
+1.  Open the **Command Palette** (`Cmd+Shift+P` on macOS or `Ctrl+Shift+P` on Windows/Linux).
+2.  Search for and run the **Chat: Install Plugin From Source** command.
+3.  Paste in our repository URL: `https://github.com/ChromeDevTools/chrome-devtools-mcp`
+
+That's it! Your agent is now supercharged with Chrome DevTools capabilities.
+
+---
+
+**Install as an MCP Server (MCP only)**
+
 **Click the button to install:**
 
 [<img src="https://img.shields.io/badge/VS_Code-VS_Code?style=flat-square&label=Install%20Server&color=0098FF" alt="Install in VS Code">](https://vscode.dev/redirect/mcp/install?name=io.github.ChromeDevTools%2Fchrome-devtools-mcp&config=%7B%22command%22%3A%22npx%22%2C%22args%22%3A%5B%22-y%22%2C%22chrome-devtools-mcp%22%5D%2C%22env%22%3A%7B%7D%7D)
@@ -243,8 +265,7 @@ Configure the following fields and press `CTRL+S` to save the configuration:
 
 **Or install manually:**
 
-Follow the MCP install <a href="https://code.visualstudio.com/docs/copilot/chat/mcp-servers#_add-an-mcp-server">guide</a>,
-with the standard config from above. You can also install the Chrome DevTools MCP server using the VS Code CLI:
+Follow the VS Code [MCP configuration guide](https://code.visualstudio.com/docs/copilot/chat/mcp-servers#_add-an-mcp-server) using the standard config from above, or use the CLI:
 
 For macOS and Linux:
 
@@ -353,6 +374,21 @@ Once connected, the Chrome DevTools MCP tools will be available in StudioAssist.
 
 </details>
 
+<details>
+  <summary>Mistral Vibe</summary>
+
+Add in ~/.vibe/config.toml:
+
+```toml
+[[mcp_servers]]
+name = "chrome-devtools"
+transport = "stdio"
+command = "npx"
+args = ["chrome-devtools-mcp@latest"]
+```
+
+</details>
+
 <details>
   <summary>OpenCode</summary>
 
@@ -402,10 +438,11 @@ qodercli mcp add -s user chrome-devtools -- npx chrome-devtools-mcp@latest
 
 <details>
   <summary>Visual Studio</summary>
-  
-  **Click the button to install:**
-  
-  [<img src="https://img.shields.io/badge/Visual_Studio-Install-C16FDE?logo=visualstudio&logoColor=white" alt="Install in Visual Studio">](https://vs-open.link/mcp-install?%7B%22name%22%3A%22chrome-devtools%22%2C%22command%22%3A%22npx%22%2C%22args%22%3A%5B%22chrome-devtools-mcp%40latest%22%5D%7D)
+
+**Click the button to install:**
+
+[<img src="https://img.shields.io/badge/Visual_Studio-Install-C16FDE?logo=visualstudio&logoColor=white" alt="Install in Visual Studio">](https://vs-open.link/mcp-install?%7B%22name%22%3A%22chrome-devtools%22%2C%22command%22%3A%22npx%22%2C%22args%22%3A%5B%22chrome-devtools-mcp%40latest%22%5D%7D)
+
 </details>
 
 <details>
@@ -431,7 +468,7 @@ Check the performance of https://developers.chrome.com
 
 Your MCP client should open the browser and record a performance trace.
 
-> [!NOTE]  
+> [!NOTE]
 > The MCP server will start the browser automatically once the MCP client uses a tool that requires a running browser instance. Connecting to the Chrome DevTools MCP server on its own will not automatically start the browser.
 
 ## Tools
@@ -460,11 +497,10 @@ If you run into any issues, checkout our [troubleshooting guide](./docs/troubles
 - **Emulation** (2 tools)
   - [`emulate`](docs/tool-reference.md#emulate)
   - [`resize_page`](docs/tool-reference.md#resize_page)
-- **Performance** (4 tools)
+- **Performance** (3 tools)
   - [`performance_analyze_insight`](docs/tool-reference.md#performance_analyze_insight)
   - [`performance_start_trace`](docs/tool-reference.md#performance_start_trace)
   - [`performance_stop_trace`](docs/tool-reference.md#performance_stop_trace)
-  - [`take_memory_snapshot`](docs/tool-reference.md#take_memory_snapshot)
 - **Network** (2 tools)
   - [`get_network_request`](docs/tool-reference.md#get_network_request)
   - [`list_network_requests`](docs/tool-reference.md#list_network_requests)
@@ -475,6 +511,14 @@ If you run into any issues, checkout our [troubleshooting guide](./docs/troubles
   - [`list_console_messages`](docs/tool-reference.md#list_console_messages)
   - [`take_screenshot`](docs/tool-reference.md#take_screenshot)
   - [`take_snapshot`](docs/tool-reference.md#take_snapshot)
+- **Extensions** (5 tools)
+  - [`install_extension`](docs/tool-reference.md#install_extension)
+  - [`list_extensions`](docs/tool-reference.md#list_extensions)
+  - [`reload_extension`](docs/tool-reference.md#reload_extension)
+  - [`trigger_extension_action`](docs/tool-reference.md#trigger_extension_action)
+  - [`uninstall_extension`](docs/tool-reference.md#uninstall_extension)
+- **Memory** (1 tools)
+  - [`take_memory_snapshot`](docs/tool-reference.md#take_memory_snapshot)
 
 <!-- END AUTO GENERATED TOOLS -->
 
@@ -485,7 +529,7 @@ The Chrome DevTools MCP server supports the following configuration option:
 <!-- BEGIN AUTO GENERATED OPTIONS -->
 
 - **`--autoConnect`/ `--auto-connect`**
-  If specified, automatically connects to a browser (Chrome 144+) running locally from the user data directory identified by the channel param (default channel is stable). Requires the remoted debugging server to be started in the Chrome instance via chrome://inspect/#remote-debugging.
+  If specified, automatically connects to a browser (Chrome 144+) running locally from the user data directory identified by the channel param (default channel is stable). Requires the remote debugging server to be started in the Chrome instance via chrome://inspect/#remote-debugging.
   - **Type:** boolean
   - **Default:** `false`
 
@@ -521,7 +565,7 @@ The Chrome DevTools MCP server supports the following configuration option:
 - **`--channel`**
   Specify a different Chrome channel that should be used. The default is the stable channel version.
   - **Type:** string
-  - **Choices:** `stable`, `canary`, `beta`, `dev`
+  - **Choices:** `canary`, `dev`, `beta`, `stable`
 
 - **`--logFile`/ `--log-file`**
   Path to a file to write debug logs to. Set the env variable `DEBUG` to `*` to enable verbose logs. Useful for submitting bug reports.
@@ -539,10 +583,22 @@ The Chrome DevTools MCP server supports the following configuration option:
   If enabled, ignores errors relative to self-signed and expired certificates. Use with caution.
   - **Type:** boolean
 
+- **`--experimentalVision`/ `--experimental-vision`**
+  Whether to enable coordinate-based tools such as click_at(x,y). Usually requires a computer-use model able to produce accurate coordinates by looking at screenshots.
+  - **Type:** boolean
+
 - **`--experimentalScreencast`/ `--experimental-screencast`**
   Exposes experimental screencast tools (requires ffmpeg). Install ffmpeg https://www.ffmpeg.org/download.html and ensure it is available in the MCP server PATH.
   - **Type:** boolean
 
+- **`--experimentalFfmpegPath`/ `--experimental-ffmpeg-path`**
+  Path to ffmpeg executable for screencast recording.
+  - **Type:** string
+
+- **`--experimentalWebmcp`/ `--experimental-webmcp`**
+  Set to true to enable debugging WebMCP tools. Requires Chrome 149+ with the following flags: `--enable-features=WebMCPTesting,DevToolsWebMCPSupport`
+  - **Type:** boolean
+
 - **`--chromeArg`/ `--chrome-arg`**
   Additional arguments for Chrome. Only applies when Chrome is launched by chrome-devtools-mcp.
   - **Type:** array
@@ -566,13 +622,18 @@ The Chrome DevTools MCP server supports the following configuration option:
   - **Type:** boolean
   - **Default:** `true`
 
+- **`--categoryExtensions`/ `--category-extensions`**
+  Set to true to include tools related to extensions. Note: This feature is currently only supported with a pipe connection. autoConnect, browserUrl, and wsEndpoint are not supported with this feature until 149 will be released.
+  - **Type:** boolean
+  - **Default:** `false`
+
 - **`--performanceCrux`/ `--performance-crux`**
   Set to false to disable sending URLs from performance traces to CrUX API to get field performance data.
   - **Type:** boolean
   - **Default:** `true`
 
 - **`--usageStatistics`/ `--usage-statistics`**
-  Set to false to opt-out of usage statistics collection. Google collects usage data to improve the tool, handled under the Google Privacy Policy (https://policies.google.com/privacy). This is independent from Chrome browser metrics. Disabled if CHROME_DEVTOOLS_MCP_NO_USAGE_STATISTICS or CI env variables are set.
+  Set to false to opt-out of usage statistics collection. Google collects usage data to improve the tool, handled under the Google Privacy Policy (https://policies.google.com/privacy). This is independent from Chrome browser metrics. Disabled if `CHROME_DEVTOOLS_MCP_NO_USAGE_STATISTICS` or `CI` env variables are set.
   - **Type:** boolean
   - **Default:** `true`
 
@@ -580,6 +641,11 @@ The Chrome DevTools MCP server supports the following configuration option:
   Exposes a "slim" set of 3 tools covering navigation, script execution and screenshots only. Useful for basic browser tasks.
   - **Type:** boolean
 
+- **`--redactNetworkHeaders`/ `--redact-network-headers`**
+  If true, redacts some of the network headers considered senstive before returning to the client.
+  - **Type:** boolean
+  - **Default:** `false`
+
 <!-- END AUTO GENERATED OPTIONS -->
 
 Pass them via the `args` property in the JSON configuration. For example:
@@ -686,7 +752,7 @@ Make sure your browser is running. Open gemini-cli and run the following prompt:
 Check the performance of https://developers.chrome.com
 ```
 
-> [!NOTE]  
+> [!NOTE]
 > The <code>autoConnect</code> option requires the user to start Chrome. If the user has multiple active profiles, the MCP server will connect to the default profile (as determined by Chrome). The MCP server has access to all open windows for the selected profile.
 
 The Chrome DevTools MCP server will try to connect to your running Chrome
@@ -722,7 +788,7 @@ Add the `--browser-url` option to your MCP client configuration. The value of th
 
 **Step 2: Start the Chrome browser**
 
-> [!WARNING]  
+> [!WARNING]
 > Enabling the remote debugging port opens up a debugging port on the running browser instance. Any application on your machine can connect to this port and control the browser. Make sure that you are not browsing any sensitive websites while the debugging port is open.
 
 Start the Chrome browser with the remote debugging port enabled. Make sure to close any running Chrome instances before starting a new one with the debugging port enabled. The port number you choose must be the same as the one you specified in the `--browser-url` option in your MCP client configuration.

package/build/src/HeapSnapshotManager.js

@@ -0,0 +1,94 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import fsSync from 'node:fs';
+import path from 'node:path';
+import { DevTools } from './third_party/index.js';
+import { createIdGenerator, stableIdSymbol, } from './utils/id.js';
+export class HeapSnapshotManager {
+    #snapshots = new Map();
+    async getSnapshot(filePath) {
+        const absolutePath = path.resolve(filePath);
+        const cached = this.#snapshots.get(absolutePath);
+        if (cached) {
+            return cached.snapshot;
+        }
+        const { snapshot, worker } = await this.#loadSnapshot(absolutePath);
+        this.#snapshots.set(absolutePath, {
+            snapshot,
+            worker,
+            uidToClassKey: new Map(),
+            classKeyToUid: new Map(),
+            idGenerator: createIdGenerator(),
+        });
+        return snapshot;
+    }
+    async getAggregates(filePath) {
+        const snapshot = await this.getSnapshot(filePath);
+        const filter = new DevTools.HeapSnapshotModel.HeapSnapshotModel.NodeFilter();
+        const aggregates = await snapshot.aggregatesWithFilter(filter);
+        for (const key of Object.keys(aggregates)) {
+            const uid = await this.getOrCreateUidForClassKey(filePath, key);
+            const aggregate = aggregates[key];
+            if (aggregate) {
+                aggregate[stableIdSymbol] = uid;
+            }
+        }
+        return aggregates;
+    }
+    async getStats(filePath) {
+        const snapshot = await this.getSnapshot(filePath);
+        return await snapshot.getStatistics();
+    }
+    async getStaticData(filePath) {
+        const snapshot = await this.getSnapshot(filePath);
+        return snapshot.staticData;
+    }
+    async getOrCreateUidForClassKey(filePath, classKey) {
+        const cached = this.#getCachedSnapshot(filePath);
+        let uid = cached.classKeyToUid.get(classKey);
+        if (!uid) {
+            uid = cached.idGenerator();
+            cached.classKeyToUid.set(classKey, uid);
+            cached.uidToClassKey.set(uid, classKey);
+        }
+        return uid;
+    }
+    #getCachedSnapshot(filePath) {
+        const absolutePath = path.resolve(filePath);
+        const cached = this.#snapshots.get(absolutePath);
+        if (!cached) {
+            throw new Error(`Snapshot not loaded for ${filePath}`);
+        }
+        return cached;
+    }
+    async #loadSnapshot(absolutePath) {
+        const workerProxy = new DevTools.HeapSnapshotModel.HeapSnapshotProxy.HeapSnapshotWorkerProxy(() => {
+            /* noop */
+        }, import.meta.resolve('./third_party/devtools-heap-snapshot-worker.js'));
+        const { promise: snapshotPromise, resolve: resolveSnapshot } = Promise.withResolvers();
+        const loaderProxy = workerProxy.createLoader(1, snapshotProxy => {
+            resolveSnapshot(snapshotProxy);
+        });
+        const fileStream = fsSync.createReadStream(absolutePath, {
+            encoding: 'utf-8',
+            highWaterMark: 1024 * 1024,
+        });
+        for await (const chunk of fileStream) {
+            await loaderProxy.write(chunk);
+        }
+        await loaderProxy.close();
+        const snapshot = await snapshotPromise;
+        return { snapshot, worker: workerProxy };
+    }
+    dispose(filePath) {
+        const absolutePath = path.resolve(filePath);
+        const cached = this.#snapshots.get(absolutePath);
+        if (cached) {
+            cached.worker.dispose();
+            this.#snapshots.delete(absolutePath);
+        }
+    }
+}

package/build/src/McpContext.js

@@ -6,31 +6,17 @@
 import fs from 'node:fs/promises';
 import path from 'node:path';
 import { UniverseManager } from './DevtoolsUtils.js';
+import { HeapSnapshotManager } from './HeapSnapshotManager.js';
 import { McpPage } from './McpPage.js';
 import { NetworkCollector, ConsoleCollector, } from './PageCollector.js';
 import { Locator } from './third_party/index.js';
 import { PredefinedNetworkConditions } from './third_party/index.js';
 import { listPages } from './tools/pages.js';
 import { CLOSE_PAGE_ERROR } from './tools/ToolDefinition.js';
-import { ExtensionRegistry, } from './utils/ExtensionRegistry.js';
-import { saveTemporaryFile } from './utils/files.js';
-import { WaitForHelper } from './WaitForHelper.js';
+import { ensureExtension, saveTemporaryFile } from './utils/files.js';
+import { getNetworkMultiplierFromString } from './WaitForHelper.js';
 const DEFAULT_TIMEOUT = 5_000;
 const NAVIGATION_TIMEOUT = 10_000;
-function getNetworkMultiplierFromString(condition) {
-    const puppeteerCondition = condition;
-    switch (puppeteerCondition) {
-        case 'Fast 4G':
-            return 1;
-        case 'Slow 4G':
-            return 2.5;
-        case 'Fast 3G':
-            return 5;
-        case 'Slow 3G':
-            return 10;
-    }
-    return 1;
-}
 export class McpContext {
     browser;
     logger;
@@ -45,18 +31,16 @@ export class McpContext {
     #networkCollector;
     #consoleCollector;
     #devtoolsUniverseManager;
-    #extensionRegistry = new ExtensionRegistry();
     #isRunningTrace = false;
     #screenRecorderData = null;
-    #inPageTools;
     #nextPageId = 1;
     #extensionPages = new WeakMap();
     #extensionServiceWorkerMap = new WeakMap();
     #nextExtensionServiceWorkerId = 1;
-    #nextSnapshotId = 1;
     #traceResults = [];
     #locatorClass;
     #options;
+    #heapSnapshotManager = new HeapSnapshotManager();
     constructor(browser, logger, options, locatorClass) {
         this.browser = browser;
         this.logger = logger;
@@ -71,7 +55,7 @@ export class McpContext {
                 uncaughtError: event => {
                     collect(event);
                 },
-                issue: event => {
+                devtoolsAggregatedIssue: event => {
                     collect(event);
                 },
             };
@@ -120,29 +104,6 @@ export class McpContext {
         }
         return this.#networkCollector.getIdForResource(request);
     }
-    resolveCdpElementId(page, cdpBackendNodeId) {
-        if (!cdpBackendNodeId) {
-            this.logger('no cdpBackendNodeId');
-            return;
-        }
-        const snapshot = page.textSnapshot;
-        if (!snapshot) {
-            this.logger('no text snapshot');
-            return;
-        }
-        // TODO: index by backendNodeId instead.
-        const queue = [snapshot.root];
-        while (queue.length) {
-            const current = queue.pop();
-            if (current.backendNodeId === cdpBackendNodeId) {
-                return current.id;
-            }
-            for (const child of current.children) {
-                queue.push(child);
-            }
-        }
-        return;
-    }
     getNetworkRequests(page, includePreservedRequests) {
         return this.#networkCollector.getData(page.pptrPage, includePreservedRequests);
     }
@@ -329,12 +290,6 @@ export class McpContext {
         this.#selectedPage = newPage;
         this.#updateSelectedPageTimeouts();
     }
-    setInPageTools(toolGroup) {
-        this.#inPageTools = toolGroup;
-    }
-    getInPageTools() {
-        return this.#inPageTools;
-    }
     #updateSelectedPageTimeouts() {
         const page = this.#getSelectedMcpPage();
         // For waiters 5sec timeout should be sufficient.
@@ -507,113 +462,12 @@ export class McpContext {
     getIsolatedContextName(page) {
         return this.#mcpPages.get(page)?.isolatedContextName;
     }
-    getDevToolsPage(page) {
-        return this.#mcpPages.get(page)?.devToolsPage;
-    }
-    async getDevToolsData(page) {
-        try {
-            this.logger('Getting DevTools UI data');
-            const devtoolsPage = this.getDevToolsPage(page.pptrPage);
-            if (!devtoolsPage) {
-                this.logger('No DevTools page detected');
-                return {};
-            }
-            const { cdpRequestId, cdpBackendNodeId } = await devtoolsPage.evaluate(async () => {
-                // @ts-expect-error no types
-                const UI = await import('/bundled/ui/legacy/legacy.js');
-                // @ts-expect-error no types
-                const SDK = await import('/bundled/core/sdk/sdk.js');
-                const request = UI.Context.Context.instance().flavor(SDK.NetworkRequest.NetworkRequest);
-                const node = UI.Context.Context.instance().flavor(SDK.DOMModel.DOMNode);
-                return {
-                    cdpRequestId: request?.requestId(),
-                    cdpBackendNodeId: node?.backendNodeId(),
-                };
-            });
-            return { cdpBackendNodeId, cdpRequestId };
-        }
-        catch (err) {
-            this.logger('error getting devtools data', err);
-        }
-        return {};
-    }
-    /**
-     * Creates a text snapshot of a page.
-     */
-    async createTextSnapshot(page, verbose = false, devtoolsData = undefined) {
-        const rootNode = await page.pptrPage.accessibility.snapshot({
-            includeIframes: true,
-            interestingOnly: !verbose,
-        });
-        if (!rootNode) {
-            return;
-        }
-        const { uniqueBackendNodeIdToMcpId } = page;
-        const snapshotId = this.#nextSnapshotId++;
-        // Iterate through the whole accessibility node tree and assign node ids that
-        // will be used for the tree serialization and mapping ids back to nodes.
-        let idCounter = 0;
-        const idToNode = new Map();
-        const seenUniqueIds = new Set();
-        const assignIds = (node) => {
-            let id = '';
-            // @ts-expect-error untyped loaderId & backendNodeId.
-            const uniqueBackendId = `${node.loaderId}_${node.backendNodeId}`;
-            if (uniqueBackendNodeIdToMcpId.has(uniqueBackendId)) {
-                // Re-use MCP exposed ID if the uniqueId is the same.
-                id = uniqueBackendNodeIdToMcpId.get(uniqueBackendId);
-            }
-            else {
-                // Only generate a new ID if we have not seen the node before.
-                id = `${snapshotId}_${idCounter++}`;
-                uniqueBackendNodeIdToMcpId.set(uniqueBackendId, id);
-            }
-            seenUniqueIds.add(uniqueBackendId);
-            const nodeWithId = {
-                ...node,
-                id,
-                children: node.children
-                    ? node.children.map(child => assignIds(child))
-                    : [],
-            };
-            // The AXNode for an option doesn't contain its `value`.
-            // Therefore, set text content of the option as value.
-            if (node.role === 'option') {
-                const optionText = node.name;
-                if (optionText) {
-                    nodeWithId.value = optionText.toString();
-                }
-            }
-            idToNode.set(nodeWithId.id, nodeWithId);
-            return nodeWithId;
-        };
-        const rootNodeWithId = assignIds(rootNode);
-        const snapshot = {
-            root: rootNodeWithId,
-            snapshotId: String(snapshotId),
-            idToNode,
-            hasSelectedElement: false,
-            verbose,
-        };
-        page.textSnapshot = snapshot;
-        const data = devtoolsData ?? (await this.getDevToolsData(page));
-        if (data?.cdpBackendNodeId) {
-            snapshot.hasSelectedElement = true;
-            snapshot.selectedElementUid = this.resolveCdpElementId(page, data?.cdpBackendNodeId);
-        }
-        // Clean up unique IDs that we did not see anymore.
-        for (const key of uniqueBackendNodeIdToMcpId.keys()) {
-            if (!seenUniqueIds.has(key)) {
-                uniqueBackendNodeIdToMcpId.delete(key);
-            }
-        }
-    }
     async saveTemporaryFile(data, filename) {
         return await saveTemporaryFile(data, filename);
     }
-    async saveFile(data, filename) {
+    async saveFile(data, clientProvidedFilePath, extension) {
         try {
-            const filePath = path.resolve(filename);
+            const filePath = ensureExtension(path.resolve(clientProvidedFilePath), extension);
             await fs.mkdir(path.dirname(filePath), { recursive: true });
             await fs.writeFile(filePath, data);
             return { filename: filePath };
@@ -631,16 +485,6 @@ export class McpContext {
     recordedTraces() {
         return this.#traceResults;
     }
-    getWaitForHelper(page, cpuMultiplier, networkMultiplier) {
-        return new WaitForHelper(page, cpuMultiplier, networkMultiplier);
-    }
-    waitForEventsAfterAction(action, options) {
-        const page = this.#getSelectedMcpPage();
-        const cpuMultiplier = page.cpuThrottlingRate;
-        const networkMultiplier = getNetworkMultiplierFromString(page.networkConditions);
-        const waitForHelper = this.getWaitForHelper(page.pptrPage, cpuMultiplier, networkMultiplier);
-        return waitForHelper.waitForEventsAfterAction(action, options);
-    }
     getNetworkRequestStableId(request) {
         return this.#networkCollector.getIdForResource(request);
     }
@@ -675,33 +519,34 @@ export class McpContext {
     }
     async installExtension(extensionPath) {
         const id = await this.browser.installExtension(extensionPath);
-        await this.#extensionRegistry.registerExtension(id, extensionPath);
         return id;
     }
     async uninstallExtension(id) {
         await this.browser.uninstallExtension(id);
-        this.#extensionRegistry.remove(id);
     }
     async triggerExtensionAction(id) {
-        const page = this.getSelectedPptrPage();
-        // @ts-expect-error internal puppeteer api is needed since we don't have a way to get
-        // a tab id at the moment
-        const theTarget = page._tabId;
-        const session = await this.browser.target().createCDPSession();
-        try {
-            await session.send('Extensions.triggerAction', {
-                id,
-                targetId: theTarget,
-            });
-        }
-        finally {
-            await session.detach();
+        const extensions = await this.browser.extensions();
+        const extension = extensions.get(id);
+        if (!extension) {
+            throw new Error(`Extension with ID ${id} not found.`);
         }
+        const page = this.getSelectedPptrPage();
+        await extension.triggerAction(page);
     }
     listExtensions() {
-        return this.#extensionRegistry.list();
+        return this.browser.extensions();
+    }
+    async getExtension(id) {
+        const pptrExtensions = await this.browser.extensions();
+        return pptrExtensions.get(id);
+    }
+    async getHeapSnapshotAggregates(filePath) {
+        return await this.#heapSnapshotManager.getAggregates(filePath);
+    }
+    async getHeapSnapshotStats(filePath) {
+        return await this.#heapSnapshotManager.getStats(filePath);
     }
-    getExtension(id) {
-        return this.#extensionRegistry.getById(id);
+    async getHeapSnapshotStaticData(filePath) {
+        return await this.#heapSnapshotManager.getStaticData(filePath);
     }
 }