playwriter 0.0.3 → 0.0.5

package/dist/mcp-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../src/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAA;AAClE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAA;AAGhF,OAAO,IAAI,MAAM,WAAW,CAAA;AAC5B,OAAO,GAAG,MAAM,UAAU,CAAA;AAE1B,MAAM,UAAU,GAAG,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AAMrD,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,IAAc;IAIhD,MAAM,SAAS,GAAG,IAAI,oBAAoB,CAAC;QACvC,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,EAAE,GAAG,IAAI,CAAC;QAC3E,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC;QAC9C,MAAM,EAAE,MAAM;QACd,GAAG,EAAE;YACD,GAAG,OAAO,CAAC,GAAG;YACd,KAAK,EAAE,qBAAqB;YAC5B,YAAY,EAAE,GAAG;YACjB,eAAe,EAAE,GAAG;SACvB;KACJ,CAAC,CAAA;IAEF,OAAO;QACH,SAAS;QACT,MAAM,EAAE,SAAS,CAAC,MAAO;KAC5B,CAAA;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,OAAgC;IAKlE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;QACtB,IAAI,EAAE,OAAO,EAAE,UAAU,IAAI,MAAM;QACnC,OAAO,EAAE,OAAO;KACnB,CAAC,CAAA;IAEF,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,eAAe,CAAC,EAAE,CAAC,CAAA;IAEvD,IAAI,YAAY,GAAG,EAAE,CAAA;IACrB,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;QACxB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAE1B,YAAY,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAA;IACnC,CAAC,CAAC,CAAA;IAEF,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IAC/B,MAAM,MAAM,CAAC,IAAI,EAAE,CAAA;IAEnB,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;QACvB,IAAI,CAAC;YACD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAA;QACxB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAA;YAClD,+BAA+B;QACnC,CAAC;IACL,CAAC,CAAA;IAED,OAAO;QACH,MAAM;QACN,MAAM,EAAE,YAAY;QACpB,OAAO;KACV,CAAA;AACL,CAAC"}
+{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../src/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAA;AAClE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAA;AAGhF,OAAO,IAAI,MAAM,WAAW,CAAA;AAC5B,OAAO,GAAG,MAAM,UAAU,CAAA;AAE1B,MAAM,UAAU,GAAG,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AAMrD,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,IAAc;IAIlD,MAAM,SAAS,GAAG,IAAI,oBAAoB,CAAC;QACzC,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,EAAE,GAAG,IAAI,CAAC;QAC3E,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC;QAC9C,MAAM,EAAE,MAAM;QACd,GAAG,EAAE;YACH,GAAG,OAAO,CAAC,GAAG;YACd,KAAK,EAAE,qBAAqB;YAC5B,YAAY,EAAE,GAAG;YACjB,eAAe,EAAE,GAAG;SACrB;KACF,CAAC,CAAA;IAEF,OAAO;QACL,SAAS;QACT,MAAM,EAAE,SAAS,CAAC,MAAO;KAC1B,CAAA;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,OAAgC;IAKpE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;QACxB,IAAI,EAAE,OAAO,EAAE,UAAU,IAAI,MAAM;QACnC,OAAO,EAAE,OAAO;KACjB,CAAC,CAAA;IAEF,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,eAAe,CAAC,EAAE,CAAC,CAAA;IAEvD,IAAI,YAAY,GAAG,EAAE,CAAA;IACrB,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;QAC1B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAE1B,YAAY,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IAC/B,MAAM,MAAM,CAAC,IAAI,EAAE,CAAA;IAEnB,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;QACzB,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAA;QACtB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAA;YACpD,+BAA+B;QACjC,CAAC;IACH,CAAC,CAAA;IAED,OAAO;QACL,MAAM;QACN,MAAM,EAAE,YAAY;QACpB,OAAO;KACR,CAAA;AACH,CAAC"}

package/dist/mcp.js

@@ -1,491 +1,98 @@
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { z } from 'zod';
-import { chromium } from 'patchright-core';
+import { chromium } from 'playwright-core';
 import fs from 'node:fs';
 import path from 'node:path';
-import os from 'node:os';
-import { spawn } from 'child_process';
-import { getBrowserExecutablePath } from './browser-config.js';
+import { spawn } from 'node:child_process';
+import { createRequire } from 'node:module';
+import vm from 'node:vm';
+const require = createRequire(import.meta.url);
 const state = {
     isConnected: false,
     page: null,
     browser: null,
-    chromeProcess: null,
-    consoleLogs: new Map(),
-    networkRequests: new Map(),
+    context: null,
 };
-const CDP_PORT = 9922;
-// Check if CDP is available on the specified port
-async function isCDPAvailable() {
+const RELAY_PORT = 19988;
+async function isPortTaken(port) {
     try {
-        const response = await fetch(`http://127.0.0.1:${CDP_PORT}/json/version`);
+        const response = await fetch(`http://localhost:${port}/`);
         return response.ok;
     }
     catch {
         return false;
     }
 }
-// Launch Chrome with CDP enabled
-async function launchChromeWithCDP() {
-    const userDataDir = path.join(os.homedir(), '.playwriter');
-    if (!fs.existsSync(userDataDir)) {
-        fs.mkdirSync(userDataDir, { recursive: true });
-    }
-    const executablePath = getBrowserExecutablePath();
-    const chromeArgs = [
-        `--remote-debugging-port=${CDP_PORT}`,
-        `--user-data-dir=${userDataDir}`,
-        '--no-first-run',
-        '--no-default-browser-check',
-        '--disable-session-crashed-bubble',
-        '--disable-features=DevToolsDebuggingRestrictions',
-        '--disable-blink-features=AutomationControlled',
-        '--no-sandbox',
-        '--disable-web-security',
-        '--disable-infobars',
-        '--disable-translate',
-        '--disable-features=AutomationControlled', // disables --enable-automation
-        '--disable-background-timer-throttling',
-        '--disable-popup-blocking',
-        '--disable-backgrounding-occluded-windows',
-        '--disable-renderer-backgrounding',
-        '--disable-window-activation',
-        '--disable-focus-on-load',
-        '--no-startup-window',
-        '--window-position=0,0',
-        '--disable-site-isolation-trials',
-        '--disable-features=IsolateOrigins,site-per-process',
-    ];
-    const chromeProcess = spawn(executablePath, chromeArgs, {
+async function ensureRelayServer() {
+    const portTaken = await isPortTaken(RELAY_PORT);
+    if (portTaken) {
+        console.error('CDP relay server already running');
+        return;
+    }
+    console.error('Starting CDP relay server...');
+    const scriptPath = require.resolve('../dist/start-relay-server.js');
+    const serverProcess = spawn(process.execPath, [scriptPath], {
         detached: true,
         stdio: 'ignore',
     });
-    // Unref the process so it doesn't keep the parent process alive
-    chromeProcess.unref();
-    // Give Chrome time to start up
-    await new Promise((resolve) => setTimeout(resolve, 2000));
-    return chromeProcess;
+    serverProcess.unref();
+    // wait for extension to connect
+    await new Promise((resolve) => setTimeout(resolve, 1000));
+    console.error('CDP relay server started');
 }
-// Ensure connection to Chrome via CDP
 async function ensureConnection() {
     if (state.isConnected && state.browser && state.page) {
         return { browser: state.browser, page: state.page };
     }
-    // Check if CDP is already available
-    const cdpAvailable = await isCDPAvailable();
-    if (!cdpAvailable) {
-        // Launch Chrome with CDP
-        const chromeProcess = await launchChromeWithCDP();
-        state.chromeProcess = chromeProcess;
-    }
-    // Connect to Chrome via CDP
-    const browser = await chromium.connectOverCDP(`http://127.0.0.1:${CDP_PORT}`);
-    // Get the default context
+    await ensureRelayServer();
+    const cdpEndpoint = `ws://localhost:${RELAY_PORT}/cdp/${Date.now()}`;
+    const browser = await chromium.connectOverCDP(cdpEndpoint);
     const contexts = browser.contexts();
-    let context;
-    if (contexts.length > 0) {
-        context = contexts[0];
-    }
-    else {
-        context = await browser.newContext();
-    }
-    // Generate user agent and set it on context
-    const { default: UserAgent } = await import('user-agents');
-    const userAgent = new UserAgent({
-        platform: 'MacIntel',
-        deviceCategory: 'desktop',
-    });
-    // Get or create page
+    const context = contexts.length > 0 ? contexts[0] : await browser.newContext();
     const pages = context.pages();
-    let page;
-    if (pages.length > 0) {
-        page = pages[0];
-        // Set user agent on existing page
-        await page.setExtraHTTPHeaders({
-            'User-Agent': userAgent.toString(),
-        });
-    }
-    else {
-        page = await context.newPage();
-        // Set user agent on new page
-        await page.setExtraHTTPHeaders({
-            'User-Agent': userAgent.toString(),
-        });
-    }
-    // Set up event listeners if not already set
-    if (!state.isConnected) {
-        page.on('console', (msg) => {
-            // Get or create logs array for this page
-            let pageLogs = state.consoleLogs.get(page);
-            if (!pageLogs) {
-                pageLogs = [];
-                state.consoleLogs.set(page, pageLogs);
-            }
-            // Add new log
-            pageLogs.push({
-                type: msg.type(),
-                text: msg.text(),
-                timestamp: Date.now(),
-                location: msg.location(),
-            });
-            // Keep only last 1000 logs
-            if (pageLogs.length > 1000) {
-                pageLogs.shift();
-            }
-        });
-        // Clean up logs and network requests when page is closed
-        page.on('close', () => {
-            state.consoleLogs.delete(page);
-            state.networkRequests.delete(page);
-        });
-        page.on('request', (request) => {
-            const startTime = Date.now();
-            const entry = {
-                url: request.url(),
-                method: request.method(),
-                headers: request.headers(),
-                timestamp: startTime,
-            };
-            request
-                .response()
-                .then((response) => {
-                if (response) {
-                    entry.status = response.status();
-                    entry.duration = Date.now() - startTime;
-                    entry.size = response.headers()['content-length']
-                        ? parseInt(response.headers()['content-length'])
-                        : 0;
-                    // Get or create requests array for this page
-                    let pageRequests = state.networkRequests.get(page);
-                    if (!pageRequests) {
-                        pageRequests = [];
-                        state.networkRequests.set(page, pageRequests);
-                    }
-                    // Add new request
-                    pageRequests.push(entry);
-                    // Keep only last 1000 requests
-                    if (pageRequests.length > 1000) {
-                        pageRequests.shift();
-                    }
-                }
-            })
-                .catch(() => {
-                // Handle response errors silently
-            });
-        });
-    }
+    const page = pages.length > 0 ? pages[0] : await context.newPage();
     state.browser = browser;
     state.page = page;
+    state.context = context;
     state.isConnected = true;
     return { browser, page };
 }
-// Initialize MCP server
-const server = new McpServer({
-    name: 'playwriter',
-    title: 'Playwright MCP Server',
-    version: '1.0.0',
-});
-// Tool 1: New Page - Creates a new browser page
-server.tool('new_page', 'Create a new browser page in the shared Chrome instance', {}, async () => {
-    try {
-        const { browser, page } = await ensureConnection();
-        // Always create a new page
-        const context = browser.contexts()[0] || await browser.newContext();
-        const newPage = await context.newPage();
-        // Set user agent on new page
-        const { default: UserAgent } = await import('user-agents');
-        const userAgent = new UserAgent({
-            platform: 'MacIntel',
-            deviceCategory: 'desktop',
-        });
-        await newPage.setExtraHTTPHeaders({
-            'User-Agent': userAgent.toString()
-        });
-        // Update state to use the new page
-        state.page = newPage;
-        // Set up event listeners on the new page
-        newPage.on('console', (msg) => {
-            // Get or create logs array for this page
-            let pageLogs = state.consoleLogs.get(newPage);
-            if (!pageLogs) {
-                pageLogs = [];
-                state.consoleLogs.set(newPage, pageLogs);
-            }
-            // Add new log
-            pageLogs.push({
-                type: msg.type(),
-                text: msg.text(),
-                timestamp: Date.now(),
-                location: msg.location(),
-            });
-            // Keep only last 1000 logs
-            if (pageLogs.length > 1000) {
-                pageLogs.shift();
-            }
-        });
-        // Clean up logs and network requests when page is closed
-        newPage.on('close', () => {
-            state.consoleLogs.delete(newPage);
-            state.networkRequests.delete(newPage);
-        });
-        newPage.on('request', (request) => {
-            const startTime = Date.now();
-            const entry = {
-                url: request.url(),
-                method: request.method(),
-                headers: request.headers(),
-                timestamp: startTime,
-            };
-            request
-                .response()
-                .then((response) => {
-                if (response) {
-                    entry.status = response.status();
-                    entry.duration = Date.now() - startTime;
-                    entry.size = response.headers()['content-length']
-                        ? parseInt(response.headers()['content-length'])
-                        : 0;
-                    // Get or create requests array for this page
-                    let pageRequests = state.networkRequests.get(newPage);
-                    if (!pageRequests) {
-                        pageRequests = [];
-                        state.networkRequests.set(newPage, pageRequests);
-                    }
-                    // Add new request
-                    pageRequests.push(entry);
-                    // Keep only last 1000 requests
-                    if (pageRequests.length > 1000) {
-                        pageRequests.shift();
-                    }
-                }
-            })
-                .catch(() => {
-                // Handle response errors silently
-            });
-        });
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: `Created new page. URL: ${newPage.url()}. Total pages: ${context.pages().length}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: `Failed to create new page: ${error.message}`,
-                },
-            ],
-            isError: true,
-        };
+async function getCurrentPage() {
+    if (state.page) {
+        return state.page;
     }
-});
-// Tool 2: Console Logs
-server.tool('console_logs', 'Retrieve console messages from the page', {
-    limit: z
-        .number()
-        .default(50)
-        .describe('Maximum number of messages to return'),
-    type: z
-        .enum(['log', 'info', 'warning', 'error', 'debug'])
-        .optional()
-        .describe('Filter by message type'),
-    offset: z.number().default(0).describe('Start from this index'),
-}, async ({ limit, type, offset }) => {
-    try {
-        const { page } = await ensureConnection(); // Ensure we're connected first
-        // Get logs for current page
-        const pageLogs = state.consoleLogs.get(page) || [];
-        // Filter and paginate logs
-        let logs = [...pageLogs];
-        if (type) {
-            logs = logs.filter((log) => log.type === type);
-        }
-        const paginatedLogs = logs.slice(offset, offset + limit);
-        // Format logs to look like Chrome console output
-        let consoleOutput = '';
-        if (paginatedLogs.length === 0) {
-            consoleOutput = 'No console messages';
-        }
-        else {
-            consoleOutput = paginatedLogs
-                .map((log) => {
-                const timestamp = new Date(log.timestamp).toLocaleTimeString();
-                const location = log.location
-                    ? ` ${log.location.url}:${log.location.lineNumber}:${log.location.columnNumber}`
-                    : '';
-                return `[${log.type}]: ${log.text}${location}`;
-            })
-                .join('\n');
-            if (logs.length > paginatedLogs.length) {
-                consoleOutput += `\n\n(Showing ${paginatedLogs.length} of ${logs.length} total messages)`;
+    if (state.browser) {
+        const contexts = state.browser.contexts();
+        if (contexts.length > 0) {
+            const pages = contexts[0].pages();
+            if (pages.length > 0) {
+                const page = pages[0];
+                await page.emulateMedia({ colorScheme: null });
+                return page;
             }
         }
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: consoleOutput,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: `Failed to get console logs: ${error.message}`,
-                },
-            ],
-            isError: true,
-        };
-    }
-});
-// Tool 3: Network History
-server.tool('network_history', 'Get history of network requests', {
-    limit: z
-        .number()
-        .default(50)
-        .describe('Maximum number of requests to return'),
-    urlPattern: z
-        .string()
-        .optional()
-        .describe('Filter by URL pattern (supports wildcards)'),
-    method: z
-        .enum(['GET', 'POST', 'PUT', 'DELETE', 'PATCH'])
-        .optional()
-        .describe('Filter by HTTP method'),
-    statusCode: z
-        .object({
-        min: z.number().optional(),
-        max: z.number().optional(),
-    })
-        .optional()
-        .describe('Filter by status code range'),
-    includeBody: z
-        .boolean()
-        .default(false)
-        .describe('Include request/response bodies'),
-}, async ({ limit, urlPattern, method, statusCode, includeBody }) => {
-    try {
-        const { page } = await ensureConnection();
-        // Get requests for current page
-        const pageRequests = state.networkRequests.get(page) || [];
-        // If includeBody is requested, we need to fetch bodies for existing requests
-        if (includeBody && pageRequests.length > 0) {
-            // Note: In a real implementation, you'd store bodies during capture
-            console.warn('Body capture not implemented in this example');
-        }
-        // Filter requests
-        let requests = [...pageRequests];
-        if (urlPattern) {
-            const pattern = new RegExp(urlPattern.replace(/\*/g, '.*'));
-            requests = requests.filter((req) => pattern.test(req.url));
-        }
-        if (method) {
-            requests = requests.filter((req) => req.method === method);
-        }
-        if (statusCode) {
-            requests = requests.filter((req) => {
-                if (statusCode.min && req.status < statusCode.min)
-                    return false;
-                if (statusCode.max && req.status > statusCode.max)
-                    return false;
-                return true;
-            });
-        }
-        const limitedRequests = requests.slice(-limit);
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        total: requests.length,
-                        requests: limitedRequests,
-                    }, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: `Failed to get network history: ${error.message}`,
-                },
-            ],
-            isError: true,
-        };
-    }
-});
-// Tool 4: Accessibility Snapshot - Get page accessibility tree as JSON
-server.tool('accessibility_snapshot', 'Get the accessibility snapshot of the current page as JSON', {}, async ({}) => {
-    try {
-        const { page } = await ensureConnection();
-        // Check if the method exists
-        if (typeof page._snapshotForAI !== 'function') {
-            // Fall back to regular accessibility snapshot
-            const snapshot = await page.accessibility.snapshot({
-                interestingOnly: true,
-                root: undefined,
-            });
-            return {
-                content: [
-                    {
-                        type: 'text',
-                        text: JSON.stringify(snapshot, null, 2),
-                    },
-                ],
-            };
-        }
-        const snapshot = await page._snapshotForAI();
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: snapshot,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        console.error('Accessibility snapshot error:', error);
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: `Failed to get accessibility snapshot: ${error.message}`,
-                },
-            ],
-            isError: true,
-        };
     }
+    throw new Error('No page available');
+}
+const server = new McpServer({
+    name: 'playwriter',
+    title: 'Playwright MCP Server',
+    version: '1.0.0',
 });
-// Tool 5: Execute - Run arbitrary JavaScript code with page and context in scope
 const promptContent = fs.readFileSync(path.join(path.dirname(new URL(import.meta.url).pathname), 'prompt.md'), 'utf-8');
 server.tool('execute', promptContent, {
     code: z
         .string()
         .describe('JavaScript code to execute with page and context in scope. Should be one line, using ; to execute multiple statements. To execute complex actions call execute multiple times. '),
-    timeout: z
-        .number()
-        .default(3000)
-        .describe('Timeout in milliseconds for code execution (default: 3000ms)'),
+    timeout: z.number().default(3000).describe('Timeout in milliseconds for code execution (default: 3000ms)'),
 }, async ({ code, timeout }) => {
-    const { page } = await ensureConnection();
-    const context = page.context();
+    await ensureConnection();
+    const page = await getCurrentPage();
+    const context = state.context || page.context();
     console.error('Executing code:', code);
     try {
-        // Collect console logs during execution
         const consoleLogs = [];
-        // Create a custom console object that collects logs
         const customConsole = {
             log: (...args) => {
                 consoleLogs.push({ method: 'log', args });
@@ -503,20 +110,40 @@ server.tool('execute', promptContent, {
                 consoleLogs.push({ method: 'debug', args });
             },
         };
-        // Create a function that has page, context, and console in scope
-        const executeCode = new Function('page', 'context', 'console', `
-                return (async () => {
-                    ${code}
-                })();
-            `);
-        // Execute the code with page, context, and custom console with timeout
+        const accessibilitySnapshot = async (targetPage) => {
+            if (targetPage._snapshotForAI) {
+                const snapshot = await targetPage._snapshotForAI();
+                const snapshotStr = typeof snapshot === 'string'
+                    ? snapshot
+                    : JSON.stringify(snapshot, null, 2);
+                return snapshotStr;
+            }
+            throw new Error('accessibilitySnapshot is not available on this page');
+        };
+        const activateTab = async (targetPage) => {
+            const cdp = await context.newCDPSession(targetPage);
+            const { targetInfo } = await cdp.send('Target.getTargetInfo');
+            const targetId = targetInfo.targetId;
+            await cdp.send('Playwriter.activateTab', { targetId });
+            await cdp.detach();
+        };
+        const vmContext = vm.createContext({
+            page,
+            context,
+            state,
+            console: customConsole,
+            accessibilitySnapshot,
+            activateTab,
+        });
+        const wrappedCode = `(async () => { ${code} })()`;
         const result = await Promise.race([
-            executeCode(page, context, customConsole),
-            new Promise((_, reject) => setTimeout(() => reject(new Error(`Code execution timed out after ${timeout}ms`)), timeout))
+            vm.runInContext(wrappedCode, vmContext, {
+                timeout,
+                displayErrors: true,
+            }),
+            new Promise((_, reject) => setTimeout(() => reject(new Error(`Code execution timed out after ${timeout}ms`)), timeout)),
         ]);
-        // Format the response with both console output and return value
         let responseText = '';
-        // Add console logs if any
         if (consoleLogs.length > 0) {
             responseText += 'Console output:\n';
             consoleLogs.forEach(({ method, args }) => {
@@ -532,19 +159,28 @@ server.tool('execute', promptContent, {
             });
             responseText += '\n';
         }
-        // Add return value if any
         if (result !== undefined) {
             responseText += 'Return value:\n';
-            responseText += JSON.stringify(result, null, 2);
+            if (typeof result === 'string') {
+                responseText += result;
+            }
+            else {
+                responseText += JSON.stringify(result, null, 2);
+            }
         }
         else if (consoleLogs.length === 0) {
             responseText += 'Code executed successfully (no output)';
         }
+        const MAX_LENGTH = 1000;
+        let finalText = responseText.trim();
+        if (finalText.length > MAX_LENGTH) {
+            finalText = finalText.slice(0, MAX_LENGTH) + `\n\n[Truncated to ${MAX_LENGTH} characters. Better manage your logs or paginate them to read the full logs]`;
+        }
         return {
             content: [
                 {
                     type: 'text',
-                    text: responseText.trim(),
+                    text: finalText,
                 },
             ],
         };
@@ -554,7 +190,7 @@ server.tool('execute', promptContent, {
             content: [
                 {
                     type: 'text',
-                    text: `Error executing code: ${error.message}\n${error.stack}`,
+                    text: `Error executing code: ${error.message}\n${error.stack || ''}`,
                 },
             ],
             isError: true,
@@ -567,23 +203,16 @@ async function main() {
     await server.connect(transport);
     console.error('Playwright MCP server running on stdio');
 }
-// Cleanup function
 async function cleanup() {
     console.error('Shutting down MCP server...');
     if (state.browser) {
         try {
-            // Close the browser connection but not the Chrome process
-            // Since we're using CDP, closing the browser object just closes
-            // the connection, not the actual Chrome instance
             await state.browser.close();
         }
         catch (e) {
             // Ignore errors during browser close
         }
     }
-    // Don't kill the Chrome process - let it continue running
-    // The process was started with detached: true and unref()
-    // so it will persist after this process exits
     process.exit(0);
 }
 // Handle process termination