Adm.java

package org.objectweb.jonas.adm;


import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.rmi.RemoteException;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.Properties;
import java.util.Vector;

import javax.naming.InitialContext;
import javax.naming.NameClassPair;
import javax.naming.NamingEnumeration;
import javax.naming.NamingException;
import javax.rmi.PortableRemoteObject;

import org.objectweb.common.Cmd;
import org.objectweb.jonas.common.JProp;
import org.objectweb.jonas.common.LogManagement;
import org.objectweb.jonas.container.EJBServiceImpl;
import org.objectweb.jonas.ear.EarServiceException;
import org.objectweb.jonas.ear.EarServiceImplMBean;
import org.objectweb.jonas.jtm.TransactionServiceImpl;
import org.objectweb.jonas.mail.MailServiceImplMBean;
import org.objectweb.jonas.naming.NamingManager;
import org.objectweb.jonas.resource.ResourceServiceException;
import org.objectweb.jonas.resource.ResourceServiceImplMBean;
import org.objectweb.jonas.service.ServiceException;
import org.objectweb.jonas.service.ServiceManager;
import org.objectweb.jonas.web.AbsJWebContainerServiceImpl;
import org.objectweb.jonas.web.AbsJWebContainerServiceImplMBean;
import org.objectweb.jonas.web.JWebContainerServiceException;
import org.objectweb.jonas_ear.deployment.api.EarDeploymentDesc;
import org.objectweb.jonas_ear.deployment.api.EarDeploymentDescException;
import org.objectweb.jonas_ejb.container.Container;
import org.objectweb.jonas_ejb.genic.GenIC;
import org.objectweb.jonas_web.deployment.api.WebContainerDeploymentDesc;
import org.objectweb.jonas_web.deployment.lib.wrapper.WebManagerWrapper;
import org.objectweb.jonas_ear.deployment.lib.wrapper.EarManagerWrapper;

public class Adm extends PortableRemoteObject implements AdmInterface {

    public static final String ADMNAME_SUFFIX = "_Adm";

    private ServiceManager sm = null;
    private EJBServiceImpl ejbserv  = null;
    private TransactionServiceImpl tm = null;
    private JProp jonasProperties = null;
    private InitialContext ictx = null;
    private boolean isEJBContainer;

    // Server State
    public static final int NOT_READY = 0;
    public static final int READY = 1;
    public static final int STOPPED = 2;
    private int serverState = NOT_READY;

    private AbsJWebContainerServiceImplMBean webContainerService = null;

    private EarServiceImplMBean earService = null;

    private ResourceServiceImplMBean resourceService = null;

    private MailServiceImplMBean mailService = null;

    public Adm(JProp jp) throws RemoteException, NamingException, ServiceException, Exception {

        jonasProperties = jp;

        // Get Service references
        sm = ServiceManager.getInstance();

        // Get an Initial Context and
        // registers the JOnAS Server Name to JNDI
        String name = null;
        name = jonasProperties.getValue("jonas.name", "jonas") + ADMNAME_SUFFIX;
        try {
            ictx = NamingManager.getInstance().getInitialContext();
            ictx.rebind(name, this);
        } catch (NamingException e) {
            throw e;
        }
    }

    private TransactionServiceImpl getTM() {
        if (tm == null) {
            try {
                tm = (TransactionServiceImpl) sm.getTransactionService();
            } catch (ServiceException e) {
            }
        }
        return tm;
    }

    private AbsJWebContainerServiceImplMBean getWebContainerService() {
        if (webContainerService == null) {
            try {
                webContainerService = (AbsJWebContainerServiceImplMBean) sm.getWebContainerService();
            } catch (ServiceException e) {
                //not started
            }
        }
        return webContainerService;
    }

    private MailServiceImplMBean getMailService() {
        if (mailService == null) {
            try {
                mailService = (MailServiceImplMBean) sm.getMailService();
            } catch (Exception e) {
                //not started
            }
        }
        return mailService;
    }

