Custom socket room handling in Nest.js

socket-room.service.ts

import { Injectable } from '@nestjs/common';
import { PinoLogger } from 'nestjs-pino';
import { Server, Socket } from 'socket.io';
import { RoomType } from '../../types/socket-room.types';

@Injectable()
export class SocketRoomService {
  private socketServer: Server;

  constructor(private readonly logger: PinoLogger) {}

  setServer(server: Server) {
    this.socketServer = server;
    this.logger.debug({
      msg: '[SocketRoomService] Socket.IO server reference set.',
    });
  }

  getServer(): Server {
    if (!this.socketServer) {
      this.logger.debug({
        msg: '[SocketRoomService] Server not initialized yet.',
      });
    }
    return this.socketServer;
  }

  handleJoinRoom(client: Socket, type: RoomType, correlationId: string) {
    client.join(type);
    this.logger.debug({
      msg: `Client joined a room`,
      clientId: client.id,
      roomType: type,
      correlationId: correlationId,
    });
  }

  handleDisconnect(client: Socket) {
    this.logger.debug({
      msg: `[SocketRoomService] Client ${client.id} disconnected.`,
    });
  }

  emitToRoom(room: RoomType, event: string, payload: any) {
    if (!this.socketServer) {
      this.logger.debug({
        msg: `[SocketRoomService] Cannot emit to room "${room}" — server not initialized.`,
      });
    }
    this.socketServer.to(room).emit(event, payload);
    this.logger.debug({
      msg: `[SocketRoomService] Emitted event "${event}" to room "${room}".`,
    });
  }

  emitToClient(socketId: string, event: string, payload: any) {
    if (!this.socketServer) {
      this.logger.debug({
        msg: `[SocketRoomService] Cannot emit to client "${socketId}" — server not initialized.`,
      });
    }
    this.socketServer.to(socketId).emit(event, payload);
    this.logger.debug({
      msg: `[SocketRoomService] Emitted event "${event}" to client "${socketId}".`,
    });
  }
}