org.tequila.template.directive.freemarker.FileSection.java Source code

Introduction

Here is the source code for org.tequila.template.directive.freemarker.FileSection.java

Source

/**
 *  Copyright (c) 2007-2008 by Carlos Gmez Montiel <iberck@gmail.com>
 *
 *  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.
 *
 *  his 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 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 org.tequila.template.directive.freemarker;

import freemarker.core.Environment;
import freemarker.template.SimpleScalar;
import freemarker.template.TemplateDirectiveBody;
import freemarker.template.TemplateDirectiveModel;
import freemarker.template.TemplateException;
import freemarker.template.TemplateModel;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.StringWriter;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Observable;
import java.util.Set;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.tequila.template.directive.DirectiveException;
import org.tequila.template.directive.TemplateDirective;
import org.tequila.template.match.TemplateProcessed;
import org.tequila.template.match.TemplatesWriterPool;
import org.tequila.conf.SpringUtils;

/**
 *
 * @author iberck
 */
public class FileSection extends Observable implements TemplateDirectiveModel, TemplateDirective {

    private static final Log log = LogFactory.getLog(FileSection.class);
    private static final String PATH = "path";
    private static final String NAME = "name";
    private static final String APPEND_BEFORE_LAST = "appendBeforeLast";

    public FileSection() {
        TemplatesWriterPool pool = (TemplatesWriterPool) SpringUtils.getBean("templatesWriterPool");
        addObserver(pool);
    }

    @Override
    public void execute(Environment env, Map params, TemplateModel[] loopVars, TemplateDirectiveBody body) {
        log.debug("Executing <@" + getDirectiveName() + " .../>");

        // Check if params were given
        if (params.isEmpty()) {
            throw new DirectiveException(
                    "<@" + getDirectiveName() + " .../> 'path, name' are required parameters.");
        }

        if (loopVars.length != 0) {
            throw new DirectiveException("<@" + getDirectiveName() + " .../> doesn't allow loop variables.");
        }

        // If there is non-empty nested content:
        if (body != null) {
            try {
                // Executes the nested body. Same as <#nested> in FTL, except
                // that we use our own writer instead of the current output writer.
                String path = null;
                String name = null;
                String appendBeforeLast = null;
                Set<Entry<String, SimpleScalar>> entrySet = params.entrySet();
                for (Entry<String, SimpleScalar> entry : entrySet) {
                    String key = entry.getKey();
                    SimpleScalar value = entry.getValue();
                    if (key.equals(PATH)) {
                        path = value.getAsString();
                    } else if (key.equals(NAME)) {
                        name = value.getAsString();
                    } else if (key.equals(APPEND_BEFORE_LAST)) {
                        appendBeforeLast = value.getAsString();
                    } else {
                        throw new DirectiveException(
                                "<@" + getDirectiveName() + " .../> doesn't allow " + key + " parameter");
                    }
                }

                // write the file
                StringWriter sw = new StringWriter();
                BufferedWriter bw = new BufferedWriter(sw);
                body.render(bw);
                env.getOut().flush();
                bw.flush();
                sw.flush();
                bw.close();
                sw.close();
                TemplateProcessed tp = new TemplateProcessed();
                tp.setTemplateResult(sw.toString());
                tp.setOutputFolder(path);
                tp.setOutputFileName(name);

                // notificar a los observadores
                super.setChanged();
                super.notifyObservers(tp);

            } catch (TemplateException ex) {
                throw new DirectiveException("Error al ejecutar la directiva '" + getDirectiveName() + "'", ex);
            } catch (IOException ex) {
                throw new DirectiveException("Error al ejecutar la directiva '" + getDirectiveName() + "'", ex);
            }

        } else {
            throw new DirectiveException("missing body");
        }
    }

    @Override
    public String getDirectiveName() {
        return "FileSection";
    }
}