/*******************************************************************************
 * 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.oauth20.attributes;

import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang.StringUtils;

import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;

import at.gv.egiz.eaaf.core.api.idp.IAttributeBuilder;
import at.gv.egiz.eaaf.core.api.idp.IAttributeGenerator;
import at.gv.egiz.eaaf.core.api.idp.IAuthData;
import at.gv.egiz.eaaf.core.api.idp.ISPConfiguration;
import at.gv.egiz.eaaf.core.exceptions.AttributeBuilderException;
import at.gv.egiz.eaaf.core.impl.data.Pair;
import at.gv.egiz.eaaf.core.impl.idp.builder.attributes.BPKAttributeBuilder;
import at.gv.egiz.eaaf.core.impl.idp.builder.attributes.EIDIdentityLinkBuilder;
import at.gv.egiz.eaaf.core.impl.idp.builder.attributes.EIDIssuingNationAttributeBuilder;
import at.gv.egiz.eaaf.core.impl.idp.builder.attributes.EIDSectorForIDAttributeBuilder;
import at.gv.egiz.eaaf.core.impl.idp.builder.attributes.EIDSourcePIN;
import at.gv.egiz.eaaf.core.impl.idp.builder.attributes.EIDSourcePINType;
import at.gv.egiz.eaaf.modules.pvp2.impl.builder.PVPAttributeBuilder;
import at.gv.egovernment.moa.id.auth.stork.STORKConstants;
import at.gv.egovernment.moa.id.protocols.builder.attributes.EIDAuthBlock;
import at.gv.egovernment.moa.id.protocols.builder.attributes.EIDCcsURL;
import at.gv.egovernment.moa.id.protocols.builder.attributes.EIDCitizenQAALevelAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.EIDSTORKTOKEN;
import at.gv.egovernment.moa.id.protocols.builder.attributes.EIDSignerCertificate;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateLegalPersonFullNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateLegalPersonSourcePinAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateLegalPersonSourcePinTypeAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateNaturalPersonBPKAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateNaturalPersonBirthDateAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateNaturalPersonFamilyNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateNaturalPersonGivenNameAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateNaturalPersonSourcePinAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateNaturalPersonSourcePinTypeAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateProfRepDescAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateProfRepOIDAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateReferenceValueAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.builder.attributes.MandateTypeAttributeBuilder;
import at.gv.egovernment.moa.id.protocols.oauth20.protocol.OAuth20AuthRequest;
import at.gv.egovernment.moa.logging.Logger;

public final class OAuth20AttributeBuilder {
	
	private OAuth20AttributeBuilder() {
		throw new InstantiationError();
	}
	 
	private static IAttributeGenerator<Pair<String, JsonPrimitive>> generator = new IAttributeGenerator<Pair<String, JsonPrimitive>>() {
		
		public Pair<String, JsonPrimitive> buildStringAttribute(final String friendlyName, final String name, final String value) {
			return Pair.newInstance(friendlyName, new JsonPrimitive(value));
		}
		
		public Pair<String, JsonPrimitive> buildIntegerAttribute(final String friendlyName, final String name, final int value) {
			return Pair.newInstance(friendlyName, new JsonPrimitive(value));
		}
		
		public Pair<String, JsonPrimitive> buildLongAttribute(final String friendlyName, final String name, final long value) {
			return Pair.newInstance(friendlyName, new JsonPrimitive(value));
		}
		
		public Pair<String, JsonPrimitive> buildEmptyAttribute(final String friendlyName, final String name) {
			return Pair.newInstance(friendlyName, new JsonPrimitive(""));
		}
		
	};
	
	private static final List<IAttributeBuilder> buildersOpenId = new ArrayList<IAttributeBuilder>();
	private static final List<IAttributeBuilder> buildersProfile = new ArrayList<IAttributeBuilder>();
	private static final List<IAttributeBuilder> buildersEID = new ArrayList<IAttributeBuilder>();
	private static final List<IAttributeBuilder> buildersEIDGov = new ArrayList<IAttributeBuilder>();
	private static final List<IAttributeBuilder> buildersMandate = new ArrayList<IAttributeBuilder>();
	private static final List<IAttributeBuilder> buildersSTORK = new ArrayList<IAttributeBuilder>();
	static {
		// openId
		buildersOpenId.add(new OpenIdIssuerAttribute());
		buildersOpenId.add(new OpenIdSubjectIdentifierAttribute());
		buildersOpenId.add(new OpenIdExpirationTimeAttribute());
		buildersOpenId.add(new OpenIdIssueInstantAttribute());
		buildersOpenId.add(new OpenIdAuthenticationTimeAttribute());
		buildersOpenId.add(new OpenIdAudiencesAttribute());
		buildersOpenId.add(new OpenIdNonceAttribute());
		
		// profile
		buildersProfile.add(new ProfileGivenNameAttribute());
		buildersProfile.add(new ProfileFamilyNameAttribute());
		buildersProfile.add(new ProfileDateOfBirthAttribute());
		
		// EID
		buildersEID.add(new EIDCcsURL());
		buildersEID.add(new EIDCitizenQAALevelAttributeBuilder());
		buildersEID.add(new EIDIssuingNationAttributeBuilder());
		buildersEID.add(new EIDSectorForIDAttributeBuilder());
		buildersEID.add(new EIDAuthBlock());
		buildersEID.add(new EIDSignerCertificate());
		buildersEID.add(new BPKAttributeBuilder());
		
		// eID_gov
		buildersEIDGov.add(new EIDSourcePIN());
		buildersEIDGov.add(new EIDSourcePINType());
		buildersEIDGov.add(new EIDIdentityLinkBuilder());
		
		// mandate
		buildersMandate.add(new MandateTypeAttributeBuilder());
		buildersMandate.add(new MandateReferenceValueAttributeBuilder());
		
		buildersMandate.add(new MandateNaturalPersonSourcePinAttributeBuilder());
		buildersMandate.add(new MandateNaturalPersonSourcePinTypeAttributeBuilder());
		buildersMandate.add(new MandateNaturalPersonBPKAttributeBuilder());
		buildersMandate.add(new MandateNaturalPersonFamilyNameAttributeBuilder());
		buildersMandate.add(new MandateNaturalPersonGivenNameAttributeBuilder());
		buildersMandate.add(new MandateNaturalPersonBirthDateAttributeBuilder());
		
		buildersMandate.add(new MandateLegalPersonSourcePinAttributeBuilder());
		buildersMandate.add(new MandateLegalPersonSourcePinTypeAttributeBuilder());
		buildersMandate.add(new MandateLegalPersonFullNameAttributeBuilder());
		
		buildersMandate.add(new MandateProfRepOIDAttributeBuilder());
		buildersMandate.add(new MandateProfRepDescAttributeBuilder());
				
		// STORK
		buildersSTORK.add(new EIDSTORKTOKEN());
		
		IAttributeBuilder attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.ADOPTEDFAMILYNAME_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.AGE_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.CANONICALRESIDENCEADDRESS_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.CONTRYCODEOFBIRTH_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.FISCALNUMBER_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.GENDER_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.INHERITEDFAMILYNAME_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.ISAGEOVER_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.MARITALSTATUS_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.NATIONALITYCODE_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.PSEUDONYM_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.RESIDENCEPERMIT_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.RESIDENCEADDRESS_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
		
		attr = PVPAttributeBuilder.getAttributeBuilder(STORKConstants.TITLE_NAME);
		if (attr != null)			
			buildersSTORK.add(attr);
	}
	
	private static void addAttibutes(final List<IAttributeBuilder> builders, final JsonObject jsonObject,
			final ISPConfiguration oaParam, final IAuthData authData, OAuth20AuthRequest oAuthRequest) {
		for (IAttributeBuilder b : builders) {
			try {
				//TODO: better solution requires more refactoring :(
				Pair<String, JsonPrimitive> attribute = null;
				if (b instanceof OpenIdNonceAttribute) {
					OpenIdNonceAttribute nonceBuilder = (OpenIdNonceAttribute) b;
					attribute = nonceBuilder.build(oaParam, authData, oAuthRequest, generator);
					
				} else				
					attribute = b.build(oaParam, authData, generator);
					
				if (attribute != null && !StringUtils.isEmpty(attribute.getSecond().getAsString())) {
					jsonObject.add(attribute.getFirst(), attribute.getSecond());
				}
			}
			catch (AttributeBuilderException e) {
				Logger.info("Cannot add attribute " + b.getName());
			}
		}
	}
	
	public static void addScopeOpenId(final JsonObject jsonObject,
			final ISPConfiguration oaParam, final IAuthData authData, 
			final OAuth20AuthRequest oAuthRequest) {
		addAttibutes(buildersOpenId, jsonObject, oaParam, authData, oAuthRequest);
	}
	
	public static void addScopeProfile(final JsonObject jsonObject,
			final ISPConfiguration oaParam, final IAuthData authData) {
		addAttibutes(buildersProfile, jsonObject, oaParam, authData, null);
	}
	
	public static void addScopeEID(final JsonObject jsonObject,
			final ISPConfiguration oaParam, final IAuthData authData) {
		addAttibutes(buildersEID, jsonObject, oaParam, authData, null);
	}
	
	public static void addScopeEIDGov(final JsonObject jsonObject,
			final ISPConfiguration oaParam, final IAuthData authData) {
		addAttibutes(buildersEIDGov, jsonObject, oaParam, authData, null);
	}
	
	public static void addScopeMandate(final JsonObject jsonObject,
			final ISPConfiguration oaParam, final IAuthData authData) {
		addAttibutes(buildersMandate, jsonObject, oaParam, authData, null);
	}
	
	public static void addScopeSTORK(final JsonObject jsonObject,
			final ISPConfiguration oaParam, final IAuthData authData) {
		addAttibutes(buildersSTORK, jsonObject, oaParam, authData, null);
	}

	/**
	 * @return the buildersprofile
	 */
	public static List<IAttributeBuilder> getBuildersprofile() {
		return buildersProfile;
	}

	/**
	 * @return the builderseid
	 */
	public static List<IAttributeBuilder> getBuilderseid() {
		return buildersEID;
	}

	/**
	 * @return the builderseidgov
	 */
	public static List<IAttributeBuilder> getBuilderseidgov() {
		return buildersEIDGov;
	}

	/**
	 * @return the buildersmandate
	 */
	public static List<IAttributeBuilder> getBuildersmandate() {
		return buildersMandate;
	}

	/**
	 * @return the buildersstork
	 */
	public static List<IAttributeBuilder> getBuildersstork() {
		return buildersSTORK;
	}
	
	
}