JOnASWSDL2Java.java

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

import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import javax.wsdl.WSDLException;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;

import org.w3c.dom.Document;
import org.xml.sax.SAXException;

import org.apache.axis.wsdl.WSDL2Java;
import org.apache.axis.wsdl.gen.Parser;

import org.objectweb.jonas_ws.wsgen.WsGenException;
import org.objectweb.jonas_ws.wsgen.generator.axis.AxisWsClientGenerator;

public class JOnASWSDL2Java extends WSDL2Java {

    private JOnASEmitter jEmitter;

    private String wsdlURL = null;

    private Document wsdlDoc = null;

    public JOnASWSDL2Java() {
        super();
        // just cast it once
        jEmitter = (JOnASEmitter) getParser();
    }

    protected Parser createParser() {
        return new JOnASEmitter();
    } // createParser

    private void setup(AxisWsClientGenerator wsc) throws WsGenException {
        setupEmitter(wsc);
        setupWSDL(wsc);
    }

    private void setupWSDL(AxisWsClientGenerator wsc) throws WsGenException {
        try {
            if (wsc.getArchive().isPacked()) {
                // URL Loading
                String jarpath = wsc.getArchive().getRootFile().getCanonicalFile().toURL().toExternalForm();
                wsdlURL = "jar:" + jarpath + "!/" + wsc.getRef().getWsdlFileName();
            } else {
                // File Loading
                wsdlURL = new File(wsc.getArchive().getRootFile(), wsc.getRef().getWsdlFileName()).toURL()
                        .toExternalForm();
            }

            DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
            factory.setNamespaceAware(true);
            DocumentBuilder builder = factory.newDocumentBuilder();
            wsdlDoc = builder.parse(wsdlURL);
        } catch (ParserConfigurationException pce) {
            throw new WsGenException("", pce);
        } catch (IOException ioe) {
            throw new WsGenException("", ioe);
        } catch (SAXException se) {
            throw new WsGenException("", se);
        }
    }

    private void setupEmitter(AxisWsClientGenerator wsc) {
        jEmitter.setDebug(wsc.getConfig().isDebug());
        jEmitter.setVerbose(wsc.getConfig().isVerbose());
        jEmitter.setNamespaceMap(convert2HashMap(wsc.getRef().getMappingFile().getMappings()));
        jEmitter.setOutputDir(wsc.getSources().getPath());
    }

    private HashMap convert2HashMap(Map m) {
        HashMap ns2pkg = new HashMap();
        for (Iterator i = m.keySet().iterator(); i.hasNext();) {
            String ns = (String) i.next();
            ns2pkg.put(ns, m.get(ns));
        }
        return ns2pkg;
    }

    public void run(AxisWsClientGenerator wsc) throws WsGenException, IOException, SAXException, WSDLException, ParserConfigurationException {
        setup(wsc);
        parser.run(wsdlURL, wsdlDoc);
    } // run

}

---