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 java.io.IOException;
import java.util.Map;

public class CustomAuthenticationSuccessHandler implements AuthenticationSuccessHandler {

	private String redirectUrl;

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

	@Override
	public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication) throws IOException {
//		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());
	}
}