@nodetool-ai/websocket 0.7.0-rc.22 → 0.8.1

package/dist/ws-adapter.js

@@ -1,45 +1,142 @@
+import { WsMessageRateLimiter } from "./lib/ws-rate-limit.js";
+import { getWsConnectionHealthConfig } from "./lib/ws-connection-health.js";
+import { isNumber } from "./lib/wire-values.js";
 /**
- * Adapts a ws WebSocket to the WebSocketConnection interface used by UnifiedWebSocketRunner.
+ * Adapts a ws WebSocket to the WebSocketConnection interface used by WebSocketClientSession.
+ *
+ * Beyond the plumbing it enforces the per-connection health limits described in
+ * `lib/ws-connection-health.ts`: a ping/idle deadline that closes half-open
+ * peers, a bounded outbound buffer that makes `sendBytes`/`sendText` wait on a
+ * slow reader instead of growing the heap, and a cap on undelivered inbound
+ * frames.
  */
 export class WsAdapter {
     socket;
+    rateLimiter;
     clientState = "connected";
     applicationState = "connected";
     queue = [];
     waiters = [];
-    constructor(socket) {
+    disconnected = false;
+    health;
+    lastActivityAt = Date.now();
+    healthTimer = null;
+    constructor(socket, rateLimiter = new WsMessageRateLimiter(), health = getWsConnectionHealthConfig()) {
         this.socket = socket;
+        this.rateLimiter = rateLimiter;
+        this.health = health;
         socket.on("message", (raw, isBinary) => {
+            if (this.disconnected)
+                return;
+            this.lastActivityAt = Date.now();
+            // Cap inbound message rate per connection — a flooding client is
+            // disconnected with a policy-violation code instead of being serviced.
+            if (!this.rateLimiter.allow()) {
+                this.closeWith(1008, "Message rate limit exceeded");
+                return;
+            }
             const bytes = raw instanceof Uint8Array ? raw : new Uint8Array(raw);
             const frame = isBinary
                 ? { type: "websocket.message", bytes }
                 : { type: "websocket.message", text: bytes.toString() };
             const waiter = this.waiters.shift();
-            if (waiter)
-                waiter(frame);
-            else
-                this.queue.push(frame);
-        });
-        socket.on("error", () => {
-            this.clientState = "disconnected";
-            this.applicationState = "disconnected";
-            const frame = { type: "websocket.disconnect" };
-            for (const waiter of this.waiters) {
+            if (waiter) {
                 waiter(frame);
+                return;
             }
-            this.waiters.length = 0;
+            // Nothing is reading. Frames pile up only if the runner's receive loop
+            // has stalled; past this depth the client is better served by a close
+            // it can reconnect from than by an ever-growing queue.
+            if (this.queue.length >= this.health.maxQueuedFrames) {
+                this.closeWith(1008, "Inbound queue overflow");
+                return;
+            }
+            this.queue.push(frame);
         });
-        socket.on("close", () => {
-            this.clientState = "disconnected";
-            this.applicationState = "disconnected";
-            // Notify ALL pending waiters, not just the first one,
-            // to prevent hanging promises when the socket closes.
-            const frame = { type: "websocket.disconnect" };
-            for (const waiter of this.waiters) {
+        // Any traffic proves the peer is alive: data frames, protocol pongs, and
+        // the pings browsers send on their own.
+        socket.on("pong", () => this.markActivity());
+        socket.on("ping", () => this.markActivity());
+        socket.on("error", () => this.markDisconnected());
+        socket.on("close", () => this.markDisconnected());
+        this.startHealthWatchdog();
+    }
+    markActivity() {
+        this.lastActivityAt = Date.now();
+    }
+    closeWith(code, reason) {
+        try {
+            this.socket.close(code, reason);
+        }
+        catch {
+            // socket already closing/closed
+        }
+        this.markDisconnected();
+    }
+    /**
+     * Transition to the disconnected state exactly once and unblock consumers.
+     *
+     * Wakes every pending `receive()` with a disconnect frame. When no `receive()`
+     * is in flight (e.g. the socket drops while the runner is busy), the frame is
+     * queued so the next `receive()` resolves instead of hanging forever.
+     */
+    markDisconnected() {
+        if (this.disconnected)
+            return;
+        this.disconnected = true;
+        this.clientState = "disconnected";
+        this.applicationState = "disconnected";
+        this.stopHealthWatchdog();
+        const frame = { type: "websocket.disconnect" };
+        if (this.waiters.length > 0) {
+            for (const waiter of this.waiters)
                 waiter(frame);
-            }
             this.waiters.length = 0;
-        });
+        }
+        else {
+            this.queue.push(frame);
+        }
+    }
+    /**
+     * Probe the peer on a timer and give up on it once it has been silent past
+     * the idle deadline.
+     *
+     * `close()` is not enough for a peer that is already gone — the closing
+     * handshake has nobody to answer it and the socket can linger for minutes —
+     * so the connection is terminated outright once the deadline passes.
+     */
+    startHealthWatchdog() {
+        if (!this.health.enabled || this.health.pingIntervalMs <= 0)
+            return;
+        this.healthTimer = setInterval(() => {
+            if (this.disconnected)
+                return;
+            if (Date.now() - this.lastActivityAt >= this.health.idleTimeoutMs) {
+                this.markDisconnected();
+                try {
+                    this.socket.terminate?.();
+                }
+                catch {
+                    // socket already gone
+                }
+                return;
+            }
+            try {
+                this.socket.ping?.();
+            }
+            catch {
+                // A ping that cannot even be queued means the socket is finished.
+                this.markDisconnected();
+            }
+        }, this.health.pingIntervalMs);
+        // Never hold the process open for a watchdog.
+        this.healthTimer.unref?.();
+    }
+    stopHealthWatchdog() {
+        if (this.healthTimer) {
+            clearInterval(this.healthTimer);
+            this.healthTimer = null;
+        }
     }
     async accept() { }
     async receive() {
@@ -49,14 +146,55 @@ export class WsAdapter {
         return new Promise((resolve) => this.waiters.push(resolve));
     }
     async sendBytes(data) {
-        this.socket.send(data);
+        await this.send(data);
     }
     async sendText(data) {
-        this.socket.send(data);
+        await this.send(data);
+    }
+    /**
+     * Write one frame, waiting first if the peer is behind.
+     *
+     * The runner serializes sends behind a lock, so waiting here propagates
+     * backpressure all the way up to whatever is producing messages instead of
+     * letting them accumulate in the socket's buffer.
+     */
+    async send(data) {
+        if (this.disconnected)
+            return;
+        await this.awaitDrain();
+        if (this.disconnected)
+            return;
+        try {
+            this.socket.send(data);
+        }
+        catch {
+            // The peer went away between the drain check and the write.
+            this.markDisconnected();
+        }
+    }
+    async awaitDrain() {
+        const { maxBufferedBytes, drainTimeoutMs } = this.health;
+        if (maxBufferedBytes <= 0)
+            return;
+        if (this.bufferedAmount() < maxBufferedBytes)
+            return;
+        const deadline = Date.now() + drainTimeoutMs;
+        while (!this.disconnected && this.bufferedAmount() >= maxBufferedBytes) {
+            if (Date.now() >= deadline) {
+                // 1001 rather than a policy code: the client should treat this as a
+                // routine drop and reconnect.
+                this.closeWith(1001, "Client too slow to receive");
+                return;
+            }
+            await new Promise((resolve) => setTimeout(resolve, 50));
+        }
+    }
+    bufferedAmount() {
+        const buffered = this.socket.bufferedAmount;
+        return isNumber(buffered) ? buffered : 0;
     }
     async close(code, reason) {
-        this.clientState = "disconnected";
-        this.applicationState = "disconnected";
+        this.markDisconnected();
         this.socket.close(code, reason);
     }
 }

package/dist/ws-adapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"ws-adapter.js","sourceRoot":"","sources":["../src/ws-adapter.ts"],"names":[],"mappings":"AASA;;GAEG;AACH,MAAM,OAAO,SAAS;IAOA;IANpB,WAAW,GAAiC,WAAW,CAAC;IACxD,gBAAgB,GAAiC,WAAW,CAAC;IAErD,KAAK,GAAc,EAAE,CAAC;IACtB,OAAO,GAAoC,EAAE,CAAC;IAEtD,YAAoB,MAAiB;QAAjB,WAAM,GAAN,MAAM,CAAW;QACnC,MAAM,CAAC,EAAE,CACP,SAAS,EACT,CAAC,GAAoC,EAAE,QAAiB,EAAE,EAAE;YAC1D,MAAM,KAAK,GACT,GAAG,YAAY,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,GAAkB,CAAC,CAAC;YACvE,MAAM,KAAK,GAAY,QAAQ;gBAC7B,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE;gBACtC,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC;YAC1D,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACpC,IAAI,MAAM;gBAAE,MAAM,CAAC,KAAK,CAAC,CAAC;;gBACrB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC,CACF,CAAC;QAEF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC;YAClC,IAAI,CAAC,gBAAgB,GAAG,cAAc,CAAC;YACvC,MAAM,KAAK,GAAG,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;YAC/C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBAClC,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC;YAClC,IAAI,CAAC,gBAAgB,GAAG,cAAc,CAAC;YACvC,sDAAsD;YACtD,sDAAsD;YACtD,MAAM,KAAK,GAAG,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;YAC/C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBAClC,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,KAAmB,CAAC;IAEhC,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAgB;QAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAY;QACzB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAa,EAAE,MAAe;QACxC,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC;QAClC,IAAI,CAAC,gBAAgB,GAAG,cAAc,CAAC;QACvC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAClC,CAAC;CACF"}
+{"version":3,"file":"ws-adapter.js","sourceRoot":"","sources":["../src/ws-adapter.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EACL,2BAA2B,EAE5B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAQhD;;;;;;;;GAQG;AACH,MAAM,OAAO,SAAS;IAYV,MAAM;IACN,WAAW;IAZrB,WAAW,GAAiC,WAAW,CAAC;IACxD,gBAAgB,GAAiC,WAAW,CAAC;IAErD,KAAK,GAAc,EAAE,CAAC;IACtB,OAAO,GAAoC,EAAE,CAAC;IAC9C,YAAY,GAAG,KAAK,CAAC;IACZ,MAAM,CAA2B;IAC1C,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC5B,WAAW,GAA0B,IAAI,CAAC;IAElD,YACU,MAAiB,EACjB,WAAW,GAAyB,IAAI,oBAAoB,EAAE,EACtE,MAAM,GAA6B,2BAA2B,EAAE;sBAFxD,MAAM;2BACN,WAAW;QAGnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,MAAM,CAAC,EAAE,CACP,SAAS,EACT,CAAC,GAAoC,EAAE,QAAiB,EAAE,EAAE;YAC1D,IAAI,IAAI,CAAC,YAAY;gBAAE,OAAO;YAC9B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACjC,iEAAiE;YACjE,uEAAuE;YACvE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC9B,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,6BAA6B,CAAC,CAAC;gBACpD,OAAO;YACT,CAAC;YACD,MAAM,KAAK,GACT,GAAG,YAAY,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,GAAkB,CAAC,CAAC;YACvE,MAAM,KAAK,GAAY,QAAQ;gBAC7B,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE;gBACtC,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC;YAC1D,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACpC,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,KAAK,CAAC,CAAC;gBACd,OAAO;YACT,CAAC;YACD,uEAAuE;YACvE,sEAAsE;YACtE,uDAAuD;YACvD,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;gBACrD,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,wBAAwB,CAAC,CAAC;gBAC/C,OAAO;YACT,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC,CACF,CAAC;QAEF,yEAAyE;QACzE,wCAAwC;QACxC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QAC7C,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QAC7C,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElD,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAC7B,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACnC,CAAC;IAEO,SAAS,CAAC,IAAY,EAAE,MAAc;QAC5C,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,gCAAgC;QAClC,CAAC;QACD,IAAI,CAAC,gBAAgB,EAAE,CAAC;IAC1B,CAAC;IAED;;;;;;OAMG;IACK,gBAAgB;QACtB,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC;QAClC,IAAI,CAAC,gBAAgB,GAAG,cAAc,CAAC;QACvC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAY,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;QACxD,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO;gBAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YACjD,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,mBAAmB;QACzB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,CAAC;YAAE,OAAO;QAEpE,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE;YAClC,IAAI,IAAI,CAAC,YAAY;gBAAE,OAAO;YAE9B,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBAClE,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxB,IAAI,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;gBAC5B,CAAC;gBAAC,MAAM,CAAC;oBACP,sBAAsB;gBACxB,CAAC;gBACD,OAAO;YACT,CAAC;YAED,IAAI,CAAC;gBACH,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,kEAAkE;gBAClE,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAE/B,8CAA8C;QAC9C,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,CAAC;IAC7B,CAAC;IAEO,kBAAkB;QACxB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAChC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,KAAmB,CAAC;IAEhC,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAgB;QAC9B,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAY;QACzB,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,IAAI,CAAC,IAAyB;QAC1C,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,4DAA4D;YAC5D,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,MAAM,EAAE,gBAAgB,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QACzD,IAAI,gBAAgB,IAAI,CAAC;YAAE,OAAO;QAClC,IAAI,IAAI,CAAC,cAAc,EAAE,GAAG,gBAAgB;YAAE,OAAO;QAErD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC;QAC7C,OAAO,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,cAAc,EAAE,IAAI,gBAAgB,EAAE,CAAC;YACvE,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,QAAQ,EAAE,CAAC;gBAC3B,oEAAoE;gBACpE,8BAA8B;gBAC9B,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,4BAA4B,CAAC,CAAC;gBACnD,OAAO;YACT,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,cAAc;QACpB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;QAC5C,OAAO,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAa,EAAE,MAAe;QACxC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAClC,CAAC;CACF"}

package/package.json

@@ -1,7 +1,10 @@
 {
   "name": "@nodetool-ai/websocket",
-  "version": "0.7.0-rc.22",
+  "version": "0.8.1",
   "type": "module",
+  "engines": {
+    "node": ">=22.0.0"
+  },
   "description": "Unified websocket runner for workflow and chat operations",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -13,50 +16,69 @@
     "dev:fast": "node ../../scripts/run-websocket-dev.mjs server stale",
     "test-ui": "node ../../scripts/run-websocket-dev.mjs test-ui",
     "devserver": "npm run test-ui",
-    "test": "vitest run",
-    "test:watch": "vitest",
-    "test:integration": "vitest run --config vitest.config.integration.ts",
-    "lint": "tsc --noEmit"
+    "test": "node ../../scripts/run-vitest.mjs run",
+    "test:watch": "node ../../scripts/run-vitest.mjs",
+    "lint": "node ../../scripts/run-tsc.mjs --noEmit"
   },
   "dependencies": {
-    "@anthropic-ai/claude-agent-sdk": "latest",
     "@fastify/cors": "^10.0.2",
-    "@fastify/static": "^9.1.1",
+    "@fastify/multipart": "^9.4.0",
+    "@fastify/rate-limit": "^10.3.0",
+    "@fastify/static": "^10.1.2",
     "@fastify/websocket": "^11.0.1",
+    "@hyzyla/pdfium": "^2.1.12",
+    "@modelcontextprotocol/ext-apps": "^1.7.2",
     "@modelcontextprotocol/sdk": "^1.27.1",
-    "@nodetool-ai/agents": "^0.7.0-rc.22",
-    "@nodetool-ai/auth": "^0.7.0-rc.22",
-    "@nodetool-ai/base-nodes": "^0.7.0-rc.22",
-    "@nodetool-ai/chat": "^0.7.0-rc.22",
-    "@nodetool-ai/config": "^0.7.0-rc.22",
-    "@nodetool-ai/dsl": "^0.7.0-rc.22",
-    "@nodetool-ai/elevenlabs-nodes": "^0.7.0-rc.22",
-    "@nodetool-ai/transformers-js-nodes": "^0.7.0-rc.22",
-    "@nodetool-ai/transformers-js-provider": "^0.7.0-rc.22",
-    "@nodetool-ai/fal-nodes": "^0.7.0-rc.22",
-    "@nodetool-ai/kie-nodes": "^0.7.0-rc.22",
-    "@nodetool-ai/huggingface": "^0.7.0-rc.22",
-    "@nodetool-ai/kernel": "^0.7.0-rc.22",
-    "@nodetool-ai/models": "^0.7.0-rc.22",
-    "@nodetool-ai/node-sdk": "^0.7.0-rc.22",
-    "@nodetool-ai/protocol": "^0.7.0-rc.22",
-    "@nodetool-ai/replicate-nodes": "^0.7.0-rc.22",
-    "@nodetool-ai/runtime": "^0.7.0-rc.22",
-    "@nodetool-ai/security": "^0.7.0-rc.22",
-    "@nodetool-ai/vectorstore": "^0.7.0-rc.22",
-    "@mariozechner/pi-coding-agent": "^0.67.4",
-    "@openai/codex-sdk": "^0.118.0",
-    "@opencode-ai/sdk": "^1.3.15",
-    "fastify": "^5.8.5",
-    "@trpc/server": "^11.0.0",
+    "@nodetool-ai/agents": "^0.8.1",
+    "@nodetool-ai/app-runtime": "^0.8.1",
+    "@nodetool-ai/atlascloud-nodes": "^0.8.1",
+    "@nodetool-ai/auth": "^0.8.1",
+    "@nodetool-ai/automation-nodes": "^0.8.1",
+    "@nodetool-ai/base-nodes": "^0.8.1",
+    "@nodetool-ai/browser": "^0.8.1",
+    "@nodetool-ai/chat": "^0.8.1",
+    "@nodetool-ai/compute": "^0.8.1",
+    "@nodetool-ai/config": "^0.8.1",
+    "@nodetool-ai/document-nodes": "^0.8.1",
+    "@nodetool-ai/dsl": "^0.8.1",
+    "@nodetool-ai/elevenlabs-nodes": "^0.8.1",
+    "@nodetool-ai/execution": "^0.8.1",
+    "@nodetool-ai/fal-nodes": "^0.8.1",
+    "@nodetool-ai/game-runtime": "^0.8.1",
+    "@nodetool-ai/huggingface": "^0.8.1",
+    "@nodetool-ai/huggingface-nodes": "^0.8.1",
+    "@nodetool-ai/higgsfield-nodes": "^0.8.1",
+    "@nodetool-ai/image-editor": "^0.8.1",
+    "@nodetool-ai/kernel": "^0.8.1",
+    "@nodetool-ai/kie-nodes": "^0.8.1",
+    "@nodetool-ai/minimax-nodes": "^0.8.1",
+    "@nodetool-ai/model-pricing": "^0.8.1",
+    "@nodetool-ai/models": "^0.8.1",
+    "@nodetool-ai/node-sdk": "^0.8.1",
+    "@nodetool-ai/protocol": "^0.8.1",
+    "@nodetool-ai/replicate-nodes": "^0.8.1",
+    "@nodetool-ai/reve-nodes": "^0.8.1",
+    "@nodetool-ai/runtime": "^0.8.1",
+    "@nodetool-ai/sandbox-compiler": "^0.8.1",
+    "@nodetool-ai/security": "^0.8.1",
+    "@nodetool-ai/storage": "^0.8.1",
+    "@nodetool-ai/timeline": "^0.8.1",
+    "@nodetool-ai/together-nodes": "^0.8.1",
+    "@nodetool-ai/topaz-nodes": "^0.8.1",
+    "@nodetool-ai/transformers-js-nodes": "^0.8.1",
+    "@nodetool-ai/transformers-js-provider": "^0.8.1",
+    "@nodetool-ai/vectorstore": "^0.8.1",
+    "@trpc/server": "^11.17.0",
+    "fastify": "^5.12.1",
+    "fflate": "^0.8.2",
     "msgpackr": "^1.11.2",
-    "superjson": "^2.2.2",
+    "sharp": "^0.35.3",
     "zod": "^4.0.0"
   },
   "devDependencies": {
     "@types/node": "^22.0.0",
-    "typescript": "^5.7.2",
-    "vitest": "^4.1.2"
+    "typescript": "npm:@typescript/typescript6@6.0.2",
+    "vitest": "^5.0.1"
   },
   "exports": {
     ".": {
@@ -70,6 +92,18 @@
       "types": "./dist/trpc/router.d.ts",
       "import": "./dist/trpc/router.js",
       "default": "./dist/trpc/router.js"
+    },
+    "./assets": {
+      "nodetool-dev": "./src/lib/asset-model-interface.ts",
+      "types": "./dist/lib/asset-model-interface.d.ts",
+      "import": "./dist/lib/asset-model-interface.js",
+      "default": "./dist/lib/asset-model-interface.js"
+    },
+    "./documents": {
+      "nodetool-dev": "./src/lib/document-model-interfaces.ts",
+      "types": "./dist/lib/document-model-interfaces.d.ts",
+      "import": "./dist/lib/document-model-interfaces.js",
+      "default": "./dist/lib/document-model-interfaces.js"
     }
   },
   "license": "MIT",
@@ -88,5 +122,17 @@
   "files": [
     "dist",
     "README.md"
+  ],
+  "keywords": [
+    "nodetool",
+    "ai",
+    "workflow",
+    "workflow-automation",
+    "typescript",
+    "websocket",
+    "fastify",
+    "server",
+    "api",
+    "realtime"
   ]
 }

package/dist/agent/agent-runtime.d.ts

@@ -1,47 +0,0 @@
-/**
- * Agent runtime — manages Claude/Codex/OpenCode agent SDK sessions.
- *
- * Sessions live entirely in-process; renderers connect over WebSocket
- * (see `socket-route.ts`) and provide a transport for streaming
- * messages and bridging frontend tool execution.
- */
-import type { AgentTransport } from "./transport.js";
-import type { AgentGetSessionMessagesRequest, AgentListSessionsRequest, AgentModelDescriptor, AgentModelsRequest, AgentSessionInfoEntry, AgentSessionOptions, AgentTranscriptMessage } from "./types.js";
-export { SYSTEM_PROMPT, type AgentQuerySession, type AgentSdkProvider, } from "./sdk-provider.js";
-/**
- * AgentRuntime — manages all active agent sessions.
- *
- * One instance is shared per process. Each WebSocket connection routes its
- * commands through this runtime; sessions are addressable by their sessionId
- * from any connection (so a user can reconnect mid-stream), but every
- * mutating operation is gated by the calling user's id — sessionIds are
- * predictable (counter-based for the temp id; thread-id for LLM after the
- * first turn) so the runtime must enforce ownership rather than rely on
- * sessionId being a secret.
- */
-declare class AgentRuntime {
-    private readonly activeSessions;
-    /** Owning userId for each entry in `activeSessions`. */
-    private readonly sessionOwners;
-    private sessionCounter;
-    /**
-     * Throw a uniform "not yours" error so callers can't probe session ids.
-     * Returns the session if the caller owns it, otherwise throws.
-     */
-    private getOwnedSession;
-    createSession(options: AgentSessionOptions, userId: string): Promise<string>;
-    /**
-     * Send a message and stream replies through the given transport.
-     * Returns when the agent turn completes.
-     */
-    sendMessageStreaming(sessionId: string, message: string, transport: AgentTransport, userId: string): Promise<void>;
-    stopExecution(sessionId: string, userId: string): Promise<void>;
-    closeSession(sessionId: string, userId: string): void;
-    closeAllSessions(): void;
-    listModels(options: AgentModelsRequest, userId: string): Promise<AgentModelDescriptor[]>;
-    listSessionsForRequest(options: AgentListSessionsRequest, userId: string): Promise<AgentSessionInfoEntry[]>;
-    getSessionMessagesForRequest(options: AgentGetSessionMessagesRequest, userId: string): Promise<AgentTranscriptMessage[]>;
-}
-export declare function getAgentRuntime(): AgentRuntime;
-export type { AgentRuntime };
-//# sourceMappingURL=agent-runtime.d.ts.map

package/dist/agent/agent-runtime.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"agent-runtime.d.ts","sourceRoot":"","sources":["../../src/agent/agent-runtime.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAwCH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,KAAK,EACV,8BAA8B,EAC9B,wBAAwB,EAExB,oBAAoB,EAEpB,kBAAkB,EAClB,qBAAqB,EACrB,mBAAmB,EACnB,sBAAsB,EAEvB,MAAM,YAAY,CAAC;AASpB,OAAO,EACL,aAAa,EACb,KAAK,iBAAiB,EACtB,KAAK,gBAAgB,GACtB,MAAM,mBAAmB,CAAC;AA2hB3B;;;;;;;;;;GAUG;AACH,cAAM,YAAY;IAChB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAwC;IACvE,wDAAwD;IACxD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA6B;IAC3D,OAAO,CAAC,cAAc,CAAK;IAE3B;;;OAGG;IACH,OAAO,CAAC,eAAe;IAcjB,aAAa,CACjB,OAAO,EAAE,mBAAmB,EAC5B,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,CAAC;IA6ClB;;;OAGG;IACG,oBAAoB,CACxB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,cAAc,EACzB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAwEV,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKrE,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAcrD,gBAAgB,IAAI,IAAI;IAYlB,UAAU,CACd,OAAO,EAAE,kBAAkB,EAC3B,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAK5B,sBAAsB,CAC1B,OAAO,EAAE,wBAAwB,EACjC,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAW7B,4BAA4B,CAChC,OAAO,EAAE,8BAA8B,EACvC,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,sBAAsB,EAAE,CAAC;CASrC;AAID,wBAAgB,eAAe,IAAI,YAAY,CAK9C;AAED,YAAY,EAAE,YAAY,EAAE,CAAC"}