io.imunity.furms.unity.config.WebClientConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of furms-unity-client Show documentation
FURMS Unity DAO/Client module
There is a newer version: 4.3.1
/*
 * Copyright (c) 2020 Bixbit s.c. All rights reserved.
 *  See LICENSE file for licensing information.
 */

package io.imunity.furms.unity.config;

import io.imunity.furms.unity.client.oauth.UnityOauthCredentials;
import io.imunity.furms.unity.client.oauth.UnityOauthProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.http.client.reactive.ReactorClientHttpConnector;
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.web.util.UriUtils;
import reactor.netty.http.client.HttpClient;

import java.nio.charset.StandardCharsets;
import java.util.Base64;

import static java.lang.String.format;
import static org.springframework.http.HttpHeaders.AUTHORIZATION;
import static org.springframework.http.HttpHeaders.CONTENT_TYPE;
import static org.springframework.http.MediaType.APPLICATION_JSON_VALUE;

@Configuration
public class WebClientConfig {
	
	public static final String BASE_UNITY_CLIENT = "baseUnityClient";
	public static final String OAUTH_UNITY_CLIENT = "oauthTokenEndpointWebClient";
	public static final int SEVENTY_FIVE_MB = 75 * 1024 * 1024;

	@Bean
	@Primary
	public WebClient adminWebClient(UnityProperties unityProperties, SslContextManager sslContextManager) {
		String authorizationKey = createAuthorizationKey(unityProperties.getAdminUser(),
				unityProperties.getAdminPassword());
		return createBuilder(unityProperties.getAdminUrl(), sslContextManager)
				.defaultHeader(AUTHORIZATION, authorizationKey)
				.defaultHeader(CONTENT_TYPE, APPLICATION_JSON_VALUE)
				.build();
	}

	@Bean(name=OAUTH_UNITY_CLIENT)
	public WebClient oauthTokenEndpointWebClient(SslContextManager sslContextManager,
	                                             UnityOauthProperties unityOauthProperties,
	                                             UnityOauthCredentials unityOauthCredentials) {
		String authorizationKey = createUrlEncodedAuthorizationKey(unityOauthCredentials.getClientId(), unityOauthCredentials.getClientSecret());
		return createBuilder(unityOauthProperties.getTokenUri(), sslContextManager)
				.defaultHeader(AUTHORIZATION, authorizationKey)
				.defaultHeader(CONTENT_TYPE, APPLICATION_JSON_VALUE)
				.build();
	}
	
	@Bean(name=BASE_UNITY_CLIENT)
	public WebClient webClient(UnityProperties unityProperties, SslContextManager sslContextManager) {
		return createBuilder(unityProperties.getUrl(), sslContextManager)
				.build();
	}
	
	private WebClient.Builder createBuilder(String baseUrl, SslContextManager sslContextManager) {
		HttpClient httpClient = HttpClient.create()
				.secure(sslSpec -> sslSpec.sslContext(sslContextManager.getSslContextForWebClient()));
		return WebClient.builder()
				.clientConnector(new ReactorClientHttpConnector(httpClient))
				.baseUrl(baseUrl)
			.codecs(config -> config
				.defaultCodecs()
				.maxInMemorySize(SEVENTY_FIVE_MB)
			);
	}

	private String createAuthorizationKey(String username, String password) {
		byte[] userPass = format("%s:%s", username, password).getBytes(StandardCharsets.UTF_8);
		String token = Base64.getEncoder().encodeToString(userPass);
		return format("Basic %s", token);
	}

	private String createUrlEncodedAuthorizationKey(String username, String password) {
		String urlEncodedCredentials = UriUtils.encodePath(format("%s:%s", username, password), StandardCharsets.UTF_8);
		String token = Base64.getEncoder().encodeToString(urlEncodedCredentials.getBytes(StandardCharsets.UTF_8));
		return format("Basic %s", token);
	}

}