Drop desktop MCP support, add server WS MCP support

Author: pimterry

src/services/desktop-api.ts

@@ -1,5 +1,4 @@
 import { desktopVersion } from "./service-versions";
-import { OperationDefinition } from "./ui-api/api-types";
 
 type DesktopInjectedKey =
     | 'httpToolkitDesktopVersion'
@@ -57,11 +56,6 @@ interface DesktopApi {
     getPathForFile?: (file: File) => string | null;
 
     setComponentVersions?: (versions: Record<string, string>) => void;
-
-    setApiOperations?: (operations: OperationDefinition[]) => void;
-    onOperationRequest?: (
-        handler: (operation: string, params: Record<string, unknown>) => Promise<unknown>
-    ) => void;
 }
 
 interface NativeContextMenuDefinition {

src/services/ui-api/api-interface.ts

@@ -1,7 +1,7 @@
 import { OperationRegistry } from './api-registry';
 import { registerAllOperations } from './operations';
+import { startServerOperationBridge } from './server-operation-bridge';
 
-import { DesktopApi } from '../desktop-api';
 import { AccountStore } from '../../model/account/account-store';
 import { EventsStore } from '../../model/events/events-store';
 import { ProxyStore } from '../../model/proxy-store';
@@ -13,11 +13,6 @@ export function initializeUiApi(stores: {
     proxyStore: ProxyStore;
     interceptorStore: InterceptorStore;
 }) {
-    if (!DesktopApi.setApiOperations || !DesktopApi.onOperationRequest) {
-        console.log("UI API not available");
-        return;
-    }
-
     const { accountStore, eventsStore, proxyStore, interceptorStore } = stores;
 
     const registry = new OperationRegistry(
@@ -30,8 +25,7 @@ export function initializeUiApi(stores: {
         () => eventsStore.events
     );
 
-    DesktopApi.setApiOperations(registry.getDefinitions());
-    DesktopApi.onOperationRequest(async (operation, params) => {
-        return await registry.execute(operation, params || {});
-    });
+    // Connect to the server's WebSocket bridge (for MCP and external tool access).
+    const authToken = new URLSearchParams(window.location.search).get('authToken') ?? undefined;
+    startServerOperationBridge(registry, authToken);
 }

src/services/ui-api/server-operation-bridge.ts

@@ -0,0 +1,107 @@
+import { OperationRegistry } from './api-registry';
+
+const RECONNECT_BASE_MS = 1_000;
+const RECONNECT_MAX_MS = 30_000;
+
+export function startServerOperationBridge(
+    registry: OperationRegistry,
+    authToken: string | undefined
+) {
+    let ws: WebSocket | null = null;
+    let reconnectDelay = RECONNECT_BASE_MS;
+    let reconnectTimer: ReturnType<typeof setTimeout> | null = null;
+    let destroyed = false;
+
+    function connect() {
+        if (destroyed) return;
+
+        const url = authToken
+            ? `ws://127.0.0.1:45457/ui-operations?token=${encodeURIComponent(authToken)}`
+            : `ws://127.0.0.1:45457/ui-operations`;
+
+        try {
+            ws = new WebSocket(url);
+        } catch {
+            scheduleReconnect();
+            return;
+        }
+
+        ws.onopen = () => {
+            reconnectDelay = RECONNECT_BASE_MS;
+            sendOperations();
+        };
+
+        ws.onmessage = (event) => {
+            try {
+                const msg = JSON.parse(event.data as string);
+                handleMessage(msg);
+            } catch {}
+        };
+
+        ws.onclose = () => {
+            ws = null;
+            scheduleReconnect();
+        };
+
+        ws.onerror = () => {
+            // Error will be followed by close
+        };
+    }
+
+    function scheduleReconnect() {
+        if (destroyed || reconnectTimer) return;
+        reconnectTimer = setTimeout(() => {
+            reconnectTimer = null;
+            reconnectDelay = Math.min(reconnectDelay * 2, RECONNECT_MAX_MS);
+            connect();
+        }, reconnectDelay);
+    }
+
+    function sendOperations() {
+        if (!ws || ws.readyState !== WebSocket.OPEN) return;
+        ws.send(JSON.stringify({
+            type: 'operations',
+            operations: registry.getDefinitions()
+        }));
+    }
+
+    function handleMessage(msg: any) {
+        if (!msg || msg.type !== 'request') return;
+
+        const { id, operation, params } = msg;
+
+        registry.execute(operation, params || {}).then((result) => {
+            if (ws && ws.readyState === WebSocket.OPEN) {
+                ws.send(JSON.stringify({
+                    type: 'response',
+                    id,
+                    result
+                }));
+            }
+        }).catch((err) => {
+            if (ws && ws.readyState === WebSocket.OPEN) {
+                ws.send(JSON.stringify({
+                    type: 'response',
+                    id,
+                    error: err.message || String(err)
+                }));
+            }
+        });
+    }
+
+    connect();
+
+    return {
+        destroy() {
+            destroyed = true;
+            if (reconnectTimer) {
+                clearTimeout(reconnectTimer);
+                reconnectTimer = null;
+            }
+            if (ws) {
+                ws.close();
+                ws = null;
+            }
+        }
+    };
+}