EWSWsClientGenerator.java

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

import java.io.File;
import java.util.Iterator;
import java.util.List;

import org.w3c.dom.Document;
import org.w3c.dom.Element;

import org.apache.axis.wsdl.toJava.GeneratedFileInfo;
import org.apache.geronimo.ews.ws4j2ee.context.webservices.client.ServiceReferanceImpl;
import org.apache.geronimo.ews.ws4j2ee.context.webservices.client.interfaces.ServiceReferanceContext;

import org.objectweb.jonas_lib.I18n;
import org.objectweb.jonas_lib.deployment.api.HandlerDesc;
import org.objectweb.jonas_lib.genbase.GenBaseException;
import org.objectweb.jonas_lib.genbase.archive.Archive;
import org.objectweb.jonas_lib.genbase.archive.EjbJar;
import org.objectweb.jonas_lib.genbase.archive.J2EEArchive;
import org.objectweb.jonas_lib.genbase.archive.WebApp;
import org.objectweb.jonas_lib.genbase.generator.Config;
import org.objectweb.jonas_lib.genbase.utils.XMLUtils;
import org.objectweb.jonas_lib.loader.AbsModuleClassLoader;

import org.objectweb.jonas_ws.deployment.api.ServiceRefDesc;
import org.objectweb.jonas_ws.wsgen.WsGenException;
import org.objectweb.jonas_ws.wsgen.ddmodifier.WsClientDDModifier;
import org.objectweb.jonas_ws.wsgen.generator.WsClientGenerator;
import org.objectweb.jonas_ws.wsgen.generator.ews.wsdltoj2ee.JOnASJ2EEWebServicesContext;
import org.objectweb.jonas_ws.wsgen.generator.ews.wsdltoj2ee.emitter.FullEmitter;
import org.objectweb.jonas_ws.wsgen.generator.ews.wsdltoj2ee.factory.JOnASClientGeneratorFactory;
import org.objectweb.jonas_ws.wsgen.generator.ews.wsdltoj2ee.wscf.JOnASWSCFHandler;

import org.objectweb.jonas.common.Log;

import org.objectweb.util.monolog.api.BasicLevel;
import org.objectweb.util.monolog.api.Logger;

public class EWSWsClientGenerator extends WsClientGenerator {

    private static Logger logger = Log.getLogger(Log.JONAS_WSGEN_EWS_PREFIX);

    private static final String CLIENT_CONFIG = "axis.clientConfigFile";

    private File generated = null;

    private static final String WEB_PREFIX = "WEB-INF/";

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

    public EWSWsClientGenerator(Config config, ServiceRefDesc srd, WsClientDDModifier ddm, Archive archive)
            throws GenBaseException, WsGenException {
        super(config, srd, ddm, archive);
    }

    public void generate() throws WsGenException {

        // the source generation is possible only when a
        // WSDL Definition is provided
        // we need :
        // - a: wsdl-file
        // - b: jaxrpc-mapping-file
        // - c: service-interface != javax.xml.rpc.Service
        // - d: some port-component-ref
        // = ((a && b) && (c || d))

        boolean hasWSDL = getRef().getWsdlFileName() != null;

        if (hasWSDL) {
            // classpath creation
            J2EEArchive j2eeArchive = (J2EEArchive) getArchive();
            AbsModuleClassLoader cl = (AbsModuleClassLoader) j2eeArchive.getModuleClassloader();
            getConfig().setClasspath(getConfig().getClasspath() + cl.getClasspath());

            try {
                // WSDL + mapping file
                FullEmitter wsEmitter = new FullEmitter();
                wsEmitter.setMappingFileInputStream(getRef().getMappingFileURL().openStream());
                JOnASClientGeneratorFactory factory = new JOnASClientGeneratorFactory();
                factory.setEmitter(wsEmitter);

                // Classloader to use to check if classes are present before
                // generating java classes
                wsEmitter.setClassLoader(cl);

                wsEmitter.setFactory(factory);
                //wsEmitter.setNowrap(true);

                wsEmitter.setJOnASWsContext(prepareJ2EEWebServicesContext(getRef()));

                wsEmitter.setOutputDir(this.getSources().getCanonicalPath());
                wsEmitter.setAllowInvalidURL(true);

                wsEmitter.run(getRef().getLocalWSDLURL().toExternalForm());

                generated = new File(findClientConfigFile(wsEmitter.getGeneratedFileInfo()));

                getLogger().log(BasicLevel.INFO, "Web Services Classes successfully generated by EWS.");
                getLogger().log(BasicLevel.INFO,
                        "Webservice client WSDD file '" + generated + "' sucessfully generated by EWS.");
            } catch (Exception e) {
                String err = getI18n().getMessage("EWSWsClientGenerator.generate.WSDL2Java", e.getMessage());
                logger.log(BasicLevel.ERROR, err, e);
                throw new WsGenException(err, e);
            }

        }

    }

