WebAppModifier.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.w3c.dom.Element;

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.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.WebServicesDDModifier;
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 WebAppModifier extends ArchiveModifier {

    private WebApp web;

    public WebAppModifier(WebApp webapp) {
        super(webapp);
        web = webapp;
    }

    public Archive modify() throws GenBaseException {

        getLogger().log(BasicLevel.INFO, "Processing WebApp " + web.getName());

        GeneratorFactory gf = GeneratorFactory.getInstance();
        Document jwebapp = web.getJonasWebAppDoc();
        Document webapp = web.getWebAppDoc();
        Document webservices = web.getWebservicesDoc();

        List refs = web.getServiceRefDescs();

        for (Iterator i = refs.iterator(); i.hasNext();) {
            ServiceRefDesc ref = (ServiceRefDesc) i.next();

            // create dd modifier
            Element base = null;
            if (jwebapp != null) {
                base = jwebapp.getDocumentElement();
            }
            WsClientDDModifier ddm = new WsClientDDModifier(ref.getServiceRefName(), jwebapp, base);

            // launch generation
            Generator g = gf.newGenerator(ref, ddm, web);
            g.generate();
            g.compile();

            // add files in web archive
            g.addFiles(web);
        }

        // create dd modifier
        WsEndpointDDModifier ddm = new WsEndpointDDModifier(webapp);
        WebServicesDDModifier wsddm = null;
        if (webservices != null) {
            wsddm = new WebServicesDDModifier(webservices);
        }

        List sds = web.getServiceDescs();

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

            // launch generation
            Generator g = gf.newGenerator(sd, ddm, wsddm, web);
            g.generate();
            g.compile();

            // add files in web archive
            g.addFiles(web);

        }

        return save(gf.getConfiguration(), "webapps" + File.separator
                + web.getName());
    }
}

---