VcPortComponent.java

package org.objectweb.jonas_ws.wsgen.generator.axis;

import java.lang.reflect.Method;
import java.util.Iterator;
import java.util.List;
import java.util.Vector;

import javax.wsdl.BindingFault;
import javax.wsdl.BindingOperation;
import javax.wsdl.Definition;
import javax.wsdl.Port;
import javax.wsdl.Service;
import javax.wsdl.extensions.ExtensibilityElement;
import javax.wsdl.extensions.soap.SOAPBinding;
import javax.wsdl.extensions.soap.SOAPBody;
import javax.wsdl.extensions.soap.SOAPFault;
import javax.wsdl.extensions.soap.SOAPOperation;
import javax.xml.namespace.QName;

import org.objectweb.jonas_ejb.deployment.api.SessionStatelessDesc;

import org.objectweb.jonas_lib.deployment.api.HandlerDesc;

import org.objectweb.jonas_ws.deployment.api.PortComponentDesc;
import org.objectweb.jonas_ws.deployment.api.SSBPortComponentDesc;

public class VcPortComponent {

    private String name;

    private VcBean bean = null;

    private static final String SEP = ",";

    private String methods = null;

    private Vector handlers = new Vector();

    private String jaxRpcClassName;

    private String wsdlFilename;

    private static final String DOCUMENT_STYLE = "document";

    private String style = null;

    private String use = null;

    private String namespace = null;

    public static final String NS_URI_SOAP = "http://schemas.xmlsoap.org/wsdl/soap/";

    private static final QName SOAP_BINDING_QNAME = new QName(NS_URI_SOAP, "binding");

    private static final QName SOAP_OPERATION_QNAME = new QName(NS_URI_SOAP, "operation");

    private static final QName SOAP_BODY_QNAME = new QName(NS_URI_SOAP, "body");

    private static final QName SOAP_FAULT_QNAME = new QName(NS_URI_SOAP, "fault");

    public VcPortComponent(PortComponentDesc pcd, String wsdl) {

        // set name
        name = pcd.getServiceName();

        wsdlFilename = wsdl;

        // set bean or jaxRpcClassName
        if (pcd.hasBeanImpl()) {
            // set bean
            SessionStatelessDesc ssb = ((SSBPortComponentDesc) pcd).getSessionStatelessDesc();
            bean = new VcBean(ssb);
        } else {
            jaxRpcClassName = pcd.getSIBClassname();
        }
        // set methods
        Method[] m = pcd.getServiceEndpointInterface().getMethods();
        for (int i = 0; i < m.length; i++) {
            if (methods != null) {
                methods += SEP;
            } else {
                methods = "";
            }
            methods += m[i].getName();
        }

        // set style/use/namespace
        Definition def = pcd.getServiceDesc().getWSDL().getDefinition();
        boolean portFound = false;
        for (Iterator i = def.getServices().keySet().iterator(); i.hasNext() && !portFound;) {
            Service s = def.getService((QName) i.next());
            Port p = s.getPort(pcd.getQName().getLocalPart());
            if (p != null) {
                // we found the linked port
                portFound = true;

                // search style attribute
                style = getStyle(p);

                // search use attribute inside port
                use = getUse(p);
            }
        }
        namespace = def.getTargetNamespace();

        // set handlers
        List hs = pcd.getHandlers();
        for (Iterator itH = hs.iterator(); itH.hasNext();) {
            HandlerDesc hd = (HandlerDesc) itH.next();
            handlers.add(new VcHandler(hd));
        }

    }

