refactor(*): make WebsocketEventEmitter a super class

src/games/gateways/games.gateway.ts

@@ -1,8 +1,4 @@
-import {
-  WebSocketGateway,
-  OnGatewayInit,
-  SubscribeMessage,
-} from '@nestjs/websockets';
+import { WebSocketGateway, SubscribeMessage } from '@nestjs/websockets';
 import { Socket } from 'socket.io';
 import { WsAuthorized } from '@/auth/decorators/ws-authorized.decorator';
 import { PlayerSubstitutionService } from '../services/player-substitution.service';
@@ -14,26 +10,22 @@ import {
 } from '@nestjs/common';
 import { Events } from '@/events/events';
 import { WebsocketEvent } from '@/websocket-event';
-import { serialize } from '@/shared/serialize';
 import { SerializerInterceptor } from '@/shared/interceptors/serializer.interceptor';
 import { Serializable } from '@/shared/serializable';
 import { GameDto } from '../dto/game.dto';
 import { WebsocketEventEmitter } from '@/shared/websocket-event-emitter';
 
 @WebSocketGateway()
 export class GamesGateway
-  implements OnGatewayInit, OnModuleInit, WebsocketEventEmitter<GameDto>
+  extends WebsocketEventEmitter<GameDto>
+  implements OnModuleInit
 {
-  private socket: Socket;
-
   constructor(
     @Inject(forwardRef(() => PlayerSubstitutionService))
     private playerSubstitutionService: PlayerSubstitutionService,
     private events: Events,
-  ) {}
-
-  async emit(event: WebsocketEvent, payload: Serializable<GameDto>) {
-    this.socket.emit(event, await serialize(payload));
+  ) {
+    super();
   }
 
   @WsAuthorized()
@@ -50,10 +42,6 @@ export class GamesGateway
     );
   }
 
-  afterInit(socket: Socket) {
-    this.socket = socket;
-  }
-
   onModuleInit() {
     this.events.gameCreated.subscribe(({ game }) =>
       this.emit(WebsocketEvent.gameCreated, game),

src/players/gateways/players.gateway.ts

@@ -2,31 +2,23 @@ import {
   WebSocketGateway,
   OnGatewayConnection,
   OnGatewayDisconnect,
-  OnGatewayInit,
 } from '@nestjs/websockets';
 import { Socket } from 'socket.io';
 import { Subject } from 'rxjs';
 import { OnModuleInit } from '@nestjs/common';
 import { Events } from '@/events/events';
 import { WebsocketEvent } from '@/websocket-event';
 import { PlayersService } from '../services/players.service';
-import { serialize } from '@/shared/serialize';
 import { WebsocketEventEmitter } from '@/shared/websocket-event-emitter';
 import { PlayerDto } from '../dto/player.dto';
-import { Serializable } from '@/shared/serializable';
 
 @WebSocketGateway()
 export class PlayersGateway
-  implements
-    OnGatewayConnection,
-    OnGatewayDisconnect,
-    OnGatewayInit,
-    OnModuleInit,
-    WebsocketEventEmitter<PlayerDto>
+  extends WebsocketEventEmitter<PlayerDto>
+  implements OnGatewayConnection, OnGatewayDisconnect, OnModuleInit
 {
   private _playerConnected = new Subject<Socket>();
   private _playerDisconnected = new Subject<Socket>();
-  private socket: Socket;
 
   get playerConnected() {
     return this._playerConnected.asObservable();
@@ -36,10 +28,8 @@ export class PlayersGateway
     return this._playerDisconnected.asObservable();
   }
 
-  constructor(private events: Events, private playersService: PlayersService) {}
-
-  async emit(event: WebsocketEvent, payload: Serializable<PlayerDto>) {
-    this.socket.emit(event, await serialize(payload));
+  constructor(private events: Events, private playersService: PlayersService) {
+    super();
   }
 
   handleConnection(socket: Socket) {
@@ -50,10 +40,6 @@ export class PlayersGateway
     this._playerDisconnected.next(socket);
   }
 
-  afterInit(socket: Socket) {
-    this.socket = socket;
-  }
-
   onModuleInit() {
     this.events.playerConnects.subscribe(async ({ playerId }) =>
       this.emit(

src/shared/websocket-event-emitter.ts

@@ -1,6 +1,17 @@
 import { WebsocketEvent } from '@/websocket-event';
+import { OnGatewayInit } from '@nestjs/websockets';
+import { Socket } from 'socket.io';
 import { Serializable } from './serializable';
+import { serialize } from './serialize';
 
-export interface WebsocketEventEmitter<T> {
-  emit(event: WebsocketEvent, payload: Serializable<T>): void;
+export class WebsocketEventEmitter<T> implements OnGatewayInit<Socket> {
+  protected server: Socket;
+
+  afterInit(socket: Socket) {
+    this.server = socket;
+  }
+
+  protected async emit(event: WebsocketEvent, payload: Serializable<T>) {
+    this.server.emit(event, await serialize(payload));
+  }
 }