@wdio/mcp 2.4.1 → 2.5.0

package/README.md

@@ -85,6 +85,7 @@ appium
 - **Scrolling**: Smooth scrolling with configurable distances
 - **Attach to running Chrome**: Connect to an existing Chrome window via `--remote-debugging-port` — ideal for testing authenticated or pre-configured sessions
 - **Device emulation**: Apply mobile/tablet presets (iPhone 15, Pixel 7, etc.) to simulate responsive layouts without a physical device
+- **Session Recording**: All tool calls are automatically recorded and exportable as runnable WebdriverIO JS
 
 ### Mobile App Automation (iOS/Android)
 
@@ -458,6 +459,16 @@ This eliminates the need to manually handle permission popups during automated t
 - **Data Format:** TOON (Token-Oriented Object Notation) for efficient LLM communication
 - **Element Detection:** XML-based page source parsing with intelligent filtering and multi-strategy locator generation
 
+### Session Recording & Code Export
+
+Every tool call is automatically recorded to a session history. You can inspect sessions and export runnable code via MCP resources — no extra tool calls needed:
+
+- `wdio://sessions` — lists all recorded sessions with type, timestamps, and step count
+- `wdio://session/current/steps` — step log for the active session, plus a generated WebdriverIO JS script ready to run with `webdriverio`
+- `wdio://session/{sessionId}/steps` — same for any past session by ID
+
+The generated script reconstructs the full session — including capabilities, navigation, clicks, and inputs — as a standalone `import { remote } from 'webdriverio'` file.
+
 ## Troubleshooting
 
 **Browser automation not working?**

package/lib/server.js

@@ -157,14 +157,7 @@ var startBrowserTool = async ({
     type: "browser",
     startedAt: (/* @__PURE__ */ new Date()).toISOString(),
     capabilities: wdioBrowser.capabilities,
-    steps: [{
-      index: 1,
-      tool: "start_browser",
-      params: { browser, headless, windowWidth, windowHeight, ...navigationUrl && { navigationUrl }, ...Object.keys(userCapabilities).length > 0 && { capabilities: userCapabilities } },
-      status: "ok",
-      durationMs: 0,
-      timestamp: (/* @__PURE__ */ new Date()).toISOString()
-    }]
+    steps: []
   });
   state.currentSession = sessionId;
   let sizeNote = "";
@@ -211,6 +204,7 @@ var closeSessionTool = async (args = {}) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error closing session: ${e}` }]
     };
   }
@@ -234,6 +228,7 @@ var navigateTool = async ({ url }) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error navigating: ${e}` }]
     };
   }
@@ -264,6 +259,7 @@ var clickAction = async (selector, timeout, scrollToView = true) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error clicking element: ${e}` }]
     };
   }
@@ -297,6 +293,7 @@ var setValueTool = async ({ selector, value, scrollToView = true, timeout = defa
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error entering text: ${e}` }]
     };
   }
