/*
* Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code 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 General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package com.sun.xml.internal.ws.wsdl.writer;
import com.sun.xml.internal.ws.api.wsdl.writer.WSDLGeneratorExtension;
import com.sun.xml.internal.ws.api.wsdl.writer.WSDLGenExtnContext;
import com.sun.xml.internal.ws.api.model.JavaMethod;
import com.sun.xml.internal.ws.api.model.CheckedException;
import com.sun.xml.internal.txw2.TypedXmlWriter;
import static com.sun.xml.internal.ws.addressing.W3CAddressingMetadataConstants.*;
import com.sun.xml.internal.ws.model.JavaMethodImpl;
import com.sun.xml.internal.ws.model.CheckedExceptionImpl;
import com.sun.xml.internal.ws.addressing.WsaActionUtil;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.logging.Logger;
/**
* This extension class generates wsam:Action values for input, output and faults in the generated wsdl.
*
* @author Rama Pulavarthi
*/
public class W3CAddressingMetadataWSDLGeneratorExtension extends
WSDLGeneratorExtension {
@Override
public void start(WSDLGenExtnContext ctxt) {
TypedXmlWriter root = ctxt.getRoot();
root._namespace(WSAM_NAMESPACE_NAME, WSAM_PREFIX_NAME);
}
@Override
public void addOperationInputExtension(TypedXmlWriter input,
JavaMethod method) {
input._attribute(WSAM_ACTION_QNAME, getInputAction(method));
}
@Override
public void addOperationOutputExtension(TypedXmlWriter output,
JavaMethod method) {
output._attribute(WSAM_ACTION_QNAME, getOutputAction(method));
}
@Override
public void addOperationFaultExtension(TypedXmlWriter fault,
JavaMethod method, CheckedException ce) {
fault._attribute(WSAM_ACTION_QNAME, getFaultAction(method, ce));
}
private static final String getInputAction(JavaMethod method) {
String inputaction = ((JavaMethodImpl)method).getInputAction();
if (inputaction.equals("")) {
// Calculate default action
inputaction = getDefaultInputAction(method);
}
return inputaction;
}
protected static final String getDefaultInputAction(JavaMethod method) {
String tns = method.getOwner().getTargetNamespace();
String delim = getDelimiter(tns);
if (tns.endsWith(delim))
tns = tns.substring(0, tns.length() - 1);
//this assumes that fromjava case there won't be input name.
// if there is input name in future, then here name=inputName
//else use operation name as follows.
String name = (method.getMEP().isOneWay()) ?
method.getOperationName() : method.getOperationName() + "Request";
return new StringBuilder(tns).append(delim).append(
method.getOwner().getPortTypeName().getLocalPart()).append(
delim).append(name).toString();
}
private static final String getOutputAction(JavaMethod method) {
String outputaction = ((JavaMethodImpl)method).getOutputAction();
if(outputaction.equals(""))
outputaction = getDefaultOutputAction(method);
return outputaction;
}
protected static final String getDefaultOutputAction(JavaMethod method) {
String tns = method.getOwner().getTargetNamespace();
String delim = getDelimiter(tns);
if (tns.endsWith(delim))
tns = tns.substring(0, tns.length() - 1);
//this assumes that fromjava case there won't be output name.
// if there is input name in future, then here name=outputName
//else use operation name as follows.
String name = method.getOperationName() + "Response";
return new StringBuilder(tns).append(delim).append(
method.getOwner().getPortTypeName().getLocalPart()).append(
delim).append(name).toString();
}
private static final String getDelimiter(String tns) {
String delim = "/";
// TODO: is this the correct way to find the separator ?
try {
URI uri = new URI(tns);
if ((uri.getScheme() != null) && uri.getScheme().equalsIgnoreCase("urn"))
delim = ":";
} catch (URISyntaxException e) {
LOGGER.warning("TargetNamespace of WebService is not a valid URI");
}
return delim;
}
private static final String getFaultAction(JavaMethod method,
CheckedException ce) {
String faultaction = ((CheckedExceptionImpl)ce).getFaultAction();
if (faultaction.equals("")) {
faultaction = getDefaultFaultAction(method,ce);
}
return faultaction;
}
protected static final String getDefaultFaultAction(JavaMethod method, CheckedException ce) {
return WsaActionUtil.getDefaultFaultAction(method,ce);
}
private static final Logger LOGGER =
Logger.getLogger(W3CAddressingMetadataWSDLGeneratorExtension.class.getName());
}