    private EarServiceImplMBean getEarService() {
        if (earService == null) {
            try {
                earService = (EarServiceImplMBean) sm.getEarService();
            } catch (Exception e) {
                //not started
            }
        }
        return earService;
    }

    private ResourceServiceImplMBean getResourceService() {
        if (resourceService == null) {
            try {
                resourceService = (ResourceServiceImplMBean) sm.getResourceService();
            } catch (Exception e) {
                //not started
            }
        }
        return resourceService;
    }

    // ------------------------------------------------------------------
    //  AdmInterface implementation
    // ------------------------------------------------------------------

    public String [] getTopics() throws RemoteException {
        return LogManagement.getInstance().getTopics();
    }

    public String getTopicLevel(String topic) throws RemoteException {
        return LogManagement.getInstance().getTopicLevel(topic);
    }

    public void setTopicLevel(String topic, String l) throws RemoteException {
        LogManagement.getInstance().setTopicLevel(topic, l);
    }

    public void addBeans(String fileName) throws RemoteException {
        if (!isEJBContainer) {
            return;
        }
        try {
            ejbserv.createContainer(fileName);
        } catch (Exception e) {
            throw new RemoteException(e.toString());
        }
    }


    public void addEar(String fileName) throws RemoteException, EarServiceException {
        if (getEarService() != null) {
            try {
                getEarService().deployEarMBean(fileName);
            } catch (EarServiceException ear) {
                throw ear;
            } catch (Exception e) {
                throw new RemoteException(e.toString());
            }
        } else {
            throw new EarServiceException("Can't add the Ear file " + fileName + ". The ear service is not started");
        }
    }

    public boolean isEarLoaded(String fileName) throws RemoteException, EarServiceException {
        boolean isLoaded = false;
        if (getEarService() != null) {
            isLoaded = getEarService().isEarLoaded(fileName);
        } else {
            throw new EarServiceException("Can't test if the ear file " + fileName + " is deployed or not. The ear service is not started");
        }
        return isLoaded;
    }

    public void addRar(String fileName) throws RemoteException, ResourceServiceException {
        if (getResourceService() != null) {
            getResourceService().deployRarMBean(fileName);
        } else {
            throw new ResourceServiceException("Can't add the Rar file " + fileName + ". The resource service is not started");
        }
    }

    public boolean isRarLoaded(String fileName) throws RemoteException, ResourceServiceException {
        boolean isLoaded = false;
        if (getResourceService() != null) {
            isLoaded = getResourceService().isRarLoaded(fileName);
        } else {
            throw new ResourceServiceException("Can't test if the rar file " + fileName + " is deployed or not. The rar service is not started");
        }
        return isLoaded;
    }

    public void addWar(String fileName) throws RemoteException, JWebContainerServiceException {
        if (getWebContainerService() != null) {
            getWebContainerService().registerWarMBean(fileName);
        } else {
            throw new JWebContainerServiceException("Can't add the war file " + fileName + ". The web container service is not started");
        }
    }

    public boolean isWarLoaded(String fileName) throws RemoteException, JWebContainerServiceException {
        boolean isLoaded = false;
        if (getWebContainerService() != null) {
            isLoaded = getWebContainerService().isWarLoaded(fileName);
        } else {
            throw new JWebContainerServiceException("Can't test if the war file " + fileName + " is deployed or not. The war service is not started");
        }
        return isLoaded;
    }

    public void removeEar(String fileName) throws RemoteException, EarServiceException {
        if (getEarService() != null) {
            try {
                getEarService().unDeployEarMBean(fileName);
                // needed here to run also the distributed garbage collector
                runGC();
            } catch (EarServiceException ear) {
                throw ear;
            } catch (Exception e) {
                throw new RemoteException(e.toString());
            }
        } else {
            throw new EarServiceException("Can't remove of the Ear file " + fileName + ". The ear service is not started");
        }
    }

