feat: SecurityUtil CurrentMember method (#11)

현재 사용자의 정보를 가져오는 SecurityUtil, currentMember method 생성
Tave100Shot · Dec 14, 2023 · 38747b4 · 38747b4
1 parent f614881
commit 38747b4
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 4 deletions.
diff --git a/src/main/java/com/api/TaveShot/global/exception/ErrorType.java b/src/main/java/com/api/TaveShot/global/exception/ErrorType.java
@@ -34,8 +34,9 @@ public enum ErrorType {
 
 
     // ------------------------------------------ USER ------------------------------------------
-    _USER_NOT_FOUND(NOT_FOUND, "USER_4040", "제공된 토큰으로 사용자를 찾을 수 없습니다.")
-
+    _USER_NOT_FOUND_BY_TOKEN(NOT_FOUND, "USER_4040", "제공된 토큰으로 사용자를 찾을 수 없습니다."),
+    _UNAUTHORIZED(UNAUTHORIZED, "USER_4010", "로그인되지 않은 상태입니다."),
+    _USER_NOT_FOUND_DB(NOT_FOUND, "USER_4041", "존재하지 않는 회원입니다.")
     ;
 
     private final HttpStatus status;

diff --git a/src/main/java/com/api/TaveShot/global/jwt/JwtProvider.java b/src/main/java/com/api/TaveShot/global/jwt/JwtProvider.java
@@ -3,7 +3,7 @@
 import static com.api.TaveShot.global.constant.OauthConstant.ACCESS_TOKEN_VALID_TIME;
 import static com.api.TaveShot.global.exception.ErrorType._JWT_EXPIRED;
 import static com.api.TaveShot.global.exception.ErrorType._JWT_PARSING_ERROR;
-import static com.api.TaveShot.global.exception.ErrorType._USER_NOT_FOUND;
+import static com.api.TaveShot.global.exception.ErrorType._USER_NOT_FOUND_BY_TOKEN;
 
 import com.api.TaveShot.domain.Member.repository.MemberRepository;
 import com.api.TaveShot.global.exception.ApiException;
@@ -87,7 +87,7 @@ public void getAuthenticationFromToken(final String jwtToken) {
     private void getGitLoginId(final String jwtToken) {
         Long userId = Long.valueOf(getUserIdFromToken(jwtToken));
         memberRepository.findById(userId)
-                .orElseThrow(() -> new ApiException(_USER_NOT_FOUND));
+                .orElseThrow(() -> new ApiException(_USER_NOT_FOUND_BY_TOKEN));
     }
 
     // 토큰에서 유저 아이디 얻기

diff --git a/src/main/java/com/api/TaveShot/global/oauth2/CustomOauth2User.java b/src/main/java/com/api/TaveShot/global/oauth2/CustomOauth2User.java
@@ -36,4 +36,8 @@ public Collection<? extends GrantedAuthority> getAuthorities() {
     public String getName() {
         return member.getGitLoginId();
     }
+
+    public Member getMember() {
+        return this.member;
+    }
 }
diff --git a/src/main/java/com/api/TaveShot/global/util/SecurityUtil.java b/src/main/java/com/api/TaveShot/global/util/SecurityUtil.java
@@ -0,0 +1,27 @@
+package com.api.TaveShot.global.util;
+
+import com.api.TaveShot.domain.Member.domain.Member;
+import com.api.TaveShot.global.exception.ApiException;
+import com.api.TaveShot.global.exception.ErrorType;
+import com.api.TaveShot.global.security.oauth2.CustomOauth2User;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+
+public class SecurityUtil {
+
+    public static Member getCurrentMember() {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+
+        if (authentication == null || !authentication.isAuthenticated()) {
+            throw new ApiException(ErrorType._UNAUTHORIZED);
+        }
+
+        Object principal = authentication.getPrincipal();
+
+        if (principal instanceof CustomOauth2User) {
+            return ((CustomOauth2User) principal).getMember();
+        }
+
+        throw new ApiException(ErrorType._USER_NOT_FOUND_DB);
+    }
+}