agent-room 0.2.0

package/dist/protocols/ws-adapter.js

@@ -0,0 +1,154 @@
+import WebSocket from "ws";
+import { StreamAdapter } from "./adapter-interface.js";
+import { Logger } from "../core/logger.js";
+const log = Logger.create("ws-adapter");
+const DEFAULT_HEARTBEAT_MS = 30_000;
+const DEFAULT_MAX_RECONNECT = 10;
+const DEFAULT_INITIAL_DELAY_MS = 1_000;
+const DEFAULT_MAX_DELAY_MS = 30_000;
+/**
+ * WebSocket protocol adapter with ping/pong heartbeat and exponential-backoff reconnection.
+ */
+export class WsAdapter extends StreamAdapter {
+    _state = "disconnected";
+    _ws = null;
+    _url = "";
+    _options = {};
+    // heartbeat
+    _heartbeatTimer = null;
+    _pongReceived = true;
+    // reconnect
+    _reconnectAttempt = 0;
+    _reconnectTimer = null;
+    _intentionalDisconnect = false;
+    get state() {
+        return this._state;
+    }
+    async connect(url, options) {
+        this._url = url;
+        this._options = options ?? {};
+        this._intentionalDisconnect = false;
+        this._reconnectAttempt = 0;
+        return this._doConnect();
+    }
+    async disconnect() {
+        this._intentionalDisconnect = true;
+        this._clearTimers();
+        if (this._ws) {
+            this._ws.removeAllListeners();
+            if (this._ws.readyState === WebSocket.OPEN ||
+                this._ws.readyState === WebSocket.CONNECTING) {
+                this._ws.close(1000, "Client disconnect");
+            }
+            this._ws = null;
+        }
+        this._setState("disconnected");
+    }
+    async send(data) {
+        if (!this._ws || this._ws.readyState !== WebSocket.OPEN) {
+            throw new Error("WebSocket is not connected — cannot send");
+        }
+        this._ws.send(data);
+    }
+    // ─── Internal ────────────────────────────────────────────────────────
+    _doConnect() {
+        return new Promise((resolve, reject) => {
+            this._setState("connecting");
+            const headers = { ...this._options.headers };
+            if (this._options.authToken) {
+                headers["Authorization"] = `Bearer ${this._options.authToken}`;
+            }
+            const ws = new WebSocket(this._url, { headers });
+            this._ws = ws;
+            ws.on("open", () => {
+                this._reconnectAttempt = 0;
+                this._setState("connected");
+                this._startHeartbeat();
+                log.info("websocket connected", { url: this._url });
+                this.emit("connected");
+                resolve();
+            });
+            ws.on("message", (raw) => {
+                const data = raw instanceof Buffer ? raw.toString("utf-8") : String(raw);
+                this.emit("message", data);
+            });
+            ws.on("pong", () => {
+                this._pongReceived = true;
+            });
+            ws.on("close", (code, reason) => {
+                this._stopHeartbeat();
+                log.debug("websocket closed", { url: this._url, code, reason: reason.toString() });
+                if (!this._intentionalDisconnect) {
+                    this.emit("disconnected", "Connection closed");
+                    this._scheduleReconnect();
+                }
+            });
+            ws.on("error", (err) => {
+                log.error("websocket error", { url: this._url }, err);
+                this.emit("error", err);
+                // Only reject the initial connect promise; subsequent errors go through events
+                if (this._state === "connecting" && this._reconnectAttempt === 0) {
+                    reject(err);
+                }
+            });
+        });
+    }
+    _startHeartbeat() {
+        const interval = this._options.heartbeatIntervalMs ?? DEFAULT_HEARTBEAT_MS;
+        this._pongReceived = true;
+        this._heartbeatTimer = setInterval(() => {
+            if (!this._pongReceived) {
+                // Missed pong — connection is likely dead
+                log.warn("heartbeat timeout — missed pong, terminating", { url: this._url });
+                Logger.metrics.increment("ws.heartbeat_timeout");
+                this._ws?.terminate();
+                return;
+            }
+            this._pongReceived = false;
+            this._ws?.ping();
+        }, interval);
+    }
+    _stopHeartbeat() {
+        if (this._heartbeatTimer) {
+            clearInterval(this._heartbeatTimer);
+            this._heartbeatTimer = null;
+        }
+    }
+    _scheduleReconnect() {
+        const maxAttempts = this._options.maxReconnectAttempts ?? DEFAULT_MAX_RECONNECT;
+        if (this._reconnectAttempt >= maxAttempts) {
+            this._setState("failed");
+            log.error("max reconnect attempts reached", { url: this._url, maxAttempts });
+            this.emit("error", new Error(`Max reconnect attempts (${maxAttempts}) reached`));
+            return;
+        }
+        this._reconnectAttempt++;
+        this._setState("reconnecting");
+        this.emit("reconnecting", this._reconnectAttempt);
+        const baseDelay = this._options.initialReconnectDelayMs ?? DEFAULT_INITIAL_DELAY_MS;
+        const maxDelay = this._options.maxReconnectDelayMs ?? DEFAULT_MAX_DELAY_MS;
+        const expDelay = baseDelay * Math.pow(2, this._reconnectAttempt - 1);
+        const jitter = Math.random() * baseDelay;
+        const delay = Math.min(expDelay + jitter, maxDelay);
+        log.info("scheduling reconnect", { url: this._url, attempt: this._reconnectAttempt, delay_ms: Math.round(delay) });
+        this._reconnectTimer = setTimeout(async () => {
+            try {
+                await this._doConnect();
+            }
+            catch {
+                // _doConnect already handles scheduling next reconnect via the close handler
+            }
+        }, delay);
+    }
+    _clearTimers() {
+        this._stopHeartbeat();
+        if (this._reconnectTimer) {
+            clearTimeout(this._reconnectTimer);
+            this._reconnectTimer = null;
+        }
+    }
+    _setState(state) {
+        this._state = state;
+    }
+}
+//# sourceMappingURL=ws-adapter.js.map