    public void removeRar(String fileName) throws RemoteException, ResourceServiceException {
        if (getResourceService() != null) {
            getResourceService().unDeployRarMBean(fileName);
            // needed here to run also the distributed garbage collector
            runGC();
        } else {
            throw new ResourceServiceException("Can't remove of the Rar file " + fileName + ". The resource service is not started");
        }
    }

    public void removeWar(String fileName) throws RemoteException, JWebContainerServiceException {
        if (getWebContainerService() != null) {
            getWebContainerService().unRegisterWarMBean(fileName);
            // needed here to run also the distributed garbage collector
            runGC();
        } else {
            throw new JWebContainerServiceException("Can't remove of the war file " + fileName + ". The web container service is not started");
        }
    }


    public void removeBeans(String fileName) throws RemoteException {
        if (!isEJBContainer) {
            return;
        }
        try {
            ejbserv.removeContainer(fileName);
        } catch (Exception e) {
            throw new RemoteException(e.getMessage());
        }
        // needed here to run also the distributed garbage collector
        runGC();
    }

    public boolean isLoaded(String fileName) throws RemoteException {
        return (ejbserv.getContainer(fileName) != null);
    }

    public String [] listBeans() throws RemoteException {

        if (!isEJBContainer) {
            return new String[0];
        }

        Container[] lcont = ejbserv.listContainers();
        Vector lbeans = new Vector();
        for (int j = 0; j < lcont.length; j++) {
            String[] lbn = lcont[j].listBeanNames();
            String contname = lcont[j].getName();
            for (int k = 0; k < lbn.length; k++) {
                String s = new String(contname + ": " + lbn[k]);
                lbeans.addElement(s);
            }
        }
        String[] ret = new String[lbeans.size()];
        lbeans.copyInto(ret);

        return ret;
    }

    public Vector listContext() throws RemoteException {
        String name = null;
        Vector ret  = new Vector();
        try {
            NamingEnumeration ne;
            ne = NamingManager.getInstance().getInitialContext().list("");
            for (Enumeration e = ne; e.hasMoreElements(); ) {
                name = ((NameClassPair) e.nextElement()).getName();
                ret.addElement(name);
            }
        } catch (Exception ex) {
            throw new RemoteException("Cannot list JNDI context", ex);
        }
        return ret;
    }

    public Properties listEnv() {
        return jonasProperties.getConfigFileEnv();
    }

    public void stopServer() throws RemoteException {
        try {
            serverState = STOPPED;
            // keep registry alive a little while (for jonas admin) in case of error at launch
            Thread.sleep(5000);
            sm.stopServices();
        } catch (ServiceException e) {
            throw new RemoteException("Cannot stop services: ", e);
        } catch (InterruptedException e) {
        }
    }


    public void killServer() throws RemoteException {
        stopServer();
        System.exit(0);
    }

    public int getServerState() throws RemoteException {
        return serverState;
    }

    public boolean isEJBContainer() throws RemoteException {
        return isEJBContainer;
    }

    public void setTransactionTimeout(int timeout) throws RemoteException {
        getTM().setTimeout(timeout);
    }

    public void runGC() throws RemoteException {
        Runtime.getRuntime().gc();
    }

    public void syncAllEntities(boolean passivate) throws RemoteException {
        if (!isEJBContainer) {
            return;
        }
        ejbserv.syncAllEntities(passivate);
    }


    // ------------------------------------------------------------------
    //  other public methods
    // ------------------------------------------------------------------

    public void serverReady(boolean isEJB) {
        serverState = READY;
        isEJBContainer = isEJB;
        if (isEJBContainer) {
            ejbserv = (EJBServiceImpl) sm.getEjbService();
        }
    }

