/*
* 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.axis2.description;
import org.apache.axiom.om.OMAbstractFactory;
import org.apache.axiom.om.OMAttribute;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMFactory;
import org.apache.axiom.om.OMNamespace;
import org.apache.axiom.om.OMNode;
import org.apache.axiom.soap.SOAP11Constants;
import org.apache.axiom.soap.SOAP12Constants;
import org.apache.axis2.AxisFault;
import org.apache.axis2.addressing.AddressingConstants;
import org.apache.axis2.addressing.AddressingHelper;
import org.apache.axis2.description.java2wsdl.Java2WSDLConstants;
import org.apache.axis2.engine.AxisConfiguration;
import org.apache.axis2.namespace.Constants;
import org.apache.axis2.util.ExternalPolicySerializer;
import org.apache.axis2.util.JavaUtils;
import org.apache.axis2.util.PolicyLocator;
import org.apache.axis2.util.PolicyUtil;
import org.apache.axis2.util.WSDLSerializationUtil;
import org.apache.axis2.util.XMLUtils;
import org.apache.axis2.wsdl.SOAPHeaderMessage;
import org.apache.axis2.wsdl.WSDLConstants;
import org.apache.neethi.Policy;
import org.apache.neethi.PolicyComponent;
import org.apache.neethi.PolicyReference;
import org.apache.neethi.PolicyRegistry;
import org.apache.ws.commons.schema.XmlSchema;
import javax.xml.namespace.QName;
import java.io.StringReader;
import java.io.StringWriter;
import java.net.URI;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
public class AxisService2WSDL11 implements Java2WSDLConstants {
private AxisService axisService;
private String serviceName;
private String[] serviceEndpointURLs;
private String targetNamespace;
private OMElement definition;
private OMNamespace soap;
private OMNamespace soap12;
private OMNamespace http;
private OMNamespace mime;
private OMNamespace tns;
private OMNamespace wsdl;
private OMNamespace wsaw;
private String style = DOCUMENT;
private String use = LITERAL;
private HashMap policiesInDefinitions;
private ExternalPolicySerializer serializer;
private HashMap messagesMap;
public AxisService2WSDL11(AxisService service) throws Exception {
this.axisService = service;
this.serviceName = service.getName();
init();
}
private void init() throws AxisFault {
// the EPR list of AxisService contains REST EPRs as well. Those REST
// EPRs will be used to generated HTTPBinding
// and rest of the EPRs will be used to generate SOAP 1.1 and 1.2
// bindings. Let's first initialize those set of
// EPRs now to be used later, especially when we generate the WSDL.
serviceEndpointURLs = axisService.getEPRs();
if (serviceEndpointURLs == null) {
Map endpointMap = axisService.getEndpoints();
if (endpointMap.size() > 0) {
Iterator endpointItr = endpointMap.values().iterator();
if (endpointItr.hasNext()) {
AxisEndpoint endpoint = (AxisEndpoint) endpointItr.next();
serviceEndpointURLs = new String[] { endpoint
.getEndpointURL() };
}
} else {
serviceEndpointURLs = new String[] { axisService
.getEndpointName() };
}
}
this.targetNamespace = axisService.getTargetNamespace();
serializer = new ExternalPolicySerializer();
// CHECKME check whether service.getAxisConfiguration() return null ???
AxisConfiguration configuration = axisService.getAxisConfiguration();
if (configuration != null) {
serializer.setAssertionsToFilter(configuration
.getLocalPolicyAssertions());
}
}
public AxisService2WSDL11(AxisService service, String serviceName)
throws Exception {
this.axisService = service;
this.serviceName = serviceName;
init();
}
public OMElement generateOM() throws Exception {
OMFactory fac = OMAbstractFactory.getOMFactory();
wsdl = fac.createOMNamespace(WSDL_NAMESPACE,
DEFAULT_WSDL_NAMESPACE_PREFIX);
OMElement ele = fac.createOMElement("definitions", wsdl);
setDefinitionElement(ele);
policiesInDefinitions = new HashMap();
Map namespaceMap = axisService.getNamespaceMap();
if (namespaceMap == null)
namespaceMap = new HashMap();
WSDLSerializationUtil.populateNamespaces(ele, namespaceMap);
soap = ele.declareNamespace(URI_WSDL11_SOAP, SOAP11_PREFIX);
soap12 = ele.declareNamespace(URI_WSDL12_SOAP, SOAP12_PREFIX);
http = ele.declareNamespace(HTTP_NAMESPACE, HTTP_PREFIX);
mime = ele.declareNamespace(MIME_NAMESPACE, MIME_PREFIX);
wsaw = ele.declareNamespace(AddressingConstants.Final.WSAW_NAMESPACE,
"wsaw");
String prefix = WSDLSerializationUtil.getPrefix(axisService
.getTargetNamespace(), namespaceMap);
if (prefix == null || "".equals(prefix)) {
prefix = DEFAULT_TARGET_NAMESPACE_PREFIX;
}
namespaceMap.put(prefix, axisService.getTargetNamespace());
tns = ele.declareNamespace(axisService.getTargetNamespace(), prefix);
// axis2.xml indicated no HTTP binding?
boolean disableREST = false;
Parameter disableRESTParameter = axisService
.getParameter(org.apache.axis2.Constants.Configuration.DISABLE_REST);
if (disableRESTParameter != null
&& JavaUtils.isTrueExplicitly(disableRESTParameter.getValue())) {
disableREST = true;
}
// axis2.xml indicated no SOAP 1.2 binding?
boolean disableSOAP12 = false;
Parameter disableSOAP12Parameter = axisService
.getParameter(org.apache.axis2.Constants.Configuration.DISABLE_SOAP12);
if (disableSOAP12Parameter != null
&& JavaUtils
.isTrueExplicitly(disableSOAP12Parameter.getValue())) {
disableSOAP12 = true;
}
// adding documentation element
// <documentation><b>NEW!</b> This method accepts an ISBN
// string and returns <b>Amazon.co.uk</b> Sales Rank for
// that book.</documentation>
WSDLSerializationUtil.addWSDLDocumentationElement(axisService, ele,
fac, wsdl);
ele.addAttribute("targetNamespace", axisService.getTargetNamespace(),
null);
OMElement wsdlTypes = fac.createOMElement("types", wsdl);
ele.addChild(wsdlTypes);
// populate the schema mappings
axisService.populateSchemaMappings();
ArrayList schemas = axisService.getSchema();
for (int i = 0; i < schemas.size(); i++) {
StringWriter writer = new StringWriter();
// XmlSchema schema = (XmlSchema) schemas.get(i);
XmlSchema schema = axisService.getSchema(i);
String targetNamespace = schema.getTargetNamespace();
if (!Constants.NS_URI_XML.equals(targetNamespace)) {
schema.write(writer);
String schemaString = writer.toString();
if (!"".equals(schemaString)) {
wsdlTypes.addChild(XMLUtils.toOM(new StringReader(
schemaString)));
}
}
}
generateMessages(fac, ele);
generatePortType(fac, ele);
// generateSOAP11Binding(fac, ele);
// if (!disableSOAP12) {
// generateSOAP12Binding(fac, ele);
// }
// if (!disableREST) {
// generateHTTPBinding(fac, ele);
// }
generateService(fac, ele, disableREST, disableSOAP12);
addPoliciesToDefinitionElement(policiesInDefinitions.values()
.iterator(), definition);
return ele;
}
private void generateMessages(OMFactory fac, OMElement defintions) {
HashSet faultMessageNames = new HashSet();
messagesMap = new HashMap();
Iterator operations = axisService.getOperations();
while (operations.hasNext()) {
AxisOperation axisOperation = (AxisOperation) operations.next();
if (axisOperation.isControlOperation()) {
continue;
}
String MEP = axisOperation.getMessageExchangePattern();
if (WSDL2Constants.MEP_URI_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisMessage inaxisMessage = axisOperation
.getMessage(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
if (inaxisMessage != null) {
writeMessage(inaxisMessage, fac, defintions);
generateHeaderMessages(inaxisMessage, fac, defintions);
}
}
if (WSDL2Constants.MEP_URI_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisMessage outAxisMessage = axisOperation
.getMessage(WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
if (outAxisMessage != null) {
writeMessage(outAxisMessage, fac, defintions);
generateHeaderMessages(outAxisMessage, fac, defintions);
}
}
// generate fault Messages
ArrayList faultyMessages = axisOperation.getFaultMessages();
if (faultyMessages != null) {
for (int i = 0; i < faultyMessages.size(); i++) {
AxisMessage axisMessage = (AxisMessage) faultyMessages
.get(i);
String name = axisMessage.getName();
if (faultMessageNames.add(name)) {
writeMessage(axisMessage, fac, defintions);
generateHeaderMessages(axisMessage, fac, defintions);
}
}
}
}
}
private void generateHeaderMessages(AxisMessage axismessage, OMFactory fac,
OMElement defintions) {
ArrayList extList = axismessage.getSoapHeaders();
for (int i = 0; i < extList.size(); i++) {
SOAPHeaderMessage header = (SOAPHeaderMessage) extList.get(i);
OMElement messageElement = fac.createOMElement(MESSAGE_LOCAL_NAME,
wsdl);
messageElement.addAttribute(ATTRIBUTE_NAME, header.getMessage()
.getLocalPart(), null);
defintions.addChild(messageElement);
OMElement messagePart = fac.createOMElement(PART_ATTRIBUTE_NAME,
wsdl);
messageElement.addChild(messagePart);
messagePart.addAttribute(ATTRIBUTE_NAME, header.part(), null);
if (header.getElement() == null) {
throw new RuntimeException(ELEMENT_ATTRIBUTE_NAME
+ " is null for " + header.getMessage());
}
messagePart.addAttribute(ELEMENT_ATTRIBUTE_NAME,
WSDLSerializationUtil.getPrefix(header.getElement()
.getNamespaceURI(), axisService.getNamespaceMap())
+ ":" + header.getElement().getLocalPart(), null);
}
}
private void writeMessage(AxisMessage axismessage, OMFactory fac,
OMElement defintions) {
if (messagesMap.get(axismessage.getName()) == null) {
messagesMap.put(axismessage.getName(), axismessage);
QName schemaElementName = axismessage.getElementQName();
OMElement messageElement = fac.createOMElement(MESSAGE_LOCAL_NAME,
wsdl);
messageElement.addAttribute(ATTRIBUTE_NAME, axismessage.getName(),
null);
defintions.addChild(messageElement);
if (schemaElementName != null) {
OMElement messagePart = fac.createOMElement(
PART_ATTRIBUTE_NAME, wsdl);
messageElement.addChild(messagePart);
if (axismessage.getMessagePartName() != null) {
messagePart.addAttribute(ATTRIBUTE_NAME, axismessage
.getMessagePartName(), null);
} else {
messagePart.addAttribute(ATTRIBUTE_NAME, axismessage
.getPartName(), null);
}
messagePart.addAttribute(ELEMENT_ATTRIBUTE_NAME,
WSDLSerializationUtil.getPrefix(schemaElementName
.getNamespaceURI(), axisService
.getNamespaceMap())
+ ":" + schemaElementName.getLocalPart(), null);
}
}
}
/**
* Builds the <portType> element in the passed WSDL definition. When
* this returns successfully, there will be a new child element under
* definitons for the portType.
*
* @param fac
* the active OMFactory
* @param defintions
* the WSDL <definitions> element
* @throws Exception
* if there's a problem
*/
private void generatePortType(OMFactory fac, OMElement defintions)
throws Exception {
OMElement portType = fac.createOMElement(PORT_TYPE_LOCAL_NAME, wsdl);
defintions.addChild(portType);
portType.addAttribute(ATTRIBUTE_NAME, serviceName + PORT_TYPE_SUFFIX,
null);
addPolicyAsExtAttribute(axisService, portType, fac);
for (Iterator operations = axisService.getOperations(); operations
.hasNext();) {
AxisOperation axisOperation = (AxisOperation) operations.next();
if (axisOperation.isControlOperation()
|| axisOperation.getName() == null) {
continue;
}
String operationName = axisOperation.getName().getLocalPart();
OMElement operation = fac.createOMElement(OPERATION_LOCAL_NAME,
wsdl);
WSDLSerializationUtil.addWSDLDocumentationElement(axisOperation,
operation, fac, wsdl);
portType.addChild(operation);
operation.addAttribute(ATTRIBUTE_NAME, operationName, null);
//addPolicyAsExtElement(axisOperation, operation);
addPolicyAsExtAttribute(axisOperation, operation, fac);
String MEP = axisOperation.getMessageExchangePattern();
if (WSDL2Constants.MEP_URI_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisMessage inaxisMessage = axisOperation
.getMessage(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
if (inaxisMessage != null) {
OMElement input = fac.createOMElement(IN_PUT_LOCAL_NAME,
wsdl);
WSDLSerializationUtil.addWSDLDocumentationElement(
inaxisMessage, input, fac, wsdl);
input.addAttribute(MESSAGE_LOCAL_NAME, tns.getPrefix()
+ ":" + inaxisMessage.getName(), null);
//addPolicyAsExtElement(inaxisMessage, input);
addPolicyAsExtAttribute(inaxisMessage, input, fac);
WSDLSerializationUtil.addWSAWActionAttribute(input,
axisOperation.getInputAction(), wsaw);
operation.addChild(input);
}
}
if (WSDL2Constants.MEP_URI_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisMessage outAxisMessage = axisOperation
.getMessage(WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
if (outAxisMessage != null) {
OMElement output = fac.createOMElement(OUT_PUT_LOCAL_NAME,
wsdl);
WSDLSerializationUtil.addWSDLDocumentationElement(
outAxisMessage, output, fac, wsdl);
output.addAttribute(MESSAGE_LOCAL_NAME, tns.getPrefix()
+ ":" + outAxisMessage.getName(), null);
//addPolicyAsExtElement(outAxisMessage, output);
addPolicyAsExtAttribute(outAxisMessage, output, fac);
WSDLSerializationUtil.addWSAWActionAttribute(output,
axisOperation.getOutputAction(), wsaw);
operation.addChild(output);
}
}
// generate fault Messages
ArrayList faultMessages = axisOperation.getFaultMessages();
if (faultMessages != null) {
for (int i = 0; i < faultMessages.size(); i++) {
AxisMessage faultyMessage = (AxisMessage) faultMessages
.get(i);
OMElement fault = fac.createOMElement(FAULT_LOCAL_NAME,
wsdl);
WSDLSerializationUtil.addWSDLDocumentationElement(
faultyMessage, fault, fac, wsdl);
fault.addAttribute(MESSAGE_LOCAL_NAME, tns.getPrefix()
+ ":" + faultyMessage.getName(), null);
fault.addAttribute(ATTRIBUTE_NAME, faultyMessage.getName(),
null);
WSDLSerializationUtil.addWSAWActionAttribute(fault,
axisOperation.getFaultAction(faultyMessage
.getName()), wsaw);
// TODO add policies for fault messages
operation.addChild(fault);
}
}
}
}
/**
* Generate the WSDL <service> element
*
* @param fac
* the active OMFactory
* @param defintions
* the WSDL <definitions> element under which to put the
* service
* @param disableREST
* if false, generate REST binding, if true, don't
* @param disableSOAP12
* if false, generate SOAP 1.2 binding, if true, don't
* @throws Exception
* if there's a problem
*/
public void generateService(OMFactory fac, OMElement defintions,
boolean disableREST, boolean disableSOAP12) throws Exception {
OMElement service = fac.createOMElement(SERVICE_LOCAL_NAME, wsdl);
defintions.addChild(service);
service.addAttribute(ATTRIBUTE_NAME, serviceName, null);
generateSoap11Port(fac, defintions, service);
if (!disableSOAP12) {
// generateSOAP12Ports(fac, service);
generateSoap12Port(fac, defintions, service);
}
addPolicyAsExtElement(PolicyInclude.SERVICE_POLICY, axisService
.getPolicyInclude(), service);
// addPolicyAsExtElement(PolicyInclude.AXIS_SERVICE_POLICY, axisService.
// getPolicyInclude(), service);
if (!disableREST) {
// generateHTTPPorts(fac, service);
generateHttpPort(fac, definition, service);
}
}
private void writeSoapHeaders(AxisMessage inaxisMessage, OMFactory fac,
OMElement input, OMNamespace soapNameSpace) throws Exception {
ArrayList extElementList;
extElementList = inaxisMessage.getSoapHeaders();
if (extElementList != null) {
Iterator elements = extElementList.iterator();
while (elements.hasNext()) {
SOAPHeaderMessage soapheader = (SOAPHeaderMessage) elements
.next();
addSOAPHeader(fac, input, soapheader, soapNameSpace);
}
}
}
private void addExtensionElement(OMFactory fac, OMElement element,
String name, String att1Name, String att1Value, String att2Name,
String att2Value, OMNamespace soapNameSpace) {
OMElement soapbinding = fac.createOMElement(name, soapNameSpace);
element.addChild(soapbinding);
soapbinding.addAttribute(att1Name, att1Value, null);
if (att2Name != null) {
soapbinding.addAttribute(att2Name, att2Value, null);
}
}
private void setDefinitionElement(OMElement defintion) {
this.definition = defintion;
}
private void addSOAPHeader(OMFactory fac, OMElement element,
SOAPHeaderMessage header, OMNamespace soapNameSpace) {
OMElement extElement = fac.createOMElement("header", soapNameSpace);
element.addChild(extElement);
String use = header.getUse();
if (use != null) {
extElement.addAttribute("use", use, null);
}
if (header.part() != null) {
extElement.addAttribute("part", header.part(), null);
}
if (header.getMessage() != null) {
extElement.addAttribute("message", WSDLSerializationUtil.getPrefix(
targetNamespace, axisService.getNamespaceMap())
+ ":" + header.getMessage().getLocalPart(), null);
}
}
private void addPolicyAsExtElement(int type, PolicyInclude policyInclude,
OMElement parentElement) throws Exception {
ArrayList elementList = policyInclude.getPolicyElements(type);
for (Iterator iterator = elementList.iterator(); iterator.hasNext();) {
Object policyElement = iterator.next();
if (policyElement instanceof Policy) {
OMElement child = PolicyUtil.getPolicyComponentAsOMElement(
(PolicyComponent) policyElement, serializer);
OMNode firstChildElem = parentElement.getFirstElement();
// if (firstChildElem == null) {
// parentElement.addChild(child);
// } else {
// firstChildElem.insertSiblingBefore(child);
// }
// there is a problem with the OM insertSiblingBefore element
// with
// drops the already exists elements.
// since there is no any techical problem of adding policy
// elements after other
// children temporaliy fix this as it is.
// one OM fix this issue we can revert this change.
parentElement.addChild(child);
} else if (policyElement instanceof PolicyReference) {
OMElement child = PolicyUtil
.getPolicyComponentAsOMElement((PolicyComponent) policyElement);
OMElement firstChildElem = parentElement.getFirstElement();
if (firstChildElem == null) {
parentElement.addChild(child);
} else {
firstChildElem.insertSiblingBefore(child);
}
PolicyRegistry reg = policyInclude.getPolicyRegistry();
String key = ((PolicyReference) policyElement).getURI();
if (key.startsWith("#")) {
key = key.substring(key.indexOf("#") + 1);
}
Policy p = reg.lookup(key);
if (p == null) {
throw new Exception("Policy not found for uri : " + key);
}
addPolicyToDefinitionElement(key, p);
}
}
}
private void addPoliciesToDefinitionElement(Iterator iterator,
OMElement definitionElement) throws Exception {
Policy policy;
OMElement policyElement;
OMNode firstChild;
ArrayList keys = new ArrayList();
for (; iterator.hasNext();) {
policy = (Policy) iterator.next();
String key = policy.getId();
if (keys.contains(key)) {
continue;
}
keys.add(key);
policyElement = PolicyUtil.getPolicyComponentAsOMElement(policy,
serializer);
firstChild = definition.getFirstOMChild();
if (firstChild != null) {
firstChild.insertSiblingBefore(policyElement);
} else {
definitionElement.addChild(policyElement);
}
}
}
private void addPolicyToDefinitionElement(String key, Policy policy) {
policiesInDefinitions.put(key, policy);
}
public String getStyle() {
return style;
}
public void setStyle(String style) {
this.style = style;
}
public String getUse() {
return use;
}
public void setUse(String use) {
this.use = use;
}
private void generateSoap11Port(OMFactory fac, OMElement definition,
OMElement service) throws Exception {
Iterator iterator = axisService.getEndpoints().values().iterator();
AxisEndpoint axisEndpoint = null;
AxisBinding axisBinding = null;
for (; iterator.hasNext();) {
axisEndpoint = (AxisEndpoint) iterator.next();
/*
* Some transports might not be active at runtime.
*/
if (!axisEndpoint.isActive()) {
continue;
}
axisBinding = axisEndpoint.getBinding();
String type = axisBinding.getType();
if (Java2WSDLConstants.TRANSPORT_URI.equals(type)
|| WSDL2Constants.URI_WSDL2_SOAP.equals(type)) {
String version = (String) axisBinding
.getProperty(WSDL2Constants.ATTR_WSOAP_VERSION);
if (SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(version)) {
OMElement port = fac.createOMElement(PORT, wsdl);
service.addChild(port);
port.addAttribute(ATTRIBUTE_NAME, axisEndpoint.getName(),
null);
QName qname = axisBinding.getName();
port.addAttribute(BINDING_LOCAL_NAME, tns.getPrefix() + ":"
+ qname.getLocalPart(), null);
String endpointURL = axisEndpoint.getEndpointURL();
WSDLSerializationUtil.addExtensionElement(fac, port,
SOAP_ADDRESS, LOCATION, (endpointURL == null) ? ""
: endpointURL, soap);
addPolicyAsExtElement(axisEndpoint, port);
generateSoap11Binding(fac, definition, axisEndpoint
.getBinding());
}
}
}
}
private void generateSoap12Port(OMFactory fac, OMElement definition,
OMElement service) throws Exception {
// /////////////////// FIXME //////////////////////////////////////////
Iterator iterator = axisService.getEndpoints().values().iterator();
AxisEndpoint axisEndpoint = null;
AxisBinding axisBinding = null;
for (; iterator.hasNext();) {
axisEndpoint = (AxisEndpoint) iterator.next();
/*
*
*/
if (!axisEndpoint.isActive()) {
continue;
}
axisBinding = axisEndpoint.getBinding();
String type = axisBinding.getType();
if (Java2WSDLConstants.TRANSPORT_URI.equals(type)
|| WSDL2Constants.URI_WSDL2_SOAP.equals(type)) {
String version = (String) axisBinding
.getProperty(WSDL2Constants.ATTR_WSOAP_VERSION);
if (SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(version)) {
OMElement port = fac.createOMElement(PORT, wsdl);
service.addChild(port);
port.addAttribute(ATTRIBUTE_NAME, axisEndpoint.getName(),
null);
QName qname = axisBinding.getName();
port.addAttribute(BINDING_LOCAL_NAME, tns.getPrefix() + ":"
+ qname.getLocalPart(), null);
String endpointURL = axisEndpoint.getEndpointURL();
WSDLSerializationUtil.addExtensionElement(fac, port,
SOAP_ADDRESS, LOCATION, (endpointURL == null) ? ""
: endpointURL, soap12);
addPolicyAsExtElement(axisEndpoint, port);
generateSoap12Binding(fac, definition, axisEndpoint
.getBinding());
}
}
}
}
private void generateHttpPort(OMFactory fac, OMElement definition,
OMElement service) throws Exception {
// /////////////////// FIXME //////////////////////////////////////////
Iterator iterator = axisService.getEndpoints().values().iterator();
AxisEndpoint axisEndpoint = null;
AxisBinding axisBinding = null;
for (; iterator.hasNext();) {
axisEndpoint = (AxisEndpoint) iterator.next();
/*
*
*/
if (!axisEndpoint.isActive()) {
continue;
}
axisBinding = axisEndpoint.getBinding();
String type = axisBinding.getType();
if (WSDL2Constants.URI_WSDL2_HTTP.equals(type)) {
OMElement port = fac.createOMElement(PORT, wsdl);
service.addChild(port);
port.addAttribute(ATTRIBUTE_NAME, axisEndpoint.getName(), null);
QName qname = axisBinding.getName();
port.addAttribute(BINDING_LOCAL_NAME, tns.getPrefix() + ":"
+ qname.getLocalPart(), null);
OMElement extElement = fac.createOMElement("address", http);
String endpointURL = axisEndpoint.getEndpointURL();
extElement.addAttribute("location", (endpointURL == null) ? ""
: endpointURL, null);
port.addChild(extElement);
addPolicyAsExtElement(axisEndpoint, port);
generateHttpBinding(fac, definition, axisEndpoint.getBinding());
}
}
}
private void generateSoap11Binding(OMFactory fac, OMElement defintions,
AxisBinding axisBinding) throws Exception {
if (isAlreadyAdded(axisBinding, defintions)) {
return;
}
OMElement binding = fac.createOMElement(BINDING_LOCAL_NAME, wsdl);
OMElement serviceElement = defintions.getFirstChildWithName(new QName(
wsdl.getNamespaceURI(), SERVICE_LOCAL_NAME));
serviceElement.insertSiblingBefore(binding);
QName qname = axisBinding.getName();
binding.addAttribute(ATTRIBUTE_NAME, qname.getLocalPart(), null);
binding.addAttribute("type", tns.getPrefix() + ":" + serviceName
+ PORT_TYPE_SUFFIX, null);
// Adding ext elements
addPolicyAsExtElement(axisBinding, binding);
addExtensionElement(fac, binding, BINDING_LOCAL_NAME, TRANSPORT,
TRANSPORT_URI, STYLE, style, soap);
// /////////////////////////////////////////////////////////////////////
// Add WS-Addressing UsingAddressing element if appropriate
// SHOULD be on the binding element per the specification
if (AddressingHelper
.getAddressingRequirementParemeterValue(axisService).equals(
AddressingConstants.ADDRESSING_OPTIONAL)) {
WSDLSerializationUtil.addExtensionElement(fac, binding,
AddressingConstants.USING_ADDRESSING,
DEFAULT_WSDL_NAMESPACE_PREFIX + ":required", "true", wsaw);
} else if (AddressingHelper.getAddressingRequirementParemeterValue(
axisService).equals(AddressingConstants.ADDRESSING_REQUIRED)) {
WSDLSerializationUtil.addExtensionElement(fac, binding,
AddressingConstants.USING_ADDRESSING,
DEFAULT_WSDL_NAMESPACE_PREFIX + ":required", "true", wsaw);
}
// //////////////////////////////////////////////////////////////////////
for (Iterator axisBindingOperations = axisBinding.getChildren(); axisBindingOperations
.hasNext();) {
AxisBindingOperation axisBindingOperation = (AxisBindingOperation) axisBindingOperations
.next();
AxisOperation axisOperation = axisBindingOperation
.getAxisOperation();
if (axisOperation.isControlOperation()
|| axisOperation.getName() == null) {
continue;
}
String opeartionName = axisOperation.getName().getLocalPart();
OMElement operation = fac.createOMElement(OPERATION_LOCAL_NAME,
wsdl);
binding.addChild(operation);
String soapAction = axisOperation.getSoapAction();
if (soapAction == null) {
soapAction = "";
}
addPolicyAsExtElement(axisBindingOperation, operation);
addExtensionElement(fac, operation, OPERATION_LOCAL_NAME,
SOAP_ACTION, soapAction, STYLE, style, soap);
// addPolicyAsExtElement(PolicyInclude.BINDING_OPERATION_POLICY,
// axisOperation.getPolicyInclude(), operation);
String MEP = axisOperation.getMessageExchangePattern();
if (WSDL2Constants.MEP_URI_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisBindingMessage axisBindingInMessage = (AxisBindingMessage) axisBindingOperation
.getChild(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
if (axisBindingInMessage != null) {
AxisMessage inaxisMessage = axisBindingInMessage
.getAxisMessage();
if (inaxisMessage != null) {
operation.addAttribute(ATTRIBUTE_NAME, opeartionName,
null);
OMElement input = fac.createOMElement(
IN_PUT_LOCAL_NAME, wsdl);
addPolicyAsExtElement(axisBindingInMessage, input);
addExtensionElement(fac, input, SOAP_BODY, SOAP_USE,
use, null, targetNamespace, soap);
// addPolicyAsExtElement(PolicyInclude.BINDING_INPUT_POLICY,
// inaxisMessage.getPolicyInclude(), input);
operation.addChild(input);
writeSoapHeaders(inaxisMessage, fac, input, soap12);
}
}
}
if (WSDL2Constants.MEP_URI_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisBindingMessage axisBindingOutMessage = (AxisBindingMessage) axisBindingOperation
.getChild(WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
if (axisBindingOutMessage != null) {
AxisMessage outAxisMessage = axisBindingOutMessage
.getAxisMessage();
if (outAxisMessage != null) {
OMElement output = fac.createOMElement(
OUT_PUT_LOCAL_NAME, wsdl);
addPolicyAsExtElement(axisBindingOutMessage, output);
addExtensionElement(fac, output, SOAP_BODY, SOAP_USE,
use, null, targetNamespace, soap);
// addPolicyAsExtElement(PolicyInclude.BINDING_OUTPUT_POLICY,
// outAxisMessage.getPolicyInclude(), output);
operation.addChild(output);
writeSoapHeaders(outAxisMessage, fac, output, soap12);
}
}
}
// generate fault Messages
ArrayList faultyMessages = axisBindingOperation.getFaults();
if (faultyMessages != null) {
for (int i = 0; i < faultyMessages.size(); i++) {
AxisBindingMessage bindingFaultyMessage = (AxisBindingMessage) faultyMessages
.get(i);
if (bindingFaultyMessage != null) {
AxisMessage faultyMessage = bindingFaultyMessage
.getAxisMessage();
OMElement fault = fac.createOMElement(FAULT_LOCAL_NAME,
wsdl);
addPolicyAsExtElement(bindingFaultyMessage, fault);
addExtensionElement(fac, fault, FAULT_LOCAL_NAME,
SOAP_USE, use, ATTRIBUTE_NAME, faultyMessage
.getName(), soap);
fault.addAttribute(ATTRIBUTE_NAME, faultyMessage
.getName(), null);
// add policies for fault messages
operation.addChild(fault);
writeSoapHeaders(faultyMessage, fac, fault, soap);
}
}
}
}
}
private void generateSoap12Binding(OMFactory fac, OMElement definitions,
AxisBinding axisBinding) throws Exception {
if (isAlreadyAdded(axisBinding, definitions)) {
return;
}
OMElement binding = fac.createOMElement(BINDING_LOCAL_NAME, wsdl);
OMElement serviceElement = definitions.getFirstChildWithName(new QName(
wsdl.getNamespaceURI(), SERVICE_LOCAL_NAME));
serviceElement.insertSiblingBefore(binding);
QName qname = axisBinding.getName();
binding.addAttribute(ATTRIBUTE_NAME, qname.getLocalPart(), null);
binding.addAttribute("type", tns.getPrefix() + ":" + serviceName
+ PORT_TYPE_SUFFIX, null);
// Adding ext elements
addPolicyAsExtElement(axisBinding, binding);
addExtensionElement(fac, binding, BINDING_LOCAL_NAME, TRANSPORT,
TRANSPORT_URI, STYLE, style, soap12);
// /////////////////////////////////////////////////////////////////////
// Add WS-Addressing UsingAddressing element if appropriate
// SHOULD be on the binding element per the specification
if (AddressingHelper
.getAddressingRequirementParemeterValue(axisService).equals(
AddressingConstants.ADDRESSING_OPTIONAL)) {
WSDLSerializationUtil.addExtensionElement(fac, binding,
AddressingConstants.USING_ADDRESSING,
DEFAULT_WSDL_NAMESPACE_PREFIX + ":required", "true", wsaw);
} else if (AddressingHelper.getAddressingRequirementParemeterValue(
axisService).equals(AddressingConstants.ADDRESSING_REQUIRED)) {
WSDLSerializationUtil.addExtensionElement(fac, binding,
AddressingConstants.USING_ADDRESSING,
DEFAULT_WSDL_NAMESPACE_PREFIX + ":required", "true", wsaw);
}
// //////////////////////////////////////////////////////////////////////
for (Iterator axisBindingOperations = axisBinding.getChildren(); axisBindingOperations
.hasNext();) {
AxisBindingOperation axisBindingOperation = (AxisBindingOperation) axisBindingOperations
.next();
AxisOperation axisOperation = axisBindingOperation
.getAxisOperation();
if (axisOperation.isControlOperation()
|| axisOperation.getName() == null) {
continue;
}
String opeartionName = axisOperation.getName().getLocalPart();
OMElement operation = fac.createOMElement(OPERATION_LOCAL_NAME,
wsdl);
binding.addChild(operation);
String soapAction = axisOperation.getSoapAction();
if (soapAction == null) {
soapAction = "";
}
addPolicyAsExtElement(axisBindingOperation, operation);
addExtensionElement(fac, operation, OPERATION_LOCAL_NAME,
SOAP_ACTION, soapAction, STYLE, style, soap12);
// addPolicyAsExtElement(PolicyInclude.BINDING_OPERATION_POLICY,
// axisOperation.getPolicyInclude(), operation);
String MEP = axisOperation.getMessageExchangePattern();
if (WSDL2Constants.MEP_URI_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisBindingMessage axisBindingInMessage = (AxisBindingMessage) axisBindingOperation
.getChild(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
if (axisBindingInMessage != null) {
AxisMessage inaxisMessage = axisBindingInMessage
.getAxisMessage();
if (inaxisMessage != null) {
operation.addAttribute(ATTRIBUTE_NAME, opeartionName,
null);
OMElement input = fac.createOMElement(
IN_PUT_LOCAL_NAME, wsdl);
addPolicyAsExtElement(axisBindingInMessage, input);
addExtensionElement(fac, input, SOAP_BODY, SOAP_USE,
use, null, targetNamespace, soap12);
operation.addChild(input);
writeSoapHeaders(inaxisMessage, fac, input, soap12);
}
}
}
if (WSDL2Constants.MEP_URI_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisBindingMessage axisBindingOutMessage = (AxisBindingMessage) axisBindingOperation
.getChild(WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
if (axisBindingOutMessage != null) {
AxisMessage outAxisMessage = axisBindingOutMessage
.getAxisMessage();
if (outAxisMessage != null) {
OMElement output = fac.createOMElement(
OUT_PUT_LOCAL_NAME, wsdl);
addPolicyAsExtElement(axisBindingOutMessage, output);
addExtensionElement(fac, output, SOAP_BODY, SOAP_USE,
use, null, targetNamespace, soap12);
// addPolicyAsExtElement(PolicyInclude.BINDING_OUTPUT_POLICY,
// outAxisMessage.getPolicyInclude(), output);
operation.addChild(output);
writeSoapHeaders(outAxisMessage, fac, output, soap12);
}
}
}
// generate fault Messages
ArrayList faultyMessages = axisBindingOperation.getFaults();
if (faultyMessages != null) {
for (int i = 0; i < faultyMessages.size(); i++) {
AxisBindingMessage bindingFaultyMessage = (AxisBindingMessage) faultyMessages
.get(i);
if (bindingFaultyMessage != null) {
AxisMessage faultyMessage = bindingFaultyMessage
.getAxisMessage();
OMElement fault = fac.createOMElement(FAULT_LOCAL_NAME,
wsdl);
addPolicyAsExtElement(bindingFaultyMessage, fault);
addExtensionElement(fac, fault, FAULT_LOCAL_NAME,
SOAP_USE, use, ATTRIBUTE_NAME, faultyMessage
.getName(), soap12);
fault.addAttribute(ATTRIBUTE_NAME, faultyMessage
.getName(), null);
// add policies for fault messages
operation.addChild(fault);
writeSoapHeaders(faultyMessage, fac, fault, soap12);
}
}
}
}
}
private void generateHttpBinding(OMFactory fac, OMElement definitions,
AxisBinding axisBinding) throws Exception {
if (isAlreadyAdded(axisBinding, definitions)) {
return;
}
OMElement binding = fac.createOMElement(BINDING_LOCAL_NAME, wsdl);
OMElement serviceElement = definitions.getFirstChildWithName(new QName(
wsdl.getNamespaceURI(), SERVICE_LOCAL_NAME));
serviceElement.insertSiblingBefore(binding);
QName qname = axisBinding.getName();
binding.addAttribute(ATTRIBUTE_NAME, qname.getLocalPart(), null);
binding.addAttribute("type", tns.getPrefix() + ":" + serviceName
+ PORT_TYPE_SUFFIX, null);
OMElement httpBinding = fac.createOMElement("binding", http);
binding.addChild(httpBinding);
httpBinding.addAttribute("verb", "POST", null);
for (Iterator axisBindingOperations = axisBinding.getChildren(); axisBindingOperations
.hasNext();) {
AxisBindingOperation axisBindingOperation = (AxisBindingOperation) axisBindingOperations
.next();
AxisOperation axisOperation = axisBindingOperation
.getAxisOperation();
if (axisOperation.isControlOperation()
|| axisOperation.getName() == null) {
continue;
}
String opeartionName = axisOperation.getName().getLocalPart();
OMElement operation = fac.createOMElement(OPERATION_LOCAL_NAME,
wsdl);
binding.addChild(operation);
OMElement httpOperation = fac.createOMElement("operation", http);
operation.addChild(httpOperation);
httpOperation.addAttribute("location", serviceName + "/"
+ axisOperation.getName().getLocalPart(), null);
String MEP = axisOperation.getMessageExchangePattern();
if (WSDL2Constants.MEP_URI_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_IN_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisBindingMessage axisBindingInMessage = (AxisBindingMessage) axisBindingOperation
.getChild(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
if (axisBindingInMessage != null) {
AxisMessage inaxisMessage = axisBindingInMessage
.getAxisMessage();
if (inaxisMessage != null) {
operation.addAttribute(ATTRIBUTE_NAME, opeartionName,
null);
OMElement input = fac.createOMElement(
IN_PUT_LOCAL_NAME, wsdl);
OMElement inputelement = fac.createOMElement("content",
mime);
input.addChild(inputelement);
inputelement.addAttribute("type", "text/xml", null);
inputelement.addAttribute("part", axisOperation
.getName().getLocalPart(), null);
operation.addChild(input);
}
}
}
if (WSDL2Constants.MEP_URI_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_OUT_OPTIONAL_IN.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OPTIONAL_OUT.equals(MEP)
|| WSDL2Constants.MEP_URI_ROBUST_OUT_ONLY.equals(MEP)
|| WSDL2Constants.MEP_URI_IN_OUT.equals(MEP)) {
AxisBindingMessage axisBindingOutMessage = (AxisBindingMessage) axisBindingOperation
.getChild(WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
if (axisBindingOutMessage != null) {
AxisMessage outAxisMessage = axisBindingOutMessage
.getAxisMessage();
if (outAxisMessage != null) {
OMElement output = fac.createOMElement(
OUT_PUT_LOCAL_NAME, wsdl);
OMElement outElement = fac.createOMElement("content",
mime);
outElement.addChild(outElement);
outElement.addAttribute("type", "text/xml", null);
outElement.addAttribute("part", axisOperation.getName()
.getLocalPart(), null);
output.addChild(outElement);
operation.addChild(output);
}
}
}
}
}
private void addPolicyAsExtElement(AxisDescription axisDescription,
OMElement wsdlElement) throws Exception {
PolicySubject policySubject = axisDescription.getPolicySubject();
Collection attachPolicyComponents = policySubject
.getAttachedPolicyComponents();
for (Iterator iterator = attachPolicyComponents.iterator(); iterator
.hasNext();) {
Object policyElement = iterator.next();
if (policyElement instanceof Policy) {
PolicyReference policyReference = PolicyUtil
.createPolicyReference((Policy) policyElement);
OMElement policyRefElement = PolicyUtil
.getPolicyComponentAsOMElement(
(PolicyComponent) policyReference, serializer);
OMNode firstChildElem = wsdlElement.getFirstElement();
if (firstChildElem == null) {
wsdlElement.addChild(policyRefElement);
} else {
firstChildElem.insertSiblingBefore(policyRefElement);
}
String key = ((PolicyReference) policyReference).getURI();
if (key.startsWith("#")) {
key = key.substring(key.indexOf("#") + 1);
}
addPolicyToDefinitionElement(key, (Policy) policyElement);
} else if (policyElement instanceof PolicyReference) {
OMElement child = PolicyUtil.getPolicyComponentAsOMElement(
(PolicyComponent) policyElement, serializer);
OMElement firstChildElem = wsdlElement.getFirstElement();
if (firstChildElem == null) {
wsdlElement.addChild(child);
} else {
firstChildElem.insertSiblingBefore(child);
}
String key = ((PolicyReference) policyElement).getURI();
if (key.startsWith("#")) {
key = key.substring(key.indexOf("#") + 1);
}
PolicyLocator locator = new PolicyLocator(axisService);
Policy p = locator.lookup(key);
if (p == null) {
throw new Exception("Policy not found for uri : " + key);
}
addPolicyToDefinitionElement(key, p);
}
}
}
private void addPolicyAsExtAttribute(AxisDescription axisDescription,
OMElement element, OMFactory factory) throws Exception {
PolicySubject policySubject = axisDescription.getPolicySubject();
ArrayList policyURIs = new ArrayList();
for (Iterator iterator = policySubject.getAttachedPolicyComponents()
.iterator(); iterator.hasNext();) {
Object policyElement = iterator.next();
String key;
if (policyElement instanceof Policy) {
Policy p = (Policy) policyElement;
if (p.getId() != null) {
key = "#" + p.getId();
} else if (p.getName() != null) {
key = p.getName();
} else {
throw new RuntimeException(
"Can't add the Policy as an extensibility attribute since it doesn't have a id or a name attribute");
}
policyURIs.add(key);
addPolicyToDefinitionElement(key, p);
} else {
String uri = ((PolicyReference) policyElement).getURI();
PolicyLocator locator = new PolicyLocator(axisService);
if (uri.startsWith("#")) {
key = uri.substring(uri.indexOf('#') + 1);
} else {
key = uri;
}
Policy p = locator.lookup(key);
if (p == null) {
throw new RuntimeException("Cannot resolve " + uri
+ " to a Policy");
}
addPolicyToDefinitionElement(key, p);
}
}
if (!policyURIs.isEmpty()) {
String value = null;
/*
* We need to create a String that is like 'uri1 uri2 .." to set as
* the value of the wsp:PolicyURIs attribute.
*/
for (Iterator iterator = policyURIs.iterator(); iterator.hasNext();) {
String uri = (String) iterator.next();
value = (value == null) ? uri : value + " " + uri;
}
OMNamespace ns = factory.createOMNamespace(
org.apache.neethi.Constants.URI_POLICY_NS,
org.apache.neethi.Constants.ATTR_WSP);
OMAttribute URIs = factory.createOMAttribute("PolicyURIs", ns,
value);
element.addAttribute(URIs);
}
}
private boolean isAlreadyAdded(AxisBinding axisBinding,
OMElement definitionElement) {
QName bindingName = axisBinding.getName();
QName name = new QName("name");
for (Iterator iterator = definitionElement
.getChildrenWithName(new QName(wsdl.getNamespaceURI(),
BINDING_LOCAL_NAME)); iterator.hasNext();) {
OMElement element = (OMElement) iterator.next();
String value = element.getAttributeValue(name);
if (bindingName.getLocalPart().equals(value)) {
return true;
}
}
return false;
}
}