Source.java

package org.objectweb.jonas_ejb.genic;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;

import org.objectweb.jonas_ejb.deployment.api.BeanDesc;
import org.objectweb.jonas_ejb.deployment.api.EntityDesc;
import org.objectweb.jonas_ejb.deployment.api.EntityJdbcCmp1Desc;
import org.objectweb.jonas_ejb.deployment.api.EntityJdbcCmp2Desc;
import org.objectweb.jonas_ejb.deployment.api.SessionDesc;
import org.objectweb.jonas_ejb.deployment.api.SessionStatefulDesc;
import org.objectweb.jonas_ejb.deployment.api.SessionStatelessDesc;

import org.objectweb.jonas.common.Log;

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

class Source {

    static final int HOME = 0;
    static final int LOCAL_HOME = 1;
    static final int REMOTE = 2;
    static final int LOCAL = 3;
    static final int ENTITY_HANDLE = 4;
    static final int ENTITY_CMP_JDBC = 5;
    static final int ITF_COH_CMP2_ENTITY = 7;
    static final int CLUSTER_HOME = 8;
    static final int SERVICE_ENDPOINT = 9;
    static final int SERVICE_ENDPOINT_HOME = 10;

    private BeanDesc dd = null;

    private String srcFileName = null;

    private int srcType;

    private VelocityEngine vEngine = null;

    private VelocityContext vctx = null;

    private Logger logger = null;

    Source(BeanDesc beanDesc, String fileName, int type, VelocityEngine ve) throws GenICException {
        dd = beanDesc;
        srcFileName = fileName;
        srcType = type;
        vctx = VContextFactory.create(dd, srcType);
        vEngine = ve;
        logger = Log.getLogger(Log.JONAS_GENIC_PREFIX);
        // Trace of the Velocity Context
        Logger vLogger = Log.getLogger(Log.JONAS_GENIC_VELOCITY_PREFIX);
        if (vLogger.getCurrentIntLevel() == BasicLevel.DEBUG) {
            vLogger.log(BasicLevel.DEBUG, "Source(..,fileName=" + fileName + ", type = " + type + ", ..)");
            vLogger.log(BasicLevel.DEBUG, "VELOCITY CONTEXT = \n(" + VContextFactory.toString(vctx) + "\n)");
        }
    }

    void generate() throws GenICException {

        String tmplName = null;
        Template tmpl = null;
        FileWriter fwriter = null;

        switch (srcType) {
            case CLUSTER_HOME:
                if (dd instanceof EntityDesc) {
                    tmplName = "JEntityClusterHome.vm";
                } else if (dd instanceof SessionStatefulDesc) {
                    tmplName = "JStatefulClusterHome.vm";
                } else if (dd instanceof SessionStatelessDesc) {
                    tmplName = "JStatelessClusterHome.vm";
                }
                break;
            case HOME:
                if (dd instanceof EntityDesc) {
                    tmplName = "JEntityHome.vm";
                } else if (dd instanceof SessionStatefulDesc) {
                    tmplName = "JStatefulHome.vm";
                } else if (dd instanceof SessionStatelessDesc) {
                    tmplName = "JStatelessHome.vm";
                }
                break;
            case LOCAL_HOME:
                if (dd instanceof EntityDesc) {
                    tmplName = "JEntityLocalHome.vm";
                } else if (dd instanceof SessionStatefulDesc) {
                    tmplName = "JStatefulLocalHome.vm";
                } else if (dd instanceof SessionStatelessDesc) {
                    tmplName = "JStatelessLocalHome.vm";
                }
                break;
            case REMOTE:
                if (dd instanceof EntityDesc) {
                    tmplName = "JEntityRemote.vm";
                } else if (dd instanceof SessionDesc) {
                    tmplName = "JSessionRemote.vm";
                }
                break;
            case LOCAL:
                if (dd instanceof EntityDesc) {
                    tmplName = "JEntityLocal.vm";
                } else if (dd instanceof SessionDesc) {
                    tmplName = "JSessionLocal.vm";
                }
                break;
            case SERVICE_ENDPOINT:
                if (dd instanceof SessionStatelessDesc) {
                    tmplName = "JServiceEndpoint.vm";
                }
                break;
            case SERVICE_ENDPOINT_HOME:
                if (dd instanceof SessionStatelessDesc) {
                    tmplName = "JServiceEndpointHome.vm";
                }
                break;
            case ENTITY_HANDLE:
                if (dd instanceof EntityDesc) {
                    tmplName = "JEntityHandle.vm";
                }
                break;
            case ENTITY_CMP_JDBC:
                if (dd instanceof EntityJdbcCmp1Desc) {
                    tmplName = "JEntityCmpJdbc.vm";
                }
                if (dd instanceof EntityJdbcCmp2Desc) {
                    tmplName = "JEntityCmp2.vm";
                }
                break;
            case ITF_COH_CMP2_ENTITY:
                if (dd instanceof EntityJdbcCmp2Desc) {
                    tmplName = "JEntityCmp2CoherenceItf.vm";
                }
                break;
            default:
                break;
        }
        if (tmplName == null) {
            throw new GenICException("No template for  '" + srcFileName + " !!!'");
        }

        try {
            tmpl = vEngine.getTemplate(tmplName);
        } catch (Exception e) {
            throw new GenICException("Cannot get the '" + tmplName + "' template file", e);
        }

        try {
            // Create before the parent directory
            File fs = new File(srcFileName);
            File fdir = fs.getParentFile();
            if (fdir != null) {
                if (!fdir.exists()) {
                    if (!fdir.mkdirs()) {
                        throw new IOException("Cannot create the directory '" + fdir.getPath() + "'");
                    }
                }
            }
            fwriter = new FileWriter(srcFileName);
        } catch (IOException e) {
            e.printStackTrace();
            throw new GenICException("Cannot create the '" + srcFileName + "' source file", e);
        }
        logger.log(BasicLevel.DEBUG, "Generate the file " + srcFileName);
        try {
            tmpl.merge(vctx, fwriter);
        } catch (Exception e) {
            throw new GenICException("Cannot generate the '" + srcFileName + "' source from the '" + tmplName
                    + "' template", e);
        }

        try {
            fwriter.flush();
            fwriter.close();
        } catch (IOException e) {
            throw new GenICException("Cannot close the '" + srcFileName + "' source file", e);
        }
    }

}

---