    public String deployFile(int type, byte[] bfile, String filename) throws RemoteException , EarServiceException, JWebContainerServiceException {
        try {
            String directory = "";

            if (type == TYPE_EJB) {
                directory = ejbserv.getEjbjarsDirectory();
            } else if (type == TYPE_EAR) {
                directory = getEarService().getAppsDirectory();
            } else if (type == TYPE_WAR) {
                directory = getWebContainerService().getWebappsDirectory();
            } else if (type == TYPE_RAR) {
                directory = getResourceService().getRarsDirectory();
            } else if (type == TYPE_CAR) {
                throw new UnsupportedOperationException("Not Supported yet");
            }

            File file = new File(directory + filename);

            FileOutputStream out = new FileOutputStream(file);
            out.write(bfile);
            out.close();
            if (type == TYPE_EJB) {
                String args[] = new String[1];
                args[0] = directory + filename;
                GenIC.main(args);
            } else if (type == TYPE_EAR) {
                unpackAndCompileEar(file);
            }

            return file.getAbsolutePath();

        } catch (Exception e) {
            e.printStackTrace();
            throw new RemoteException(e.toString(), e);
        }
    }

    private void unpackAndCompileEar(File file) throws EarServiceException, IOException, EarDeploymentDescException {
        URL earUrl[] = new URL[1];

        try {
            earUrl[0] = file.toURL();
        } catch (MalformedURLException e) {
            String err = "Invalid ear file name '" + file;
            throw new EarServiceException(err, e);
        }

        //Create classLoader
        //parent classloader is the current classloader
        ClassLoader currentLoader = Thread.currentThread().getContextClassLoader();
        URLClassLoader loaderCls = new URLClassLoader(earUrl, currentLoader);

        EarDeploymentDesc desc = EarManagerWrapper.getDeploymentDesc(earUrl[0].getFile(), loaderCls);

        String[] ejbTags = desc.getEjbTags();
        /*
         * Example of a jar command:
         *   jar -uf ../output/ejbjars/sb.jar -C /tmp/genic1547.tmp a/b/C1.class
         *                                    -C /tmp/genic1547.tmp a/b/C2.class
         *                                    .....
         */
        String javaHomeBin = System.getProperty("java.home", "");
        if (!("".equals(javaHomeBin))) {
            javaHomeBin = javaHomeBin + File.separator + ".."
                + File.separator + "bin" + File.separator;
        }

        Cmd cmd = null;
        cmd = new Cmd(javaHomeBin + "jar");
        cmd.addArgument("-xf");
        cmd.addArgument(file.getAbsolutePath());

        for (int i = 0; i < ejbTags.length; i++) {
            cmd.addArgument(ejbTags[i]);
        }

        boolean exitCmd = cmd.run();
        // Analyse the result of the jar command
        if (!exitCmd) {
            throw new EarServiceException("Failed when extracting the the ejb jar "
                                          + "in the given jar file '" + file + "'.");
        }

        for (int i = 0; i < ejbTags.length; i++) {
            String args[] = new String[1];
            args[0] = ejbTags[i];
            GenIC.main(args);
        }

        cmd = new Cmd(javaHomeBin + "jar");
        cmd.addArgument("-uf");
        cmd.addArgument(file.getAbsolutePath());

        for (int i = 0; i < ejbTags.length; i++) {
            cmd.addArgument(ejbTags[i]);
        }
        exitCmd = cmd.run();
        // Analyse the result of the jar command
        if (!exitCmd) {
            throw new EarServiceException("Failed when extracting the the ejb jar "
                                          + "in the given jar file '" + file + "'.");
        }

        for (int i = 0; i < ejbTags.length; i++) {
            File del = new File(ejbTags[i]);
            del.delete();
        }

    }

