/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * The contents of this file are subject to the terms of either the GNU

 * General Public License Version 2 only ("GPL") or the Common

 * Development and Distribution License("CDDL") (collectively, the

 * "License"). You may not use this file except in compliance with the

 * License. You can obtain a copy of the License at

 * http://www.netbeans.org/cddl-gplv2.html

 * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the

 * specific language governing permissions and limitations under the

 * License.  When distributing the software, include this License Header

 * Notice in each file and include the License file at

 * nbbuild/licenses/CDDL-GPL-2-CP.  Sun designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Sun in the GPL Version 2 section of the License file that

 * accompanied this code. If applicable, add the following below the

 * License Header, with the fields enclosed by brackets [] replaced by

 * your own identifying information:

 * "Portions Copyrighted [year] [name of copyright owner]"

 *

 * Contributor(s):

 *

 * Portions Copyrighted 2009 Jaroslav Tulach

 */

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);

        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);

        }

    }

}