프리미엄 여부 조회

JaeMyeongSon · Dec 18, 2023 · f1ae464 · f1ae464
1 parent c57d110
commit f1ae464
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/users/dto/users.dto.ts b/src/users/dto/users.dto.ts
@@ -2,7 +2,7 @@ import { ApiProperty } from '@nestjs/swagger';
 import { User } from '../../entities/user';
 
 export class UsersDto {
-  static fromEntity({ id, email, rateLimit }: User) {
+  static fromEntity({ id, email, rateLimit, isPremium }: User) {
     const userDto = new UsersDto();
 
     userDto.id = id;
@@ -39,4 +39,11 @@ export class UsersDto {
     description: '현재 사용한 채팅 횟수',
   })
   public currentUsedCount: number;
+
+  @ApiProperty({
+    required: true,
+    example: true,
+    description: '프리미엄 유무',
+  })
+  public isPremium: boolean;
 }
diff --git a/src/users/users.controller.ts b/src/users/users.controller.ts
@@ -62,6 +62,7 @@ export class UsersController {
 
     const userDto = UsersDto.fromEntity(user);
 
+    userDto.isPremium = await this.usersService.existsPremium(userId);
     userDto.currentUsedCount = usedChatCount;
 
     return userDto;