@everydaydevopsio/pilot 0.6.2 → 0.7.0

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 }]
       };