net.rumati.maven.plugins.velocity.VelocityMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of velocity-maven-plugin Show documentation
A Maven plugin that uses Apache Velocity to process templates, populating variables from the Maven POM.
There is a newer version: 0.3.1
Show newest version
package net.rumati.maven.plugins.velocity;

import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.Reader;
import java.io.Writer;
import java.nio.charset.Charset;
import java.util.Map;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;
import org.apache.velocity.runtime.RuntimeServices;
import org.apache.velocity.runtime.log.LogChute;

/**
 * Processes a Velocity template
 *
 * @goal velocity
 * @phase process-resources
 */
public class VelocityMojo
    extends AbstractMojo
{
    /**
     * Template file
     * @parameter
     * @required
     */
    private File template;

    /**
     * Output file
     * @parameter
     * @required
     */
    private File outputFile;

    /**
     * Properties to populate in the template
     * @parameter
     * @required
     */
    private Map properties;

    public void execute()
        throws MojoExecutionException
    {
        File parentDirectory = outputFile.getParentFile();
        if (!parentDirectory.isDirectory() && !parentDirectory.mkdirs()){
            throw new MojoExecutionException("Error creating output directory: " + parentDirectory.getAbsolutePath());
        }
        
        try {
            Reader reader = new InputStreamReader(new FileInputStream(template), Charset.forName("UTF-8"));
            try {
                Writer writer = new OutputStreamWriter(new FileOutputStream(outputFile), Charset.forName("UTF-8"));
                try {
                    VelocityEngine engine = new VelocityEngine();
                    engine.setProperty(VelocityEngine.RUNTIME_LOG_LOGSYSTEM, new LogChute() {

                        public void init(RuntimeServices rs)
                                throws Exception
                        {
                            /* do nothing */
                        }

                        public void log(int i, String string)
                        {
                            switch (i){
                                case LogChute.INFO_ID:
                                    VelocityMojo.this.getLog().info(string);
                                    break;
                                case LogChute.WARN_ID:
                                    VelocityMojo.this.getLog().warn(string);
                                    break;
                                case LogChute.ERROR_ID:
                                    VelocityMojo.this.getLog().error(string);
                                    break;
                            }
                        }

                        public void log(int i, String string, Throwable thrwbl)
                        {
                            switch (i){
                                case LogChute.INFO_ID:
                                    VelocityMojo.this.getLog().info(string, thrwbl);
                                    break;
                                case LogChute.WARN_ID:
                                    VelocityMojo.this.getLog().warn(string, thrwbl);
                                    break;
                                case LogChute.ERROR_ID:
                                    VelocityMojo.this.getLog().error(string, thrwbl);
                                    break;
                            }
                        }

                        public boolean isLevelEnabled(int i)
                        {
                            return i > LogChute.DEBUG_ID;
                        }
                    });
                    engine.init();
                    VelocityContext ctx = new VelocityContext();
                    for (Map.Entry e : properties.entrySet()){
                        ctx.put(e.getKey().toString(), e.getValue());
                    }
                    engine.evaluate(ctx, writer, "velocity-maven-plugin", reader);
                }finally{
                    writer.close();
                }
            }finally{
                reader.close();
            }
        }catch (Exception e){
            throw new MojoExecutionException("Error processing template", e);
        }
    }
}