@webiny/api-websockets-standalone 0.0.0-unstable.5387e5f7c8 → 0.0.0-unstable.7c2def32a7

package/adapter/NodeWsAdapter.js

@@ -1,6 +1,19 @@
-import { WebSocketServer } from "ws";
+import { WebSocket, WebSocketServer } from "ws";
 import { WebsocketsServerAdapter } from "./abstractions.js";
 const MAX_PAYLOAD_BYTES = 65536;
+const SERVICE_RESTART = 1012;
+const CLOSE_WAIT_MS = 1000;
+const closeForRestart = (client)=>new Promise((resolve)=>{
+        if (client.readyState === WebSocket.CLOSED) return void resolve();
+        const timer = setTimeout(()=>{
+            client.terminate();
+        }, CLOSE_WAIT_MS);
+        client.once("close", ()=>{
+            clearTimeout(timer);
+            resolve();
+        });
+        client.close(SERVICE_RESTART, "Server restarting");
+    });
 class NodeWsAdapterImpl {
     getWss() {
         if (!this.wss) throw new Error("WebSocketServer has not been started. Call start() first.");
@@ -17,10 +30,16 @@ class NodeWsAdapterImpl {
             this.getWss().emit("connection", ws, request);
         });
     }
-    stop() {
-        const wss = this.getWss();
-        for (const client of wss.clients)client.terminate();
-        return new Promise((resolve, reject)=>{
+    async stop() {
+        const wss = this.wss;
+        if (!wss) return;
+        this.wss = void 0;
+        const clients = [
+            ...wss.clients
+        ];
+        const closing = clients.map(closeForRestart);
+        await Promise.all(closing);
+        await new Promise((resolve, reject)=>{
             wss.close((err)=>{
                 if (err) return void reject(err);
                 resolve();

package/adapter/NodeWsAdapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"adapter/NodeWsAdapter.js","sources":["../../src/adapter/NodeWsAdapter.ts"],"sourcesContent":["import type { Server as HttpServer } from \"node:http\";\nimport type { IncomingMessage } from \"node:http\";\nimport type { Duplex } from \"node:stream\";\nimport { WebSocketServer } from \"ws\";\nimport { WebSocket } from \"ws\";\nimport { WebsocketsServerAdapter } from \"./abstractions.js\";\n\n/*\n * Clients only send small JSON envelopes (a ping carrying a token). The `ws` default is 100 MiB,\n * which any client, authenticated or not, could make the server buffer per message.\n */\nconst MAX_PAYLOAD_BYTES = 64 * 1024;\n\nclass NodeWsAdapterImpl implements WebsocketsServerAdapter.Interface<WebSocket> {\n    private wss: WebSocketServer | undefined;\n\n    private getWss(): WebSocketServer {\n        if (!this.wss) {\n            throw new Error(\"WebSocketServer has not been started. Call start() first.\");\n        }\n        return this.wss;\n    }\n\n    /* The server param is kept for interface compliance but not used —\n       the HTTP upgrade event is handled externally via handleUpgrade(). */\n    public start(_server: HttpServer): void {\n        this.wss = new WebSocketServer({ noServer: true, maxPayload: MAX_PAYLOAD_BYTES });\n    }\n\n    public handleUpgrade(request: IncomingMessage, socket: Duplex, head: Buffer): void {\n        this.getWss().handleUpgrade(request, socket, head, ws => {\n            this.getWss().emit(\"connection\", ws, request);\n        });\n    }\n\n    public stop(): Promise<void> {\n        const wss = this.getWss();\n        /* Terminate all open client connections so wss.close() resolves immediately. */\n        for (const client of wss.clients) {\n            client.terminate();\n        }\n        return new Promise((resolve, reject) => {\n            wss.close(err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    public onConnection(\n        cb: (socket: WebSocket, request: import(\"node:http\").IncomingMessage) => void\n    ): void {\n        this.getWss().on(\"connection\", cb);\n    }\n\n    public onMessage(socket: WebSocket, cb: (data: Buffer) => void): void {\n        socket.on(\"message\", cb);\n    }\n\n    public onClose(socket: WebSocket, cb: (code: number, reason: Buffer) => void): void {\n        socket.on(\"close\", cb);\n    }\n\n    public onError(socket: WebSocket, cb: (error: Error) => void): void {\n        socket.on(\"error\", cb);\n    }\n\n    public send(socket: WebSocket, data: string): Promise<void> {\n        return new Promise((resolve, reject) => {\n            socket.send(data, err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    public close(socket: WebSocket, code?: number, reason?: string): void {\n        socket.close(code, reason);\n    }\n}\n\nexport { NodeWsAdapterImpl };\n\nexport const NodeWsAdapter = WebsocketsServerAdapter.createImplementation({\n    implementation: NodeWsAdapterImpl,\n    dependencies: []\n});\n"],"names":["MAX_PAYLOAD_BYTES","NodeWsAdapterImpl","Error","_server","WebSocketServer","request","socket","head","ws","wss","client","Promise","resolve","reject","err","cb","data","code","reason","NodeWsAdapter","WebsocketsServerAdapter"],"mappings":";;AAWA,MAAMA,oBAAoB;AAE1B,MAAMC;IAGM,SAA0B;QAC9B,IAAI,CAAC,IAAI,CAAC,GAAG,EACT,MAAM,IAAIC,MAAM;QAEpB,OAAO,IAAI,CAAC,GAAG;IACnB;IAIO,MAAMC,OAAmB,EAAQ;QACpC,IAAI,CAAC,GAAG,GAAG,IAAIC,gBAAgB;YAAE,UAAU;YAAM,YAAYJ;QAAkB;IACnF;IAEO,cAAcK,OAAwB,EAAEC,MAAc,EAAEC,IAAY,EAAQ;QAC/E,IAAI,CAAC,MAAM,GAAG,aAAa,CAACF,SAASC,QAAQC,MAAMC,CAAAA;YAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,cAAcA,IAAIH;QACzC;IACJ;IAEO,OAAsB;QACzB,MAAMI,MAAM,IAAI,CAAC,MAAM;QAEvB,KAAK,MAAMC,UAAUD,IAAI,OAAO,CAC5BC,OAAO,SAAS;QAEpB,OAAO,IAAIC,QAAQ,CAACC,SAASC;YACzBJ,IAAI,KAAK,CAACK,CAAAA;gBACN,IAAIA,KAAK,YACLD,OAAOC;gBAGXF;YACJ;QACJ;IACJ;IAEO,aACHG,EAA6E,EACzE;QACJ,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,cAAcA;IACnC;IAEO,UAAUT,MAAiB,EAAES,EAA0B,EAAQ;QAClET,OAAO,EAAE,CAAC,WAAWS;IACzB;IAEO,QAAQT,MAAiB,EAAES,EAA0C,EAAQ;QAChFT,OAAO,EAAE,CAAC,SAASS;IACvB;IAEO,QAAQT,MAAiB,EAAES,EAA0B,EAAQ;QAChET,OAAO,EAAE,CAAC,SAASS;IACvB;IAEO,KAAKT,MAAiB,EAAEU,IAAY,EAAiB;QACxD,OAAO,IAAIL,QAAQ,CAACC,SAASC;YACzBP,OAAO,IAAI,CAACU,MAAMF,CAAAA;gBACd,IAAIA,KAAK,YACLD,OAAOC;gBAGXF;YACJ;QACJ;IACJ;IAEO,MAAMN,MAAiB,EAAEW,IAAa,EAAEC,MAAe,EAAQ;QAClEZ,OAAO,KAAK,CAACW,MAAMC;IACvB;AACJ;AAIO,MAAMC,gBAAgBC,wBAAwB,oBAAoB,CAAC;IACtE,gBAAgBnB;IAChB,cAAc,EAAE;AACpB"}
+{"version":3,"file":"adapter/NodeWsAdapter.js","sources":["../../src/adapter/NodeWsAdapter.ts"],"sourcesContent":["import type { Server as HttpServer } from \"node:http\";\nimport type { IncomingMessage } from \"node:http\";\nimport type { Duplex } from \"node:stream\";\nimport { WebSocketServer } from \"ws\";\nimport { WebSocket } from \"ws\";\nimport { WebsocketsServerAdapter } from \"./abstractions.js\";\n\n/*\n * Clients only send small JSON envelopes (a ping carrying a token). The `ws` default is 100 MiB,\n * which any client, authenticated or not, could make the server buffer per message.\n */\nconst MAX_PAYLOAD_BYTES = 64 * 1024;\n\n/*\n * Sent to every client when the server stops. 1012 means \"service restart\": the client reconnects,\n * and behind a load balancer it reaches a process that is still up.\n */\nconst SERVICE_RESTART = 1012;\n\n// How long a client gets to answer the close before its socket is cut.\nconst CLOSE_WAIT_MS = 1000;\n\n/*\n * Closes the socket with `SERVICE_RESTART` and resolves once it is closed. A client that has not\n * answered within `CLOSE_WAIT_MS` (a frozen tab, a dead network) is cut.\n */\nconst closeForRestart = (client: WebSocket): Promise<void> => {\n    return new Promise(resolve => {\n        if (client.readyState === WebSocket.CLOSED) {\n            resolve();\n            return;\n        }\n        const timer = setTimeout(() => {\n            client.terminate();\n        }, CLOSE_WAIT_MS);\n        client.once(\"close\", () => {\n            clearTimeout(timer);\n            resolve();\n        });\n        client.close(SERVICE_RESTART, \"Server restarting\");\n    });\n};\n\nclass NodeWsAdapterImpl implements WebsocketsServerAdapter.Interface<WebSocket> {\n    private wss: WebSocketServer | undefined;\n\n    private getWss(): WebSocketServer {\n        if (!this.wss) {\n            throw new Error(\"WebSocketServer has not been started. Call start() first.\");\n        }\n        return this.wss;\n    }\n\n    /* The server param is kept for interface compliance but not used —\n       the HTTP upgrade event is handled externally via handleUpgrade(). */\n    public start(_server: HttpServer): void {\n        this.wss = new WebSocketServer({ noServer: true, maxPayload: MAX_PAYLOAD_BYTES });\n    }\n\n    public handleUpgrade(request: IncomingMessage, socket: Duplex, head: Buffer): void {\n        this.getWss().handleUpgrade(request, socket, head, ws => {\n            this.getWss().emit(\"connection\", ws, request);\n        });\n    }\n\n    // Safe to call again: a stopped adapter has nothing left to close.\n    public async stop(): Promise<void> {\n        const wss = this.wss;\n        if (!wss) {\n            return;\n        }\n        this.wss = undefined;\n\n        const clients = [...wss.clients];\n        const closing = clients.map(closeForRestart);\n        await Promise.all(closing);\n\n        await new Promise<void>((resolve, reject) => {\n            wss.close(err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    public onConnection(\n        cb: (socket: WebSocket, request: import(\"node:http\").IncomingMessage) => void\n    ): void {\n        this.getWss().on(\"connection\", cb);\n    }\n\n    public onMessage(socket: WebSocket, cb: (data: Buffer) => void): void {\n        socket.on(\"message\", cb);\n    }\n\n    public onClose(socket: WebSocket, cb: (code: number, reason: Buffer) => void): void {\n        socket.on(\"close\", cb);\n    }\n\n    public onError(socket: WebSocket, cb: (error: Error) => void): void {\n        socket.on(\"error\", cb);\n    }\n\n    public send(socket: WebSocket, data: string): Promise<void> {\n        return new Promise((resolve, reject) => {\n            socket.send(data, err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    public close(socket: WebSocket, code?: number, reason?: string): void {\n        socket.close(code, reason);\n    }\n}\n\nexport { NodeWsAdapterImpl };\n\nexport const NodeWsAdapter = WebsocketsServerAdapter.createImplementation({\n    implementation: NodeWsAdapterImpl,\n    dependencies: []\n});\n"],"names":["MAX_PAYLOAD_BYTES","SERVICE_RESTART","CLOSE_WAIT_MS","closeForRestart","client","Promise","resolve","WebSocket","timer","setTimeout","clearTimeout","NodeWsAdapterImpl","Error","_server","WebSocketServer","request","socket","head","ws","wss","undefined","clients","closing","reject","err","cb","data","code","reason","NodeWsAdapter","WebsocketsServerAdapter"],"mappings":";;AAWA,MAAMA,oBAAoB;AAM1B,MAAMC,kBAAkB;AAGxB,MAAMC,gBAAgB;AAMtB,MAAMC,kBAAkB,CAACC,SACd,IAAIC,QAAQC,CAAAA;QACf,IAAIF,OAAO,UAAU,KAAKG,UAAU,MAAM,EAAE,YACxCD;QAGJ,MAAME,QAAQC,WAAW;YACrBL,OAAO,SAAS;QACpB,GAAGF;QACHE,OAAO,IAAI,CAAC,SAAS;YACjBM,aAAaF;YACbF;QACJ;QACAF,OAAO,KAAK,CAACH,iBAAiB;IAClC;AAGJ,MAAMU;IAGM,SAA0B;QAC9B,IAAI,CAAC,IAAI,CAAC,GAAG,EACT,MAAM,IAAIC,MAAM;QAEpB,OAAO,IAAI,CAAC,GAAG;IACnB;IAIO,MAAMC,OAAmB,EAAQ;QACpC,IAAI,CAAC,GAAG,GAAG,IAAIC,gBAAgB;YAAE,UAAU;YAAM,YAAYd;QAAkB;IACnF;IAEO,cAAce,OAAwB,EAAEC,MAAc,EAAEC,IAAY,EAAQ;QAC/E,IAAI,CAAC,MAAM,GAAG,aAAa,CAACF,SAASC,QAAQC,MAAMC,CAAAA;YAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,cAAcA,IAAIH;QACzC;IACJ;IAGA,MAAa,OAAsB;QAC/B,MAAMI,MAAM,IAAI,CAAC,GAAG;QACpB,IAAI,CAACA,KACD;QAEJ,IAAI,CAAC,GAAG,GAAGC;QAEX,MAAMC,UAAU;eAAIF,IAAI,OAAO;SAAC;QAChC,MAAMG,UAAUD,QAAQ,GAAG,CAAClB;QAC5B,MAAME,QAAQ,GAAG,CAACiB;QAElB,MAAM,IAAIjB,QAAc,CAACC,SAASiB;YAC9BJ,IAAI,KAAK,CAACK,CAAAA;gBACN,IAAIA,KAAK,YACLD,OAAOC;gBAGXlB;YACJ;QACJ;IACJ;IAEO,aACHmB,EAA6E,EACzE;QACJ,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,cAAcA;IACnC;IAEO,UAAUT,MAAiB,EAAES,EAA0B,EAAQ;QAClET,OAAO,EAAE,CAAC,WAAWS;IACzB;IAEO,QAAQT,MAAiB,EAAES,EAA0C,EAAQ;QAChFT,OAAO,EAAE,CAAC,SAASS;IACvB;IAEO,QAAQT,MAAiB,EAAES,EAA0B,EAAQ;QAChET,OAAO,EAAE,CAAC,SAASS;IACvB;IAEO,KAAKT,MAAiB,EAAEU,IAAY,EAAiB;QACxD,OAAO,IAAIrB,QAAQ,CAACC,SAASiB;YACzBP,OAAO,IAAI,CAACU,MAAMF,CAAAA;gBACd,IAAIA,KAAK,YACLD,OAAOC;gBAGXlB;YACJ;QACJ;IACJ;IAEO,MAAMU,MAAiB,EAAEW,IAAa,EAAEC,MAAe,EAAQ;QAClEZ,OAAO,KAAK,CAACW,MAAMC;IACvB;AACJ;AAIO,MAAMC,gBAAgBC,wBAAwB,oBAAoB,CAAC;IACtE,gBAAgBnB;IAChB,cAAc,EAAE;AACpB"}

package/connectionManager/ServerConnectionManager.d.ts

@@ -1,12 +1,15 @@
 import { ConnectionRegistry } from "@webiny/api-websockets/exports/api.js";
+import { ProcessOwner } from "@webiny/event-handler-standalone";
 import { WebsocketsConnectionManager } from "./abstractions.js";
 export declare class ServerConnectionManagerImpl implements WebsocketsConnectionManager.Interface<unknown> {
     private readonly registry;
+    private readonly owner;
     private readonly sockets;
     private readonly metadata;
-    constructor(registry: ConnectionRegistry.Interface);
+    constructor(registry: ConnectionRegistry.Interface, owner: ProcessOwner.Interface);
     add(params: WebsocketsConnectionManager.AddParams<unknown>): Promise<void>;
     remove(connectionId: string): Promise<void>;
+    removeOwnedHere(connectionIds: string[]): Promise<void>;
     getSocket(connectionId: string): unknown | undefined;
     getMetadata(connectionId: string): WebsocketsConnectionManager.ConnectionMetadata | undefined;
     updateLastSeen(connectionId: string): Promise<void>;

package/connectionManager/ServerConnectionManager.js

@@ -1,8 +1,10 @@
 import { ConnectionRegistry } from "@webiny/api-websockets/exports/api.js";
+import { ProcessOwner } from "@webiny/event-handler-standalone";
 import { WebsocketsConnectionManager } from "./abstractions.js";
 class ServerConnectionManagerImpl {
-    constructor(registry){
+    constructor(registry, owner){
         this.registry = registry;
+        this.owner = owner;
         this.sockets = new Map();
         this.metadata = new Map();
     }
@@ -20,7 +22,8 @@ class ServerConnectionManagerImpl {
             identity: params.identity,
             tenant: params.tenant,
             endpoint: params.endpoint,
-            connectedOn: new Date(params.connectedAt).toISOString()
+            connectedOn: new Date(params.connectedAt).toISOString(),
+            owner: this.owner.id
         });
     }
     async remove(connectionId) {
@@ -35,6 +38,12 @@ class ServerConnectionManagerImpl {
             throw err;
         }
     }
+    async removeOwnedHere(connectionIds) {
+        if (0 === connectionIds.length) return;
+        const connections = await this.registry.listViaConnections(connectionIds);
+        const owned = connections.filter((connection)=>connection.owner === this.owner.id);
+        for (const connection of owned)await this.remove(connection.connectionId);
+    }
     getSocket(connectionId) {
         return this.sockets.get(connectionId);
     }
@@ -61,7 +70,8 @@ class ServerConnectionManagerImpl {
 const ServerConnectionManager = WebsocketsConnectionManager.createImplementation({
     implementation: ServerConnectionManagerImpl,
     dependencies: [
-        ConnectionRegistry
+        ConnectionRegistry,
+        ProcessOwner
     ]
 });
 export { ServerConnectionManager, ServerConnectionManagerImpl };

package/connectionManager/ServerConnectionManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"connectionManager/ServerConnectionManager.js","sources":["../../src/connectionManager/ServerConnectionManager.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { WebsocketsConnectionManager } from \"./abstractions.js\";\n\nexport class ServerConnectionManagerImpl implements WebsocketsConnectionManager.Interface<unknown> {\n    private readonly sockets = new Map<string, unknown>();\n    private readonly metadata = new Map<string, WebsocketsConnectionManager.ConnectionMetadata>();\n\n    public constructor(private readonly registry: ConnectionRegistry.Interface) {}\n\n    public async add(params: WebsocketsConnectionManager.AddParams<unknown>): Promise<void> {\n        this.sockets.set(params.connectionId, params.socket);\n        this.metadata.set(params.connectionId, {\n            connectionId: params.connectionId,\n            endpoint: params.endpoint,\n            connectedAt: params.connectedAt,\n            host: params.host,\n            headers: params.headers\n        });\n\n        // Save the connection to the shared registry. This is what lets the server later push a\n        // message to a specific user — e.g. when a background task finishes and calls\n        // SendToIdentity. That flow looks connections up in the registry by identity id to get their\n        // connection ids, then comes back here to fetch each live socket by connection id and send.\n        // The in-memory map above only answers \"socket for this connection id?\" — it can't be\n        // searched by identity, so without this registry row a targeted send finds nobody. The\n        // registry stores `connectedOn` as an ISO string (it compares connections against a recency\n        // cutoff), so we convert the epoch-ms `connectedAt`.\n        await this.registry.register({\n            connectionId: params.connectionId,\n            identity: params.identity,\n            tenant: params.tenant,\n            endpoint: params.endpoint,\n            connectedOn: new Date(params.connectedAt).toISOString()\n        });\n    }\n\n    public async remove(connectionId: string): Promise<void> {\n        this.sockets.delete(connectionId);\n        this.metadata.delete(connectionId);\n        try {\n            await this.registry.unregister({ connectionId });\n        } catch (err) {\n            if ((err as any).code === \"CONNECTION_NOT_FOUND\") {\n                return;\n            }\n            throw err;\n        }\n    }\n\n    public getSocket(connectionId: string): unknown | undefined {\n        return this.sockets.get(connectionId);\n    }\n\n    public getMetadata(\n        connectionId: string\n    ): WebsocketsConnectionManager.ConnectionMetadata | undefined {\n        return this.metadata.get(connectionId);\n    }\n\n    public async updateLastSeen(connectionId: string): Promise<void> {\n        await this.registry.updateLastSeen(connectionId);\n    }\n\n    public async cleanup(maxAge: number): Promise<string[]> {\n        const olderThan = new Date(Date.now() - maxAge);\n        const stale = await this.registry.listStale(olderThan);\n        const removed: string[] = [];\n        for (const entry of stale) {\n            await this.remove(entry.connectionId);\n            removed.push(entry.connectionId);\n        }\n        return removed;\n    }\n\n    public getActiveConnectionIds(): string[] {\n        return Array.from(this.sockets.keys());\n    }\n}\n\nexport const ServerConnectionManager = WebsocketsConnectionManager.createImplementation({\n    implementation: ServerConnectionManagerImpl,\n    dependencies: [ConnectionRegistry]\n});\n"],"names":["ServerConnectionManagerImpl","registry","Map","params","Date","connectionId","err","maxAge","olderThan","stale","removed","entry","Array","ServerConnectionManager","WebsocketsConnectionManager","ConnectionRegistry"],"mappings":";;AAGO,MAAMA;IAIT,YAAoCC,QAAsC,CAAE;aAAxCA,QAAQ,GAARA;aAHnB,OAAO,GAAG,IAAIC;aACd,QAAQ,GAAG,IAAIA;IAE6C;IAE7E,MAAa,IAAIC,MAAsD,EAAiB;QACpF,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA,OAAO,YAAY,EAAEA,OAAO,MAAM;QACnD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA,OAAO,YAAY,EAAE;YACnC,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,aAAaA,OAAO,WAAW;YAC/B,MAAMA,OAAO,IAAI;YACjB,SAASA,OAAO,OAAO;QAC3B;QAUA,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACzB,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,QAAQA,OAAO,MAAM;YACrB,UAAUA,OAAO,QAAQ;YACzB,aAAa,IAAIC,KAAKD,OAAO,WAAW,EAAE,WAAW;QACzD;IACJ;IAEA,MAAa,OAAOE,YAAoB,EAAiB;QACrD,IAAI,CAAC,OAAO,CAAC,MAAM,CAACA;QACpB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAACA;QACrB,IAAI;YACA,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAAEA;YAAa;QAClD,EAAE,OAAOC,KAAK;YACV,IAAKA,AAAqB,2BAArBA,IAAY,IAAI,EACjB;YAEJ,MAAMA;QACV;IACJ;IAEO,UAAUD,YAAoB,EAAuB;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA;IAC5B;IAEO,YACHA,YAAoB,EACsC;QAC1D,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA;IAC7B;IAEA,MAAa,eAAeA,YAAoB,EAAiB;QAC7D,MAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,CAACA;IACvC;IAEA,MAAa,QAAQE,MAAc,EAAqB;QACpD,MAAMC,YAAY,IAAIJ,KAAKA,KAAK,GAAG,KAAKG;QACxC,MAAME,QAAQ,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAACD;QAC5C,MAAME,UAAoB,EAAE;QAC5B,KAAK,MAAMC,SAASF,MAAO;YACvB,MAAM,IAAI,CAAC,MAAM,CAACE,MAAM,YAAY;YACpCD,QAAQ,IAAI,CAACC,MAAM,YAAY;QACnC;QACA,OAAOD;IACX;IAEO,yBAAmC;QACtC,OAAOE,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI;IACvC;AACJ;AAEO,MAAMC,0BAA0BC,4BAA4B,oBAAoB,CAAC;IACpF,gBAAgBd;IAChB,cAAc;QAACe;KAAmB;AACtC"}
+{"version":3,"file":"connectionManager/ServerConnectionManager.js","sources":["../../src/connectionManager/ServerConnectionManager.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { ProcessOwner } from \"@webiny/event-handler-standalone\";\nimport { WebsocketsConnectionManager } from \"./abstractions.js\";\n\nexport class ServerConnectionManagerImpl implements WebsocketsConnectionManager.Interface<unknown> {\n    private readonly sockets = new Map<string, unknown>();\n    private readonly metadata = new Map<string, WebsocketsConnectionManager.ConnectionMetadata>();\n\n    public constructor(\n        private readonly registry: ConnectionRegistry.Interface,\n        private readonly owner: ProcessOwner.Interface\n    ) {}\n\n    public async add(params: WebsocketsConnectionManager.AddParams<unknown>): Promise<void> {\n        this.sockets.set(params.connectionId, params.socket);\n        this.metadata.set(params.connectionId, {\n            connectionId: params.connectionId,\n            endpoint: params.endpoint,\n            connectedAt: params.connectedAt,\n            host: params.host,\n            headers: params.headers\n        });\n\n        // Save the connection to the shared registry. This is what lets the server later push a\n        // message to a specific user — e.g. when a background task finishes and calls\n        // SendToIdentity. That flow looks connections up in the registry by identity id to get their\n        // connection ids, then comes back here to fetch each live socket by connection id and send.\n        // The in-memory map above only answers \"socket for this connection id?\" — it can't be\n        // searched by identity, so without this registry row a targeted send finds nobody. The\n        // registry stores `connectedOn` as an ISO string (it compares connections against a recency\n        // cutoff), so we convert the epoch-ms `connectedAt`.\n        await this.registry.register({\n            connectionId: params.connectionId,\n            identity: params.identity,\n            tenant: params.tenant,\n            endpoint: params.endpoint,\n            connectedOn: new Date(params.connectedAt).toISOString(),\n            owner: this.owner.id\n        });\n    }\n\n    public async remove(connectionId: string): Promise<void> {\n        this.sockets.delete(connectionId);\n        this.metadata.delete(connectionId);\n        try {\n            await this.registry.unregister({ connectionId });\n        } catch (err) {\n            if ((err as any).code === \"CONNECTION_NOT_FOUND\") {\n                return;\n            }\n            throw err;\n        }\n    }\n\n    /*\n     * For connections this process has no socket for. Only the process that holds a socket can tell\n     * that it is gone, so a connection is removed only when this process is its owner. Another\n     * process's connections are left alone: with several processes, a socket missing here is\n     * usually open there. Rows without an owner are left to the heartbeat's stale cleanup.\n     */\n    public async removeOwnedHere(connectionIds: string[]): Promise<void> {\n        if (connectionIds.length === 0) {\n            return;\n        }\n        const connections = await this.registry.listViaConnections(connectionIds);\n        const owned = connections.filter(connection => connection.owner === this.owner.id);\n        for (const connection of owned) {\n            await this.remove(connection.connectionId);\n        }\n    }\n\n    public getSocket(connectionId: string): unknown | undefined {\n        return this.sockets.get(connectionId);\n    }\n\n    public getMetadata(\n        connectionId: string\n    ): WebsocketsConnectionManager.ConnectionMetadata | undefined {\n        return this.metadata.get(connectionId);\n    }\n\n    public async updateLastSeen(connectionId: string): Promise<void> {\n        await this.registry.updateLastSeen(connectionId);\n    }\n\n    public async cleanup(maxAge: number): Promise<string[]> {\n        const olderThan = new Date(Date.now() - maxAge);\n        const stale = await this.registry.listStale(olderThan);\n        const removed: string[] = [];\n        for (const entry of stale) {\n            await this.remove(entry.connectionId);\n            removed.push(entry.connectionId);\n        }\n        return removed;\n    }\n\n    public getActiveConnectionIds(): string[] {\n        return Array.from(this.sockets.keys());\n    }\n}\n\nexport const ServerConnectionManager = WebsocketsConnectionManager.createImplementation({\n    implementation: ServerConnectionManagerImpl,\n    dependencies: [ConnectionRegistry, ProcessOwner]\n});\n"],"names":["ServerConnectionManagerImpl","registry","owner","Map","params","Date","connectionId","err","connectionIds","connections","owned","connection","maxAge","olderThan","stale","removed","entry","Array","ServerConnectionManager","WebsocketsConnectionManager","ConnectionRegistry","ProcessOwner"],"mappings":";;;AAIO,MAAMA;IAIT,YACqBC,QAAsC,EACtCC,KAA6B,CAChD;aAFmBD,QAAQ,GAARA;aACAC,KAAK,GAALA;aALJ,OAAO,GAAG,IAAIC;aACd,QAAQ,GAAG,IAAIA;IAK7B;IAEH,MAAa,IAAIC,MAAsD,EAAiB;QACpF,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA,OAAO,YAAY,EAAEA,OAAO,MAAM;QACnD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA,OAAO,YAAY,EAAE;YACnC,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,aAAaA,OAAO,WAAW;YAC/B,MAAMA,OAAO,IAAI;YACjB,SAASA,OAAO,OAAO;QAC3B;QAUA,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACzB,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,QAAQA,OAAO,MAAM;YACrB,UAAUA,OAAO,QAAQ;YACzB,aAAa,IAAIC,KAAKD,OAAO,WAAW,EAAE,WAAW;YACrD,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE;QACxB;IACJ;IAEA,MAAa,OAAOE,YAAoB,EAAiB;QACrD,IAAI,CAAC,OAAO,CAAC,MAAM,CAACA;QACpB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAACA;QACrB,IAAI;YACA,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAAEA;YAAa;QAClD,EAAE,OAAOC,KAAK;YACV,IAAKA,AAAqB,2BAArBA,IAAY,IAAI,EACjB;YAEJ,MAAMA;QACV;IACJ;IAQA,MAAa,gBAAgBC,aAAuB,EAAiB;QACjE,IAAIA,AAAyB,MAAzBA,cAAc,MAAM,EACpB;QAEJ,MAAMC,cAAc,MAAM,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAACD;QAC3D,MAAME,QAAQD,YAAY,MAAM,CAACE,CAAAA,aAAcA,WAAW,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,EAAE;QACjF,KAAK,MAAMA,cAAcD,MACrB,MAAM,IAAI,CAAC,MAAM,CAACC,WAAW,YAAY;IAEjD;IAEO,UAAUL,YAAoB,EAAuB;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA;IAC5B;IAEO,YACHA,YAAoB,EACsC;QAC1D,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA;IAC7B;IAEA,MAAa,eAAeA,YAAoB,EAAiB;QAC7D,MAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,CAACA;IACvC;IAEA,MAAa,QAAQM,MAAc,EAAqB;QACpD,MAAMC,YAAY,IAAIR,KAAKA,KAAK,GAAG,KAAKO;QACxC,MAAME,QAAQ,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAACD;QAC5C,MAAME,UAAoB,EAAE;QAC5B,KAAK,MAAMC,SAASF,MAAO;YACvB,MAAM,IAAI,CAAC,MAAM,CAACE,MAAM,YAAY;YACpCD,QAAQ,IAAI,CAACC,MAAM,YAAY;QACnC;QACA,OAAOD;IACX;IAEO,yBAAmC;QACtC,OAAOE,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI;IACvC;AACJ;AAEO,MAAMC,0BAA0BC,4BAA4B,oBAAoB,CAAC;IACpF,gBAAgBnB;IAChB,cAAc;QAACoB;QAAoBC;KAAa;AACpD"}

package/connectionManager/abstractions.d.ts

@@ -19,6 +19,7 @@ export interface IWebsocketsConnectionManagerMetadata {
 export interface IWebsocketsConnectionManager<TSocket> {
     add(params: IWebsocketsConnectionManagerAddParams<TSocket>): Promise<void>;
     remove(connectionId: string): Promise<void>;
+    removeOwnedHere(connectionIds: string[]): Promise<void>;
     getSocket(connectionId: string): TSocket | undefined;
     getMetadata(connectionId: string): IWebsocketsConnectionManagerMetadata | undefined;
     updateLastSeen(connectionId: string): Promise<void>;

package/connectionManager/abstractions.js.map

@@ -1 +1 @@
-{"version":3,"file":"connectionManager/abstractions.js","sources":["../../src/connectionManager/abstractions.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { createAbstraction } from \"@webiny/feature/api\";\n\nexport interface IWebsocketsConnectionManagerAddParams<TSocket> {\n    connectionId: string;\n    socket: TSocket;\n    endpoint: string;\n    identity: ConnectionRegistry.Identity;\n    tenant: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManagerMetadata {\n    connectionId: string;\n    endpoint: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManager<TSocket> {\n    add(params: IWebsocketsConnectionManagerAddParams<TSocket>): Promise<void>;\n    remove(connectionId: string): Promise<void>;\n    getSocket(connectionId: string): TSocket | undefined;\n    getMetadata(connectionId: string): IWebsocketsConnectionManagerMetadata | undefined;\n    updateLastSeen(connectionId: string): Promise<void>;\n    cleanup(maxAge: number): Promise<string[]>;\n    getActiveConnectionIds(): string[];\n}\n\nexport const WebsocketsConnectionManager = createAbstraction<IWebsocketsConnectionManager<unknown>>(\n    \"WebsocketsConnectionManager\"\n);\n\nexport namespace WebsocketsConnectionManager {\n    export type Interface<TSocket> = IWebsocketsConnectionManager<TSocket>;\n    export type AddParams<TSocket> = IWebsocketsConnectionManagerAddParams<TSocket>;\n    export type ConnectionMetadata = IWebsocketsConnectionManagerMetadata;\n}\n"],"names":["WebsocketsConnectionManager","createAbstraction"],"mappings":";AAgCO,MAAMA,8BAA8BC,kBACvC"}
+{"version":3,"file":"connectionManager/abstractions.js","sources":["../../src/connectionManager/abstractions.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { createAbstraction } from \"@webiny/feature/api\";\n\nexport interface IWebsocketsConnectionManagerAddParams<TSocket> {\n    connectionId: string;\n    socket: TSocket;\n    endpoint: string;\n    identity: ConnectionRegistry.Identity;\n    tenant: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManagerMetadata {\n    connectionId: string;\n    endpoint: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManager<TSocket> {\n    add(params: IWebsocketsConnectionManagerAddParams<TSocket>): Promise<void>;\n    remove(connectionId: string): Promise<void>;\n    removeOwnedHere(connectionIds: string[]): Promise<void>;\n    getSocket(connectionId: string): TSocket | undefined;\n    getMetadata(connectionId: string): IWebsocketsConnectionManagerMetadata | undefined;\n    updateLastSeen(connectionId: string): Promise<void>;\n    cleanup(maxAge: number): Promise<string[]>;\n    getActiveConnectionIds(): string[];\n}\n\nexport const WebsocketsConnectionManager = createAbstraction<IWebsocketsConnectionManager<unknown>>(\n    \"WebsocketsConnectionManager\"\n);\n\nexport namespace WebsocketsConnectionManager {\n    export type Interface<TSocket> = IWebsocketsConnectionManager<TSocket>;\n    export type AddParams<TSocket> = IWebsocketsConnectionManagerAddParams<TSocket>;\n    export type ConnectionMetadata = IWebsocketsConnectionManagerMetadata;\n}\n"],"names":["WebsocketsConnectionManager","createAbstraction"],"mappings":";AAiCO,MAAMA,8BAA8BC,kBACvC"}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@webiny/api-websockets-standalone",
-	"version": "0.0.0-unstable.5387e5f7c8",
+	"version": "0.0.0-unstable.7c2def32a7",
 	"type": "module",
 	"repository": {
 		"type": "git",
@@ -16,17 +16,18 @@
 	},
 	"license": "MIT",
 	"dependencies": {
-		"@webiny/api": "0.0.0-unstable.5387e5f7c8",
-		"@webiny/api-websockets": "0.0.0-unstable.5387e5f7c8",
-		"@webiny/feature": "0.0.0-unstable.5387e5f7c8",
-		"@webiny/plugins": "0.0.0-unstable.5387e5f7c8",
-		"@webiny/utils": "0.0.0-unstable.5387e5f7c8",
+		"@webiny/api": "0.0.0-unstable.7c2def32a7",
+		"@webiny/api-websockets": "0.0.0-unstable.7c2def32a7",
+		"@webiny/event-handler-standalone": "0.0.0-unstable.7c2def32a7",
+		"@webiny/feature": "0.0.0-unstable.7c2def32a7",
+		"@webiny/plugins": "0.0.0-unstable.7c2def32a7",
+		"@webiny/utils": "0.0.0-unstable.7c2def32a7",
 		"ws": "8.21.3",
 		"zod": "4.4.3"
 	},
 	"devDependencies": {
 		"@types/ws": "8.18.1",
-		"@webiny/build-tools": "0.0.0-unstable.5387e5f7c8",
+		"@webiny/build-tools": "0.0.0-unstable.7c2def32a7",
 		"rimraf": "6.1.3",
 		"typescript": "7.0.2",
 		"vitest": "4.1.11"

package/server/WebsocketsServer.js

@@ -43,11 +43,13 @@ class WebsocketsServer {
     async stop() {
         this.shuttingDown = true;
         this.heartbeat?.stop();
-        await this.adapter.stop();
-        if (this.connectionManager) {
-            const activeIds = this.connectionManager.getActiveConnectionIds();
-            for (const connectionId of activeIds)await this.connectionManager.remove(connectionId);
-        }
+        const activeIds = this.connectionManager?.getActiveConnectionIds() ?? [];
+        const removals = activeIds.map((connectionId)=>this.removeConnection(connectionId));
+        const closing = this.adapter.stop();
+        await Promise.all([
+            closing,
+            ...removals
+        ]);
         if (this.ownsHttpServer) await this.closeHttpServer();
         if (this.debug) console.log("[WebsocketsServer] stopped");
     }

package/server/WebsocketsServer.js.map

@@ -1 +1 @@
-{"version":3,"file":"server/WebsocketsServer.js","sources":["../../src/server/WebsocketsServer.ts"],"sourcesContent":["import { createServer } from \"node:http\";\nimport type { Server as HttpServer, IncomingMessage } from \"node:http\";\nimport type { Duplex } from \"node:stream\";\nimport { mdbid } from \"@webiny/utils\";\nimport type { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\nimport type { WebsocketsUpgradeHandler } from \"~/upgradeHandler/abstractions.js\";\nimport type { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport type { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { NodeWsAdapterImpl } from \"~/adapter/NodeWsAdapter.js\";\nimport { DefaultUpgradeHandlerImpl } from \"~/upgradeHandler/DefaultUpgradeHandler.js\";\nimport { HeartbeatManager } from \"~/heartbeat/HeartbeatManager.js\";\nimport type {\n    CreateWebsocketsServerParams,\n    AttachWebsocketsServerParams,\n    IWebsocketsServer,\n    WebsocketsConnectionAuthenticator\n} from \"./types.js\";\n\nconst ANONYMOUS_IDENTITY: ConnectionRegistry.Identity = { id: \"\", displayName: \"\", type: \"\" };\n\nconst DEFAULT_HEARTBEAT_INTERVAL = 60_000;\n// Three admin ping intervals (5 minutes each): one late ping is not enough to evict a live tab.\nconst DEFAULT_STALE_AFTER = 15 * 60_000;\n\nconst toHeaders = (raw: IncomingMessage[\"headers\"]): Record<string, string> => {\n    const headers: Record<string, string> = {};\n    for (const [key, value] of Object.entries(raw)) {\n        if (typeof value === \"string\") {\n            headers[key] = value;\n        }\n    }\n    return headers;\n};\n\ninterface WebsocketsServerParams {\n    httpServer: HttpServer;\n    ownsHttpServer: boolean;\n    adapter: WebsocketsServerAdapter.Interface<unknown>;\n    upgradeHandler: WebsocketsUpgradeHandler.Interface;\n    connectionManager?: WebsocketsConnectionManager.Interface<unknown>;\n    authenticate?: WebsocketsConnectionAuthenticator;\n    heartbeatInterval: number;\n    staleAfter: number;\n    debug: boolean;\n    port: number;\n    host: string;\n}\n\nclass WebsocketsServer implements IWebsocketsServer {\n    private readonly httpServer: HttpServer;\n    private readonly ownsHttpServer: boolean;\n    private readonly adapter: WebsocketsServerAdapter.Interface<unknown>;\n    private readonly upgradeHandler: WebsocketsUpgradeHandler.Interface;\n    private readonly connectionManager: WebsocketsConnectionManager.Interface<unknown> | undefined;\n    private readonly authenticate: WebsocketsConnectionAuthenticator | undefined;\n    private readonly heartbeat: HeartbeatManager | undefined;\n    private readonly debug: boolean;\n    private readonly requestedPort: number;\n    private readonly requestedHost: string;\n\n    private shuttingDown: boolean = false;\n    private endpoint: string = \"\";\n    private resolvedPort: number = 0;\n\n    public constructor(params: WebsocketsServerParams) {\n        this.httpServer = params.httpServer;\n        this.ownsHttpServer = params.ownsHttpServer;\n        this.adapter = params.adapter;\n        this.upgradeHandler = params.upgradeHandler;\n        this.connectionManager = params.connectionManager;\n        this.authenticate = params.authenticate;\n        this.debug = params.debug;\n        this.requestedPort = params.port;\n        this.requestedHost = params.host;\n\n        if (params.connectionManager) {\n            this.heartbeat = new HeartbeatManager(\n                params.connectionManager,\n                params.adapter,\n                params.heartbeatInterval,\n                params.staleAfter\n            );\n        }\n    }\n\n    public async start(): Promise<void> {\n        this.adapter.start(this.httpServer);\n        this.wireUpgradeHandler();\n        this.wireConnectionHandler();\n\n        if (this.ownsHttpServer) {\n            await this.listen();\n        } else {\n            this.resolvePortFromExistingServer();\n        }\n\n        this.heartbeat?.start();\n\n        if (this.debug) {\n            console.log(`[WebsocketsServer] started on ${this.endpoint}`);\n        }\n    }\n\n    public async stop(): Promise<void> {\n        this.shuttingDown = true;\n        this.heartbeat?.stop();\n\n        await this.adapter.stop();\n\n        if (this.connectionManager) {\n            const activeIds = this.connectionManager.getActiveConnectionIds();\n            for (const connectionId of activeIds) {\n                await this.connectionManager.remove(connectionId);\n            }\n        }\n\n        if (this.ownsHttpServer) {\n            await this.closeHttpServer();\n        }\n\n        if (this.debug) {\n            console.log(\"[WebsocketsServer] stopped\");\n        }\n    }\n\n    public port(): number {\n        return this.resolvedPort;\n    }\n\n    private wireUpgradeHandler(): void {\n        this.httpServer.on(\n            \"upgrade\",\n            async (request: IncomingMessage, socket: Duplex, head: Buffer) => {\n                if (this.shuttingDown) {\n                    socket.destroy();\n                    return;\n                }\n\n                /*\n                 * Node drops its own error listener before emitting `upgrade`, so an error on the\n                 * raw socket before `ws` takes it over would be unhandled and end the process.\n                 */\n                socket.on(\"error\", error => {\n                    console.error(\"WebSocket upgrade socket error:\", error.message);\n                });\n\n                try {\n                    const decision = await this.upgradeHandler.shouldUpgrade(request);\n                    if (!decision.allowed) {\n                        socket.write(`HTTP/1.1 ${decision.statusCode} ${decision.reason}\\r\\n\\r\\n`);\n                        socket.destroy();\n                        return;\n                    }\n\n                    this.adapter.handleUpgrade(request, socket, head);\n                } catch (error) {\n                    console.error(\"WebSocket upgrade failed:\", error);\n                    socket.destroy();\n                }\n            }\n        );\n    }\n\n    private wireConnectionHandler(): void {\n        this.adapter.onConnection((socket, request) => {\n            if (this.shuttingDown) {\n                return;\n            }\n\n            const connectionId = mdbid();\n            const connectedAt = Date.now();\n            const host = request.headers.host || \"localhost\";\n            const headers = toHeaders(request.headers);\n\n            // Register the connection under its real identity, decoded from the `?token` JWT. Both\n            // the authentication and the registry write are async, so we kick this off without\n            // awaiting it and wire the socket handlers below synchronously — that way we don't miss a\n            // close or message that arrives during the gap. If a message does arrive before\n            // registration finishes, the `getMetadata` guard below just drops it, which is fine since\n            // the client doesn't send anything meaningful right at connect time.\n            const registration = this.registerConnection({\n                connectionId,\n                socket,\n                request,\n                connectedAt,\n                host,\n                headers\n            });\n\n            this.adapter.onMessage(socket, data => {\n                if (this.shuttingDown) {\n                    return;\n                }\n\n                const metadata = this.connectionManager?.getMetadata(connectionId);\n                if (!metadata) {\n                    return;\n                }\n\n                try {\n                    JSON.parse(data.toString());\n                } catch {\n                    return;\n                }\n\n                this.connectionManager?.updateLastSeen(connectionId).catch(error => {\n                    console.error(\n                        `Failed to update last seen for WebSocket connection \"${connectionId}\":`,\n                        error\n                    );\n                });\n            });\n\n            this.adapter.onClose(socket, () => {\n                if (this.shuttingDown) {\n                    return;\n                }\n                /*\n                 * Wait for registration: a socket that closes while it is still authenticating\n                 * would otherwise be removed first and then registered, leaving a dead entry.\n                 */\n                void registration.then(() => this.removeConnection(connectionId));\n            });\n\n            this.adapter.onError(socket, error => {\n                console.error(`WebSocket error for connection \"${connectionId}\":`, error.message);\n            });\n        });\n    }\n\n    /**\n     * Authenticates the upgrade using the `?token`/`?tenant` query the app-websockets client sends,\n     * then registers the live socket in the shared connection manager under the resolved identity and\n     * tenant. Getting both right matters for targeted server→client sends: `SendToIdentity` finds\n     * connections by identity id in the registry, so a connection that's anonymous (or never made it\n     * into the registry) can't be reached. Everything here is guarded — if something goes wrong we\n     * don't crash the server or the upgrade, the connection just ends up unaddressable.\n     */\n    private async registerConnection(params: {\n        connectionId: string;\n        socket: unknown;\n        request: IncomingMessage;\n        connectedAt: number;\n        host: string;\n        headers: Record<string, string>;\n    }): Promise<void> {\n        const { connectionId, socket, request, connectedAt, host, headers } = params;\n\n        let identity: ConnectionRegistry.Identity = ANONYMOUS_IDENTITY;\n        let tenant = \"\";\n\n        try {\n            const url = new URL(request.url ?? \"\", `http://${host}`);\n            tenant = url.searchParams.get(\"tenant\") ?? \"\";\n            const token = url.searchParams.get(\"token\") ?? \"\";\n\n            if (this.authenticate && token) {\n                const authenticated = await this.authenticate(token);\n                if (authenticated?.id) {\n                    identity = authenticated;\n                }\n            }\n\n            await this.connectionManager?.add({\n                connectionId,\n                socket,\n                endpoint: this.endpoint,\n                identity,\n                tenant,\n                connectedAt,\n                host,\n                headers\n            });\n        } catch (error) {\n            console.error(\n                `Failed to register WebSocket connection \"${connectionId}\":`,\n                error instanceof Error ? error.message : String(error)\n            );\n        }\n    }\n\n    private async removeConnection(connectionId: string): Promise<void> {\n        try {\n            await this.connectionManager?.remove(connectionId);\n        } catch (error) {\n            console.error(`Failed to remove WebSocket connection \"${connectionId}\":`, error);\n        }\n    }\n\n    private resolvePortFromExistingServer(): void {\n        const address = this.httpServer.address();\n        if (address && typeof address !== \"string\") {\n            this.resolvedPort = address.port;\n            this.endpoint = `ws://127.0.0.1:${this.resolvedPort}`;\n        }\n    }\n\n    private listen(): Promise<void> {\n        return new Promise((resolve, reject) => {\n            this.httpServer.listen(this.requestedPort, this.requestedHost, () => {\n                const address = this.httpServer.address();\n                if (!address || typeof address === \"string\") {\n                    reject(new Error(\"Unexpected server address format.\"));\n                    return;\n                }\n                this.resolvedPort = address.port;\n                this.endpoint = `ws://${this.requestedHost}:${this.resolvedPort}`;\n                resolve();\n            });\n        });\n    }\n\n    private closeHttpServer(): Promise<void> {\n        return new Promise((resolve, reject) => {\n            this.httpServer.close(err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n}\n\nexport { WebsocketsServer };\n\nexport const createWebsocketsServer = (params: CreateWebsocketsServerParams): IWebsocketsServer => {\n    const httpServer = createServer();\n    const server = new WebsocketsServer({\n        httpServer,\n        ownsHttpServer: true,\n        adapter: new NodeWsAdapterImpl(),\n        upgradeHandler: new DefaultUpgradeHandlerImpl(),\n        connectionManager: params.connectionManager,\n        authenticate: params.authenticate,\n        heartbeatInterval: params.heartbeatInterval ?? DEFAULT_HEARTBEAT_INTERVAL,\n        staleAfter: params.staleAfter ?? DEFAULT_STALE_AFTER,\n        debug: params.debug ?? false,\n        port: params.port ?? 0,\n        host: params.host ?? \"127.0.0.1\"\n    });\n    return server;\n};\n\nexport const attachWebsocketsServer = (params: AttachWebsocketsServerParams): IWebsocketsServer => {\n    const server = new WebsocketsServer({\n        httpServer: params.server,\n        ownsHttpServer: false,\n        adapter: new NodeWsAdapterImpl(),\n        upgradeHandler: new DefaultUpgradeHandlerImpl(),\n        connectionManager: params.connectionManager,\n        authenticate: params.authenticate,\n        heartbeatInterval: params.heartbeatInterval ?? DEFAULT_HEARTBEAT_INTERVAL,\n        staleAfter: params.staleAfter ?? DEFAULT_STALE_AFTER,\n        debug: params.debug ?? false,\n        port: 0,\n        host: \"127.0.0.1\"\n    });\n    return server;\n};\n"],"names":["ANONYMOUS_IDENTITY","DEFAULT_HEARTBEAT_INTERVAL","DEFAULT_STALE_AFTER","toHeaders","raw","headers","key","value","Object","WebsocketsServer","params","HeartbeatManager","console","activeIds","connectionId","request","socket","head","error","decision","mdbid","connectedAt","Date","host","registration","data","metadata","JSON","identity","tenant","url","URL","token","authenticated","Error","String","address","Promise","resolve","reject","err","createWebsocketsServer","httpServer","createServer","server","NodeWsAdapterImpl","DefaultUpgradeHandlerImpl","attachWebsocketsServer"],"mappings":";;;;;AAkBA,MAAMA,qBAAkD;IAAE,IAAI;IAAI,aAAa;IAAI,MAAM;AAAG;AAE5F,MAAMC,6BAA6B;AAEnC,MAAMC,sBAAsB;AAE5B,MAAMC,YAAY,CAACC;IACf,MAAMC,UAAkC,CAAC;IACzC,KAAK,MAAM,CAACC,KAAKC,MAAM,IAAIC,OAAO,OAAO,CAACJ,KACtC,IAAI,AAAiB,YAAjB,OAAOG,OACPF,OAAO,CAACC,IAAI,GAAGC;IAGvB,OAAOF;AACX;AAgBA,MAAMI;IAgBF,YAAmBC,MAA8B,CAAE;aAJ3C,YAAY,GAAY;aACxB,QAAQ,GAAW;aACnB,YAAY,GAAW;QAG3B,IAAI,CAAC,UAAU,GAAGA,OAAO,UAAU;QACnC,IAAI,CAAC,cAAc,GAAGA,OAAO,cAAc;QAC3C,IAAI,CAAC,OAAO,GAAGA,OAAO,OAAO;QAC7B,IAAI,CAAC,cAAc,GAAGA,OAAO,cAAc;QAC3C,IAAI,CAAC,iBAAiB,GAAGA,OAAO,iBAAiB;QACjD,IAAI,CAAC,YAAY,GAAGA,OAAO,YAAY;QACvC,IAAI,CAAC,KAAK,GAAGA,OAAO,KAAK;QACzB,IAAI,CAAC,aAAa,GAAGA,OAAO,IAAI;QAChC,IAAI,CAAC,aAAa,GAAGA,OAAO,IAAI;QAEhC,IAAIA,OAAO,iBAAiB,EACxB,IAAI,CAAC,SAAS,GAAG,IAAIC,iBACjBD,OAAO,iBAAiB,EACxBA,OAAO,OAAO,EACdA,OAAO,iBAAiB,EACxBA,OAAO,UAAU;IAG7B;IAEA,MAAa,QAAuB;QAChC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU;QAClC,IAAI,CAAC,kBAAkB;QACvB,IAAI,CAAC,qBAAqB;QAE1B,IAAI,IAAI,CAAC,cAAc,EACnB,MAAM,IAAI,CAAC,MAAM;aAEjB,IAAI,CAAC,6BAA6B;QAGtC,IAAI,CAAC,SAAS,EAAE;QAEhB,IAAI,IAAI,CAAC,KAAK,EACVE,QAAQ,GAAG,CAAC,CAAC,8BAA8B,EAAE,IAAI,CAAC,QAAQ,EAAE;IAEpE;IAEA,MAAa,OAAsB;QAC/B,IAAI,CAAC,YAAY,GAAG;QACpB,IAAI,CAAC,SAAS,EAAE;QAEhB,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI;QAEvB,IAAI,IAAI,CAAC,iBAAiB,EAAE;YACxB,MAAMC,YAAY,IAAI,CAAC,iBAAiB,CAAC,sBAAsB;YAC/D,KAAK,MAAMC,gBAAgBD,UACvB,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAACC;QAE5C;QAEA,IAAI,IAAI,CAAC,cAAc,EACnB,MAAM,IAAI,CAAC,eAAe;QAG9B,IAAI,IAAI,CAAC,KAAK,EACVF,QAAQ,GAAG,CAAC;IAEpB;IAEO,OAAe;QAClB,OAAO,IAAI,CAAC,YAAY;IAC5B;IAEQ,qBAA2B;QAC/B,IAAI,CAAC,UAAU,CAAC,EAAE,CACd,WACA,OAAOG,SAA0BC,QAAgBC;YAC7C,IAAI,IAAI,CAAC,YAAY,EAAE,YACnBD,OAAO,OAAO;YAQlBA,OAAO,EAAE,CAAC,SAASE,CAAAA;gBACfN,QAAQ,KAAK,CAAC,mCAAmCM,MAAM,OAAO;YAClE;YAEA,IAAI;gBACA,MAAMC,WAAW,MAAM,IAAI,CAAC,cAAc,CAAC,aAAa,CAACJ;gBACzD,IAAI,CAACI,SAAS,OAAO,EAAE;oBACnBH,OAAO,KAAK,CAAC,CAAC,SAAS,EAAEG,SAAS,UAAU,CAAC,CAAC,EAAEA,SAAS,MAAM,CAAC,QAAQ,CAAC;oBACzEH,OAAO,OAAO;oBACd;gBACJ;gBAEA,IAAI,CAAC,OAAO,CAAC,aAAa,CAACD,SAASC,QAAQC;YAChD,EAAE,OAAOC,OAAO;gBACZN,QAAQ,KAAK,CAAC,6BAA6BM;gBAC3CF,OAAO,OAAO;YAClB;QACJ;IAER;IAEQ,wBAA8B;QAClC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAACA,QAAQD;YAC/B,IAAI,IAAI,CAAC,YAAY,EACjB;YAGJ,MAAMD,eAAeM;YACrB,MAAMC,cAAcC,KAAK,GAAG;YAC5B,MAAMC,OAAOR,QAAQ,OAAO,CAAC,IAAI,IAAI;YACrC,MAAMV,UAAUF,UAAUY,QAAQ,OAAO;YAQzC,MAAMS,eAAe,IAAI,CAAC,kBAAkB,CAAC;gBACzCV;gBACAE;gBACAD;gBACAM;gBACAE;gBACAlB;YACJ;YAEA,IAAI,CAAC,OAAO,CAAC,SAAS,CAACW,QAAQS,CAAAA;gBAC3B,IAAI,IAAI,CAAC,YAAY,EACjB;gBAGJ,MAAMC,WAAW,IAAI,CAAC,iBAAiB,EAAE,YAAYZ;gBACrD,IAAI,CAACY,UACD;gBAGJ,IAAI;oBACAC,KAAK,KAAK,CAACF,KAAK,QAAQ;gBAC5B,EAAE,OAAM;oBACJ;gBACJ;gBAEA,IAAI,CAAC,iBAAiB,EAAE,eAAeX,cAAc,MAAMI,CAAAA;oBACvDN,QAAQ,KAAK,CACT,CAAC,qDAAqD,EAAEE,aAAa,EAAE,CAAC,EACxEI;gBAER;YACJ;YAEA,IAAI,CAAC,OAAO,CAAC,OAAO,CAACF,QAAQ;gBACzB,IAAI,IAAI,CAAC,YAAY,EACjB;gBAMCQ,aAAa,IAAI,CAAC,IAAM,IAAI,CAAC,gBAAgB,CAACV;YACvD;YAEA,IAAI,CAAC,OAAO,CAAC,OAAO,CAACE,QAAQE,CAAAA;gBACzBN,QAAQ,KAAK,CAAC,CAAC,gCAAgC,EAAEE,aAAa,EAAE,CAAC,EAAEI,MAAM,OAAO;YACpF;QACJ;IACJ;IAUA,MAAc,mBAAmBR,MAOhC,EAAiB;QACd,MAAM,EAAEI,YAAY,EAAEE,MAAM,EAAED,OAAO,EAAEM,WAAW,EAAEE,IAAI,EAAElB,OAAO,EAAE,GAAGK;QAEtE,IAAIkB,WAAwC5B;QAC5C,IAAI6B,SAAS;QAEb,IAAI;YACA,MAAMC,MAAM,IAAIC,IAAIhB,QAAQ,GAAG,IAAI,IAAI,CAAC,OAAO,EAAEQ,MAAM;YACvDM,SAASC,IAAI,YAAY,CAAC,GAAG,CAAC,aAAa;YAC3C,MAAME,QAAQF,IAAI,YAAY,CAAC,GAAG,CAAC,YAAY;YAE/C,IAAI,IAAI,CAAC,YAAY,IAAIE,OAAO;gBAC5B,MAAMC,gBAAgB,MAAM,IAAI,CAAC,YAAY,CAACD;gBAC9C,IAAIC,eAAe,IACfL,WAAWK;YAEnB;YAEA,MAAM,IAAI,CAAC,iBAAiB,EAAE,IAAI;gBAC9BnB;gBACAE;gBACA,UAAU,IAAI,CAAC,QAAQ;gBACvBY;gBACAC;gBACAR;gBACAE;gBACAlB;YACJ;QACJ,EAAE,OAAOa,OAAO;YACZN,QAAQ,KAAK,CACT,CAAC,yCAAyC,EAAEE,aAAa,EAAE,CAAC,EAC5DI,iBAAiBgB,QAAQhB,MAAM,OAAO,GAAGiB,OAAOjB;QAExD;IACJ;IAEA,MAAc,iBAAiBJ,YAAoB,EAAiB;QAChE,IAAI;YACA,MAAM,IAAI,CAAC,iBAAiB,EAAE,OAAOA;QACzC,EAAE,OAAOI,OAAO;YACZN,QAAQ,KAAK,CAAC,CAAC,uCAAuC,EAAEE,aAAa,EAAE,CAAC,EAAEI;QAC9E;IACJ;IAEQ,gCAAsC;QAC1C,MAAMkB,UAAU,IAAI,CAAC,UAAU,CAAC,OAAO;QACvC,IAAIA,WAAW,AAAmB,YAAnB,OAAOA,SAAsB;YACxC,IAAI,CAAC,YAAY,GAAGA,QAAQ,IAAI;YAChC,IAAI,CAAC,QAAQ,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,YAAY,EAAE;QACzD;IACJ;IAEQ,SAAwB;QAC5B,OAAO,IAAIC,QAAQ,CAACC,SAASC;YACzB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE;gBAC3D,MAAMH,UAAU,IAAI,CAAC,UAAU,CAAC,OAAO;gBACvC,IAAI,CAACA,WAAW,AAAmB,YAAnB,OAAOA,SAAsB,YACzCG,OAAO,IAAIL,MAAM;gBAGrB,IAAI,CAAC,YAAY,GAAGE,QAAQ,IAAI;gBAChC,IAAI,CAAC,QAAQ,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,EAAE;gBACjEE;YACJ;QACJ;IACJ;IAEQ,kBAAiC;QACrC,OAAO,IAAID,QAAQ,CAACC,SAASC;YACzB,IAAI,CAAC,UAAU,CAAC,KAAK,CAACC,CAAAA;gBAClB,IAAIA,KAAK,YACLD,OAAOC;gBAGXF;YACJ;QACJ;IACJ;AACJ;AAIO,MAAMG,yBAAyB,CAAC/B;IACnC,MAAMgC,aAAaC;IACnB,MAAMC,SAAS,IAAInC,iBAAiB;QAChCiC;QACA,gBAAgB;QAChB,SAAS,IAAIG;QACb,gBAAgB,IAAIC;QACpB,mBAAmBpC,OAAO,iBAAiB;QAC3C,cAAcA,OAAO,YAAY;QACjC,mBAAmBA,OAAO,iBAAiB,IAAIT;QAC/C,YAAYS,OAAO,UAAU,IAAIR;QACjC,OAAOQ,OAAO,KAAK,IAAI;QACvB,MAAMA,OAAO,IAAI,IAAI;QACrB,MAAMA,OAAO,IAAI,IAAI;IACzB;IACA,OAAOkC;AACX;AAEO,MAAMG,yBAAyB,CAACrC;IACnC,MAAMkC,SAAS,IAAInC,iBAAiB;QAChC,YAAYC,OAAO,MAAM;QACzB,gBAAgB;QAChB,SAAS,IAAImC;QACb,gBAAgB,IAAIC;QACpB,mBAAmBpC,OAAO,iBAAiB;QAC3C,cAAcA,OAAO,YAAY;QACjC,mBAAmBA,OAAO,iBAAiB,IAAIT;QAC/C,YAAYS,OAAO,UAAU,IAAIR;QACjC,OAAOQ,OAAO,KAAK,IAAI;QACvB,MAAM;QACN,MAAM;IACV;IACA,OAAOkC;AACX"}
+{"version":3,"file":"server/WebsocketsServer.js","sources":["../../src/server/WebsocketsServer.ts"],"sourcesContent":["import { createServer } from \"node:http\";\nimport type { Server as HttpServer, IncomingMessage } from \"node:http\";\nimport type { Duplex } from \"node:stream\";\nimport { mdbid } from \"@webiny/utils\";\nimport type { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\nimport type { WebsocketsUpgradeHandler } from \"~/upgradeHandler/abstractions.js\";\nimport type { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport type { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { NodeWsAdapterImpl } from \"~/adapter/NodeWsAdapter.js\";\nimport { DefaultUpgradeHandlerImpl } from \"~/upgradeHandler/DefaultUpgradeHandler.js\";\nimport { HeartbeatManager } from \"~/heartbeat/HeartbeatManager.js\";\nimport type {\n    CreateWebsocketsServerParams,\n    AttachWebsocketsServerParams,\n    IWebsocketsServer,\n    WebsocketsConnectionAuthenticator\n} from \"./types.js\";\n\nconst ANONYMOUS_IDENTITY: ConnectionRegistry.Identity = { id: \"\", displayName: \"\", type: \"\" };\n\nconst DEFAULT_HEARTBEAT_INTERVAL = 60_000;\n// Three admin ping intervals (5 minutes each): one late ping is not enough to evict a live tab.\nconst DEFAULT_STALE_AFTER = 15 * 60_000;\n\nconst toHeaders = (raw: IncomingMessage[\"headers\"]): Record<string, string> => {\n    const headers: Record<string, string> = {};\n    for (const [key, value] of Object.entries(raw)) {\n        if (typeof value === \"string\") {\n            headers[key] = value;\n        }\n    }\n    return headers;\n};\n\ninterface WebsocketsServerParams {\n    httpServer: HttpServer;\n    ownsHttpServer: boolean;\n    adapter: WebsocketsServerAdapter.Interface<unknown>;\n    upgradeHandler: WebsocketsUpgradeHandler.Interface;\n    connectionManager?: WebsocketsConnectionManager.Interface<unknown>;\n    authenticate?: WebsocketsConnectionAuthenticator;\n    heartbeatInterval: number;\n    staleAfter: number;\n    debug: boolean;\n    port: number;\n    host: string;\n}\n\nclass WebsocketsServer implements IWebsocketsServer {\n    private readonly httpServer: HttpServer;\n    private readonly ownsHttpServer: boolean;\n    private readonly adapter: WebsocketsServerAdapter.Interface<unknown>;\n    private readonly upgradeHandler: WebsocketsUpgradeHandler.Interface;\n    private readonly connectionManager: WebsocketsConnectionManager.Interface<unknown> | undefined;\n    private readonly authenticate: WebsocketsConnectionAuthenticator | undefined;\n    private readonly heartbeat: HeartbeatManager | undefined;\n    private readonly debug: boolean;\n    private readonly requestedPort: number;\n    private readonly requestedHost: string;\n\n    private shuttingDown: boolean = false;\n    private endpoint: string = \"\";\n    private resolvedPort: number = 0;\n\n    public constructor(params: WebsocketsServerParams) {\n        this.httpServer = params.httpServer;\n        this.ownsHttpServer = params.ownsHttpServer;\n        this.adapter = params.adapter;\n        this.upgradeHandler = params.upgradeHandler;\n        this.connectionManager = params.connectionManager;\n        this.authenticate = params.authenticate;\n        this.debug = params.debug;\n        this.requestedPort = params.port;\n        this.requestedHost = params.host;\n\n        if (params.connectionManager) {\n            this.heartbeat = new HeartbeatManager(\n                params.connectionManager,\n                params.adapter,\n                params.heartbeatInterval,\n                params.staleAfter\n            );\n        }\n    }\n\n    public async start(): Promise<void> {\n        this.adapter.start(this.httpServer);\n        this.wireUpgradeHandler();\n        this.wireConnectionHandler();\n\n        if (this.ownsHttpServer) {\n            await this.listen();\n        } else {\n            this.resolvePortFromExistingServer();\n        }\n\n        this.heartbeat?.start();\n\n        if (this.debug) {\n            console.log(`[WebsocketsServer] started on ${this.endpoint}`);\n        }\n    }\n\n    public async stop(): Promise<void> {\n        this.shuttingDown = true;\n        this.heartbeat?.stop();\n\n        /*\n         * Close the sockets and take this process's connections out of the registry at the same\n         * time, so both fit in the shutdown. Without the registry step, the rows stay until another\n         * process's heartbeat finds them stale, and sends to those users go nowhere until then.\n         */\n        const activeIds = this.connectionManager?.getActiveConnectionIds() ?? [];\n        const removals = activeIds.map(connectionId => this.removeConnection(connectionId));\n        const closing = this.adapter.stop();\n        await Promise.all([closing, ...removals]);\n\n        if (this.ownsHttpServer) {\n            await this.closeHttpServer();\n        }\n\n        if (this.debug) {\n            console.log(\"[WebsocketsServer] stopped\");\n        }\n    }\n\n    public port(): number {\n        return this.resolvedPort;\n    }\n\n    private wireUpgradeHandler(): void {\n        this.httpServer.on(\n            \"upgrade\",\n            async (request: IncomingMessage, socket: Duplex, head: Buffer) => {\n                if (this.shuttingDown) {\n                    socket.destroy();\n                    return;\n                }\n\n                /*\n                 * Node drops its own error listener before emitting `upgrade`, so an error on the\n                 * raw socket before `ws` takes it over would be unhandled and end the process.\n                 */\n                socket.on(\"error\", error => {\n                    console.error(\"WebSocket upgrade socket error:\", error.message);\n                });\n\n                try {\n                    const decision = await this.upgradeHandler.shouldUpgrade(request);\n                    if (!decision.allowed) {\n                        socket.write(`HTTP/1.1 ${decision.statusCode} ${decision.reason}\\r\\n\\r\\n`);\n                        socket.destroy();\n                        return;\n                    }\n\n                    this.adapter.handleUpgrade(request, socket, head);\n                } catch (error) {\n                    console.error(\"WebSocket upgrade failed:\", error);\n                    socket.destroy();\n                }\n            }\n        );\n    }\n\n    private wireConnectionHandler(): void {\n        this.adapter.onConnection((socket, request) => {\n            if (this.shuttingDown) {\n                return;\n            }\n\n            const connectionId = mdbid();\n            const connectedAt = Date.now();\n            const host = request.headers.host || \"localhost\";\n            const headers = toHeaders(request.headers);\n\n            // Register the connection under its real identity, decoded from the `?token` JWT. Both\n            // the authentication and the registry write are async, so we kick this off without\n            // awaiting it and wire the socket handlers below synchronously — that way we don't miss a\n            // close or message that arrives during the gap. If a message does arrive before\n            // registration finishes, the `getMetadata` guard below just drops it, which is fine since\n            // the client doesn't send anything meaningful right at connect time.\n            const registration = this.registerConnection({\n                connectionId,\n                socket,\n                request,\n                connectedAt,\n                host,\n                headers\n            });\n\n            this.adapter.onMessage(socket, data => {\n                if (this.shuttingDown) {\n                    return;\n                }\n\n                const metadata = this.connectionManager?.getMetadata(connectionId);\n                if (!metadata) {\n                    return;\n                }\n\n                try {\n                    JSON.parse(data.toString());\n                } catch {\n                    return;\n                }\n\n                this.connectionManager?.updateLastSeen(connectionId).catch(error => {\n                    console.error(\n                        `Failed to update last seen for WebSocket connection \"${connectionId}\":`,\n                        error\n                    );\n                });\n            });\n\n            this.adapter.onClose(socket, () => {\n                if (this.shuttingDown) {\n                    return;\n                }\n                /*\n                 * Wait for registration: a socket that closes while it is still authenticating\n                 * would otherwise be removed first and then registered, leaving a dead entry.\n                 */\n                void registration.then(() => this.removeConnection(connectionId));\n            });\n\n            this.adapter.onError(socket, error => {\n                console.error(`WebSocket error for connection \"${connectionId}\":`, error.message);\n            });\n        });\n    }\n\n    /**\n     * Authenticates the upgrade using the `?token`/`?tenant` query the app-websockets client sends,\n     * then registers the live socket in the shared connection manager under the resolved identity and\n     * tenant. Getting both right matters for targeted server→client sends: `SendToIdentity` finds\n     * connections by identity id in the registry, so a connection that's anonymous (or never made it\n     * into the registry) can't be reached. Everything here is guarded — if something goes wrong we\n     * don't crash the server or the upgrade, the connection just ends up unaddressable.\n     */\n    private async registerConnection(params: {\n        connectionId: string;\n        socket: unknown;\n        request: IncomingMessage;\n        connectedAt: number;\n        host: string;\n        headers: Record<string, string>;\n    }): Promise<void> {\n        const { connectionId, socket, request, connectedAt, host, headers } = params;\n\n        let identity: ConnectionRegistry.Identity = ANONYMOUS_IDENTITY;\n        let tenant = \"\";\n\n        try {\n            const url = new URL(request.url ?? \"\", `http://${host}`);\n            tenant = url.searchParams.get(\"tenant\") ?? \"\";\n            const token = url.searchParams.get(\"token\") ?? \"\";\n\n            if (this.authenticate && token) {\n                const authenticated = await this.authenticate(token);\n                if (authenticated?.id) {\n                    identity = authenticated;\n                }\n            }\n\n            await this.connectionManager?.add({\n                connectionId,\n                socket,\n                endpoint: this.endpoint,\n                identity,\n                tenant,\n                connectedAt,\n                host,\n                headers\n            });\n        } catch (error) {\n            console.error(\n                `Failed to register WebSocket connection \"${connectionId}\":`,\n                error instanceof Error ? error.message : String(error)\n            );\n        }\n    }\n\n    private async removeConnection(connectionId: string): Promise<void> {\n        try {\n            await this.connectionManager?.remove(connectionId);\n        } catch (error) {\n            console.error(`Failed to remove WebSocket connection \"${connectionId}\":`, error);\n        }\n    }\n\n    private resolvePortFromExistingServer(): void {\n        const address = this.httpServer.address();\n        if (address && typeof address !== \"string\") {\n            this.resolvedPort = address.port;\n            this.endpoint = `ws://127.0.0.1:${this.resolvedPort}`;\n        }\n    }\n\n    private listen(): Promise<void> {\n        return new Promise((resolve, reject) => {\n            this.httpServer.listen(this.requestedPort, this.requestedHost, () => {\n                const address = this.httpServer.address();\n                if (!address || typeof address === \"string\") {\n                    reject(new Error(\"Unexpected server address format.\"));\n                    return;\n                }\n                this.resolvedPort = address.port;\n                this.endpoint = `ws://${this.requestedHost}:${this.resolvedPort}`;\n                resolve();\n            });\n        });\n    }\n\n    private closeHttpServer(): Promise<void> {\n        return new Promise((resolve, reject) => {\n            this.httpServer.close(err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n}\n\nexport { WebsocketsServer };\n\nexport const createWebsocketsServer = (params: CreateWebsocketsServerParams): IWebsocketsServer => {\n    const httpServer = createServer();\n    const server = new WebsocketsServer({\n        httpServer,\n        ownsHttpServer: true,\n        adapter: new NodeWsAdapterImpl(),\n        upgradeHandler: new DefaultUpgradeHandlerImpl(),\n        connectionManager: params.connectionManager,\n        authenticate: params.authenticate,\n        heartbeatInterval: params.heartbeatInterval ?? DEFAULT_HEARTBEAT_INTERVAL,\n        staleAfter: params.staleAfter ?? DEFAULT_STALE_AFTER,\n        debug: params.debug ?? false,\n        port: params.port ?? 0,\n        host: params.host ?? \"127.0.0.1\"\n    });\n    return server;\n};\n\nexport const attachWebsocketsServer = (params: AttachWebsocketsServerParams): IWebsocketsServer => {\n    const server = new WebsocketsServer({\n        httpServer: params.server,\n        ownsHttpServer: false,\n        adapter: new NodeWsAdapterImpl(),\n        upgradeHandler: new DefaultUpgradeHandlerImpl(),\n        connectionManager: params.connectionManager,\n        authenticate: params.authenticate,\n        heartbeatInterval: params.heartbeatInterval ?? DEFAULT_HEARTBEAT_INTERVAL,\n        staleAfter: params.staleAfter ?? DEFAULT_STALE_AFTER,\n        debug: params.debug ?? false,\n        port: 0,\n        host: \"127.0.0.1\"\n    });\n    return server;\n};\n"],"names":["ANONYMOUS_IDENTITY","DEFAULT_HEARTBEAT_INTERVAL","DEFAULT_STALE_AFTER","toHeaders","raw","headers","key","value","Object","WebsocketsServer","params","HeartbeatManager","console","activeIds","removals","connectionId","closing","Promise","request","socket","head","error","decision","mdbid","connectedAt","Date","host","registration","data","metadata","JSON","identity","tenant","url","URL","token","authenticated","Error","String","address","resolve","reject","err","createWebsocketsServer","httpServer","createServer","server","NodeWsAdapterImpl","DefaultUpgradeHandlerImpl","attachWebsocketsServer"],"mappings":";;;;;AAkBA,MAAMA,qBAAkD;IAAE,IAAI;IAAI,aAAa;IAAI,MAAM;AAAG;AAE5F,MAAMC,6BAA6B;AAEnC,MAAMC,sBAAsB;AAE5B,MAAMC,YAAY,CAACC;IACf,MAAMC,UAAkC,CAAC;IACzC,KAAK,MAAM,CAACC,KAAKC,MAAM,IAAIC,OAAO,OAAO,CAACJ,KACtC,IAAI,AAAiB,YAAjB,OAAOG,OACPF,OAAO,CAACC,IAAI,GAAGC;IAGvB,OAAOF;AACX;AAgBA,MAAMI;IAgBF,YAAmBC,MAA8B,CAAE;aAJ3C,YAAY,GAAY;aACxB,QAAQ,GAAW;aACnB,YAAY,GAAW;QAG3B,IAAI,CAAC,UAAU,GAAGA,OAAO,UAAU;QACnC,IAAI,CAAC,cAAc,GAAGA,OAAO,cAAc;QAC3C,IAAI,CAAC,OAAO,GAAGA,OAAO,OAAO;QAC7B,IAAI,CAAC,cAAc,GAAGA,OAAO,cAAc;QAC3C,IAAI,CAAC,iBAAiB,GAAGA,OAAO,iBAAiB;QACjD,IAAI,CAAC,YAAY,GAAGA,OAAO,YAAY;QACvC,IAAI,CAAC,KAAK,GAAGA,OAAO,KAAK;QACzB,IAAI,CAAC,aAAa,GAAGA,OAAO,IAAI;QAChC,IAAI,CAAC,aAAa,GAAGA,OAAO,IAAI;QAEhC,IAAIA,OAAO,iBAAiB,EACxB,IAAI,CAAC,SAAS,GAAG,IAAIC,iBACjBD,OAAO,iBAAiB,EACxBA,OAAO,OAAO,EACdA,OAAO,iBAAiB,EACxBA,OAAO,UAAU;IAG7B;IAEA,MAAa,QAAuB;QAChC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU;QAClC,IAAI,CAAC,kBAAkB;QACvB,IAAI,CAAC,qBAAqB;QAE1B,IAAI,IAAI,CAAC,cAAc,EACnB,MAAM,IAAI,CAAC,MAAM;aAEjB,IAAI,CAAC,6BAA6B;QAGtC,IAAI,CAAC,SAAS,EAAE;QAEhB,IAAI,IAAI,CAAC,KAAK,EACVE,QAAQ,GAAG,CAAC,CAAC,8BAA8B,EAAE,IAAI,CAAC,QAAQ,EAAE;IAEpE;IAEA,MAAa,OAAsB;QAC/B,IAAI,CAAC,YAAY,GAAG;QACpB,IAAI,CAAC,SAAS,EAAE;QAOhB,MAAMC,YAAY,IAAI,CAAC,iBAAiB,EAAE,4BAA4B,EAAE;QACxE,MAAMC,WAAWD,UAAU,GAAG,CAACE,CAAAA,eAAgB,IAAI,CAAC,gBAAgB,CAACA;QACrE,MAAMC,UAAU,IAAI,CAAC,OAAO,CAAC,IAAI;QACjC,MAAMC,QAAQ,GAAG,CAAC;YAACD;eAAYF;SAAS;QAExC,IAAI,IAAI,CAAC,cAAc,EACnB,MAAM,IAAI,CAAC,eAAe;QAG9B,IAAI,IAAI,CAAC,KAAK,EACVF,QAAQ,GAAG,CAAC;IAEpB;IAEO,OAAe;QAClB,OAAO,IAAI,CAAC,YAAY;IAC5B;IAEQ,qBAA2B;QAC/B,IAAI,CAAC,UAAU,CAAC,EAAE,CACd,WACA,OAAOM,SAA0BC,QAAgBC;YAC7C,IAAI,IAAI,CAAC,YAAY,EAAE,YACnBD,OAAO,OAAO;YAQlBA,OAAO,EAAE,CAAC,SAASE,CAAAA;gBACfT,QAAQ,KAAK,CAAC,mCAAmCS,MAAM,OAAO;YAClE;YAEA,IAAI;gBACA,MAAMC,WAAW,MAAM,IAAI,CAAC,cAAc,CAAC,aAAa,CAACJ;gBACzD,IAAI,CAACI,SAAS,OAAO,EAAE;oBACnBH,OAAO,KAAK,CAAC,CAAC,SAAS,EAAEG,SAAS,UAAU,CAAC,CAAC,EAAEA,SAAS,MAAM,CAAC,QAAQ,CAAC;oBACzEH,OAAO,OAAO;oBACd;gBACJ;gBAEA,IAAI,CAAC,OAAO,CAAC,aAAa,CAACD,SAASC,QAAQC;YAChD,EAAE,OAAOC,OAAO;gBACZT,QAAQ,KAAK,CAAC,6BAA6BS;gBAC3CF,OAAO,OAAO;YAClB;QACJ;IAER;IAEQ,wBAA8B;QAClC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAACA,QAAQD;YAC/B,IAAI,IAAI,CAAC,YAAY,EACjB;YAGJ,MAAMH,eAAeQ;YACrB,MAAMC,cAAcC,KAAK,GAAG;YAC5B,MAAMC,OAAOR,QAAQ,OAAO,CAAC,IAAI,IAAI;YACrC,MAAMb,UAAUF,UAAUe,QAAQ,OAAO;YAQzC,MAAMS,eAAe,IAAI,CAAC,kBAAkB,CAAC;gBACzCZ;gBACAI;gBACAD;gBACAM;gBACAE;gBACArB;YACJ;YAEA,IAAI,CAAC,OAAO,CAAC,SAAS,CAACc,QAAQS,CAAAA;gBAC3B,IAAI,IAAI,CAAC,YAAY,EACjB;gBAGJ,MAAMC,WAAW,IAAI,CAAC,iBAAiB,EAAE,YAAYd;gBACrD,IAAI,CAACc,UACD;gBAGJ,IAAI;oBACAC,KAAK,KAAK,CAACF,KAAK,QAAQ;gBAC5B,EAAE,OAAM;oBACJ;gBACJ;gBAEA,IAAI,CAAC,iBAAiB,EAAE,eAAeb,cAAc,MAAMM,CAAAA;oBACvDT,QAAQ,KAAK,CACT,CAAC,qDAAqD,EAAEG,aAAa,EAAE,CAAC,EACxEM;gBAER;YACJ;YAEA,IAAI,CAAC,OAAO,CAAC,OAAO,CAACF,QAAQ;gBACzB,IAAI,IAAI,CAAC,YAAY,EACjB;gBAMCQ,aAAa,IAAI,CAAC,IAAM,IAAI,CAAC,gBAAgB,CAACZ;YACvD;YAEA,IAAI,CAAC,OAAO,CAAC,OAAO,CAACI,QAAQE,CAAAA;gBACzBT,QAAQ,KAAK,CAAC,CAAC,gCAAgC,EAAEG,aAAa,EAAE,CAAC,EAAEM,MAAM,OAAO;YACpF;QACJ;IACJ;IAUA,MAAc,mBAAmBX,MAOhC,EAAiB;QACd,MAAM,EAAEK,YAAY,EAAEI,MAAM,EAAED,OAAO,EAAEM,WAAW,EAAEE,IAAI,EAAErB,OAAO,EAAE,GAAGK;QAEtE,IAAIqB,WAAwC/B;QAC5C,IAAIgC,SAAS;QAEb,IAAI;YACA,MAAMC,MAAM,IAAIC,IAAIhB,QAAQ,GAAG,IAAI,IAAI,CAAC,OAAO,EAAEQ,MAAM;YACvDM,SAASC,IAAI,YAAY,CAAC,GAAG,CAAC,aAAa;YAC3C,MAAME,QAAQF,IAAI,YAAY,CAAC,GAAG,CAAC,YAAY;YAE/C,IAAI,IAAI,CAAC,YAAY,IAAIE,OAAO;gBAC5B,MAAMC,gBAAgB,MAAM,IAAI,CAAC,YAAY,CAACD;gBAC9C,IAAIC,eAAe,IACfL,WAAWK;YAEnB;YAEA,MAAM,IAAI,CAAC,iBAAiB,EAAE,IAAI;gBAC9BrB;gBACAI;gBACA,UAAU,IAAI,CAAC,QAAQ;gBACvBY;gBACAC;gBACAR;gBACAE;gBACArB;YACJ;QACJ,EAAE,OAAOgB,OAAO;YACZT,QAAQ,KAAK,CACT,CAAC,yCAAyC,EAAEG,aAAa,EAAE,CAAC,EAC5DM,iBAAiBgB,QAAQhB,MAAM,OAAO,GAAGiB,OAAOjB;QAExD;IACJ;IAEA,MAAc,iBAAiBN,YAAoB,EAAiB;QAChE,IAAI;YACA,MAAM,IAAI,CAAC,iBAAiB,EAAE,OAAOA;QACzC,EAAE,OAAOM,OAAO;YACZT,QAAQ,KAAK,CAAC,CAAC,uCAAuC,EAAEG,aAAa,EAAE,CAAC,EAAEM;QAC9E;IACJ;IAEQ,gCAAsC;QAC1C,MAAMkB,UAAU,IAAI,CAAC,UAAU,CAAC,OAAO;QACvC,IAAIA,WAAW,AAAmB,YAAnB,OAAOA,SAAsB;YACxC,IAAI,CAAC,YAAY,GAAGA,QAAQ,IAAI;YAChC,IAAI,CAAC,QAAQ,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,YAAY,EAAE;QACzD;IACJ;IAEQ,SAAwB;QAC5B,OAAO,IAAItB,QAAQ,CAACuB,SAASC;YACzB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE;gBAC3D,MAAMF,UAAU,IAAI,CAAC,UAAU,CAAC,OAAO;gBACvC,IAAI,CAACA,WAAW,AAAmB,YAAnB,OAAOA,SAAsB,YACzCE,OAAO,IAAIJ,MAAM;gBAGrB,IAAI,CAAC,YAAY,GAAGE,QAAQ,IAAI;gBAChC,IAAI,CAAC,QAAQ,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,EAAE;gBACjEC;YACJ;QACJ;IACJ;IAEQ,kBAAiC;QACrC,OAAO,IAAIvB,QAAQ,CAACuB,SAASC;YACzB,IAAI,CAAC,UAAU,CAAC,KAAK,CAACC,CAAAA;gBAClB,IAAIA,KAAK,YACLD,OAAOC;gBAGXF;YACJ;QACJ;IACJ;AACJ;AAIO,MAAMG,yBAAyB,CAACjC;IACnC,MAAMkC,aAAaC;IACnB,MAAMC,SAAS,IAAIrC,iBAAiB;QAChCmC;QACA,gBAAgB;QAChB,SAAS,IAAIG;QACb,gBAAgB,IAAIC;QACpB,mBAAmBtC,OAAO,iBAAiB;QAC3C,cAAcA,OAAO,YAAY;QACjC,mBAAmBA,OAAO,iBAAiB,IAAIT;QAC/C,YAAYS,OAAO,UAAU,IAAIR;QACjC,OAAOQ,OAAO,KAAK,IAAI;QACvB,MAAMA,OAAO,IAAI,IAAI;QACrB,MAAMA,OAAO,IAAI,IAAI;IACzB;IACA,OAAOoC;AACX;AAEO,MAAMG,yBAAyB,CAACvC;IACnC,MAAMoC,SAAS,IAAIrC,iBAAiB;QAChC,YAAYC,OAAO,MAAM;QACzB,gBAAgB;QAChB,SAAS,IAAIqC;QACb,gBAAgB,IAAIC;QACpB,mBAAmBtC,OAAO,iBAAiB;QAC3C,cAAcA,OAAO,YAAY;QACjC,mBAAmBA,OAAO,iBAAiB,IAAIT;QAC/C,YAAYS,OAAO,UAAU,IAAIR;QACjC,OAAOQ,OAAO,KAAK,IAAI;QACvB,MAAM;QACN,MAAM;IACV;IACA,OAAOoC;AACX"}

package/transport/ServerWebsocketsTransport.js

@@ -7,11 +7,12 @@ class ServerWebsocketsTransportImpl {
         this.adapter = adapter;
     }
     async send(connections, data) {
+        const missing = [];
         for (const connection of connections){
             const { connectionId } = connection;
             const socket = this.connectionManager.getSocket(connectionId);
             if (void 0 === socket) {
-                await this.connectionManager.remove(connectionId);
+                missing.push(connectionId);
                 continue;
             }
             try {
@@ -21,13 +22,15 @@ class ServerWebsocketsTransportImpl {
                 console.log(ex);
             }
         }
+        await this.connectionManager.removeOwnedHere(missing);
     }
     async disconnect(connections) {
+        const missing = [];
         for (const connection of connections){
             const { connectionId } = connection;
             const socket = this.connectionManager.getSocket(connectionId);
             if (void 0 === socket) {
-                await this.connectionManager.remove(connectionId);
+                missing.push(connectionId);
                 continue;
             }
             try {
@@ -38,6 +41,7 @@ class ServerWebsocketsTransportImpl {
                 console.log(ex);
             }
         }
+        await this.connectionManager.removeOwnedHere(missing);
     }
 }
 const ServerWebsocketsTransport = WebsocketsTransport.createImplementation({

package/transport/ServerWebsocketsTransport.js.map

@@ -1 +1 @@
-{"version":3,"file":"transport/ServerWebsocketsTransport.js","sources":["../../src/transport/ServerWebsocketsTransport.ts"],"sourcesContent":["import { WebsocketsTransport } from \"@webiny/api-websockets/exports/api.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\nimport { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\n\nexport class ServerWebsocketsTransportImpl implements WebsocketsTransport.Interface {\n    private readonly connectionManager: WebsocketsConnectionManager.Interface<unknown>;\n    private readonly adapter: WebsocketsServerAdapter.Interface<unknown>;\n\n    public constructor(\n        connectionManager: WebsocketsConnectionManager.Interface<unknown>,\n        adapter: WebsocketsServerAdapter.Interface<unknown>\n    ) {\n        this.connectionManager = connectionManager;\n        this.adapter = adapter;\n    }\n\n    public async send<T extends GenericRecord = GenericRecord>(\n        connections: WebsocketsTransport.SendConnection[],\n        data: WebsocketsTransport.SendData<T>\n    ): Promise<void> {\n        for (const connection of connections) {\n            const { connectionId } = connection;\n            const socket = this.connectionManager.getSocket(connectionId);\n            if (socket === undefined) {\n                await this.connectionManager.remove(connectionId);\n                continue;\n            }\n            try {\n                await this.adapter.send(socket, JSON.stringify(data));\n            } catch (ex) {\n                console.error(\n                    `Failed to send message to connection \"${connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n    }\n\n    public async disconnect(\n        connections: WebsocketsTransport.DisconnectConnection[]\n    ): Promise<void> {\n        for (const connection of connections) {\n            const { connectionId } = connection;\n            const socket = this.connectionManager.getSocket(connectionId);\n            if (socket === undefined) {\n                await this.connectionManager.remove(connectionId);\n                continue;\n            }\n            try {\n                this.adapter.close(socket);\n                await this.connectionManager.remove(connectionId);\n            } catch (ex) {\n                console.error(\n                    `Failed to disconnect connection \"${connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n    }\n}\n\nexport const ServerWebsocketsTransport = WebsocketsTransport.createImplementation({\n    implementation: ServerWebsocketsTransportImpl,\n    dependencies: [WebsocketsConnectionManager, WebsocketsServerAdapter]\n});\n"],"names":["ServerWebsocketsTransportImpl","connectionManager","adapter","connections","data","connection","connectionId","socket","undefined","JSON","ex","console","ServerWebsocketsTransport","WebsocketsTransport","WebsocketsConnectionManager","WebsocketsServerAdapter"],"mappings":";;;AAKO,MAAMA;IAIT,YACIC,iBAAiE,EACjEC,OAAmD,CACrD;QACE,IAAI,CAAC,iBAAiB,GAAGD;QACzB,IAAI,CAAC,OAAO,GAAGC;IACnB;IAEA,MAAa,KACTC,WAAiD,EACjDC,IAAqC,EACxB;QACb,KAAK,MAAMC,cAAcF,YAAa;YAClC,MAAM,EAAEG,YAAY,EAAE,GAAGD;YACzB,MAAME,SAAS,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAACD;YAChD,IAAIC,AAAWC,WAAXD,QAAsB;gBACtB,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAACD;gBACpC;YACJ;YACA,IAAI;gBACA,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAACC,QAAQE,KAAK,SAAS,CAACL;YACnD,EAAE,OAAOM,IAAI;gBACTC,QAAQ,KAAK,CACT,CAAC,sCAAsC,EAAEL,aAAa,mCAAmC,CAAC;gBAE9FK,QAAQ,GAAG,CAACD;YAChB;QACJ;IACJ;IAEA,MAAa,WACTP,WAAuD,EAC1C;QACb,KAAK,MAAME,cAAcF,YAAa;YAClC,MAAM,EAAEG,YAAY,EAAE,GAAGD;YACzB,MAAME,SAAS,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAACD;YAChD,IAAIC,AAAWC,WAAXD,QAAsB;gBACtB,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAACD;gBACpC;YACJ;YACA,IAAI;gBACA,IAAI,CAAC,OAAO,CAAC,KAAK,CAACC;gBACnB,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAACD;YACxC,EAAE,OAAOI,IAAI;gBACTC,QAAQ,KAAK,CACT,CAAC,iCAAiC,EAAEL,aAAa,mCAAmC,CAAC;gBAEzFK,QAAQ,GAAG,CAACD;YAChB;QACJ;IACJ;AACJ;AAEO,MAAME,4BAA4BC,oBAAoB,oBAAoB,CAAC;IAC9E,gBAAgBb;IAChB,cAAc;QAACc;QAA6BC;KAAwB;AACxE"}
+{"version":3,"file":"transport/ServerWebsocketsTransport.js","sources":["../../src/transport/ServerWebsocketsTransport.ts"],"sourcesContent":["import { WebsocketsTransport } from \"@webiny/api-websockets/exports/api.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\nimport { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\n\nexport class ServerWebsocketsTransportImpl implements WebsocketsTransport.Interface {\n    private readonly connectionManager: WebsocketsConnectionManager.Interface<unknown>;\n    private readonly adapter: WebsocketsServerAdapter.Interface<unknown>;\n\n    public constructor(\n        connectionManager: WebsocketsConnectionManager.Interface<unknown>,\n        adapter: WebsocketsServerAdapter.Interface<unknown>\n    ) {\n        this.connectionManager = connectionManager;\n        this.adapter = adapter;\n    }\n\n    public async send<T extends GenericRecord = GenericRecord>(\n        connections: WebsocketsTransport.SendConnection[],\n        data: WebsocketsTransport.SendData<T>\n    ): Promise<void> {\n        // Connections with no socket in this process; see `removeOwnedHere`.\n        const missing: string[] = [];\n        for (const connection of connections) {\n            const { connectionId } = connection;\n            const socket = this.connectionManager.getSocket(connectionId);\n            if (socket === undefined) {\n                missing.push(connectionId);\n                continue;\n            }\n            try {\n                await this.adapter.send(socket, JSON.stringify(data));\n            } catch (ex) {\n                console.error(\n                    `Failed to send message to connection \"${connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n        await this.connectionManager.removeOwnedHere(missing);\n    }\n\n    public async disconnect(\n        connections: WebsocketsTransport.DisconnectConnection[]\n    ): Promise<void> {\n        const missing: string[] = [];\n        for (const connection of connections) {\n            const { connectionId } = connection;\n            const socket = this.connectionManager.getSocket(connectionId);\n            if (socket === undefined) {\n                missing.push(connectionId);\n                continue;\n            }\n            try {\n                this.adapter.close(socket);\n                await this.connectionManager.remove(connectionId);\n            } catch (ex) {\n                console.error(\n                    `Failed to disconnect connection \"${connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n        await this.connectionManager.removeOwnedHere(missing);\n    }\n}\n\nexport const ServerWebsocketsTransport = WebsocketsTransport.createImplementation({\n    implementation: ServerWebsocketsTransportImpl,\n    dependencies: [WebsocketsConnectionManager, WebsocketsServerAdapter]\n});\n"],"names":["ServerWebsocketsTransportImpl","connectionManager","adapter","connections","data","missing","connection","connectionId","socket","undefined","JSON","ex","console","ServerWebsocketsTransport","WebsocketsTransport","WebsocketsConnectionManager","WebsocketsServerAdapter"],"mappings":";;;AAKO,MAAMA;IAIT,YACIC,iBAAiE,EACjEC,OAAmD,CACrD;QACE,IAAI,CAAC,iBAAiB,GAAGD;QACzB,IAAI,CAAC,OAAO,GAAGC;IACnB;IAEA,MAAa,KACTC,WAAiD,EACjDC,IAAqC,EACxB;QAEb,MAAMC,UAAoB,EAAE;QAC5B,KAAK,MAAMC,cAAcH,YAAa;YAClC,MAAM,EAAEI,YAAY,EAAE,GAAGD;YACzB,MAAME,SAAS,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAACD;YAChD,IAAIC,AAAWC,WAAXD,QAAsB;gBACtBH,QAAQ,IAAI,CAACE;gBACb;YACJ;YACA,IAAI;gBACA,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAACC,QAAQE,KAAK,SAAS,CAACN;YACnD,EAAE,OAAOO,IAAI;gBACTC,QAAQ,KAAK,CACT,CAAC,sCAAsC,EAAEL,aAAa,mCAAmC,CAAC;gBAE9FK,QAAQ,GAAG,CAACD;YAChB;QACJ;QACA,MAAM,IAAI,CAAC,iBAAiB,CAAC,eAAe,CAACN;IACjD;IAEA,MAAa,WACTF,WAAuD,EAC1C;QACb,MAAME,UAAoB,EAAE;QAC5B,KAAK,MAAMC,cAAcH,YAAa;YAClC,MAAM,EAAEI,YAAY,EAAE,GAAGD;YACzB,MAAME,SAAS,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAACD;YAChD,IAAIC,AAAWC,WAAXD,QAAsB;gBACtBH,QAAQ,IAAI,CAACE;gBACb;YACJ;YACA,IAAI;gBACA,IAAI,CAAC,OAAO,CAAC,KAAK,CAACC;gBACnB,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAACD;YACxC,EAAE,OAAOI,IAAI;gBACTC,QAAQ,KAAK,CACT,CAAC,iCAAiC,EAAEL,aAAa,mCAAmC,CAAC;gBAEzFK,QAAQ,GAAG,CAACD;YAChB;QACJ;QACA,MAAM,IAAI,CAAC,iBAAiB,CAAC,eAAe,CAACN;IACjD;AACJ;AAEO,MAAMQ,4BAA4BC,oBAAoB,oBAAoB,CAAC;IAC9E,gBAAgBd;IAChB,cAAc;QAACe;QAA6BC;KAAwB;AACxE"}