org.krakenapps.docxcod.FreeMarkerRunner.java Source code

Introduction

Here is the source code for org.krakenapps.docxcod.FreeMarkerRunner.java

Source

/*
 * Copyright 2013 Future Systems
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.krakenapps.docxcod;

import java.io.Closeable;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import org.apache.commons.io.FilenameUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;
import freemarker.template.TemplateException;

public class FreeMarkerRunner implements OOXMLProcessor {
    private Logger logger = LoggerFactory.getLogger(getClass().getName());
    private Set<String> targets = new HashSet<String>();

    public FreeMarkerRunner(String string) {
        targets.add(FilenameUtils.normalize(string));
    }

    @Override
    public void process(OOXMLPackage docx, Map<String, Object> rootMap) {
        // TODO Auto-generated method stub
        Configuration cfg = new Configuration();
        cfg.setObjectWrapper(new DefaultObjectWrapper());

        for (String s : docx.listParts("")) {

            if (!targets.isEmpty() && !targets.contains(s))
                continue;

            InputStreamReader templateReader = null;

            PrintWriter writer = null;
            File sf = new File(docx.getDataDir(), s);
            File outf = new File(sf.getPath() + ".new");
            try {
                templateReader = new InputStreamReader(new FileInputStream(sf));
                writer = new PrintWriter(new FileOutputStream(outf));
                logger.info("process: try freemarker template processing: {}", sf);
                Template t = new Template(s, templateReader, cfg);
                t.process(rootMap, writer);
                logger.trace("process: freemarker template processing completed");
                safeClose(writer);
                safeClose(templateReader);
                writer = null;
                templateReader = null;

                logger.trace("process: rename {} to {}", outf, sf);
                boolean deleteResult = sf.delete();
                if (deleteResult) {
                    logger.trace("process: deleting old file success: {}", sf);
                    outf.renameTo(sf);
                } else {
                    logger.error("process: deleting old file failed: {}", sf);
                }
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            } catch (TemplateException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            } finally {
                safeClose(templateReader);
                safeClose(writer);
            }
        }
    }

    private void safeClose(Closeable o) {
        try {
            if (o != null)
                o.close();
        } catch (IOException e) {
            // ignore
        }
    }
}