machine-bridge-mcp 1.2.1 → 1.2.5

package/src/worker/observability.ts

@@ -5,8 +5,8 @@ const SENSITIVE_FIELD = /(?:authorization|cookie|credential|password|secret|toke
 export class WorkerObservability {
   private readonly startedAt = performance.now();
   private readonly requests = { total: 0, successful: 0, client_error: 0, server_error: 0 };
-  private readonly calls = { started: 0, completed: 0, failed: 0, cancelled: 0, timed_out: 0 };
-  private readonly sockets = { candidates: 0, authenticated: 0, disconnected: 0, protocol_errors: 0 };
+  private readonly calls = { started: 0, completed: 0, failed: 0, cancelled: 0, timed_out: 0, unmatched_results: 0 };
+  private readonly sockets = { candidates: 0, authenticated: 0, ready: 0, disconnected: 0, protocol_errors: 0 };
   private readonly errors = new Map<string, number>();
   private readonly tools = new Map<string, { started: number; completed: number; failed: number; active: number }>();
 
@@ -39,8 +39,11 @@ export class WorkerObservability {
     this.incrementError(code);
   }
 
+  unmatchedResult(): void { this.calls.unmatched_results += 1; }
+
   socketCandidate(): void { this.sockets.candidates += 1; }
   socketAuthenticated(): void { this.sockets.authenticated += 1; }
+  socketReady(): void { this.sockets.ready += 1; }
   socketDisconnected(): void { this.sockets.disconnected += 1; }
   socketProtocolError(code: string): void {
     this.sockets.protocol_errors += 1;

package/src/worker/pending-calls.ts

@@ -19,6 +19,8 @@ export interface PendingCallRecord {
   timeout: ReturnType<typeof setTimeout>;
   resolve: (value: unknown) => void;
   reject: (error: Error) => void;
+  signal?: AbortSignal;
+  abortHandler?: () => void;
 }
 
 interface RegisterPendingCall {
@@ -28,6 +30,8 @@ interface RegisterPendingCall {
   tool: string;
   timeoutMs: number;
   onTimeout: (record: PendingCallRecord) => Error;
+  signal?: AbortSignal;
+  onAbort?: (record: PendingCallRecord) => Error;
 }
 
 export class PendingCallRegistry {
@@ -63,6 +67,14 @@ export class PendingCallRegistry {
         catch { error = new Error("pending daemon call timed out"); }
         reject(error instanceof Error ? error : new Error("pending daemon call timed out"));
       }, input.timeoutMs);
+      const abortHandler = () => {
+        const record = this.take(input.id);
+        if (!record) return;
+        let error: unknown;
+        try { error = input.onAbort?.(record); }
+        catch { error = new Error("pending daemon call was cancelled"); }
+        reject(error instanceof Error ? error : new Error("pending daemon call was cancelled"));
+      };
       const record: PendingCallRecord = {
         id: input.id,
         socket: input.socket,
@@ -72,9 +84,13 @@ export class PendingCallRegistry {
         timeout,
         resolve,
         reject,
+        signal: input.signal,
+        abortHandler: input.signal ? abortHandler : undefined,
       };
       this.byId.set(input.id, record);
       if (input.clientRequestKey) this.byRequestKey.set(input.clientRequestKey, input.id);
+      if (input.signal?.aborted) abortHandler();
+      else input.signal?.addEventListener("abort", abortHandler, { once: true });
     });
   }
 
@@ -131,6 +147,7 @@ export class PendingCallRegistry {
     const record = this.byId.get(id);
     if (!record) return undefined;
     clearTimeout(record.timeout);
+    if (record.signal && record.abortHandler) record.signal.removeEventListener("abort", record.abortHandler);
     this.byId.delete(id);
     if (record.clientRequestKey && this.byRequestKey.get(record.clientRequestKey) === id) {
       this.byRequestKey.delete(record.clientRequestKey);

package/wrangler.jsonc

@@ -3,7 +3,7 @@
   "name": "machine-bridge-mcp",
   "main": "src/worker/index.ts",
   "compatibility_date": "2026-07-11",
-  "compatibility_flags": ["nodejs_compat"],
+  "compatibility_flags": ["nodejs_compat", "enable_request_signal", "request_signal_passthrough"],
   "workers_dev": true,
   "durable_objects": {
     "bindings": [