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

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.joda.time.DateTime;
import org.opensaml.common.binding.artifact.SAMLArtifactMap.SAMLArtifactMapEntry;
import org.opensaml.saml2.core.ArtifactResolve;
import org.opensaml.saml2.core.ArtifactResponse;

import at.gv.egovernment.moa.id.auth.exception.MOAIDException;
import at.gv.egovernment.moa.id.data.IAuthData;
import at.gv.egovernment.moa.id.data.SLOInformationInterface;
import at.gv.egovernment.moa.id.protocols.pvp2x.PVPAssertionStorage;
import at.gv.egovernment.moa.id.protocols.pvp2x.PVPTargetConfiguration;
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.id.protocols.pvp2x.binding.SoapBinding;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.RequestDeniedException;
import at.gv.egovernment.moa.id.protocols.pvp2x.utils.SAML2Utils;
import at.gv.egovernment.moa.logging.Logger;

public class ArtifactResolution implements IRequestHandler {

	public boolean handleObject(InboundMessage obj) {
		return (obj instanceof MOARequest && 
				((MOARequest)obj).getSamlRequest() instanceof ArtifactResolve);
	}

	public SLOInformationInterface process(PVPTargetConfiguration obj, HttpServletRequest req,
			HttpServletResponse resp, IAuthData authData) throws MOAIDException {
		if (!handleObject(obj.getRequest())) {
			throw new MOAIDException("pvp2.13", null);
		}
		
		ArtifactResolve artifactResolve = (ArtifactResolve) ((MOARequest)obj.getRequest()).getSamlRequest();
		String artifactID = artifactResolve.getArtifact().getArtifact();

		PVPAssertionStorage pvpAssertion = PVPAssertionStorage.getInstance();

		if (!pvpAssertion.contains(artifactID)) {
			throw new RequestDeniedException();
		} else {
			try {
				SAMLArtifactMapEntry assertion = pvpAssertion.get(artifactID);
				ArtifactResponse response = SAML2Utils
						.createSAMLObject(ArtifactResponse.class);
				response.setMessage(assertion.getSamlMessage());
				response.setIssueInstant(new DateTime());
				SoapBinding encoder = new SoapBinding();
				encoder.encodeRespone(req, resp, response, null, null);
			} catch (Exception e) {
				Logger.error("Failed to resolve artifact", e);
			}
		}
		
		return null;
	}

}