BeanSources.java

package org.objectweb.jonas_ejb.genic;

import java.io.File;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;

import org.apache.velocity.app.VelocityEngine;

import org.objectweb.jonas_ejb.deployment.api.BeanDesc;
import org.objectweb.jonas_ejb.deployment.api.EntityCmp2Desc;
import org.objectweb.jonas_ejb.deployment.api.EntityDesc;
import org.objectweb.jonas_ejb.deployment.api.EntityJdbcCmp1Desc;
import org.objectweb.jonas_ejb.deployment.api.SessionStatelessDesc;
import org.objectweb.jonas_ejb.lib.BeanNaming;
import org.objectweb.jonas_ejb.lib.RdbMappingBuilder;

import org.objectweb.jonas.common.Log;

import org.objectweb.jorm.compiler.api.JormCompilerParameter;
import org.objectweb.jorm.compiler.lib.JormCompiler;

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

public class BeanSources {

    private static final String JAVA_SUFFIX = ".java";

    private static final String RDB_PREFIX = "rdb";

    private static final String CLUSTER_SUFFIX = "_Cluster.xml";

    private String wrpRemoteFileName = null;

    private String wrpLocalFileName = null;

    private String wrpHomeFileName = null;

    private String wrpHomeClusterConfFileName = null;

    private String wrpLocalHomeFileName = null;

    private String wrpHandleFileName = null;

    private String wrpBeanFileName = null;

    private String wrpServiceEndpointFileName = null;
    
    private String wrpSEHomeFileName = null;

    private String itfCohCmp2Entity = null;

    private ArrayList sources;

    private ArrayList noRemoteJava;

    private String outdir = null;

    private BeanDesc dd = null;

    private JormCompiler jormCompiler;

    private Logger logger = null;

    public BeanSources(BeanDesc beanDesc, String dirOutputName, VelocityEngine ve, JormCompiler jormCompiler)
            throws GenICException {
        this(beanDesc, dirOutputName, ve);
        this.jormCompiler = jormCompiler;
    }

