Merge pull request #369 from boostcampwm2023/BE-feature/socket-connec…

…tion

소켓 연결 시 인증 추가

nestjs-BE/server/src/board-trees/board-trees.gateway.ts

@@ -1,4 +1,7 @@
+import { UnauthorizedException } from '@nestjs/common';
+import { JwtService } from '@nestjs/jwt';
 import {
+  OnGatewayConnection,
   SubscribeMessage,
   WebSocketGateway,
   WebSocketServer,
@@ -13,12 +16,28 @@ import {
 } from '../crdt/operation';
 
 @WebSocketGateway({ namespace: 'board' })
-export class BoardTreesGateway {
-  constructor(private boardTreesService: BoardTreesService) {}
+export class BoardTreesGateway implements OnGatewayConnection {
+  constructor(
+    private boardTreesService: BoardTreesService,
+    private jwtService: JwtService,
+  ) {}
 
   @WebSocketServer()
   server: Server;
 
+  handleConnection(client: Socket, token: string) {
+    if (!token) {
+      client.disconnect();
+      throw new UnauthorizedException();
+    }
+    try {
+      this.jwtService.verify(token);
+    } catch (error) {
+      client.disconnect();
+      throw new UnauthorizedException();
+    }
+  }
+
   @SubscribeMessage('joinBoard')
   async handleJoinBoard(client: Socket, payload: string) {
     const payloadObject = JSON.parse(payload);

nestjs-BE/server/src/board-trees/board-trees.module.ts

@@ -1,4 +1,5 @@
 import { Module } from '@nestjs/common';
+import { JwtModule } from '@nestjs/jwt';
 import { MongooseModule } from '@nestjs/mongoose';
 import { BoardTreesService } from './board-trees.service';
 import { BoardTreesGateway } from './board-trees.gateway';
@@ -9,6 +10,7 @@ import { BoardTree, BoardTreeSchema } from './schemas/board-tree.schema';
     MongooseModule.forFeature([
       { name: BoardTree.name, schema: BoardTreeSchema },
     ]),
+    JwtModule,
   ],
   providers: [BoardTreesService, BoardTreesGateway],
 })