/**
* 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.cxf.tools.wsdlto.frontend.jaxws.processor.internal;
import java.io.File;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.xml.namespace.QName;
import org.w3c.dom.DOMException;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.apache.cxf.helpers.CastUtils;
import org.apache.cxf.helpers.JavaUtils;
import org.apache.cxf.jaxb.JAXBUtils;
import org.apache.cxf.service.model.MessagePartInfo;
import org.apache.cxf.service.model.ServiceInfo;
import org.apache.cxf.tools.common.ToolConstants;
import org.apache.cxf.tools.common.ToolContext;
import org.apache.cxf.tools.common.model.DefaultValueWriter;
import org.apache.cxf.tools.util.ClassCollector;
import org.apache.cxf.tools.util.NameUtil;
import org.apache.cxf.tools.util.URIParserUtil;
import org.apache.cxf.tools.wsdlto.core.DataBindingProfile;
import org.apache.ws.commons.schema.XmlSchema;
import org.apache.ws.commons.schema.XmlSchemaCollection;
import org.apache.ws.commons.schema.XmlSchemaComplexType;
import org.apache.ws.commons.schema.XmlSchemaElement;
import org.apache.ws.commons.schema.XmlSchemaForm;
import org.apache.ws.commons.schema.XmlSchemaObjectCollection;
import org.apache.ws.commons.schema.XmlSchemaSequence;
public final class ProcessorUtil {
private static final String KEYWORDS_PREFIX = "_";
private ProcessorUtil() {
}
public static String resolvePartName(MessagePartInfo part) {
return NameUtil.mangleNameToVariableName(part.getName().getLocalPart());
}
public static String getPartType(MessagePartInfo part) {
return part.getConcreteName().getLocalPart();
}
public static String resolvePartType(MessagePartInfo part) {
return NameUtil.mangleNameToClassName(getPartType(part));
}
public static String getType(MessagePartInfo part, ToolContext context, boolean fullname) {
String type = "";
DataBindingProfile dataBinding = context.get(DataBindingProfile.class);
if (part.isElement()) {
type = dataBinding.getType(getElementName(part), true);
} else {
type = dataBinding.getType(part.getTypeQName(), false);
}
if (type == null) {
type = resolvePartType(part);
}
return type;
}
public static DefaultValueWriter getDefaultValueWriter(MessagePartInfo part,
ToolContext context) {
DataBindingProfile dataBinding = context.get(DataBindingProfile.class);
if (part.isElement()) {
return dataBinding.createDefaultValueWriter(getElementName(part), true);
}
return dataBinding.createDefaultValueWriter(part.getTypeQName(), false);
}
public static DefaultValueWriter getDefaultValueWriterForWrappedElement(MessagePartInfo part,
ToolContext context,
QName subElement) {
DataBindingProfile dataBinding = context.get(DataBindingProfile.class);
return dataBinding.createDefaultValueWriterForWrappedElement(part.getElementQName(), subElement);
}
public static QName getElementName(MessagePartInfo part) {
return part == null ? null : part.getConcreteName();
}
//
// support multiple -p options
// if user change the package name through -p namespace=package name
//
public static QName getMappedElementName(MessagePartInfo part, ToolContext env) {
QName origin = getElementName(part);
if (origin == null) {
return null;
}
if (!env.hasNamespace(origin.getNamespaceURI())) {
return origin;
}
return new QName(env.getCustomizedNS(origin.getNamespaceURI()), origin.getLocalPart());
}
public static String resolvePartType(MessagePartInfo part, ToolContext env) {
if (env != null) {
return resolvePartType(part, env, false);
} else {
return resolvePartType(part);
}
}
public static String resolvePartType(MessagePartInfo part, ToolContext context, boolean fullName) {
DataBindingProfile dataBinding = context.get(DataBindingProfile.class);
if (dataBinding == null) {
String primitiveType = JAXBUtils.builtInTypeToJavaType(part.getTypeQName().getLocalPart());
if (part.getTypeQName() != null && primitiveType != null) {
return primitiveType;
} else {
return resolvePartType(part);
}
}
String name = "";
if (part.isElement()) {
name = dataBinding.getType(getElementName(part), true);
} else {
name = dataBinding.getType(part.getTypeQName(), false);
}
return name;
}
public static String resolvePartNamespace(MessagePartInfo part) {
return part.getConcreteName().getNamespaceURI();
}
public static String mangleNameToVariableName(String vName) {
String result = NameUtil.mangleNameToVariableName(vName);
if (JavaUtils.isJavaKeyword(result)) {
return KEYWORDS_PREFIX + result;
} else {
return result;
}
}
public static String parsePackageName(String namespace, String defaultPackageName) {
String packageName = (defaultPackageName != null && defaultPackageName.trim().length() > 0)
? defaultPackageName : null;
if (packageName == null) {
packageName = URIParserUtil.getPackageName(namespace);
}
return packageName;
}
public static String getAbsolutePath(String location) throws IOException {
if (location.startsWith("http://")) {
return location;
} else {
return resolvePath(new File(location).getAbsolutePath());
}
}
public static URL getWSDLURL(String location) throws Exception {
if (location.startsWith("http://")) {
return new URL(location);
} else {
return new File(getAbsolutePath(location)).toURI().toURL();
}
}
private static String resolvePath(String path) {
return path.replace('\\', '/');
}
public static String classNameToFilePath(String className) {
String str;
if (className.indexOf(".") < 0) {
return className;
} else {
str = className.replaceAll("\\.", "/");
}
return str;
}
//
// the non-wrapper style will get the type info from the part directly
//
public static String getFullClzName(MessagePartInfo part, ToolContext context, boolean primitiveType) {
DataBindingProfile dataBinding = context.get(DataBindingProfile.class);
String jtype = null;
QName xmlTypeName = getElementName(part);
// if this flag is true , mapping to java Type first;
// if not found , findd the primitive type : int ,long
// if not found, find in the generated class
if (!primitiveType && dataBinding != null) {
jtype = dataBinding.getType(xmlTypeName, true);
}
if (!primitiveType && dataBinding == null) {
Class holderClass = JAXBUtils.holderClass(xmlTypeName.getLocalPart());
jtype = holderClass == null ? null : holderClass.getName();
if (jtype == null) {
jtype = JAXBUtils.builtInTypeToJavaType(xmlTypeName.getLocalPart());
}
}
if (primitiveType) {
jtype = JAXBUtils.builtInTypeToJavaType(xmlTypeName.getLocalPart());
}
String namespace = xmlTypeName.getNamespaceURI();
String type = resolvePartType(part, context, true);
String userPackage = context.mapPackageName(namespace);
ClassCollector collector = context.get(ClassCollector.class);
if (jtype == null) {
jtype = collector.getTypesFullClassName(parsePackageName(namespace, userPackage), type);
}
if (jtype == null) {
if (!resolvePartType(part).equals(type)) {
jtype = resolvePartType(part, context, true);
} else {
jtype = parsePackageName(namespace, userPackage) + "." + type;
}
}
return jtype;
}
public static String getFileOrURLName(String fileOrURL) {
try {
try {
return escapeSpace(new URL(fileOrURL).toExternalForm());
} catch (MalformedURLException e) {
return new File(fileOrURL).getCanonicalFile().toURI().toURL().toExternalForm();
}
} catch (Exception e) {
return fileOrURL;
}
}
private static String escapeSpace(String url) {
StringBuffer buf = new StringBuffer();
for (int i = 0; i < url.length(); i++) {
if (url.charAt(i) == ' ') {
buf.append("%20");
} else {
buf.append(url.charAt(i));
}
}
return buf.toString();
}
public static String absolutize(String name) {
// absolutize all the system IDs in the input,
// so that we can map system IDs to DOM trees.
try {
URL baseURL = new File(".").getCanonicalFile().toURI().toURL();
return new URL(baseURL, name.replaceAll(" ", "%20")).toExternalForm();
} catch (IOException e) {
// ignore
}
return name;
}
public static String getHandlerConfigFileName(String name) {
return name + "_handler";
}
public static Node cloneNode(Document document, Node node, boolean deep) throws DOMException {
if (document == null || node == null) {
return null;
}
int type = node.getNodeType();
if (node.getOwnerDocument() == document) {
return node.cloneNode(deep);
}
Node clone;
switch (type) {
case Node.CDATA_SECTION_NODE:
clone = document.createCDATASection(node.getNodeValue());
break;
case Node.COMMENT_NODE:
clone = document.createComment(node.getNodeValue());
break;
case Node.ENTITY_REFERENCE_NODE:
clone = document.createEntityReference(node.getNodeName());
break;
case Node.ELEMENT_NODE:
clone = document.createElement(node.getNodeName());
NamedNodeMap attributes = node.getAttributes();
for (int i = 0; i < attributes.getLength(); i++) {
((Element)clone).setAttribute(attributes.item(i).getNodeName(), attributes.item(i)
.getNodeValue());
}
break;
case Node.TEXT_NODE:
clone = document.createTextNode(node.getNodeValue());
break;
default:
return null;
}
if (deep && type == Node.ELEMENT_NODE) {
Node child = node.getFirstChild();
while (child != null) {
clone.appendChild(cloneNode(document, child, true));
child = child.getNextSibling();
}
}
return clone;
}
public static List<QName> getWrappedElementQNames(ToolContext context, QName partElement) {
List<QName> qnames = new ArrayList<QName>();
for (WrapperElement element : getWrappedElement(context, partElement)) {
qnames.add(element.getElementName());
}
return qnames;
}
public static List<WrapperElement> getWrappedElement(ToolContext context, QName partElement) {
List<WrapperElement> qnames = new ArrayList<WrapperElement>();
ServiceInfo serviceInfo = (ServiceInfo)context.get(ServiceInfo.class);
XmlSchemaCollection schema = serviceInfo.getXmlSchemaCollection();
XmlSchemaElement elementByName = schema.getElementByQName(partElement);
//is elementByName is null,it could be generate from serviceInfo, we need read the schema element
//into schemaCollection
if (elementByName == null) {
Map<String, Element> maps = CastUtils.cast((Map<?, ?>)context.get(ToolConstants.SCHEMA_MAP));
if (maps != null) {
for (Element ele : maps.values()) {
schema.read(ele);
}
elementByName = schema.getElementByQName(partElement);
}
}
XmlSchemaComplexType type = (XmlSchemaComplexType)elementByName.getSchemaType();
XmlSchemaSequence seq = (XmlSchemaSequence)type.getParticle();
if (seq != null) {
XmlSchemaObjectCollection items = seq.getItems();
Iterator ite = items.getIterator();
while (ite.hasNext()) {
XmlSchemaElement subElement = (XmlSchemaElement)ite.next();
if (subElement.getQName() != null) {
qnames.add(new WrapperElement(subElement.getQName(), subElement.getSchemaTypeName()));
} else {
qnames.add(new WrapperElement(subElement.getRefName(), subElement.getSchemaTypeName()));
}
}
}
return qnames;
}
public static boolean isSchemaFormQualified(ToolContext context, QName partElement) {
ServiceInfo serviceInfo = (ServiceInfo)context.get(ServiceInfo.class);
XmlSchemaCollection schemaCol = serviceInfo.getXmlSchemaCollection();
for (int i = 0; i < schemaCol.getXmlSchemas().length; i++) {
XmlSchema schema = schemaCol.getXmlSchemas().clone()[i];
if (schema.getElementByName(partElement) != null) {
return schema.getElementFormDefault().getValue().equals(XmlSchemaForm.QUALIFIED);
}
}
return false;
}
}