Spring Security 소셜 로그인 로직 구현하기

@EnableWebSecurity
@RequiredArgsConstructor
public class SecurityConfig {

    @Value("${jwt.token.secret}")
    private String secretKey;


    private final EmployeeService employeeService;

    private final Oauth2SuccessHandler oauth2SuccessHandler;
    private final Oauth2FailureHandler oauth2FailureHandler;

    private final CustomOAuth2UserService customOAuth2UserService;

    @Bean
    public SecurityFilterChain securityFilterChain(HttpSecurity httpSecurity) throws Exception {
        return httpSecurity
                .httpBasic().disable()
                .csrf().disable()
                .cors().and()
                .sessionManagement()
                .sessionCreationPolicy(SessionCreationPolicy.STATELESS)
                .and()

			   ......
            
                .oauth2Login().loginPage("/login")
                .and()
                .oauth2Login()
                .userInfoEndpoint()
                .userService(customOAuth2UserService)
                .and()
                .successHandler(oauth2SuccessHandler)
                .failureHandler(oauth2FailureHandler)
                .and()
            
                .....

                .addFilterBefore(new JwtTokenFilter(employeeService,secretKey), UsernamePasswordAuthenticationFilter.class)
                .build();

    }

}

package com.project.myacademy.global.configuration.oauth;

import lombok.Builder;
import lombok.Getter;

@Builder
@Getter
public class EmployeeProfile {
    //Resource Server마다 제공하는 정보가 다르므로 통일시키기 위한 profile
    private final String name;
    private final String email;

    public EmployeeProfile(String name, String email) {

        this.name = name;
        this.email = email;

    }
}

public enum OAuthAttributes {

    GOOGLE("google", (attributes) -> {
        return new EmployeeProfile(
                (String) attributes.get("name"),
                (String) attributes.get("email")
        );
    }),
    NAVER("naver", (attributes) -> {
        Map<String, Object> response = (Map<String, Object>) attributes.get("response");
        return new EmployeeProfile(
                (String) response.get("name"),
                (String) response.get("email")
        );
    });

    private final String registrationId;
    private final Function<Map<String, Object>, EmployeeProfile> of;

    OAuthAttributes(String registrationId, Function<Map<String, Object>, EmployeeProfile> of) {
        this.registrationId = registrationId;
        this.of = of;
    }

    public static EmployeeProfile extract(String registrationId, Map<String, Object> attributes) {
        return Arrays.stream(values())
                .filter(provider -> registrationId.equals(provider.registrationId))
                .findFirst()
                .orElseThrow(IllegalArgumentException::new)
                .of.apply(attributes);
    }
}

@Service
@RequiredArgsConstructor
@Slf4j
public class CustomOAuth2UserService implements OAuth2UserService<OAuth2UserRequest, OAuth2User> {

    private final EmployeeRepository employeeRepository;
    private final HttpSession httpSession;

    @Value("${jwt.token.secret}")
    private String key;

    @Override
    public OAuth2User loadUser(OAuth2UserRequest userRequest) throws OAuth2AuthenticationException {
        OAuth2UserService<OAuth2UserRequest,OAuth2User> delegate =
                new DefaultOAuth2UserService();

        // 소셜에서 인증받아서 가져온 유저 정보를 담고 있다.
        OAuth2User oAuth2User = delegate.loadUser(userRequest);

        // 어떤 서비스인지(구글, 네이버 등등) -> 로그 찍어보면 naver 혹은 google 이 나온다.
        String registrationId = userRequest.getClientRegistration()
                .getRegistrationId();

        log.info("🌈 소셜 인증한 서비스 [{}]",registrationId);


        // OAuth2 로그인 진행 시 키가 되는 필드 값, 구글은 sub 네이버는 response 라는 이름을 갖는다.
        String userNameAttributeName = userRequest.getClientRegistration()
                .getProviderDetails()
                .getUserInfoEndpoint()
                .getUserNameAttributeName();

        Map<String, Object> attributes = oAuth2User.getAttributes();

        //서비스 마다 다른 키와 벨류 값을 변환하여 객체를 만든다.
        EmployeeProfile employeeProfile = OAuthAttributes.extract(registrationId, attributes);

        // 소셜 로그인으로 들어온 이메일과 같은 이메일로 가입된 회원의 실명 + 이메일이 동일한 경우 승인
        Employee employee = check(employeeProfile);


        // 로그인 성공하면 세션에 회원 실명 저장
        httpSession.setAttribute("name",employee.getName());

        // 해당 계정이 갖고 있는 권한 그대로 주입
        return new DefaultOAuth2User(
                Collections.singleton(new SimpleGrantedAuthority(employee.getEmployeeRole().name())),
                attributes,
                userNameAttributeName);

    }

    // 소셜 로그인 시, 소셜 로그인에 등록된 실명과 이메일로 가입한 회원이 존재하지 않을 경우 null
    private Employee check(EmployeeProfile employeeProfile) {
        Employee employee = employeeRepository.findByNameAndEmail(employeeProfile.getName(), employeeProfile.getEmail())
                .orElseThrow(() -> new OAuth2AuthenticationException("가입된 회원이 아닙니다."));

        return employee;
    }

}

@Component
@Slf4j
@RequiredArgsConstructor
public class Oauth2SuccessHandler extends SimpleUrlAuthenticationSuccessHandler {

    @Value("${jwt.token.secret}")
    private String key;

    @Value("${app.oauth2.authorizedRedirectUri}")
    private String redirectUri;

    private String targetUrl;

    private final EmployeeRepository employeeRepository;

    @Override
    public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication)
            throws IOException, ServletException {

        OAuth2User oAuth2User = (OAuth2User)authentication.getPrincipal();

        // 구글 인증시 아래 로직
        String realName = (String)oAuth2User.getAttribute("name");
        String email = (String)oAuth2User.getAttribute("email");
        log.info("🌈 구글 인증 시 이름 추출 [{}] || 이메일 추출 [{}]",realName,email);

        Map<String, Object> attributes = oAuth2User.getAttributes();

        // 네이버 인증 시 아래 로직
        if(attributes.get("sub")==null){
            Map<String, Object> response2 =(Map<String, Object>) attributes.get("response");
            realName = (String) response2.get("name");
            email = (String) response2.get("email");
            log.info("🌈 네이버 인증 시 이름 추출 [{}] || 이메일 추출 [{}]",realName,email);
        }

        // 이름과 이메일이 둘다 일치하는 회원이 저장되어있을 것
        Employee foundEmployee = employeeRepository.findByNameAndEmail(realName, email).get();
        String foundAccount = foundEmployee.getAccount();
        log.info("🌈 소셜 로그인 인증한 계정명 [{}]",foundAccount);


        // 회원 계정으로 토큰 생성 후 쿼리 파라미터로 보냄
        String token = JwtTokenUtil.createToken(foundAccount,key,1000*60*60);

        targetUrl = UriComponentsBuilder.fromUriString(redirectUri)
                .queryParam("token", token)
                .build().toUriString();

        clearAuthenticationAttributes(request);
        getRedirectStrategy().sendRedirect(request, response, targetUrl);
    }
}

@Component
@Slf4j
@RequiredArgsConstructor
public class Oauth2FailureHandler implements AuthenticationFailureHandler {

    @Override
    public void onAuthenticationFailure(HttpServletRequest request, HttpServletResponse response, AuthenticationException exception) throws IOException, ServletException {
            response.sendRedirect("/oauthFail");
    }
}