10:55 a.m.
Author: alessio.soldano(a)jboss.com
Date: 2010-03-31 11:55:22 -0400 (Wed, 31 Mar 2010)
New Revision: 11910
Added:
common/trunk/src/main/java/org/jboss/ws/core/utils/AbstractWSDLFilePublisher.java
Modified:
common/trunk/pom.xml
Log:
[JBWS-2986] Adding AbstractWSDLFilePublisher to common
Modified: common/trunk/pom.xml
===================================================================
--- common/trunk/pom.xml 2010-03-31 13:57:29 UTC (rev 11909)
+++ common/trunk/pom.xml 2010-03-31 15:55:22 UTC (rev 11910)
@@ -34,6 +34,7 @@
<jboss.web.version>2.1.3.GA</jboss.web.version>
<jbossws.spi.version>1.3.0-SNAPSHOT</jbossws.spi.version>
<junit.version>3.8.2</junit.version>
+ <wsdl4j.version>1.6.2</wsdl4j.version>
</properties>
<!-- Dependencies -->
@@ -92,6 +93,12 @@
<version>${jboss.web.version}</version>
<scope>provided</scope>
</dependency>
+ <dependency>
+ <groupId>wsdl4j</groupId>
+ <artifactId>wsdl4j</artifactId>
+ <version>${wsdl4j.version}</version>
+ <scope>provided</scope>
+ </dependency>
<!-- test dependencies -->
<dependency>
Added: common/trunk/src/main/java/org/jboss/ws/core/utils/AbstractWSDLFilePublisher.java
===================================================================
--- common/trunk/src/main/java/org/jboss/ws/core/utils/AbstractWSDLFilePublisher.java
(rev 0)
+++
common/trunk/src/main/java/org/jboss/ws/core/utils/AbstractWSDLFilePublisher.java 2010-03-31
15:55:22 UTC (rev 11910)
@@ -0,0 +1,244 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2010, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.ws.core.utils;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URL;
+import java.util.Iterator;
+import java.util.List;
+
+import javax.wsdl.Definition;
+import javax.wsdl.Import;
+import javax.wsdl.factory.WSDLFactory;
+
+import org.jboss.logging.Logger;
+import org.jboss.wsf.common.DOMUtils;
+import org.jboss.wsf.common.IOUtils;
+import org.jboss.wsf.spi.SPIProvider;
+import org.jboss.wsf.spi.SPIProviderResolver;
+import org.jboss.wsf.spi.deployment.ArchiveDeployment;
+import org.jboss.wsf.spi.management.ServerConfig;
+import org.jboss.wsf.spi.management.ServerConfigFactory;
+import org.w3c.dom.Element;
+
+/**
+ * Abstract WSDL file publisher
+ *
+ * @author alessio.soldano(a)jboss.com
+ * @since 25-Mar-2010
+ *
+ */
+public abstract class AbstractWSDLFilePublisher
+{
+ private static final Logger log = Logger.getLogger(AbstractWSDLFilePublisher.class);
+
+ // The deployment info for the web service archive
+ protected ArchiveDeployment dep;
+ // The expected wsdl location in the deployment
+ protected String expLocation;
+ // The server config
+ protected ServerConfig serverConfig;
+
+ public AbstractWSDLFilePublisher(ArchiveDeployment dep)
+ {
+ this.dep = dep;
+
+ SPIProvider spiProvider = SPIProviderResolver.getInstance().getProvider();
+ serverConfig = spiProvider.getSPI(ServerConfigFactory.class).getServerConfig();
+
+ if (dep.getType().toString().endsWith("JSE"))
+ {
+ expLocation = "WEB-INF/wsdl/";
+ }
+ else
+ {
+ expLocation = "META-INF/wsdl/";
+ }
+ }
+
+ /** Publish the wsdl imports for a given wsdl definition
+ */
+ @SuppressWarnings("unchecked")
+ protected void publishWsdlImports(URL parentURL, Definition parentDefinition,
List<String> published) throws Exception
+ {
+ String baseURI = parentURL.toExternalForm();
+
+ Iterator it = parentDefinition.getImports().values().iterator();
+ while (it.hasNext())
+ {
+ for (Import wsdlImport : (List<Import>)it.next())
+ {
+ String locationURI = wsdlImport.getLocationURI();
+ Definition subdef = wsdlImport.getDefinition();
+
+ // its an external import, don't publish locally
+ if (locationURI.startsWith("http://") == false)
+ {
+ // infinity loops prevention
+ if (published.contains(locationURI))
+ {
+ continue;
+ }
+ else
+ {
+ published.add(locationURI);
+ }
+
+ URL targetURL = new URL(baseURI.substring(0,
baseURI.lastIndexOf("/") + 1) + locationURI);
+ File targetFile = new File(targetURL.getPath());
+ targetFile.getParentFile().mkdirs();
+
+ WSDLFactory wsdlFactory = WSDLFactory.newInstance();
+ javax.wsdl.xml.WSDLWriter wsdlWriter = wsdlFactory.newWSDLWriter();
+ FileWriter fw = new FileWriter(targetFile);
+ wsdlWriter.writeWSDL(subdef, fw);
+ fw.close();
+
+ log.debug("WSDL import published to: " + targetURL);
+
+ // recursively publish imports
+ publishWsdlImports(targetURL, subdef, published);
+
+ // Publish XMLSchema imports
+ Element subdoc = DOMUtils.parse(targetURL.openStream());
+ publishSchemaImports(targetURL, subdoc, published);
+ }
+ }
+ }
+ }
+
+ /** Publish the schema imports for a given wsdl definition
+ */
+ protected void publishSchemaImports(URL parentURL, Element element, List<String>
published) throws Exception
+ {
+ String baseURI = parentURL.toExternalForm();
+
+ Iterator<Element> it = DOMUtils.getChildElements(element);
+ while (it.hasNext())
+ {
+ Element childElement = (Element)it.next();
+ if ("import".equals(childElement.getLocalName()) ||
"include".equals(childElement.getLocalName()))
+ {
+ String schemaLocation =
childElement.getAttribute("schemaLocation");
+ if (schemaLocation.length() > 0)
+ {
+ if (schemaLocation.startsWith("http://") == false)
+ {
+ // infinity loops prevention
+ if (published.contains(schemaLocation))
+ {
+ continue;
+ }
+ else
+ {
+ published.add(schemaLocation);
+ }
+
+ URL xsdURL = new URL(baseURI.substring(0,
baseURI.lastIndexOf("/") + 1) + schemaLocation);
+ File targetFile = new File(xsdURL.getPath());
+ targetFile.getParentFile().mkdirs();
+
+ String deploymentName = dep.getCanonicalName();
+
+ // get the resource path including the separator
+ int index = baseURI.indexOf(deploymentName) + 1;
+ String resourcePath = baseURI.substring(index +
deploymentName.length());
+ //check for sub-directories
+ resourcePath = resourcePath.substring(0,
resourcePath.lastIndexOf("/") + 1);
+
+ resourcePath = expLocation + resourcePath + schemaLocation;
+ while (resourcePath.indexOf("//") != -1)
+ {
+ resourcePath = resourcePath.replace("//", "/");
+ }
+ URL resourceURL = dep.getResourceResolver().resolve(resourcePath);
+// URL resourceURL = dep.getMetaDataFileURL(resourcePath);
+ InputStream is = new ResourceURL(resourceURL).openStream();
+ if (is == null)
+ throw new IllegalArgumentException("Cannot find schema import
in deployment: " + resourcePath);
+
+ FileOutputStream fos = null;
+ try
+ {
+ fos = new FileOutputStream(targetFile);
+ IOUtils.copyStream(fos, is);
+ }
+ finally
+ {
+ if (fos != null) fos.close();
+ }
+
+ log.debug("XMLSchema import published to: " + xsdURL);
+
+ // recursively publish imports
+ Element subdoc = DOMUtils.parse(xsdURL.openStream());
+ publishSchemaImports(xsdURL, subdoc, published);
+ }
+ }
+ }
+ else
+ {
+ publishSchemaImports(parentURL, childElement, published);
+ }
+ }
+ }
+
+ /**
+ * Delete the published wsdl
+ */
+ public void unpublishWsdlFiles() throws IOException
+ {
+ String deploymentDir = (dep.getParent() != null ? dep.getParent().getSimpleName() :
dep.getSimpleName());
+
+ File serviceDir = new File(serverConfig.getServerDataDir().getCanonicalPath() +
"/wsdl/" + deploymentDir);
+ deleteWsdlPublishDirectory(serviceDir);
+ }
+
+ /**
+ * Delete the published wsdl document, traversing down the dir structure
+ */
+ protected void deleteWsdlPublishDirectory(File dir) throws IOException
+ {
+ String[] files = dir.list();
+ for (int i = 0; files != null && i < files.length; i++)
+ {
+ String fileName = files[i];
+ File file = new File(dir + "/" + fileName);
+ if (file.isDirectory())
+ {
+ deleteWsdlPublishDirectory(file);
+ }
+ else
+ {
+ if (file.delete() == false)
+ log.warn("Cannot delete published wsdl document: " +
file.toURL());
+ }
+ }
+
+ // delete the directory as well
+ dir.delete();
+ }
+}
Property changes on:
common/trunk/src/main/java/org/jboss/ws/core/utils/AbstractWSDLFilePublisher.java
___________________________________________________________________
Name: svn:keywords
+ Id Revision
Name: svn:eol-style
+ LF