Use new template engine.
[Sone.git] / src / main / java / net / pterodactylus / sone / web / page / TemplatePage.java
index 7dfd94b..08e026d 100644 (file)
@@ -26,6 +26,8 @@ import java.util.logging.Logger;
 import net.pterodactylus.sone.web.page.Page.Request.Method;
 import net.pterodactylus.util.logging.Logging;
 import net.pterodactylus.util.template.Template;
+import net.pterodactylus.util.template.TemplateContext;
+import net.pterodactylus.util.template.TemplateContextFactory;
 import freenet.clients.http.LinkEnabledCallback;
 import freenet.clients.http.PageMaker;
 import freenet.clients.http.PageNode;
@@ -45,8 +47,11 @@ public class TemplatePage implements Page, LinkEnabledCallback {
        /** The path of the page. */
        private final String path;
 
+       /** The template context factory. */
+       private final TemplateContextFactory templateContextFactory;
+
        /** The template to render. */
-       protected final Template template;
+       private final Template template;
 
        /** The L10n handler. */
        private final BaseL10n l10n;
@@ -62,6 +67,8 @@ public class TemplatePage implements Page, LinkEnabledCallback {
         *
         * @param path
         *            The path of the page
+        * @param templateContextFactory
+        *            The template context factory
         * @param template
         *            The template to render
         * @param l10n
@@ -72,8 +79,9 @@ public class TemplatePage implements Page, LinkEnabledCallback {
         *            The target to redirect to if a POST request does not contain
         *            the correct form password
         */
-       public TemplatePage(String path, Template template, BaseL10n l10n, String pageTitleKey, String invalidFormPasswordRedirectTarget) {
+       public TemplatePage(String path, TemplateContextFactory templateContextFactory, Template template, BaseL10n l10n, String pageTitleKey, String invalidFormPasswordRedirectTarget) {
                this.path = path;
+               this.templateContextFactory = templateContextFactory;
                this.template = template;
                this.l10n = l10n;
                this.pageTitleKey = pageTitleKey;
@@ -116,21 +124,23 @@ public class TemplatePage implements Page, LinkEnabledCallback {
                        pageNode.addForwardLink("icon", shortcutIcon);
                }
 
+               TemplateContext templateContext = templateContextFactory.createTemplateContext();
+               templateContext.mergeContext(template.getInitialContext());
                try {
                        long start = System.nanoTime();
-                       processTemplate(request, template);
+                       processTemplate(request, templateContext);
                        long finish = System.nanoTime();
                        logger.log(Level.FINEST, "Template was rendered in " + ((finish - start) / 1000) / 1000.0 + "ms.");
                } catch (RedirectException re1) {
                        return new RedirectResponse(re1.getTarget());
                }
 
-               postProcess(request, template);
-
                StringWriter stringWriter = new StringWriter();
-               template.render(stringWriter);
+               template.render(templateContext, stringWriter);
                pageNode.content.addChild("%", stringWriter.toString());
 
+               postProcess(request, templateContext);
+
                return new Response(200, "OK", "text/html", pageNode.outer.generate());
        }
 
@@ -159,29 +169,29 @@ public class TemplatePage implements Page, LinkEnabledCallback {
         *
         * @param request
         *            The request that is rendered
-        * @param template
-        *            The template to set variables in
+        * @param templateContext
+        *            The template context to set variables in
         * @throws RedirectException
         *             if the processing page wants to redirect after processing
         */
-       protected void processTemplate(Request request, Template template) throws RedirectException {
+       protected void processTemplate(Request request, TemplateContext templateContext) throws RedirectException {
                /* do nothing. */
        }
 
        /**
         * This method will be called after
-        * {@link #processTemplate(net.pterodactylus.sone.web.page.Page.Request, Template)}
+        * {@link #processTemplate(net.pterodactylus.sone.web.page.Page.Request, DataProvider)}
         * has processed the template and the template was rendered. This method
         * will not be called if
-        * {@link #processTemplate(net.pterodactylus.sone.web.page.Page.Request, Template)}
+        * {@link #processTemplate(net.pterodactylus.sone.web.page.Page.Request, DataProvider)}
         * throws a {@link RedirectException}!
         *
         * @param request
         *            The request being processed
-        * @param template
-        *            The template that was rendered
+        * @param templateContext
+        *            The template context that supplied the rendered data
         */
-       protected void postProcess(Request request, Template template) {
+       protected void postProcess(Request request, TemplateContext templateContext) {
                /* do nothing. */
        }
 
@@ -212,7 +222,7 @@ public class TemplatePage implements Page, LinkEnabledCallback {
        /**
         * Exception that can be thrown to signal that a subclassed {@link Page}
         * wants to redirect the user during the
-        * {@link TemplatePage#processTemplate(net.pterodactylus.sone.web.page.Page.Request, Template)}
+        * {@link TemplatePage#processTemplate(net.pterodactylus.sone.web.page.Page.Request, TemplateContext)}
         * method call.
         *
         * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>