refactor: JwtOIDCProvider, OauthOIDCHelper 코드 리팩토링

ki-met-hoon · Mar 20, 2024 · 4461050 · 4461050
1 parent 0adebc6
commit 4461050
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 16 deletions.
diff --git a/src/main/java/com/example/pnuunivmiryangcampus/support/token/JwtOIDCProvider.java b/src/main/java/com/example/pnuunivmiryangcampus/support/token/JwtOIDCProvider.java
@@ -1,6 +1,7 @@
 package com.example.pnuunivmiryangcampus.support.token;
 
 import com.example.pnuunivmiryangcampus.auth.OIDCDecodePayload;
+import com.example.pnuunivmiryangcampus.auth.OIDCPublicKeyDto;
 import io.jsonwebtoken.Claims;
 import io.jsonwebtoken.ExpiredJwtException;
 import io.jsonwebtoken.Jws;
@@ -22,27 +23,22 @@
 public class JwtOIDCProvider {
 
     public String getKidFromTokenHeader(String token) {
-
         String KID = "kid";
-        String[] splitToken = token.split("\\.");
-        String header = splitToken[0];
-
-        byte[] decodeJson = Base64.getDecoder().decode(header);
-        String decodeHeader = new String(decodeJson);
+        String encodedHeader = getEncodedHeader(token);
+        String decodedHeader = getDecodedHeader(encodedHeader);
 
         try {
-            JSONObject jsonObject = new JSONObject(decodeHeader);
+            JSONObject jsonObject = new JSONObject(decodedHeader);
             return jsonObject.get(KID).toString();
         } catch (JSONException e) {
             return e.toString();
         }
     }
 
-    public Jws<Claims> getOIDCTokenJws(String token, String modulus, String exponent, String iss, String aud) {
-
+    public Jws<Claims> getOIDCTokenJws(String token, OIDCPublicKeyDto oidcPublicKeyDto, String iss, String aud) {
         try {
             return Jwts.parser()
-                    .verifyWith(getRSAPublicKey(modulus, exponent))
+                    .verifyWith(getRSAPublicKey(oidcPublicKeyDto.n(), oidcPublicKeyDto.e()))
                     .requireAudience(aud)
                     .requireIssuer(iss)
                     .build()
@@ -55,9 +51,8 @@ public Jws<Claims> getOIDCTokenJws(String token, String modulus, String exponent
         }
     }
 
-    public OIDCDecodePayload getOIDCTokenBody(String token, String modulus, String exponent, String iss, String aud) {
-
-        Claims payload = getOIDCTokenJws(token, modulus, exponent, iss, aud).getPayload();
+    public OIDCDecodePayload getOIDCTokenBody(String token, OIDCPublicKeyDto oidcPublicKeyDto, String iss, String aud) {
+        Claims payload = getOIDCTokenJws(token, oidcPublicKeyDto, iss, aud).getPayload();
 
         return new OIDCDecodePayload(
                 payload.getIssuer(),
@@ -66,8 +61,17 @@ public OIDCDecodePayload getOIDCTokenBody(String token, String modulus, String e
                 payload.get("email", String.class));
     }
 
-    private PublicKey getRSAPublicKey(String modulus, String exponent) throws NoSuchAlgorithmException, InvalidKeySpecException {
+    private static String getDecodedHeader(String encodedHeader) {
+        byte[] decodedHeaderBytes = Base64.getDecoder().decode(encodedHeader);
+        return new String(decodedHeaderBytes);
+    }
 
+    private static String getEncodedHeader(String token) {
+        String[] splitToken = token.split("\\.");
+        return splitToken[0];
+    }
+
+    private PublicKey getRSAPublicKey(String modulus, String exponent) throws NoSuchAlgorithmException, InvalidKeySpecException {
         KeyFactory keyFactory = KeyFactory.getInstance("RSA");
         byte[] decodeN = Base64.getUrlDecoder().decode(modulus);
         byte[] decodeE = Base64.getUrlDecoder().decode(exponent);

diff --git a/src/main/java/com/example/pnuunivmiryangcampus/support/token/OauthOIDCHelper.java b/src/main/java/com/example/pnuunivmiryangcampus/support/token/OauthOIDCHelper.java
@@ -24,7 +24,7 @@ private OIDCDecodePayload getPayloadFromIdToken(String token, String iss, String
                         .findFirst()
                         .orElseThrow();
 
-        return jwtOIDCProvider.getOIDCTokenBody(token, oidcPublicKeyDto.n(), oidcPublicKeyDto.e(), iss, aud);
+        return jwtOIDCProvider.getOIDCTokenBody(token, oidcPublicKeyDto, iss, aud);
     }
 
     public OIDCDecodePayload getKakaoOIDCDecodePayload(String token) {
@@ -34,6 +34,7 @@ public OIDCDecodePayload getKakaoOIDCDecodePayload(String token) {
                 token,
                 kakaoProperties.getBaseUrl(),
                 kakaoProperties.getRestApiKey(),
-                oidcPublicKeysResponse);
+                oidcPublicKeysResponse
+        );
     }
 }