/*******************************************************************************
 * 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 org.opensaml.common.SAMLObject;
import org.opensaml.common.binding.artifact.SAMLArtifactMap;
import org.opensaml.xml.io.MarshallingException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import at.gv.egiz.eaaf.core.api.storage.ITransactionStorage;
import at.gv.egiz.eaaf.core.exceptions.EAAFException;
import at.gv.egovernment.moa.id.protocols.pvp2x.utils.StoredAssertion;

@Service("PVPAssertionStorage")
public class PVPAssertionStorage implements SAMLArtifactMap {
	
	@Autowired private ITransactionStorage transactionStorage;
		
	public boolean contains(String artifact) {
		return transactionStorage.containsKey(artifact);
	}

	public void put(String artifact, String relyingPartyId, String issuerId,
			SAMLObject samlMessage) throws MarshallingException {
		SAMLArtifactMapEntry assertion = new StoredAssertion(artifact,
				relyingPartyId,
				issuerId,
				samlMessage);
		 
		try { 
			transactionStorage.put(artifact, assertion, -1);
			
		} catch (EAAFException e) {
			// TODO Insert Error Handling, if Assertion could not be stored
			throw new MarshallingException("Assertion are not stored in Database.",e);
		}
	}

	public SAMLArtifactMapEntry get(String artifact) {
		try {
			return transactionStorage.get(artifact, SAMLArtifactMapEntry.class);
			
		} catch (EAAFException e) {
			// TODO Insert Error Handling, if Assertion could not be read
			e.printStackTrace();
			return null;
		}
	}

	public void remove(String artifact) {
		transactionStorage.remove(artifact);
	}
	
}