@rlanz/socket 0.0.1-3 → 0.0.1-5

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.
Files changed (55) hide show
  1. package/README.md +467 -41
  2. package/build/chunk-ANT2E3LT.js +136 -0
  3. package/build/chunk-ANT2E3LT.js.map +1 -0
  4. package/build/chunk-B4Y3TNDI.js +159 -0
  5. package/build/chunk-B4Y3TNDI.js.map +1 -0
  6. package/build/{chunk-GKAD2UOA.js → chunk-D3HUBCBW.js} +1 -1
  7. package/build/chunk-D3HUBCBW.js.map +1 -0
  8. package/build/chunk-SFAY2ZA4.js +28 -0
  9. package/build/chunk-SFAY2ZA4.js.map +1 -0
  10. package/build/{chunk-XUDFDJME.js → chunk-SHH6U4CI.js} +12 -9
  11. package/build/chunk-SHH6U4CI.js.map +1 -0
  12. package/build/chunk-YAX5EHHB.js +453 -0
  13. package/build/chunk-YAX5EHHB.js.map +1 -0
  14. package/build/framework-CvxgHwbv.d.ts +12 -0
  15. package/build/index.d.ts +11 -3
  16. package/build/index.js +32 -6
  17. package/build/index.js.map +1 -1
  18. package/build/providers/socket_provider.d.ts +4 -3
  19. package/build/providers/socket_provider.js +1774 -57
  20. package/build/providers/socket_provider.js.map +1 -1
  21. package/build/services/socket.d.ts +4 -3
  22. package/build/socket_service-mybqv1hj.d.ts +105 -0
  23. package/build/src/assembler_hook.d.ts +13 -0
  24. package/build/src/assembler_hook.js +249 -0
  25. package/build/src/assembler_hook.js.map +1 -0
  26. package/build/src/client/index.d.ts +46 -12
  27. package/build/src/client/index.js +223 -81
  28. package/build/src/client/index.js.map +1 -1
  29. package/build/src/client/react.d.ts +27 -0
  30. package/build/src/client/react.js +69 -0
  31. package/build/src/client/react.js.map +1 -0
  32. package/build/src/client/svelte.d.ts +22 -0
  33. package/build/src/client/svelte.js +82 -0
  34. package/build/src/client/svelte.js.map +1 -0
  35. package/build/src/client/types.d.ts +68 -10
  36. package/build/src/client/vue.d.ts +25 -0
  37. package/build/src/client/vue.js +81 -0
  38. package/build/src/client/vue.js.map +1 -0
  39. package/build/src/decorators.d.ts +4 -4
  40. package/build/src/decorators.js +1 -1
  41. package/build/src/health_check.d.ts +4 -3
  42. package/build/src/otel.js +4 -5
  43. package/build/src/otel.js.map +1 -1
  44. package/build/src/testing.d.ts +54 -0
  45. package/build/src/testing.js +7 -0
  46. package/build/src/testing.js.map +1 -0
  47. package/build/src/types.d.ts +2 -2
  48. package/build/{types-BnOi46bE.d.ts → types-C7Q36ryo.d.ts} +144 -79
  49. package/package.json +43 -8
  50. package/build/chunk-GKAD2UOA.js.map +0 -1
  51. package/build/chunk-NRWRLZAO.js +0 -1794
  52. package/build/chunk-NRWRLZAO.js.map +0 -1
  53. package/build/chunk-XUDFDJME.js.map +0 -1
  54. package/build/shared_types-Dw9AphfO.d.ts +0 -21
  55. package/build/socket_service-BN1g3ENk.d.ts +0 -144
