@playwright/mcp 0.0.36 → 0.0.37-alpha-2025-09-09

package/lib/extension/cdpRelay.js

@@ -1,358 +0,0 @@
-/**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-/**
- * WebSocket server that bridges Playwright MCP and Chrome Extension
- *
- * Endpoints:
- * - /cdp/guid - Full CDP interface for Playwright MCP
- * - /extension/guid - Extension connection for chrome.debugger forwarding
- */
-import { spawn } from 'child_process';
-import debug from 'debug';
-import { WebSocket, WebSocketServer } from 'ws';
-import { httpAddressToString } from '../mcp/http.js';
-import { logUnhandledError } from '../utils/log.js';
-import { ManualPromise } from '../mcp/manualPromise.js';
-import * as protocol from './protocol.js';
-// @ts-ignore
-const { registry } = await import('playwright-core/lib/server/registry/index');
-const debugLogger = debug('pw:mcp:relay');
-export class CDPRelayServer {
-    _wsHost;
-    _browserChannel;
-    _userDataDir;
-    _executablePath;
-    _cdpPath;
-    _extensionPath;
-    _wss;
-    _playwrightConnection = null;
-    _extensionConnection = null;
-    _connectedTabInfo;
-    _nextSessionId = 1;
-    _extensionConnectionPromise;
-    constructor(server, browserChannel, userDataDir, executablePath) {
-        this._wsHost = httpAddressToString(server.address()).replace(/^http/, 'ws');
-        this._browserChannel = browserChannel;
-        this._userDataDir = userDataDir;
-        this._executablePath = executablePath;
-        const uuid = crypto.randomUUID();
-        this._cdpPath = `/cdp/${uuid}`;
-        this._extensionPath = `/extension/${uuid}`;
-        this._resetExtensionConnection();
-        this._wss = new WebSocketServer({ server });
-        this._wss.on('connection', this._onConnection.bind(this));
-    }
-    cdpEndpoint() {
-        return `${this._wsHost}${this._cdpPath}`;
-    }
-    extensionEndpoint() {
-        return `${this._wsHost}${this._extensionPath}`;
-    }
-    async ensureExtensionConnectionForMCPContext(clientInfo, abortSignal, toolName) {
-        debugLogger('Ensuring extension connection for MCP context');
-        if (this._extensionConnection)
-            return;
-        this._connectBrowser(clientInfo, toolName);
-        debugLogger('Waiting for incoming extension connection');
-        await Promise.race([
-            this._extensionConnectionPromise,
-            new Promise((_, reject) => setTimeout(() => {
-                reject(new Error(`Extension connection timeout. Make sure the "Playwright MCP Bridge" extension is installed. See https://github.com/microsoft/playwright-mcp/blob/main/extension/README.md for installation instructions.`));
-            }, process.env.PWMCP_TEST_CONNECTION_TIMEOUT ? parseInt(process.env.PWMCP_TEST_CONNECTION_TIMEOUT, 10) : 5_000)),
-            new Promise((_, reject) => abortSignal.addEventListener('abort', reject))
-        ]);
-        debugLogger('Extension connection established');
-    }
-    _connectBrowser(clientInfo, toolName) {
-        const mcpRelayEndpoint = `${this._wsHost}${this._extensionPath}`;
-        // Need to specify "key" in the manifest.json to make the id stable when loading from file.
-        const url = new URL('chrome-extension://jakfalbnbhgkpmoaakfflhflbfpkailf/connect.html');
-        url.searchParams.set('mcpRelayUrl', mcpRelayEndpoint);
-        const client = {
-            name: clientInfo.name,
-            version: clientInfo.version,
-        };
-        url.searchParams.set('client', JSON.stringify(client));
-        url.searchParams.set('protocolVersion', process.env.PWMCP_TEST_PROTOCOL_VERSION ?? protocol.VERSION.toString());
-        if (toolName)
-            url.searchParams.set('newTab', String(toolName === 'browser_navigate'));
-        const href = url.toString();
-        let executablePath = this._executablePath;
-        if (!executablePath) {
-            const executableInfo = registry.findExecutable(this._browserChannel);
-            if (!executableInfo)
-                throw new Error(`Unsupported channel: "${this._browserChannel}"`);
-            executablePath = executableInfo.executablePath();
-            if (!executablePath)
-                throw new Error(`"${this._browserChannel}" executable not found. Make sure it is installed at a standard location.`);
-        }
-        const args = [];
-        if (this._userDataDir)
-            args.push(`--user-data-dir=${this._userDataDir}`);
-        args.push(href);
-        spawn(executablePath, args, {
-            windowsHide: true,
-            detached: true,
-            shell: false,
-            stdio: 'ignore',
-        });
-    }
-    stop() {
-        this.closeConnections('Server stopped');
-        this._wss.close();
-    }
-    closeConnections(reason) {
-        this._closePlaywrightConnection(reason);
-        this._closeExtensionConnection(reason);
-    }
-    _onConnection(ws, request) {
-        const url = new URL(`http://localhost${request.url}`);
-        debugLogger(`New connection to ${url.pathname}`);
-        if (url.pathname === this._cdpPath) {
-            this._handlePlaywrightConnection(ws);
-        }
-        else if (url.pathname === this._extensionPath) {
-            this._handleExtensionConnection(ws);
-        }
-        else {
-            debugLogger(`Invalid path: ${url.pathname}`);
-            ws.close(4004, 'Invalid path');
-        }
-    }
-    _handlePlaywrightConnection(ws) {
-        if (this._playwrightConnection) {
-            debugLogger('Rejecting second Playwright connection');
-            ws.close(1000, 'Another CDP client already connected');
-            return;
-        }
-        this._playwrightConnection = ws;
-        ws.on('message', async (data) => {
-            try {
-                const message = JSON.parse(data.toString());
-                await this._handlePlaywrightMessage(message);
-            }
-            catch (error) {
-                debugLogger(`Error while handling Playwright message\n${data.toString()}\n`, error);
-            }
-        });
-        ws.on('close', () => {
-            if (this._playwrightConnection !== ws)
-                return;
-            this._playwrightConnection = null;
-            this._closeExtensionConnection('Playwright client disconnected');
-            debugLogger('Playwright WebSocket closed');
-        });
-        ws.on('error', error => {
-            debugLogger('Playwright WebSocket error:', error);
-        });
-        debugLogger('Playwright MCP connected');
-    }
-    _closeExtensionConnection(reason) {
-        this._extensionConnection?.close(reason);
-        this._extensionConnectionPromise.reject(new Error(reason));
-        this._resetExtensionConnection();
-    }
-    _resetExtensionConnection() {
-        this._connectedTabInfo = undefined;
-        this._extensionConnection = null;
-        this._extensionConnectionPromise = new ManualPromise();
-        void this._extensionConnectionPromise.catch(logUnhandledError);
-    }
-    _closePlaywrightConnection(reason) {
-        if (this._playwrightConnection?.readyState === WebSocket.OPEN)
-            this._playwrightConnection.close(1000, reason);
-        this._playwrightConnection = null;
-    }
-    _handleExtensionConnection(ws) {
-        if (this._extensionConnection) {
-            ws.close(1000, 'Another extension connection already established');
-            return;
-        }
-        this._extensionConnection = new ExtensionConnection(ws);
-        this._extensionConnection.onclose = (c, reason) => {
-            debugLogger('Extension WebSocket closed:', reason, c === this._extensionConnection);
-            if (this._extensionConnection !== c)
-                return;
-            this._resetExtensionConnection();
-            this._closePlaywrightConnection(`Extension disconnected: ${reason}`);
-        };
-        this._extensionConnection.onmessage = this._handleExtensionMessage.bind(this);
-        this._extensionConnectionPromise.resolve();
-    }
-    _handleExtensionMessage(method, params) {
-        switch (method) {
-            case 'forwardCDPEvent':
-                const sessionId = params.sessionId || this._connectedTabInfo?.sessionId;
-                this._sendToPlaywright({
-                    sessionId,
-                    method: params.method,
-                    params: params.params
-                });
-                break;
-        }
-    }
-    async _handlePlaywrightMessage(message) {
-        debugLogger('← Playwright:', `${message.method} (id=${message.id})`);
-        const { id, sessionId, method, params } = message;
-        try {
-            const result = await this._handleCDPCommand(method, params, sessionId);
-            this._sendToPlaywright({ id, sessionId, result });
-        }
-        catch (e) {
-            debugLogger('Error in the extension:', e);
-            this._sendToPlaywright({
-                id,
-                sessionId,
-                error: { message: e.message }
-            });
-        }
-    }
-    async _handleCDPCommand(method, params, sessionId) {
-        switch (method) {
-            case 'Browser.getVersion': {
-                return {
-                    protocolVersion: '1.3',
-                    product: 'Chrome/Extension-Bridge',
-                    userAgent: 'CDP-Bridge-Server/1.0.0',
-                };
-            }
-            case 'Browser.setDownloadBehavior': {
-                return {};
-            }
-            case 'Target.setAutoAttach': {
-                // Forward child session handling.
-                if (sessionId)
-                    break;
-                // Simulate auto-attach behavior with real target info
-                const { targetInfo } = await this._extensionConnection.send('attachToTab', {});
-                this._connectedTabInfo = {
-                    targetInfo,
-                    sessionId: `pw-tab-${this._nextSessionId++}`,
-                };
-                debugLogger('Simulating auto-attach');
-                this._sendToPlaywright({
-                    method: 'Target.attachedToTarget',
-                    params: {
-                        sessionId: this._connectedTabInfo.sessionId,
-                        targetInfo: {
-                            ...this._connectedTabInfo.targetInfo,
-                            attached: true,
-                        },
-                        waitingForDebugger: false
-                    }
-                });
-                return {};
-            }
-            case 'Target.getTargetInfo': {
-                return this._connectedTabInfo?.targetInfo;
-            }
-        }
-        return await this._forwardToExtension(method, params, sessionId);
-    }
-    async _forwardToExtension(method, params, sessionId) {
-        if (!this._extensionConnection)
-            throw new Error('Extension not connected');
-        // Top level sessionId is only passed between the relay and the client.
-        if (this._connectedTabInfo?.sessionId === sessionId)
-            sessionId = undefined;
-        return await this._extensionConnection.send('forwardCDPCommand', { sessionId, method, params });
-    }
-    _sendToPlaywright(message) {
-        debugLogger('→ Playwright:', `${message.method ?? `response(id=${message.id})`}`);
-        this._playwrightConnection?.send(JSON.stringify(message));
-    }
-}
-class ExtensionConnection {
-    _ws;
-    _callbacks = new Map();
-    _lastId = 0;
-    onmessage;
-    onclose;
-    constructor(ws) {
-        this._ws = ws;
-        this._ws.on('message', this._onMessage.bind(this));
-        this._ws.on('close', this._onClose.bind(this));
-        this._ws.on('error', this._onError.bind(this));
-    }
-    async send(method, params) {
-        if (this._ws.readyState !== WebSocket.OPEN)
-            throw new Error(`Unexpected WebSocket state: ${this._ws.readyState}`);
-        const id = ++this._lastId;
-        this._ws.send(JSON.stringify({ id, method, params }));
-        const error = new Error(`Protocol error: ${method}`);
-        return new Promise((resolve, reject) => {
-            this._callbacks.set(id, { resolve, reject, error });
-        });
-    }
-    close(message) {
-        debugLogger('closing extension connection:', message);
-        if (this._ws.readyState === WebSocket.OPEN)
-            this._ws.close(1000, message);
-    }
-    _onMessage(event) {
-        const eventData = event.toString();
-        let parsedJson;
-        try {
-            parsedJson = JSON.parse(eventData);
-        }
-        catch (e) {
-            debugLogger(`<closing ws> Closing websocket due to malformed JSON. eventData=${eventData} e=${e?.message}`);
-            this._ws.close();
-            return;
-        }
-        try {
-            this._handleParsedMessage(parsedJson);
-        }
-        catch (e) {
-            debugLogger(`<closing ws> Closing websocket due to failed onmessage callback. eventData=${eventData} e=${e?.message}`);
-            this._ws.close();
-        }
-    }
-    _handleParsedMessage(object) {
-        if (object.id && this._callbacks.has(object.id)) {
-            const callback = this._callbacks.get(object.id);
-            this._callbacks.delete(object.id);
-            if (object.error) {
-                const error = callback.error;
-                error.message = object.error;
-                callback.reject(error);
-            }
-            else {
-                callback.resolve(object.result);
-            }
-        }
-        else if (object.id) {
-            debugLogger('← Extension: unexpected response', object);
-        }
-        else {
-            this.onmessage?.(object.method, object.params);
-        }
-    }
-    _onClose(event) {
-        debugLogger(`<ws closed> code=${event.code} reason=${event.reason}`);
-        this._dispose();
-        this.onclose?.(this, event.reason);
-    }
-    _onError(event) {
-        debugLogger(`<ws error> message=${event.message} type=${event.type} target=${event.target}`);
-        this._dispose();
-    }
-    _dispose() {
-        for (const callback of this._callbacks.values())
-            callback.reject(new Error('WebSocket closed'));
-        this._callbacks.clear();
-    }
-}

