/* * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. * * Copyright (c) 1997-2013 Oracle and/or its affiliates. All rights reserved. * * The contents of this file are subject to the terms of either the GNU * General Public License Version 2 only ("GPL") or the Common Development * and Distribution License("CDDL") (collectively, the "License"). You * may not use this file except in compliance with the License. You can * obtain a copy of the License at * http://glassfish.java.net/public/CDDL+GPL_1_1.html * or packager/legal/LICENSE.txt. See the License for the specific * language governing permissions and limitations under the License. * * When distributing the software, include this License Header Notice in each * file and include the License file at packager/legal/LICENSE.txt. * * GPL Classpath Exception: * Oracle designates this particular file as subject to the "Classpath" * exception as provided by Oracle in the GPL Version 2 section of the License * file that accompanied this code. * * Modifications: * If applicable, add the following below the License Header, with the fields * enclosed by brackets [] replaced by your own identifying information: * "Portions Copyright [year] [name of copyright owner]" * * Contributor(s): * If you wish your version of this file to be governed by only the CDDL or * only the GPL Version 2, indicate your decision by adding "[Contributor] * elects to include this software in this distribution under the [CDDL or GPL * Version 2] license." If you don't indicate a single choice of license, a * recipient has the option to distribute your version of this file under * either the CDDL, the GPL Version 2 or to extend the choice of license to * its licensees as provided above. However, if you add GPL Version 2 code * and therefore, elected the GPL Version 2 license, then the option applies * only if the new code is made subject to such option by the copyright * holder. */ package com.sun.tools.ws.util; import com.sun.istack.NotNull; import com.sun.tools.ws.resources.WscompileMessages; import com.sun.tools.ws.wscompile.WsimportListener; import com.sun.tools.ws.wscompile.WsimportOptions; import com.sun.tools.ws.wsdl.parser.DOMForest; import com.sun.tools.ws.wsdl.parser.MetadataFinder; import com.sun.xml.txw2.output.IndentingXMLStreamWriter; import com.sun.xml.ws.api.server.PortAddressResolver; import com.sun.xml.ws.streaming.SourceReaderFactory; import com.sun.xml.ws.wsdl.parser.WSDLConstants; import com.sun.xml.ws.wsdl.writer.DocumentLocationResolver; import com.sun.xml.ws.wsdl.writer.WSDLPatcher; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.Node; import org.w3c.dom.NodeList; import javax.xml.namespace.QName; import javax.xml.stream.XMLOutputFactory; import javax.xml.stream.XMLStreamException; import javax.xml.stream.XMLStreamReader; import javax.xml.stream.XMLStreamWriter; import javax.xml.transform.dom.DOMSource; import java.io.*; import java.net.MalformedURLException; import java.net.URL; import java.util.HashMap; import java.util.Map; import java.util.Set; /** * @author Rama Pulavarthi */ public class WSDLFetcher { private WsimportOptions options; private WsimportListener listener; public WSDLFetcher(WsimportOptions options, WsimportListener listener) { this.options = options; this.listener = listener; } /** * Fetches the wsdls in the DOMForest to the options.destDir * @param forest * @return location of fetched root WSDL document * @throws IOException * @throws XMLStreamException * @throws FileNotFoundException */ public String fetchWsdls(MetadataFinder forest) throws IOException, XMLStreamException { String rootWsdl = null; for(String root: forest.getRootDocuments()) { rootWsdl = root; } Set<String> externalRefs = forest.getExternalReferences(); Map<String,String> documentMap = createDocumentMap(forest, getWSDLDownloadDir(), rootWsdl, externalRefs); String rootWsdlName = fetchFile(rootWsdl,forest, documentMap,getWSDLDownloadDir()); for(String reference: forest.getExternalReferences()) { fetchFile(reference,forest,documentMap,getWSDLDownloadDir()); } return WSDL_PATH +"/" + rootWsdlName; } private String fetchFile(final String doc, DOMForest forest, final Map<String, String> documentMap, File destDir) throws IOException, XMLStreamException { DocumentLocationResolver docLocator = createDocResolver(doc, forest, documentMap); WSDLPatcher wsdlPatcher = new WSDLPatcher(new PortAddressResolver() { @Override public String getAddressFor(@NotNull QName serviceName, @NotNull String portName) { return null; } }, docLocator); XMLStreamReader xsr = null; XMLStreamWriter xsw = null; OutputStream os = null; String resolvedRootWsdl = null; try { XMLOutputFactory writerfactory; xsr = SourceReaderFactory.createSourceReader(new DOMSource(forest.get(doc)), false); writerfactory = XMLOutputFactory.newInstance(); resolvedRootWsdl = docLocator.getLocationFor(null, doc); File outFile = new File(destDir, resolvedRootWsdl); os = new FileOutputStream(outFile); if(options.verbose) { listener.message(WscompileMessages.WSIMPORT_DOCUMENT_DOWNLOAD(doc,outFile)); } xsw = writerfactory.createXMLStreamWriter(os); //DOMForest eats away the whitespace loosing all the indentation, so write it through // indenting writer for better readability of fetched documents IndentingXMLStreamWriter indentingWriter = new IndentingXMLStreamWriter(xsw); wsdlPatcher.bridge(xsr, indentingWriter); options.addGeneratedFile(outFile); } finally { try { if (xsr != null) {xsr.close();} if (xsw != null) {xsw.close();} } finally { if (os != null) {os.close();} } } return resolvedRootWsdl; } private Map<String,String> createDocumentMap(MetadataFinder forest, File baseDir, final String rootWsdl, Set<String> externalReferences) { Map<String,String> map = new HashMap<String,String>(); String rootWsdlFileName = rootWsdl; String rootWsdlName; int slashIndex = rootWsdl.lastIndexOf("/"); if( slashIndex >= 0) { rootWsdlFileName = rootWsdl.substring(slashIndex+1); } if(!rootWsdlFileName.endsWith(WSDL_FILE_EXTENSION)) { Document rootWsdlDoc = forest.get(rootWsdl); NodeList serviceNodes = rootWsdlDoc.getElementsByTagNameNS(WSDLConstants.QNAME_SERVICE.getNamespaceURI(),WSDLConstants.QNAME_SERVICE.getLocalPart()); if (serviceNodes.getLength() == 0) { rootWsdlName = "Service"; } else { Node serviceNode = serviceNodes.item(0); String serviceName = ((Element)serviceNode).getAttribute( WSDLConstants.ATTR_NAME); rootWsdlName = serviceName; } rootWsdlFileName = rootWsdlName+ WSDL_FILE_EXTENSION; } else { rootWsdlName = rootWsdlFileName.substring(0,rootWsdlFileName.length()-5); } map.put(rootWsdl,sanitize(rootWsdlFileName)); int i =1; for(String ref: externalReferences) { Document refDoc = forest.get(ref); Element rootEl = refDoc.getDocumentElement(); String fileExtn; String fileName = null; int index = ref.lastIndexOf("/"); if (index >= 0) { fileName = ref.substring(index + 1); } if(rootEl.getLocalName().equals(WSDLConstants.QNAME_DEFINITIONS.getLocalPart()) && rootEl.getNamespaceURI().equals(WSDLConstants.NS_WSDL)) { fileExtn = WSDL_FILE_EXTENSION; } else if(rootEl.getLocalName().equals(WSDLConstants.QNAME_SCHEMA.getLocalPart()) && rootEl.getNamespaceURI().equals(WSDLConstants.NS_XMLNS)) { fileExtn = SCHEMA_FILE_EXTENSION; } else { fileExtn = ".xml"; } if(fileName != null && (fileName.endsWith(WSDL_FILE_EXTENSION) || fileName.endsWith(SCHEMA_FILE_EXTENSION))) { map.put(ref, rootWsdlName+"_"+fileName); } else { map.put(ref, rootWsdlName+"_metadata"+ (i++) + fileExtn); } } return map; } private DocumentLocationResolver createDocResolver(final String baseWsdl, final DOMForest forest, final Map<String,String> documentMap) { return new DocumentLocationResolver() { @Override public String getLocationFor(String namespaceURI, String systemId) { try { URL reference = new URL(new URL(baseWsdl),systemId); systemId = reference.toExternalForm(); } catch (MalformedURLException e) { throw new RuntimeException(e); } if(documentMap.get(systemId) != null) { return documentMap.get(systemId); } else { String parsedEntity = forest.getReferencedEntityMap().get(systemId); return documentMap.get(parsedEntity); } } }; } private String sanitize(String fileName) { fileName = fileName.replace('?', '.'); StringBuilder sb = new StringBuilder(fileName); for (int i = 0; i < sb.length(); i++) { char c = sb.charAt(i); if (Character.isLetterOrDigit(c) || (c == '/') || (c == '.') || (c == '_') || (c == ' ') || (c == '-')) { continue; } else { sb.setCharAt(i, '_'); } } return sb.toString(); } private File getWSDLDownloadDir() { File wsdlDir = new File(options.destDir, WSDL_PATH); boolean created = wsdlDir.mkdirs(); if (options.verbose && !created) { listener.message(WscompileMessages.WSCOMPILE_NO_SUCH_DIRECTORY(wsdlDir)); } return wsdlDir; } private static String WSDL_PATH="META-INF/wsdl"; private static String WSDL_FILE_EXTENSION=".wsdl"; private static String SCHEMA_FILE_EXTENSION=".xsd"; }