X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpage%2FTemplatePage.java;h=08e026d0d524fdd838250616e935afdc30c2a95c;hb=9e4db46b86d084eba9029906e779ec1d96f78ac4;hp=761e89b6b3b4162ffe99bf49e16ab085d2ffa68e;hpb=62603dd80cc0a5539d5032e4537c88b86c158085;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/web/page/TemplatePage.java b/src/main/java/net/pterodactylus/sone/web/page/TemplatePage.java
index 761e89b..08e026d 100644
--- a/src/main/java/net/pterodactylus/sone/web/page/TemplatePage.java
+++ b/src/main/java/net/pterodactylus/sone/web/page/TemplatePage.java
@@ -20,9 +20,14 @@ package net.pterodactylus.sone.web.page;
 import java.io.StringWriter;
 import java.util.Collection;
 import java.util.Collections;
+import java.util.logging.Level;
+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;
@@ -36,11 +41,17 @@ import freenet.l10n.BaseL10n;
  */
 public class TemplatePage implements Page, LinkEnabledCallback {
 
+	/** The logger. */
+	private static final Logger logger = Logging.getLogger(TemplatePage.class);
+
 	/** 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;
@@ -56,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
@@ -66,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;
@@ -110,16 +124,23 @@ public class TemplatePage implements Page, LinkEnabledCallback {
 			pageNode.addForwardLink("icon", shortcutIcon);
 		}
 
+		TemplateContext templateContext = templateContextFactory.createTemplateContext();
+		templateContext.mergeContext(template.getInitialContext());
 		try {
-			processTemplate(request, template);
+			long start = System.nanoTime();
+			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());
 		}
 
 		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());
 	}
 
@@ -148,12 +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, 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, DataProvider)}
+	 * throws a {@link RedirectException}!
+	 *
+	 * @param request
+	 *            The request being processed
+	 * @param templateContext
+	 *            The template context that supplied the rendered data
+	 */
+	protected void postProcess(Request request, TemplateContext templateContext) {
 		/* do nothing. */
 	}
 
@@ -184,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>