AxisWsClientGenerator.java

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

import java.io.File;

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

import org.apache.velocity.VelocityContext;

import org.objectweb.jonas_lib.I18n;
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.MappingFile;
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.axis.wsdl2java.JOnASWSDL2Java;

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

public class AxisWsClientGenerator extends WsClientGenerator {

    private static JVelocity jvelocity = null;

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

    private static final String WSDD_SUFFIX = ".wsdd";

    private static final String WSDD_PREFIX = "deploy-client-";

    private static int count = 0;

    private File generated = null;

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

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

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

        // init velocity
        if (jvelocity == null) {
            jvelocity = new JVelocity("deploy_client.vm");
        }
    }

    public void generate() throws WsGenException {

        // the source generation is possible only when a
        // WSDL Definition is provided
        if ((getRef().getWsdlFileName() != null)
                && !getRef().getServiceInterface().getName().equals("javax.xml.rpc.Service")) {

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


                JOnASWSDL2Java jWsdl2Java = new JOnASWSDL2Java();
                jWsdl2Java.run(this);
                getLogger().log(BasicLevel.INFO, "Web Services Classes successfully generated by Axis.");
            } catch (Exception e) {
                String err = getI18n().getMessage("AxisWsClientGenerator.generate.WSDL2Java");
                e.printStackTrace(System.err);
                throw new WsGenException(err, e);
            }
        }

        // Client-config.wsdd ??
        // only if handlers are specified in ServiceRef
        // but if noConfig set to true, assume that
        // user configure himself the client. (expert)
        // Could we create configuration ?
        if (hasClientConfigFile(getRef())) {
            // create deploy.wsdd
            // use velocity template
            // build a unique file name
            String filename = WSDD_PREFIX + (count++) + WSDD_SUFFIX;

            VelocityContext vctx = VContextFactory.getContext(getRef());

            getLogger().log(BasicLevel.DEBUG, "Creating '" + filename + "'");
            generated = new File(getSources(), filename);

            jvelocity.generate(generated, vctx);

        }
        // End !noConfig
    }

    private boolean hasClientConfigFile(ServiceRefDesc ref) {
        // no configuration asked
        if (getConfig().isNoConfig()) {
            return false;
        }

        MappingFile mf = ref.getMappingFile();

        if (mf != null) {
            // mapping file defined
            if (mf.getXmlTypeMappings().hasNext()) {
                // we have 1 mapping at least
                return true;
            } else {
                // no mapping
                // return true if ref has Handlers
                return (ref.getHandlerRefs().size() != 0);
            }
        } else {
            // no mapping file
            // return true if ref has Handlers
            return (ref.getHandlerRefs().size() != 0);
        }
    }

    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;
    }
}

---