    public List listModules(int type, int state) throws RemoteException {
        try {

            //Should look at refactoring this logic into the services instead.
            List modules = new ArrayList();
            if (type == TYPE_EJB) {
                // Will need to do parent element in future
                List jars = null;
                if (state == STATUS_RUNNING || state == STATUS_ALL) {
                    jars = ejbserv.getDeployedJars();
                    for (int i = 0; i < jars.size(); i++) {
                        ModuleDesc desc = new ModuleDesc((String)jars.get(i), STATUS_RUNNING, null);
                        modules.add(desc);
                    }
                } else if (state == STATUS_STOPPED || state == STATUS_ALL) {
                    jars = ejbserv.getDeployableJars();
                    for (int i = 0; i < jars.size(); i++) {
                        ModuleDesc desc = new ModuleDesc((String)jars.get(i), STATUS_STOPPED, null);
                        modules.add(desc);
                    }
                }
            } else if (type == TYPE_EAR) {
                // Will need to do sub elements in future
                List ears = null;
                if (state == STATUS_RUNNING || state == STATUS_ALL) {
                    ears = getEarService().getDeployedEars();
                    for (int i = 0; i < ears.size(); i++) {
                        ModuleDesc desc = new ModuleDesc((String)ears.get(i), STATUS_RUNNING, null);
                        modules.add(desc);
                    }
                } else if (state == STATUS_STOPPED || state == STATUS_ALL) {
                    ears = getEarService().getDeployableEars();
                    for (int i = 0; i < ears.size(); i++) {
                        ModuleDesc desc = new ModuleDesc((String)ears.get(i), STATUS_STOPPED, null);
                        modules.add(desc);
                    }
                }
            } else if (type == TYPE_WAR) {
                // Will need to do parent element in future
                List wars = null;
                AbsJWebContainerServiceImpl webService = (AbsJWebContainerServiceImpl)getWebContainerService(); // Very evil.
                if (state == STATUS_RUNNING || state == STATUS_ALL) {
                    wars = webService.getDeployedWars();
                    for (int i = 0; i < wars.size(); i++) {
                        // Need to get the Context root of the web app as well.
                        String warName = (String)wars.get(i);
                        ClassLoader cl = webService.getClassLoader(new File(warName).toURL(), null, null);
                        WebContainerDeploymentDesc wdesc = WebManagerWrapper.getDeploymentDesc(warName, cl);
                        int start = warName.lastIndexOf("/");
                        if (start < 0) {
                            start = 0;
                        }
                        String url = wdesc.getContextRoot() != null ? wdesc.getContextRoot(): warName.substring(start, warName.indexOf(".war"));
                        ModuleDesc desc = new ModuleDesc(warName, STATUS_RUNNING, url);
                        modules.add(desc);
                    }
                } else if (state == STATUS_STOPPED || state == STATUS_ALL) {
                    wars = webService.getDeployableWars();
                    for (int i = 0; i < wars.size(); i++) {
                        // Need to get the Context root of the web app as well.
                        String warName = (String)wars.get(i);
                        ClassLoader cl = webService.getClassLoader(new File(warName).toURL(), null, null);
                        WebContainerDeploymentDesc wdesc = WebManagerWrapper.getDeploymentDesc(warName, cl);
                        int start = warName.lastIndexOf("/");
                        if (start < 0) {
                            start = 0;
                        }

                        String url = wdesc.getContextRoot() != null ? wdesc.getContextRoot(): warName.substring(start, warName.indexOf(".war"));
                        ModuleDesc desc = new ModuleDesc(warName, STATUS_STOPPED, url);
                        modules.add(desc);
                    }
                }
            } else if (type == TYPE_RAR) {
                // Will need to do parent element in future
                List rars = getResourceService().getInstalledRars();
                for (int i = 0; i < rars.size(); i++) {
                    ModuleDesc desc = new ModuleDesc((String)rars.get(i), STATUS_RUNNING, null);
                    modules.add(desc);
                }
            } else if (type == TYPE_CAR) {
                throw new UnsupportedOperationException("Not Supported yet");
            }

            return modules;
        } catch (Exception e) {
            //Ugly
            throw new RemoteException("Failed to list modules", e);
        }


    }

    public void undeployFile(String filename) throws RemoteException {
        try {
            File f = new File(filename);
            f.delete();
        } catch (Exception ioe) {
            throw new RemoteException("Failed to remove the module " + filename + ":" + ioe.getMessage(), ioe);
        }

    }


}

---