/*******************************************************************************
 * Copyright 2014 Federal Chancellery Austria
 * MOA-ID has been developed in a cooperation between BRZ, the Federal
 * Chancellery Austria - ICT staff unit, and Graz University of Technology.
 *
 * Licensed under the EUPL, Version 1.1 or - as soon they will be approved by
 * the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * http://www.osor.eu/eupl/
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 *
 * This product combines work with different licenses. See the "NOTICE" text
 * file for details on the various modules and licenses.
 * The "NOTICE" text file is part of the distribution. Any derivative works
 * that you distribute must include a readable copy of the "NOTICE" text file.
 *******************************************************************************/
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.config.auth.OAAuthParameter;
import at.gv.egovernment.moa.id.data.IAuthData;
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.EIDAuthBlock;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDCcsURL;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDCitizenQAALevelAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDIdentityLinkBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDIssuingNationAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDSTORKTOKEN;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDSectorForIDAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDSignerCertificate;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDSourcePIN;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.EIDSourcePINType;
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.IAttributeGenerator;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateFullMandateAttributeBuilder;
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.MandateNaturalPersonSourcePinAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.MandateNaturalPersonSourcePinTypeAttributeBuilder;
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.builder.attributes.STORKAdoptedFamilyNameAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKAgeAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKCanonicalResidenceAddressAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKCountryCodeOfBirthAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKFiscalNumberAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKGenderAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKInhertedFamilyNameAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKIsAgeOverAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKMaritalStatusAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKNationalityCodeAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKPseudonymAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKResidencePermitAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKTextResidenceAddressAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.STORKTitleAttributBuilder;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.SamlAttributeGenerator;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.exceptions.AttributeException;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.exceptions.InvalidDateFormatAttributeException;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.exceptions.NoMandateDataAttributeException;
import at.gv.egovernment.moa.id.protocols.pvp2x.builder.attributes.exceptions.UnavailableAttributeException;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.InvalidDateFormatException;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.NoMandateDataAvailableException;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.PVP2Exception;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.UnprovideableAttributeException;

public class PVPAttributeBuilder {
	
	private static IAttributeGenerator<Attribute> generator = new SamlAttributeGenerator();
	
	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());
		addBuilder(new EIDIdentityLinkBuilder());
		addBuilder(new EIDAuthBlock());
		addBuilder(new EIDCcsURL());
		addBuilder(new EIDSignerCertificate());
		addBuilder(new EIDSourcePIN());
		addBuilder(new EIDSourcePINType());
		
		// 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 MandateNaturalPersonSourcePinAttributeBuilder());
		addBuilder(new MandateNaturalPersonSourcePinTypeAttributeBuilder());
		addBuilder(new MandateTypeAttributeBuilder());
		addBuilder(new MandateProfRepOIDAttributeBuilder());
		addBuilder(new MandateProfRepDescAttributeBuilder());
		addBuilder(new MandateReferenceValueAttributeBuilder());
		addBuilder(new MandateFullMandateAttributeBuilder());
		
		//STORK Attributes
		addBuilder(new EIDSTORKTOKEN());
		addBuilder(new STORKAdoptedFamilyNameAttributBuilder());
		addBuilder(new STORKAgeAttributBuilder());
		addBuilder(new STORKCanonicalResidenceAddressAttributBuilder());
		addBuilder(new STORKCountryCodeOfBirthAttributBuilder());
		addBuilder(new STORKFiscalNumberAttributBuilder());
		addBuilder(new STORKGenderAttributBuilder());
		addBuilder(new STORKInhertedFamilyNameAttributBuilder());
		addBuilder(new STORKIsAgeOverAttributBuilder());
		addBuilder(new STORKMaritalStatusAttributBuilder());
		addBuilder(new STORKNationalityCodeAttributBuilder());
		addBuilder(new STORKPseudonymAttributBuilder());
		addBuilder(new STORKResidencePermitAttributBuilder());
		addBuilder(new STORKTextResidenceAddressAttributBuilder());
		addBuilder(new STORKTitleAttributBuilder());		
	}
	
	public static Attribute buildAttribute(String name, OAAuthParameter oaParam,
			IAuthData authData) throws PVP2Exception {
		if (builders.containsKey(name)) {
			try {
				return builders.get(name).build(oaParam, authData, generator);
			}
			catch (AttributeException e) {
				if (e instanceof UnavailableAttributeException) {
					throw new UnprovideableAttributeException(((UnavailableAttributeException) e).getAttributeName());
				} else if (e instanceof InvalidDateFormatAttributeException) {
					throw new InvalidDateFormatException();
				} else if (e instanceof NoMandateDataAttributeException) {
					throw new NoMandateDataAvailableException();
				} else {
					throw new UnprovideableAttributeException(name);
				}
			}
		}
		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(generator);
			if (emptyAttribute != null) {
				attributes.add(emptyAttribute);
			}
		}
		return attributes;
	}
	
}