JVelocity.java

package org.objectweb.jonas_ws.wsgen.generator.axis;

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.apache.velocity.runtime.RuntimeConstants;

import org.objectweb.jonas_lib.I18n;

import org.objectweb.jonas_ws.wsgen.WsGenException;

import org.objectweb.jonas.common.Log;

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

public class JVelocity {

    private static I18n i18n = I18n.getInstance(JVelocity.class);

    private static Logger logger =  Log.getLogger(Log.JONAS_WSGEN_PREFIX);

    private VelocityEngine vEngine;

    private Template template;

    public JVelocity(String tmplName) throws WsGenException {
        // Prepare Velocity Engine
        String jonasRoot = System.getProperty("install.root");

        if (jonasRoot == null) {
            String err = i18n.getMessage("JVelocity.constr.notset");
            throw new WsGenException(err);
        }

        String path2Tmpl = new String(jonasRoot + File.separatorChar + "templates" + File.separatorChar + "wsgen"
                + File.separatorChar + "generator" + File.separatorChar + "axis");

        // instanciate the engine
        vEngine = new VelocityEngine();
        vEngine.setProperty(RuntimeConstants.VM_LIBRARY, "");
        vEngine.setProperty(RuntimeConstants.RESOURCE_LOADER, "file");
        vEngine.setProperty(RuntimeConstants.FILE_RESOURCE_LOADER_PATH, path2Tmpl);

        try {
            vEngine.init();
        } catch (Exception e) {
            String err = i18n.getMessage("JVelocity.constr.initFailure");
            throw new WsGenException(err, e);
        }

        // Create the Template
        try {
            template = vEngine.getTemplate(tmplName);
        } catch (Exception e) {
            String err = i18n.getMessage("JVelocity.constr.tmplError", tmplName);
            throw new WsGenException(err, e);
        }
    }

    public void generate(File fs, VelocityContext context) throws WsGenException {
        FileWriter fwriter = null;

        try {
            // Create before the parent directory
            File fdir = fs.getParentFile();

            if (fdir != null) {
                if (!fdir.exists()) {
                    if (!fdir.mkdirs()) {
                        String err = i18n.getMessage("JVelocity.generate.directories", fdir.getPath());
                        throw new WsGenException(err);
                    }
                }
            }

            fwriter = new FileWriter(fs);
        } catch (IOException e) {
            String err = i18n.getMessage("JVelocity.generate.file", fs);
            throw new WsGenException(err, e);
        }

        try {
            template.merge(context, fwriter);
        } catch (Exception e) {
            String err = i18n.getMessage("JVelocity.generate.cannot", fs);
            throw new WsGenException(err, e);
        }

        try {
            fwriter.flush();
            fwriter.close();
        } catch (IOException e) {
            // do nothing, just a warn
            String err = i18n.getMessage("JVelocity.generate.close", fs);
            logger.log(BasicLevel.WARN, err);
        }
    }
}

---