    public BeanSources(BeanDesc beanDesc, String dirOutputName, VelocityEngine ve) throws GenICException {

        logger = Log.getLogger(Log.JONAS_GENIC_PREFIX);
        dd = beanDesc;
        outdir = dirOutputName;
        sources = new ArrayList();
        noRemoteJava = new ArrayList();

        // generated Home
        if (dd.getHomeClass() != null) {
            if (outdir.length() > 0) {
                wrpHomeFileName = outdir + File.separatorChar + BeanNaming.getPath(dd.getFullWrpHomeName());
            } else {
                wrpHomeFileName = dd.getWrpHomeName();
            }
            wrpHomeClusterConfFileName = wrpHomeFileName.concat(CLUSTER_SUFFIX);
            sources.add(new Source(dd, wrpHomeClusterConfFileName, Source.CLUSTER_HOME, ve));
            wrpHomeFileName = wrpHomeFileName.concat(JAVA_SUFFIX);
            sources.add(new Source(dd, wrpHomeFileName, Source.HOME, ve));
            // generated Handle for Remote Entity bean
            if (dd instanceof EntityDesc) {
                if (outdir.length() > 0) {
                    wrpHandleFileName = outdir + File.separatorChar + BeanNaming.getPath(dd.getFullWrpHandleName());
                } else {
                    wrpHandleFileName = dd.getWrpHandleName();
                }
                wrpHandleFileName = wrpHandleFileName.concat(JAVA_SUFFIX);
                noRemoteJava.add(wrpHandleFileName);
                sources.add(new Source(dd, wrpHandleFileName, Source.ENTITY_HANDLE, ve));
            }
        }

        // generated EJBObject (Remote)
        if (dd.getRemoteClass() != null) {
            if (outdir.length() > 0) {
                wrpRemoteFileName = outdir + File.separatorChar + BeanNaming.getPath(dd.getFullWrpRemoteName());
            } else {
                wrpRemoteFileName = dd.getWrpRemoteName();
            }
            wrpRemoteFileName = wrpRemoteFileName.concat(JAVA_SUFFIX);
            sources.add(new Source(dd, wrpRemoteFileName, Source.REMOTE, ve));
        }

        // generated LocalHome
        if (dd.getLocalHomeClass() != null) {
            if (outdir.length() > 0) {
                wrpLocalHomeFileName = outdir + File.separatorChar + BeanNaming.getPath(dd.getFullWrpLocalHomeName());
            } else {
                wrpLocalHomeFileName = dd.getWrpLocalHomeName();
            }
            wrpLocalHomeFileName = wrpLocalHomeFileName.concat(JAVA_SUFFIX);
            noRemoteJava.add(wrpLocalHomeFileName);
            sources.add(new Source(dd, wrpLocalHomeFileName, Source.LOCAL_HOME, ve));
        }

        // generated EJBLocalObject (Local)
        if (dd.getLocalClass() != null) {
            if (outdir.length() > 0) {
                wrpLocalFileName = outdir + File.separatorChar + BeanNaming.getPath(dd.getFullWrpLocalName());
            } else {
                wrpLocalFileName = dd.getWrpLocalName();
            }
            wrpLocalFileName = wrpLocalFileName.concat(JAVA_SUFFIX);
            noRemoteJava.add(wrpLocalFileName);
            sources.add(new Source(dd, wrpLocalFileName, Source.LOCAL, ve));
        }

        // generated ServiceEndpoint and SEHome
        if (dd instanceof SessionStatelessDesc) {
            SessionStatelessDesc ssd = (SessionStatelessDesc) dd;
            if (ssd.getServiceEndpointClass() != null) {
                // ServiceEndpoint
                if (outdir.length() > 0) {
                    wrpServiceEndpointFileName = outdir + File.separatorChar
                            + BeanNaming.getPath(ssd.getFullWrpServiceEndpointName());
                } else {
                    wrpServiceEndpointFileName = ssd.getWrpServiceEndpointName();
                }
                wrpServiceEndpointFileName = wrpServiceEndpointFileName.concat(JAVA_SUFFIX);
                //noRemoteJava.add(wrpServiceEndpointFileName); // Accessed locally only
                sources.add(new Source(dd, wrpServiceEndpointFileName, Source.SERVICE_ENDPOINT, ve));
                // ServiceEndpointHome
                if (outdir.length() > 0) {
                    wrpSEHomeFileName = outdir + File.separatorChar
                            + BeanNaming.getPath(ssd.getFullWrpSEHomeName());
                } else {
                    wrpSEHomeFileName = ssd.getWrpSEHomeName();
                }
                wrpSEHomeFileName = wrpSEHomeFileName.concat(JAVA_SUFFIX);
                noRemoteJava.add(wrpSEHomeFileName);    // Accessed locally only
                sources.add(new Source(dd, wrpSEHomeFileName, Source.SERVICE_ENDPOINT_HOME, ve));
            }
        }

        // generated derived bean class for Entity Bean CMP
        if (dd instanceof EntityJdbcCmp1Desc) {
            if (outdir.length() > 0) {
                wrpBeanFileName = outdir + File.separatorChar + BeanNaming.getPath(dd.getFullDerivedBeanName());
            } else {
                wrpBeanFileName = dd.getDerivedBeanName();
            }
            wrpBeanFileName = wrpBeanFileName.concat(JAVA_SUFFIX);
            noRemoteJava.add(wrpBeanFileName);
            sources.add(new Source(dd, wrpBeanFileName, Source.ENTITY_CMP_JDBC, ve));
        }
        if (dd instanceof EntityCmp2Desc) {
            EntityCmp2Desc ecd = (EntityCmp2Desc) dd;
            if (outdir.length() > 0) {
                wrpBeanFileName = outdir + File.separatorChar + BeanNaming.getPath(dd.getFullDerivedBeanName())
                        + JAVA_SUFFIX;
                if (ecd.needJormCoherenceHelper()) {
                    itfCohCmp2Entity = outdir + File.separatorChar
                            + BeanNaming.getPath(ecd.getJormCoherenceHelperFQItfName()) + JAVA_SUFFIX;
                }
            } else {
                wrpBeanFileName = BeanNaming.getPath(dd.getDerivedBeanName()) + JAVA_SUFFIX;
                if (ecd.needJormCoherenceHelper()) {
                    itfCohCmp2Entity = BeanNaming.getPath(ecd.getJormCoherenceHelperItfName()) + JAVA_SUFFIX;
                }
            }
            if (ecd.needJormCoherenceHelper()) {
                sources.add(new Source(dd, itfCohCmp2Entity, Source.ITF_COH_CMP2_ENTITY, ve));
                noRemoteJava.add(itfCohCmp2Entity);
            }
            noRemoteJava.add(wrpBeanFileName);
            sources.add(new Source(dd, wrpBeanFileName, Source.ENTITY_CMP_JDBC, ve));
        }
    }

