Merge pull request #17 from MeTLaBPJ/dongkyun

[Fix] Merge 충돌 해결

Author: d6nggyun

src/main/java/org/inffy/domain/chat/handler/StompHandler.java

@@ -31,7 +31,7 @@ public Message<?> preSend(Message<?> message, MessageChannel messageChannel){
             String authorization = jwtTokenProvider.getJwtFromStompRequest(accessor);
             jwtTokenProvider.validateStompJwt(authorization);
 
-            Authentication authentication = jwtTokenProvider.getAuthenticationJwt(authorization);
+            Authentication authentication = jwtTokenProvider.getAuthentication(authorization);
             accessor.setUser(authentication);
         }
         return message;

src/main/java/org/inffy/global/security/jwt/util/JwtTokenProvider.java

@@ -7,9 +7,11 @@
 import org.inffy.domain.member.entity.Member;
 import org.inffy.domain.member.repository.MemberRepository;
 import org.inffy.global.exception.entity.RestApiException;
+import org.inffy.global.exception.entity.StompJwtException;
 import org.inffy.global.exception.error.CustomErrorCode;
 import org.springframework.beans.factory.InitializingBean;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
 import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
 import org.springframework.security.core.Authentication;
 import org.springframework.security.core.GrantedAuthority;
@@ -124,4 +126,31 @@ public boolean validateToken(String token) {
             throw new RestApiException(CustomErrorCode.JWT_NOT_VALID); // 유효하지 않은 토큰
         }
     }
+
+    public boolean validateStompJwt(String token) {
+        if(token == null) {
+            throw new JwtException("Jwt AccessToken not found");
+        }
+
+        try {
+            Jwts.parserBuilder()
+                    .setSigningKey(key)
+                    .build()
+                    .parseClaimsJws(token)
+                    .getBody();
+            return true;
+        } catch (ExpiredJwtException e) {
+            throw new StompJwtException(CustomErrorCode.JWT_ACCESS_TOKEN_EXPIRED);
+        } catch (MalformedJwtException e) {
+            throw new StompJwtException(CustomErrorCode.JWT_MALFORMED);
+        } catch (SignatureException | SecurityException e) {
+            throw new StompJwtException(CustomErrorCode.JWT_SIGNATURE);
+        } catch (UnsupportedJwtException e) {
+            throw new StompJwtException(CustomErrorCode.JWT_UNSUPPORTED);
+        }
+    }
+
+    public String getJwtFromStompRequest(final StompHeaderAccessor accessor){
+        return accessor.getFirstNativeHeader("Authorization").substring(7);
+    }
 }