EjbJarModifier.java

package org.objectweb.jonas_ws.wsgen.modifier;

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

import org.w3c.dom.Document;

import org.objectweb.jonas_lib.genbase.GenBaseException;
import org.objectweb.jonas_lib.genbase.archive.Application;
import org.objectweb.jonas_lib.genbase.archive.Archive;
import org.objectweb.jonas_lib.genbase.archive.DummyApplication;
import org.objectweb.jonas_lib.genbase.archive.DummyWebApp;
import org.objectweb.jonas_lib.genbase.archive.Ejb;
import org.objectweb.jonas_lib.genbase.archive.EjbJar;
import org.objectweb.jonas_lib.genbase.archive.WebApp;
import org.objectweb.jonas_lib.genbase.modifier.ArchiveModifier;

import org.objectweb.jonas_ws.deployment.api.ServiceDesc;
import org.objectweb.jonas_ws.deployment.api.ServiceRefDesc;
import org.objectweb.jonas_ws.wsgen.ddmodifier.WsClientDDModifier;
import org.objectweb.jonas_ws.wsgen.ddmodifier.WsEndpointDDModifier;
import org.objectweb.jonas_ws.wsgen.generator.Generator;
import org.objectweb.jonas_ws.wsgen.generator.GeneratorFactory;

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

public class EjbJarModifier extends ArchiveModifier {

    private EjbJar ejbjar = null;

    public EjbJarModifier(EjbJar ejbjar) {
        super(ejbjar);
        this.ejbjar = ejbjar;
    }

    public Archive modify() throws GenBaseException {

        getLogger().log(BasicLevel.INFO, "Processing EjbJar " + ejbjar.getName());

        GeneratorFactory gf = GeneratorFactory.getInstance();
        Document jejbjar = ejbjar.getJonasEjbJarDoc();

        // Webservices endpoint
        List sds = ejbjar.getServiceDescs();

        if (sds.size() != 0) {
            if (ejbjar.getApplication() == null) {
                // we have webservices outside of an application

                String ejbName = ejbjar.getRootFile().getName();
                String earName = ejbName.substring(0, ejbName.length() - ".jar".length()) + ".ear";
                Application application = new DummyApplication(earName);
                ejbjar.setApplication(application);
                application.addEjbJar(ejbjar);
                ApplicationModifier am = new ApplicationModifier(application);

                return am.modify();

            } else {
                // we have webservices inside of an application

                String warName = null;
                String war = ejbjar.getWarName();
                if (war != null) {
                    warName = war;
                } else {
                    String ejbName = ejbjar.getRootFile().getName();
                    String contextName = ejbName.substring(0, ejbName.length() - ".jar".length());
                    warName = contextName + ".war";
                }

                Application application = ejbjar.getApplication();
                WebApp web = new DummyWebApp(application, warName);

                for (Iterator i = sds.iterator(); i.hasNext();) {
                    ServiceDesc sd = (ServiceDesc) i.next();

                    // create dd modifier
                    WsEndpointDDModifier ddm = new WsEndpointDDModifier(web.getWebAppDoc());

                    // launch generation
                    Generator g = gf.newGenerator(sd, ddm, null, ejbjar);
                    g.generate();
                    g.compile();
                    // add files in web archive
                    g.addFiles(web);

                }
                // save webapp
                application.addWebApp(new WebApp(save(gf.getConfiguration(),
                        "webapps" + File.separator + web.getName(), web)), ejbjar.getContextRoot());
            }
        }

        // add client files
        List ejbs = ejbjar.getEjbs();
        for (Iterator i = ejbs.iterator(); i.hasNext();) {
            Ejb ejb = (Ejb) i.next();
            List refs = ejb.getServiceRefDescs();
            for (Iterator j = refs.iterator(); j.hasNext();) {
                ServiceRefDesc ref = (ServiceRefDesc) j.next();

                // create dd modifier
                WsClientDDModifier ddm = new WsClientDDModifier(ref.getServiceRefName(), jejbjar, ejb.getJonasBeanElement());

                // launch generation
                Generator g = gf.newGenerator(ref, ddm, ejbjar);
                g.generate();
                g.compile();
                // add files in web archive
                g.addFiles(ejbjar);
            }
        }

        return save(gf.getConfiguration(), "ejbjars" + File.separator + ejbjar.getRootFile().getName());

    }
}

---