J2EEServerMBean.java

package org.objectweb.jonas.server;

// Java imports
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Arrays;

import javax.enterprise.deploy.shared.ModuleType;
import javax.management.MBeanException;
import javax.management.Notification;
import javax.management.NotificationFilter;
import javax.management.NotificationListener;

import org.apache.commons.modeler.BaseModelMBean;

import org.objectweb.common.Cmd;

import org.objectweb.jonas_ear.deployment.api.EarDeploymentDesc;
import org.objectweb.jonas_ear.deployment.api.EarDeploymentDescException;
import org.objectweb.jonas_ear.deployment.lib.wrapper.EarManagerWrapper;

import org.objectweb.jonas_ejb.genic.wrapper.GenicServiceWrapper;

import org.objectweb.jonas_lib.genclientstub.wrapper.ClientGenStubWrapper;

import org.objectweb.jonas_web.deployment.api.WebContainerDeploymentDescException;
import org.objectweb.jonas_web.deployment.lib.wrapper.WebManagerWrapper;

import org.objectweb.jonas_ws.wsgen.wrapper.WsGenWrapper;

import org.objectweb.jonas.common.Log;
import org.objectweb.jonas.container.EJBService;
import org.objectweb.jonas.ear.EarService;
import org.objectweb.jonas.ear.EarServiceException;
import org.objectweb.jonas.resource.ResourceService;
import org.objectweb.jonas.service.ServiceException;
import org.objectweb.jonas.service.ServiceManager;
import org.objectweb.jonas.web.JWebContainerService;

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

public class J2EEServerMBean extends BaseModelMBean {

    private static final int BUFFER_SIZE = 1024;

    private static Logger mgtLogger = Log.getLogger(Log.JONAS_MANAGEMENT_PREFIX);

    public J2EEServerMBean() throws MBeanException {
        super();
    }

    // ------------------------------------------------------------- Public
    // methods

    public String[] getDeployedObjects() {
        ArrayList al = (ArrayList) ((J2EEServer) (this.resource)).getDeployedObjects();
        return (String[]) al.toArray(new String[al.size()]);
    }

    public String[] getResources() {
        ArrayList al = (ArrayList) ((J2EEServer) (this.resource)).getResources();
        return (String[]) al.toArray(new String[al.size()]);
    }

    public String[] getJavaVMs() {
        ArrayList al = (ArrayList) ((J2EEServer) (this.resource)).getJavaVMs();
        return (String[]) al.toArray(new String[al.size()]);
    }

    public void sendNotification(Notification pNotification) {
        ((J2EEServer) (this.resource)).sendNotification(pNotification);
    }

    public void addNotificationListener(NotificationListener pListner, NotificationFilter pFilter,
            java.lang.Object pHandback) throws java.lang.IllegalArgumentException {
        ((J2EEServer) (this.resource)).addNotificationListener(pListner, pFilter, pHandback);
    }

    public String deployJar(String fileName) throws Exception {
        return ((J2EEServer) (this.resource)).deployJar(fileName);
    }

    public void deployWar(String fileName) throws Exception {
        ((J2EEServer) (this.resource)).deployWar(fileName);
    }

    public String deployEar(String fileName) throws Exception {
        return ((J2EEServer) (this.resource)).deployEar(fileName);
    }

    public String deployRar(String fileName) throws Exception {
        return ((J2EEServer) (this.resource)).deployRar(fileName);
    }

    public Boolean isEarDeployed(String fileName) throws Exception {
        return ((J2EEServer) (this.resource)).isEarDeployed(fileName);
    }

    public Boolean isRarDeployed(String fileName) throws Exception {
        return ((J2EEServer) (this.resource)).isRarDeployed(fileName);
    }

    public Boolean isJarDeployed(String fileName) throws Exception {
        return ((J2EEServer) (this.resource)).isJarDeployed(fileName);
    }

    public Boolean isWarDeployed(String fileName) throws Exception {
        return ((J2EEServer) (this.resource)).isWarDeployed(fileName);
    }