    private static String findClientConfigFile(GeneratedFileInfo generatedFileInfo) {
        List generated = generatedFileInfo.getFileNames();
        boolean found = false;
        String filename = null;
        for (Iterator i = generated.iterator(); i.hasNext() && !found;) {
            String entry = (String) i.next();
            // */deploy-client-*.wsdd
            if (entry.matches(".*/deploy-client-\\d+\\.wsdd$")) {
                found = true;
                filename = entry;
            }
        }
        return filename;
    }

    private JOnASJ2EEWebServicesContext prepareJ2EEWebServicesContext(ServiceRefDesc ref) {

        JOnASJ2EEWebServicesContext wsCtx = new JOnASJ2EEWebServicesContext();
        ServiceReferanceContext sRefCtx = new ServiceReferanceImpl();
        sRefCtx.setServcieQName(ref.getServiceQName());

        for (Iterator i = ref.getHandlerRefs().iterator(); i.hasNext();) {
            HandlerDesc handler = (HandlerDesc) i.next();
            sRefCtx.addHandler(new JOnASWSCFHandler(handler));
        }
        wsCtx.addServiceReferanceContext(sRefCtx);
        return wsCtx;
    }

    public void addFiles(Archive archive) throws WsGenException {
        if (archive instanceof WebApp) {
            archive.addDirectoryIn("WEB-INF/classes/", getClasses());

            if (generated != null) {
                archive.addFileIn("WEB-INF/", generated);

                // ensure the optionnal descriptor exists
                if (!getModifier().hasJonasServiceRef()) {
                    if (!getArchive().getContainedFiles().contains("WEB-INF/jonas-web.xml")) {
                        // jonas-web.xml doesn't exists
                        createEmptyJonasWeb((J2EEArchive) archive);
                    }
                    Element jsr = getModifier().createJonasServiceRef(getRef().getServiceRefName());
                    // update
                    getModifier().setElement(jsr);

                }

                // add init param
                getModifier().addJonasInitParam(CLIENT_CONFIG, WEB_PREFIX + generated.getName());
            }
        } else if (archive instanceof EjbJar) {
            archive.addDirectory(getClasses());

            if (generated != null) {
                archive.addFileIn("META-INF/", generated);

                // ensure the optionnal descriptor exists
                if (!getModifier().hasJonasServiceRef()) {
                    Element jsr = getModifier().createJonasServiceRef(getRef().getServiceRefName());
                    // update
                    getModifier().setElement(jsr);
                }

                // add init param
                getModifier().addJonasInitParam(CLIENT_CONFIG, "META-INF/" + generated.getName());
            }
        } else {
            archive.addDirectory(getClasses());

            if (generated != null) {
                archive.addFileIn("META-INF/", generated);

                // ensure the optionnal descriptor exists
                if (!getModifier().hasJonasServiceRef()) {
                    if (!getArchive().getContainedFiles().contains("META-INF/jonas-client.xml")) {
                        // jonas-client.xml doesn't exists
                        createEmptyJonasClient((J2EEArchive) archive);
                    }
                    Element jsr = getModifier().createJonasServiceRef(getRef().getServiceRefName());
                    // update
                    getModifier().setElement(jsr);

                }

                // add init param
                getModifier().addJonasInitParam(CLIENT_CONFIG, "META-INF/" + generated.getName());
            }
        }
    }

    private void createEmptyJonasWeb(J2EEArchive archive) {
        Document doc = XMLUtils.newJonasWeb();
        archive.getDescriptors().put("WEB-INF/jonas-web.xml", doc);
        getModifier().setDocument(doc);
    }

    private void createEmptyJonasClient(J2EEArchive archive) {
        Document doc = XMLUtils.newJonasClient();
        archive.getDescriptors().put("META-INF/jonas-client.xml", doc);
        getModifier().setDocument(doc);
    }

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

---