@kadi.build/core 0.0.1-alpha.1 → 0.0.1-alpha.10

package/dist/broker/BrokerConnection.js

@@ -0,0 +1,434 @@
+/**
+ * Broker Connection
+ *
+ * Wraps a single WebSocket connection to a KADI broker.
+ * Handles connection lifecycle, message sending/receiving, and state management.
+ *
+ * @module broker/BrokerConnection
+ */
+import { EventEmitter } from 'events';
+import WebSocket from 'ws';
+import { ConnectionState } from '../types/index.js';
+import { KadiError, ErrorCode } from '../types/index.js';
+/**
+ * Broker Connection
+ *
+ * Manages a single WebSocket connection to a broker.
+ * Handles low-level connection lifecycle and message framing.
+ *
+ * Does NOT handle:
+ * - Protocol logic (handshake, auth) - that's BrokerProtocol's job
+ * - Multiple connections - that's BrokerConnectionManager's job
+ *
+ * @example
+ * ```typescript
+ * const connection = new BrokerConnection({
+ *   url: 'ws://localhost:8080',
+ *   name: 'local',
+ *   connectionTimeout: 10000
+ * });
+ *
+ * connection.on('connected', () => {
+ *   console.log('Connected to broker');
+ * });
+ *
+ * connection.on('message', (message) => {
+ *   console.log('Received:', message);
+ * });
+ *
+ * await connection.connect();
+ * ```
+ */
+export class BrokerConnection extends EventEmitter {
+    /**
+     * Broker configuration
+     */
+    config;
+    /**
+     * WebSocket instance
+     */
+    ws = null;
+    /**
+     * Current connection state
+     */
+    _state = ConnectionState.DISCONNECTED;
+    /**
+     * Assigned agent ID (set after handshake)
+     */
+    _agentId = null;
+    /**
+     * Heartbeat timer
+     */
+    heartbeatTimer = null;
+    /**
+     * Connection timestamp
+     */
+    _connectedAt = null;
+    /**
+     * Last heartbeat timestamp
+     */
+    _lastHeartbeat = null;
+    /**
+     * Pending requests (for matching responses)
+     */
+    pendingRequests = new Map();
+    /**
+     * Request ID counter
+     */
+    requestIdCounter = 0;
+    /**
+     * Create a new BrokerConnection
+     *
+     * @param config - Broker configuration
+     */
+    constructor(config) {
+        super();
+        this.config = {
+            ...config,
+            connectionTimeout: config.connectionTimeout ?? 10000,
+            requestTimeout: config.requestTimeout ?? 30000,
+            heartbeatInterval: config.heartbeatInterval ?? 30000,
+            autoReconnect: config.autoReconnect ?? true,
+            maxReconnectAttempts: config.maxReconnectAttempts ?? 5
+        };
+    }
+    /**
+     * Connect to the broker
+     *
+     * @throws {KadiError} If connection fails
+     *
+     * @example
+     * ```typescript
+     * await connection.connect();
+     * console.log('Connected!');
+     * ```
+     */
+    async connect() {
+        if (this._state !== ConnectionState.DISCONNECTED) {
+            throw new KadiError('Connection already in progress or established', ErrorCode.CONNECTION_FAILED, 400, { currentState: this._state, broker: this.config.name });
+        }
+        this.setState(ConnectionState.CONNECTING);
+        return new Promise((resolve, reject) => {
+            const timeout = setTimeout(() => {
+                this.cleanup();
+                reject(new KadiError(`Connection timeout after ${this.config.connectionTimeout}ms`, ErrorCode.CONNECTION_TIMEOUT, 408, { broker: this.config.name, url: this.config.url }));
+            }, this.config.connectionTimeout);
+            try {
+                this.ws = new WebSocket(this.config.url);
+                this.ws.on('open', () => {
+                    clearTimeout(timeout);
+                    this._connectedAt = Date.now();
+                    this.setState(ConnectionState.CONNECTED);
+                    this.setupHeartbeat();
+                    this.emit('connected');
+                    resolve();
+                });
+                this.ws.on('error', (error) => {
+                    clearTimeout(timeout);
+                    this.emit('error', error);
+                    reject(new KadiError(`WebSocket error: ${error.message}`, ErrorCode.CONNECTION_FAILED, 500, { broker: this.config.name, url: this.config.url, originalError: error.message }));
+                });
+                this.ws.on('close', (code, reason) => {
+                    clearTimeout(timeout);
+                    this.handleClose(code, reason.toString());
+                });
+                this.ws.on('message', (data) => {
+                    this.handleMessage(data);
+                });
+            }
+            catch (error) {
+                clearTimeout(timeout);
+                this.cleanup();
+                reject(KadiError.from(error, ErrorCode.CONNECTION_FAILED));
+            }
+        });
+    }
+    /**
+     * Disconnect from the broker
+     *
+     * @example
+     * ```typescript
+     * await connection.disconnect();
+     * ```
+     */
+    async disconnect() {
+        this.cleanup();
+    }
+    /**
+     * Send a JSON-RPC request and wait for response
+     *
+     * @template T - Response result type
+     * @param request - JSON-RPC request
+     * @returns Promise resolving to the response result
+     *
+     * @throws {KadiError} If request fails or times out
+     *
+     * @example
+     * ```typescript
+     * const result = await connection.sendRequest({
+     *   jsonrpc: '2.0',
+     *   method: 'kadi.handshake',
+     *   params: { name: 'my-agent' },
+     *   id: 1
+     * });
+     * ```
+     */
+    async sendRequest(request) {
+        if (!this.isConnected) {
+            throw new KadiError('Not connected to broker', ErrorCode.BROKER_NOT_CONNECTED, 503, { broker: this.config.name });
+        }
+        // Generate ID if not provided
+        if (request.id === undefined) {
+            request.id = ++this.requestIdCounter;
+        }
+        return new Promise((resolve, reject) => {
+            const timeout = setTimeout(() => {
+                this.pendingRequests.delete(request.id);
+                reject(new KadiError(`Request timeout after ${this.config.requestTimeout}ms`, ErrorCode.BROKER_TIMEOUT, 408, { broker: this.config.name, method: request.method, requestId: request.id }));
+            }, this.config.requestTimeout);
+            this.pendingRequests.set(request.id, {
+                id: request.id,
+                resolve: resolve,
+                reject,
+                timeout,
+                timestamp: Date.now()
+            });
+            this.send(request);
+        });
+    }
+    /**
+     * Send a JSON-RPC notification (no response expected)
+     *
+     * @param notification - JSON-RPC notification
+     *
+     * @example
+     * ```typescript
+     * connection.sendNotification({
+     *   jsonrpc: '2.0',
+     *   method: 'kadi.heartbeat',
+     *   params: { timestamp: Date.now() }
+     * });
+     * ```
+     */
+    sendNotification(notification) {
+        if (!this.isConnected) {
+            throw new KadiError('Not connected to broker', ErrorCode.BROKER_NOT_CONNECTED, 503, { broker: this.config.name });
+        }
+        this.send(notification);
+    }
+    /**
+     * Send a tool result back to broker as JSON-RPC response
+     *
+     * Used when this client acts as a tool provider and needs to respond to invocations.
+     * The requestId must match the incoming request's id field for proper routing.
+     *
+     * @param requestId - Request ID to respond to (CRITICAL: must match incoming request id)
+     * @param result - Result data to send back
+     */
+    sendResponse(requestId, result) {
+        if (!requestId) {
+            // No ID = broker can't route response, so silently drop
+            return;
+        }
+        if (!this.isConnected) {
+            throw new KadiError('Not connected to broker', ErrorCode.BROKER_NOT_CONNECTED, 503, { broker: this.config.name });
+        }
+        // Send proper JSON-RPC response with matching id
+        const response = {
+            jsonrpc: '2.0',
+            result,
+            id: requestId
+        };
+        this.send(response);
+    }
+    /**
+     * Send a tool error back to broker as JSON-RPC error response
+     *
+     * Used when tool execution fails and we need to report the error back.
+     * The requestId must match the incoming request's id field for proper routing.
+     *
+     * @param requestId - Request ID to respond to (CRITICAL: must match incoming request id)
+     * @param code - Error code (e.g., ErrorCode.TOOL_NOT_FOUND)
+     * @param message - Human-readable error message
+     * @param data - Optional additional error context
+     */
+    sendError(requestId, code, message, data) {
+        if (!requestId) {
+            // No ID = broker can't route error, so silently drop
+            return;
+        }
+        if (!this.isConnected) {
+            throw new KadiError('Not connected to broker', ErrorCode.BROKER_NOT_CONNECTED, 503, { broker: this.config.name });
+        }
+        // Send proper JSON-RPC error response with matching id
+        const response = {
+            jsonrpc: '2.0',
+            error: {
+                code: -32000, // Application error code range
+                message: `${code}: ${message}`,
+                data
+            },
+            id: requestId
+        };
+        this.send(response);
+    }
+    /**
+     * Send raw data to broker
+     *
+     * @param data - Data to send (request, response, or notification)
+     */
+    send(data) {
+        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+            throw new KadiError('WebSocket not open', ErrorCode.BROKER_NOT_CONNECTED, 503, { broker: this.config.name });
+        }
+        try {
+            const message = JSON.stringify(data);
+            this.ws.send(message);
+        }
+        catch (error) {
+            throw new KadiError(`Failed to send message: ${error instanceof Error ? error.message : String(error)}`, ErrorCode.TRANSPORT_SEND_FAILED, 500, { broker: this.config.name });
+        }
+    }
+    /**
+     * Handle incoming WebSocket message
+     *
+     * @param data - Raw message data
+     */
+    handleMessage(data) {
+        try {
+            const message = JSON.parse(data.toString());
+            // Check if this is a response to a pending request
+            if ('id' in message && message.id !== undefined) {
+                const pending = this.pendingRequests.get(message.id);
+                if (pending) {
+                    clearTimeout(pending.timeout);
+                    this.pendingRequests.delete(message.id);
+                    const response = message;
+                    if (response.error) {
+                        pending.reject(new KadiError(response.error.message, ErrorCode.BROKER_RESPONSE_ERROR, 500, { broker: this.config.name, errorCode: response.error.code, errorData: response.error.data }));
+                    }
+                    else {
+                        pending.resolve(response.result);
+                    }
+                    return;
+                }
+            }
+            // Emit as general message (notification or unsolicited response)
+            this.emit('message', message);
+        }
+        catch (error) {
+            this.emit('error', new KadiError(`Failed to parse message: ${error instanceof Error ? error.message : String(error)}`, ErrorCode.INVALID_MESSAGE_FORMAT, 400, { broker: this.config.name }));
+        }
+    }
+    /**
+     * Handle WebSocket close event
+     *
+     * @param code - Close code
+     * @param reason - Close reason
+     */
+    handleClose(code, reason) {
+        this.setState(ConnectionState.DISCONNECTED);
+        this.cleanup();
+        this.emit('disconnected', code, reason);
+    }
+    /**
+     * Setup heartbeat timer
+     */
+    setupHeartbeat() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+        }
+        if (this.config.heartbeatInterval && this.config.heartbeatInterval > 0) {
+            this.heartbeatTimer = setInterval(() => {
+                this._lastHeartbeat = Date.now();
+                // Heartbeat logic handled by BrokerProtocol
+            }, this.config.heartbeatInterval);
+        }
+    }
+    /**
+     * Cleanup connection resources
+     */
+    cleanup() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+            this.heartbeatTimer = null;
+        }
+        // Reject all pending requests
+        for (const [id, pending] of this.pendingRequests) {
+            clearTimeout(pending.timeout);
+            pending.reject(new KadiError('Connection closed', ErrorCode.CONNECTION_CLOSED, 503, { broker: this.config.name, requestId: id }));
+        }
+        this.pendingRequests.clear();
+        if (this.ws) {
+            this.ws.removeAllListeners();
+            if (this.ws.readyState === WebSocket.OPEN || this.ws.readyState === WebSocket.CONNECTING) {
+                this.ws.close();
+            }
+            this.ws = null;
+        }
+        this._connectedAt = null;
+        this._lastHeartbeat = null;
+        this.setState(ConnectionState.DISCONNECTED);
+    }
+    /**
+     * Set connection state and emit event
+     *
+     * @param newState - New state
+     */
+    setState(newState) {
+        const oldState = this._state;
+        if (oldState !== newState) {
+            this._state = newState;
+            this.emit('stateChange', oldState, newState);
+        }
+    }
+    /**
+     * Get broker name
+     */
+    get name() {
+        return this.config.name;
+    }
+    /**
+     * Get broker URL
+     */
+    get url() {
+        return this.config.url;
+    }
+    /**
+     * Get current connection state
+     */
+    get state() {
+        return this._state;
+    }
+    /**
+     * Check if connected
+     */
+    get isConnected() {
+        return this._state === ConnectionState.CONNECTED || this._state === ConnectionState.AUTHENTICATED;
+    }
+    /**
+     * Get assigned agent ID
+     */
+    get agentId() {
+        return this._agentId;
+    }
+    /**
+     * Set agent ID (called after handshake)
+     */
+    set agentId(id) {
+        this._agentId = id;
+    }
+    /**
+     * Get connection timestamp
+     */
+    get connectedAt() {
+        return this._connectedAt;
+    }
+    /**
+     * Get last heartbeat timestamp
+     */
+    get lastHeartbeat() {
+        return this._lastHeartbeat;
+    }
+}
+//# sourceMappingURL=BrokerConnection.js.map