    public void generate() throws GenICException {
        // Generates the java sources
        for (Iterator it = sources.iterator(); it.hasNext();) {
            Source src = (Source) it.next();
            src.generate();
        }

        // In case of Cmp2, call jorm compiler.
        if (dd instanceof EntityCmp2Desc) {
            EntityCmp2Desc ecd = (EntityCmp2Desc) dd;

            // Build and fill a JormCompilerParameter dedicated to the bean
            JormCompilerParameter cp = jormCompiler.getCompilerParameter();
            cp.setProjectName(RdbMappingBuilder.getProjectName());
            cp.setKeepSrc(true);
            cp.setOutput(outdir);
            cp.setClassMappingInheritance("org.objectweb.jonas_ejb.container.jorm.RdbFactory");
            switch (ecd.getLockPolicy()) {
                case EntityDesc.LOCK_CONTAINER_READ_UNCOMMITTED:
                    cp.setBindingInheritance("org.objectweb.jonas_ejb.container.JEntitySwitchCRU");
                    break;
                case EntityDesc.LOCK_CONTAINER_SERIALIZED:
                    cp.setBindingInheritance("org.objectweb.jonas_ejb.container.JEntitySwitchCS");
                    break;
                case EntityDesc.LOCK_CONTAINER_READ_COMMITTED:
                    cp.setBindingInheritance("org.objectweb.jonas_ejb.container.JEntitySwitchCRC");
                    break;
                case EntityDesc.LOCK_DATABASE:
                    cp.setBindingInheritance("org.objectweb.jonas_ejb.container.JEntitySwitchDB");
                    break;
                case EntityDesc.LOCK_READ_ONLY:
                    cp.setBindingInheritance("org.objectweb.jonas_ejb.container.JEntitySwitchRO");
                    break;
                default:
                    throw new GenICException("Cannot find JEntitySwitch: Unknown lock policy");
            }

            // Run the Jorm Compiler
            Collection jormflist = null;
            try {
                // For debug only: Generate the '.pd' Jorm Files
                if (Log.getLogger("org.objectweb.jonas_ejb.mijorm").isLoggable(BasicLevel.DEBUG)) {
                    jormCompiler.generateJormFiles(ecd.getJormList());
                }
                jormflist = jormCompiler.generateFiles(ecd.getJormList());
            } catch (Exception e) {
                throw new GenICException("Problem during jorm generation", e);
            }
            // Save the list of Jorm generated files
            for (Iterator i = jormflist.iterator(); i.hasNext();) {
                String file = (String) i.next();
                logger.log(BasicLevel.DEBUG, "Jorm generated file: " + file);
                noRemoteJava.add(file);
            }
        }
    }

    public String getEjbName() {
        return dd.getEjbName();
    }

    public String getWrpHomeFileName() {
        return wrpHomeFileName;
    }

    public String getWrpHomeClusterConfFileName() {
        return wrpHomeClusterConfFileName;
    }

    public String getWrpRemoteFileName() {
        return wrpRemoteFileName;
    }

    public String getWrpRemoteClassName() {
        return dd.getFullWrpRemoteName();
    }

    public String getWrpServiceEndpointFileName() {
        return wrpServiceEndpointFileName;
    }

    public String getWrpSEHomeFileName() {
        return wrpSEHomeFileName;
    }

    public String getWrpServiceEndpointClassName() {
        if (dd instanceof SessionStatelessDesc) {
            return ((SessionStatelessDesc) dd).getFullWrpServiceEndpointName();
        }
        return null;
    }

    public String getWrpHomeClassName() {
        return dd.getFullWrpHomeName();
    }

    public Collection getNoRemoteJavas() {
        return noRemoteJava;
    }
}

---