@@ -343,7 +340,7 @@ function buildIOSCapabilities(appPath, options) {
     capabilities["appium:autoAcceptAlerts"] = void 0;
   }
   for (const [key, value] of Object.entries(options)) {
-    if (!["deviceName", "platformVersion", "automationName", "autoAcceptAlerts", "autoDismissAlerts", "udid", "noReset", "fullReset", "newCommandTimeout"].includes(
+    if (!["deviceName", "platformVersion", "automationName", "autoGrantPermissions", "autoAcceptAlerts", "autoDismissAlerts", "udid", "noReset", "fullReset", "newCommandTimeout"].includes(
       key
     )) {
       capabilities[`appium:${key}`] = value;
@@ -380,7 +377,7 @@ function buildAndroidCapabilities(appPath, options) {
     capabilities["appium:appWaitActivity"] = options.appWaitActivity;
   }
   for (const [key, value] of Object.entries(options)) {
-    if (!["deviceName", "platformVersion", "automationName", "autoGrantPermissions", "appWaitActivity", "noReset", "fullReset", "newCommandTimeout"].includes(
+    if (!["deviceName", "platformVersion", "automationName", "autoGrantPermissions", "autoAcceptAlerts", "autoDismissAlerts", "appWaitActivity", "noReset", "fullReset", "newCommandTimeout"].includes(
       key
     )) {
       capabilities[`appium:${key}`] = value;
@@ -521,32 +518,8 @@ var startAppTool = async (args) => {
       type: platform.toLowerCase(),
       startedAt: (/* @__PURE__ */ new Date()).toISOString(),
       capabilities: mergedCapabilities,
-      steps: [{
-        index: 1,
-        tool: "start_app_session",
-        params: {
-          platform,
-          deviceName,
-          ...platformVersion !== void 0 && { platformVersion },
-          ...automationName !== void 0 && { automationName },
-          ...appPath !== void 0 && { appPath },
-          ...udid !== void 0 && { udid },
-          ...noReset !== void 0 && { noReset },
-          ...fullReset !== void 0 && { fullReset },
-          ...autoGrantPermissions !== void 0 && { autoGrantPermissions },
-          ...autoAcceptAlerts !== void 0 && { autoAcceptAlerts },
-          ...autoDismissAlerts !== void 0 && { autoDismissAlerts },
-          ...appWaitActivity !== void 0 && { appWaitActivity },
-          ...newCommandTimeout !== void 0 && { newCommandTimeout },
-          ...appiumHost !== void 0 && { appiumHost },
-          ...appiumPort !== void 0 && { appiumPort },
-          ...appiumPath !== void 0 && { appiumPath },
-          ...Object.keys(userCapabilities).length > 0 && { capabilities: userCapabilities }
-        },
-        status: "ok",
-        durationMs: 0,
-        timestamp: (/* @__PURE__ */ new Date()).toISOString()
-      }]
+      appiumConfig: { hostname: serverConfig.hostname, port: serverConfig.port, path: serverConfig.path },
+      steps: []
     });
     state2.currentSession = sessionId;
     const appInfo = appPath ? `
@@ -564,6 +537,7 @@ Appium Server: ${serverConfig.hostname}:${serverConfig.port}${serverConfig.path}
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error starting app session: ${e}` }]
     };
   }
@@ -597,6 +571,7 @@ var scrollTool = async ({ direction, pixels = 500 }) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error scrolling: ${e}` }]
     };
   }
@@ -1769,6 +1744,7 @@ var getVisibleElementsTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error getting visible elements: ${e}` }]
     };
   }
@@ -2088,6 +2064,7 @@ var getAccessibilityTreeTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error getting accessibility tree: ${e}` }]
     };
   }
@@ -2148,6 +2125,7 @@ var takeScreenshotTool = async ({ outputPath }) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error taking screenshot: ${e.message}` }]
     };
   }
@@ -2187,6 +2165,7 @@ var getCookiesTool = async ({ name }) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error getting cookies: ${e}` }]
     };
   }
@@ -2224,6 +2203,7 @@ var setCookieTool = async ({
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error setting cookie: ${e}` }]
     };
   }
@@ -2250,6 +2230,7 @@ var deleteCookiesTool = async ({ name }) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error deleting cookies: ${e}` }]
     };
   }
@@ -2283,10 +2264,12 @@ var tapElementTool = async (args) => {
       };
     }
     return {
+      isError: true,
       content: [{ type: "text", text: "Error: Must provide either selector or x,y coordinates" }]
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error tapping: ${e}` }]
     };
   }
@@ -2321,6 +2304,7 @@ var swipeTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error swiping: ${e}` }]
     };
   }
@@ -2354,10 +2338,12 @@ var dragAndDropTool = async (args) => {
       };
     }
     return {
+      isError: true,
       content: [{ type: "text", text: "Error: Must provide either targetSelector or x,y coordinates" }]
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error dragging: ${e}` }]
     };
   }
