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