package/dist/broker/BrokerConnection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BrokerConnection.js","sourceRoot":"","sources":["../../src/broker/BrokerConnection.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,SAAS,MAAM,IAAI,CAAC;AAQ3B,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAgCzD;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,OAAO,gBAAiB,SAAQ,YAAY;IAChD;;OAEG;IACc,MAAM,CAAe;IAEtC;;OAEG;IACK,EAAE,GAAqB,IAAI,CAAC;IAEpC;;OAEG;IACK,MAAM,GAAoB,eAAe,CAAC,YAAY,CAAC;IAE/D;;OAEG;IACK,QAAQ,GAAkB,IAAI,CAAC;IAEvC;;OAEG;IACK,cAAc,GAA0B,IAAI,CAAC;IAErD;;OAEG;IACK,YAAY,GAAkB,IAAI,CAAC;IAE3C;;OAEG;IACK,cAAc,GAAkB,IAAI,CAAC;IAE7C;;OAEG;IACK,eAAe,GAAG,IAAI,GAAG,EAAmC,CAAC;IAErE;;OAEG;IACK,gBAAgB,GAAG,CAAC,CAAC;IAE7B;;;;OAIG;IACH,YAAY,MAAoB;QAC9B,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG;YACZ,GAAG,MAAM;YACT,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,KAAK;YACpD,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,KAAK;YAC9C,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,KAAK;YACpD,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,IAAI;YAC3C,oBAAoB,EAAE,MAAM,CAAC,oBAAoB,IAAI,CAAC;SACvD,CAAC;IACJ,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,MAAM,KAAK,eAAe,CAAC,YAAY,EAAE,CAAC;YACjD,MAAM,IAAI,SAAS,CACjB,+CAA+C,EAC/C,SAAS,CAAC,iBAAiB,EAC3B,GAAG,EACH,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CACxD,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAE1C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,SAAS,CAClB,4BAA4B,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,EAC7D,SAAS,CAAC,kBAAkB,EAC5B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CACnD,CAAC,CAAC;YACL,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;YAElC,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAEzC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;oBACtB,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC/B,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;oBACzC,IAAI,CAAC,cAAc,EAAE,CAAC;oBACtB,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;oBACvB,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBAC5B,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBAC1B,MAAM,CAAC,IAAI,SAAS,CAClB,oBAAoB,KAAK,CAAC,OAAO,EAAE,EACnC,SAAS,CAAC,iBAAiB,EAC3B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,aAAa,EAAE,KAAK,CAAC,OAAO,EAAE,CACjF,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;oBACnC,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC5C,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;oBAC7B,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC,CAAC,CAAC;YAEL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED;;;;;;;;;;;;;;;;;;OAkBG;IACH,KAAK,CAAC,WAAW,CAAc,OAAuB;QACpD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,SAAS,CACjB,yBAAyB,EACzB,SAAS,CAAC,oBAAoB,EAC9B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAC7B,CAAC;QACJ,CAAC;QAED,8BAA8B;QAC9B,IAAI,OAAO,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YAC7B,OAAO,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,gBAAgB,CAAC;QACvC,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAG,CAAC,CAAC;gBACzC,MAAM,CAAC,IAAI,SAAS,CAClB,yBAAyB,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,EACvD,SAAS,CAAC,cAAc,EACxB,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,CAC5E,CAAC,CAAC;YACL,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YAE/B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAG,EAAE;gBACpC,EAAE,EAAE,OAAO,CAAC,EAAG;gBACf,OAAO,EAAE,OAAmC;gBAC5C,MAAM;gBACN,OAAO;gBACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,gBAAgB,CAAC,YAAiC;QAChD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,SAAS,CACjB,yBAAyB,EACzB,SAAS,CAAC,oBAAoB,EAC9B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAC7B,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IAED;;;;;;;;OAQG;IACH,YAAY,CAAC,SAAsC,EAAE,MAAe;QAClE,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,wDAAwD;YACxD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,SAAS,CACjB,yBAAyB,EACzB,SAAS,CAAC,oBAAoB,EAC9B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAC7B,CAAC;QACJ,CAAC;QAED,iDAAiD;QACjD,MAAM,QAAQ,GAAoB;YAChC,OAAO,EAAE,KAAK;YACd,MAAM;YACN,EAAE,EAAE,SAAS;SACd,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtB,CAAC;IAED;;;;;;;;;;OAUG;IACH,SAAS,CACP,SAAsC,EACtC,IAAY,EACZ,OAAe,EACf,IAA8B;QAE9B,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,qDAAqD;YACrD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,SAAS,CACjB,yBAAyB,EACzB,SAAS,CAAC,oBAAoB,EAC9B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAC7B,CAAC;QACJ,CAAC;QAED,uDAAuD;QACvD,MAAM,QAAQ,GAAoB;YAChC,OAAO,EAAE,KAAK;YACd,KAAK,EAAE;gBACL,IAAI,EAAE,CAAC,KAAK,EAAE,+BAA+B;gBAC7C,OAAO,EAAE,GAAG,IAAI,KAAK,OAAO,EAAE;gBAC9B,IAAI;aACL;YACD,EAAE,EAAE,SAAS;SACd,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACK,IAAI,CAAC,IAA4D;QACvE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACtD,MAAM,IAAI,SAAS,CACjB,oBAAoB,EACpB,SAAS,CAAC,oBAAoB,EAC9B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAC7B,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACrC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,SAAS,CACjB,2BAA2B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACnF,SAAS,CAAC,qBAAqB,EAC/B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAC7B,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,aAAa,CAAC,IAAoB;QACxC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAA0C,CAAC;YAErF,mDAAmD;YACnD,IAAI,IAAI,IAAI,OAAO,IAAI,OAAO,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;gBAChD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACrD,IAAI,OAAO,EAAE,CAAC;oBACZ,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAExC,MAAM,QAAQ,GAAG,OAA0B,CAAC;oBAC5C,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;wBACnB,OAAO,CAAC,MAAM,CAAC,IAAI,SAAS,CAC1B,QAAQ,CAAC,KAAK,CAAC,OAAO,EACtB,SAAS,CAAC,qBAAqB,EAC/B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAC7F,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBACnC,CAAC;oBACD,OAAO;gBACT,CAAC;YACH,CAAC;YAED,iEAAiE;YACjE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAEhC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,SAAS,CAC9B,4BAA4B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACpF,SAAS,CAAC,sBAAsB,EAChC,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAC7B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,WAAW,CAAC,IAAY,EAAE,MAAc;QAC9C,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;QAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;YACvE,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;gBACrC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACjC,4CAA4C;YAC9C,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,OAAO;QACb,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QAED,8BAA8B;QAC9B,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACjD,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,IAAI,SAAS,CAC1B,mBAAmB,EACnB,SAAS,CAAC,iBAAiB,EAC3B,GAAG,EACH,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,CAC5C,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,UAAU,EAAE,CAAC;gBACzF,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAClB,CAAC;YACD,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC;IAED;;;;OAIG;IACK,QAAQ,CAAC,QAAyB;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC;QAC7B,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED;;OAEG;IACH,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,MAAM,KAAK,eAAe,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,eAAe,CAAC,aAAa,CAAC;IACpG,CAAC;IAED;;OAEG;IACH,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,IAAI,OAAO,CAAC,EAAiB;QAC3B,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;CACF"}

