@atelierai/uco 1.0.0

package/dist/server/ws/hub.js

@@ -0,0 +1,386 @@
+/**
+ * WebSocket hub — accepts plugin connections at /hub/mcp-server.
+ *
+ * Responsibilities:
+ *   - Validate auth token from query param (?access_token=) or header
+ *   - Register connection in registry
+ *   - Dispatch incoming messages: parse envelope → route to RPC handler or
+ *     resolve pending REST-forwarded request
+ *   - Handle close/error: unregister connection, reject pending requests
+ *   - Send OnInitialClientData notification on connect
+ *   - Broadcast OnMcpClientConnected / OnMcpClientDisconnected notifications
+ */
+import { WebSocketServer } from 'ws';
+import { URL } from 'node:url';
+import { timingSafeEqual } from 'node:crypto';
+import { hasForbiddenOrigin } from '../origin.js';
+import { serverRpcHandlers } from './rpc.js';
+import { parseMessage, makeErrorResponse, makeResponse, makeNotification, serializeMessage, ErrorCode, isRequest, isResponse, isNotification, extractDurableOperationId, } from './envelope.js';
+import { isToolCallControlError, toStructuredToolCallError, } from '../../tool-call-control.js';
+import { HUB_PATH, ClientFacingMethod, HEARTBEAT_METHOD, } from '../types.js';
+export class McpHub {
+    wss = null;
+    registry;
+    pending;
+    authToken;
+    serverApiVersion;
+    serverVersion;
+    heartbeatIntervalMs;
+    clientData;
+    heartbeatTimer = null;
+    onToolsUpdated;
+    onPromptsUpdated;
+    onResourcesUpdated;
+    onPluginConnected;
+    onPluginDisconnected;
+    onCallRecordCompleted;
+    ownedRuntime;
+    constructor(opts) {
+        this.registry = opts.registry;
+        this.pending = opts.pending;
+        this.authToken = opts.authToken;
+        this.serverApiVersion = opts.serverApiVersion;
+        this.serverVersion = opts.serverVersion;
+        this.heartbeatIntervalMs = opts.heartbeatIntervalMs;
+        this.clientData = opts.clientData ?? [];
+        this.onToolsUpdated = opts.onToolsUpdated;
+        this.onPromptsUpdated = opts.onPromptsUpdated;
+        this.onResourcesUpdated = opts.onResourcesUpdated;
+        this.onPluginConnected = opts.onPluginConnected;
+        this.onPluginDisconnected = opts.onPluginDisconnected;
+        this.onCallRecordCompleted = opts.onCallRecordCompleted;
+        this.ownedRuntime = opts.ownedRuntime;
+    }
+    /**
+     * Attach to an existing HTTP server. Handles WebSocket upgrades at HUB_PATH only.
+     */
+    attach(server) {
+        this.wss = new WebSocketServer({ noServer: true });
+        this.heartbeatTimer = setInterval(() => {
+            this.broadcastNotification(HEARTBEAT_METHOD, {
+                timestampUtc: new Date().toISOString(),
+            });
+        }, this.heartbeatIntervalMs);
+        server.on('upgrade', (req, socket, head) => {
+            const { pathname, searchParams } = this.parseUrl(req);
+            if (pathname !== HUB_PATH) {
+                // Not our route — let other upgrade handlers deal with it (or destroy).
+                return;
+            }
+            if (hasForbiddenOrigin(req)) {
+                const body = JSON.stringify({
+                    ok: false,
+                    error: {
+                        code: 'forbidden-origin',
+                        message: 'Browser Origin requests are not allowed.',
+                        retryable: false,
+                    },
+                });
+                socket.write('HTTP/1.1 403 Forbidden\r\n' +
+                    'Connection: close\r\n' +
+                    'Content-Type: application/json\r\n' +
+                    `Content-Length: ${Buffer.byteLength(body)}\r\n\r\n${body}`);
+                socket.destroy();
+                return;
+            }
+            if (this.ownedRuntime?.phase === 'precommit') {
+                const body = JSON.stringify({
+                    ok: false,
+                    error: {
+                        code: 'owned-server-precommit',
+                        message: 'The uco-owned server is awaiting handoff commit.',
+                        retryable: true,
+                    },
+                });
+                socket.write('HTTP/1.1 503 Service Unavailable\r\n' +
+                    'Connection: close\r\n' +
+                    'Content-Type: application/json\r\n' +
+                    `Content-Length: ${Buffer.byteLength(body)}\r\n\r\n${body}`);
+                socket.destroy();
+                return;
+            }
+            // Auth check.
+            if (this.authToken) {
+                const token = searchParams.get('access_token') ?? this.extractBearerToken(req);
+                if (!tokensMatch(token, this.authToken)) {
+                    socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
+                    socket.destroy();
+                    return;
+                }
+            }
+            this.wss.handleUpgrade(req, socket, head, (ws) => {
+                // Extract instanceId from query or header.
+                const instanceId = searchParams.get('instanceId')
+                    ?? req.headers['mcp-instance-id']?.toString()
+                    ?? undefined;
+                const token = this.authToken
+                    ? (searchParams.get('access_token') ?? this.extractBearerToken(req) ?? undefined)
+                    : undefined;
+                this.onConnection(ws, token, instanceId);
+            });
+        });
+    }
+    onConnection(ws, token, instanceId) {
+        const connectionId = this.registry.register(ws, { token, instanceId });
+        // Send OnInitialClientData notification.
+        const notif = makeNotification(ClientFacingMethod.OnInitialClientData, {
+            clients: this.clientData,
+        });
+        this.safeSend(ws, notif);
+        // Tell every connection (this one included) that a new plugin joined.
+        // Params shape mirrors the C# OnMcpClientConnectedParams wrapper (camelCase):
+        // { connected: McpClientData, all: McpClientData[] }.
+        const connectedData = this.clientDataFor(this.registry.get(connectionId));
+        if (connectedData) {
+            this.broadcastNotification(ClientFacingMethod.OnMcpClientConnected, {
+                connected: connectedData,
+                all: this.connectionsClientData(),
+            });
+        }
+        // Track whether the disconnect callback has already fired (socket error
+        // is typically followed by close — guard against double-fire).
+        let disconnected = false;
+        const onDisconnect = (reason) => {
+            if (disconnected)
+                return;
+            disconnected = true;
+            // Capture the leaving client's data before unregistering — the registry
+            // entry is gone afterwards.
+            const leaving = this.clientDataFor(this.registry.get(connectionId));
+            this.registry.unregister(connectionId);
+            this.pending.rejectAllForConnection(connectionId, {
+                code: ErrorCode.PLUGIN_NOT_CONNECTED,
+                message: reason,
+            });
+            this.onPluginDisconnected?.(connectionId);
+            // Tell the remaining connections this plugin went away. When the entry
+            // is already gone (e.g. ForceDisconnect-replaced in single-plugin mode)
+            // there is nothing meaningful to report — skip.
+            // Params shape mirrors the C# OnMcpClientDisconnectedParams wrapper:
+            // { disconnected: McpClientData, remaining: McpClientData[] }.
+            if (leaving) {
+                this.broadcastNotification(ClientFacingMethod.OnMcpClientDisconnected, {
+                    disconnected: { ...leaving, isConnected: false },
+                    remaining: this.connectionsClientData(),
+                });
+            }
+        };
+        ws.on('message', (data, isBinary) => {
+            const text = Buffer.isBuffer(data)
+                ? data.toString('utf8')
+                : Array.isArray(data)
+                    ? Buffer.concat(data).toString('utf8')
+                    : String(data);
+            this.onMessage(ws, connectionId, text);
+        });
+        ws.on('close', () => {
+            onDisconnect('Plugin connection closed.');
+        });
+        ws.on('error', () => {
+            onDisconnect('Plugin connection error.');
+        });
+    }
+    onMessage(ws, connectionId, text) {
+        // Debug: log every incoming raw frame (truncated) to diagnose wire-format mismatches.
+        // Set WS_DEBUG to any non-empty value to enable.
+        if (process.env.WS_DEBUG) {
+            console.log(`[WS>server] ${connectionId.slice(0, 8)}: ${text.slice(0, 500)}`);
+        }
+        const parsed = parseMessage(text);
+        if (!parsed.ok) {
+            // Parse error — respond with error code (no id → use 0).
+            this.safeSend(ws, makeErrorResponse(0, parsed.error.code, parsed.error.message));
+            return;
+        }
+        const msg = parsed.message;
+        // Response: resolve pending REST-forwarded request.
+        if (isResponse(msg)) {
+            if (msg.error) {
+                this.pending.reject(msg.id, msg.error);
+            }
+            else {
+                const operationId = extractDurableOperationId(msg.result);
+                if (operationId)
+                    this.pending.bindOperation(msg.id, operationId);
+                this.pending.resolve(msg.id, msg.result);
+            }
+            return;
+        }
+        // Request: dispatch to server-facing RPC handler.
+        if (isRequest(msg)) {
+            this.dispatchRequest(ws, connectionId, msg.id, msg.method, msg.params);
+            return;
+        }
+        // Notification (no id): process but don't respond.
+        if (isNotification(msg)) {
+            this.handleNotification(connectionId, msg.method, msg.params);
+            return;
+        }
+    }
+    dispatchRequest(ws, connectionId, id, method, params) {
+        const handler = serverRpcHandlers[method];
+        if (!handler) {
+            this.safeSend(ws, makeErrorResponse(id, ErrorCode.METHOD_NOT_FOUND, `Method not found: ${method}`));
+            return;
+        }
+        const ctx = {
+            connectionId,
+            registry: this.registry,
+            pending: this.pending,
+            serverApiVersion: this.serverApiVersion,
+            serverVersion: this.serverVersion,
+            clientData: this.clientData,
+            onToolsUpdated: this.onToolsUpdated,
+            onPromptsUpdated: this.onPromptsUpdated,
+            onResourcesUpdated: this.onResourcesUpdated,
+            onPluginConnected: this.onPluginConnected,
+            onPluginDisconnected: this.onPluginDisconnected,
+            onCallRecordCompleted: this.onCallRecordCompleted,
+        };
+        try {
+            const result = handler(params, ctx);
+            if (result instanceof Promise) {
+                result
+                    .then((val) => this.safeSend(ws, makeResponse(id, val)))
+                    .catch((err) => {
+                    this.safeSend(ws, makeHandlerErrorResponse(id, err));
+                });
+            }
+            else {
+                this.safeSend(ws, makeResponse(id, result));
+            }
+        }
+        catch (err) {
+            this.safeSend(ws, makeHandlerErrorResponse(id, err));
+        }
+    }
+    handleNotification(connectionId, method, _params) {
+        // Notifications from the plugin are handled here (currently none of the 21 methods
+        // are plugin→server notifications — all 7 server-facing methods are request-response).
+        // This is a no-op for unknown notifications.
+        void connectionId;
+        void method;
+        void _params;
+    }
+    /** Send a message to a specific connection by connectionId. */
+    sendToConnection(connectionId, msg) {
+        const entry = this.registry.get(connectionId);
+        if (!entry || entry.ws.readyState !== entry.ws.OPEN)
+            return false;
+        entry.ws.send(serializeMessage(msg));
+        return true;
+    }
+    /** Close the WebSocket server. */
+    close() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+            this.heartbeatTimer = null;
+        }
+        if (this.wss) {
+            for (const client of this.wss.clients) {
+                client.close();
+            }
+            this.wss.close();
+            this.wss = null;
+        }
+    }
+    // ===== Helpers =====
+    /**
+     * Map a registry entry to the client-facing McpClientData shape
+     * (camelCase, mirrors the C# McpClientData [JsonPropertyName] attributes).
+     */
+    clientDataFor(entry) {
+        if (!entry)
+            return undefined;
+        return {
+            isConnected: true,
+            // instanceId is the stable identity plugins know; fall back to the
+            // server-assigned connection id.
+            sessionId: entry.instanceId ?? entry.connectionId,
+        };
+    }
+    /** Snapshot of all live connections as McpClientData. */
+    connectionsClientData() {
+        const out = [];
+        for (const entry of this.registry.snapshot()) {
+            const data = this.clientDataFor(entry);
+            if (data)
+                out.push(data);
+        }
+        return out;
+    }
+    /** Send a notification to every open connection. */
+    broadcastNotification(method, params) {
+        const notif = makeNotification(method, params);
+        for (const entry of this.registry.snapshot()) {
+            this.safeSend(entry.ws, notif);
+        }
+    }
+    safeSend(ws, msg) {
+        if (ws.readyState === ws.OPEN) {
+            const text = serializeMessage(msg);
+            // Debug: log every outgoing frame (truncated) to diagnose wire-format mismatches.
+            // Set WS_DEBUG to any non-empty value to enable.
+            if (process.env.WS_DEBUG) {
+                console.log(`[server>WS] ${text.slice(0, 500)}`);
+            }
+            ws.send(text);
+        }
+    }
+    parseUrl(req) {
+        const fullUrl = req.url ?? '';
+        try {
+            const parsed = new URL(fullUrl, 'http://localhost');
+            return { pathname: parsed.pathname, searchParams: parsed.searchParams };
+        }
+        catch {
+            return { pathname: fullUrl, searchParams: new URLSearchParams() };
+        }
+    }
+    extractBearerToken(req) {
+        const auth = req.headers['authorization'];
+        if (!auth)
+            return null;
+        const match = /^Bearer\s+(.+)$/i.exec(auth);
+        return match ? match[1] : null;
+    }
+}
+function makeHandlerErrorResponse(id, error) {
+    if (!isToolCallControlError(error)) {
+        const message = error instanceof Error ? error.message : String(error);
+        return makeErrorResponse(id, ErrorCode.INTERNAL_ERROR, message);
+    }
+    const structured = toStructuredToolCallError(error);
+    return makeErrorResponse(id, structuredRpcCode(structured), structured.message, structured);
+}
+function structuredRpcCode(error) {
+    switch (error.code) {
+        case 'invalid_control':
+        case 'unsupported_control_version':
+            return ErrorCode.INVALID_PARAMS;
+        case 'cancelled':
+            return ErrorCode.CALL_CANCELLED;
+        case 'deadline_exceeded':
+            return ErrorCode.CALL_DEADLINE_EXCEEDED;
+        case 'middleware_rejected':
+            return ErrorCode.MIDDLEWARE_REJECTED;
+        case 'tool_execution_failed':
+        default:
+            return ErrorCode.INTERNAL_ERROR;
+    }
+}
+/**
+ * Constant-time string comparison to mitigate timing attacks on auth tokens.
+ * Returns true when the strings are equal. When lengths differ, returns
+ * false immediately (timingSafeEqual throws on unequal-length buffers).
+ */
+function tokensMatch(a, b) {
+    if (a === null)
+        return false;
+    const aBuf = Buffer.from(a);
+    const bBuf = Buffer.from(b);
+    if (aBuf.length !== bBuf.length)
+        return false;
+    return timingSafeEqual(aBuf, bBuf);
+}
+//# sourceMappingURL=hub.js.map