package/lib/extension/extensionContextFactory.js

@@ -1,56 +0,0 @@
-/**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-import debug from 'debug';
-import * as playwright from 'playwright';
-import { startHttpServer } from '../mcp/http.js';
-import { CDPRelayServer } from './cdpRelay.js';
-const debugLogger = debug('pw:mcp:relay');
-export class ExtensionContextFactory {
-    _browserChannel;
-    _userDataDir;
-    _executablePath;
-    constructor(browserChannel, userDataDir, executablePath) {
-        this._browserChannel = browserChannel;
-        this._userDataDir = userDataDir;
-        this._executablePath = executablePath;
-    }
-    async createContext(clientInfo, abortSignal, toolName) {
-        const browser = await this._obtainBrowser(clientInfo, abortSignal, toolName);
-        return {
-            browserContext: browser.contexts()[0],
-            close: async () => {
-                debugLogger('close() called for browser context');
-                await browser.close();
-            }
-        };
-    }
-    async _obtainBrowser(clientInfo, abortSignal, toolName) {
-        const relay = await this._startRelay(abortSignal);
-        await relay.ensureExtensionConnectionForMCPContext(clientInfo, abortSignal, toolName);
-        return await playwright.chromium.connectOverCDP(relay.cdpEndpoint());
-    }
-    async _startRelay(abortSignal) {
-        const httpServer = await startHttpServer({});
-        if (abortSignal.aborted) {
-            httpServer.close();
-            throw new Error(abortSignal.reason);
-        }
-        const cdpRelayServer = new CDPRelayServer(httpServer, this._browserChannel, this._userDataDir, this._executablePath);
-        abortSignal.addEventListener('abort', () => cdpRelayServer.stop());
-        debugLogger(`CDP relay server started, extension endpoint: ${cdpRelayServer.extensionEndpoint()}.`);
-        return cdpRelayServer;
-    }
-}

package/lib/extension/protocol.js

@@ -1,18 +0,0 @@
-/**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-// Whenever the commands/events change, the version must be updated. The latest
-// extension version should be compatible with the old MCP clients.
-export const VERSION = 1;

package/lib/index.js

@@ -1,40 +0,0 @@
-/**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-import { BrowserServerBackend } from './browserServerBackend.js';
-import { resolveConfig } from './config.js';
-import { contextFactory } from './browserContextFactory.js';
-import * as mcpServer from './mcp/server.js';
-import { packageJSON } from './utils/package.js';
-export async function createConnection(userConfig = {}, contextGetter) {
-    const config = await resolveConfig(userConfig);
-    const factory = contextGetter ? new SimpleBrowserContextFactory(contextGetter) : contextFactory(config);
-    return mcpServer.createServer('Playwright', packageJSON.version, new BrowserServerBackend(config, factory), false);
-}
-class SimpleBrowserContextFactory {
-    name = 'custom';
-    description = 'Connect to a browser using a custom context getter';
-    _contextGetter;
-    constructor(contextGetter) {
-        this._contextGetter = contextGetter;
-    }
-    async createContext() {
-        const browserContext = await this._contextGetter();
-        return {
-            browserContext,
-            close: () => browserContext.close()
-        };
-    }
-}

package/lib/loop/loop.js

@@ -1,69 +0,0 @@
-/**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-import debug from 'debug';
-export async function runTask(delegate, client, task, oneShot = false) {
-    const { tools } = await client.listTools();
-    const taskContent = oneShot ? `Perform following task: ${task}.` : `Perform following task: ${task}. Once the task is complete, call the "done" tool.`;
-    const conversation = delegate.createConversation(taskContent, tools, oneShot);
-    for (let iteration = 0; iteration < 5; ++iteration) {
-        debug('history')('Making API call for iteration', iteration);
-        const toolCalls = await delegate.makeApiCall(conversation);
-        if (toolCalls.length === 0)
-            throw new Error('Call the "done" tool when the task is complete.');
-        const toolResults = [];
-        for (const toolCall of toolCalls) {
-            const doneResult = delegate.checkDoneToolCall(toolCall);
-            if (doneResult !== null)
-                return conversation.messages;
-            const { name, arguments: args, id } = toolCall;
-            try {
-                debug('tool')(name, args);
-                const response = await client.callTool({
-                    name,
-                    arguments: args,
-                });
-                const responseContent = (response.content || []);
-                debug('tool')(responseContent);
-                const text = responseContent.filter(part => part.type === 'text').map(part => part.text).join('\n');
-                toolResults.push({
-                    toolCallId: id,
-                    content: text,
-                });
-            }
-            catch (error) {
-                debug('tool')(error);
-                toolResults.push({
-                    toolCallId: id,
-                    content: `Error while executing tool "${name}": ${error instanceof Error ? error.message : String(error)}\n\nPlease try to recover and complete the task.`,
-                    isError: true,
-                });
-                // Skip remaining tool calls for this iteration
-                for (const remainingToolCall of toolCalls.slice(toolCalls.indexOf(toolCall) + 1)) {
-                    toolResults.push({
-                        toolCallId: remainingToolCall.id,
-                        content: `This tool call is skipped due to previous error.`,
-                        isError: true,
-                    });
-                }
-                break;
-            }
-        }
-        delegate.addToolResults(conversation, toolResults);
-        if (oneShot)
-            return conversation.messages;
-    }
-    throw new Error('Failed to perform step, max attempts reached');
-}