package/dist/broker/BrokerConnectionManager.d.ts

@@ -0,0 +1,216 @@
+/**
+ * Broker Connection Manager
+ *
+ * Manages multiple broker connections.
+ * Handles connection lifecycle, failover, and routing.
+ *
+ * @module broker/BrokerConnectionManager
+ */
+import { EventEmitter } from 'events';
+import type { BrokerConfig } from '../types/index.js';
+import { BrokerConnection } from './BrokerConnection.js';
+/**
+ * Broker Connection Manager
+ *
+ * Single responsibility: Manage multiple broker connections.
+ * Does NOT handle:
+ * - Protocol logic (that's BrokerProtocol's job)
+ * - Tool registration/invocation (that's ToolRegistry/Client's job)
+ *
+ * @example
+ * ```typescript
+ * const manager = new BrokerConnectionManager();
+ *
+ * // Add brokers
+ * await manager.connect({
+ *   url: 'ws://localhost:8080',
+ *   name: 'local'
+ * });
+ *
+ * await manager.connect({
+ *   url: 'wss://prod.example.com',
+ *   name: 'production'
+ * });
+ *
+ * // Get a connection
+ * const connection = manager.getConnection('local');
+ *
+ * // Get default connection
+ * const defaultConn = manager.getDefaultConnection();
+ * ```
+ */
+export declare class BrokerConnectionManager extends EventEmitter {
+    /**
+     * Active broker connections (name -> connection)
+     */
+    private connections;
+    /**
+     * Default broker name
+     */
+    private defaultBrokerName;
+    /**
+     * Connect to a broker
+     *
+     * @param config - Broker configuration
+     * @param setAsDefault - Set this as the default broker
+     * @returns The established connection
+     *
+     * @throws {KadiError} If connection fails
+     *
+     * @example
+     * ```typescript
+     * const connection = await manager.connect({
+     *   url: 'ws://localhost:8080',
+     *   name: 'local'
+     * }, true);
+     * ```
+     */
+    connect(config: BrokerConfig, setAsDefault?: boolean): Promise<BrokerConnection>;
+    /**
+     * Connect to multiple brokers
+     *
+     * @param configs - Array of broker configurations
+     * @param defaultBrokerName - Name of the default broker
+     * @returns Array of established connections
+     *
+     * @throws {KadiError} If any connection fails
+     *
+     * @example
+     * ```typescript
+     * await manager.connectMultiple([
+     *   { url: 'ws://localhost:8080', name: 'local' },
+     *   { url: 'wss://prod.example.com', name: 'prod' }
+     * ], 'local');
+     * ```
+     */
+    connectMultiple(configs: BrokerConfig[], defaultBrokerName?: string): Promise<BrokerConnection[]>;
+    /**
+     * Disconnect from a broker
+     *
+     * @param brokerName - Name of broker to disconnect from
+     *
+     * @example
+     * ```typescript
+     * await manager.disconnect('local');
+     * ```
+     */
+    disconnect(brokerName: string): Promise<void>;
+    /**
+     * Disconnect from all brokers
+     *
+     * @example
+     * ```typescript
+     * await manager.disconnectAll();
+     * ```
+     */
+    disconnectAll(): Promise<void>;
+    /**
+     * Get a broker connection by name
+     *
+     * @param brokerName - Name of the broker
+     * @returns The connection or undefined if not found
+     *
+     * @example
+     * ```typescript
+     * const conn = manager.getConnection('local');
+     * if (conn) {
+     *   await conn.sendRequest(...);
+     * }
+     * ```
+     */
+    getConnection(brokerName: string): BrokerConnection | undefined;
+    /**
+     * Get the default broker connection
+     *
+     * @returns The default connection
+     * @throws {KadiError} If no default broker is set
+     *
+     * @example
+     * ```typescript
+     * const conn = manager.getDefaultConnection();
+     * await conn.sendRequest(...);
+     * ```
+     */
+    getDefaultConnection(): BrokerConnection;
+    /**
+     * Get all broker connections
+     *
+     * @returns Array of all connections
+     *
+     * @example
+     * ```typescript
+     * const connections = manager.getAllConnections();
+     * console.log(`Connected to ${connections.length} brokers`);
+     * ```
+     */
+    getAllConnections(): BrokerConnection[];
+    /**
+     * Get all connected broker connections
+     *
+     * @returns Array of connected connections
+     *
+     * @example
+     * ```typescript
+     * const connected = manager.getConnectedBrokers();
+     * ```
+     */
+    getConnectedBrokers(): BrokerConnection[];
+    /**
+     * Check if connected to a specific broker
+     *
+     * @param brokerName - Broker name
+     * @returns true if connected
+     *
+     * @example
+     * ```typescript
+     * if (manager.isConnected('local')) {
+     *   // Use local broker
+     * }
+     * ```
+     */
+    isConnected(brokerName: string): boolean;
+    /**
+     * Check if connected to any broker
+     *
+     * @returns true if connected to at least one broker
+     *
+     * @example
+     * ```typescript
+     * if (!manager.hasConnections()) {
+     *   throw new Error('Not connected to any broker');
+     * }
+     * ```
+     */
+    hasConnections(): boolean;
+    /**
+     * Set the default broker
+     *
+     * @param brokerName - Name of broker to set as default
+     * @throws {KadiError} If broker doesn't exist
+     *
+     * @example
+     * ```typescript
+     * manager.setDefaultBroker('production');
+     * ```
+     */
+    setDefaultBroker(brokerName: string): void;
+    /**
+     * Get default broker name
+     *
+     * @returns Default broker name or null if not set
+     */
+    getDefaultBrokerName(): string | null;
+    /**
+     * Get number of active connections
+     *
+     * @returns Connection count
+     */
+    get connectionCount(): number;
+    /**
+     * Get broker names
+     *
+     * @returns Array of broker names
+     */
+    getBrokerNames(): string[];
+}
+//# sourceMappingURL=BrokerConnectionManager.d.ts.map

