EWSWsEndpointGenerator.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.Element;
import org.w3c.dom.NodeList;

import org.apache.axis.constants.Scope;
import org.apache.axis.wsdl.toJava.GeneratedFileInfo;
import org.apache.geronimo.ews.jaxrpcmapping.J2eeGeneratorFactory;
import org.apache.geronimo.ews.ws4j2ee.context.J2EEWebServiceContext;

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.J2EEArchive;
import org.objectweb.jonas_lib.genbase.archive.WebApp;
import org.objectweb.jonas_lib.genbase.generator.Config;
import org.objectweb.jonas_lib.loader.AbsModuleClassLoader;

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_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.JOnASEndpointGeneratorFactory;

import org.objectweb.jonas.common.Log;

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

public class EWSWsEndpointGenerator extends WsEndpointGenerator {

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

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

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

    private File generatedServerConfig;

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

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

    public void generate() throws WsGenException {

        // 1. Generation of wsdd server file
        String filename = null;
        try {
            FullEmitter emitter = new FullEmitter();

            emitter.setWscontext(prepareJ2EEWebServicesContext(getService()));
            emitter.setMappingFileInputStream(getService().getMappingFileURL().openStream());
            emitter.setOutputDir(this.getSources().getCanonicalPath());
            J2eeGeneratorFactory factory = new JOnASEndpointGeneratorFactory();
            factory.setEmitter(emitter);
            emitter.setFactory(factory);

            // force request scope
            emitter.setScope(Scope.REQUEST);
            
            emitter.setServerSide(true);
            emitter.run(getService().getLocalWSDLURL().toExternalForm());

            generatedServerConfig = new File(findServerConfigFile(emitter.getGeneratedFileInfo()));
            filename = generatedServerConfig.getName();
            getLogger().log(BasicLevel.INFO,
                    "Webservice endpoint WSDD file '" + filename + "' sucessfully generated by EWS.");
        } catch (Exception e) {
            String err = i18n.getMessage("EWSWsEndpointGenerator.generate.WSDL2Java", getService().getName(), e
                    .getMessage());
            logger.log(BasicLevel.ERROR, err, e);
            throw new WsGenException(err, e);
        }
        
        // 2. classpath creation
        J2EEArchive j2eeArchive = (J2EEArchive) getArchive();
        AbsModuleClassLoader cl = (AbsModuleClassLoader) j2eeArchive.getModuleClassloader();
        getConfig().setClasspath(getConfig().getClasspath() + cl.getClasspath());

        // 3. Remove existing servlets
        //  * remove the servlet used by the port (store security informations)
        //  * remove the associated servlet-mapping (store old url-pattern value)
        for (Iterator i = getService().getPortComponents().iterator(); i.hasNext();) {
            PortComponentDesc desc = (PortComponentDesc) i.next();

            String urlPatternOld = null;
            NodeList securityRoleRefs = null;

            // port-component linked to a servlet :
            if (desc.hasJaxRpcImpl()) {
                JaxRpcPortComponentDesc jax = (JaxRpcPortComponentDesc) desc;

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

                // remove servlet mapping
                urlPatternOld = getModifier().removeServletMapping(jax.getSibLink());

                // remove old servlet but pick up its security role refs
                //getModifier().removeServlet(jax.getSibLink());
                securityRoleRefs = getModifier().removeServletWithSecurity(jax.getSibLink());
            }

            // in all case :
            // create a servlet with all needed configuration
            // and use our own
            getModifier().addServlet(desc.getSibLink(), AXIS_SERVLET_CLASS);
            getModifier().addServletParam(desc.getSibLink(), SERVER_CONFIG, filename);

            String mapping = "/" + desc.getName() + "/*";
            getModifier().addServletMapping(desc.getSibLink(), mapping);

            if (desc.hasJaxRpcImpl()) {
                // add the first element in the servletToSecurityRoleRefs map
                // (currently consider that only one servlet with security
                // constraints was removed !! - if there were more, keep the
                // constraints of the last removed)
                if (securityRoleRefs != null) {
                    getModifier().addServletSecurityRoleRefs(desc.getSibLink(), securityRoleRefs);
                }
                // update url-pattern
                getModifier().updateSecurityConstraint(urlPatternOld, mapping);
            }
        }
    }

    private static String findServerConfigFile(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-server-*.wsdd
            if (entry.matches(".*/deploy-server-\\d+\\.wsdd$")) {
                found = true;
                filename = entry;
            }
        }
        return filename;
    }

    private J2EEWebServiceContext prepareJ2EEWebServicesContext(ServiceDesc service) {
        JOnASJ2EEWebServicesContext ctx = new JOnASJ2EEWebServicesContext();
        ctx.setServiceDesc(service);
        return ctx;
    }

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

        web.addDirectoryIn("WEB-INF/classes/", getClasses());
    }
}

---