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

import org.opensaml.saml2.metadata.provider.FilterException;
import org.opensaml.saml2.metadata.provider.MetadataFilter;
import org.opensaml.xml.XMLObject;

import javax.xml.transform.dom.DOMSource;
import javax.xml.validation.Schema;
import javax.xml.validation.Validator;

import org.opensaml.common.xml.SAMLSchemaBuilder;

import org.xml.sax.SAXException;

import at.gv.egovernment.moa.id.config.ConfigurationException;
import at.gv.egovernment.moa.id.config.auth.AuthConfigurationProviderFactory;
import at.gv.egovernment.moa.id.protocols.pvp2x.exceptions.filter.SchemaValidationException;
import at.gv.egovernment.moa.logging.Logger;

/**
 * @author tlenz
 *
 */
public class SchemaValidationFilter implements MetadataFilter {

	private boolean isActive = true;
	
	public SchemaValidationFilter() {
		try {
			isActive = AuthConfigurationProviderFactory.getInstance().isPVPSchemaValidationActive();
			
		} catch (ConfigurationException e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * 
	 */
	public SchemaValidationFilter(boolean useSchemaValidation) {
		this.isActive = useSchemaValidation;
	}
	
	
	/* (non-Javadoc)
	 * @see org.opensaml.saml2.metadata.provider.MetadataFilter#doFilter(org.opensaml.xml.XMLObject)
	 */
	@Override
	public void doFilter(XMLObject arg0) throws SchemaValidationException {
		
		String errString = null;
		
		if (isActive) {
			try {
				Schema test = SAMLSchemaBuilder.getSAML11Schema();
				Validator val = test.newValidator();
				DOMSource source = new DOMSource(arg0.getDOM());		
				val.validate(source);
				Logger.info("Metadata Schema validation check done OK");
				return;
			
			} catch (SAXException e) {
				if (Logger.isDebugEnabled() || Logger.isTraceEnabled())
					Logger.warn("Metadata Schema validation FAILED with exception:", e);
				else
					Logger.warn("Metadata Schema validation FAILED with message: "+ e.getMessage());

				errString = e.getMessage();
				
			} catch (Exception e) {
				if (Logger.isDebugEnabled() || Logger.isTraceEnabled())
					Logger.warn("Metadata Schema validation FAILED with exception:", e);
				else
					Logger.warn("Metadata Schema validation FAILED with message: "+ e.getMessage());
				
				errString = e.getMessage();
				
			}
			
			throw new SchemaValidationException("Metadata Schema validation FAILED with message: "+ errString);
			
		} else		
			Logger.info("Metadata Schema validation check is DEACTIVATED!");
		
	}

}