    private static String getUse(Port p) {
        String use = null;

        // use is declared inside
        // - wsdl:definition/wsdl:binding/wsdl:operation/wsdl:input/soap:body@use
        // - wsdl:definition/wsdl:binding/wsdl:operation/wsdl:output/soap:body@use
        // - wsdl:definition/wsdl:binding/wsdl:operation/wsdl:fault/soap:fault@use
        // take the first one.
        List ops = p.getBinding().getBindingOperations();
        for (Iterator j = ops.iterator(); j.hasNext() && (use == null);) {
            BindingOperation bop = (BindingOperation) j.next();

            // search input
            List inputExt = bop.getBindingInput().getExtensibilityElements();
            for (Iterator k = inputExt.iterator(); k.hasNext() && (use == null);) {
                ExtensibilityElement ext = (ExtensibilityElement) k.next();
                if (ext.getElementType().equals(SOAP_BODY_QNAME)) {
                    SOAPBody sb = (SOAPBody) ext;
                    use = sb.getUse();
                }
            }

            // search output
            List outputExt = bop.getBindingOutput().getExtensibilityElements();
            for (Iterator k = outputExt.iterator(); k.hasNext() && (use == null);) {
                ExtensibilityElement ext = (ExtensibilityElement) k.next();
                if (ext.getElementType().equals(SOAP_BODY_QNAME)) {
                    SOAPBody sb = (SOAPBody) ext;
                    use = sb.getUse();
                }
            }

            // search faults
            for (Iterator k = bop.getBindingFaults().keySet().iterator(); k.hasNext() && (use == null);) {
                BindingFault bf = bop.getBindingFault((String) k.next());

                // search soap:fault
                List faultExt = bf.getExtensibilityElements();
                for (Iterator i = faultExt.iterator(); i.hasNext() && (use == null);) {
                    ExtensibilityElement ext = (ExtensibilityElement) i.next();
                    if (ext.getElementType().equals(SOAP_FAULT_QNAME)) {
                        SOAPFault sf = (SOAPFault) ext;
                        use = sf.getUse();
                    }
                }
            }
        }
        // use cannot be null as it is required !

        return use;
    }

    private static String getStyle(Port p) {
        String style = null;

        // we need to explore soap:operation and get the first that set a style attribute
        List bindingOps = p.getBinding().getBindingOperations();
        for (Iterator i = bindingOps.iterator(); i.hasNext() && (style == null);) {

            BindingOperation bop = (BindingOperation) i.next();
            // browse soap:operation
            List extElements = bop.getExtensibilityElements();
            for (Iterator j = extElements.iterator(); j.hasNext();) {
                ExtensibilityElement ext = (ExtensibilityElement) j.next();
                if (ext.getElementType().equals(SOAP_OPERATION_QNAME)) {

                    // Got a soap:operation element
                    SOAPOperation sop = (SOAPOperation) ext;
                    style = sop.getStyle();
                }
            }
        }

        // browse extensibility elements of port's binding
        List soapElements = p.getBinding().getExtensibilityElements();
        for (Iterator i = soapElements.iterator(); i.hasNext() && (style == null);) {

            ExtensibilityElement ext = (ExtensibilityElement) i.next();
            if (ext.getElementType().equals(SOAP_BINDING_QNAME)) {

                // Got a soap:binding element
                SOAPBinding soapb = (SOAPBinding) ext;
                style = soapb.getStyle();
            }

        }

        if (style == null) {
            style = DOCUMENT_STYLE;
        }

        return style;
    }

    public VcPortComponent(String name, List hrs) {

        // set name
        this.name = name;

        // set handlers
        for (Iterator itH = hrs.iterator(); itH.hasNext();) {
            HandlerDesc hr = (HandlerDesc) itH.next();
            handlers.add(new VcHandler(hr));
        }

    }

    public String getName() {
        return name;
    }

    public VcBean getBean() {
        return bean;
    }

    public String getWSDLFilename() {
        return wsdlFilename;
    }

    public String getMethods() {
        return methods;
    }

    public Vector getHandlers() {
        return handlers;
    }

    public String getJaxRpcClassName() {
        return jaxRpcClassName;
    }

    public String getStyle() {
        return style;
    }

    public String getUse() {
        return use;
    }

    public String getNamespace() {
        return namespace;
    }
}

---