hotfix: add updatePushAllow

Repick-official · Aug 16, 2024 · e78b3bf · e78b3bf
1 parent bd6abdf
commit e78b3bf
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 4 deletions.
diff --git a/src/main/java/com/example/repick/domain/fcmtoken/entity/UserFcmTokenInfo.java b/src/main/java/com/example/repick/domain/fcmtoken/entity/UserFcmTokenInfo.java
@@ -11,6 +11,7 @@
 public class UserFcmTokenInfo {
     private Long userId;
     private String fcmToken;
+    private Boolean pushAllow = false;
 
     @DynamoDBHashKey
     public Long getUserId() {
@@ -22,10 +23,19 @@ public String getFcmToken() {
         return fcmToken;
     }
 
-    public void update(String fcmToken) {
+    @DynamoDBAttribute
+    public Boolean getPushAllow() {
+        return pushAllow;
+    }
+
+    public void updateFcmToken(String fcmToken) {
         this.fcmToken = fcmToken == null ? this.fcmToken : fcmToken;
     }
 
+    public void updatePushAllow(Boolean pushAllow) {
+        this.pushAllow = pushAllow;
+    }
+
     public void setUserId(Long userId) {
         this.userId = userId;
     }

diff --git a/src/main/java/com/example/repick/domain/fcmtoken/service/UserFcmTokenInfoService.java b/src/main/java/com/example/repick/domain/fcmtoken/service/UserFcmTokenInfoService.java
@@ -19,20 +19,27 @@ public UserFcmTokenInfo getFcmToken(Long userId) {
     }
 
     public void save(Long userId, String fcmToken) {
-        userFcmTokenInfoRepository.save(new UserFcmTokenInfo(userId, fcmToken));
+        userFcmTokenInfoRepository.save(new UserFcmTokenInfo(userId, fcmToken, false));
     }
 
     public void saveOrUpdate(Long userId, String fcmToken) {
 
         UserFcmTokenInfo userFcmTokenInfo = userFcmTokenInfoRepository.findById(userId)
-                .orElse(new UserFcmTokenInfo(userId, fcmToken));
+                .orElse(new UserFcmTokenInfo(userId, fcmToken, false));
 
-        userFcmTokenInfo.update(fcmToken);
+        userFcmTokenInfo.updateFcmToken(fcmToken);
 
         userFcmTokenInfoRepository.save(userFcmTokenInfo);
 
     }
 
+    public void updatePushAllow(Long userId, Boolean pushAllow) {
+        UserFcmTokenInfo userFcmTokenInfo = userFcmTokenInfoRepository.findById(userId)
+                .orElseThrow(() -> new CustomException(USER_FCM_TOKEN_NOT_FOUND));
+        userFcmTokenInfo.updatePushAllow(pushAllow);
+        userFcmTokenInfoRepository.save(userFcmTokenInfo);
+    }
+
     public void delete(Long userId) {
         try {
             userFcmTokenInfoRepository.deleteById(userId);

diff --git a/src/main/java/com/example/repick/domain/user/api/UserController.java b/src/main/java/com/example/repick/domain/user/api/UserController.java
@@ -1,5 +1,6 @@
 package com.example.repick.domain.user.api;
 
+import com.example.repick.domain.fcmtoken.service.UserFcmTokenInfoService;
 import com.example.repick.domain.user.dto.*;
 import com.example.repick.domain.user.service.UserService;
 import com.example.repick.global.jwt.TokenResponse;
@@ -34,6 +35,7 @@ public class UserController {
     private final AppleUserService appleUserService;
     private final NaverUserService naverUserService;
     private final GoogleUserService googleUserService;
+    private final UserFcmTokenInfoService userFcmTokenInfoService;
 
     @Operation(summary = "구글 엑세스 토큰으로 내부 토큰 발급하기",
             description = """
@@ -264,4 +266,17 @@ public SuccessResponse<GetMyPage> getMyPage() {
         return SuccessResponse.success(userService.getMyPage());
     }
 
+    @Operation(summary = "푸시 알림 허용 여부 수정하기",
+            description = """
+                    푸시 알림 허용 여부를 수정합니다.
+                    
+                    요청값:
+                    - (Query Parameter) pushAllow: 푸시 알림 허용 여부
+                    """)
+    @PatchMapping("/pushAllow")
+    public SuccessResponse<Boolean> patchPushAllow(@Parameter(name = "pushAllow", description = "푸시 알림 허용 여부", required = true)
+                                        @RequestParam boolean pushAllow) {
+        return SuccessResponse.success(userService.updatePushAllow(pushAllow));
+    }
+
 }
diff --git a/src/main/java/com/example/repick/domain/user/service/UserService.java b/src/main/java/com/example/repick/domain/user/service/UserService.java
@@ -184,4 +184,13 @@ public GetMyPage getMyPage() {
         // 닉네임, 포인트, 배송 정보
         return GetMyPage.of(user.getNickname(), user.getSettlement(), preparing, shipping, delivered, confirmed);
     }
+
+    public Boolean updatePushAllow(boolean pushAllow) {
+        User user = userRepository.findByProviderId(SecurityContextHolder.getContext().getAuthentication().getName())
+                .orElseThrow(() -> new CustomException(ErrorCode.USER_NOT_FOUND));
+
+        userFcmTokenInfoService.updatePushAllow(user.getId(), pushAllow);
+
+        return true;
+    }
 }