com.github.sviperll.mavem.plugin.mustache.RenderMustacheMojo Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 Victor Nazarov .
 */
package com.github.sviperll.mavem.plugin.mustache;

import com.github.mustachejava.DefaultMustacheFactory;
import com.github.mustachejava.Mustache;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.nio.charset.Charset;
import java.text.MessageFormat;
import java.util.Map;
import java.util.Properties;
import java.util.TreeMap;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;

/**
 * Goal which writes properties to file.
 *
 * @author Victor Nazarov 
 */
@Mojo(name = "render", defaultPhase = LifecyclePhase.GENERATE_RESOURCES)
public class RenderMustacheMojo extends ConfiguredMustacheMojo {

    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        try {
            Charset charset = Charset.forName(encoding);
            MustacheLoader mustacheReader = new MustacheLoader(new DefaultMustacheFactory(), getLog());
            if (contexts != null) {
                for (Context contextConfig: contexts) {
                    Object contextObject = loadContext(contextConfig);
                    if (contextConfig.templates != null) {
                        for (Template template: contextConfig.templates) {
                            Mustache mustache = mustacheReader.load(template.inputFile, charset);
                            renderMustache(mustache, contextObject, template.outputFile, charset);
                        }
                    }
                }
            }
        } catch (IOException ex) {
            throw new MojoExecutionException("Error writing mustache", ex);
        }
    }

    private Object loadContext(Context context) throws MojoFailureException, IOException {
        Charset charset = Charset.forName(encoding);
        if (context.type.equals("properties"))
            return loadPropertiesContext(context.file, charset);
        else if (context.type.equals("json"))
            return loadJsonContext(context.file, charset);
        else
            throw new MojoFailureException(MessageFormat.format("Unsupported context type: {0}, use one of json, properties",
                                                                context.type));
    }
}