1 | "use strict";
|
2 | Object.defineProperty(exports, "__esModule", { value: true });
|
3 | exports.WebSocketsController = void 0;
|
4 | const shared_utils_1 = require("@nestjs/common/utils/shared.utils");
|
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.metadataExplorer = new gateway_metadata_explorer_1.GatewayMetadataExplorer(new metadata_scanner_1.MetadataScanner());
|
18 | }
|
19 | connectGatewayToServer(instance, metatype, moduleKey) {
|
20 | const options = Reflect.getMetadata(constants_1.GATEWAY_OPTIONS, metatype) || {};
|
21 | const port = Reflect.getMetadata(constants_1.PORT_METADATA, metatype) || 0;
|
22 | if (!Number.isInteger(port)) {
|
23 | throw new invalid_socket_port_exception_1.InvalidSocketPortException(port, metatype);
|
24 | }
|
25 | this.subscribeToServerEvents(instance, options, port, moduleKey);
|
26 | }
|
27 | subscribeToServerEvents(instance, options, port, moduleKey) {
|
28 | const nativeMessageHandlers = this.metadataExplorer.explore(instance);
|
29 | const messageHandlers = nativeMessageHandlers.map(({ callback, message, methodName }) => ({
|
30 | message,
|
31 | methodName,
|
32 | callback: this.contextCreator.create(instance, callback, moduleKey, methodName),
|
33 | }));
|
34 | const observableServer = this.socketServerProvider.scanForSocketServer(options, port);
|
35 | this.assignServerToProperties(instance, observableServer.server);
|
36 | this.subscribeEvents(instance, messageHandlers, observableServer);
|
37 | }
|
38 | subscribeEvents(instance, subscribersMap, observableServer) {
|
39 | const { init, disconnect, connection, server } = observableServer;
|
40 | const adapter = this.config.getIoAdapter();
|
41 | this.subscribeInitEvent(instance, init);
|
42 | this.subscribeConnectionEvent(instance, connection);
|
43 | this.subscribeDisconnectEvent(instance, disconnect);
|
44 | const handler = this.getConnectionHandler(this, instance, subscribersMap, disconnect, connection);
|
45 | adapter.bindClientConnect(server, handler);
|
46 | }
|
47 | getConnectionHandler(context, instance, subscribersMap, disconnect, connection) {
|
48 | const adapter = this.config.getIoAdapter();
|
49 | return (...args) => {
|
50 | const [client] = args;
|
51 | connection.next(args);
|
52 | context.subscribeMessages(subscribersMap, client, instance);
|
53 | const disconnectHook = adapter.bindClientDisconnect;
|
54 | disconnectHook &&
|
55 | disconnectHook.call(adapter, client, () => disconnect.next(client));
|
56 | };
|
57 | }
|
58 | subscribeInitEvent(instance, event) {
|
59 | if (instance.afterInit) {
|
60 | event.subscribe(instance.afterInit.bind(instance));
|
61 | }
|
62 | }
|
63 | subscribeConnectionEvent(instance, event) {
|
64 | if (instance.handleConnection) {
|
65 | event
|
66 | .pipe(operators_1.distinctUntilChanged((prev, curr) => compare_element_util_1.compareElementAt(prev, curr, 0)))
|
67 | .subscribe((args) => instance.handleConnection(...args));
|
68 | }
|
69 | }
|
70 | subscribeDisconnectEvent(instance, event) {
|
71 | if (instance.handleDisconnect) {
|
72 | event
|
73 | .pipe(operators_1.distinctUntilChanged())
|
74 | .subscribe(instance.handleDisconnect.bind(instance));
|
75 | }
|
76 | }
|
77 | subscribeMessages(subscribersMap, client, instance) {
|
78 | const adapter = this.config.getIoAdapter();
|
79 | const handlers = subscribersMap.map(({ callback, message }) => ({
|
80 | message,
|
81 | callback: callback.bind(instance, client),
|
82 | }));
|
83 | adapter.bindMessageHandlers(client, handlers, data => rxjs_1.from(this.pickResult(data)).pipe(operators_1.mergeAll()));
|
84 | }
|
85 | async pickResult(defferedResult) {
|
86 | const result = await defferedResult;
|
87 | if (result && shared_utils_1.isFunction(result.subscribe)) {
|
88 | return result;
|
89 | }
|
90 | if (result instanceof Promise) {
|
91 | return rxjs_1.from(result);
|
92 | }
|
93 | return rxjs_1.of(result);
|
94 | }
|
95 | assignServerToProperties(instance, server) {
|
96 | for (const propertyKey of this.metadataExplorer.scanForServerHooks(instance)) {
|
97 | Reflect.set(instance, propertyKey, server);
|
98 | }
|
99 | }
|
100 | }
|
101 | exports.WebSocketsController = WebSocketsController;
|