package/dist/server/ws/hub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hub.js","sourceRoot":"","sources":["../../../src/server/ws/hub.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,eAAe,EAAa,MAAM,IAAI,CAAC;AAGhD,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAKlD,OAAO,EAAE,iBAAiB,EAAmB,MAAM,UAAU,CAAC;AAC9D,OAAO,EACL,YAAY,EACZ,iBAAiB,EACjB,YAAY,EACZ,gBAAgB,EAChB,gBAAgB,EAChB,SAAS,EACT,SAAS,EACT,UAAU,EACV,cAAc,EACd,yBAAyB,GAC1B,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,sBAAsB,EACtB,yBAAyB,GAE1B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,QAAQ,EACR,kBAAkB,EAClB,gBAAgB,GAEjB,MAAM,aAAa,CAAC;AAoBrB,MAAM,OAAO,MAAM;IACT,GAAG,GAA2B,IAAI,CAAC;IAC1B,QAAQ,CAAqB;IAC7B,OAAO,CAAiB;IACxB,SAAS,CAAU;IACnB,gBAAgB,CAAS;IACzB,aAAa,CAAS;IACtB,mBAAmB,CAAS;IAC5B,UAAU,CAAkB;IACrC,cAAc,GAA0C,IAAI,CAAC;IAEpD,cAAc,CAAoD;IAClE,gBAAgB,CAAoD;IACpE,kBAAkB,CAAwD;IAC1E,iBAAiB,CAA0G;IAC3H,oBAAoB,CAAkC;IACtD,qBAAqB,CAAmG;IACxH,YAAY,CAAsB;IAEnD,YAAY,IAAgB;QAC1B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC9C,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,CAAC;QACpD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC9C,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;QAClD,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAChD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,oBAAoB,CAAC;QACtD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACxD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,MAAc;QACnB,IAAI,CAAC,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE;gBAC3C,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACvC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAE7B,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;YAC1D,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YACtD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC1B,wEAAwE;gBACxE,OAAO;YACT,CAAC;YAED,IAAI,kBAAkB,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC1B,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE;wBACL,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,0CAA0C;wBACnD,SAAS,EAAE,KAAK;qBACjB;iBACF,CAAC,CAAC;gBACH,MAAM,CAAC,KAAK,CACV,4BAA4B;oBAC5B,uBAAuB;oBACvB,oCAAoC;oBACpC,mBAAmB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAC5D,CAAC;gBACF,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,IAAI,IAAI,CAAC,YAAY,EAAE,KAAK,KAAK,WAAW,EAAE,CAAC;gBAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC1B,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE;wBACL,IAAI,EAAE,wBAAwB;wBAC9B,OAAO,EAAE,kDAAkD;wBAC3D,SAAS,EAAE,IAAI;qBAChB;iBACF,CAAC,CAAC;gBACH,MAAM,CAAC,KAAK,CACV,sCAAsC;oBACtC,uBAAuB;oBACvB,oCAAoC;oBACpC,mBAAmB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAC5D,CAAC;gBACF,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,cAAc;YACd,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,MAAM,KAAK,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;gBAC/E,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;oBACxC,MAAM,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;oBAClD,MAAM,CAAC,OAAO,EAAE,CAAC;oBACjB,OAAO;gBACT,CAAC;YACH,CAAC;YAED,IAAI,CAAC,GAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAa,EAAE,EAAE;gBAC3D,2CAA2C;gBAC3C,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC;uBAC5C,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE;uBAC1C,SAAS,CAAC;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS;oBAC1B,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC;oBACjF,CAAC,CAAC,SAAS,CAAC;gBAEd,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,YAAY,CAAC,EAAa,EAAE,KAAyB,EAAE,UAA8B;QAC3F,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAEvE,yCAAyC;QACzC,MAAM,KAAK,GAAG,gBAAgB,CAAC,kBAAkB,CAAC,mBAAmB,EAAE;YACrE,OAAO,EAAE,IAAI,CAAC,UAAU;SACzB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAEzB,sEAAsE;QACtE,8EAA8E;QAC9E,sDAAsD;QACtD,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;QAC1E,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,oBAAoB,EAAE;gBAClE,SAAS,EAAE,aAAa;gBACxB,GAAG,EAAE,IAAI,CAAC,qBAAqB,EAAE;aAClC,CAAC,CAAC;QACL,CAAC;QAED,wEAAwE;QACxE,+DAA+D;QAC/D,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,MAAM,YAAY,GAAG,CAAC,MAAc,EAAQ,EAAE;YAC5C,IAAI,YAAY;gBAAE,OAAO;YACzB,YAAY,GAAG,IAAI,CAAC;YACpB,wEAAwE;YACxE,4BAA4B;YAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;YACpE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YACvC,IAAI,CAAC,OAAO,CAAC,sBAAsB,CAAC,YAAY,EAAE;gBAChD,IAAI,EAAE,SAAS,CAAC,oBAAoB;gBACpC,OAAO,EAAE,MAAM;aAChB,CAAC,CAAC;YACH,IAAI,CAAC,oBAAoB,EAAE,CAAC,YAAY,CAAC,CAAC;YAC1C,uEAAuE;YACvE,wEAAwE;YACxE,gDAAgD;YAChD,qEAAqE;YACrE,+DAA+D;YAC/D,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,uBAAuB,EAAE;oBACrE,YAAY,EAAE,EAAE,GAAG,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE;oBAChD,SAAS,EAAE,IAAI,CAAC,qBAAqB,EAAE;iBACxC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC;QAEF,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAuB,EAAE,QAAiB,EAAE,EAAE;YAC9D,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAChC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACvB,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;oBACnB,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;oBACtC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACnB,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,YAAY,CAAC,2BAA2B,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,YAAY,CAAC,0BAA0B,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,SAAS,CAAC,EAAa,EAAE,YAAoB,EAAE,IAAY;QACjE,sFAAsF;QACtF,iDAAiD;QACjD,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,eAAe,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QAElC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,yDAAyD;YACzD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,iBAAiB,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;YACjF,OAAO;QACT,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC;QAE3B,oDAAoD;QACpD,IAAI,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACpB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACd,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;YACzC,CAAC;iBAAM,CAAC;gBACN,MAAM,WAAW,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC1D,IAAI,WAAW;oBAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;gBACjE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;YAC3C,CAAC;YACD,OAAO;QACT,CAAC;QAED,kDAAkD;QAClD,IAAI,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;YACnB,IAAI,CAAC,eAAe,CAAC,EAAE,EAAE,YAAY,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;YACvE,OAAO;QACT,CAAC;QAED,mDAAmD;QACnD,IAAI,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;YAC9D,OAAO;QACT,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,EAAa,EACb,YAAoB,EACpB,EAAmB,EACnB,MAAc,EACd,MAAe;QAEf,MAAM,OAAO,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,iBAAiB,CAAC,EAAE,EAAE,SAAS,CAAC,gBAAgB,EAAE,qBAAqB,MAAM,EAAE,CAAC,CAAC,CAAC;YACpG,OAAO;QACT,CAAC;QAED,MAAM,GAAG,GAAe;YACtB,YAAY;YACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;YAC3C,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;YAC/C,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;SAClD,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YACpC,IAAI,MAAM,YAAY,OAAO,EAAE,CAAC;gBAC9B,MAAM;qBACH,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;qBACvD,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;oBACtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,wBAAwB,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;gBACvD,CAAC,CAAC,CAAC;YACP,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,wBAAwB,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,YAAoB,EAAE,MAAc,EAAE,OAAgB;QAC/E,mFAAmF;QACnF,uFAAuF;QACvF,6CAA6C;QAC7C,KAAK,YAAY,CAAC;QAClB,KAAK,MAAM,CAAC;QACZ,KAAK,OAAO,CAAC;IACf,CAAC;IAED,+DAA+D;IAC/D,gBAAgB,CAAC,YAAoB,EAAE,GAAY;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,EAAE,CAAC,UAAU,KAAK,KAAK,CAAC,EAAE,CAAC,IAAI;YAAE,OAAO,KAAK,CAAC;QAClE,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAY,CAAC,CAAC,CAAC;QAC9C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kCAAkC;IAClC,KAAK;QACH,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;gBACtC,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,CAAC;YACD,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;YACjB,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;QAClB,CAAC;IACH,CAAC;IAED,sBAAsB;IAEtB;;;OAGG;IACK,aAAa,CAAC,KAAkC;QACtD,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,OAAO;YACL,WAAW,EAAE,IAAI;YACjB,mEAAmE;YACnE,iCAAiC;YACjC,SAAS,EAAE,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,YAAY;SAClD,CAAC;IACJ,CAAC;IAED,yDAAyD;IACjD,qBAAqB;QAC3B,MAAM,GAAG,GAAoB,EAAE,CAAC;QAChC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YACvC,IAAI,IAAI;gBAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,oDAAoD;IAC5C,qBAAqB,CAAC,MAAc,EAAE,MAAe;QAC3D,MAAM,KAAK,GAAG,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/C,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAEO,QAAQ,CAAC,EAAa,EAAE,GAAY;QAC1C,IAAI,EAAE,CAAC,UAAU,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAY,CAAC,CAAC;YAC5C,kFAAkF;YAClF,iDAAiD;YACjD,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YACnD,CAAC;YACD,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;IACH,CAAC;IAEO,QAAQ,CAAC,GAAoB;QACnC,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC;YACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC;QAC1E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,eAAe,EAAE,EAAE,CAAC;QACpE,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,GAAoB;QAC7C,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QACvB,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACjC,CAAC;CACF;AAED,SAAS,wBAAwB,CAAC,EAAmB,EAAE,KAAc;IACnE,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,OAAO,iBAAiB,CAAC,EAAE,EAAE,SAAS,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,UAAU,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;IACpD,OAAO,iBAAiB,CACtB,EAAE,EACF,iBAAiB,CAAC,UAAU,CAAC,EAC7B,UAAU,CAAC,OAAO,EAClB,UAAU,CACX,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,KAA8B;IACvD,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,iBAAiB,CAAC;QACvB,KAAK,6BAA6B;YAChC,OAAO,SAAS,CAAC,cAAc,CAAC;QAClC,KAAK,WAAW;YACd,OAAO,SAAS,CAAC,cAAc,CAAC;QAClC,KAAK,mBAAmB;YACtB,OAAO,SAAS,CAAC,sBAAsB,CAAC;QAC1C,KAAK,qBAAqB;YACxB,OAAO,SAAS,CAAC,mBAAmB,CAAC;QACvC,KAAK,uBAAuB,CAAC;QAC7B;YACE,OAAO,SAAS,CAAC,cAAc,CAAC;IACpC,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,CAAgB,EAAE,CAAS;IAC9C,IAAI,CAAC,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC;IAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC5B,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IAC9C,OAAO,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;AACrC,CAAC"}

