pl.edu.icm.unity.oauth.as.OAuthClientProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of unity-server-oauth Show documentation
Client and server OAuth support
The newest version!
/*
 * Copyright (c) 2021 Bixbit - Krzysztof Benedyczak. All rights reserved.
 * See LICENCE.txt file for licensing information.
 */

package pl.edu.icm.unity.oauth.as;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Component;

import pl.edu.icm.unity.base.attribute.AttributeExt;
import pl.edu.icm.unity.base.entity.Entity;
import pl.edu.icm.unity.base.entity.EntityParam;
import pl.edu.icm.unity.base.exceptions.EngineException;
import pl.edu.icm.unity.base.identity.Identity;
import pl.edu.icm.unity.base.identity.IdentityTaV;
import pl.edu.icm.unity.base.utils.Log;
import pl.edu.icm.unity.engine.api.AttributesManagement;
import pl.edu.icm.unity.engine.api.EntityManagement;
import pl.edu.icm.unity.stdext.identity.UsernameIdentity;

@Component
public class OAuthClientProvider
{
	private static final Logger log = Log.getLogger(Log.U_SERVER_OAUTH, OAuthClientProvider.class);
	private EntityManagement identitiesMan;
	private AttributesManagement attributesMan;

	public OAuthClientProvider(@Qualifier("insecure") EntityManagement identitiesManagement,
			@Qualifier("insecure") AttributesManagement attributesManagement)
	{
		this.identitiesMan = identitiesManagement;
		this.attributesMan = attributesManagement;
	}

	public OAuthClient getClient(long entityId, String oauthGroup) throws OAuthValidationException
	{
		EntityParam clientEntity = new EntityParam(entityId);
		Entity clientResolvedEntity = getClient(clientEntity);

		Identity username = clientResolvedEntity.getIdentities().stream()
				.filter(i -> i.getTypeId().equals(UsernameIdentity.ID)).findFirst().orElse(null);

		return new OAuthClient(username != null ? username.getComparableValue() : null,

				clientResolvedEntity.getId(), getAttributesNoAuthZ(clientEntity, oauthGroup));

	}

	public OAuthClient getClient(String username, String oauthGroup) throws OAuthValidationException
	{
		EntityParam clientEntity = new EntityParam(new IdentityTaV(UsernameIdentity.ID, username));
		return new OAuthClient(username, getClient(clientEntity).getId(),
				getAttributesNoAuthZ(clientEntity, oauthGroup));

	}

	private Entity getClient(EntityParam clientEntity) throws OAuthValidationException
	{
		try
		{
			return identitiesMan.getEntity(clientEntity);

		} catch (IllegalArgumentException e)
		{
			throw new OAuthValidationException("The client '" + clientEntity + "' is unknown");
		} catch (Exception e)
		{
			log.error("Problem retrieving identity of the OAuth client", e);
			throw new OAuthValidationException("Internal error, can not retrieve OAuth client's data");
		}
	}

	private Map getAttributesNoAuthZ(EntityParam clientEntity, String oauthGroup)
			throws OAuthValidationException
	{

		Collection attrs;
		try
		{
			attrs = attributesMan.getAllAttributes(clientEntity, true, oauthGroup, null, false);
		} catch (EngineException e)
		{
			log.error("Problem retrieving attributes of the OAuth client", e);
			throw new OAuthValidationException("Internal error, can not retrieve OAuth client's data");
		}
		Map ret = new HashMap<>();
		attrs.stream().forEach(a -> ret.put(a.getName(), a));
		return ret;
	}

}