    public String deployFile(Integer typeparam, java.lang.Byte[] bfile, String filename, String[] genicArgs) {
        String directory = "";
        int type = typeparam.intValue();

        ServiceManager serviceManager = null;
        try {
            serviceManager = ServiceManager.getInstance();
        } catch (Exception e) {
            // TODO
            mgtLogger.log(BasicLevel.ERROR, "Cannot get the instance of the Service Manager : '" + e.getMessage()
                    + "'.");
        }

        boolean isJar = false;
        if (type == ModuleType.EJB.getValue()) {
                directory = ((EJBService) serviceManager.getEjbService()).getEjbjarsDirectory();
                isJar = true;
        } else if (type == ModuleType.EAR.getValue()) {
                directory = ((EarService) serviceManager.getEarService()).getAppsDirectory();
        } else if (type == ModuleType.WAR.getValue()) {
                directory = ((JWebContainerService) serviceManager.getWebContainerService()).getWebappsDirectory();
        } else if (type == ModuleType.RAR.getValue()) {
                directory = ((ResourceService) serviceManager.getResourceService()).getRarsDirectory();
        } else if (type == ModuleType.CAR.getValue()) {
            // do nothing
        }

        File file = new File(directory + filename);
        try {
            FileOutputStream out = new FileOutputStream(file);
            byte[] bfileused = new byte[bfile.length];
            for (int i = 0; i < bfile.length; i++) {
                bfileused[i] = bfile[i].byteValue();
            }
            out.write(bfileused);
            out.close();
            if (type == ModuleType.EJB.getValue()) {
                    try {
                        callGenic(genicArgs, directory + filename);
                    } catch (ServiceException e) {
                        mgtLogger.log(BasicLevel.WARN, "deploy File error " + filename + e.getClass().getName() + " "
                                + e.getMessage());
                    }
            } else if (type == ModuleType.EAR.getValue()) {
                unpackAndCompileEar(file, genicArgs);
            } else if (type == ModuleType.WAR.getValue()) {
                 checkWebAppDeploymentDesc(file);
            } else {
                mgtLogger.log(BasicLevel.DEBUG, "Deployment not yet implemented for type '" + type + ".");
            }
        } catch (IOException ioe) {
            mgtLogger.log(BasicLevel.ERROR, "Cannot dump to outputstream the file '" + file + "' : '"
                    + ioe.getMessage() + "'.", ioe);
            // TODO
        }

        // Ugly but don't work without it ...
        try {
            Thread.sleep(1000);
        } catch (InterruptedException e1) {
            e1.printStackTrace();
        }

        if (!filename.endsWith(".rar")) {
            String modifiedArchive = null;
            ClientGenStubWrapper clientWrapper = new ClientGenStubWrapper();
            try {
                modifiedArchive = clientWrapper.callClientGenStubExecute(file.getPath());
            } catch (Exception e) {
                mgtLogger.log(BasicLevel.ERROR, "Cannot launch ClientGenStub on the file '" + file + "' : '"
                        + e.getCause().getMessage() + "'.", e);
            }

            boolean modified = false;
            try {
                modified = clientWrapper.callClientGenStubIsInputModifed();
            } catch (Exception e) {
                mgtLogger.log(BasicLevel.ERROR, "Cannot launch ClientGenStub.isInputModified '" + file + "' : '"
                        + e.getCause().getMessage() + "'.", e);
                // assume file has been modified
                modified = true;
            }

            if (modified) {
                // assume also that we can call WsGen after
                File newFile = new File(modifiedArchive);
                try {
                    copy(newFile, file);
                } catch (Exception e) {
                    mgtLogger.log(BasicLevel.ERROR, "Error when copying file '" + newFile + "'. : " + e.getMessage());
                }

                // Ugly but don't work without it ...
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e1) {
                    e1.printStackTrace();
                }

                WsGenWrapper ww = new WsGenWrapper();
                try {
                    modifiedArchive = ww.callWsGenExecute(file.getPath());
                } catch (Exception e) {
                    mgtLogger.log(BasicLevel.ERROR, "Cannot launch WsGen on the file '" + file + "' : '"
                            + e.getCause().getMessage() + "'.", e);
                }
                
                try {
                    modified = ww.callWsGenIsInputModifed();
                } catch (Exception e) {
                    mgtLogger.log(BasicLevel.ERROR, "Cannot launch WsGen on the file '" + file + "' : '"
                            + e.getCause().getMessage() + "'.", e);
                }
                
                if (modified) {
                    // assume also that we can call WsGen after
                    File newFile2 = new File(modifiedArchive);
                        // Check for modifiedArchive suffix
                        if (modifiedArchive.endsWith(".ear")) {
                                // check if input file is a jar
                                if (isJar) {
                                        // this case is special: we got a JAR in input and generate a EAR in output
                                        // the output goes to apps directory
                                        String modifiedDirectory = ((EarService) serviceManager.getEarService()).getAppsDirectory();
                                        file = new File(modifiedDirectory + newFile2.getName());
                                }
                        }
                    try {
                        copy(newFile2, file);
                    } catch (Exception e) {
                        mgtLogger.log(BasicLevel.ERROR, "Error when copying file '" + newFile2 + "'. : " + e.getMessage());
                    }
                    return file.getPath();
                }

                return modifiedArchive;
            } else {
                // don't call wsgen as it will fail with DTD too
                return file.getPath();
            }
        } else {
            return file.getPath();
        }
    }

    void copy(File src, File dst) throws IOException {
        InputStream is = new FileInputStream(src);
        OutputStream os = new FileOutputStream(dst);

        // bytes to transfer
        byte[] buf = new byte[BUFFER_SIZE];
        int len;
        while ((len = is.read(buf)) > 0) {
            os.write(buf, 0, len);
        }
        is.close();
        os.close();
    }

    private void callGenic(String[] genicArgs, String jarPath) {
        String[] args;
        if (genicArgs != null) {
            args = new String[genicArgs.length + 1];
            for (int i = 0; i < genicArgs.length; i++) {
                args[i] = genicArgs[i];
            }
            args[genicArgs.length] = jarPath;
        } else {
            args = new String[1];
            args[0] = jarPath;
        }
        mgtLogger.log(BasicLevel.DEBUG, "Calling GenIC with arguments :" + Arrays.asList(args));
        GenicServiceWrapper.callGenic(args);
    }

    private void unpackAndCompileEar(File file, String[] genicArgs) throws ServiceException {
        URL[] earUrl = new URL[1];

        try {
            earUrl[0] = file.toURL();
        } catch (MalformedURLException e) {
            // TODO
            mgtLogger.log(BasicLevel.ERROR, "The url for the file '" + file + "' cannot be built : '" + e.getMessage()
                    + "'.");
        }

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

        EarDeploymentDesc desc = null;
        try {
            desc = EarManagerWrapper.getDeploymentDesc(earUrl[0].getFile(), loaderCls);
        } catch (EarDeploymentDescException e) {
            // TODO
            mgtLogger.log(BasicLevel.ERROR, "Cannot get deployment descriptor for the Ear file '" + file + "' : '"
                    + e.getMessage() + "'.");
        }

        String[] ejbTags = desc.getEjbTags();
        if (ejbTags.length != 0) {
            //  need to call GENIC
            /*
             * 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++) {
                try {
                    callGenic(genicArgs, ejbTags[i]);
                } catch (ServiceException e) {
                    throw (e);
                }
            }
            // add modified jar into ear
            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();
            }
        }
    }

    private void checkWebAppDeploymentDesc(File file) {
        URL warUrl = null;

        try {
            warUrl = file.toURL();
        } catch (MalformedURLException mue) {
            mgtLogger.log(BasicLevel.ERROR, "The url for the file '" + file + "' cannot be built : '"
                    + mue.getMessage() + "'.");
            throw new RuntimeException(mue);
        }

        URLClassLoader warCl = new URLClassLoader(new URL[] {warUrl});

        try {
            WebManagerWrapper.getDeploymentDesc(warUrl, warCl, null);
        } catch (WebContainerDeploymentDescException e) {
            String err = "Cannot read the deployment descriptors '" + warUrl.getFile() + "'";
            mgtLogger.log(BasicLevel.ERROR, err + ": " + e);
            e.printStackTrace(System.err);
            throw new RuntimeException(err, e);
        }

    }

    //
    //    public static void addResource(String pObjectName) {
    //        try {
    //            Server oServer = Server.getInstance();
    //            J2EEServer oJ2EEServer = oServer.getJ2EEServer();
    //            oJ2EEServer.addResource(pObjectName);
    //        } catch (Exception e) {
    //            mgtLogger.log(BasicLevel.WARN
    //                , "Add resource error (object=" + pObjectName + ") : "
    //                + e.getClass().getName() + " " + e.getMessage());
    //        }
    //    }
    //    public static String removeResource(String pObjectName) {
    //        String sRet = null;
    //        try {
    //            Server oServer = Server.getInstance();
    //            J2EEServer oJ2EEServer = oServer.getJ2EEServer();
    //            sRet = oJ2EEServer.removeResource(pObjectName);
    //        } catch (Exception e) {
    //            mgtLogger.log(BasicLevel.WARN
    //                , "Remove resource object error (object=" + pObjectName + ") : "
    //                + e.getClass().getName() + " " + e.getMessage());
    //        }
    //        return sRet;
    //    }
}

---