package at.gv.egovernment.moa.id.protocols.pvp2x.builder;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;

import org.opensaml.saml2.core.Attribute;

import at.gv.egovernment.moa.id.auth.data.AuthenticationSession;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.BPKAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.BirthdateAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDCitizenQAALevelAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDIssuingNationAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDSectorForIDAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.GivenNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.IAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateLegalPersonFullNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateLegalPersonSourcePinAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateLegalPersonSourcePinTypeAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateNaturalPersonBPKAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateNaturalPersonBirthDateAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateNaturalPersonFamilyNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateNaturalPersonGivenNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateProfRepDescAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateProfRepOIDAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateReferenceValueAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateTypeAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.PVPVersionAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.PrincipalNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.PVP2Exception;

public class PVPAttributeBuilder {

	private static HashMap<String, IAttributeBuilder> builders;

	private static void addBuilder(IAttributeBuilder builder) {
		builders.put(builder.getName(), builder);
	}

	static {
		builders = new HashMap<String, IAttributeBuilder>();
		// Citizen Token normal
		addBuilder(new PVPVersionAttributeBuilder());
		addBuilder(new PrincipalNameAttributeBuilder());
		addBuilder(new GivenNameAttributeBuilder());
		addBuilder(new BirthdateAttributeBuilder());
		addBuilder(new BPKAttributeBuilder());
		addBuilder(new EIDCitizenQAALevelAttributeBuilder());
		addBuilder(new EIDIssuingNationAttributeBuilder());
		addBuilder(new EIDSectorForIDAttributeBuilder());
		
		// Mandate Attributes
		addBuilder(new MandateTypeAttributeBuilder());
		addBuilder(new MandateLegalPersonFullNameAttributeBuilder());
		addBuilder(new MandateLegalPersonSourcePinAttributeBuilder());
		addBuilder(new MandateLegalPersonSourcePinTypeAttributeBuilder());
		addBuilder(new MandateNaturalPersonBirthDateAttributeBuilder());
		addBuilder(new MandateNaturalPersonBPKAttributeBuilder());
		addBuilder(new MandateNaturalPersonFamilyNameAttributeBuilder());
		addBuilder(new MandateNaturalPersonGivenNameAttributeBuilder());
		addBuilder(new MandateTypeAttributeBuilder());
		addBuilder(new MandateProfRepOIDAttributeBuilder());
		addBuilder(new MandateProfRepDescAttributeBuilder());
		addBuilder(new MandateReferenceValueAttributeBuilder());
	}

	public static Attribute buildAttribute(String name,
			AuthenticationSession authSession) throws PVP2Exception {
		if (builders.containsKey(name)) {
			return builders.get(name).build(authSession);
		}
		return null;
	}

	public static List<Attribute> buildSupportedEmptyAttributes() {
		List<Attribute> attributes = new ArrayList<Attribute>();
		Iterator<IAttributeBuilder> builderIt = builders.values().iterator();
		while (builderIt.hasNext()) {
			IAttributeBuilder builder = builderIt.next();
			Attribute emptyAttribute = builder.buildEmpty();
			if (emptyAttribute != null) {
				attributes.add(emptyAttribute);
			}
		}
		return attributes;
	}

}