package/dist/server/ws/pending.d.ts

@@ -0,0 +1,47 @@
+/**
+ * Bounded pending request correlation. Unity owns durable operations; every map
+ * here is ephemeral and is removed by one atomic settle path.
+ */
+import type { RpcError } from './envelope.js';
+export declare const MAX_PENDING_PROCESS = 1024;
+export declare const MAX_PENDING_PER_CONNECTION = 256;
+export interface PendingIdentity {
+    envelopeId: string | number;
+    requestID?: string;
+    operationID?: string;
+    callId?: string;
+    cancellationId?: string;
+    connectionId?: string;
+    generation?: number;
+}
+export interface PendingTrackOptions {
+    signal?: AbortSignal;
+    timeoutError?: RpcError;
+    abortError?: RpcError;
+    requestID?: string;
+    operationID?: string;
+    callId?: string;
+    cancellationId?: string;
+    generation?: number;
+    onAbortAfterDispatch?: (entry: PendingIdentity) => void;
+    onResolved?: (result: unknown, entry: PendingIdentity) => void;
+}
+export declare class PendingTracker {
+    private readonly byEnvelopeId;
+    private readonly byRequestID;
+    private readonly byOperationID;
+    private readonly byConnection;
+    track(id: string | number, timeoutMs: number, method: string, connectionId?: string, options?: PendingTrackOptions | AbortSignal): Promise<unknown>;
+    markDispatched(id: string | number): boolean;
+    trackDeferred(requestID: string, envelopeId: string | number, operationID?: string): void;
+    resolve(id: string | number, result: unknown): boolean;
+    bindOperation(id: string | number, operationID: string): boolean;
+    resolveDeferred(requestID: string, result: unknown, operationID?: string): boolean;
+    reject(id: string | number, error: RpcError): boolean;
+    rejectAllForConnection(connectionId: string, error: RpcError): void;
+    get size(): number;
+    private settle;
+    private removeEntry;
+    private addIndex;
+    private removeIndex;
+}