@@ -1,1794 +0,0 @@
1
- import {
2
- getDecoratedChannelHandlers
3
- } from "./chunk-XUDFDJME.js";
4
- import {
5
- broadcastChannel,
6
- channelMessageChannel,
7
- connectChannel,
8
- disconnectChannel,
9
- subscribeChannel,
10
- unsubscribeChannel
11
- } from "./chunk-GKAD2UOA.js";
12
-
13
- // src/presence_manager.ts
14
- var PRESENCE_DATA_KEY = "__rlanzSocketPresence";
15
- var PresenceManager = class {
16
- #presence = /* @__PURE__ */ new Map();
17
- #fetchSockets = null;
18
- /**
19
- * Sets the socket fetcher used to build distributed snapshots.
20
- */
21
- setSocketFetcher(fetchSockets) {
22
- this.#fetchSockets = fetchSockets;
23
- }
24
- /**
25
- * Adds a user to a channel.
26
- */
27
- join(channel, socket, member) {
28
- if (!this.#presence.has(channel)) {
29
- this.#presence.set(channel, /* @__PURE__ */ new Map());
30
- }
31
- const presenceUser = typeof member === "string" ? { id: socket.id, name: member } : {
32
- ...member,
33
- id: member.id || socket.id
34
- };
35
- const user = {
36
- ...presenceUser,
37
- joinedAt: /* @__PURE__ */ new Date()
38
- };
39
- this.#presence.get(channel).set(socket.id, user);
40
- this.#setSocketPresence(socket, channel, user);
41
- }
42
- /**
43
- * Removes a user from a channel.
44
- */
45
- leave(channel, socket) {
46
- const channelPresence = this.#presence.get(channel);
47
- if (channelPresence) {
48
- channelPresence.delete(socket.id);
49
- if (channelPresence.size === 0) {
50
- this.#presence.delete(channel);
51
- }
52
- }
53
- this.#deleteSocketPresence(socket, channel);
54
- }
55
- /**
56
- * Removes a user from every channel.
57
- * Returns the list of channels the user left.
58
- */
59
- leaveAll(socket) {
60
- const leftChannels = [];
61
- for (const [channel, users] of this.#presence) {
62
- if (users.has(socket.id)) {
63
- users.delete(socket.id);
64
- leftChannels.push(channel);
65
- if (users.size === 0) {
66
- this.#presence.delete(channel);
67
- }
68
- this.#deleteSocketPresence(socket, channel);
69
- }
70
- }
71
- return leftChannels;
72
- }
73
- /**
74
- * Gets presence data for a channel.
75
- */
76
- async snapshot(channel) {
77
- if (this.#fetchSockets) {
78
- return this.#getFromSockets(channel);
79
- }
80
- return this.#localSnapshot(channel);
81
- }
82
- /**
83
- * Gets the number of users in a channel.
84
- */
85
- async count(channel) {
86
- return (await this.snapshot(channel)).count;
87
- }
88
- /**
89
- * Checks whether a socket is locally present in a channel.
90
- */
91
- hasLocal(channel, socketId) {
92
- return this.#presence.get(channel)?.has(socketId) || false;
93
- }
94
- /**
95
- * Lists every local channel where a socket is present.
96
- */
97
- getLocalChannelsForSocket(socketId) {
98
- const channels = [];
99
- for (const [channel, users] of this.#presence) {
100
- if (users.has(socketId)) {
101
- channels.push(channel);
102
- }
103
- }
104
- return channels;
105
- }
106
- /**
107
- * Lists sockets that are locally present in a channel.
108
- */
109
- getLocalSockets(channel) {
110
- const users = this.#presence.get(channel);
111
- if (!users) {
112
- return [];
113
- }
114
- return Array.from(users.entries()).map(([socketId, user]) => ({
115
- id: socketId,
116
- data: {
117
- [PRESENCE_DATA_KEY]: {
118
- [channel]: user
119
- }
120
- }
121
- }));
122
- }
123
- #localSnapshot(channel) {
124
- const users = this.#presence.get(channel);
125
- return {
126
- channel,
127
- users: users ? Array.from(users.values()) : [],
128
- count: users?.size || 0
129
- };
130
- }
131
- async #getFromSockets(channel) {
132
- const sockets = await this.#fetchSockets(channel);
133
- const users = sockets.map((socket) => socket.data?.[PRESENCE_DATA_KEY]?.[channel]).filter((user) => Boolean(user));
134
- return {
135
- channel,
136
- users,
137
- count: users.length
138
- };
139
- }
140
- #setSocketPresence(socket, channel, user) {
141
- socket.data[PRESENCE_DATA_KEY] ??= {};
142
- socket.data[PRESENCE_DATA_KEY][channel] = user;
143
- }
144
- #deleteSocketPresence(socket, channel) {
145
- if (!socket.data[PRESENCE_DATA_KEY]) {
146
- return;
147
- }
148
- delete socket.data[PRESENCE_DATA_KEY][channel];
149
- if (Object.keys(socket.data[PRESENCE_DATA_KEY]).length === 0) {
150
- delete socket.data[PRESENCE_DATA_KEY];
151
- }
152
- }
153
- };
154
-
155
- // src/base_channel.ts
156
- var MissingPresenceInfoError = class extends Error {
157
- constructor() {
158
- super("Presence channels must implement getPresenceInfo(socket)");
159
- }
160
- };
161
- var BaseChannel = class {
162
- /**
163
- * URL-like channel pattern, for example: 'chat/:roomId'.
164
- */
165
- static pattern;
166
- /**
167
- * Channel options.
168
- */
169
- static options;
170
- /**
171
- * Middlewares to run before subscription.
172
- */
173
- static middlewares;
174
- /**
175
- * Event handler map.
176
- * Ex: { 'chat:send': 'handleSend' }
177
- */
178
- handlers = {};
179
- /**
180
- * Injected socket service.
181
- */
182
- socket;
183
- /**
184
- * Full channel name, for example: 'chat/general'.
185
- */
186
- channelName;
187
- /**
188
- * Parameters extracted from the pattern, for example: { roomId: 'general' }.
189
- */
190
- params;
191
- /**
192
- * Injected presence manager when presence is enabled.
193
- */
194
- presence;
195
- /**
196
- * Injects dependencies.
197
- */
198
- $setContext(socket, channelName, params, presence) {
199
- this.socket = socket;
200
- this.channelName = channelName;
201
- this.params = params;
202
- if (presence) {
203
- this.presence = presence;
204
- }
205
- }
206
- /**
207
- * Emits to every channel member.
208
- */
209
- broadcast(event, data) {
210
- this.socket.to(this.channelName).emit(event, data);
211
- }
212
- /**
213
- * Emits to every channel member except one socket.
214
- */
215
- broadcastExcept(socketId, event, data) {
216
- this.socket.to(this.channelName).except(socketId).emit(event, data);
217
- }
218
- /**
219
- * Runs the middleware chain.
220
- *
221
- * middleware[0] -> middleware[1] -> ... -> subscribe
222
- */
223
- async $runMiddlewares(ctx) {
224
- const middlewares = this.constructor.middlewares || [];
225
- let index = 0;
226
- const next = async () => {
227
- if (index >= middlewares.length) return;
228
- const middleware = middlewares[index++];
229
- if (typeof middleware === "function") {
230
- await middleware(ctx, next);
231
- } else {
232
- await middleware.handle(ctx, next);
233
- }
234
- };
235
- await next();
236
- }
237
- /**
238
- * Dispatches a message to the matching handler.
239
- * Called by SocketService. Do not override.
240
- */
241
- async $handleMessage(socket, event, data) {
242
- const handlers = {
243
- ...this.handlers,
244
- ...getDecoratedChannelHandlers(this)
245
- };
246
- const handler = handlers[event];
247
- if (typeof handler === "function") {
248
- return handler.call(this, socket, data);
249
- }
250
- return this.onMessage?.(socket, event, data);
251
- }
252
- /**
253
- * Runs the generic join hook. ChannelSubscriptions owns subscription lifecycle.
254
- */
255
- async $handleJoin(socket, ...params) {
256
- await this.onJoin?.(socket, ...params);
257
- }
258
- /**
259
- * Runs the generic leave hook. ChannelSubscriptions owns subscription lifecycle.
260
- */
261
- async $handleLeave(socket, ...params) {
262
- await this.onLeave?.(socket, ...params);
263
- }
264
- $getPresenceInfo(socket) {
265
- if (!this.getPresenceInfo) {
266
- throw new MissingPresenceInfoError();
267
- }
268
- return this.getPresenceInfo(socket);
269
- }
270
- };
271
-
272
- // src/fake_socket.ts
273
- import { AssertionError } from "assert";
274
- import { isDeepStrictEqual } from "util";
275
- var SocketFake = class {
276
- #emissions = [];
277
- #restore;
278
- constructor(restore) {
279
- this.#restore = restore;
280
- }
281
- get emissions() {
282
- return [...this.#emissions];
283
- }
284
- record(emission) {
285
- this.#emissions.push({
286
- ...emission,
287
- ..."except" in emission && emission.except ? { except: [...emission.except] } : {}
288
- });
289
- }
290
- dispatch(emission) {
291
- this.record(emission);
292
- }
293
- clear() {
294
- this.#emissions = [];
295
- }
296
- restore() {
297
- this.#restore();
298
- }
299
- [Symbol.dispose]() {
300
- this.restore();
301
- }
302
- assertBroadcasted(event, query) {
303
- this.#assertEmitted(
304
- (emission) => emission.target === "global" && emission.event === event,
305
- query,
306
- `Expected socket event "${event}" to have been broadcasted`
307
- );
308
- }
309
- assertNotBroadcasted(event, query) {
310
- this.#assertNotEmitted(
311
- (emission) => emission.target === "global" && emission.event === event,
312
- query,
313
- `Expected socket event "${event}" not to have been broadcasted`
314
- );
315
- }
316
- assertEmittedTo(channel, event, query) {
317
- this.#assertEmitted(
318
- (emission) => emission.target === "channel" && emission.channel === channel && emission.event === event,
319
- query,
320
- `Expected socket event "${event}" to have been emitted to channel "${channel}"`
321
- );
322
- }
323
- assertNotEmittedTo(channel, event, query) {
324
- this.#assertNotEmitted(
325
- (emission) => emission.target === "channel" && emission.channel === channel && emission.event === event,
326
- query,
327
- `Expected socket event "${event}" not to have been emitted to channel "${channel}"`
328
- );
329
- }
330
- assertEmittedToUser(userId, event, query) {
331
- this.#assertEmitted(
332
- (emission) => emission.target === "user" && emission.userId === userId && emission.event === event,
333
- query,
334
- `Expected socket event "${event}" to have been emitted to user "${String(userId)}"`
335
- );
336
- }
337
- assertNotEmittedToUser(userId, event, query) {
338
- this.#assertNotEmitted(
339
- (emission) => emission.target === "user" && emission.userId === userId && emission.event === event,
340
- query,
341
- `Expected socket event "${event}" not to have been emitted to user "${String(userId)}"`
342
- );
343
- }
344
- assertEmittedCount(count, query) {
345
- const actual = this.#matching(query).length;
346
- if (actual !== count) {
347
- throw new AssertionError({
348
- message: `Expected ${count} socket event(s) to have been emitted, but found ${actual}`,
349
- actual,
350
- expected: count,
351
- operator: "strictEqual"
352
- });
353
- }
354
- }
355
- assertNothingEmitted() {
356
- this.assertEmittedCount(0);
357
- }
358
- #assertEmitted(predicate, query, message) {
359
- const matching = this.#emissions.filter((emission) => {
360
- return predicate(emission) && this.#matchesQuery(emission, query);
361
- });
362
- if (matching.length === 0) {
363
- throw new AssertionError({
364
- message,
365
- actual: this.#emissions,
366
- expected: query,
367
- operator: "includes"
368
- });
369
- }
370
- }
371
- #assertNotEmitted(predicate, query, message) {
372
- const matching = this.#emissions.filter((emission) => {
373
- return predicate(emission) && this.#matchesQuery(emission, query);
374
- });
375
- if (matching.length > 0) {
376
- throw new AssertionError({
377
- message,
378
- actual: matching,
379
- expected: [],
380
- operator: "doesNotInclude"
381
- });
382
- }
383
- }
384
- #matching(query) {
385
- return this.#emissions.filter((emission) => {
386
- if (query?.target && emission.target !== query.target) {
387
- return false;
388
- }
389
- if (query?.event && emission.event !== query.event) {
390
- return false;
391
- }
392
- if (query?.channel && (!("channel" in emission) || emission.channel !== query.channel)) {
393
- return false;
394
- }
395
- if (query?.userId !== void 0 && (!("userId" in emission) || emission.userId !== query.userId)) {
396
- return false;
397
- }
398
- return this.#matchesQuery(emission, query);
399
- });
400
- }
401
- #matchesQuery(emission, query) {
402
- if (!query) {
403
- return true;
404
- }
405
- if ("data" in query) {
406
- if (typeof query.data === "function") {
407
- if (!query.data(emission.data)) {
408
- return false;
409
- }
410
- } else if (!isDeepStrictEqual(emission.data, query.data)) {
411
- return false;
412
- }
413
- }
414
- if ("except" in query) {
415
- if (typeof query.except === "function") {
416
- if (!query.except("except" in emission ? emission.except : void 0)) {
417
- return false;
418
- }
419
- } else if (!isDeepStrictEqual(("except" in emission ? emission.except : void 0) ?? [], query.except)) {
420
- return false;
421
- }
422
- }
423
- return true;
424
- }
425
- };
426
-
427
- // src/socket_service.ts
428
- import { randomUUID as randomUUID2 } from "crypto";
429
- import Emittery from "emittery";
430
- import { WebSocket, WebSocketServer } from "ws";
431
-
432
- // src/channel_subscription_storage.ts
433
- var ChannelSubscriptionStorage = class {
434
- #subscriptions = /* @__PURE__ */ new Map();
435
- #subscribers = /* @__PURE__ */ new Map();
436
- has(socketId, channelName) {
437
- return this.#subscriptions.get(socketId)?.has(channelName) ?? false;
438
- }
439
- get(socketId, channelName) {
440
- return this.#subscriptions.get(socketId)?.get(channelName);
441
- }
442
- set(socketId, channelName, subscription) {
443
- if (!this.#subscriptions.has(socketId)) {
444
- this.#subscriptions.set(socketId, /* @__PURE__ */ new Map());
445
- }
446
- this.#subscriptions.get(socketId).set(channelName, subscription);
447
- if (!this.#subscribers.has(channelName)) {
448
- this.#subscribers.set(channelName, /* @__PURE__ */ new Set());
449
- }
450
- this.#subscribers.get(channelName).add(socketId);
451
- }
452
- delete(socketId, channelName) {
453
- const subscriptions = this.#subscriptions.get(socketId);
454
- if (!subscriptions) {
455
- return;
456
- }
457
- subscriptions.delete(channelName);
458
- if (subscriptions.size === 0) {
459
- this.#subscriptions.delete(socketId);
460
- }
461
- const subscribers = this.#subscribers.get(channelName);
462
- subscribers?.delete(socketId);
463
- if (subscribers?.size === 0) {
464
- this.#subscribers.delete(channelName);
465
- }
466
- }
467
- deleteSocket(socketId) {
468
- for (const channelName of this.channelNamesFor(socketId)) {
469
- this.delete(socketId, channelName);
470
- }
471
- }
472
- channelNamesFor(socketId) {
473
- return [...this.#subscriptions.get(socketId)?.keys() ?? []];
474
- }
475
- socketIdsFor(channelName) {
476
- return [...this.#subscribers.get(channelName) ?? []];
477
- }
478
- get channelsCount() {
479
- return this.#subscribers.size;
480
- }
481
- clear() {
482
- this.#subscriptions.clear();
483
- this.#subscribers.clear();
484
- }
485
- };
486
-
487
- // src/channel_subscriptions.ts
488
- var ChannelSubscriptions = class {
489
- constructor(socketService, channelRouter, logger, presenceManager) {
490
- this.socketService = socketService;
491
- this.channelRouter = channelRouter;
492
- this.logger = logger;
493
- this.presenceManager = presenceManager;
494
- }
495
- #storage = new ChannelSubscriptionStorage();
496
- setPresenceManager(presenceManager) {
497
- this.presenceManager = presenceManager;
498
- }
499
- async subscribe(socket, channelName) {
500
- if (this.#storage.has(socket.id, channelName)) {
501
- return {
502
- ack: {
503
- ok: true,
504
- presenceData: await this.#getPresenceData(socket, channelName)
505
- },
506
- created: false
507
- };
508
- }
509
- const result = await this.channelRouter.authorize(socket, channelName);
510
- if (!result.success) {
511
- return {
512
- ack: { ok: false, error: result.error },
513
- created: false
514
- };
515
- }
516
- result.instance.$setContext(
517
- this.socketService,
518
- channelName,
519
- result.params,
520
- this.presenceManager
521
- );
522
- let presenceData = result.presenceData;
523
- let presenceJoined = false;
524
- try {
525
- if (this.#usesPresence(result.instance)) {
526
- const member = this.#getPresenceMember(result.instance, socket);
527
- this.presenceManager.join(channelName, socket.raw, member);
528
- presenceJoined = true;
529
- await result.instance.onMemberJoin?.(socket, member, ...result.paramValues);
530
- await result.instance.onJoin?.(socket, ...result.paramValues);
531
- presenceData = await this.presenceManager.snapshot(channelName);
532
- } else {
533
- await result.instance.onJoin?.(socket, ...result.paramValues);
534
- }
535
- this.#storage.set(socket.id, channelName, {
536
- channelName,
537
- instance: result.instance,
538
- paramValues: result.paramValues
539
- });
540
- socket.channels.add(channelName);
541
- if (this.#usesPresence(result.instance) && presenceData) {
542
- this.socketService.to(channelName).except(socket.id).emit("presence:update", presenceData);
543
- }
544
- } catch (error) {
545
- if (presenceJoined) {
546
- this.presenceManager.leave(channelName, socket.raw);
547
- const rollbackData = await this.presenceManager.snapshot(channelName).catch(() => null);
548
- if (rollbackData) {
549
- this.socketService.to(channelName).emit("presence:update", rollbackData);
550
- }
551
- }
552
- this.#storage.delete(socket.id, channelName);
553
- socket.channels.delete(channelName);
554
- return {
555
- ack: {
556
- ok: false,
557
- error: error instanceof MissingPresenceInfoError ? error.message : "Join failed"
558
- },
559
- created: false
560
- };
561
- }
562
- return {
563
- ack: { ok: true, presenceData },
564
- created: true
565
- };
566
- }
567
- async leave(socket, channelName) {
568
- const subscription = this.#storage.get(socket.id, channelName);
569
- if (subscription) {
570
- this.#storage.delete(socket.id, channelName);
571
- socket.channels.delete(channelName);
572
- try {
573
- if (this.#usesPresence(subscription.instance)) {
574
- const member = this.#getPresenceMember(subscription.instance, socket);
575
- this.presenceManager.leave(channelName, socket.raw);
576
- const presenceData = await this.presenceManager.snapshot(channelName);
577
- this.socketService.to(channelName).emit("presence:update", presenceData);
578
- await subscription.instance.onMemberLeave?.(socket, member, ...subscription.paramValues);
579
- await subscription.instance.onLeave?.(socket, ...subscription.paramValues);
580
- } else {
581
- await subscription.instance.onLeave?.(socket, ...subscription.paramValues);
582
- }
583
- } catch (error) {
584
- this.logger.warn(`failed to leave channel "${channelName}": %s`, error);
585
- }
586
- } else {
587
- socket.channels.delete(channelName);
588
- }
589
- }
590
- async leaveAll(socket) {
591
- for (const channelName of this.getChannelNames(socket.id)) {
592
- await this.leave(socket, channelName);
593
- }
594
- }
595
- async handleMessage(socket, payload) {
596
- const subscription = this.#storage.get(socket.id, payload.channel);
597
- if (!subscription) {
598
- return { ok: false, error: "Not subscribed" };
599
- }
600
- try {
601
- const result = await subscription.instance.$handleMessage(socket, payload.event, payload.data);
602
- return { ok: true, data: result };
603
- } catch (error) {
604
- return {
605
- ok: false,
606
- error: error instanceof Error ? error.message : "Handler error"
607
- };
608
- }
609
- }
610
- relayWhisper(socket, payload) {
611
- if (!this.#storage.has(socket.id, payload.channel)) {
612
- return { ok: false, error: "Not subscribed" };
613
- }
614
- this.socketService.to(payload.channel).except(socket.id).emit(`client:${payload.event}`, payload.data);
615
- return { ok: true };
616
- }
617
- deleteSocket(socketId) {
618
- this.#storage.deleteSocket(socketId);
619
- }
620
- clear() {
621
- this.#storage.clear();
622
- }
623
- getSocketIds(channelName) {
624
- return this.#storage.socketIdsFor(channelName);
625
- }
626
- getChannelNames(socketId) {
627
- return this.#storage.channelNamesFor(socketId);
628
- }
629
- get channelsCount() {
630
- return this.#storage.channelsCount;
631
- }
632
- #usesPresence(instance) {
633
- return Boolean(
634
- instance.constructor.options?.presence && this.presenceManager
635
- );
636
- }
637
- #getPresenceMember(instance, socket) {
638
- return instance.$getPresenceInfo(socket);
639
- }
640
- async #getPresenceData(socket, channelName) {
641
- if (!this.presenceManager?.hasLocal(channelName, socket.id)) {
642
- return void 0;
643
- }
644
- return this.presenceManager.snapshot(channelName);
645
- }
646
- };
647
-
648
- // src/protocol/reply.ts
649
- var INVALID_SOCKET_MESSAGE = "Invalid socket message";
650
- var Reply = class _Reply {
651
- constructor(id, ok, data, error) {
652
- this.id = id;
653
- this.ok = ok;
654
- this.data = data;
655
- this.error = error;
656
- }
657
- static ok(id, data) {
658
- return new _Reply(id, true, data, void 0);
659
- }
660
- static error(id, error) {
661
- return new _Reply(id, false, void 0, error);
662
- }
663
- static invalidMessage(id) {
664
- if (id === void 0) {
665
- return { type: "error", error: INVALID_SOCKET_MESSAGE };
666
- }
667
- return _Reply.error(id, INVALID_SOCKET_MESSAGE).toFrame();
668
- }
669
- static fromSubscribeResult(id, result) {
670
- if (!result) {
671
- return _Reply.error(id, "Socket service is not initialized");
672
- }
673
- if (!result.ack.ok) {
674
- return _Reply.error(id, result.ack.error ?? "Subscribe failed");
675
- }
676
- const data = result.ack.presenceData ? {
677
- presenceData: result.ack.presenceData
678
- } : void 0;
679
- return _Reply.ok(id, data);
680
- }
681
- static fromChannelAck(id, ack) {
682
- return ack.ok ? _Reply.ok(id, ack.data) : _Reply.error(id, ack.error ?? "Handler error");
683
- }
684
- toFrame() {
685
- const frame = {
686
- id: this.id,
687
- type: "ack",
688
- ok: this.ok
689
- };
690
- if (this.data !== void 0) {
691
- frame.data = this.data;
692
- }
693
- if (this.error !== void 0) {
694
- frame.error = this.error;
695
- }
696
- return frame;
697
- }
698
- };
699
-
700
- // src/protocol/message.ts
701
- function isRecord(value) {
702
- return typeof value === "object" && value !== null && !Array.isArray(value);
703
- }
704
- var Message = class _Message {
705
- static fromTransport(raw) {
706
- try {
707
- return _Message.fromParsedTransport(JSON.parse(raw));
708
- } catch {
709
- return new InvalidMessage(void 0);
710
- }
711
- }
712
- static fromParsedTransport(value) {
713
- if (!isRecord(value)) {
714
- return new InvalidMessage(void 0);
715
- }
716
- const id = typeof value.id === "string" ? value.id : void 0;
717
- if ("id" in value && typeof value.id !== "string") {
718
- return new InvalidMessage(void 0);
719
- }
720
- if (!isClientMessageType(value.type)) {
721
- return new InvalidMessage(id);
722
- }
723
- if (value.type === "ping") {
724
- return new PingMessage(id);
725
- }
726
- if (value.type === "subscribe" && typeof value.channel === "string") {
727
- return new SubscribeMessage(id, value.channel);
728
- }
729
- if (value.type === "unsubscribe" && typeof value.channel === "string") {
730
- return new UnsubscribeMessage(id, value.channel);
731
- }
732
- if ((value.type === "message" || value.type === "whisper") && typeof value.channel === "string" && typeof value.event === "string") {
733
- return new ChannelProtocolMessage(id, value.type, value.channel, value.event, value.data);
734
- }
735
- return new InvalidMessage(id);
736
- }
737
- };
738
- var InvalidMessage = class extends Message {
739
- constructor(id) {
740
- super();
741
- this.id = id;
742
- }
743
- valid = false;
744
- type = "invalid";
745
- toRejectionFrame() {
746
- return Reply.invalidMessage(this.id);
747
- }
748
- };
749
- var PingMessage = class extends Message {
750
- constructor(id) {
751
- super();
752
- this.id = id;
753
- }
754
- valid = true;
755
- type = "ping";
756
- };
757
- var SubscribeMessage = class extends Message {
758
- constructor(id, channel) {
759
- super();
760
- this.id = id;
761
- this.channel = channel;
762
- }
763
- valid = true;
764
- type = "subscribe";
765
- };
766
- var UnsubscribeMessage = class extends Message {
767
- constructor(id, channel) {
768
- super();
769
- this.id = id;
770
- this.channel = channel;
771
- }
772
- valid = true;
773
- type = "unsubscribe";
774
- };
775
- var ChannelProtocolMessage = class extends Message {
776
- constructor(id, type, channel, event, data) {
777
- super();
778
- this.id = id;
779
- this.type = type;
780
- this.channel = channel;
781
- this.event = event;
782
- this.data = data;
783
- }
784
- valid = true;
785
- toChannelMessage() {
786
- return {
787
- channel: this.channel,
788
- event: this.event,
789
- data: this.data
790
- };
791
- }
792
- };
793
- function isClientMessageType(value) {
794
- return value === "ping" || value === "subscribe" || value === "unsubscribe" || value === "message" || value === "whisper";
795
- }
796
-
797
- // src/message_queue.ts
798
- var MessageQueue = class {
799
- #queues = /* @__PURE__ */ new Map();
800
- enqueue(key, task) {
801
- const previous = this.#queues.get(key) ?? Promise.resolve();
802
- const next = previous.catch(() => {
803
- }).then(task).catch(() => {
804
- }).finally(() => {
805
- if (this.#queues.get(key) === next) {
806
- this.#queues.delete(key);
807
- }
808
- });
809
- this.#queues.set(key, next);
810
- }
811
- delete(key) {
812
- this.#queues.delete(key);
813
- }
814
- clear() {
815
- this.#queues.clear();
816
- }
817
- };
818
-
819
- // src/broadcasting.ts
820
- function userRoom(userId) {
821
- return `socket:user:${String(userId)}`;
822
- }
823
-
824
- // src/socket_bus.ts
825
- import { randomUUID } from "crypto";
826
- import { Bus } from "@boringnode/bus";
827
- var DEFAULT_CHANNEL = "socket::broadcast";
828
- var DEFAULT_PRESENCE_TIMEOUT = 100;
829
- var SocketBus = class {
830
- constructor(transport, handlers) {
831
- this.handlers = handlers;
832
- this.#channel = transport.channel ?? DEFAULT_CHANNEL;
833
- this.#presenceTimeout = transport.presenceTimeout ?? DEFAULT_PRESENCE_TIMEOUT;
834
- this.#bus = new Bus(transport.driver(), { retryQueue: { enabled: true } });
835
- }
836
- #origin = randomUUID();
837
- #channel;
838
- #bus;
839
- #presenceTimeout;
840
- #pendingPresenceRequests = /* @__PURE__ */ new Map();
841
- async start() {
842
- await this.#bus.subscribe(this.#channel, async (payload) => {
843
- const message = payload;
844
- if (message.origin === this.#origin) {
845
- return;
846
- }
847
- switch (message.type) {
848
- case "channel:event":
849
- this.handlers.channel(message);
850
- return;
851
- case "user:event":
852
- this.handlers.user(message);
853
- return;
854
- case "broadcast:event":
855
- this.handlers.broadcast(message);
856
- return;
857
- case "presence:sockets:request":
858
- await this.#respondToPresenceRequest(message);
859
- return;
860
- case "presence:sockets:response":
861
- this.#handlePresenceResponse(message);
862
- return;
863
- }
864
- });
865
- }
866
- publishChannel(channel, event, data, except) {
867
- const message = {
868
- type: "channel:event",
869
- origin: this.#origin,
870
- channel,
871
- event,
872
- data
873
- };
874
- if (except) {
875
- message.except = except;
876
- }
877
- void this.#bus.publish(this.#channel, message);
878
- }
879
- publishUser(room, event, data) {
880
- void this.#bus.publish(this.#channel, {
881
- type: "user:event",
882
- origin: this.#origin,
883
- room,
884
- event,
885
- data
886
- });
887
- }
888
- publishBroadcast(event, data) {
889
- void this.#bus.publish(this.#channel, {
890
- type: "broadcast:event",
891
- origin: this.#origin,
892
- event,
893
- data
894
- });
895
- }
896
- fetchPresenceSockets(channel) {
897
- const requestId = randomUUID();
898
- return new Promise((resolve) => {
899
- const timeout = setTimeout(() => {
900
- const request = this.#pendingPresenceRequests.get(requestId);
901
- if (!request) {
902
- return;
903
- }
904
- this.#pendingPresenceRequests.delete(requestId);
905
- resolve(request.sockets.map((socket) => this.#deserializePresenceSocket(socket)));
906
- }, this.#presenceTimeout);
907
- this.#pendingPresenceRequests.set(requestId, {
908
- sockets: [],
909
- resolve,
910
- timeout
911
- });
912
- void this.#bus.publish(this.#channel, {
913
- type: "presence:sockets:request",
914
- origin: this.#origin,
915
- requestId,
916
- channel
917
- });
918
- });
919
- }
920
- async close() {
921
- for (const [requestId, request] of this.#pendingPresenceRequests) {
922
- clearTimeout(request.timeout);
923
- request.resolve(request.sockets.map((socket) => this.#deserializePresenceSocket(socket)));
924
- this.#pendingPresenceRequests.delete(requestId);
925
- }
926
- await this.#bus.disconnect();
927
- }
928
- async #respondToPresenceRequest(message) {
929
- await this.#bus.publish(this.#channel, {
930
- type: "presence:sockets:response",
931
- origin: this.#origin,
932
- target: message.origin,
933
- requestId: message.requestId,
934
- sockets: this.handlers.presenceSockets(message.channel)
935
- });
936
- }
937
- #handlePresenceResponse(message) {
938
- if (message.target !== this.#origin) {
939
- return;
940
- }
941
- const request = this.#pendingPresenceRequests.get(message.requestId);
942
- if (!request) {
943
- return;
944
- }
945
- request.sockets.push(...message.sockets);
946
- }
947
- #deserializePresenceSocket(socket) {
948
- const presence = socket.data[PRESENCE_DATA_KEY];
949
- return {
950
- id: socket.id,
951
- data: {
952
- [PRESENCE_DATA_KEY]: presence ? Object.fromEntries(
953
- Object.entries(presence).map(([channel, user]) => [
954
- channel,
955
- {
956
- ...user,
957
- joinedAt: new Date(user.joinedAt)
958
- }
959
- ])
960
- ) : void 0
961
- }
962
- };
963
- }
964
- };
965
-
966
- // src/socket_upgrader.ts
967
- var DEFAULT_WEBSOCKET_PATH = "/socket";
968
- var SocketUpgrader = class _SocketUpgrader {
969
- constructor(server, config) {
970
- this.server = server;
971
- this.config = config;
972
- this.#path = config?.path ?? DEFAULT_WEBSOCKET_PATH;
973
- }
974
- #path;
975
- async handle(request, socket, head, accept) {
976
- if (!this.#matchesPath(request)) {
977
- return false;
978
- }
979
- const accepted = await this.#authenticate(request);
980
- if (!accepted) {
981
- _SocketUpgrader.reject(socket, 401, "Unauthorized");
982
- return true;
983
- }
984
- this.server.handleUpgrade(request, socket, head, (connection) => {
985
- accept(connection, request, accepted);
986
- });
987
- return true;
988
- }
989
- static reject(socket, statusCode, reason) {
990
- if (!socket.writableEnded) {
991
- socket.write(`HTTP/1.1 ${statusCode} ${reason}\r
992
- \r
993
- `);
994
- }
995
- socket.destroy();
996
- }
997
- #matchesPath(request) {
998
- const url = new URL(request.url ?? "/", "ws://localhost");
999
- return url.pathname === this.#path;
1000
- }
1001
- async #authenticate(request) {
1002
- const auth = this.#parseAuth(request);
1003
- const config = this.config;
1004
- if (!config) {
1005
- return { auth };
1006
- }
1007
- if (!config.authenticate && !config.createContextOnUpgrade) {
1008
- return { auth };
1009
- }
1010
- let httpContext;
1011
- try {
1012
- httpContext = await config.createContext?.(request);
1013
- if (!config.authenticate) {
1014
- return { auth, httpContext };
1015
- }
1016
- const ctx = {
1017
- request,
1018
- auth,
1019
- httpContext
1020
- };
1021
- const result = await config.authenticate(ctx);
1022
- if (result === false || result === null) {
1023
- return null;
1024
- }
1025
- return {
1026
- auth: result.auth ?? auth,
1027
- data: result.data,
1028
- httpContext,
1029
- user: result.user
1030
- };
1031
- } catch {
1032
- return null;
1033
- }
1034
- }
1035
- #parseAuth(request) {
1036
- const url = new URL(request.url ?? "/", "ws://localhost");
1037
- const auth = url.searchParams.get("auth");
1038
- if (!auth) {
1039
- return void 0;
1040
- }
1041
- try {
1042
- const parsed = JSON.parse(auth);
1043
- return parsed && typeof parsed === "object" ? parsed : void 0;
1044
- } catch {
1045
- return void 0;
1046
- }
1047
- }
1048
- };
1049
-
1050
- // src/socket_service.ts
1051
- var DEFAULT_PING_INTERVAL = 25e3;
1052
- var DEFAULT_PING_TIMEOUT = 5e3;
1053
- function sendFrame(connection, frame) {
1054
- if (connection.readyState !== WebSocket.OPEN) {
1055
- return;
1056
- }
1057
- connection.send(JSON.stringify(frame));
1058
- }
1059
- function validateDuration(name, value) {
1060
- if (value === void 0) {
1061
- return void 0;
1062
- }
1063
- if (!Number.isFinite(value) || value <= 0) {
1064
- throw new Error(`websocket.${name} must be a positive number of milliseconds`);
1065
- }
1066
- return value;
1067
- }
1068
- function resolveHeartbeatConfig(config) {
1069
- const pingInterval = validateDuration("pingInterval", config?.pingInterval);
1070
- const pingTimeout = validateDuration("pingTimeout", config?.pingTimeout);
1071
- if (pingInterval === void 0 && pingTimeout === void 0) {
1072
- return null;
1073
- }
1074
- return {
1075
- interval: pingInterval ?? DEFAULT_PING_INTERVAL,
1076
- timeout: pingTimeout ?? DEFAULT_PING_TIMEOUT
1077
- };
1078
- }
1079
- var SocketService = class extends Emittery {
1080
- #server = null;
1081
- #httpServer = null;
1082
- #upgradeHandler = null;
1083
- #sockets = /* @__PURE__ */ new Map();
1084
- #userRooms = /* @__PURE__ */ new Map();
1085
- #socketUserRoom = /* @__PURE__ */ new Map();
1086
- #channelSubscriptions = null;
1087
- #presenceManager = null;
1088
- #bus = null;
1089
- #fake = null;
1090
- #deliverySink = {
1091
- dispatch: (emission) => {
1092
- this.#deliverBroadcast(emission);
1093
- }
1094
- };
1095
- #broadcastSink = this.#deliverySink;
1096
- #status = "stopped";
1097
- #lastError = null;
1098
- #heartbeatInterval = null;
1099
- #heartbeatTimeouts = /* @__PURE__ */ new Map();
1100
- #messageQueue = new MessageQueue();
1101
- /**
1102
- * Accesses the underlying ws server.
1103
- */
1104
- get server() {
1105
- if (!this.#server) {
1106
- throw new Error("WebSocket server is not initialized. Call boot() first.");
1107
- }
1108
- return this.#server;
1109
- }
1110
- setPresenceManager(presenceManager) {
1111
- this.#presenceManager = presenceManager;
1112
- this.#channelSubscriptions?.setPresenceManager(presenceManager);
1113
- this.#configurePresenceFetcher();
1114
- }
1115
- async boot(httpServer, config, channelRouter, logger) {
1116
- this.#status = "starting";
1117
- this.#lastError = null;
1118
- this.#channelSubscriptions = new ChannelSubscriptions(
1119
- this,
1120
- channelRouter,
1121
- logger,
1122
- this.#presenceManager ?? void 0
1123
- );
1124
- const websocketConfig = config.websocket;
1125
- const heartbeatConfig = resolveHeartbeatConfig(websocketConfig);
1126
- try {
1127
- if (config.transport) {
1128
- this.#bus = new SocketBus(config.transport, {
1129
- channel: (message) => {
1130
- this.#traceBroadcast(
1131
- {
1132
- target: "channel",
1133
- channel: message.channel,
1134
- event: message.event,
1135
- via: "bus",
1136
- except: message.except
1137
- },
1138
- () => this.#emitToChannelLocally(
1139
- message.channel,
1140
- message.event,
1141
- message.data,
1142
- message.except
1143
- )
1144
- );
1145
- },
1146
- user: (message) => {
1147
- this.#traceBroadcast(
1148
- { target: "user", room: message.room, event: message.event, via: "bus" },
1149
- () => this.#emitToUserRoomLocally(message.room, message.event, message.data)
1150
- );
1151
- },
1152
- broadcast: (message) => {
1153
- this.#traceBroadcast(
1154
- { target: "global", event: message.event, via: "bus" },
1155
- () => this.#broadcastLocally(message.event, message.data)
1156
- );
1157
- },
1158
- presenceSockets: (channel) => {
1159
- return this.#getLocalPresenceSockets(channel);
1160
- }
1161
- });
1162
- await this.#bus.start();
1163
- this.#configurePresenceFetcher();
1164
- }
1165
- const websocketServer = new WebSocketServer({
1166
- noServer: true
1167
- });
1168
- this.#server = websocketServer;
1169
- this.#httpServer = httpServer;
1170
- const upgrader = new SocketUpgrader(websocketServer, websocketConfig);
1171
- this.#upgradeHandler = (request, socket, head) => {
1172
- upgrader.handle(request, socket, head, (connection, upgradeRequest, accepted) => {
1173
- this.#handleConnection(connection, upgradeRequest, accepted);
1174
- }).catch((error) => {
1175
- logger.warn("failed to upgrade socket connection: %s", error);
1176
- SocketUpgrader.reject(socket, 500, "Internal Server Error");
1177
- });
1178
- };
1179
- httpServer.on("upgrade", this.#upgradeHandler);
1180
- this.#startHeartbeat(heartbeatConfig);
1181
- this.#status = "ready";
1182
- } catch (error) {
1183
- this.#lastError = error instanceof Error ? error : new Error("Socket service boot failed");
1184
- this.#status = "failed";
1185
- await this.close();
1186
- throw error;
1187
- }
1188
- }
1189
- #handleConnection(connection, request, upgrade) {
1190
- const socket = this.#wrapSocket(connection, request, upgrade);
1191
- connectChannel.traceSync(
1192
- () => {
1193
- this.#sockets.set(socket.id, socket);
1194
- this.emit("connect", { socket });
1195
- },
1196
- { socketId: socket.id }
1197
- );
1198
- connection.on("message", (data) => {
1199
- const message = Message.fromTransport(data.toString());
1200
- if (message.type === "ping") {
1201
- sendFrame(socket.raw.connection, { id: message.id, type: "pong" });
1202
- return;
1203
- }
1204
- this.#messageQueue.enqueue(socket.id, async () => {
1205
- await this.#handleMessage(socket, message).catch((error) => {
1206
- sendFrame(connection, {
1207
- type: "error",
1208
- error: error instanceof Error ? error.message : "Unexpected socket error"
1209
- });
1210
- });
1211
- });
1212
- });
1213
- connection.on("pong", () => {
1214
- this.#clearHeartbeatTimeout(socket.id);
1215
- });
1216
- connection.on("close", async () => {
1217
- const traceMessage = {
1218
- socketId: socket.id,
1219
- reason: "close",
1220
- subscriptions: socket.channels.size
1221
- };
1222
- await disconnectChannel.tracePromise(async () => {
1223
- this.#clearHeartbeatTimeout(socket.id);
1224
- this.#messageQueue.delete(socket.id);
1225
- await this.#channelSubscriptions?.leaveAll(socket);
1226
- this.#channelSubscriptions?.deleteSocket(socket.id);
1227
- this.#sockets.delete(socket.id);
1228
- this.#leaveUserRoom(socket.id);
1229
- this.emit("disconnect", { socket, reason: "close" });
1230
- }, traceMessage);
1231
- });
1232
- }
1233
- async #handleMessage(socket, message) {
1234
- if (!message.valid) {
1235
- sendFrame(socket.raw.connection, message.toRejectionFrame());
1236
- return;
1237
- }
1238
- switch (message.type) {
1239
- case "ping":
1240
- return;
1241
- case "subscribe": {
1242
- const traceMessage = {
1243
- socketId: socket.id,
1244
- channel: message.channel
1245
- };
1246
- await subscribeChannel.tracePromise(async () => {
1247
- const result = await this.#channelSubscriptions?.subscribe(socket, message.channel);
1248
- traceMessage.created = result?.created ?? false;
1249
- traceMessage.ok = result?.ack.ok ?? false;
1250
- if (!result?.ack.ok) {
1251
- traceMessage.error = result?.ack.error ?? "Socket service is not initialized";
1252
- }
1253
- if (result?.created) {
1254
- this.emit("subscribe", { socket, channel: message.channel });
1255
- }
1256
- sendFrame(socket.raw.connection, Reply.fromSubscribeResult(message.id, result).toFrame());
1257
- }, traceMessage);
1258
- return;
1259
- }
1260
- case "unsubscribe": {
1261
- const traceMessage = {
1262
- socketId: socket.id,
1263
- channel: message.channel
1264
- };
1265
- await unsubscribeChannel.tracePromise(async () => {
1266
- traceMessage.removed = socket.channels.has(message.channel);
1267
- await this.#channelSubscriptions?.leave(socket, message.channel);
1268
- traceMessage.ok = true;
1269
- this.emit("unsubscribe", { socket, channel: message.channel });
1270
- sendFrame(socket.raw.connection, Reply.ok(message.id).toFrame());
1271
- }, traceMessage);
1272
- return;
1273
- }
1274
- case "message":
1275
- case "whisper": {
1276
- await this.#handleChannelFrame(socket, message);
1277
- return;
1278
- }
1279
- }
1280
- }
1281
- async #handleChannelFrame(socket, message) {
1282
- const payload = message.toChannelMessage();
1283
- const traceMessage = {
1284
- socketId: socket.id,
1285
- channel: payload.channel,
1286
- event: payload.event
1287
- };
1288
- await channelMessageChannel.tracePromise(async () => {
1289
- const ack = await this.#handleChannelPayload(socket, message.type, payload);
1290
- traceMessage.ok = ack.ok;
1291
- if (!ack.ok) {
1292
- traceMessage.error = ack.error ?? "Handler error";
1293
- }
1294
- sendFrame(socket.raw.connection, Reply.fromChannelAck(message.id, ack).toFrame());
1295
- }, traceMessage);
1296
- }
1297
- async #handleChannelPayload(socket, type, payload) {
1298
- if (typeof payload.channel !== "string" || typeof payload.event !== "string") {
1299
- return { ok: false, error: `Invalid channel ${type}` };
1300
- }
1301
- if (type === "message") {
1302
- return this.#channelSubscriptions.handleMessage(socket, payload);
1303
- }
1304
- return this.#channelSubscriptions.relayWhisper(socket, payload);
1305
- }
1306
- #wrapSocket(connection, request, upgrade) {
1307
- const raw = {
1308
- id: randomUUID2(),
1309
- data: upgrade.data ?? {},
1310
- auth: upgrade.auth,
1311
- connection,
1312
- request,
1313
- httpContext: upgrade.httpContext
1314
- };
1315
- return {
1316
- id: raw.id,
1317
- user: upgrade.user,
1318
- channels: /* @__PURE__ */ new Set(),
1319
- emit(event, data) {
1320
- sendFrame(connection, { type: "event", event, data });
1321
- },
1322
- joinUserRoom: (userId) => {
1323
- this.#joinUserRoom(raw.id, userRoom(userId));
1324
- return Promise.resolve();
1325
- },
1326
- leaveUserRoom: () => {
1327
- this.#leaveUserRoom(raw.id);
1328
- return Promise.resolve();
1329
- },
1330
- disconnect() {
1331
- connection.close();
1332
- },
1333
- raw
1334
- };
1335
- }
1336
- #joinUserRoom(socketId, room) {
1337
- this.#leaveUserRoom(socketId);
1338
- if (!this.#userRooms.has(room)) {
1339
- this.#userRooms.set(room, /* @__PURE__ */ new Set());
1340
- }
1341
- this.#userRooms.get(room).add(socketId);
1342
- this.#socketUserRoom.set(socketId, room);
1343
- }
1344
- #leaveUserRoom(socketId) {
1345
- const room = this.#socketUserRoom.get(socketId);
1346
- if (!room) {
1347
- return;
1348
- }
1349
- const sockets = this.#userRooms.get(room);
1350
- sockets?.delete(socketId);
1351
- if (sockets?.size === 0) {
1352
- this.#userRooms.delete(room);
1353
- }
1354
- this.#socketUserRoom.delete(socketId);
1355
- }
1356
- to(channel) {
1357
- return {
1358
- emit: (event, data) => {
1359
- this.#broadcastSink.dispatch({ target: "channel", channel, event, data });
1360
- },
1361
- except: (socketId) => ({
1362
- emit: (event, data) => {
1363
- this.#broadcastSink.dispatch({
1364
- target: "channel",
1365
- channel,
1366
- event,
1367
- data,
1368
- except: [socketId]
1369
- });
1370
- }
1371
- })
1372
- };
1373
- }
1374
- toUser(userId) {
1375
- const room = userRoom(userId);
1376
- return {
1377
- emit: (event, data) => {
1378
- this.#broadcastSink.dispatch({ target: "user", userId, room, event, data });
1379
- }
1380
- };
1381
- }
1382
- broadcast(event, data) {
1383
- this.#broadcastSink.dispatch({ target: "global", event, data });
1384
- }
1385
- #deliverBroadcast(emission) {
1386
- switch (emission.target) {
1387
- case "channel": {
1388
- const channel = emission.channel;
1389
- this.#bus?.publishChannel(channel, emission.event, emission.data, emission.except);
1390
- const traceMessage = {
1391
- target: "channel",
1392
- channel,
1393
- event: emission.event,
1394
- via: "local"
1395
- };
1396
- if (emission.except) {
1397
- traceMessage.except = emission.except;
1398
- }
1399
- this.#traceBroadcast(
1400
- traceMessage,
1401
- () => this.#emitToChannelLocally(channel, emission.event, emission.data, emission.except)
1402
- );
1403
- return;
1404
- }
1405
- case "user": {
1406
- const room = emission.room;
1407
- this.#bus?.publishUser(room, emission.event, emission.data);
1408
- this.#traceBroadcast(
1409
- { target: "user", room, event: emission.event, via: "local" },
1410
- () => this.#emitToUserRoomLocally(room, emission.event, emission.data)
1411
- );
1412
- return;
1413
- }
1414
- case "global":
1415
- this.#bus?.publishBroadcast(emission.event, emission.data);
1416
- this.#traceBroadcast(
1417
- { target: "global", event: emission.event, via: "local" },
1418
- () => this.#broadcastLocally(emission.event, emission.data)
1419
- );
1420
- }
1421
- }
1422
- #traceBroadcast(message, deliver) {
1423
- broadcastChannel.traceSync(() => {
1424
- message.delivered = deliver();
1425
- }, message);
1426
- }
1427
- #emitToChannelLocally(channel, event, data, except = []) {
1428
- let delivered = 0;
1429
- for (const socketId of this.#channelSubscriptions?.getSocketIds(channel) ?? []) {
1430
- if (except.includes(socketId)) {
1431
- continue;
1432
- }
1433
- const socket = this.#sockets.get(socketId);
1434
- if (socket) {
1435
- sendFrame(socket.raw.connection, { type: "event", channel, event, data });
1436
- delivered += 1;
1437
- }
1438
- }
1439
- return delivered;
1440
- }
1441
- #emitToUserRoomLocally(room, event, data) {
1442
- let delivered = 0;
1443
- for (const socketId of this.#userRooms.get(room) ?? []) {
1444
- const socket = this.#sockets.get(socketId);
1445
- if (socket) {
1446
- sendFrame(socket.raw.connection, { type: "event", event, data });
1447
- delivered += 1;
1448
- }
1449
- }
1450
- return delivered;
1451
- }
1452
- #broadcastLocally(event, data) {
1453
- let delivered = 0;
1454
- for (const socket of this.#sockets.values()) {
1455
- sendFrame(socket.raw.connection, { type: "event", event, data });
1456
- delivered += 1;
1457
- }
1458
- return delivered;
1459
- }
1460
- #configurePresenceFetcher() {
1461
- if (!this.#presenceManager || !this.#bus) {
1462
- return;
1463
- }
1464
- this.#presenceManager.setSocketFetcher(async (channel) => {
1465
- return [
1466
- ...this.#presenceManager.getLocalSockets(channel),
1467
- ...await this.#bus.fetchPresenceSockets(channel)
1468
- ];
1469
- });
1470
- }
1471
- #getLocalPresenceSockets(channel) {
1472
- return (this.#presenceManager?.getLocalSockets(channel) ?? []).map((socket) => {
1473
- return this.#serializePresenceSocket(channel, socket);
1474
- });
1475
- }
1476
- #serializePresenceSocket(channel, socket) {
1477
- const user = socket.data[PRESENCE_DATA_KEY][channel];
1478
- return {
1479
- id: socket.id,
1480
- data: {
1481
- [PRESENCE_DATA_KEY]: {
1482
- [channel]: {
1483
- ...user,
1484
- joinedAt: user.joinedAt.toISOString()
1485
- }
1486
- }
1487
- }
1488
- };
1489
- }
1490
- getSocket(socketId) {
1491
- return this.#sockets.get(socketId);
1492
- }
1493
- get connectionsCount() {
1494
- return this.#sockets.size;
1495
- }
1496
- get status() {
1497
- return this.#status;
1498
- }
1499
- get ready() {
1500
- return this.#status === "ready" && Boolean(this.#server);
1501
- }
1502
- fake() {
1503
- const fake = new SocketFake(() => {
1504
- if (this.#fake === fake) {
1505
- this.restore();
1506
- }
1507
- });
1508
- this.#fake = fake;
1509
- this.#broadcastSink = fake;
1510
- return fake;
1511
- }
1512
- restore() {
1513
- this.#fake = null;
1514
- this.#broadcastSink = this.#deliverySink;
1515
- }
1516
- markFailed(error) {
1517
- this.#lastError = error instanceof Error ? error : new Error("Socket service failed");
1518
- this.#status = "failed";
1519
- }
1520
- health() {
1521
- return {
1522
- status: this.#status,
1523
- ready: this.ready,
1524
- connections: this.#sockets.size,
1525
- channels: this.#channelSubscriptions?.channelsCount ?? 0,
1526
- ...this.#lastError ? { lastError: this.#lastError.message } : {}
1527
- };
1528
- }
1529
- #startHeartbeat(config) {
1530
- this.#stopHeartbeat();
1531
- if (!config) {
1532
- return;
1533
- }
1534
- this.#heartbeatInterval = setInterval(() => {
1535
- this.#pingSockets(config.timeout);
1536
- }, config.interval);
1537
- }
1538
- #pingSockets(timeout) {
1539
- for (const socket of this.#sockets.values()) {
1540
- const { connection } = socket.raw;
1541
- if (connection.readyState !== WebSocket.OPEN) {
1542
- this.#clearHeartbeatTimeout(socket.id);
1543
- continue;
1544
- }
1545
- if (this.#heartbeatTimeouts.has(socket.id)) {
1546
- continue;
1547
- }
1548
- const timeoutId = setTimeout(() => {
1549
- this.#heartbeatTimeouts.delete(socket.id);
1550
- connection.terminate();
1551
- }, timeout);
1552
- this.#heartbeatTimeouts.set(socket.id, timeoutId);
1553
- try {
1554
- connection.ping();
1555
- } catch {
1556
- this.#clearHeartbeatTimeout(socket.id);
1557
- connection.terminate();
1558
- }
1559
- }
1560
- }
1561
- #clearHeartbeatTimeout(socketId) {
1562
- const timeout = this.#heartbeatTimeouts.get(socketId);
1563
- if (!timeout) {
1564
- return;
1565
- }
1566
- clearTimeout(timeout);
1567
- this.#heartbeatTimeouts.delete(socketId);
1568
- }
1569
- #stopHeartbeat() {
1570
- if (this.#heartbeatInterval) {
1571
- clearInterval(this.#heartbeatInterval);
1572
- this.#heartbeatInterval = null;
1573
- }
1574
- for (const timeout of this.#heartbeatTimeouts.values()) {
1575
- clearTimeout(timeout);
1576
- }
1577
- this.#heartbeatTimeouts.clear();
1578
- }
1579
- #cleanup() {
1580
- this.#stopHeartbeat();
1581
- for (const socket of this.#sockets.values()) {
1582
- socket.raw.connection.terminate();
1583
- }
1584
- this.#sockets.clear();
1585
- this.#userRooms.clear();
1586
- this.#socketUserRoom.clear();
1587
- this.#messageQueue.clear();
1588
- this.#channelSubscriptions?.clear();
1589
- }
1590
- async close() {
1591
- const server = this.#server;
1592
- const bus = this.#bus;
1593
- const httpServer = this.#httpServer;
1594
- const upgradeHandler = this.#upgradeHandler;
1595
- if (this.#status !== "failed") {
1596
- this.#status = "stopping";
1597
- }
1598
- this.#server = null;
1599
- this.#bus = null;
1600
- this.#httpServer = null;
1601
- this.#upgradeHandler = null;
1602
- if (httpServer && upgradeHandler) {
1603
- httpServer.off("upgrade", upgradeHandler);
1604
- }
1605
- this.#cleanup();
1606
- try {
1607
- await bus?.close();
1608
- if (!server) {
1609
- if (this.#status !== "failed") {
1610
- this.#status = "stopped";
1611
- }
1612
- return;
1613
- }
1614
- await new Promise((resolve) => {
1615
- server.close(() => {
1616
- resolve();
1617
- });
1618
- });
1619
- if (this.#status !== "failed") {
1620
- this.#status = "stopped";
1621
- }
1622
- } catch (error) {
1623
- this.markFailed(error);
1624
- throw error;
1625
- }
1626
- }
1627
- };
1628
-
1629
- // src/channel_pattern.ts
1630
- import { exec, match, parse } from "@poppinss/matchit";
1631
- var ChannelPattern = class _ChannelPattern {
1632
- constructor(value) {
1633
- this.value = value;
1634
- this.route = parse(value);
1635
- this.score = this.route.map((segment) => {
1636
- switch (segment.type) {
1637
- case 0:
1638
- return 40;
1639
- case 1:
1640
- return 30;
1641
- case 3:
1642
- return 20;
1643
- case 2:
1644
- return 0;
1645
- }
1646
- });
1647
- }
1648
- route;
1649
- score;
1650
- static from(value) {
1651
- return new _ChannelPattern(value);
1652
- }
1653
- match(channelName) {
1654
- const matched = match(channelName, [this.route]);
1655
- if (matched.length === 0) {
1656
- return null;
1657
- }
1658
- return exec(channelName, matched, true);
1659
- }
1660
- /**
1661
- * Returns the most specific matching candidate. Static segments outrank params,
1662
- * params outrank optional params, and optional params outrank wildcards.
1663
- * Insertion order is preserved for ties.
1664
- */
1665
- static firstMatch(channelName, candidates) {
1666
- const sorted = [...candidates].sort((left, right) => {
1667
- return _ChannelPattern.#compareSpecificity(left.pattern, right.pattern);
1668
- });
1669
- const routes = sorted.map((candidate) => candidate.pattern.route);
1670
- const matched = match(channelName, routes);
1671
- if (matched.length === 0) {
1672
- return null;
1673
- }
1674
- for (const { pattern, channel } of sorted) {
1675
- if (pattern.route !== matched) {
1676
- continue;
1677
- }
1678
- const params = pattern.match(channelName);
1679
- if (params) {
1680
- return { pattern, channel, params };
1681
- }
1682
- }
1683
- return null;
1684
- }
1685
- static #compareSpecificity(left, right) {
1686
- const length = Math.max(left.score.length, right.score.length);
1687
- for (let index = 0; index < length; index++) {
1688
- const scoreDifference = (right.score[index] ?? -1) - (left.score[index] ?? -1);
1689
- if (scoreDifference !== 0) {
1690
- return scoreDifference;
1691
- }
1692
- }
1693
- return right.score.length - left.score.length;
1694
- }
1695
- };
1696
-
1697
- // src/channel_router.ts
1698
- var ChannelRouter = class {
1699
- /**
1700
- * Registered channels keyed by their pattern.
1701
- */
1702
- #channels = /* @__PURE__ */ new Map();
1703
- /**
1704
- * Registers a channel.
1705
- */
1706
- register(channel) {
1707
- const patternValue = channel.pattern;
1708
- if (!patternValue) {
1709
- throw new Error(`Channel ${channel.name} must define a static pattern`);
1710
- }
1711
- if (this.#channels.has(patternValue)) {
1712
- throw new Error(`Channel pattern "${patternValue}" is already registered`);
1713
- }
1714
- const pattern = ChannelPattern.from(patternValue);
1715
- this.#channels.set(pattern.value, { pattern, channel });
1716
- }
1717
- /**
1718
- * Matches a channel name against registered patterns.
1719
- */
1720
- match(channelName) {
1721
- const matched = ChannelPattern.firstMatch(channelName, this.#channels.values());
1722
- if (!matched) {
1723
- return null;
1724
- }
1725
- return {
1726
- channel: matched.channel,
1727
- pattern: matched.pattern.value,
1728
- params: matched.params
1729
- };
1730
- }
1731
- /**
1732
- * Authorizes a socket before it joins a channel.
1733
- */
1734
- async authorize(socket, channelName) {
1735
- const matched = this.match(channelName);
1736
- if (!matched) {
1737
- return {
1738
- success: false,
1739
- error: "Channel not found"
1740
- };
1741
- }
1742
- const instance = new matched.channel();
1743
- const paramValues = Object.values(matched.params);
1744
- const ctx = {
1745
- socket,
1746
- channel: channelName,
1747
- params: matched.params,
1748
- presenceData: void 0,
1749
- setPresenceData(data) {
1750
- ctx.presenceData = data;
1751
- }
1752
- };
1753
- try {
1754
- await instance.$runMiddlewares(ctx);
1755
- } catch (error) {
1756
- return {
1757
- success: false,
1758
- error: error instanceof Error ? error.message : "Middleware rejected"
1759
- };
1760
- }
1761
- return {
1762
- success: true,
1763
- instance,
1764
- params: matched.params,
1765
- paramValues,
1766
- presenceData: ctx.presenceData ?? null
1767
- };
1768
- }
1769
- /**
1770
- * Lists all registered channels.
1771
- */
1772
- list() {
1773
- return [...this.#channels.values()].map(({ pattern, channel }) => ({
1774
- pattern: pattern.value,
1775
- name: channel.name,
1776
- options: channel.options
1777
- }));
1778
- }
1779
- /**
1780
- * Number of registered channels.
1781
- */
1782
- get size() {
1783
- return this.#channels.size;
1784
- }
1785
- };
1786
-
1787
- export {
1788
- PresenceManager,
1789
- BaseChannel,
1790
- SocketFake,
1791
- SocketService,
1792
- ChannelRouter
1793
- };
1794
- //# sourceMappingURL=chunk-NRWRLZAO.js.map