feat(#60): add interceptors for set success response type in chat-con…

…troller and socket-gateway

src/chat/controllers/chat.controller.ts

@@ -31,9 +31,11 @@ import { JwtAccessTokenGuard } from 'src/config/guards/jwt-access-token.guard';
 import { GetNotificationsResponseDto } from '../dto/get-notifications-response.dto';
 import { ApiGetChatNotifications } from '../swagger-decorators/get-chat-notifications.decorator';
 import { ApiCreateChatImage } from '../swagger-decorators/create-chat-image.decorators';
+import { SuccessResponseInterceptor } from 'src/common/interceptors/success-response.interceptor';
 
 @ApiTags('CHAT')
 @UsePipes(ValidationPipe)
+@UseInterceptors(SuccessResponseInterceptor)
 @Controller('chat-room')
 export class ChatController {
   constructor(

src/chat/events/events.gateway.ts

@@ -15,12 +15,14 @@ import { AsyncApiSub } from 'nestjs-asyncapi';
 import {
   BadRequestException,
   UseFilters,
+  UseGuards,
   UsePipes,
   ValidationPipe,
 } from '@nestjs/common';
 import { LoginChatRoomDto } from '../dto/login-chat-room.dto';
 import { WebSocketExceptionFilter } from '../exceptions/websocket-exception.filter';
 import mongoose from 'mongoose';
+import { WebSocketJwtAccessTokenGuard } from 'src/config/guards/web-socket-jwt-access-token.guard';
 
 @WebSocketGateway({ namespace: /\/ch-.+/, cors: true })
 @UsePipes(ValidationPipe)
@@ -78,17 +80,20 @@ export class EventsGateway
       payload: PostChatDto,
     },
   })
+  @UseGuards(WebSocketJwtAccessTokenGuard)
   @SubscribeMessage('message')
   async handleMessage(
     @MessageBody() postChatDto: PostChatDto,
     @ConnectedSocket() socket: Socket,
   ) {
     if (postChatDto.hasOwnProperty('content')) {
       const returnedChat = await this.chatService.createChat(postChatDto);
-      socket.to(postChatDto.roomId.toString()).emit('message', returnedChat);
+      const data = returnedChat;
+      socket.to(postChatDto.roomId.toString()).emit('message', { data });
     } else {
       const returnedChat = await this.chatService.findChatImage(postChatDto);
-      socket.to(postChatDto.roomId.toString()).emit('message', returnedChat);
+      const data = returnedChat;
+      socket.to(postChatDto.roomId.toString()).emit('message', { data });
     }
   }
 

src/chat/services/chat.service.ts

@@ -139,15 +139,15 @@ export class ChatService {
       receiverId,
     );
 
-    const chat = {
+    const data = {
       content: returnedChat.content,
       sender: returnedChat.sender,
       receiver: returnedChat.receiver,
     };
 
     if (returnedChat) this.subject.next(returnedChat);
 
-    return chat;
+    return data;
   }
 
   async createChatImage(

src/common/interceptors/success-response.interceptor.ts

@@ -0,0 +1,21 @@
+import {
+  Injectable,
+  NestInterceptor,
+  ExecutionContext,
+  CallHandler,
+} from '@nestjs/common';
+import { Observable } from 'rxjs';
+import { map } from 'rxjs/operators';
+
+@Injectable()
+export class SuccessResponseInterceptor implements NestInterceptor {
+  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
+    const statusCode = context.getArgByIndex(1).statusCode;
+    return next.handle().pipe(
+      map((data) => ({
+        statusCode,
+        data,
+      })),
+    );
+  }
+}

src/common/s3/s3.service.ts

@@ -44,6 +44,7 @@ export class S3Service {
 
       return { url: fileUrl, key: filename };
     } catch (error) {
+      console.error(error);
       throw new Error('S3 업로드 오류');
     }
   }

src/config/guards/web-socket-jwt-access-token.guard.ts

@@ -0,0 +1,22 @@
+import { ExecutionContext, Injectable } from '@nestjs/common';
+import { TokenService } from 'src/auth/services/token.service';
+
+@Injectable()
+export class WebSocketJwtAccessTokenGuard {
+  constructor(private tokenService: TokenService) {}
+
+  async canActivate(context: ExecutionContext) {
+    const request = context.switchToWs().getClient().handshake;
+    const accessToken = request.headers['access_token'];
+
+    if (!accessToken) {
+      return false;
+    }
+
+    const userId = await this.tokenService.decodeToken(accessToken);
+    request.user = { userId };
+    console.log(request.user);
+
+    return true;
+  }
+}