EjbJar.java

package org.objectweb.jonas_lib.genbase.archive;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import javax.xml.parsers.ParserConfigurationException;

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

import org.objectweb.jonas_ejb.deployment.api.BeanDesc;
import org.objectweb.jonas_ejb.deployment.api.DeploymentDesc;
import org.objectweb.jonas_ejb.deployment.lib.EjbDeploymentDescManager;

import org.objectweb.jonas_lib.deployment.api.DeploymentDescException;
import org.objectweb.jonas_lib.genbase.GenBaseException;
import org.objectweb.jonas_lib.genbase.utils.XMLUtils;
import org.objectweb.jonas_lib.loader.EjbJarClassLoader;

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 EjbJar extends J2EEArchive implements WsEndpoint {

    private Application app;

    private WSDeploymentDesc wsDD = null;

    private DeploymentDesc ejbDD = null;

    private List ejbs;

    private Map descriptors;

    private Document jEjbJar;

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

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

    private void init() throws GenBaseException {
        loadDescriptors();
    }


    public void initialize() throws GenBaseException {

        try {
            if (app == null) {
                // simple ejb case
                setModuleClassloader(new EjbJarClassLoader(new URL[] {getArchive()
                        .getRootFile().toURL()}, Thread.currentThread()
                        .getContextClassLoader()));
            } else {
                // embedded ejb case
                setModuleClassloader(app.getEJBClassLoader());
            }
        } catch (IOException ioe) {
            String err = getI18n().getMessage("EjbJar.init.loader", getArchive()
                    .getRootFile());
            throw new GenBaseException(err, ioe);
        }

        try {
            ejbDD = EjbDeploymentDescManager.getDeploymentDesc(getRootFile()
                    .getAbsolutePath(), getModuleClassloader());
        } catch (DeploymentDescException dde) {
            throw new GenBaseException(dde);
        }

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

        // create inner EJBs
        ejbs = new Vector();

        BeanDesc[] bd = ejbDD.getBeanDesc();

        for (int i = 0; i < bd.length; i++) {
            ejbs.add(new Ejb(bd[i], XMLUtils.getBeanElement(jEjbJar
                    .getDocumentElement(), bd[i].getEjbName())));
        }

    }


    private void loadDescriptors() throws GenBaseException {
        try {
            jEjbJar = XMLUtils.newDocument(getJonasEjbJarInputStream(), "META-INF/jonas-ejb-jar.xml", isDTDsAllowed());
        } catch (SAXException saxe) {
            String err = getI18n().getMessage("EjbJar.loadDescriptors.parseError");
            throw new GenBaseException(err, saxe);
        } catch (ParserConfigurationException pce) {
            String err = getI18n().getMessage("EjbJar.loadDescriptors.prepare");
            throw new GenBaseException(err, pce);
        } catch (IOException ioe) {
            String err = getI18n().getMessage("EjbJar.loadDescriptors.parseError");
            throw new GenBaseException(err, ioe);
        }

        descriptors = new Hashtable();
        descriptors.put("META-INF/jonas-ejb-jar.xml", jEjbJar);
    }

    public List getEjbs() {
        return ejbs;
    }

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

    public void addClasses(File classes) {
        addDirectory(classes);
    }

    public String getWarName() {
       if (wsDD != null) {
           return wsDD.getWarFile();
       } else {
           return null;
       }
    }

    public String getContextRoot() {
        // compute context root name from ejbjar name
        String archiveName = this.getArchive().getName();
        String root = null;
        if (archiveName.toLowerCase().endsWith(".jar")) {
            root = archiveName.toLowerCase().substring(0, archiveName.length() - ".jar".length());
        } else {
            root = archiveName;
        }
        if (wsDD != null) {
            String croot = wsDD.getContextRoot();
            if (croot != null) {
                root = croot;
            }
        }
        getLogger().log(BasicLevel.DEBUG, "Computed context root : " + root);
        return root;
    }

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

    public Application getApplication() {
        return app;
    }

    public Map getDescriptors() {
        return descriptors;
    }

    public boolean omit(String name) {
        return (name.equals("META-INF/jonas-ejb-jar.xml")
                || name.equals("META-INF\\jonas-ejb-jar.xml"));
    }

    public Document getJonasEjbJarDoc() {
        return jEjbJar;
    }

    public InputStream getJonasEjbJarInputStream() throws IOException {
        InputStream is = null;

        if (isPacked()) {
            is = getInputStream("META-INF/jonas-ejb-jar.xml");
        } else {
            is = getInputStream("META-INF" + File.separator
                    + "jonas-ejb-jar.xml");
        }
        if (is == null) {
            throw new IOException("Cannot read jonas specific DD for EJB as the entry is not present");
        }
        return is;
    }
}

---