ref

dist/src/services/webSocketServer/index.js

@@ -10,19 +10,18 @@ const enums_1 = require("../../enums");
 const client_1 = require("../../models/client");
 const WS_PATH = 'peerjs';
 class WebSocketServer extends events_1.default {
-    constructor({ server, realm, config, messagesTransport }) {
+    constructor({ server, realm, config }) {
         var _a;
         super();
         this.setMaxListeners(0);
         this.realm = realm;
         this.config = config;
-        this.messagesTransport = messagesTransport;
         const path = this.config.path;
         this.path = `${path}${path.endsWith('/') ? "" : "/"}${WS_PATH}`;
         this.socketServer = new ws_1.default.Server({ path: this.path, server });
         this.socketServer.on("connection", (socket, req) => this._onSocketConnection(socket, req));
         this.socketServer.on("error", (error) => this._onSocketError(error));
-        (_a = this.messagesTransport) === null || _a === void 0 ? void 0 : _a.registerHanadler((message) => this._handleMessage(message));
+        (_a = this.config.messagesTransport) === null || _a === void 0 ? void 0 : _a.registerHanadler((message) => this._handleMessage(message));
     }
     _onSocketConnection(socket, req) {
         var _a;
@@ -77,8 +76,8 @@ class WebSocketServer extends events_1.default {
             try {
                 const message = JSON.parse(data);
                 message.src = client.getId();
-                if (message.type !== "HEARTBEAT" && this.messagesTransport) {
-                    this.messagesTransport.sendMessage(message);
+                if (message.type !== "HEARTBEAT" && this.config.messagesTransport) {
+                    this.config.messagesTransport.sendMessage(message);
                     return;
                 }
                 this.emit("message", client, message);

index.d.ts

@@ -26,6 +26,7 @@ declare interface IConfig {
     cert: string;
   };
   readonly generateClientId?: () => string;
+  readonly messagesTransport?: MessagesTransport;
 }
 
 declare interface IClient {
@@ -65,6 +66,11 @@ declare interface CustomExpress extends Express.Express {
   on(event: 'error', callback: (error: Error) => void): this;
 }
 
+declare type MessagesTransport = {
+  registerHanadler(handler: (message: { type: string; dst?: string; } & any) => boolean): void;
+  sendMessage(message: { type: string; src: string; } & any): void;
+};
+
 declare function ExpressPeerServer(server: Server, options?: IConfig): CustomExpress;
 declare function PeerServer(options?: Optional<IConfig>, callback?: (server: Server) => void): CustomExpress;
 

src/config/index.ts

@@ -1,3 +1,8 @@
+type MessagesTransport = {
+  registerHanadler(handler: (message: { type: string; dst?: string; } & any) => boolean): void;
+  sendMessage(message: { type: string; src: string; } & any): void;
+};
+
 export interface IConfig {
   readonly host: string;
   readonly port: number;
@@ -14,6 +19,7 @@ export interface IConfig {
     cert: string;
   };
   readonly generateClientId?: () => string;
+  readonly messagesTransport?: MessagesTransport;
 }
 
 const defaultConfig: IConfig = {

src/services/webSocketServer/index.ts

@@ -18,12 +18,7 @@ interface IAuthParams {
   key?: string;
 }
 
-type CustomConfig = Pick<IConfig, 'path' | 'key' | 'concurrent_limit'>;
-
-type MessagesTransport = {
-  registerHanadler(handler: (message: { type: string; dst?: string; } & any) => boolean): void;
-  sendMessage(message: { type: string; src: string; } & any): void;
-};
+type CustomConfig = Pick<IConfig, 'path' | 'key' | 'concurrent_limit' | 'messagesTransport'>;
 
 const WS_PATH = 'peerjs';
 
@@ -31,7 +26,6 @@ type Dependencies = {
   server: any;
   realm: IRealm;
   config: CustomConfig;
-  messagesTransport?: MessagesTransport;
 };
 
 export class WebSocketServer extends EventEmitter implements IWebSocketServer {
@@ -41,16 +35,14 @@ export class WebSocketServer extends EventEmitter implements IWebSocketServer {
 
   private readonly realm: IRealm;
   private readonly config: CustomConfig;
-  private readonly messagesTransport?: MessagesTransport;
 
-  constructor({ server, realm, config, messagesTransport }: Dependencies) {
+  constructor({ server, realm, config }: Dependencies) {
     super();
 
     this.setMaxListeners(0);
 
     this.realm = realm;
     this.config = config;
-    this.messagesTransport = messagesTransport;
 
     const path = this.config.path;
     this.path = `${path}${path.endsWith('/') ? "" : "/"}${WS_PATH}`;
@@ -60,7 +52,7 @@ export class WebSocketServer extends EventEmitter implements IWebSocketServer {
     this.socketServer.on("connection", (socket: MyWebSocket, req) => this._onSocketConnection(socket, req));
     this.socketServer.on("error", (error: Error) => this._onSocketError(error));
 
-    this.messagesTransport?.registerHanadler((message) => this._handleMessage(message));
+    this.config.messagesTransport?.registerHanadler((message) => this._handleMessage(message));
   }
 
   private _onSocketConnection(socket: MyWebSocket, req: IncomingMessage): void {
@@ -138,8 +130,8 @@ export class WebSocketServer extends EventEmitter implements IWebSocketServer {
 
         message.src = client.getId();
 
-        if (message.type !== "HEARTBEAT" && this.messagesTransport) {
-          this.messagesTransport.sendMessage(message);
+        if (message.type !== "HEARTBEAT" && this.config.messagesTransport) {
+          this.config.messagesTransport.sendMessage(message);
           return;
         }