pl.edu.icm.unity.engine.credential.CredentialReqRepository Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2017 Bixbit - Krzysztof Benedyczak All rights reserved.
 * See LICENCE.txt file for licensing information.
 */
package pl.edu.icm.unity.engine.credential;

import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import pl.edu.icm.unity.MessageSource;
import pl.edu.icm.unity.exceptions.EngineException;
import pl.edu.icm.unity.store.api.generic.CredentialRequirementDB;
import pl.edu.icm.unity.store.api.tx.Transactional;
import pl.edu.icm.unity.types.authn.CredentialRequirements;

/**
 * Allows read all credential requirements from DB and one special credential
 * requirements which contains all credentials
 * 
 * @author P.Piernik
 *
 */
@Component
public class CredentialReqRepository
{

	private CredentialRepository credentialRepo;
	private CredentialRequirementDB credentialRequirementDB;
	private MessageSource msg;

	@Autowired
	public CredentialReqRepository(CredentialRepository credentialRepo,
			CredentialRequirementDB credentialRequirementDB, MessageSource msg)
	{

		this.credentialRepo = credentialRepo;
		this.credentialRequirementDB = credentialRequirementDB;
		this.msg = msg;
	}

	@Transactional
	public Collection getCredentialRequirements() throws EngineException
	{
		List res = credentialRequirementDB.getAll();
		res.add(getSystemAllCredReq());
		return res;
	}

	@Transactional
	public CredentialRequirements get(String name) throws EngineException
	{
		if (SystemAllCredentialRequirements.NAME.equals(name))
			return getSystemAllCredReq();
		return credentialRequirementDB.get(name);
	}

	private CredentialRequirements getSystemAllCredReq() throws EngineException
	{
		Set all = credentialRepo.getCredentialDefinitions().stream()
				.map(c -> c.getName()).collect(Collectors.toSet());
		return new SystemAllCredentialRequirements(msg, all);
	}
	
	public void assertExist(String name) throws EngineException
	{
		if (!getAllNames().contains(name))
			throw new IllegalArgumentException(
					"There is no required credential set with id " + name);
	}

	private Set getAllNames() throws EngineException
	{
		return getCredentialRequirements().stream().map(cr -> cr.getName())
				.collect(Collectors.toSet());
	}

}