package/dist/protocols/ws-adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-adapter.js","sourceRoot":"","sources":["../../src/protocols/ws-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,IAAI,CAAC;AAC3B,OAAO,EAAE,aAAa,EAA6C,MAAM,wBAAwB,CAAC;AAClG,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAE3C,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;AAExC,MAAM,oBAAoB,GAAG,MAAM,CAAC;AACpC,MAAM,qBAAqB,GAAG,EAAE,CAAC;AACjC,MAAM,wBAAwB,GAAG,KAAK,CAAC;AACvC,MAAM,oBAAoB,GAAG,MAAM,CAAC;AAEpC;;GAEG;AACH,MAAM,OAAO,SAAU,SAAQ,aAAa;IAClC,MAAM,GAAoB,cAAc,CAAC;IACzC,GAAG,GAAqB,IAAI,CAAC;IAC7B,IAAI,GAAG,EAAE,CAAC;IACV,QAAQ,GAAmB,EAAE,CAAC;IAEtC,YAAY;IACJ,eAAe,GAA0C,IAAI,CAAC;IAC9D,aAAa,GAAG,IAAI,CAAC;IAE7B,YAAY;IACJ,iBAAiB,GAAG,CAAC,CAAC;IACtB,eAAe,GAAyC,IAAI,CAAC;IAC7D,sBAAsB,GAAG,KAAK,CAAC;IAEvC,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,GAAW,EAAE,OAAwB;QACjD,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC;QAChB,IAAI,CAAC,QAAQ,GAAG,OAAO,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,sBAAsB,GAAG,KAAK,CAAC;QACpC,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;QAE3B,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC;QACnC,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,kBAAkB,EAAE,CAAC;YAC9B,IACE,IAAI,CAAC,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI;gBACtC,IAAI,CAAC,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,UAAU,EAC5C,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;YAC5C,CAAC;YACD,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;QAClB,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,IAAqB;QAC9B,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;IAED,wEAAwE;IAEhE,UAAU;QAChB,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAE7B,MAAM,OAAO,GAA2B,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrE,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;gBAC5B,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YACjE,CAAC;YAED,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YACjD,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC;YAEd,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACjB,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBAC3B,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBAC5B,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBACpD,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBACvB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAsB,EAAE,EAAE;gBAC1C,MAAM,IAAI,GACR,GAAG,YAAY,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC9D,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7B,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACjB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAY,EAAE,MAAc,EAAE,EAAE;gBAC9C,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,GAAG,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAEnF,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;oBACjC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;oBAC/C,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;gBAC5B,GAAG,CAAC,KAAK,CAAC,iBAAiB,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC;gBACtD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;gBACxB,+EAA+E;gBAC/E,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,IAAI,IAAI,CAAC,iBAAiB,KAAK,CAAC,EAAE,CAAC;oBACjE,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,MAAM,QAAQ,GACZ,IAAI,CAAC,QAAQ,CAAC,mBAAmB,IAAI,oBAAoB,CAAC;QAC5D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAE1B,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE;YACtC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;gBACxB,0CAA0C;gBAC1C,GAAG,CAAC,IAAI,CAAC,8CAA8C,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC7E,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;gBACjD,IAAI,CAAC,GAAG,EAAE,SAAS,EAAE,CAAC;gBACtB,OAAO;YACT,CAAC;YACD,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;YAC3B,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QACnB,CAAC,EAAE,QAAQ,CAAC,CAAC;IACf,CAAC;IAEO,cAAc;QACpB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,kBAAkB;QACxB,MAAM,WAAW,GACf,IAAI,CAAC,QAAQ,CAAC,oBAAoB,IAAI,qBAAqB,CAAC;QAE9D,IAAI,IAAI,CAAC,iBAAiB,IAAI,WAAW,EAAE,CAAC;YAC1C,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACzB,GAAG,CAAC,KAAK,CAAC,gCAAgC,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YAC7E,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,2BAA2B,WAAW,WAAW,CAAC,CAAC,CAAC;YACjF,OAAO;QACT,CAAC;QAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC/B,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAElD,MAAM,SAAS,GACb,IAAI,CAAC,QAAQ,CAAC,uBAAuB,IAAI,wBAAwB,CAAC;QACpE,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,mBAAmB,IAAI,oBAAoB,CAAC;QAC3E,MAAM,QAAQ,GAAG,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,CAAC,CAAC;QAEpD,GAAG,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAEnH,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;YAC3C,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;YAC1B,CAAC;YAAC,MAAM,CAAC;gBACP,6EAA6E;YAC/E,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,YAAY,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACnC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,KAAsB;QACtC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;CACF"}

package/dist/server.d.ts

@@ -0,0 +1,10 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export interface AgentRoomServerOptions {
+    /** Default Service WebSocket URL. Read from AGENT_ROOM_URL env var or --service-url CLI arg. */
+    serviceUrl?: string;
+}
+/**
+ * Create and configure the AgentRoom MCP server with all tools and resources.
+ */
+export declare function createAgentRoomServer(options?: AgentRoomServerOptions): McpServer;
+//# sourceMappingURL=server.d.ts.map

package/dist/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAoB,MAAM,yCAAyC,CAAC;AAiEtF,MAAM,WAAW,sBAAsB;IACrC,gGAAgG;IAChG,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,GAAE,sBAA2B,GAAG,SAAS,CAw/BrF"}