com.c4_soft.springaddons.rest.SpringAddonsWebClientSupport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-addons-starter-rest Show documentation
The newest version!
package com.c4_soft.springaddons.rest;

import java.util.Optional;

import org.springframework.security.oauth2.client.OAuth2AuthorizedClientManager;
import org.springframework.web.reactive.function.client.ClientRequest;
import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
import org.springframework.web.reactive.function.client.ExchangeFunction;
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.web.service.annotation.HttpExchange;

/**
 * 
 * Provides with {@link WebClient} builder instances pre-configured with:
 * 
 * 
 * HTTP conector if proxy properties or environment variables are set
 * Base URL
 * authorization exchange function if Basic or OAuth2 Bearer
 * 
 * 
 * 

 * Also provides with helper methods to get {@link HttpExchange @@HttpExchange} proxies with {@link WebClient}
 * 
 * 
 * /!\ Auto-configured only in servlet (WebMVC) applications and only if some {@link SpringAddonsRestProperties} are present /!\
 * 
 *
 * @author Jerome Wacongne chl4mp@c4-soft.com
 * @see    ReactiveSpringAddonsWebClientSupport an equivalent for reactive (Webflux) applications
 */
public class SpringAddonsWebClientSupport extends AbstractSpringAddonsWebClientSupport {

	private final Optional authorizedClientManager;

	public SpringAddonsWebClientSupport(
			SystemProxyProperties systemProxyProperties,
			SpringAddonsRestProperties addonsProperties,
			BearerProvider forwardingBearerProvider,
			Optional authorizedClientManager) {
		super(systemProxyProperties, addonsProperties, forwardingBearerProvider);
		this.authorizedClientManager = authorizedClientManager;
	}

	@Override
	protected ExchangeFilterFunction oauth2RegistrationFilter(String registrationId) {
		return (ClientRequest request, ExchangeFunction next) -> {
			final var provider = authorizedClientManager.map(acm -> new AuthorizedClientBearerProvider(acm, registrationId));
			if (provider.flatMap(AuthorizedClientBearerProvider::getBearer).isPresent()) {
				final var modified = ClientRequest.from(request).headers(headers -> headers.setBearerAuth(provider.get().getBearer().get())).build();
				return next.exchange(modified);
			}
			return next.exchange(request);
		};
	}
}