de.thomasvolk.genexample.bericht.templates.Template.java Source code

Introduction

Here is the source code for de.thomasvolk.genexample.bericht.templates.Template.java

Source

/**
 * Copyright (C) 2015 Thomas Volk
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package de.thomasvolk.genexample.bericht.templates;

import groovy.lang.Writable;
import groovy.text.SimpleTemplateEngine;
import org.apache.commons.io.FilenameUtils;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Writer;
import java.util.HashMap;
import java.util.Map;

public class Template {
    private final String name;
    private final String dir;
    private final String extension;
    private final String path;

    public Template(String path) {
        this.name = FilenameUtils.getBaseName(path);
        this.dir = FilenameUtils.getPath(path);
        this.extension = FilenameUtils.getExtension(path);
        this.path = path;
    }

    public String getName() {
        return name;
    }

    public String getExtension() {
        return extension;
    }

    protected String getTargetPath(String zielPfad, String name) {
        return FilenameUtils.concat(zielPfad, name);
    }

    public void generiere(String zielPfad) {
        generiere(zielPfad, (Object) null);
    }

    public void generiere(String zielPfad, Object ctx) {
        generiere(zielPfad, ctx, getName());
    }

    public void generiere(String zielPfad, Object ctx, String newName) {
        try (FileWriter fileWriter = new FileWriter(getTargetPath(zielPfad, newName + "." + getExtension()))) {
            Map<String, Object> binding = new HashMap<>();
            binding.put("ctx", ctx);
            binding.put("templateName", newName);
            generiere(fileWriter, binding);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    public void generiere(Writer out, Map<String, Object> binding) {
        String tpl = "de/thomasvolk/genexample/bericht/templates/report/" + path;
        try (InputStreamReader reader = new InputStreamReader(
                Template.class.getClassLoader().getResourceAsStream(tpl))) {
            SimpleTemplateEngine templateEngine = new SimpleTemplateEngine();
            groovy.text.Template template = templateEngine.createTemplate(reader);
            Writable writable = template.make(binding);
            writable.writeTo(out);
        } catch (Exception e) {
            throw new RuntimeException(tpl, e);
        }
    }

    public String getDir() {
        return dir;
    }
}