AxisWsEndpointGenerator.java

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

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

import org.apache.velocity.VelocityContext;

import org.objectweb.jonas_lib.genbase.GenBaseException;
import org.objectweb.jonas_lib.genbase.archive.Archive;
import org.objectweb.jonas_lib.genbase.archive.WebApp;
import org.objectweb.jonas_lib.genbase.generator.Config;

import org.objectweb.jonas_ws.deployment.api.JaxRpcPortComponentDesc;
import org.objectweb.jonas_ws.deployment.api.PortComponentDesc;
import org.objectweb.jonas_ws.deployment.api.ServiceDesc;
import org.objectweb.jonas_ws.wsgen.WsGenException;
import org.objectweb.jonas_ws.wsgen.ddmodifier.WebServicesDDModifier;
import org.objectweb.jonas_ws.wsgen.ddmodifier.WsEndpointDDModifier;
import org.objectweb.jonas_ws.wsgen.generator.WsEndpointGenerator;

import org.objectweb.jonas.common.Log;

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

public class AxisWsEndpointGenerator extends WsEndpointGenerator {

    private static JVelocity jvelocity = null;

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

    private static final String AXIS_SERVLET_CLASS = "org.objectweb.jonas.ws.axis.JAxisServlet";

    private static final String SERVER_CONFIG = "axis.serverConfigFile";

    private static final String WSDD_SUFFIX = ".wsdd";

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

    private static int count = 0;

    private File generatedServerConfig;

    public AxisWsEndpointGenerator(Config config, ServiceDesc serviceDesc, WsEndpointDDModifier ddm,
            WebServicesDDModifier wsddm, Archive arch) throws GenBaseException, WsGenException {
        super(config, serviceDesc, ddm, wsddm, arch);

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

    public void generate() throws WsGenException {
        String sName = getService().getName();

        // construct VelocityContext
        VelocityContext vc = VContextFactory.getContext(getService());

        // Generate file
        String filename = WSDD_PREFIX + (count++) + WSDD_SUFFIX;
        generatedServerConfig = new File(getSources(), filename);
        jvelocity.generate(generatedServerConfig, vc);

        // remove existing servlets
        for (Iterator i = getService().getPortComponents().iterator(); i.hasNext();) {
            Object obj = i.next();
            if (obj instanceof JaxRpcPortComponentDesc) {
                JaxRpcPortComponentDesc jax = (JaxRpcPortComponentDesc) obj;

                if (logger.isLoggable(BasicLevel.DEBUG)) {
                    logger.log(BasicLevel.DEBUG, "Removing servlet '" + jax.getSibLink() + "'");
                }

                // remove servlet mapping
                getModifier().removeServletMapping(jax.getSibLink());
                // remove old servlet
                getModifier().removeServlet(jax.getSibLink());
                // change sib-link
                getWsModifier().changeServletLink(sName, jax.getSibLink(), sName);
            }
        }

        // and use our own
        getModifier().addServlet(sName, AXIS_SERVLET_CLASS);
        getModifier().addServletParam(sName, SERVER_CONFIG, filename);

        // setup servlet-mappings
        boolean requireDefaultMapping = false;
        List usedServletMappings = new Vector();
        for (Iterator i = getService().getPortComponents().iterator(); i.hasNext();) {
            PortComponentDesc pcd = (PortComponentDesc) i.next();
            String mapping = pcd.getMapping();
            if (mapping != null) {
                // port has specified an endpoint URI
                // use it for servlet-mapping
                if (!usedServletMappings.contains(mapping)) {
                    usedServletMappings.add(mapping);

                    if (logger.isLoggable(BasicLevel.DEBUG)) {
                        logger.log(BasicLevel.DEBUG, "Adding servlet-mapping for '" + sName + "' -> '" + mapping + "'");
                    }

                    getModifier().addServletMapping(sName, mapping);
                }
            } else {
                // no specified endpoint uri for the port
                requireDefaultMapping = true;
            }
        }
        if (requireDefaultMapping) {
            // try to set a default mapping
            String defaultEndpointURI = getService().getEndpointURI();
            if (defaultEndpointURI == null) {
                // default behavior
                getModifier().addServletMapping(sName, "/" + sName + "/*");
            } else {
                // use retieved value
                getModifier().addServletMapping(sName, defaultEndpointURI);
            }
        }

    }

    public void addFiles(Archive archive) throws WsGenException {
        // archive must be a WebApp
        if (!(archive instanceof WebApp)) {
            String err = getI18n().getMessage("AxisWsEndpointGenerator.addFiles.illegal", archive.getRootFile());
            throw new IllegalArgumentException(err);
        }

        WebApp web = (WebApp) archive;
        web.addFileIn("WEB-INF/", generatedServerConfig);

    }
}

---