package org.cateproject.marshal;
   
   import java.io.IOException;
   
   import javax.xml.bind.JAXBException;
   import javax.xml.bind.Marshaller;
   import javax.xml.bind.Unmarshaller;
   import javax.xml.parsers.ParserConfigurationException;
   import javax.xml.parsers.SAXParser;
  import javax.xml.parsers.SAXParserFactory;
  import javax.xml.transform.Source;
  import javax.xml.transform.sax.SAXSource;
  import javax.xml.validation.Schema;
  
  import org.apache.xml.resolver.tools.CatalogResolver;
  import org.cateproject.model.DataSet;
  import org.springframework.core.io.ClassPathResource;
  import org.springframework.core.io.Resource;
  import org.springframework.oxm.UncategorizedXmlMappingException;
  import org.springframework.oxm.XmlMappingException;
  import org.springframework.oxm.jaxb.Jaxb2Marshaller;
  import org.springframework.oxm.jaxb.JaxbSystemException;
  import org.springframework.xml.validation.SchemaLoaderUtils;
  import org.xml.sax.SAXException;
  import org.xml.sax.SAXNotRecognizedException;
  import org.xml.sax.XMLReader;
  
  import eu.etaxonomy.cdm.io.jaxb.CdmDocumentBuilder;
  import eu.etaxonomy.cdm.io.jaxb.CdmMarshallerListener;
  import eu.etaxonomy.cdm.io.jaxb.DefaultErrorHandler;
  import eu.etaxonomy.cdm.io.jaxb.WarningTolerantValidationEventHandler;
  import eu.etaxonomy.cdm.jaxb.CdmNamespacePrefixMapper;
  import eu.etaxonomy.cdm.jaxb.FormattedText;
  import eu.etaxonomy.cdm.jaxb.MultilanguageTextElement;
  
  
  /**
   * Initializes a JaxbContext with one class (org.cateproject.model.DataSet). 
   * 
   * @author ben.clark
   */
  public class CateDocumentBuilder extends Jaxb2Marshaller {
  	private boolean formattedOutput = Boolean.TRUE;
  	private String encoding = "UTF-8";
  	
  	public static String[] CATE_SCHEMA_FILES = {
  		                                        "/schema/xml/xinclude.xsd",
  		                                        "/schema/xml/xml.xsd",
  		                                        "/schema/xml/XMLSchema.xsd",
  		                                        "/schema/cate/cate.xsd",
  		                                        "/schema/cdm/agent.xsd",
                                                  "/schema/cdm/cdm.xsd",
                                                  "/schema/cdm/common.xsd",
                                                  "/schema/cdm/description.xsd",
                                                  "/schema/cdm/location.xsd",
                                                  "/schema/cdm/media.xsd",
                                                  "/schema/cdm/molecular.xsd",
                                                  "/schema/cdm/name.xsd",
                                                  "/schema/cdm/occurrence.xsd",
                                                  "/schema/cdm/reference.xsd",
                                                  "/schema/cdm/taxon.xsd"};
  	private Resource schemas[];
  	
  	public static Class[] CONTEXT_CLASSES = {DataSet.class,eu.etaxonomy.cdm.io.jaxb.DataSet.class,FormattedText.class,MultilanguageTextElement.class};
  
  	public CateDocumentBuilder() {
          schemas = new Resource[CATE_SCHEMA_FILES.length];
  		
  		for(int i = 0; i < CATE_SCHEMA_FILES.length; i++) {
  			schemas[i] = new ClassPathResource(CATE_SCHEMA_FILES[i]);
  		}
  		
  		super.setSchemas(schemas);
  		super.setClassesToBeBound(CONTEXT_CLASSES);
  		super.setSchemaLanguage("http://www.w3.org/2001/XMLSchema");
  	}
  	@Override
  	protected void initJaxbUnmarshaller(Unmarshaller unmarshaller) throws JAXBException { 
          unmarshaller.setEventHandler(new WarningTolerantValidationEventHandler());
  	}
  	
  	@Override
  	public Object unmarshal(Source source) throws XmlMappingException {
  		try {
  		SAXParserFactory saxParserFactory = SAXParserFactory.newInstance();
  		Schema schema  = SchemaLoaderUtils.loadSchema(schemas, "http://www.w3.org/2001/XMLSchema");
  			
          saxParserFactory.setNamespaceAware(true);
          saxParserFactory.setXIncludeAware(true);
          saxParserFactory.setValidating(true);
          saxParserFactory.setSchema(schema);
          
          SAXParser saxParser = saxParserFactory.newSAXParser();
          XMLReader xmlReader = saxParser.getXMLReader();
  	    xmlReader.setEntityResolver(new CatalogResolver());
  	    xmlReader.setErrorHandler(new DefaultErrorHandler());
  		SAXSource saxSource = new SAXSource( xmlReader, SAXSource.sourceToInputSource(source));
  		saxSource.setSystemId(source.getSystemId());
  		
 		return super.unmarshal(saxSource);
 		} catch (ParserConfigurationException pce) {
 			throw new JaxbSystemException(new JAXBException(pce));
 		} catch (SAXNotRecognizedException saxnre) {
 			throw new JaxbSystemException(new JAXBException(saxnre));
 		} catch (SAXException saxe) {
 			throw new JaxbSystemException(new JAXBException(saxe));
 		} catch (IOException ioe) {
 			throw new JaxbSystemException(new JAXBException(ioe));
 		}
 	}
 	
 	protected void initJaxbMarshaller(Marshaller marshaller) throws JAXBException {
 //        marshaller.setProperty("com.sun.xml.bind.namespacePrefixMapper", new CdmNamespacePrefixMapper() );
 //        
 //		// For test purposes insert newlines to make the XML output readable
 //		marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, formattedOutput);
 //		marshaller.setProperty(Marshaller.JAXB_SCHEMA_LOCATION,"http://etaxonomy.eu/cdm/model/1.0 schema/cdm/cdm.xsd");
 //		marshaller.setProperty(Marshaller.JAXB_ENCODING, encoding);
 //		
 //		CdmMarshallerListener marshallerListener = new CdmMarshallerListener();
 //		marshaller.setListener(marshallerListener);		
 //		marshaller.setEventHandler(new WarningTolerantValidationEventHandler());
 	}
 }