FIRE-156 Add screenshare gateway

package.json

@@ -43,6 +43,7 @@
     "@types/jest": "27.5.0",
     "@types/node": "^16.0.0",
     "@types/supertest": "^2.0.11",
+    "@types/ws": "^8.5.3",
     "@typescript-eslint/eslint-plugin": "^5.0.0",
     "@typescript-eslint/parser": "^5.0.0",
     "eslint": "^8.0.1",

src/app.module.ts

@@ -4,6 +4,7 @@ import { AppController } from './app.controller';
 import { AppService } from './app.service';
 import { RoomsController } from './rooms/rooms.controller';
 import { RoomsService } from './rooms/rooms.service';
+import { ScreenShareGateway } from './gateways/screenshare.gateway';
 
 @Module({
   imports: [
@@ -13,6 +14,6 @@ import { RoomsService } from './rooms/rooms.service';
     }),
   ],
   controllers: [AppController, RoomsController],
-  providers: [AppService, RoomsService],
+  providers: [AppService, RoomsService, ScreenShareGateway],
 })
 export class AppModule {}

src/gateways/screenshare.gateway.ts

@@ -0,0 +1,88 @@
+import {
+  OnGatewayDisconnect,
+  OnGatewayInit,
+  SubscribeMessage,
+  WebSocketGateway,
+  WebSocketServer,
+} from '@nestjs/websockets';
+
+import { Logger } from '@nestjs/common';
+import { Server } from 'ws';
+import { Socket } from 'socket.io';
+
+@WebSocketGateway({ namespace: 'screen-share' })
+export class ScreenShareGateway implements OnGatewayInit, OnGatewayDisconnect {
+  @WebSocketServer()
+  server: Server;
+
+  private activeSockets: { room: string; id: string }[] = [];
+
+  private logger: Logger = new Logger('ScreenShareGateway');
+
+  @SubscribeMessage('joinRoom')
+  public joinRoom(client: Socket, room: string): void {
+    const existingSocket = this.activeSockets?.find(
+      (socket) => socket.room === room && socket.id === client.id,
+    );
+
+    if (!existingSocket) {
+      this.activeSockets = [...this.activeSockets, { id: client.id, room }];
+      client.emit(`${room}-update-user-list`, {
+        users: this.activeSockets
+          .filter((socket) => socket.room === room && socket.id !== client.id)
+          .map((existingSocket) => existingSocket.id),
+        current: client.id,
+      });
+
+      client.broadcast.emit(`${room}-add-user`, {
+        user: client.id,
+      });
+    }
+    return this.logger.log(`Client ${client.id} joined ${room}`);
+  }
+
+  @SubscribeMessage('call-user')
+  public callUser(client: Socket, data: any): void {
+    client.to(data.to).emit('call-made', {
+      offer: data.offer,
+      socket: client.id,
+    });
+  }
+
+  @SubscribeMessage('make-answer')
+  public makeAnswer(client: Socket, data: any): void {
+    client.to(data.to).emit('answer-made', {
+      socket: client.id,
+      answer: data.answer,
+    });
+  }
+
+  @SubscribeMessage('reject-call')
+  public rejectCall(client: Socket, data: any): void {
+    client.to(data.from).emit('call-rejected', {
+      socket: client.id,
+    });
+  }
+
+  public afterInit(): void {
+    this.logger.log('Init');
+  }
+
+  public handleDisconnect(client: Socket): void {
+    const existingSocket = this.activeSockets.find(
+      (socket) => socket.id === client.id,
+    );
+
+    if (!existingSocket) return;
+
+    this.activeSockets = this.activeSockets.filter(
+      (socket) => socket.id !== client.id,
+    );
+
+    client.broadcast.emit(`${existingSocket.room}-remove-user`, {
+      socketId: client.id,
+    });
+
+    this.logger.log(`Client disconnected: ${client.id}`);
+  }
+}

yarn.lock

@@ -1089,6 +1089,13 @@
   dependencies:
     "@types/superagent" "*"
 
+"@types/ws@^8.5.3":
+  version "8.5.3"
+  resolved "https://registry.yarnpkg.com/@types/ws/-/ws-8.5.3.tgz#7d25a1ffbecd3c4f2d35068d0b283c037003274d"
+  integrity sha512-6YOoWjruKj1uLf3INHH7D3qTXwFfEsg1kf3c0uDdSBJwfa/llkwIjrAGV7j7mVgGNbzTQ3HiHKKDXl6bJPD97w==
+  dependencies:
+    "@types/node" "*"
+
 "@types/yargs-parser@*":
   version "21.0.0"
   resolved "https://registry.yarnpkg.com/@types/yargs-parser/-/yargs-parser-21.0.0.tgz#0c60e537fa790f5f9472ed2776c2b71ec117351b"