package org.springblade.auth.config;

import org.apache.logging.log4j.util.Strings;
import org.springframework.security.core.Authentication;
import org.springframework.security.web.authentication.AuthenticationSuccessHandler;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;
import java.util.Map;

import static org.springframework.security.web.context.HttpSessionSecurityContextRepository.SPRING_SECURITY_CONTEXT_KEY;

public class CustomAuthenticationSuccessHandler implements AuthenticationSuccessHandler {

	private String redirectUrl;

	private String springSecurityContextKey = SPRING_SECURITY_CONTEXT_KEY;

	public CustomAuthenticationSuccessHandler(String redirectUrl) {
		this.redirectUrl = redirectUrl;
	}

	@Override
	public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication) throws IOException {
		// 获取session
//		HttpSession session = request.getSession();
		// 从authentication中获取principal信息
//		Object principal = authentication.getPrincipal();
		// 将principal信息存入session
//		session.setAttribute(this.springSecurityContextKey, principal);
		//		String requestURI = request.getRequestURI();
		Map<String, String[]> parameterMap = request.getParameterMap();
		StringBuilder builder = new StringBuilder();
		if (parameterMap.size()>0){
			builder.append("?");
		}
		String client_id = request.getHeader("client_id");
		String response_type = request.getHeader("response_type");
		String redirect_uri = request.getHeader("redirect_uri");
		// 拼接
		if (!Strings.isBlank(client_id)){
			builder.append("client_id=").append(client_id)
				.append("&")
				.append("response_type=").append(response_type)
				.append("&")
				.append("redirect_uri=").append(redirect_uri);
		}
		// 跳转
		response.sendRedirect(redirectUrl + builder.toString());
	}
}