refactor: Update UserController to update user's FCM token

adego-project · Jul 19, 2024 · 70b5433 · 70b5433
1 parent 7d48759
commit 70b5433
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/modules/user/user.controller.ts b/src/modules/user/user.controller.ts
@@ -75,7 +75,7 @@ export class UserController {
         @Body() data: UpdateUserFCMTokenDTO,
     ): Promise<ResponseDTO<null>> {
         try {
-            await this.userService.updateUserById(user.id, data);
+            await this.userService.updateUserFCMTokenById(user.id, data.FCMToken);
             return { status: 'success', data: null };
         } catch (error) {
             return { status: 'error', data: null };

diff --git a/src/modules/user/user.service.ts b/src/modules/user/user.service.ts
@@ -33,6 +33,16 @@ export class UserService {
         return data;
     }
 
+    async updateUserFCMTokenById(id: string, FCMToken: string) {
+        const user = await this.prisma.user.findUnique({ where: { id } });
+        if (!user) throw new HttpException('User not found', HttpStatus.NOT_FOUND);
+
+        await this.prisma.user.update({
+            where: { id },
+            data: { FCMToken },
+        });
+    }
+
     /**
      * Update user profile image by id
      * @param id User id