@everydaydevopsio/pilot 0.6.3 → 0.7.0

package/src/mcp/tools/interaction.ts

@@ -1,21 +1,7 @@
 import { z } from 'zod';
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { BrowserContext } from '../server.js';
-import { executeFill } from '../../commands/fill.js';
-import { executeHover } from '../../commands/hover.js';
-import { executePressKey } from '../../commands/press_key.js';
-import { executeSelect } from '../../commands/select.js';
-import { executeCheck } from '../../commands/check.js';
-import { executeScroll } from '../../commands/scroll.js';
-
-function requireContext(context: BrowserContext) {
-  const manager = context.manager;
-  if (!manager || !manager.isConnected()) {
-    throw new Error('Browser not started. Call browser_start first.');
-  }
-  const client = manager.getClient()!;
-  return { client, refMap: context.elementRefMap };
-}
+import { requireDriver } from './driver-context.js';
 
 const fillShape = {
   ref: z.string().describe('Element ref from browser_snapshot (e.g. "e3")'),
@@ -81,8 +67,7 @@ export function registerInteractionTools(
     'Fill a form field by element ref. Replaces the entire field value and fires input/change events compatible with React, Vue, and Svelte. Use browser_snapshot first to get refs.',
     fillShape,
     async ({ ref, value }) => {
-      const { client, refMap } = requireContext(context);
-      const result = await executeFill(client, refMap, { ref, value });
+      const result = await requireDriver(context).fill({ ref, value });
       return {
         content: [
           {
@@ -113,8 +98,7 @@ export function registerInteractionTools(
           isError: true
         };
       }
-      const { client, refMap } = requireContext(context);
-      const result = await executeHover(client, refMap, params);
+      const result = await requireDriver(context).hover(params);
       return {
         content: [
           {
@@ -131,8 +115,7 @@ export function registerInteractionTools(
     'Press a key or key combination. Supports named keys (Enter, Tab, Escape, etc.) and modifier combos (Control+a, Meta+c, Shift+Enter).',
     pressKeyShape,
     async ({ key }) => {
-      const { client } = requireContext(context);
-      const result = await executePressKey(client, { key });
+      const result = await requireDriver(context).pressKey({ key });
       const modStr =
         result.modifiers.length > 0
           ? ` with modifiers: ${result.modifiers.join('+')}`
@@ -165,8 +148,7 @@ export function registerInteractionTools(
           isError: true
         };
       }
-      const { client, refMap } = requireContext(context);
-      const result = await executeSelect(client, refMap, params);
+      const result = await requireDriver(context).select(params);
       return {
         content: [
           {
@@ -183,8 +165,7 @@ export function registerInteractionTools(
     'Toggle a checkbox or radio button by ref. Optionally set to a specific checked state.',
     checkShape,
     async ({ ref, checked }) => {
-      const { client, refMap } = requireContext(context);
-      const result = await executeCheck(client, refMap, { ref, checked });
+      const result = await requireDriver(context).check({ ref, checked });
       return {
         content: [
           {
@@ -217,8 +198,7 @@ export function registerInteractionTools(
           isError: true
         };
       }
-      const { client, refMap } = requireContext(context);
-      const result = await executeScroll(client, refMap, params);
+      const result = await requireDriver(context).scroll(params);
       return {
         content: [
           {

package/src/mcp/tools/network.ts

@@ -1,8 +1,8 @@
 import { z } from 'zod';
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { BrowserContext } from '../server.js';
+import { requireDriver } from './driver-context.js';
 import {
-  executeNetwork,
   formatNetworkList,
   formatNetworkDetail
 } from '../../commands/network.js';
@@ -48,15 +48,6 @@ const networkShape = {
     .describe('Maximum number of results (list only)')
 };
 
-function requireContext(context: BrowserContext) {
-  const manager = context.manager;
-  if (!manager || !manager.isConnected()) {
-    throw new Error('Browser not started. Call browser_start first.');
-  }
-  const client = manager.getClient()!;
-  return { client, networkBuffer: context.networkBuffer };
-}
-
 export function registerNetworkTools(
   server: McpServer,
   context: BrowserContext
@@ -66,8 +57,7 @@ export function registerNetworkTools(
     'Inspect captured network requests. Actions: "list" shows request summaries with filters (url, method, status, resourceType, failed); "get" returns full detail for a request ID including headers and response body; "clear" resets the buffer.',
     networkShape,
     async (params) => {
-      const { client, networkBuffer } = requireContext(context);
-      const result = await executeNetwork(client, networkBuffer, params);
+      const result = await requireDriver(context).network(params);
 
       let text: string;
       if (result.action === 'list') {

package/src/mcp/tools/performance.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { BrowserContext } from '../server.js';
-import { executePerformance } from '../../commands/performance.js';
+import { requireDriver } from './driver-context.js';
 
 const performanceShape = {
   action: z
@@ -11,14 +11,6 @@ const performanceShape = {
     )
 };
 
-function requireContext(context: BrowserContext) {
-  const manager = context.manager;
-  if (!manager || !manager.isConnected()) {
-    throw new Error('Browser not started. Call browser_start first.');
-  }
-  return { client: manager.getClient()! };
-}
-
 export function registerPerformanceTools(
   server: McpServer,
   context: BrowserContext
@@ -28,10 +20,7 @@ export function registerPerformanceTools(
     'Inspect page performance. Actions: "start" begins tracing (call before navigating), "stop" ends collection, "analyze" returns an agent-friendly summary with navigation timing, long tasks, slow requests, large resources, and JS execution time.',
     performanceShape,
     async ({ action }) => {
-      const { client } = requireContext(context);
-      const result = await executePerformance(client, context.traceState, {
-        action
-      });
+      const result = await requireDriver(context).performance({ action });
       return {
         content: [{ type: 'text' as const, text: result.text }]
       };

package/src/mcp/tools/snapshot.ts

@@ -1,8 +1,7 @@
 import { z } from 'zod';
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { BrowserContext } from '../server.js';
-import { executeSnapshot } from '../../commands/snapshot.js';
-import { executeFind } from '../../commands/find.js';
+import { requireDriver } from './driver-context.js';
 
 const findShape = {
   role: z
@@ -23,15 +22,6 @@ const findShape = {
     )
 };
 
-function requireContext(context: BrowserContext) {
-  const manager = context.manager;
-  if (!manager || !manager.isConnected()) {
-    throw new Error('Browser not started. Call browser_start first.');
-  }
-  const client = manager.getClient()!;
-  return { client, refMap: context.elementRefMap };
-}
-
 export function registerSnapshotTools(
   server: McpServer,
   context: BrowserContext
@@ -41,8 +31,7 @@ export function registerSnapshotTools(
     'Capture a structured accessibility snapshot of the current page. Returns interactive and meaningful elements with short refs (e1, e2, ...) that can be used with browser_click and other interaction tools instead of CSS selectors.',
     {},
     async () => {
-      const { client, refMap } = requireContext(context);
-      const result = await executeSnapshot(client, refMap);
+      const result = await requireDriver(context).snapshot();
       return {
         content: [{ type: 'text' as const, text: result.text }]
       };
@@ -54,8 +43,7 @@ export function registerSnapshotTools(
     'Find elements in the current page matching a query. Takes a new snapshot and filters by role, name, or text. Returns matching elements with refs.',
     findShape,
     async ({ role, name, text }) => {
-      const { client, refMap } = requireContext(context);
-      const result = await executeFind(client, refMap, { role, name, text });
+      const result = await requireDriver(context).find({ role, name, text });
       return {
         content: [{ type: 'text' as const, text: result.text }]
       };

package/src/mcp/tools/styles.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { BrowserContext } from '../server.js';
-import { executeStyles } from '../../commands/styles.js';
+import { requireDriver } from './driver-context.js';
 
 const stylesShape = {
   ref: z.string().describe('Element ref from browser_snapshot (e.g. "e3")'),
@@ -13,15 +13,6 @@ const stylesShape = {
     )
 };
 
-function requireContext(context: BrowserContext) {
-  const manager = context.manager;
-  if (!manager || !manager.isConnected()) {
-    throw new Error('Browser not started. Call browser_start first.');
-  }
-  const client = manager.getClient()!;
-  return { client, refMap: context.elementRefMap };
-}
-
 export function registerStylesTools(
   server: McpServer,
   context: BrowserContext
@@ -31,8 +22,7 @@ export function registerStylesTools(
     'Inspect computed styles, matched CSS rules, and box model for an element by ref. Helps diagnose visibility issues (display:none, opacity:0), layout problems (margins, positioning, overflow), and identify which CSS rule is responsible for a property. Use browser_snapshot first to get refs.',
     stylesShape,
     async ({ ref, properties }) => {
-      const { client, refMap } = requireContext(context);
-      const result = await executeStyles(client, refMap, { ref, properties });
+      const result = await requireDriver(context).styles({ ref, properties });
       return {
         content: [{ type: 'text' as const, text: result.text }]
       };