socket pager

src/modules/pager/pager.gateway.ts

@@ -0,0 +1,139 @@
+import {
+  WebSocketGateway,
+  WebSocketServer,
+  SubscribeMessage,
+  OnGatewayConnection,
+  OnGatewayDisconnect,
+  MessageBody,
+  ConnectedSocket,
+} from '@nestjs/websockets';
+import { Server, Socket } from 'socket.io';
+import { Logger, UseGuards } from '@nestjs/common';
+import { Pager } from './entities/pager.entity';
+import { WsAdminAuthGuard, type AuthenticatedSocket } from './guards/ws-admin-auth.guard';
+import { WsRestId } from './decorators/ws-rest-id.decorator';
+
+@WebSocketGateway({
+  cors: {
+    origin: true,
+    credentials: true,
+  },
+  namespace: '/pager',
+})
+export class PagerGateway implements OnGatewayConnection, OnGatewayDisconnect {
+  @WebSocketServer()
+  server!: Server;
+
+  private readonly logger = new Logger(PagerGateway.name);
+
+  handleConnection(client: Socket) {
+    this.logger.log(`Client connected: ${client.id}`);
+  }
+
+  handleDisconnect(client: Socket) {
+    this.logger.log(`Client disconnected: ${client.id}`);
+  }
+
+  /**
+   * Join a room for a specific restaurant (admin/staff)
+   * Requires admin authentication via JWT token
+   * Restaurant ID is extracted from the authenticated admin's token
+   * Room format: `restaurant:${restId}`
+   */
+  @UseGuards(WsAdminAuthGuard)
+  @SubscribeMessage('join:restaurant')
+  handleJoinRestaurant(@ConnectedSocket() client: AuthenticatedSocket, @WsRestId() restId: string) {
+    const room = `restaurant:${restId}`;
+    void client.join(room);
+    this.logger.log(`Admin ${client.adminId} (Client ${client.id}) joined room: ${room}`);
+    void client.emit('joined', { room, message: 'Successfully joined restaurant room' });
+  }
+
+  /**
+   * Join a room for a specific cookie/session (public user)
+   * Room format: `cookie:${cookieId}`
+   */
+  @SubscribeMessage('join:session')
+  handleJoinSession(@ConnectedSocket() client: Socket, @MessageBody() data: { cookieId: string }) {
+    if (!data.cookieId) {
+      void client.emit('error', { message: 'Cookie ID is required' });
+      return;
+    }
+
+    const room = `cookie:${data.cookieId}`;
+    void client.join(room);
+    this.logger.log(`Client ${client.id} joined room: ${room}`);
+    void client.emit('joined', { room, message: 'Successfully joined session room' });
+  }
+
+  /**
+   * Leave a room
+   */
+  @SubscribeMessage('leave:room')
+  handleLeaveRoom(@ConnectedSocket() client: Socket, @MessageBody() data: { room: string }) {
+    if (data.room) {
+      void client.leave(data.room);
+      this.logger.log(`Client ${client.id} left room: ${data.room}`);
+      void client.emit('left', { room: data.room, message: 'Successfully left room' });
+    }
+  }
+
+  /**
+   * Emit pager created event to restaurant room
+   */
+  emitPagerCreated(pager: Pager) {
+    const room = `restaurant:${pager.restaurant.id}`;
+    this.server.to(room).emit('pager:created', {
+      pager: {
+        id: pager.id,
+        tableNumber: pager.tableNumber,
+        message: pager.message,
+        status: pager.status,
+        createdAt: pager.createdAt,
+        user: pager.user
+          ? {
+              id: pager.user.id,
+              firstName: pager.user.firstName,
+              lastName: pager.user.lastName,
+            }
+          : null,
+      },
+    });
+    this.logger.log(`Emitted pager:created to room: ${room}`);
+  }
+
+  /**
+   * Emit pager status updated event to both restaurant and cookie rooms
+   */
+  emitPagerStatusUpdated(pager: Pager) {
+    const restaurantRoom = `restaurant:${pager.restaurant.id}`;
+    const cookieRoom = `cookie:${pager.cookieId}`;
+
+    const pagerData = {
+      id: pager.id,
+      tableNumber: pager.tableNumber,
+      message: pager.message,
+      status: pager.status,
+      updatedAt: pager.updatedAt,
+      staff: pager.staff
+        ? {
+            id: pager.staff.id,
+            firstName: pager.staff.firstName,
+            lastName: pager.staff.lastName,
+          }
+        : null,
+    };
+
+    // Emit to restaurant room (admin/staff)
+    this.server.to(restaurantRoom).emit('pager:status:updated', {
+      pager: pagerData,
+    });
+    this.logger.log(`Emitted pager:status:updated to room: ${restaurantRoom}`);
+
+    // Emit to cookie room (public user)
+    this.server.to(cookieRoom).emit('pager:status:updated', {
+      pager: pagerData,
+    });
+    this.logger.log(`Emitted pager:status:updated to room: ${cookieRoom}`);
+  }
+}