package org.webslinger.ext.template.freemarker;

 import java.io.Writer;

 import java.io.IOException;

 import java.util.Map;

 import org.apache.bsf.BSFException;

 import org.webslinger.template.TemplateMacro;

 import org.webslinger.template.TemplateManager;

 import freemarker.core.Environment;

 import freemarker.template.TemplateModelException;

 import freemarker.template.TemplateTransformModel;

 public class FreemarkerTransformMacro implements TemplateTransformModel {

     private final FreemarkerTemplateEngine engine;

     private final TemplateMacro macro;;

     public FreemarkerTransformMacro(FreemarkerTemplateEngine engine, TemplateMacro macro) {

         this.engine = engine;

         this.macro = macro;

     }

     public FreemarkerTemplateEngine getEngine() {

         return engine;

     }

     public TemplateMacro getMacro() {

         return macro;

     }

     public String getName() {

         return getMacro().getName();

     }

     public Writer getWriter(final Writer out, final Map args) throws TemplateModelException, IOException {

         //final TemplateRequest request = (TemplateRequest) args.get(TemplateRequest.class.getName());

         //final TemplateManager manager = (TemplateManager) args.get(TemplateManager.class.getName());

         //final String templateName = Environment.getCurrentEnvironment().getTemplate().getName();

         final Object currentResource = null;

         return new Writer() {

             protected final StringBuilder buf = new StringBuilder();

             protected final TemplateMacro.Body body = new TemplateMacro.Body() {

                 public boolean render(Writer writer, Map<String, Object> context) throws IOException {

                     writer.write(buf.toString());

                     return true;

                 }

                 public Object getSource() {

                     return currentResource;

                 }

             };

             public void write(char[] buf, int offset, int length) throws IOException {

                 this.buf.append(buf, offset, length);

             }

             public void flush() throws IOException {

             }

             public void close() throws IOException {

                 try {

                     // FIXME: separate parameters/args

                     getMacro().render((TemplateManager) getEngine().getManager(), currentResource, out, new TemplateMacro.Args(null, args), args, body);

                     //manager.render(getMacro(), new FreemarkerTemplateNode(templateName, 0, 0, args, buf), out, new FreemarkerTemplateRequest(getEngine(), request));

                 } catch (BSFException e) {

                     throw (IOException) new IOException(e.getMessage()).initCause(e);

                 }

             }

         };

     }

 }