package/dist/broker/BrokerConnectionManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BrokerConnectionManager.d.ts","sourceRoot":"","sources":["../../src/broker/BrokerConnectionManager.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,qBAAa,uBAAwB,SAAQ,YAAY;IACvD;;OAEG;IACH,OAAO,CAAC,WAAW,CAAuC;IAE1D;;OAEG;IACH,OAAO,CAAC,iBAAiB,CAAuB;IAEhD;;;;;;;;;;;;;;;;OAgBG;IACG,OAAO,CAAC,MAAM,EAAE,YAAY,EAAE,YAAY,UAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA6CpF;;;;;;;;;;;;;;;;OAgBG;IACG,eAAe,CACnB,OAAO,EAAE,YAAY,EAAE,EACvB,iBAAiB,CAAC,EAAE,MAAM,GACzB,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAY9B;;;;;;;;;OASG;IACG,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBnD;;;;;;;OAOG;IACG,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAQpC;;;;;;;;;;;;;OAaG;IACH,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,gBAAgB,GAAG,SAAS;IAI/D;;;;;;;;;;;OAWG;IACH,oBAAoB,IAAI,gBAAgB;IAsBxC;;;;;;;;;;OAUG;IACH,iBAAiB,IAAI,gBAAgB,EAAE;IAIvC;;;;;;;;;OASG;IACH,mBAAmB,IAAI,gBAAgB,EAAE;IAIzC;;;;;;;;;;;;OAYG;IACH,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;IAKxC;;;;;;;;;;;OAWG;IACH,cAAc,IAAI,OAAO;IAIzB;;;;;;;;;;OAUG;IACH,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAa1C;;;;OAIG;IACH,oBAAoB,IAAI,MAAM,GAAG,IAAI;IAIrC;;;;OAIG;IACH,IAAI,eAAe,IAAI,MAAM,CAE5B;IAED;;;;OAIG;IACH,cAAc,IAAI,MAAM,EAAE;CAG3B"}