WSDLFile.java

package org.objectweb.jonas_ws.deployment.api;

import java.io.StringWriter;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import javax.wsdl.Definition;
import javax.wsdl.Message;
import javax.wsdl.Part;
import javax.wsdl.Port;
import javax.wsdl.Service;
import javax.wsdl.WSDLException;
import javax.wsdl.extensions.soap.SOAPAddress;
import javax.wsdl.extensions.soap.SOAPBinding;
import javax.wsdl.factory.WSDLFactory;
import javax.wsdl.xml.WSDLReader;
import javax.wsdl.xml.WSDLWriter;
import javax.xml.namespace.QName;

import org.objectweb.jonas_lib.I18n;

public class WSDLFile {


    private String name;

    private List wsdlPorts = null;

    private Definition def = null;

    private static final boolean VERBOSE = false;

    private static I18n i18n = I18n.getInstance(WSDLFile.class);

    public WSDLFile(ClassLoader cl, String name) throws WSDeploymentDescException {
            this(cl.getResource(name), name);
    }

    public WSDLFile(URL url, String name) throws WSDeploymentDescException {
        this.name = name;

        try {
            WSDLFactory f = WSDLFactory.newInstance();
            WSDLReader r = f.newWSDLReader();

            r.setFeature("javax.wsdl.verbose", VERBOSE); //$NON-NLS-1$
            r.setFeature("javax.wsdl.importDocuments", true); //$NON-NLS-1$

            if (url == null) {
                throw new WSDeploymentDescException(getI18n().getMessage("WSDLFile.notFound", name)); //$NON-NLS-1$
            }

            def = r.readWSDL(url.toExternalForm(), url.toExternalForm());
        } catch (WSDLException e) {
            throw new WSDeploymentDescException(getI18n().getMessage("WSDLFile.WSDLParsingError", name), e); //$NON-NLS-1$
        }

        wsdlPorts = new Vector();
        fillWsdlPorts();
    }

    public boolean hasPort(QName portQName) {
        return hasPort(portQName.getLocalPart());
    }

    public boolean hasPort(String portName) {
        return getPort(portName) != null;
    }

    public boolean hasService(QName srvQName) {
        return def.getService(srvQName) != null;
    }

    public boolean hasSOAPHeader(QName shQName) {
        Map msgs = def.getMessages();

        for (Iterator m = msgs.values().iterator(); m.hasNext();) {
            Message msg = (Message) m.next();

            if (msg.getQName().getNamespaceURI() == shQName.getNamespaceURI()) {
                Part p = msg.getPart(shQName.getLocalPart());

                if (p != null) {
                    return true;
                }
            }
        }

        return false;
    }

    public boolean hasPortsIncludedIn(List portList) {
        return portList.containsAll(wsdlPorts);
    }

    public boolean hasSOAPBinding(QName portQName) {
        boolean isSoapBinding = false;
        Port port = getPort(portQName.getLocalPart());

        // get Port
        if (port != null) {
            List ee = port.getBinding().getExtensibilityElements();
            Iterator eeIt = ee.iterator();

            while (eeIt.hasNext() && !isSoapBinding) {
                // verify that it uses a soap binding
                Object elem = eeIt.next();

                if (elem != null) {
                    isSoapBinding = elem instanceof SOAPBinding;
                }
            }
        }

        return isSoapBinding;
    }

    public Definition getDefinition() {
        return def;
    }

    public int getNbServices() {
        return def.getServices().size();
    }

    public QName getServiceQname() {
        Map srvs = def.getServices();
        Iterator svcIt = srvs.values().iterator();
        QName res = null;

        if (svcIt.hasNext()) {
            Service svc = (Service) svcIt.next();
            res = svc.getQName();
        }

        return res;
    }

    public URL getLocation(QName portQName) throws WSDeploymentDescException {
        Port port = getPort(portQName.getLocalPart());

        // get portQName port
        if (port != null) {
            List ee = port.getExtensibilityElements();
            Iterator eeIt = ee.iterator();

            while (eeIt.hasNext()) {
                // find the soap address element
                Object elem = eeIt.next();

                if ((elem != null) && elem instanceof SOAPAddress) {
                    try {
                        return new URL(((SOAPAddress) elem).getLocationURI());
                    } catch (MalformedURLException e) {
                        throw new WSDeploymentDescException(getI18n().getMessage("WSDLFile.MalformedPortLocation", portQName)); //$NON-NLS-1$
                    }
                }
            }
        }

        return null;
    }

    public void setLocation(QName portQName, URL loc) {
        Port port = getPort(portQName.getLocalPart());

        if (port != null) {
            List ee = port.getExtensibilityElements();
            Iterator eeIt = ee.iterator();

            while (eeIt.hasNext()) {
                // find the soap address element
                Object elem = eeIt.next();

                if ((elem != null) && elem instanceof SOAPAddress) {
                    ((SOAPAddress) elem).setLocationURI(loc.toString());
                }
            }
        }
    }

    private void fillWsdlPorts() {
        Map svcs = def.getServices();
        Iterator svcsIt = svcs.values().iterator();

        while (svcsIt.hasNext()) {
            Service svc = (Service) svcsIt.next();

            if (svc != null) {
                for (Iterator j = svc.getPorts().values().iterator(); j.hasNext();) {
                    Port p = (Port) j.next();
                    wsdlPorts.add(new QName(def.getTargetNamespace(), p.getName()));
                }
            }
        }
    }

    private Port getPort(String portName) {
        Map svcs = def.getServices();
        Iterator svcsIt = svcs.values().iterator();

        while (svcsIt.hasNext()) {
            Service svc = (Service) svcsIt.next();

            if (svc != null) {
                Port port = svc.getPort(portName);

                // get the port identified by portQname
                return port;
            }
        }

        return null;
    }

    public String getName() {
        return name;
    }

    public String toString() {
        StringBuffer sb = new StringBuffer();

        sb.append("\n" + getClass().getName()); //$NON-NLS-1$
        sb.append("\ngetName()=" + getName()); //$NON-NLS-1$

        StringWriter sw = new StringWriter();
        // Write Definition
        try {
            WSDLFactory factory = WSDLFactory.newInstance();
            WSDLWriter writer = factory.newWSDLWriter();

            writer.writeWSDL(def, sw);

        } catch (WSDLException e) {
            sb.append(getI18n().getMessage("WSDLFile.writeDefError")); //$NON-NLS-1$
        }

        sb.append(sw.getBuffer().toString());

        return sb.toString();
    }

    public boolean equals(Object other) {
        if (other == null) {
            return false;
        }

        if (!(other instanceof WSDLFile)) {
            return false;
        }

        WSDLFile ref = (WSDLFile) other;
        Definition odef = ref.getDefinition();

        if (def.getServices().size() != odef.getServices().size()) {
            return false;
        }

        if (def.getPortTypes().size() != odef.getPortTypes().size()) {
            return false;
        }

        if (def.getMessages().size() != odef.getMessages().size()) {
            return false;
        }

        if (def.getBindings().size() != odef.getBindings().size()) {
            return false;
        }

        // After all theses tests, the 2 objects are equals in value
        return true;
    }

    protected static I18n getI18n() {
        return i18n;
    }
}

---