@Configuration
@EnableOAuth2Client
public class OAuth2ClientConfig
{
	@Autowired
	OAuth2ClientContext oauth2ClientContext;

	// 앞서 전장에서 만들었던 계정 서비스입니다.
	@Autowired
	AccountService accountService;

	@Bean
	@ConfigurationProperties("facebook.client")
	AuthorizationCodeResourceDetails facebook()
	{
		return new AuthorizationCodeResourceDetails();
	}

	@Bean
	@ConfigurationProperties("facebook.resource")
	ResourceServerProperties facebookResource()
	{
		return new ResourceServerProperties();
	}

	@Bean
	@ConfigurationProperties("naver.client")
	AuthorizationCodeResourceDetails naver()
	{
		return new AuthorizationCodeResourceDetails();
	}

	@Bean
	@ConfigurationProperties("naver.resource")
	ResourceServerProperties naverResource()
	{
		return new ResourceServerProperties();
	}

	@Bean
	FilterRegistrationBean oauth2ClientFilterRegistration(OAuth2ClientContextFilter filter)
	{
		FilterRegistrationBean registration = new FilterRegistrationBean();
		registration.setFilter(filter);
		// 스프링 사이트에 의하면 다른 필터보다 우선순위를 올리기위해 -100을 주었다고 나옵니다.
		registration.setOrder(-100);
		return registration;
	}

	@Bean("sso.filter")
	Filter ssoFilter()
	{
		List<Filter> filters = new ArrayList<>();

		// 페이스북
		OAuth2ClientAuthenticationProcessingFilter facebook
			= new OAuth2ClientAuthenticationProcessingFilter("/sign-in/facebook");
		facebook.setRestTemplate(new OAuth2RestTemplate(facebook(), oauth2ClientContext));
		facebook.setTokenServices(new UserTokenServices(facebookResource().getUserInfoUri(), facebook().getClientId()));
		facebook.setAuthenticationSuccessHandler(new OAuth2SuccessHandler("facebook", accountService));
		filters.add(facebook);

		// 네이버
		OAuth2ClientAuthenticationProcessingFilter naver
			= new OAuth2ClientAuthenticationProcessingFilter("/sign-in/naver");
		naver.setRestTemplate(new OAuth2RestTemplate(naver(), oauth2ClientContext));
		naver.setTokenServices(new UserTokenServices(naverResource().getUserInfoUri(), naver().getClientId()));
		naver.setAuthenticationSuccessHandler(new OAuth2SuccessHandler("naver", accountService));
		filters.add(naver);

		CompositeFilter filter = new CompositeFilter();
		filter.setFilters(filters);
		return filter;
	}
}

public class UserTokenServices extends UserInfoTokenServices
{
	public UserTokenServices(String userInfoEndpointUrl, String clientId)
	{
		super(userInfoEndpointUrl, clientId);
		setAuthoritiesExtractor(new OAuth2AuthoritiesExtractor());
	}

	public static class OAuth2AuthoritiesExtractor implements AuthoritiesExtractor
	{
		@Override
		public List<GrantedAuthority> extractAuthorities(Map<String, Object> map)
		{
			return AuthorityUtils.createAuthorityList(RoleType.OAUTH.toString());
		}
	}
}

public class OAuth2SuccessHandler implements AuthenticationSuccessHandler
{
	final String type;
	final AccountService accountService;

	public OAuth2SuccessHandler(String type, AccountService accountService)
	{
		this.type = type;
		this.accountService = accountService;
	}

	@Override
	public void onAuthenticationSuccess(HttpServletRequest req, HttpServletResponse res, Authentication auth)
			throws IOException, ServletException
	{
		// 연결되어 있는 계정이 있는지 확인합니다.
		// 이전 강의 AccountService.getAccountByOAuthId 참고!!
		Account account = accountService.getAccountByOAuthId(type, auth.getName());

		// 연결되어 있는 계정이 있는경우.
		if (account != null)
		{
			// 기존 인증을 바꿉니다.
			// 이전 강의의 일반 로그인과 동일한 방식으로 로그인한 것으로 간주하여 처리합니다.
			// 기존 인증이 날아가기 때문에 OAUTH ROLE은 증발하며, USER ROLE 이 적용됩니다.
			SecurityContextHolder.getContext().setAuthentication(new UsernamePasswordAuthenticationToken
			(
				new UserDetails(account), null, UserDetails.getAuthorities(account.getRoles()))
			);
			res.sendRedirect("/");
		}
		// 연결된 계정이 없는경우
		else
		{
			// 회원가입 페이지로 보냅니다.
			// ROLE 은 OAUTH 상태입니다.
			res.sendRedirect("/sign-up/oauth");

			// 특별히 추가정보를 받아서 가입해야할 일이없다면,
			// 즉석으로 계정을 생성한 후 성공처리 해준다면 사용자 입장에서 좋을 것 같습니다.
		}
	}
}

@RequestMapping(path="/test", produces="text/plain")
@ResponseBody
String test(Authentication authentication)
{
	return authentication;
}