e1c-test-tools
Version:
Набор инструментов для тестирования (предварительная версия)
93 lines (92 loc) • 4.01 kB
JavaScript
;
var __rest = (this && this.__rest) || function (s, e) {
var t = {};
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
t[p] = s[p];
if (s != null && typeof Object.getOwnPropertySymbols === "function")
for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
t[p[i]] = s[p[i]];
}
return t;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.SocketIoAdapter = void 0;
const shared_utils_1 = require("@nestjs/common/utils/shared.utils");
const websockets_1 = require("@nestjs/websockets");
const constants_1 = require("@nestjs/websockets/constants");
const rxjs_1 = require("rxjs");
const operators_1 = require("rxjs/operators");
const socket_io_1 = require("socket.io");
class SocketIoAdapter extends websockets_1.AbstractWsAdapter {
constructor(appOrHttpServer, corsOrigins = []) {
super(appOrHttpServer);
this.corsOrigins = corsOrigins;
}
create(port, options) {
if (!options) {
return this.createIOServer(port);
}
const { namespace, server } = options, opt = __rest(options, ["namespace", "server"]);
// eslint-disable-next-line no-nested-ternary
return server && shared_utils_1.isFunction(server.of)
? server.of(namespace)
: namespace
? this.createIOServer(port, opt).of(namespace)
: this.createIOServer(port, opt);
}
createIOServer(port, options) {
if (this.httpServer && port === 0) {
const s = new socket_io_1.Server(this.httpServer, {
'cors': {
'origin': this.corsOrigins,
'methods': ['GET', 'POST'],
'credentials': true,
},
'cookie': {
// 'domain': 'io',
'httpOnly': true,
'path': '/',
},
// Allow 1MB of data per request.
'maxHttpBufferSize': 1e6,
});
return s;
}
return new socket_io_1.Server(port, options);
}
bindMessageHandlers(client, handlers, transform) {
const disconnect$ = rxjs_1.fromEvent(client, constants_1.DISCONNECT_EVENT).pipe(operators_1.share(), operators_1.first());
handlers.forEach(({ message, callback }) => {
const source$ = rxjs_1.fromEvent(client, message).pipe(operators_1.mergeMap((payload) => {
const { data, ack } = this.mapPayload(payload);
return transform(callback(data, ack)).pipe(operators_1.filter((response) => !shared_utils_1.isNil(response)), operators_1.map((response) => [response, ack]));
}), operators_1.takeUntil(disconnect$));
// eslint-disable-next-line consistent-return
source$.subscribe(([response, ack]) => {
if (response.event) {
return client.emit(response.event, response.data);
}
// eslint-disable-next-line no-unused-expressions
shared_utils_1.isFunction(ack) && ack(response);
});
});
}
// eslint-disable-next-line class-methods-use-this
mapPayload(payload) {
if (!Array.isArray(payload)) {
return { 'data': payload };
}
const lastElement = payload[payload.length - 1];
const isAck = shared_utils_1.isFunction(lastElement);
if (isAck) {
const size = payload.length - 1;
return {
'data': size === 1 ? payload[0] : payload.slice(0, size),
'ack': lastElement,
};
}
return { 'data': payload };
}
}
exports.SocketIoAdapter = SocketIoAdapter;