server/openejb-axis/src/main/java/org/apache/openejb/server/axis/assembler/CommonsSchemaLoader.java - tomee - Git at Google

 /**
  *
  * Licensed to the Apache Software Foundation (ASF) under one or more
  * contributor license agreements.  See the NOTICE file distributed with
  * this work for additional information regarding copyright ownership.
  * The ASF licenses this file to You under the Apache License, Version 2.0
  * (the "License"); you may not use this file except in compliance with
  * the License.  You may obtain a copy of the License at
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  *  Unless required by applicable law or agreed to in writing, software
  *  distributed under the License is distributed on an "AS IS" BASIS,
  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  *  See the License for the specific language governing permissions and
  *  limitations under the License.
  */
 package org.apache.openejb.server.axis.assembler;

 import com.ibm.wsdl.extensions.PopulatedExtensionRegistry;
 import com.ibm.wsdl.extensions.schema.SchemaConstants;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.apache.openejb.OpenEJBException;
 import org.apache.openejb.server.ServerRuntimeException;
 import org.apache.ws.commons.schema.XmlSchemaCollection;
 import org.w3c.dom.Element;
 import org.xml.sax.InputSource;

 import javax.wsdl.Definition;
 import javax.wsdl.Import;
 import javax.wsdl.Types;
 import javax.wsdl.WSDLException;
 import javax.wsdl.extensions.ExtensionRegistry;
 import javax.wsdl.extensions.UnknownExtensibilityElement;
 import javax.wsdl.extensions.schema.Schema;
 import javax.wsdl.factory.WSDLFactory;
 import javax.wsdl.xml.WSDLLocator;
 import javax.wsdl.xml.WSDLReader;
 import javax.xml.namespace.QName;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URI;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.jar.JarFile;
 import java.util.zip.ZipEntry;

 public class CommonsSchemaLoader {
     private static final Log LOG = LogFactory.getLog(CommonsSchemaLoader.class);

     private final URI wsdlUri;
     private final JarFile moduleFile;
     private final XmlSchemaCollection xmlSchemaCollection = new XmlSchemaCollection();

     public CommonsSchemaLoader(URI wsdlUri, JarFile moduleFile) {
         this.wsdlUri = wsdlUri;
         this.moduleFile = moduleFile;
     }

     public XmlSchemaCollection loadSchema() throws OpenEJBException {
         Definition definition = readWsdl(wsdlUri);
         addImportsFromDefinition(definition);
         return xmlSchemaCollection;
     }

     private void addImportsFromDefinition(Definition definition) throws OpenEJBException {
         Types types = definition.getTypes();
         if (types != null) {
             for (Object extensibilityElement : types.getExtensibilityElements()) {
                 if (extensibilityElement instanceof Schema) {
                     Schema unknownExtensibilityElement = (Schema) extensibilityElement;
                     QName elementType = unknownExtensibilityElement.getElementType();
                     if (new QName("http://www.w3.org/2001/XMLSchema", "schema").equals(elementType)) {
                         Element element = unknownExtensibilityElement.getElement();
                         xmlSchemaCollection.read(element);
                     }
                 } else if (extensibilityElement instanceof UnknownExtensibilityElement) {
                     //This is allegedly obsolete as of axis-wsdl4j-1.2-RC3.jar which includes the Schema extension above.
                     //The change notes imply that imported schemas should end up in Schema elements.  They don't, so this is still needed.
                     UnknownExtensibilityElement unknownExtensibilityElement = (UnknownExtensibilityElement) extensibilityElement;
                     Element element = unknownExtensibilityElement.getElement();
                     String elementNamespace = element.getNamespaceURI();
                     String elementLocalName = element.getNodeName();
                     if ("http://www.w3.org/2001/XMLSchema".equals(elementNamespace) && "schema".equals(elementLocalName)) {
                         xmlSchemaCollection.read(element);
                     }
                 }
             }
         }

         //noinspection unchecked
         Map<String, List<Import>> imports = definition.getImports();
         if (imports != null) {
             for (Map.Entry<String, List<Import>> entry : imports.entrySet()) {
                 String namespaceURI = entry.getKey();
                 List<Import> importList = entry.getValue();
                 for (Import anImport : importList) {
                     //according to the 1.1 jwsdl mr shcema imports are supposed to show up here,
                     //but according to the 1.0 spec there is supposed to be no Definition.
                     Definition importedDef = anImport.getDefinition();
                     if (importedDef != null) {
                         addImportsFromDefinition(importedDef);
                     } else {
                         LOG.warn("Missing definition in import for namespace " + namespaceURI);
                     }
                 }
             }
         }
     }

     private Definition readWsdl(URI wsdlURI) throws OpenEJBException {
         Definition definition;
         WSDLFactory wsdlFactory;
         try {
             wsdlFactory = WSDLFactory.newInstance();
         } catch (WSDLException e) {
             throw new OpenEJBException("Could not create WSDLFactory", e);
         }
         WSDLReader wsdlReaderNoImport = wsdlFactory.newWSDLReader();
         wsdlReaderNoImport.setFeature("javax.wsdl.importDocuments", false);
         ExtensionRegistry extensionRegistry = new PopulatedExtensionRegistry();
         extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_1999,
             UnknownExtensibilityElement.class);
         extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_1999,
             extensionRegistry.getDefaultDeserializer());
         extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_1999,
             extensionRegistry.getDefaultSerializer());

         extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_2000,
             UnknownExtensibilityElement.class);
         extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_2000,
             extensionRegistry.getDefaultDeserializer());
         extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_2000,
             extensionRegistry.getDefaultSerializer());

         extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_2001,
             UnknownExtensibilityElement.class);
         extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_2001,
             extensionRegistry.getDefaultDeserializer());
         extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_2001,
             extensionRegistry.getDefaultSerializer());
         wsdlReaderNoImport.setExtensionRegistry(extensionRegistry);

         JarWSDLLocator wsdlLocator = new JarWSDLLocator(wsdlURI);
         WSDLReader wsdlReader = wsdlFactory.newWSDLReader();

         Thread thread = Thread.currentThread();
         ClassLoader oldCl = thread.getContextClassLoader();
         thread.setContextClassLoader(this.getClass().getClassLoader());
         try {
             try {
                 definition = wsdlReader.readWSDL(wsdlLocator);
             } catch (WSDLException e) {
                 throw new OpenEJBException("Failed to read wsdl document", e);
             } catch (RuntimeException e) {
                 throw new OpenEJBException(e.getMessage(), e);
             }
         } finally {
             thread.setContextClassLoader(oldCl);
         }

         return definition;
     }

     class JarWSDLLocator implements WSDLLocator {

         private final List<InputStream> streams = new ArrayList<>();
         private final URI wsdlURI;
         private URI latestImportURI;

         public JarWSDLLocator(URI wsdlURI) {
             this.wsdlURI = wsdlURI;
         }

         @Override
         public InputSource getBaseInputSource() {
             ZipEntry entry = moduleFile.getEntry(wsdlURI.toString());
             if (entry == null) {
                 throw new ServerRuntimeException("The webservices.xml file points to a non-existant WSDL file " + wsdlURI.toString());
             }

             InputStream wsdlInputStream;
             try {
                 wsdlInputStream = moduleFile.getInputStream(entry);
                 streams.add(wsdlInputStream);
             } catch (Exception e) {
                 throw new ServerRuntimeException("Could not open stream to wsdl file", e);
             }
             return new InputSource(wsdlInputStream);
         }

         @Override
         public String getBaseURI() {
             return wsdlURI.toString();
         }

         @Override
         public InputSource getImportInputSource(String parentLocation, String relativeLocation) {
             URI parentURI = URI.create(parentLocation);
             latestImportURI = parentURI.resolve(relativeLocation);

             InputStream importInputStream;
             try {
                 ZipEntry entry = moduleFile.getEntry(latestImportURI.toString());
                 importInputStream = moduleFile.getInputStream(entry);
                 streams.add(importInputStream);
             } catch (Exception e) {
                 throw new ServerRuntimeException("Could not open stream to import file", e);
             }

             InputSource inputSource = new InputSource(importInputStream);
             inputSource.setSystemId(getLatestImportURI());
             return inputSource;
         }

         @Override
         public String getLatestImportURI() {
             return latestImportURI.toString();
         }

         @Override
         public void close() {
             for (InputStream inputStream : streams) {
                 try {
                     inputStream.close();
                 } catch (IOException e) {
                     //ignore
                 }
             }
             streams.clear();
         }
     }
 }
	/**
	*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You under the Apache License, Version 2.0
	* (the "License"); you may not use this file except in compliance with
	* the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	package org.apache.openejb.server.axis.assembler;

	import com.ibm.wsdl.extensions.PopulatedExtensionRegistry;
	import com.ibm.wsdl.extensions.schema.SchemaConstants;
	import org.apache.commons.logging.Log;
	import org.apache.commons.logging.LogFactory;
	import org.apache.openejb.OpenEJBException;
	import org.apache.openejb.server.ServerRuntimeException;
	import org.apache.ws.commons.schema.XmlSchemaCollection;
	import org.w3c.dom.Element;
	import org.xml.sax.InputSource;

	import javax.wsdl.Definition;
	import javax.wsdl.Import;
	import javax.wsdl.Types;
	import javax.wsdl.WSDLException;
	import javax.wsdl.extensions.ExtensionRegistry;
	import javax.wsdl.extensions.UnknownExtensibilityElement;
	import javax.wsdl.extensions.schema.Schema;
	import javax.wsdl.factory.WSDLFactory;
	import javax.wsdl.xml.WSDLLocator;
	import javax.wsdl.xml.WSDLReader;
	import javax.xml.namespace.QName;
	import java.io.IOException;
	import java.io.InputStream;
	import java.net.URI;
	import java.util.ArrayList;
	import java.util.List;
	import java.util.Map;
	import java.util.jar.JarFile;
	import java.util.zip.ZipEntry;

	public class CommonsSchemaLoader {
	private static final Log LOG = LogFactory.getLog(CommonsSchemaLoader.class);

	private final URI wsdlUri;
	private final JarFile moduleFile;
	private final XmlSchemaCollection xmlSchemaCollection = new XmlSchemaCollection();

	public CommonsSchemaLoader(URI wsdlUri, JarFile moduleFile) {
	this.wsdlUri = wsdlUri;
	this.moduleFile = moduleFile;
	}

	public XmlSchemaCollection loadSchema() throws OpenEJBException {
	Definition definition = readWsdl(wsdlUri);
	addImportsFromDefinition(definition);
	return xmlSchemaCollection;
	}

	private void addImportsFromDefinition(Definition definition) throws OpenEJBException {
	Types types = definition.getTypes();
	if (types != null) {
	for (Object extensibilityElement : types.getExtensibilityElements()) {
	if (extensibilityElement instanceof Schema) {
	Schema unknownExtensibilityElement = (Schema) extensibilityElement;
	QName elementType = unknownExtensibilityElement.getElementType();
	if (new QName("http://www.w3.org/2001/XMLSchema", "schema").equals(elementType)) {
	Element element = unknownExtensibilityElement.getElement();
	xmlSchemaCollection.read(element);
	}
	} else if (extensibilityElement instanceof UnknownExtensibilityElement) {
	//This is allegedly obsolete as of axis-wsdl4j-1.2-RC3.jar which includes the Schema extension above.
	//The change notes imply that imported schemas should end up in Schema elements. They don't, so this is still needed.
	UnknownExtensibilityElement unknownExtensibilityElement = (UnknownExtensibilityElement) extensibilityElement;
	Element element = unknownExtensibilityElement.getElement();
	String elementNamespace = element.getNamespaceURI();
	String elementLocalName = element.getNodeName();
	if ("http://www.w3.org/2001/XMLSchema".equals(elementNamespace) && "schema".equals(elementLocalName)) {
	xmlSchemaCollection.read(element);
	}
	}
	}
	}

	//noinspection unchecked
	Map<String, List<Import>> imports = definition.getImports();
	if (imports != null) {
	for (Map.Entry<String, List<Import>> entry : imports.entrySet()) {
	String namespaceURI = entry.getKey();
	List<Import> importList = entry.getValue();
	for (Import anImport : importList) {
	//according to the 1.1 jwsdl mr shcema imports are supposed to show up here,
	//but according to the 1.0 spec there is supposed to be no Definition.
	Definition importedDef = anImport.getDefinition();
	if (importedDef != null) {
	addImportsFromDefinition(importedDef);
	} else {
	LOG.warn("Missing definition in import for namespace " + namespaceURI);
	}
	}
	}
	}
	}

	private Definition readWsdl(URI wsdlURI) throws OpenEJBException {
	Definition definition;
	WSDLFactory wsdlFactory;
	try {
	wsdlFactory = WSDLFactory.newInstance();
	} catch (WSDLException e) {
	throw new OpenEJBException("Could not create WSDLFactory", e);
	}
	WSDLReader wsdlReaderNoImport = wsdlFactory.newWSDLReader();
	wsdlReaderNoImport.setFeature("javax.wsdl.importDocuments", false);
	ExtensionRegistry extensionRegistry = new PopulatedExtensionRegistry();
	extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_1999,
	UnknownExtensibilityElement.class);
	extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_1999,
	extensionRegistry.getDefaultDeserializer());
	extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_1999,
	extensionRegistry.getDefaultSerializer());

	extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_2000,
	UnknownExtensibilityElement.class);
	extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_2000,
	extensionRegistry.getDefaultDeserializer());
	extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_2000,
	extensionRegistry.getDefaultSerializer());

	extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_2001,
	UnknownExtensibilityElement.class);
	extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_2001,
	extensionRegistry.getDefaultDeserializer());
	extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_2001,
	extensionRegistry.getDefaultSerializer());
	wsdlReaderNoImport.setExtensionRegistry(extensionRegistry);

	JarWSDLLocator wsdlLocator = new JarWSDLLocator(wsdlURI);
	WSDLReader wsdlReader = wsdlFactory.newWSDLReader();

	Thread thread = Thread.currentThread();
	ClassLoader oldCl = thread.getContextClassLoader();
	thread.setContextClassLoader(this.getClass().getClassLoader());
	try {
	try {
	definition = wsdlReader.readWSDL(wsdlLocator);
	} catch (WSDLException e) {
	throw new OpenEJBException("Failed to read wsdl document", e);
	} catch (RuntimeException e) {
	throw new OpenEJBException(e.getMessage(), e);
	}
	} finally {
	thread.setContextClassLoader(oldCl);
	}

	return definition;
	}

	class JarWSDLLocator implements WSDLLocator {

	private final List<InputStream> streams = new ArrayList<>();
	private final URI wsdlURI;
	private URI latestImportURI;

	public JarWSDLLocator(URI wsdlURI) {
	this.wsdlURI = wsdlURI;
	}

	@Override
	public InputSource getBaseInputSource() {
	ZipEntry entry = moduleFile.getEntry(wsdlURI.toString());
	if (entry == null) {
	throw new ServerRuntimeException("The webservices.xml file points to a non-existant WSDL file " + wsdlURI.toString());
	}

	InputStream wsdlInputStream;
	try {
	wsdlInputStream = moduleFile.getInputStream(entry);
	streams.add(wsdlInputStream);
	} catch (Exception e) {
	throw new ServerRuntimeException("Could not open stream to wsdl file", e);
	}
	return new InputSource(wsdlInputStream);
	}

	@Override
	public String getBaseURI() {
	return wsdlURI.toString();
	}

	@Override
	public InputSource getImportInputSource(String parentLocation, String relativeLocation) {
	URI parentURI = URI.create(parentLocation);
	latestImportURI = parentURI.resolve(relativeLocation);

	InputStream importInputStream;
	try {
	ZipEntry entry = moduleFile.getEntry(latestImportURI.toString());
	importInputStream = moduleFile.getInputStream(entry);
	streams.add(importInputStream);
	} catch (Exception e) {
	throw new ServerRuntimeException("Could not open stream to import file", e);
	}

	InputSource inputSource = new InputSource(importInputStream);
	inputSource.setSystemId(getLatestImportURI());
	return inputSource;
	}

	@Override
	public String getLatestImportURI() {
	return latestImportURI.toString();
	}

	@Override
	public void close() {
	for (InputStream inputStream : streams) {
	try {
	inputStream.close();
	} catch (IOException e) {
	//ignore
	}
	}
	streams.clear();
	}
	}
	}