@@ -2395,6 +2381,7 @@ var getAppStateTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error getting app state: ${e}` }]
     };
   }
@@ -2436,6 +2423,7 @@ ${contexts.map((ctx, idx) => `${idx + 1}. ${ctx}`).join("\n")}`
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error getting contexts: ${e}` }]
     };
   }
@@ -2449,6 +2437,7 @@ var getCurrentContextTool = async () => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error getting current context: ${e}` }]
     };
   }
@@ -2464,14 +2453,7 @@ var switchContextTool = async (args) => {
       if (index >= 0 && index < contexts.length) {
         targetContext = contexts[index];
       } else {
-        return {
-          content: [
-            {
-              type: "text",
-              text: `Error: Invalid context index ${context}. Available contexts: ${contexts.length}`
-            }
-          ]
-        };
+        throw new Error(`Error: Invalid context index ${context}. Available contexts: ${contexts.length}`);
       }
     }
     await browser.switchContext(targetContext);
@@ -2480,6 +2462,7 @@ var switchContextTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error switching context: ${e}` }]
     };
   }
@@ -2523,6 +2506,7 @@ var rotateDeviceTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error rotating device: ${e}` }]
     };
   }
@@ -2536,6 +2520,7 @@ var hideKeyboardTool = async () => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error hiding keyboard: ${e}` }]
     };
   }
@@ -2557,6 +2542,7 @@ var getGeolocationTool = async () => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error getting geolocation: ${e}` }]
     };
   }
@@ -2579,6 +2565,7 @@ var setGeolocationTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error setting geolocation: ${e}` }]
     };
   }
@@ -2644,6 +2631,7 @@ var executeScriptTool = async (args) => {
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error executing script: ${e}` }]
     };
   }
@@ -2718,6 +2706,19 @@ var attachBrowserTool = async ({
       capabilities: browser.capabilities,
       isAttached: true
     });
+    state2.sessionHistory.set(sessionId, {
+      sessionId,
+      type: "browser",
+      startedAt: (/* @__PURE__ */ new Date()).toISOString(),
+      capabilities: {
+        browserName: "chrome",
+        "goog:chromeOptions": {
+          debuggerAddress: `${host}:${port}`,
+          args: [`--user-data-dir=${userDataDir}`]
+        }
+      },
+      steps: []
+    });
     if (activeUrl) {
       await browser.url(activeUrl);
     }
@@ -2733,6 +2734,7 @@ Current page: "${title}" (${url})`
     };
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error attaching to browser: ${e}` }]
     };
   }
@@ -2767,11 +2769,13 @@ var emulateDeviceTool = async ({
     const metadata = state2.sessionMetadata.get(sessionId);
     if (metadata?.type === "ios" || metadata?.type === "android") {
       return {
+        isError: true,
         content: [{ type: "text", text: "Error: emulate_device is only supported for web browser sessions, not iOS/Android." }]
       };
     }
     if (!browser.isBidi) {
       return {
+        isError: true,
         content: [{
           type: "text",
           text: "Error: emulate_device requires a BiDi-enabled session.\nRestart the browser with: start_browser({ capabilities: { webSocketUrl: true } })"
@@ -2791,7 +2795,7 @@ var emulateDeviceTool = async ({
 ${names.join("\n")}` }]
           };
         }
-        return { content: [{ type: "text", text: `Error listing devices: ${e}` }] };
+        return { isError: true, content: [{ type: "text", text: `Error listing devices: ${e}` }] };
       }
       return { content: [{ type: "text", text: "Could not retrieve device list." }] };
     }
@@ -2820,10 +2824,11 @@ ${names.join("\n")}` }]
           }]
         };
       }
-      return { content: [{ type: "text", text: `Error: ${e}` }] };
+      return { isError: true, content: [{ type: "text", text: `Error: ${e}` }] };
     }
   } catch (e) {
     return {
+      isError: true,
       content: [{ type: "text", text: `Error: ${e}` }]
     };
   }
@@ -2837,7 +2842,7 @@ var package_default = {
     type: "git",
     url: "git://github.com/webdriverio/mcp.git"
   },
