Spring Security 안쓰고 AOP로 OAuth 구현하기 - 2

/** * 관리자 인증을 수행하는 컨트롤러 클래스입니다. */ @RestController @RequestMapping("/api/admin/auth") @RequiredArgsConstructor public class AdminAuthController { private final TokenProvider tokenProvider; private final OauthService oauthService; private final CookieManager cookieManager; private final SiteUrlProperties siteUrlProperties; private final GoogleApiProperties googleApiProperties; private final JwtProperties jwtProperties; /** * 구글 로그인 페이지로 리다이렉트합니다. * * @param response 요청 시의 서블렛 {@link HttpServletResponse} * @throws IOException 입출력 예외 */ @GetMapping("/login") public void login(HttpServletResponse response) throws IOException { oauthService.sendToGoogleApi(response); } /** * 구글 로그인 성공 시에 콜백을 처리합니다. * <br> * 구글 API로부터 받은 인증 코드를 이용하여 구글 API에게 인증 토큰을 요청하고, * <br> * 인증 토큰을 이용하여 구글 API에게 프로필 정보를 요청합니다. * <br> * 프로필 정보를 이용하여 JWT 토큰을 생성하고, JWT 토큰을 쿠키에 저장합니다. * <br> * 완료되면, 프론트엔드의 메인 화면으로 리다이렉트합니다. * * @param code 구글 API로부터 쿼리로 받은 인증 코드 * @param req 요청 시의 서블렛 {@link HttpServletRequest} * @param res 요청 시의 서블렛 {@link HttpServletResponse} * @throws IOException 입출력 예외 */ @GetMapping("/login/callback") public void loginCallback(@RequestParam String code, HttpServletRequest req, HttpServletResponse res) throws IOException { String apiToken = oauthService.getGoogleToken(code); JsonNode profile = oauthService.getGoogleProfile(apiToken); String accessToken = tokenProvider.createToken(googleApiProperties.getProviderName(), profile, DateUtil.getNow()); String serverName = req.getServerName(); cookieManager.setCookie(res, jwtProperties.getAdminTokenName(), accessToken, "/", serverName); res.sendRedirect(siteUrlProperties.getFeHost() + "/main"); } }

