io.imunity.furms.unity.client.oauth.UnityOauthAccessTokenRepository 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) 2021 Bixbit s.c. All rights reserved.
 *  See LICENSE file for licensing information.
 */

package io.imunity.furms.unity.client.oauth;

import io.imunity.furms.domain.authz.token.TokenRefreshResponse;
import io.imunity.furms.spi.tokens.AccessTokenRepository;
import io.imunity.furms.unity.client.UnityClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.util.UriComponentsBuilder;

import java.lang.invoke.MethodHandles;
import java.net.URI;
import java.util.LinkedHashMap;

@Component
class UnityOauthAccessTokenRepository implements AccessTokenRepository {

	private static final Logger LOG = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());

	private final UnityClient unityClient;
	private final UnityOauthClient unityOauthClient;
	private final UnityOauthProperties unityOauthProperties;

	UnityOauthAccessTokenRepository(UnityClient unityClient,
	                                       UnityOauthClient unityOauthClient,
	                                       UnityOauthProperties unityOauthProperties) {
		this.unityClient = unityClient;
		this.unityOauthClient = unityOauthClient;
		this.unityOauthProperties = unityOauthProperties;
	}

	@Override
	public void revoke(String accessToken, String clientId) {
		final URI uri = UriComponentsBuilder.fromUriString(unityOauthProperties.getRevoke())
			.queryParam("token", accessToken)
			.queryParam("client_id", clientId)
			.queryParam("token_type_hint", "access_token")
			.queryParam("logout", "true")
			.build()
			.toUri();

		unityClient.post(uri, MediaType.APPLICATION_FORM_URLENCODED);
	}

	@Override
	public TokenRefreshResponse refresh(String refreshToken, String clientId) throws Exception {
		try {
			final URI uri = new URI(unityOauthProperties.getTokenUri());
			final MultiValueMap formData = new LinkedMultiValueMap<>();
			formData.add("grant_type", "refresh_token");
			formData.add("refresh_token", refreshToken);

			@SuppressWarnings("unchecked")
			LinkedHashMap response = unityOauthClient.postForObject(uri, LinkedHashMap.class, 
					formData, MediaType.APPLICATION_FORM_URLENCODED);

			return new TokenRefreshResponse(response);
		} catch (final Exception e) {
			LOG.error("Could not refresh token: ", e);
			throw e;
		}
	}

}