package at.gv.egovernment.moa.id.auth.modules.sl20_auth;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.Map;

import org.junit.BeforeClass;
import org.junit.Test;
import org.opensaml.DefaultBootstrap;
import org.opensaml.saml2.core.Assertion;
import org.springframework.beans.factory.annotation.Autowired;

import at.gv.egiz.eaaf.core.api.idp.auth.data.IIdentityLink;
import at.gv.egiz.eaaf.modules.pvp2.sp.impl.utils.AssertionAttributeExtractor;
import at.gv.egovernment.moa.id.auth.modules.sl20_auth.dummydata.DummyAuthConfig;
import at.gv.egovernment.moa.id.auth.modules.sl20_auth.dummydata.DummyOA;
import at.gv.egovernment.moa.id.auth.modules.sl20_auth.sl20.SL20Constants;
import at.gv.egovernment.moa.id.auth.modules.sl20_auth.sl20.verifier.QualifiedeIDVerifier;
import at.gv.egovernment.moa.id.auth.parser.IdentityLinkAssertionParser;
import at.gv.egovernment.moa.id.commons.api.IOAAuthParameters;
import at.gv.egovernment.moa.id.commons.api.data.IVerifiyXMLSignatureResponse;
import at.gv.egovernment.moa.id.commons.api.exceptions.ConfigurationException;
import at.gv.egovernment.moa.logging.Logger;
import at.gv.egovernment.moa.spss.MOAException;
import at.gv.egovernment.moa.spss.api.Configurator;
import at.gv.egovernment.moa.util.Base64Utils;
import at.gv.egovernment.moa.util.MiscUtil;
import at.gv.egovernment.moaspss.logging.LoggingContext;
import at.gv.egovernment.moaspss.logging.LoggingContextManager;
import iaik.security.ec.provider.ECCelerate;
import iaik.security.provider.IAIK;

public abstract class eIDDataVerifierTest {
	
	protected Map<String, String> eIDData = null;

	@Autowired DummyAuthConfig authConfig;
	
	@BeforeClass
	public static void moaSPSSInitialize() throws ConfigurationException, org.opensaml.xml.ConfigurationException, IOException {
		Logger.info("Loading Java security providers.");
		//System.setProperty("moa.spss.server.configuration", "F:\\Projekte\\configs\\moa-spss\\MOASPSSConfiguration.xml");
		String current = new java.io.File( "." ).getCanonicalPath();
		System.setProperty("moa.spss.server.configuration", current + "/src/test/resources/moaspss_config/MOASPSSConfiguration.xml");
		
	    IAIK.addAsProvider();                
	    ECCelerate.addAsProvider();
	    DefaultBootstrap.bootstrap();
	    
        try {
        	LoggingContextManager.getInstance().setLoggingContext(
                    new LoggingContext("startup"));
        	Logger.debug("Starting MOA-SPSS initialization process ... ");
        	Configurator.getInstance().init();        	
        	Logger.info("MOA-SPSS initialization complete ");
        	                       
         } catch (MOAException e) {
        	 Logger.error("MOA-SP initialization FAILED!", e.getWrapped()); 
             throw new ConfigurationException("config.10", new Object[] { e
                      .toString() }, e);
		}
       
	}
	
	@Test
	public void dummyTest() throws Exception {
		
		
	}
			
	@Test
	public void parseIdl() throws Exception {
		String idlB64 = eIDData.get(SL20Constants.SL20_COMMAND_PARAM_EID_RESULT_IDL);
		if (MiscUtil.isEmpty(idlB64))
			throw new Exception("NO IDL found");
		
		IIdentityLink idl = new IdentityLinkAssertionParser(new ByteArrayInputStream(Base64Utils.decode(idlB64, false))).parseIdentityLink();
		//IIdentityLink idl = new IdentityLinkAssertionParser(new ByteArrayInputStream(Base64Url.decode(idlB64))).parseIdentityLink();
		if (idl == null) 
			throw new Exception("IDL parsing FAILED");
			
	}
	
	@Test
	public void verifyIdl() throws Exception {
		String idlB64 = eIDData.get(SL20Constants.SL20_COMMAND_PARAM_EID_RESULT_IDL);
		if (MiscUtil.isEmpty(idlB64))
			throw new Exception("NO IDL found");
		
		IIdentityLink idl = new IdentityLinkAssertionParser(new ByteArrayInputStream(Base64Utils.decode(idlB64, false))).parseIdentityLink();
//		IIdentityLink idl = new IdentityLinkAssertionParser(new ByteArrayInputStream(Base64Url.decode(idlB64))).parseIdentityLink();
		
		if (idl == null) 
			throw new Exception("IDL parsing FAILED");
		
		IOAAuthParameters dummyOA = new DummyOA();
		QualifiedeIDVerifier.verifyIdentityLink(idl, dummyOA , authConfig);
		
	}
	
	@Test
	public void parseAuthBlock() throws Exception {
		String authBlockB64 = eIDData.get(SL20Constants.SL20_COMMAND_PARAM_EID_RESULT_AUTHBLOCK);
		if (MiscUtil.isEmpty(authBlockB64))
			throw new Exception("NO AuthBlock found");
		
		Assertion authBlock = QualifiedeIDVerifier.parseAuthBlockToSaml2Assertion(authBlockB64);
		new AssertionAttributeExtractor(authBlock);
				
	}
	
	
	
	@Test
	public void verifyAuthBlock() throws Exception {
		String authBlockB64 = eIDData.get(SL20Constants.SL20_COMMAND_PARAM_EID_RESULT_AUTHBLOCK);
		if (MiscUtil.isEmpty(authBlockB64))
			throw new Exception("NO AuthBlock found");
			
		IOAAuthParameters dummyOA = new DummyOA();		
		authConfig.setRequireAuthBlockQC(false);
		QualifiedeIDVerifier.verifyAuthBlock(authBlockB64, dummyOA , authConfig);
		authConfig.setRequireAuthBlockQC(true);
		
	}
	
	@Test
	public void checkIDLAgainstAuthblock() throws Exception {
		String authBlockB64 = eIDData.get(SL20Constants.SL20_COMMAND_PARAM_EID_RESULT_AUTHBLOCK);
		String idlB64 = eIDData.get(SL20Constants.SL20_COMMAND_PARAM_EID_RESULT_IDL);
		if (MiscUtil.isEmpty(idlB64))
			throw new Exception("NO IDL found");	
		if (MiscUtil.isEmpty(authBlockB64))
			throw new Exception("NO AuthBlock found");
		
		IIdentityLink idl = new IdentityLinkAssertionParser(new ByteArrayInputStream(Base64Utils.decode(idlB64, false))).parseIdentityLink();
		Assertion authBlock = QualifiedeIDVerifier.parseAuthBlockToSaml2Assertion(authBlockB64);
		AssertionAttributeExtractor authBlockExtractor = new AssertionAttributeExtractor(authBlock);				
		IVerifiyXMLSignatureResponse authBlockVerificationResult = QualifiedeIDVerifier.verifyAuthBlock(authBlockB64,  new DummyOA() , authConfig);
		QualifiedeIDVerifier.checkConsistencyOfeIDData(getSl20ReqId(), idl, authBlockExtractor, authBlockVerificationResult);
		
		
		
	}
	
	protected abstract String getSl20ReqId();
}