From: David ‘Bombe’ Roden <bombe@pterodactylus.net>
Date: Sat, 16 Oct 2010 14:08:48 +0000 (+0200)
Subject: Add classpath-based template provider for template inclusions.
X-Git-Tag: 0.1-RC1~271
X-Git-Url: https://git.pterodactylus.net/?a=commitdiff_plain;h=54e3312d42e82c36da964874b03bf69b9068359c;p=Sone.git

Add classpath-based template provider for template inclusions.
---

diff --git a/src/main/java/net/pterodactylus/sone/web/WebInterface.java b/src/main/java/net/pterodactylus/sone/web/WebInterface.java
index 83a942a..cbd8b42 100644
--- a/src/main/java/net/pterodactylus/sone/web/WebInterface.java
+++ b/src/main/java/net/pterodactylus/sone/web/WebInterface.java
@@ -44,6 +44,9 @@ import net.pterodactylus.util.template.DateFilter;
 import net.pterodactylus.util.template.DefaultTemplateFactory;
 import net.pterodactylus.util.template.ReflectionAccessor;
 import net.pterodactylus.util.template.Template;
+import net.pterodactylus.util.template.TemplateException;
+import net.pterodactylus.util.template.TemplateFactory;
+import net.pterodactylus.util.template.TemplateProvider;
 import freenet.clients.http.LinkEnabledCallback;
 import freenet.clients.http.SessionManager;
 import freenet.clients.http.ToadletContainer;
@@ -148,6 +151,7 @@ public class WebInterface extends AbstractService {
 		templateFactory.addAccessor(Post.class, new PostAccessor(core()));
 		templateFactory.addFilter("date", new DateFilter());
 		templateFactory.addFilter("l10n", new L10nFilter(l10n()));
+		templateFactory.setTemplateProvider(new ClassPathTemplateProvider(templateFactory));
 
 		String formPassword = sonePlugin.pluginRespirator().getToadletContainer().getFormPassword();
 
@@ -266,4 +270,47 @@ public class WebInterface extends AbstractService {
 		}
 	}
 
+	/**
+	 * Template provider implementation that uses
+	 * {@link WebInterface#createReader(String)} to load templates for
+	 * inclusion.
+	 *
+	 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+	 */
+	private class ClassPathTemplateProvider implements TemplateProvider {
+
+		/** The template factory. */
+		private final TemplateFactory templateFactory;
+
+		/**
+		 * Creates a new template provider that locates templates on the
+		 * classpath.
+		 *
+		 * @param templateFactory
+		 *            The template factory to create the templates
+		 */
+		public ClassPathTemplateProvider(TemplateFactory templateFactory) {
+			this.templateFactory = templateFactory;
+		}
+
+		/**
+		 * {@inheritDoc}
+		 */
+		@Override
+		@SuppressWarnings("synthetic-access")
+		public Template getTemplate(String templateName) {
+			Reader templateReader = createReader("/templates/" + templateName);
+			if (templateReader == null) {
+				return null;
+			}
+			Template template = templateFactory.createTemplate(templateReader);
+			try {
+				template.parse();
+			} catch (TemplateException te1) {
+				logger.log(Level.WARNING, "Could not parse template “" + templateName + "” for inclusion!", te1);
+			}
+			return template;
+		}
+
+	}
 }