/*******************************************************************************
 * 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;

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

import org.opensaml.common.xml.SAMLConstants;
import org.opensaml.saml2.core.impl.AuthnRequestImpl;
import org.opensaml.saml2.metadata.AttributeConsumingService;
import org.opensaml.saml2.metadata.RequestedAttribute;
import org.opensaml.saml2.metadata.SPSSODescriptor;
import org.opensaml.saml2.metadata.provider.MetadataProvider;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;

import at.gv.egovernment.moa.id.moduls.RequestImpl;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.NoMetadataInformationException;
import at.gv.egovernment.moa.id.protocols.pvp2x.messages.InboundMessage;
import at.gv.egovernment.moa.id.protocols.pvp2x.messages.MOARequest;
import at.gv.egovernment.moa.logging.Logger;

@Component("PVPTargetConfiguration")
@Scope(value = BeanDefinition.SCOPE_PROTOTYPE)
public class PVPTargetConfiguration extends RequestImpl {

	public static final String DATAID_INTERFEDERATION_MINIMAL_FRONTCHANNEL_RESP = "useMinimalFrontChannelResponse";
	public static final String DATAID_INTERFEDERATION_NAMEID = "federatedNameID";
	public static final String DATAID_INTERFEDERATION_QAALEVEL = "federatedQAALevel";	
	
	public static final String DATAID_INTERFEDERATION_REQUESTID = "authnReqID";
	
	private static final long serialVersionUID = 4889919265919638188L;
	
	InboundMessage request;
	String binding;
	String consumerURL;
	
	public InboundMessage getRequest() {
		return request;
	}

	public void setRequest(InboundMessage request) {
		this.request = request;
	}

	public String getBinding() {
		return binding;
	}

	public void setBinding(String binding) {
		this.binding = binding;
	}

	public String getConsumerURL() {
		return consumerURL;
	}

	public void setConsumerURL(String consumerURL) {
		this.consumerURL = consumerURL;
		
	}

	/* (non-Javadoc)
	 * @see at.gv.egovernment.moa.id.moduls.RequestImpl#getRequestedAttributes()
	 */
	@Override
	public Collection<String> getRequestedAttributes(MetadataProvider metadataProvider) {

		Map<String, String> reqAttr = new HashMap<String, String>();
		for (String el : PVP2XProtocol.DEFAULTREQUESTEDATTRFORINTERFEDERATION)
			reqAttr.put(el, "");
						
		try {			
			SPSSODescriptor spSSODescriptor = getRequest().getEntityMetadata(metadataProvider).getSPSSODescriptor(SAMLConstants.SAML20P_NS);
			if (spSSODescriptor.getAttributeConsumingServices() != null && 
					spSSODescriptor.getAttributeConsumingServices().size() > 0) {
							
				Integer aIdx = null;
				if (getRequest() instanceof MOARequest && 
						((MOARequest)getRequest()).getSamlRequest() instanceof AuthnRequestImpl) {					
					AuthnRequestImpl authnRequest = (AuthnRequestImpl)((MOARequest)getRequest()).getSamlRequest();					
					aIdx = authnRequest.getAttributeConsumingServiceIndex();
					
				} else {
					Logger.error("MOARequest is NOT of type AuthnRequest");
				}
				
				int idx = 0;

				AttributeConsumingService attributeConsumingService = null;
				
				if (aIdx != null) {
					idx = aIdx.intValue();
					attributeConsumingService = spSSODescriptor
							.getAttributeConsumingServices().get(idx);
					
				} else {
					List<AttributeConsumingService> attrConsumingServiceList = spSSODescriptor.getAttributeConsumingServices();
					for (AttributeConsumingService el : attrConsumingServiceList) {
						if (el.isDefault())
							attributeConsumingService = el;
					}				
				}
				
				for ( RequestedAttribute attr : attributeConsumingService.getRequestAttributes())
					reqAttr.put(attr.getName(), "");
			}
			
			//return attributQueryBuilder.buildSAML2AttributeList(this.getOnlineApplicationConfiguration(), reqAttr.keySet().iterator());
			return reqAttr.keySet();
			
		} catch (NoMetadataInformationException e) {
			Logger.warn("NO metadata found for Entity " + getRequest().getEntityID());
			return null;
					
		}
		
	}

}