WebApp.java

package org.objectweb.jonas_lib.genbase.archive;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;
import java.util.Vector;
import java.util.jar.JarFile;

import javax.xml.parsers.ParserConfigurationException;

import org.w3c.dom.Document;
import org.xml.sax.SAXException;

import org.objectweb.jonas_lib.deployment.api.DeploymentDescException;
import org.objectweb.jonas_lib.deployment.api.EjbRefDesc;
import org.objectweb.jonas_lib.genbase.GenBaseException;
import org.objectweb.jonas_lib.genbase.utils.FileUtils;
import org.objectweb.jonas_lib.genbase.utils.XMLUtils;
import org.objectweb.jonas_lib.loader.WebappClassLoader;

import org.objectweb.jonas_web.deployment.api.WebContainerDeploymentDesc;
import org.objectweb.jonas_web.deployment.lib.WebDeploymentDescManager;

import org.objectweb.jonas_ws.deployment.api.ServiceRefDesc;
import org.objectweb.jonas_ws.deployment.api.WSDeploymentDesc;
import org.objectweb.jonas_ws.deployment.lib.WSDeploymentDescManager;

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

public class WebApp extends J2EEArchive implements EjbRefModule, WsClient, WsEndpoint {

    private Application app = null;

    private String webFilename;

    private WebContainerDeploymentDesc webDD;

    private WSDeploymentDesc wsDD = null;

    private List sRefs;

    private Document webApp = null;

    private Document jonasWebApp = null;

    private Document webservices = null;

    private Map descriptors;

    private List ejbRefs;

    public WebApp(Archive archive) throws GenBaseException {
        super(archive);
        webFilename = archive.getName();
        getLogger().log(BasicLevel.DEBUG, "Wrapping '" + archive.getName() + "' in WebApp");
        init();
    }

    public WebApp(Archive archive, Application app) throws GenBaseException {
        super(archive);
        webFilename = archive.getName();
        setApplication(app);
        getLogger().log(BasicLevel.DEBUG, "Wrapping '" + archive.getName() + "' in WebApp");
        init();
    }

    private void init() throws GenBaseException {
        // load Deployment Descs
        loadDescriptors();
    }

    private void loadDescriptors() throws GenBaseException {
        try {
            webApp = XMLUtils.newDocument(getWebInputStream(), "WEB-INF/web.xml", isDTDsAllowed());

            // jonas-web.xml (optionnal)
            InputStream is = getJonasWebInputStream();

            if (is != null) {
                jonasWebApp = XMLUtils.newDocument(is, "WEB-INF/jonas-web.xml", isDTDsAllowed());
            }

            // webservices.xml (optionnal)
            InputStream isWS = getWebservicesInputStream();

            if (isWS != null) {
                webservices = XMLUtils.newDocument(isWS, "WEB-INF/webservices.xml", isDTDsAllowed());
            }
        } catch (SAXException saxe) {
            String err = getI18n().getMessage("WebApp.loadDescriptors.parseError");
            throw new GenBaseException(err, saxe);
        } catch (ParserConfigurationException pce) {
            String err = getI18n().getMessage("WebApp.loadDescriptors.prepare");
            throw new GenBaseException(err, pce);
        } catch (IOException ioe) {
            String err = getI18n().getMessage("WebApp.loadDescriptors.parseError");
            throw new GenBaseException(err, ioe);
        }

        descriptors = new Hashtable();
        descriptors.put("WEB-INF/web.xml", webApp);

        if (jonasWebApp != null) {
            descriptors.put("WEB-INF/jonas-web.xml", jonasWebApp);
        }

        if (webservices != null) {
            descriptors.put("WEB-INF/webservices.xml", webservices);
        }
    }

    public String getName() {
        return webFilename;
    }

    public void setApplication(Application app) {
        this.app = app;
    }

    public Application getApplication() {
        return app;
    }

    public List getServiceRefDescs() {
        return sRefs;
    }

    public List getServiceDescs() {
        if (wsDD != null) {
            return wsDD.getServiceDescs();
        } else {
            return new Vector();
        }
    }

    public void addClasses(File classes) {
        addDirectoryIn("WEB-INF/classes", classes);
    }

    public Document getWebAppDoc() {
        return webApp;
    }

    public Document getJonasWebAppDoc() {
        return jonasWebApp;
    }

    public Document getWebservicesDoc() {
        return webservices;
    }

    private InputStream getWebInputStream() throws IOException {
        InputStream is = null;

        if (isPacked()) {
            is = getInputStream("WEB-INF/web.xml");
        } else {
            is = getInputStream("WEB-INF" + File.separator + "web.xml");
        }

        return is;
    }

    private InputStream getJonasWebInputStream() throws IOException {
        InputStream is = null;

        if (isPacked()) {
            is = getInputStream("WEB-INF/jonas-web.xml");
        } else {
            is = getInputStream("WEB-INF" + File.separator + "jonas-web.xml");
        }

        return is;
    }

    private InputStream getWebservicesInputStream() throws IOException {
        InputStream is = null;

        if (isPacked()) {
            is = getInputStream("WEB-INF/webservices.xml");
        } else {
            is = getInputStream("WEB-INF" + File.separator + "webservices.xml");
        }

        return is;
    }

    public Map getDescriptors() {
        return descriptors;
    }

    public boolean omit(String name) {
        return (name.equals("WEB-INF/web.xml") || name.equals("WEB-INF\\web.xml")
                || name.equals("WEB-INF/jonas-web.xml") || name.equals("WEB-INF\\jonas-web.xml")
                || name.equals("WEB-INF/webservices.xml") || name.equals("WEB-INF\\webservices.xml"));
    }

    public void initialize() throws GenBaseException {
        File webappUnpackDir = null;
        try {

            if (getArchive().isPacked()) {
                JarFile jf = new JarFile(getRootFile());
                webappUnpackDir = FileUtils.unpack(jf);
                jf.close();
                setArchive(new FileArchive(webappUnpackDir));
            }

            if (app == null) {
                // simple webapp case
                setModuleClassloader(new WebappClassLoader(webappUnpackDir.toURL(), Thread.currentThread()
                        .getContextClassLoader()));
            } else {
                // embedded webapp case
                setModuleClassloader(new WebappClassLoader(webappUnpackDir.toURL(), app.getEJBClassLoader()));
            }
        } catch (IOException ioe) {
            String err = getI18n().getMessage("WebApp.init.loader", getArchive().getRootFile());
            throw new GenBaseException(err, ioe);
        }

        try {
            webDD = WebDeploymentDescManager.getDeploymentDesc(webappUnpackDir.getAbsolutePath(), getModuleClassloader());
        } catch (DeploymentDescException dde) {
            throw new GenBaseException(dde);
        }

        try {
            wsDD = WSDeploymentDescManager.getDeploymentDesc(webappUnpackDir.getAbsolutePath(), getModuleClassloader());
        } catch (DeploymentDescException dde) {
            throw new GenBaseException(dde);
        }

        // we want a List of service-ref
        sRefs = new Vector();

        ServiceRefDesc[] refs = webDD.getServiceRefDesc();

        for (int i = 0; i < refs.length; i++) {
            sRefs.add(refs[i]);
        }

        // List of ejb-refs
        ejbRefs = new Vector();
        EjbRefDesc[] refDesc = webDD.getEjbRefDesc();

        for (int i = 0; i < refDesc.length; i++) {
            ejbRefs.add(refDesc[i]);
        }

    }

    public String getContextRoot() {
        return this.wsDD.getContextRoot();
    }

    public List getEjbRefDescs() {
        return ejbRefs;
    }
}

---