1 | "use strict";
|
2 | Object.defineProperty(exports, "__esModule", { value: true });
|
3 | exports.WebSocketsController = void 0;
|
4 | const logger_service_1 = require("@nestjs/common/services/logger.service");
|
5 | const metadata_scanner_1 = require("@nestjs/core/metadata-scanner");
|
6 | const rxjs_1 = require("rxjs");
|
7 | const operators_1 = require("rxjs/operators");
|
8 | const constants_1 = require("./constants");
|
9 | const invalid_socket_port_exception_1 = require("./errors/invalid-socket-port.exception");
|
10 | const gateway_metadata_explorer_1 = require("./gateway-metadata-explorer");
|
11 | const compare_element_util_1 = require("./utils/compare-element.util");
|
12 | class WebSocketsController {
|
13 | constructor(socketServerProvider, config, contextCreator) {
|
14 | this.socketServerProvider = socketServerProvider;
|
15 | this.config = config;
|
16 | this.contextCreator = contextCreator;
|
17 | this.logger = new logger_service_1.Logger(WebSocketsController.name, {
|
18 | timestamp: true,
|
19 | });
|
20 | this.metadataExplorer = new gateway_metadata_explorer_1.GatewayMetadataExplorer(new metadata_scanner_1.MetadataScanner());
|
21 | }
|
22 | connectGatewayToServer(instance, metatype, moduleKey) {
|
23 | const options = Reflect.getMetadata(constants_1.GATEWAY_OPTIONS, metatype) || {};
|
24 | const port = Reflect.getMetadata(constants_1.PORT_METADATA, metatype) || 0;
|
25 | if (!Number.isInteger(port)) {
|
26 | throw new invalid_socket_port_exception_1.InvalidSocketPortException(port, metatype);
|
27 | }
|
28 | this.subscribeToServerEvents(instance, options, port, moduleKey);
|
29 | }
|
30 | subscribeToServerEvents(instance, options, port, moduleKey) {
|
31 | const nativeMessageHandlers = this.metadataExplorer.explore(instance);
|
32 | const messageHandlers = nativeMessageHandlers.map(({ callback, message, methodName }) => ({
|
33 | message,
|
34 | methodName,
|
35 | callback: this.contextCreator.create(instance, callback, moduleKey, methodName),
|
36 | }));
|
37 | const observableServer = this.socketServerProvider.scanForSocketServer(options, port);
|
38 | this.assignServerToProperties(instance, observableServer.server);
|
39 | this.subscribeEvents(instance, messageHandlers, observableServer);
|
40 | }
|
41 | subscribeEvents(instance, subscribersMap, observableServer) {
|
42 | const { init, disconnect, connection, server } = observableServer;
|
43 | const adapter = this.config.getIoAdapter();
|
44 | this.subscribeInitEvent(instance, init);
|
45 | this.subscribeConnectionEvent(instance, connection);
|
46 | this.subscribeDisconnectEvent(instance, disconnect);
|
47 | const handler = this.getConnectionHandler(this, instance, subscribersMap, disconnect, connection);
|
48 | adapter.bindClientConnect(server, handler);
|
49 | this.printSubscriptionLogs(instance, subscribersMap);
|
50 | }
|
51 | getConnectionHandler(context, instance, subscribersMap, disconnect, connection) {
|
52 | const adapter = this.config.getIoAdapter();
|
53 | return (...args) => {
|
54 | const [client] = args;
|
55 | connection.next(args);
|
56 | context.subscribeMessages(subscribersMap, client, instance);
|
57 | const disconnectHook = adapter.bindClientDisconnect;
|
58 | disconnectHook &&
|
59 | disconnectHook.call(adapter, client, () => disconnect.next(client));
|
60 | };
|
61 | }
|
62 | subscribeInitEvent(instance, event) {
|
63 | if (instance.afterInit) {
|
64 | event.subscribe(instance.afterInit.bind(instance));
|
65 | }
|
66 | }
|
67 | subscribeConnectionEvent(instance, event) {
|
68 | if (instance.handleConnection) {
|
69 | event
|
70 | .pipe((0, operators_1.distinctUntilChanged)((prev, curr) => (0, compare_element_util_1.compareElementAt)(prev, curr, 0)))
|
71 | .subscribe((args) => instance.handleConnection(...args));
|
72 | }
|
73 | }
|
74 | subscribeDisconnectEvent(instance, event) {
|
75 | if (instance.handleDisconnect) {
|
76 | event
|
77 | .pipe((0, operators_1.distinctUntilChanged)())
|
78 | .subscribe(instance.handleDisconnect.bind(instance));
|
79 | }
|
80 | }
|
81 | subscribeMessages(subscribersMap, client, instance) {
|
82 | const adapter = this.config.getIoAdapter();
|
83 | const handlers = subscribersMap.map(({ callback, message }) => ({
|
84 | message,
|
85 | callback: callback.bind(instance, client),
|
86 | }));
|
87 | adapter.bindMessageHandlers(client, handlers, data => (0, rxjs_1.from)(this.pickResult(data)).pipe((0, operators_1.mergeAll)()));
|
88 | }
|
89 | async pickResult(deferredResult) {
|
90 | const result = await deferredResult;
|
91 | if ((0, rxjs_1.isObservable)(result)) {
|
92 | return result;
|
93 | }
|
94 | if (result instanceof Promise) {
|
95 | return (0, rxjs_1.from)(result);
|
96 | }
|
97 | return (0, rxjs_1.of)(result);
|
98 | }
|
99 | assignServerToProperties(instance, server) {
|
100 | for (const propertyKey of this.metadataExplorer.scanForServerHooks(instance)) {
|
101 | Reflect.set(instance, propertyKey, server);
|
102 | }
|
103 | }
|
104 | printSubscriptionLogs(instance, subscribersMap) {
|
105 | var _a;
|
106 | const gatewayClassName = (_a = instance === null || instance === void 0 ? void 0 : instance.constructor) === null || _a === void 0 ? void 0 : _a.name;
|
107 | if (!gatewayClassName) {
|
108 | return;
|
109 | }
|
110 | subscribersMap.forEach(({ message }) => this.logger.log(`${gatewayClassName} subscribed to the "${message}" message`));
|
111 | }
|
112 | }
|
113 | exports.WebSocketsController = WebSocketsController;
|