-  version: "2.4.0",
+  version: "2.4.1",
   description: "MCP server with WebdriverIO for browser and mobile app automation (iOS/Android via Appium)",
   main: "./lib/server.js",
   module: "./lib/server.js",
@@ -2940,9 +2945,7 @@ function withRecording(toolName, callback) {
   return async (params, extra) => {
     const start = Date.now();
     const result = await callback(params, extra);
-    const isError = result.content.some(
-      (c) => c.type === "text" && typeof c.text === "string" && c.text.startsWith("Error")
-    );
+    const isError = result.isError === true;
     appendStep(
       toolName,
       params,
@@ -2964,7 +2967,7 @@ function formatParams(params) {
 function indentJson(value) {
   return JSON.stringify(value, null, 2).split("\n").map((line, i) => i > 0 ? `  ${line}` : line).join("\n");
 }
-function generateStep(step) {
+function generateStep(step, history) {
   if (step.tool === "__session_transition__") {
     const newId = step.params.newSessionId ?? "unknown";
     return `// --- new session: ${newId} started at ${step.timestamp} ---`;
@@ -2975,52 +2978,29 @@ function generateStep(step) {
   const p = step.params;
   switch (step.tool) {
     case "start_browser": {
-      const browserName = p.browser === "edge" ? "msedge" : String(p.browser ?? "chrome");
-      const headless = p.headless !== false;
-      const width = p.windowWidth ?? 1920;
-      const height = p.windowHeight ?? 1080;
-      const args = [`--window-size=${width},${height}`];
-      if (headless && browserName !== "safari") {
-        args.push("--headless=new", "--disable-gpu", "--disable-dev-shm-usage");
-      }
-      const caps = { browserName };
-      if (browserName === "chrome") caps["goog:chromeOptions"] = { args };
-      else if (browserName === "msedge") caps["ms:edgeOptions"] = { args };
-      else if (browserName === "firefox" && headless) caps["moz:firefoxOptions"] = { args: ["-headless"] };
-      const extra = p.capabilities;
-      const merged = extra ? { ...caps, ...extra } : caps;
       const nav = p.navigationUrl ? `
 await browser.url('${escapeStr(p.navigationUrl)}');` : "";
       return `const browser = await remote({
-  capabilities: ${indentJson(merged)}
+  capabilities: ${indentJson(history.capabilities)}
 });${nav}`;
     }
     case "start_app_session": {
-      const caps = {
-        platformName: p.platform,
-        "appium:deviceName": p.deviceName,
-        ...p.platformVersion !== void 0 && { "appium:platformVersion": p.platformVersion },
-        ...p.automationName !== void 0 && { "appium:automationName": p.automationName },
-        ...p.appPath !== void 0 && { "appium:app": p.appPath },
-        ...p.udid !== void 0 && { "appium:udid": p.udid },
-        ...p.noReset !== void 0 && { "appium:noReset": p.noReset },
-        ...p.fullReset !== void 0 && { "appium:fullReset": p.fullReset },
-        ...p.autoGrantPermissions !== void 0 && { "appium:autoGrantPermissions": p.autoGrantPermissions },
-        ...p.autoAcceptAlerts !== void 0 && { "appium:autoAcceptAlerts": p.autoAcceptAlerts },
-        ...p.autoDismissAlerts !== void 0 && { "appium:autoDismissAlerts": p.autoDismissAlerts },
-        ...p.appWaitActivity !== void 0 && { "appium:appWaitActivity": p.appWaitActivity },
-        ...p.newCommandTimeout !== void 0 && { "appium:newCommandTimeout": p.newCommandTimeout },
-        ...p.capabilities ?? {}
-      };
       const config = {
         protocol: "http",
-        hostname: p.appiumHost ?? "localhost",
-        port: p.appiumPort ?? 4723,
-        path: p.appiumPath ?? "/",
-        capabilities: caps
+        hostname: history.appiumConfig?.hostname ?? "localhost",
+        port: history.appiumConfig?.port ?? 4723,
+        path: history.appiumConfig?.path ?? "/",
+        capabilities: history.capabilities
       };
       return `const browser = await remote(${indentJson(config)});`;
     }
+    case "attach_browser": {
+      const nav = p.navigationUrl ? `
+await browser.url('${escapeStr(p.navigationUrl)}');` : "";
+      return `const browser = await remote({
+  capabilities: ${indentJson(history.capabilities)}
+});${nav}`;
+    }
     case "navigate":
       return `await browser.url('${escapeStr(p.url)}');`;
     case "click_element":
@@ -3048,7 +3028,7 @@ await browser.url('${escapeStr(p.navigationUrl)}');` : "";
   }
 }
 function generateCode(history) {
-  const steps = history.steps.map(generateStep).join("\n");
+  const steps = history.steps.map((step) => generateStep(step, history)).join("\n");
   return `import { remote } from 'webdriverio';
 
 ${steps}
@@ -3062,47 +3042,37 @@ function getCurrentSessionId() {
 }
 function buildSessionsIndex() {
   const histories = getSessionHistory();
-  if (histories.size === 0) return "No sessions recorded.";
   const currentId = getCurrentSessionId();
-  const lines = [`Sessions (${histories.size} total):
-`];
-  for (const [id, h] of histories) {
-    const ended = h.endedAt ?? "-";
-    const current = id === currentId ? "  [current]" : "";
-    lines.push(`- ${id}  ${h.type}  started: ${h.startedAt}  ended: ${ended}  ${h.steps.length} steps${current}`);
-  }
-  return lines.join("\n");
+  const sessions = Array.from(histories.values()).map((h) => ({
+    sessionId: h.sessionId,
+    type: h.type,
+    startedAt: h.startedAt,
+    ...h.endedAt ? { endedAt: h.endedAt } : {},
+    stepCount: h.steps.length,
+    isCurrent: h.sessionId === currentId
+  }));
+  return JSON.stringify({ sessions });
 }
 function buildCurrentSessionSteps() {
   const currentId = getCurrentSessionId();
-  if (!currentId) return "No active session.";
+  if (!currentId) return null;
   return buildSessionStepsById(currentId);
 }
 function buildSessionStepsById(sessionId) {
   const history = getSessionHistory().get(sessionId);
-  if (!history) return `Session not found: ${sessionId}`;
-  return formatSessionSteps(history);
-}
-function formatSessionSteps(history) {
-  const header = `Session: ${history.sessionId} (${history.type}) \u2014 ${history.steps.length} steps
-`;
-  const stepLines = history.steps.map((step) => {
-    if (step.tool === "__session_transition__") {
-      return `--- session transitioned to ${step.params.newSessionId ?? "unknown"} at ${step.timestamp} ---`;
-    }
-    const statusLabel = step.status === "ok" ? "[ok]   " : "[error]";
-    const params = Object.entries(step.params).map(([k, v]) => `${k}="${v}"`).join(" ");
-    const errorSuffix = step.error ? ` \u2014 ${step.error}` : "";
-    return `${step.index}. ${statusLabel}  ${step.tool.padEnd(24)} ${params}${errorSuffix}  ${step.durationMs}ms`;
+  if (!history) return null;
+  return buildSessionPayload(history);
+}
+function buildSessionPayload(history) {
+  const stepsJson = JSON.stringify({
+    sessionId: history.sessionId,
+    type: history.type,
+    startedAt: history.startedAt,
+    ...history.endedAt ? { endedAt: history.endedAt } : {},
+    stepCount: history.steps.length,
+    steps: history.steps
   });
-  const stepsText = stepLines.length > 0 ? stepLines.join("\n") : "(no steps yet)";
-  const jsCode = generateCode(history);
-  return `${header}
-Steps:
-${stepsText}
-
---- Generated WebdriverIO JS ---
-${jsCode}`;
+  return { stepsJson, generatedJs: generateCode(history) };
 }
 
 // src/server.ts
@@ -3127,10 +3097,10 @@ var registerTool = (definition, callback) => server.registerTool(definition.name
   description: definition.description,
   inputSchema: definition.inputSchema
 }, callback);
-registerTool(startBrowserToolDefinition, startBrowserTool);
-registerTool(startAppToolDefinition, startAppTool);
+registerTool(startBrowserToolDefinition, withRecording("start_browser", startBrowserTool));
+registerTool(startAppToolDefinition, withRecording("start_app_session", startAppTool));
 registerTool(closeSessionToolDefinition, closeSessionTool);
-registerTool(attachBrowserToolDefinition, attachBrowserTool);
+registerTool(attachBrowserToolDefinition, withRecording("attach_browser", attachBrowserTool));
 registerTool(emulateDeviceToolDefinition, emulateDeviceTool);
 registerTool(navigateToolDefinition, withRecording("navigate", navigateTool));
 registerTool(getVisibleElementsToolDefinition, getVisibleElementsTool);
@@ -3157,30 +3127,54 @@ registerTool(executeScriptToolDefinition, executeScriptTool);
 server.registerResource(
   "sessions",
   "wdio://sessions",
-  { description: "Index of all browser and app sessions with step counts" },
+  { description: "JSON index of all browser and app sessions with metadata and step counts" },
   async () => ({
-    contents: [{ uri: "wdio://sessions", mimeType: "text/plain", text: buildSessionsIndex() }]
+    contents: [{ uri: "wdio://sessions", mimeType: "application/json", text: buildSessionsIndex() }]
   })
 );
 server.registerResource(
   "session-current-steps",
   "wdio://session/current/steps",
-  { description: "Steps for the currently active session with generated WebdriverIO JS" },
-  async () => ({
-    contents: [{ uri: "wdio://session/current/steps", mimeType: "text/plain", text: buildCurrentSessionSteps() }]
-  })
+  { description: "JSON step log for the currently active session" },
+  async () => {
+    const payload = buildCurrentSessionSteps();
+    return {
+      contents: [{ uri: "wdio://session/current/steps", mimeType: "application/json", text: payload?.stepsJson ?? '{"error":"No active session"}' }]
+    };
+  }
+);
+server.registerResource(
+  "session-current-code",
+  "wdio://session/current/code",
+  { description: "Generated WebdriverIO JS code for the currently active session" },
+  async () => {
+    const payload = buildCurrentSessionSteps();
+    return {
+      contents: [{ uri: "wdio://session/current/code", mimeType: "text/plain", text: payload?.generatedJs ?? "// No active session" }]
+    };
+  }
 );
 server.registerResource(
   "session-steps",
   new ResourceTemplate("wdio://session/{sessionId}/steps", { list: void 0 }),
-  { description: "Steps for a specific session by ID with generated WebdriverIO JS" },
-  async (uri, { sessionId }) => ({
-    contents: [{
-      uri: uri.href,
-      mimeType: "text/plain",
-      text: buildSessionStepsById(sessionId)
-    }]
-  })
+  { description: "JSON step log for a specific session by ID" },
+  async (uri, { sessionId }) => {
+    const payload = buildSessionStepsById(sessionId);
+    return {
+      contents: [{ uri: uri.href, mimeType: "application/json", text: payload?.stepsJson ?? `{"error":"Session not found: ${sessionId}"}` }]
+    };
+  }
+);
+server.registerResource(
+  "session-code",
+  new ResourceTemplate("wdio://session/{sessionId}/code", { list: void 0 }),
+  { description: "Generated WebdriverIO JS code for a specific session by ID" },
+  async (uri, { sessionId }) => {
+    const payload = buildSessionStepsById(sessionId);
+    return {
+      contents: [{ uri: uri.href, mimeType: "text/plain", text: payload?.generatedJs ?? `// Session not found: ${sessionId}` }]
+    };
+  }
 );
 async function main() {
   const transport = new StdioServerTransport();