@nestjs/websockets
Version:
Nest - modern, fast, powerful node.js web framework (@websockets)
53 lines (52 loc) • 2.61 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.SocketServerProvider = void 0;
const tslib_1 = require("tslib");
const shared_utils_1 = require("@nestjs/common/utils/shared.utils");
const server_and_event_streams_factory_1 = require("./factories/server-and-event-streams-factory");
class SocketServerProvider {
constructor(socketsContainer, applicationConfig) {
this.socketsContainer = socketsContainer;
this.applicationConfig = applicationConfig;
}
scanForSocketServer(options, port) {
const serverAndStreamsHost = this.socketsContainer.getOneByConfig({
port,
path: options.path,
});
if (serverAndStreamsHost && options.namespace) {
return this.decorateWithNamespace(options, port, serverAndStreamsHost.server);
}
return serverAndStreamsHost
? serverAndStreamsHost
: this.createSocketServer(options, port);
}
createSocketServer(options, port) {
const adapter = this.applicationConfig.getIoAdapter();
const _a = options, { namespace, server } = _a, partialOptions = tslib_1.__rest(_a, ["namespace", "server"]);
const ioServer = adapter.create(port, partialOptions);
const serverAndEventStreamsHost = server_and_event_streams_factory_1.ServerAndEventStreamsFactory.create(ioServer);
this.socketsContainer.addOne({ port, path: options.path }, serverAndEventStreamsHost);
if (!namespace) {
return serverAndEventStreamsHost;
}
return this.decorateWithNamespace(options, port, ioServer);
}
decorateWithNamespace(options, port, targetServer) {
const namespaceServer = this.getServerOfNamespace(options, port, targetServer);
const serverAndEventStreamsHost = server_and_event_streams_factory_1.ServerAndEventStreamsFactory.create(namespaceServer);
this.socketsContainer.addOne({ port, path: options.path, namespace: options.namespace }, serverAndEventStreamsHost);
return serverAndEventStreamsHost;
}
getServerOfNamespace(options, port, server) {
const adapter = this.applicationConfig.getIoAdapter();
return adapter.create(port, Object.assign(Object.assign({}, options), { namespace: this.validateNamespace(options.namespace || ''), server }));
}
validateNamespace(namespace) {
if (!(0, shared_utils_1.isString)(namespace)) {
return namespace;
}
return (0, shared_utils_1.addLeadingSlash)(namespace);
}
}
exports.SocketServerProvider = SocketServerProvider;