/*

 * Quoridor server and related libraries

 * Copyright (C) 2009-2010 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

 *

 * 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.

 *

 * 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program. Look for COPYING file in the top folder.

 * If not, see http://www.gnu.org/licenses/.

 */

package cz.xelfi.quoridor.freemarkerdor;

import com.sun.jersey.spi.template.TemplateProcessor;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.io.OutputStream;

import java.io.OutputStreamWriter;

import java.io.Writer;

import java.util.Map;

import javax.script.Bindings;

import javax.script.ScriptContext;

import javax.script.ScriptEngine;

import javax.script.ScriptEngineManager;

import javax.script.ScriptException;

import javax.ws.rs.ext.Provider;

import org.openide.filesystems.FileUtil;

import org.w3c.dom.Document;

/**

 *

 * @author Jaroslav Tulach <jtulach@netbeans.org>

 */

@Provider

public class FreemarkerProcessor implements TemplateProcessor {

    private ScriptEngineManager manager;

    public String resolve(String name) {

        return name;

    }

    public void writeTo(String fqn, Object model, OutputStream out) throws IOException {

        synchronized (this) {

            if (manager == null) {

                manager = new ScriptEngineManager();

            }

        }

        ScriptEngine eng = manager.getEngineByName("freemarker");

        if (eng == null) {

            throw new IOException("freemarker not found!");

        }

        Bindings bind = eng.getContext().getBindings(ScriptContext.ENGINE_SCOPE);

        if (model instanceof Map) {

            bind.putAll((Map<? extends String, ? extends Object>)model);

        }

        if (model instanceof Document) {

            bind.put("doc", model);

        } else {

            bind.put("model", model);

        }

        Writer w = new OutputStreamWriter(out, "UTF-8");

        eng.getContext().setWriter(w);

        eng.getContext().setAttribute(ScriptEngine.FILENAME, fqn, ScriptContext.ENGINE_SCOPE);

        // silly workaround for

        // http://openide.netbeans.org/issues/show_bug.cgi?id=170177

        eng.getContext().setAttribute("org.openide.filesystems.FileObject", FileUtil.createData(FileUtil.getConfigRoot(), fqn), ScriptContext.ENGINE_SCOPE);

        InputStream is = FreemarkerProcessor.class.getResourceAsStream(fqn);

        if (is == null) {

            throw new IOException("Not found " + fqn); // NOI18N

        }

        InputStreamReader r = new InputStreamReader(is, "UTF-8");

        try {

            eng.eval(r);

        } catch (ScriptException ex) {

            System.err.println("dump data: " + model);

            ex.printStackTrace();

            throw new IOException(ex);

        }

    }

}