@Component @Getter public class GoogleApiProperties { @Value("${oauth2.client.registration.google.name}") private String providerName; @Value("${oauth2.client.registration.google.client-id}") private String clientId; @Value("${oauth2.client.registration.google.client-secret}") private String clientSecret; //... 생략 ...

/** * OAuth를 수행하는 서비스 클래스입니다. */ @Service @RequiredArgsConstructor public class OauthService { private final GoogleApiProperties googleApiProperties; private final FtApiProperties ftApiProperties; /** * 구글 OAuth 인증을 위한 URL을 생성하고, HttpServletResponse에 리다이렉트합니다. * * @param response {@link HttpServletResponse} * @throws IOException 입출력 예외 */ public void sendToGoogleApi(HttpServletResponse response) throws IOException { response.sendRedirect( ApiUriBuilder .builder() .authUri(googleApiProperties.getAuthUri()) .clientId(googleApiProperties.getClientId()) .redirectUri(googleApiProperties.getRedirectUri()) .scope(googleApiProperties.getScope()) .grantType(googleApiProperties.getGrantType()) .build() .getCodeRequestUri()); } /** * 구글 OAuth 인증을 위한 토큰을 요청합니다. * * @param code 인증 코드 * @return API 액세스 토큰 * @throws ServiceException API 요청에 에러가 반환됐을 때 발생하는 예외 */ public String getGoogleToken(String code) { ObjectMapper objectMapper = new ObjectMapper(); RestTemplate restTemplate = new RestTemplate(); HttpHeaders headers = new HttpHeaders(); headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED); MultiValueMap<String, String> map = new LinkedMultiValueMap<>(); map.add("grant_type", "authorization_code"); map.add("client_id", googleApiProperties.getClientId()); map.add("client_secret", googleApiProperties.getClientSecret()); map.add("redirect_uri", googleApiProperties.getRedirectUri()); map.add("code", code); HttpEntity<MultiValueMap<String, String>> request = new HttpEntity<>(map, headers); try { ResponseEntity<String> response = restTemplate.postForEntity( googleApiProperties.getTokenUri(), request, String.class); return objectMapper.readTree(response.getBody()) .get(googleApiProperties.getAccessTokenName()).asText(); } catch (Exception e) { throw new ServiceException(ExceptionStatus.OAUTH_BAD_GATEWAY); } } /** * 구글 OAuth 인증을 통해 받은 토큰을 이용해 사용자 정보를 요청합니다. * * @param token 토큰 * @return 사용자 정보 * @throws ServiceException API 요청에 에러가 반환됐을 때 발생하는 예외 */ public JsonNode getGoogleProfile(String token) { ObjectMapper objectMapper = new ObjectMapper(); RestTemplate restTemplate = new RestTemplate(); HttpHeaders headers = new HttpHeaders(); headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED); headers.setBearerAuth(token); HttpEntity<MultiValueMap<String, String>> request = new HttpEntity<>(null, headers); try { ResponseEntity<String> response = restTemplate.exchange( googleApiProperties.getUserInfoUri(), HttpMethod.GET, request, String.class); return objectMapper.readTree(response.getBody()); } catch (Exception e) { throw new ServiceException(ExceptionStatus.OAUTH_BAD_GATEWAY); } // ... 42 인증은 생략 ...

/** * API 제공자에 따라 JWT 토큰을 생성하는 클래스입니다. */ @Component @RequiredArgsConstructor public class TokenProvider { private final JwtProperties jwtProperties; private final GoogleApiProperties googleApiProperties; private final FtApiProperties ftApiProperties; /** * JWT 토큰에 담을 클레임(Payload)을 생성합니다. * * @param provider API 제공자 이름 * @param profile API 제공자로부터 받은 프로필 * @return JWT 클레임(Payload) */ public Map<String, Object> makeClaimsByProviderProfile(String provider, JsonNode profile) { Map<String, Object> claims = new HashMap<>(); if (provider.equals(googleApiProperties.getProviderName())) { claims.put("email", profile.get("email").asText()); } if (provider.equals(ftApiProperties.getProviderName())) { claims.put("name", profile.get("login").asText()); claims.put("email", profile.get("email").asText()); claims.put("blackholedAt", profile.get("cursus_users").get(1).get("blackholed_at") != null ? profile.get("cursus_users").get(1).get("blackholed_at").asText() : null); claims.put("role", UserRole.USER); } return claims; } /** * JWT 토큰을 생성합니다. * * @param provider API 제공자 이름 * @param profile API 제공자로부터 받은 프로필 * @param now 현재 시각 * @return JWT 토큰 */ public String createToken(String provider, JsonNode profile, Date now) { return Jwts.builder() .setClaims(makeClaimsByProviderProfile(provider, profile)) .signWith(jwtProperties.getSigningKey(), SignatureAlgorithm.HS256) .setExpiration(DateUtil.addDaysToDate(now, jwtProperties.getExpiry())) .compact(); } }

/** * 토큰의 유효성을 검사하는 클래스입니다. */ @Component @RequiredArgsConstructor @Slf4j public class TokenValidator { private final JwtProperties jwtProperties; /** * 토큰의 유효성을 검사합니다. * <br> * 매 요청시 헤더에 Bearer 토큰으로 인증을 시도하기 때문에, * <br> * 헤더에 bearer 방식으로 유효하게 토큰이 전달되었는지 검사합니다. * * @param req {@link HttpServletRequest} * @return 정상적인 방식의 토큰 요청인지, 유효한 토큰인지 여부 */ public Boolean isTokenValid(HttpServletRequest req) { String authHeader = req.getHeader("Authorization"); if (authHeader == null || authHeader.startsWith("Bearer ") == false) { return false; } String token = authHeader.substring(7); if (token == null || checkTokenValidity(token) == false) { return false; } return true; } /** * 토큰의 유효성을 검사합니다. * <br> * JWT ParseBuilder의 parseClaimJws를 통해 토큰을 검사합니다. * <br> * 만료되었거나, 잘못된(위, 변조된) 토큰이거스나, 지원되지 않는 토큰이면 false를 반환합니다. * * @param token 검사할 토큰 * @return 토큰이 만료되거나 유효한지 아닌지 여부 */ public Boolean checkTokenValidity(String token) { try { Jwts.parserBuilder().setSigningKey(jwtProperties.getSigningKey()).build() .parseClaimsJws(token); return true; } catch (MalformedJwtException e) { log.info("잘못된 JWT 서명입니다."); } catch (ExpiredJwtException e) { log.info("만료된 JWT 토큰입니다."); } catch (UnsupportedJwtException e) { log.info("지원되지 않는 JWT 토큰입니다."); } catch (IllegalArgumentException e) { log.info("JWT 토큰이 잘못되었습니다."); } return false; } public JsonNode getPayloadJson(final String token) throws JsonProcessingException { ObjectMapper objectMapper = new ObjectMapper(); final String payloadJWT = token.split("\\.")[1]; Base64.Decoder decoder = Base64.getUrlDecoder(); return objectMapper.readTree(new String(decoder.decode(payloadJWT))); } }