From a7ddfb5647d1af9611096d593b9042a4a21de282 Mon Sep 17 00:00:00 2001
From: toychip <dlawnsgud427@naver.com>
Date: Wed, 29 Nov 2023 19:15:09 +0900
Subject: [PATCH] =?UTF-8?q?feat:=20JwtToken=EC=9C=BC=EB=A1=9C=EB=B6=80?=
 =?UTF-8?q?=ED=84=B0=20=EC=9C=A0=EC=A0=80=EC=9D=98=20=EC=A0=95=EB=B3=B4=20?=
 =?UTF-8?q?=EC=B6=94=EC=B6=9C=20=ED=9B=84=20=EA=B2=80=EC=A6=9D=20=EA=B5=AC?=
 =?UTF-8?q?=ED=98=84=20(#7)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../api/TaveShot/global/jwt/JwtProvider.java  | 47 +++++++++++++++++--
 1 file changed, 44 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/api/TaveShot/global/jwt/JwtProvider.java b/src/main/java/com/api/TaveShot/global/jwt/JwtProvider.java
index 2f40702..1b1a4de 100644
--- a/src/main/java/com/api/TaveShot/global/jwt/JwtProvider.java
+++ b/src/main/java/com/api/TaveShot/global/jwt/JwtProvider.java
@@ -2,24 +2,35 @@
 
 import static com.api.TaveShot.global.constant.OauthConstant.ACCESS_TOKEN_VALID_TIME;
 
+import com.api.TaveShot.domain.Member.domain.Member;
+import com.api.TaveShot.domain.Member.repository.MemberRepository;
 import io.jsonwebtoken.Claims;
 import io.jsonwebtoken.ExpiredJwtException;
 import io.jsonwebtoken.Jwts;
 import io.jsonwebtoken.SignatureAlgorithm;
 import io.jsonwebtoken.security.Keys;
 import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
 import java.util.Date;
 import javax.crypto.SecretKey;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
+import org.springframework.security.core.context.SecurityContextHolder;
 import org.springframework.stereotype.Component;
 
 @Component
+@Slf4j
+@RequiredArgsConstructor
 public class JwtProvider {
 
+    private final MemberRepository memberRepository;
+
     @Value("${jwt.secret.key}")
     private String SECRET_KEY;
 
-    public String generateAccessToken(String id) {
+    public String generateJwtToken(String id) {
         Claims claims = createClaims(id);
         Date now = new Date();
         long expiredDate = calculateExpirationDate(now);
@@ -48,13 +59,13 @@ private SecretKey generateKey() {
     }
 
     // 토큰의 유효성 검사
-    public void isValidToken(String token) {
+    public void isValidToken(String jwtToken) {
         try {
             SecretKey key = generateKey();
             Jwts.parserBuilder()
                     .setSigningKey(key)
                     .build()
-                    .parseClaimsJws(token);
+                    .parseClaimsJws(jwtToken);
 
         } catch (ExpiredJwtException e) { // 어세스 토큰 만료
             throw new IllegalArgumentException("Access Token expired");
@@ -62,4 +73,34 @@ public void isValidToken(String token) {
             throw new IllegalArgumentException("User Not Authorized");
         }
     }
+
+    public void getAuthenticationFromToken(String jwtToken) {
+
+        Long userId = Long.valueOf(getUserIdFromToken(jwtToken));
+        Member findMember = memberRepository.findById(userId).orElseThrow(() -> new RuntimeException("token 으로 Member를 찾을 수 없음"));
+        String gitName = findMember.getGitName();
+
+        log.info("-------------- getAuthenticationFromToken 어세스토큰: " + jwtToken);
+
+        // JWT 토큰이 유효하면, 사용자 정보를 연결 세션에 추가
+        UsernamePasswordAuthenticationToken authenticationToken =
+                new UsernamePasswordAuthenticationToken(gitName, jwtToken, new ArrayList<>());
+
+        SecurityContextHolder.getContext().setAuthentication(authenticationToken);
+
+    }
+
+    // 토큰에서 유저 아이디 얻기
+    public String getUserIdFromToken(String jwtToken) {
+        SecretKey key = generateKey();
+
+        Claims claims = Jwts.parserBuilder()
+                .setSigningKey(key)
+                .build()
+                .parseClaimsJws(jwtToken)
+                .getBody();
+
+        log.info("-------------- JwtProvider.getUserIdFromAccessToken: " + claims.getSubject());
+        return claims.getSubject();
+    }
 }