@digo-labs/ws 0.1.8 → 0.1.10
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/server.d.ts +3 -0
- package/dist/server.js +20 -8
- package/dist/server.js.map +1 -1
- package/dist/start-hub.cjs +2753 -191
- package/dist/start-hub.cjs.map +1 -1
- package/dist/start-hub.css +240 -0
- package/dist/start-hub.css.map +1 -1
- package/package.json +2 -2
package/dist/server.d.ts
CHANGED
|
@@ -7,6 +7,8 @@ export declare class WebsocketHub {
|
|
|
7
7
|
private logs;
|
|
8
8
|
private messageHandlers;
|
|
9
9
|
private heartbeatInterval;
|
|
10
|
+
private statusTimer;
|
|
11
|
+
private statusDirty;
|
|
10
12
|
constructor(config: WebsocketsHubConfig);
|
|
11
13
|
private heartbeat;
|
|
12
14
|
private handleConnection;
|
|
@@ -14,6 +16,7 @@ export declare class WebsocketHub {
|
|
|
14
16
|
private addLog;
|
|
15
17
|
private broadcast;
|
|
16
18
|
private broadcastStatus;
|
|
19
|
+
private sendStatus;
|
|
17
20
|
onMessage(handler: (channel: string, data: unknown) => void): void;
|
|
18
21
|
getState(channel: string): unknown;
|
|
19
22
|
close(): void;
|
package/dist/server.js
CHANGED
|
@@ -10,6 +10,8 @@ var s = class {
|
|
|
10
10
|
logs = [];
|
|
11
11
|
messageHandlers = [];
|
|
12
12
|
heartbeatInterval;
|
|
13
|
+
statusTimer = null;
|
|
14
|
+
statusDirty = !1;
|
|
13
15
|
constructor(e) {
|
|
14
16
|
this.httpServer = i.createServer((e, t) => {
|
|
15
17
|
if (e.method === "GET" && e.url === "/health") {
|
|
@@ -51,14 +53,14 @@ var s = class {
|
|
|
51
53
|
}
|
|
52
54
|
this.handleMessage(t, e, r);
|
|
53
55
|
}), e.on("close", () => {
|
|
54
|
-
this.clients.delete(t), this.addLog(t
|
|
56
|
+
this.clients.delete(t), this.addLog(t, r.DISCONNECT, ""), this.broadcastStatus();
|
|
55
57
|
});
|
|
56
58
|
}
|
|
57
59
|
handleMessage(e, i, a) {
|
|
58
60
|
switch (a.type) {
|
|
59
61
|
case n.IDENTIFY: {
|
|
60
62
|
let t = [...this.clients].find((t) => t.id === a.id && t !== e);
|
|
61
|
-
t && (t.ws.close(4e3), this.clients.delete(t)), e.id = a.id, e.app = a.app, this.addLog(
|
|
63
|
+
t && (t.ws.close(4e3), this.clients.delete(t)), e.id = a.id, e.app = a.app, this.addLog(e, r.CONNECT, ""), this.broadcastStatus();
|
|
62
64
|
break;
|
|
63
65
|
}
|
|
64
66
|
case n.SUBSCRIBE:
|
|
@@ -71,10 +73,10 @@ var s = class {
|
|
|
71
73
|
};
|
|
72
74
|
i.send(JSON.stringify(e));
|
|
73
75
|
}
|
|
74
|
-
}), this.addLog(e
|
|
76
|
+
}), this.addLog(e, r.SUBSCRIBE, a.channels.join(", ")), this.broadcastStatus();
|
|
75
77
|
break;
|
|
76
78
|
case n.UNSUBSCRIBE:
|
|
77
|
-
a.channels.forEach((t) => e.channels.delete(t)), this.addLog(e
|
|
79
|
+
a.channels.forEach((t) => e.channels.delete(t)), this.addLog(e, r.UNSUBSCRIBE, a.channels.join(", ")), this.broadcastStatus();
|
|
78
80
|
break;
|
|
79
81
|
case n.MESSAGE: {
|
|
80
82
|
let n = {
|
|
@@ -82,7 +84,7 @@ var s = class {
|
|
|
82
84
|
channel: a.channel,
|
|
83
85
|
data: a.data
|
|
84
86
|
};
|
|
85
|
-
this.broadcast(e, a.channel, n), this.messageHandlers.forEach((e) => e(a.channel, a.data)), this.addLog(e
|
|
87
|
+
this.broadcast(e, a.channel, n), this.messageHandlers.forEach((e) => e(a.channel, a.data)), this.addLog(e, r.MESSAGE, a.channel, a.data), this.broadcastStatus();
|
|
86
88
|
break;
|
|
87
89
|
}
|
|
88
90
|
case n.STATE: {
|
|
@@ -92,7 +94,7 @@ var s = class {
|
|
|
92
94
|
channel: a.channel,
|
|
93
95
|
data: a.data
|
|
94
96
|
};
|
|
95
|
-
this.broadcast(e, a.channel, n), this.messageHandlers.forEach((e) => e(a.channel, a.data)), this.addLog(e
|
|
97
|
+
this.broadcast(e, a.channel, n), this.messageHandlers.forEach((e) => e(a.channel, a.data)), this.addLog(e, r.STATE, a.channel, a.data), this.broadcastStatus();
|
|
96
98
|
break;
|
|
97
99
|
}
|
|
98
100
|
}
|
|
@@ -100,7 +102,8 @@ var s = class {
|
|
|
100
102
|
addLog(e, t, n, r) {
|
|
101
103
|
this.logs.push({
|
|
102
104
|
timestamp: Date.now(),
|
|
103
|
-
sender: e,
|
|
105
|
+
sender: e.id,
|
|
106
|
+
app: e.app,
|
|
104
107
|
type: t,
|
|
105
108
|
channel: n,
|
|
106
109
|
data: r
|
|
@@ -113,6 +116,15 @@ var s = class {
|
|
|
113
116
|
});
|
|
114
117
|
}
|
|
115
118
|
broadcastStatus() {
|
|
119
|
+
if (this.statusTimer) {
|
|
120
|
+
this.statusDirty = !0;
|
|
121
|
+
return;
|
|
122
|
+
}
|
|
123
|
+
this.sendStatus(), this.statusTimer = setTimeout(() => {
|
|
124
|
+
this.statusTimer = null, this.statusDirty && (this.statusDirty = !1, this.broadcastStatus());
|
|
125
|
+
}, 200);
|
|
126
|
+
}
|
|
127
|
+
sendStatus() {
|
|
116
128
|
let n = {
|
|
117
129
|
clients: [...this.clients].map((e) => ({
|
|
118
130
|
id: e.id,
|
|
@@ -139,7 +151,7 @@ var s = class {
|
|
|
139
151
|
return this.state.get(e);
|
|
140
152
|
}
|
|
141
153
|
close() {
|
|
142
|
-
clearInterval(this.heartbeatInterval), this.clients.forEach((e) => {
|
|
154
|
+
clearInterval(this.heartbeatInterval), this.statusTimer && clearTimeout(this.statusTimer), this.clients.forEach((e) => {
|
|
143
155
|
e.ws.readyState === a.OPEN && e.ws.close(1001, "Hub shutting down");
|
|
144
156
|
}), this.wss.close(), this.httpServer.close();
|
|
145
157
|
}
|
package/dist/server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.js","names":[],"sources":["../src/server.ts"],"sourcesContent":["import http from 'node:http';\nimport { WebSocketServer, WebSocket } from 'ws';\n\nimport { WebsocketClientMessage, WebsocketClientEvent, WebSocketHubEvent, WebsocketHubMessage, WebsocketHubLogEntry, WebsocketLogEvent, WebsocketHubStatus, WebsocketsHubConfig, WEBSOCKET_HUB_CHANNEL } from '@digo-labs/common';\n\ninterface ConnectedClient {\n id: string;\n app: string;\n ws: WebSocket;\n channels: Set<string>;\n alive: boolean;\n}\n\nexport class WebsocketHub {\n private httpServer: http.Server;\n private wss: WebSocketServer;\n\n private clients = new Set<ConnectedClient>();\n private state = new Map<string, unknown>();\n\n private logs: WebsocketHubLogEntry[] = [];\n private messageHandlers: ((channel: string, data: unknown) => void)[] = [];\n private heartbeatInterval: ReturnType<typeof setInterval>;\n\n constructor(config: WebsocketsHubConfig) {\n this.httpServer = http.createServer((req, res) => {\n if (req.method === 'GET' && req.url === '/health') {\n res.writeHead(200, { 'Content-Type': 'application/json' });\n res.end(JSON.stringify({ ok: true, clients: this.clients.size }));\n return;\n }\n\n res.writeHead(404);\n res.end();\n });\n\n this.wss = new WebSocketServer({ server: this.httpServer });\n\n this.heartbeatInterval = setInterval(() => this.heartbeat(), 30000);\n\n this.wss.on('close', () => clearInterval(this.heartbeatInterval));\n this.wss.on('connection', socket => this.handleConnection(socket));\n\n this.httpServer.listen(config.port);\n }\n\n private heartbeat(): void {\n let terminated = false;\n\n this.clients.forEach((client) => {\n if (!client.alive) {\n client.ws.terminate();\n\n this.clients.delete(client);\n\n terminated = true;\n\n return;\n }\n\n client.alive = false;\n\n client.ws.ping();\n });\n\n if (terminated) this.broadcastStatus();\n }\n\n private handleConnection(socket: WebSocket): void {\n const client: ConnectedClient = {\n id: 'unknown',\n app: 'unknown',\n ws: socket,\n channels: new Set(),\n alive: true,\n };\n\n this.clients.add(client);\n\n socket.on('error', () => {});\n\n socket.on('pong', () => {\n client.alive = true;\n });\n\n socket.on('message', (raw) => {\n let message: WebsocketClientMessage;\n\n try {\n message = JSON.parse(raw.toString());\n } catch { return; }\n\n this.handleMessage(client, socket, message);\n });\n\n socket.on('close', () => {\n this.clients.delete(client);\n\n this.addLog(client.id, WebsocketLogEvent.DISCONNECT, '');\n\n this.broadcastStatus();\n });\n }\n\n private handleMessage(client: ConnectedClient, socket: WebSocket, message: WebsocketClientMessage): void {\n switch (message.type) {\n case WebsocketClientEvent.IDENTIFY: {\n const existing = [...this.clients].find(c => c.id === message.id && c !== client);\n\n if (existing) {\n existing.ws.close(4000);\n\n this.clients.delete(existing);\n }\n\n client.id = message.id;\n client.app = message.app;\n\n this.addLog(message.id, WebsocketLogEvent.CONNECT, '');\n\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.SUBSCRIBE: {\n message.channels.forEach((channel) => {\n client.channels.add(channel);\n\n if (this.state.has(channel)) {\n const stateMessage: WebsocketHubMessage = {\n type: WebSocketHubEvent.STATE,\n channel: channel,\n data: this.state.get(channel),\n };\n\n socket.send(JSON.stringify(stateMessage));\n }\n });\n\n this.addLog(client.id, WebsocketLogEvent.SUBSCRIBE, message.channels.join(', '));\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.UNSUBSCRIBE: {\n message.channels.forEach(channel => client.channels.delete(channel));\n\n this.addLog(client.id, WebsocketLogEvent.UNSUBSCRIBE, message.channels.join(', '));\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.MESSAGE: {\n const outgoing: WebsocketHubMessage = {\n type: WebSocketHubEvent.MESSAGE,\n channel: message.channel,\n data: message.data,\n };\n\n this.broadcast(client, message.channel, outgoing);\n\n this.messageHandlers.forEach(h => h(message.channel, message.data));\n\n this.addLog(client.id, WebsocketLogEvent.MESSAGE, message.channel, message.data);\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.STATE: {\n this.state.set(message.channel, message.data);\n\n const outgoing: WebsocketHubMessage = {\n type: WebSocketHubEvent.STATE,\n channel: message.channel,\n data: message.data,\n };\n\n this.broadcast(client, message.channel, outgoing);\n\n this.messageHandlers.forEach(h => h(message.channel, message.data));\n\n this.addLog(client.id, WebsocketLogEvent.STATE, message.channel, message.data);\n\n this.broadcastStatus();\n break;\n }\n }\n }\n\n private addLog(sender: string, type: WebsocketLogEvent, channel: string, data?: unknown): void {\n this.logs.push({ timestamp: Date.now(), sender: sender, type: type, channel: channel, data: data });\n\n if (this.logs.length > 808) this.logs.shift();\n }\n\n private broadcast(sender: ConnectedClient, channel: string, message: WebsocketHubMessage): void {\n const payload = JSON.stringify(message);\n\n this.clients.forEach((client) => {\n if (client !== sender && client.channels.has(channel) && client.ws.readyState === WebSocket.OPEN) {\n client.ws.send(payload);\n }\n });\n }\n\n private broadcastStatus(): void {\n const status: WebsocketHubStatus = {\n clients: [...this.clients].map(client => ({ id: client.id, app: client.app, channels: [...client.channels] })),\n state: Object.fromEntries([...this.state].filter(([k]) => k !== WEBSOCKET_HUB_CHANNEL)),\n logs: this.logs,\n };\n\n this.state.set(WEBSOCKET_HUB_CHANNEL, status);\n\n const message: WebsocketHubMessage = {\n type: WebSocketHubEvent.STATE,\n channel: WEBSOCKET_HUB_CHANNEL,\n data: status,\n };\n\n const payload = JSON.stringify(message);\n\n this.clients.forEach((client) => {\n if (client.channels.has(WEBSOCKET_HUB_CHANNEL) && client.ws.readyState === WebSocket.OPEN) {\n client.ws.send(payload);\n }\n });\n }\n\n public onMessage(handler: (channel: string, data: unknown) => void): void { this.messageHandlers.push(handler); }\n\n public getState(channel: string): unknown { return this.state.get(channel); }\n\n public close(): void {\n clearInterval(this.heartbeatInterval);\n\n this.clients.forEach((client) => {\n if (client.ws.readyState === WebSocket.OPEN) {\n client.ws.close(1001, 'Hub shutting down');\n }\n });\n\n this.wss.close();\n this.httpServer.close();\n }\n}\n"],"mappings":";;;;AAaA,IAAa,IAAb,MAA0B;CACxB;CACA;CAEA,0BAAkB,IAAI,IAAqB;CAC3C,wBAAkB,IAAI,IAAqB;CAE3C,OAAoD,CAAC;CACrD,kBAA0E,CAAC;CAC3E;CAEA,YAAY,GAA6B;EAmBvC,AAlBA,KAAK,aAAa,EAAK,cAAc,GAAK,MAAQ;GAChD,IAAI,EAAI,WAAW,SAAS,EAAI,QAAQ,WAAW;IAEjD,AADA,EAAI,UAAU,KAAK,EAAE,gBAAgB,mBAAmB,CAAC,GACzD,EAAI,IAAI,KAAK,UAAU;KAAE,IAAI;KAAM,SAAS,KAAK,QAAQ;IAAK,CAAC,CAAC;IAChE;GACF;GAGA,AADA,EAAI,UAAU,GAAG,GACjB,EAAI,IAAI;EACV,CAAC,GAED,KAAK,MAAM,IAAI,EAAgB,EAAE,QAAQ,KAAK,WAAW,CAAC,GAE1D,KAAK,oBAAoB,kBAAkB,KAAK,UAAU,GAAG,GAAK,GAElE,KAAK,IAAI,GAAG,eAAoB,cAAc,KAAK,iBAAiB,CAAC,GACrE,KAAK,IAAI,GAAG,eAAc,MAAU,KAAK,iBAAiB,CAAM,CAAC,GAEjE,KAAK,WAAW,OAAO,EAAO,IAAI;CACpC;CAEA,YAA0B;EACxB,IAAI,IAAa;EAkBjB,AAhBA,KAAK,QAAQ,SAAS,MAAW;GAC/B,IAAI,CAAC,EAAO,OAAO;IAKjB,AAJA,EAAO,GAAG,UAAU,GAEpB,KAAK,QAAQ,OAAO,CAAM,GAE1B,IAAa;IAEb;GACF;GAIA,AAFA,EAAO,QAAQ,IAEf,EAAO,GAAG,KAAK;EACjB,CAAC,GAEG,KAAY,KAAK,gBAAgB;CACvC;CAEA,iBAAyB,GAAyB;EAChD,IAAM,IAA0B;GAC9B,IAAU;GACV,KAAU;GACV,IAAU;GACV,0BAAU,IAAI,IAAI;GAClB,OAAU;EACZ;EAoBA,AAlBA,KAAK,QAAQ,IAAI,CAAM,GAEvB,EAAO,GAAG,eAAe,CAAC,CAAC,GAE3B,EAAO,GAAG,cAAc;GACtB,EAAO,QAAQ;EACjB,CAAC,GAED,EAAO,GAAG,YAAY,MAAQ;GAC5B,IAAI;GAEJ,IAAI;IACF,IAAU,KAAK,MAAM,EAAI,SAAS,CAAC;GACrC,QAAQ;IAAE;GAAQ;GAElB,KAAK,cAAc,GAAQ,GAAQ,CAAO;EAC5C,CAAC,GAED,EAAO,GAAG,eAAe;GAKvB,AAJA,KAAK,QAAQ,OAAO,CAAM,GAE1B,KAAK,OAAO,EAAO,IAAI,EAAkB,YAAY,EAAE,GAEvD,KAAK,gBAAgB;EACvB,CAAC;CACH;CAEA,cAAsB,GAAyB,GAAmB,GAAuC;EACvG,QAAQ,EAAQ,MAAhB;GACE,KAAK,EAAqB,UAAU;IAClC,IAAM,IAAW,CAAC,GAAG,KAAK,OAAO,CAAC,CAAC,MAAK,MAAK,EAAE,OAAO,EAAQ,MAAM,MAAM,CAAM;IAahF,AAXI,MACF,EAAS,GAAG,MAAM,GAAI,GAEtB,KAAK,QAAQ,OAAO,CAAQ,IAG9B,EAAO,KAAM,EAAQ,IACrB,EAAO,MAAM,EAAQ,KAErB,KAAK,OAAO,EAAQ,IAAI,EAAkB,SAAS,EAAE,GAErD,KAAK,gBAAgB;IACrB;GACF;GAEA,KAAK,EAAqB;IAgBxB,AAfA,EAAQ,SAAS,SAAS,MAAY;KAGpC,IAFA,EAAO,SAAS,IAAI,CAAO,GAEvB,KAAK,MAAM,IAAI,CAAO,GAAG;MAC3B,IAAM,IAAoC;OACxC,MAAS,EAAkB;OAClB;OACT,MAAS,KAAK,MAAM,IAAI,CAAO;MACjC;MAEA,EAAO,KAAK,KAAK,UAAU,CAAY,CAAC;KAC1C;IACF,CAAC,GAED,KAAK,OAAO,EAAO,IAAI,EAAkB,WAAW,EAAQ,SAAS,KAAK,IAAI,CAAC,GAC/E,KAAK,gBAAgB;IACrB;GAGF,KAAK,EAAqB;IAIxB,AAHA,EAAQ,SAAS,SAAQ,MAAW,EAAO,SAAS,OAAO,CAAO,CAAC,GAEnE,KAAK,OAAO,EAAO,IAAI,EAAkB,aAAa,EAAQ,SAAS,KAAK,IAAI,CAAC,GACjF,KAAK,gBAAgB;IACrB;GAGF,KAAK,EAAqB,SAAS;IACjC,IAAM,IAAgC;KACpC,MAAS,EAAkB;KAC3B,SAAS,EAAQ;KACjB,MAAS,EAAQ;IACnB;IAOA,AALA,KAAK,UAAU,GAAQ,EAAQ,SAAS,CAAQ,GAEhD,KAAK,gBAAgB,SAAQ,MAAK,EAAE,EAAQ,SAAS,EAAQ,IAAI,CAAC,GAElE,KAAK,OAAO,EAAO,IAAI,EAAkB,SAAS,EAAQ,SAAS,EAAQ,IAAI,GAC/E,KAAK,gBAAgB;IACrB;GACF;GAEA,KAAK,EAAqB,OAAO;IAC/B,KAAK,MAAM,IAAI,EAAQ,SAAS,EAAQ,IAAI;IAE5C,IAAM,IAAgC;KACpC,MAAS,EAAkB;KAC3B,SAAS,EAAQ;KACjB,MAAS,EAAQ;IACnB;IAQA,AANA,KAAK,UAAU,GAAQ,EAAQ,SAAS,CAAQ,GAEhD,KAAK,gBAAgB,SAAQ,MAAK,EAAE,EAAQ,SAAS,EAAQ,IAAI,CAAC,GAElE,KAAK,OAAO,EAAO,IAAI,EAAkB,OAAO,EAAQ,SAAS,EAAQ,IAAI,GAE7E,KAAK,gBAAgB;IACrB;GACF;EACF;CACF;CAEA,OAAe,GAAgB,GAAyB,GAAiB,GAAsB;EAG7F,AAFA,KAAK,KAAK,KAAK;GAAE,WAAW,KAAK,IAAI;GAAW;GAAc;GAAe;GAAe;EAAK,CAAC,GAE9F,KAAK,KAAK,SAAS,OAAK,KAAK,KAAK,MAAM;CAC9C;CAEA,UAAkB,GAAyB,GAAiB,GAAoC;EAC9F,IAAM,IAAU,KAAK,UAAU,CAAO;EAEtC,KAAK,QAAQ,SAAS,MAAW;GAC/B,AAAI,MAAW,KAAU,EAAO,SAAS,IAAI,CAAO,KAAK,EAAO,GAAG,eAAe,EAAU,QAC1F,EAAO,GAAG,KAAK,CAAO;EAE1B,CAAC;CACH;CAEA,kBAAgC;EAC9B,IAAM,IAA6B;GACjC,SAAS,CAAC,GAAG,KAAK,OAAO,CAAC,CAAC,KAAI,OAAW;IAAE,IAAI,EAAO;IAAI,KAAK,EAAO;IAAK,UAAU,CAAC,GAAG,EAAO,QAAQ;GAAE,EAAE;GAC7G,OAAS,OAAO,YAAY,CAAC,GAAG,KAAK,KAAK,CAAC,CAAC,QAAQ,CAAC,OAAO,MAAM,CAAqB,CAAC;GACxF,MAAS,KAAK;EAChB;EAEA,KAAK,MAAM,IAAI,GAAuB,CAAM;EAE5C,IAAM,IAA+B;GACnC,MAAS,EAAkB;GAC3B,SAAS;GACT,MAAS;EACX,GAEM,IAAU,KAAK,UAAU,CAAO;EAEtC,KAAK,QAAQ,SAAS,MAAW;GAC/B,AAAI,EAAO,SAAS,IAAI,CAAqB,KAAK,EAAO,GAAG,eAAe,EAAU,QACnF,EAAO,GAAG,KAAK,CAAO;EAE1B,CAAC;CACH;CAEA,UAAiB,GAAyD;EAAE,KAAK,gBAAgB,KAAK,CAAO;CAAG;CAEhH,SAAgB,GAA0B;EAAE,OAAO,KAAK,MAAM,IAAI,CAAO;CAAG;CAE5E,QAAqB;EAUnB,AATA,cAAc,KAAK,iBAAiB,GAEpC,KAAK,QAAQ,SAAS,MAAW;GAC/B,AAAI,EAAO,GAAG,eAAe,EAAU,QACrC,EAAO,GAAG,MAAM,MAAM,mBAAmB;EAE7C,CAAC,GAED,KAAK,IAAI,MAAM,GACf,KAAK,WAAW,MAAM;CACxB;AACF"}
|
|
1
|
+
{"version":3,"file":"server.js","names":[],"sources":["../src/server.ts"],"sourcesContent":["import http from 'node:http';\nimport { WebSocketServer, WebSocket } from 'ws';\n\nimport { WebsocketClientMessage, WebsocketClientEvent, WebSocketHubEvent, WebsocketHubMessage, WebsocketHubLogEntry, WebsocketLogEvent, WebsocketHubStatus, WebsocketsHubConfig, WEBSOCKET_HUB_CHANNEL } from '@digo-labs/common';\n\ninterface ConnectedClient {\n id: string;\n app: string;\n ws: WebSocket;\n channels: Set<string>;\n alive: boolean;\n}\n\nexport class WebsocketHub {\n private httpServer: http.Server;\n private wss: WebSocketServer;\n\n private clients = new Set<ConnectedClient>();\n private state = new Map<string, unknown>();\n\n private logs: WebsocketHubLogEntry[] = [];\n private messageHandlers: ((channel: string, data: unknown) => void)[] = [];\n private heartbeatInterval: ReturnType<typeof setInterval>;\n private statusTimer: ReturnType<typeof setTimeout> | null = null;\n private statusDirty = false;\n\n constructor(config: WebsocketsHubConfig) {\n this.httpServer = http.createServer((req, res) => {\n if (req.method === 'GET' && req.url === '/health') {\n res.writeHead(200, { 'Content-Type': 'application/json' });\n res.end(JSON.stringify({ ok: true, clients: this.clients.size }));\n return;\n }\n\n res.writeHead(404);\n res.end();\n });\n\n this.wss = new WebSocketServer({ server: this.httpServer });\n\n this.heartbeatInterval = setInterval(() => this.heartbeat(), 30000);\n\n this.wss.on('close', () => clearInterval(this.heartbeatInterval));\n this.wss.on('connection', socket => this.handleConnection(socket));\n\n this.httpServer.listen(config.port);\n }\n\n private heartbeat(): void {\n let terminated = false;\n\n this.clients.forEach((client) => {\n if (!client.alive) {\n client.ws.terminate();\n\n this.clients.delete(client);\n\n terminated = true;\n\n return;\n }\n\n client.alive = false;\n\n client.ws.ping();\n });\n\n if (terminated) this.broadcastStatus();\n }\n\n private handleConnection(socket: WebSocket): void {\n const client: ConnectedClient = {\n id: 'unknown',\n app: 'unknown',\n ws: socket,\n channels: new Set(),\n alive: true,\n };\n\n this.clients.add(client);\n\n socket.on('error', () => {});\n\n socket.on('pong', () => {\n client.alive = true;\n });\n\n socket.on('message', (raw) => {\n let message: WebsocketClientMessage;\n\n try {\n message = JSON.parse(raw.toString());\n } catch { return; }\n\n this.handleMessage(client, socket, message);\n });\n\n socket.on('close', () => {\n this.clients.delete(client);\n\n this.addLog(client, WebsocketLogEvent.DISCONNECT, '');\n\n this.broadcastStatus();\n });\n }\n\n private handleMessage(client: ConnectedClient, socket: WebSocket, message: WebsocketClientMessage): void {\n switch (message.type) {\n case WebsocketClientEvent.IDENTIFY: {\n const existing = [...this.clients].find(c => c.id === message.id && c !== client);\n\n if (existing) {\n existing.ws.close(4000);\n\n this.clients.delete(existing);\n }\n\n client.id = message.id;\n client.app = message.app;\n\n this.addLog(client, WebsocketLogEvent.CONNECT, '');\n\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.SUBSCRIBE: {\n message.channels.forEach((channel) => {\n client.channels.add(channel);\n\n if (this.state.has(channel)) {\n const stateMessage: WebsocketHubMessage = {\n type: WebSocketHubEvent.STATE,\n channel: channel,\n data: this.state.get(channel),\n };\n\n socket.send(JSON.stringify(stateMessage));\n }\n });\n\n this.addLog(client, WebsocketLogEvent.SUBSCRIBE, message.channels.join(', '));\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.UNSUBSCRIBE: {\n message.channels.forEach(channel => client.channels.delete(channel));\n\n this.addLog(client, WebsocketLogEvent.UNSUBSCRIBE, message.channels.join(', '));\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.MESSAGE: {\n const outgoing: WebsocketHubMessage = {\n type: WebSocketHubEvent.MESSAGE,\n channel: message.channel,\n data: message.data,\n };\n\n this.broadcast(client, message.channel, outgoing);\n\n this.messageHandlers.forEach(h => h(message.channel, message.data));\n\n this.addLog(client, WebsocketLogEvent.MESSAGE, message.channel, message.data);\n this.broadcastStatus();\n break;\n }\n\n case WebsocketClientEvent.STATE: {\n this.state.set(message.channel, message.data);\n\n const outgoing: WebsocketHubMessage = {\n type: WebSocketHubEvent.STATE,\n channel: message.channel,\n data: message.data,\n };\n\n this.broadcast(client, message.channel, outgoing);\n\n this.messageHandlers.forEach(h => h(message.channel, message.data));\n\n this.addLog(client, WebsocketLogEvent.STATE, message.channel, message.data);\n\n this.broadcastStatus();\n break;\n }\n }\n }\n\n private addLog(client: ConnectedClient, type: WebsocketLogEvent, channel: string, data?: unknown): void {\n this.logs.push({ timestamp: Date.now(), sender: client.id, app: client.app, type: type, channel: channel, data: data });\n\n if (this.logs.length > 808) this.logs.shift();\n }\n\n private broadcast(sender: ConnectedClient, channel: string, message: WebsocketHubMessage): void {\n const payload = JSON.stringify(message);\n\n this.clients.forEach((client) => {\n if (client !== sender && client.channels.has(channel) && client.ws.readyState === WebSocket.OPEN) {\n client.ws.send(payload);\n }\n });\n }\n\n private broadcastStatus(): void {\n if (this.statusTimer) {\n this.statusDirty = true;\n\n return;\n }\n\n this.sendStatus();\n\n this.statusTimer = setTimeout(() => {\n this.statusTimer = null;\n\n if (!this.statusDirty) return;\n\n this.statusDirty = false;\n\n this.broadcastStatus();\n }, 200);\n }\n\n private sendStatus(): void {\n const status: WebsocketHubStatus = {\n clients: [...this.clients].map(client => ({ id: client.id, app: client.app, channels: [...client.channels] })),\n state: Object.fromEntries([...this.state].filter(([k]) => k !== WEBSOCKET_HUB_CHANNEL)),\n logs: this.logs,\n };\n\n this.state.set(WEBSOCKET_HUB_CHANNEL, status);\n\n const message: WebsocketHubMessage = {\n type: WebSocketHubEvent.STATE,\n channel: WEBSOCKET_HUB_CHANNEL,\n data: status,\n };\n\n const payload = JSON.stringify(message);\n\n this.clients.forEach((client) => {\n if (client.channels.has(WEBSOCKET_HUB_CHANNEL) && client.ws.readyState === WebSocket.OPEN) {\n client.ws.send(payload);\n }\n });\n }\n\n public onMessage(handler: (channel: string, data: unknown) => void): void { this.messageHandlers.push(handler); }\n\n public getState(channel: string): unknown { return this.state.get(channel); }\n\n public close(): void {\n clearInterval(this.heartbeatInterval);\n\n if (this.statusTimer) clearTimeout(this.statusTimer);\n\n this.clients.forEach((client) => {\n if (client.ws.readyState === WebSocket.OPEN) {\n client.ws.close(1001, 'Hub shutting down');\n }\n });\n\n this.wss.close();\n this.httpServer.close();\n }\n}\n"],"mappings":";;;;AAaA,IAAa,IAAb,MAA0B;CACxB;CACA;CAEA,0BAAkB,IAAI,IAAqB;CAC3C,wBAAkB,IAAI,IAAqB;CAE3C,OAAoD,CAAC;CACrD,kBAA0E,CAAC;CAC3E;CACA,cAAkE;CAClE,cAAsB;CAEtB,YAAY,GAA6B;EAmBvC,AAlBA,KAAK,aAAa,EAAK,cAAc,GAAK,MAAQ;GAChD,IAAI,EAAI,WAAW,SAAS,EAAI,QAAQ,WAAW;IAEjD,AADA,EAAI,UAAU,KAAK,EAAE,gBAAgB,mBAAmB,CAAC,GACzD,EAAI,IAAI,KAAK,UAAU;KAAE,IAAI;KAAM,SAAS,KAAK,QAAQ;IAAK,CAAC,CAAC;IAChE;GACF;GAGA,AADA,EAAI,UAAU,GAAG,GACjB,EAAI,IAAI;EACV,CAAC,GAED,KAAK,MAAM,IAAI,EAAgB,EAAE,QAAQ,KAAK,WAAW,CAAC,GAE1D,KAAK,oBAAoB,kBAAkB,KAAK,UAAU,GAAG,GAAK,GAElE,KAAK,IAAI,GAAG,eAAoB,cAAc,KAAK,iBAAiB,CAAC,GACrE,KAAK,IAAI,GAAG,eAAc,MAAU,KAAK,iBAAiB,CAAM,CAAC,GAEjE,KAAK,WAAW,OAAO,EAAO,IAAI;CACpC;CAEA,YAA0B;EACxB,IAAI,IAAa;EAkBjB,AAhBA,KAAK,QAAQ,SAAS,MAAW;GAC/B,IAAI,CAAC,EAAO,OAAO;IAKjB,AAJA,EAAO,GAAG,UAAU,GAEpB,KAAK,QAAQ,OAAO,CAAM,GAE1B,IAAa;IAEb;GACF;GAIA,AAFA,EAAO,QAAQ,IAEf,EAAO,GAAG,KAAK;EACjB,CAAC,GAEG,KAAY,KAAK,gBAAgB;CACvC;CAEA,iBAAyB,GAAyB;EAChD,IAAM,IAA0B;GAC9B,IAAU;GACV,KAAU;GACV,IAAU;GACV,0BAAU,IAAI,IAAI;GAClB,OAAU;EACZ;EAoBA,AAlBA,KAAK,QAAQ,IAAI,CAAM,GAEvB,EAAO,GAAG,eAAe,CAAC,CAAC,GAE3B,EAAO,GAAG,cAAc;GACtB,EAAO,QAAQ;EACjB,CAAC,GAED,EAAO,GAAG,YAAY,MAAQ;GAC5B,IAAI;GAEJ,IAAI;IACF,IAAU,KAAK,MAAM,EAAI,SAAS,CAAC;GACrC,QAAQ;IAAE;GAAQ;GAElB,KAAK,cAAc,GAAQ,GAAQ,CAAO;EAC5C,CAAC,GAED,EAAO,GAAG,eAAe;GAKvB,AAJA,KAAK,QAAQ,OAAO,CAAM,GAE1B,KAAK,OAAO,GAAQ,EAAkB,YAAY,EAAE,GAEpD,KAAK,gBAAgB;EACvB,CAAC;CACH;CAEA,cAAsB,GAAyB,GAAmB,GAAuC;EACvG,QAAQ,EAAQ,MAAhB;GACE,KAAK,EAAqB,UAAU;IAClC,IAAM,IAAW,CAAC,GAAG,KAAK,OAAO,CAAC,CAAC,MAAK,MAAK,EAAE,OAAO,EAAQ,MAAM,MAAM,CAAM;IAahF,AAXI,MACF,EAAS,GAAG,MAAM,GAAI,GAEtB,KAAK,QAAQ,OAAO,CAAQ,IAG9B,EAAO,KAAM,EAAQ,IACrB,EAAO,MAAM,EAAQ,KAErB,KAAK,OAAO,GAAQ,EAAkB,SAAS,EAAE,GAEjD,KAAK,gBAAgB;IACrB;GACF;GAEA,KAAK,EAAqB;IAgBxB,AAfA,EAAQ,SAAS,SAAS,MAAY;KAGpC,IAFA,EAAO,SAAS,IAAI,CAAO,GAEvB,KAAK,MAAM,IAAI,CAAO,GAAG;MAC3B,IAAM,IAAoC;OACxC,MAAS,EAAkB;OAClB;OACT,MAAS,KAAK,MAAM,IAAI,CAAO;MACjC;MAEA,EAAO,KAAK,KAAK,UAAU,CAAY,CAAC;KAC1C;IACF,CAAC,GAED,KAAK,OAAO,GAAQ,EAAkB,WAAW,EAAQ,SAAS,KAAK,IAAI,CAAC,GAC5E,KAAK,gBAAgB;IACrB;GAGF,KAAK,EAAqB;IAIxB,AAHA,EAAQ,SAAS,SAAQ,MAAW,EAAO,SAAS,OAAO,CAAO,CAAC,GAEnE,KAAK,OAAO,GAAQ,EAAkB,aAAa,EAAQ,SAAS,KAAK,IAAI,CAAC,GAC9E,KAAK,gBAAgB;IACrB;GAGF,KAAK,EAAqB,SAAS;IACjC,IAAM,IAAgC;KACpC,MAAS,EAAkB;KAC3B,SAAS,EAAQ;KACjB,MAAS,EAAQ;IACnB;IAOA,AALA,KAAK,UAAU,GAAQ,EAAQ,SAAS,CAAQ,GAEhD,KAAK,gBAAgB,SAAQ,MAAK,EAAE,EAAQ,SAAS,EAAQ,IAAI,CAAC,GAElE,KAAK,OAAO,GAAQ,EAAkB,SAAS,EAAQ,SAAS,EAAQ,IAAI,GAC5E,KAAK,gBAAgB;IACrB;GACF;GAEA,KAAK,EAAqB,OAAO;IAC/B,KAAK,MAAM,IAAI,EAAQ,SAAS,EAAQ,IAAI;IAE5C,IAAM,IAAgC;KACpC,MAAS,EAAkB;KAC3B,SAAS,EAAQ;KACjB,MAAS,EAAQ;IACnB;IAQA,AANA,KAAK,UAAU,GAAQ,EAAQ,SAAS,CAAQ,GAEhD,KAAK,gBAAgB,SAAQ,MAAK,EAAE,EAAQ,SAAS,EAAQ,IAAI,CAAC,GAElE,KAAK,OAAO,GAAQ,EAAkB,OAAO,EAAQ,SAAS,EAAQ,IAAI,GAE1E,KAAK,gBAAgB;IACrB;GACF;EACF;CACF;CAEA,OAAe,GAAyB,GAAyB,GAAiB,GAAsB;EAGtG,AAFA,KAAK,KAAK,KAAK;GAAE,WAAW,KAAK,IAAI;GAAG,QAAQ,EAAO;GAAI,KAAK,EAAO;GAAW;GAAe;GAAe;EAAK,CAAC,GAElH,KAAK,KAAK,SAAS,OAAK,KAAK,KAAK,MAAM;CAC9C;CAEA,UAAkB,GAAyB,GAAiB,GAAoC;EAC9F,IAAM,IAAU,KAAK,UAAU,CAAO;EAEtC,KAAK,QAAQ,SAAS,MAAW;GAC/B,AAAI,MAAW,KAAU,EAAO,SAAS,IAAI,CAAO,KAAK,EAAO,GAAG,eAAe,EAAU,QAC1F,EAAO,GAAG,KAAK,CAAO;EAE1B,CAAC;CACH;CAEA,kBAAgC;EAC9B,IAAI,KAAK,aAAa;GACpB,KAAK,cAAc;GAEnB;EACF;EAIA,AAFA,KAAK,WAAW,GAEhB,KAAK,cAAc,iBAAiB;GAClC,KAAK,cAAc,MAEd,KAAK,gBAEV,KAAK,cAAc,IAEnB,KAAK,gBAAgB;EACvB,GAAG,GAAG;CACR;CAEA,aAA2B;EACzB,IAAM,IAA6B;GACjC,SAAS,CAAC,GAAG,KAAK,OAAO,CAAC,CAAC,KAAI,OAAW;IAAE,IAAI,EAAO;IAAI,KAAK,EAAO;IAAK,UAAU,CAAC,GAAG,EAAO,QAAQ;GAAE,EAAE;GAC7G,OAAS,OAAO,YAAY,CAAC,GAAG,KAAK,KAAK,CAAC,CAAC,QAAQ,CAAC,OAAO,MAAM,CAAqB,CAAC;GACxF,MAAS,KAAK;EAChB;EAEA,KAAK,MAAM,IAAI,GAAuB,CAAM;EAE5C,IAAM,IAA+B;GACnC,MAAS,EAAkB;GAC3B,SAAS;GACT,MAAS;EACX,GAEM,IAAU,KAAK,UAAU,CAAO;EAEtC,KAAK,QAAQ,SAAS,MAAW;GAC/B,AAAI,EAAO,SAAS,IAAI,CAAqB,KAAK,EAAO,GAAG,eAAe,EAAU,QACnF,EAAO,GAAG,KAAK,CAAO;EAE1B,CAAC;CACH;CAEA,UAAiB,GAAyD;EAAE,KAAK,gBAAgB,KAAK,CAAO;CAAG;CAEhH,SAAgB,GAA0B;EAAE,OAAO,KAAK,MAAM,IAAI,CAAO;CAAG;CAE5E,QAAqB;EAYnB,AAXA,cAAc,KAAK,iBAAiB,GAEhC,KAAK,eAAa,aAAa,KAAK,WAAW,GAEnD,KAAK,QAAQ,SAAS,MAAW;GAC/B,AAAI,EAAO,GAAG,eAAe,EAAU,QACrC,EAAO,GAAG,MAAM,MAAM,mBAAmB;EAE7C,CAAC,GAED,KAAK,IAAI,MAAM,GACf,KAAK,WAAW,MAAM;CACxB;AACF"}
|