X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FImageLinkFilter.java;h=fc4b803ffcfb354be9ec1b839fbedd2fe3ac4ed2;hp=d230f24cbc10515369dcc71082f97baa987659db;hb=6940f7d3df4f6164bfc69a9393a0f5796627b880;hpb=047787883cea2957540331b4720d9a20167eb042

diff --git a/src/main/java/net/pterodactylus/sone/template/ImageLinkFilter.java b/src/main/java/net/pterodactylus/sone/template/ImageLinkFilter.java
index d230f24..fc4b803 100644
--- a/src/main/java/net/pterodactylus/sone/template/ImageLinkFilter.java
+++ b/src/main/java/net/pterodactylus/sone/template/ImageLinkFilter.java
@@ -28,6 +28,7 @@ import java.util.Map;
 import net.pterodactylus.sone.core.Core;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.util.template.Filter;
+import net.pterodactylus.util.template.HtmlFilter;
 import net.pterodactylus.util.template.Template;
 import net.pterodactylus.util.template.TemplateContext;
 import net.pterodactylus.util.template.TemplateContextFactory;
@@ -49,19 +50,12 @@ public class ImageLinkFilter implements Filter {
 	private final Core core;
 
 	/** The template context factory. */
-	private final TemplateContextFactory templateContextFactory;
+	private final TemplateContextFactory templateContextFactory = new TemplateContextFactory();
 
-	/**
-	 * Creates a new image link filter.
-	 *
-	 * @param core
-	 *            The core
-	 * @param templateContextFactory
-	 *            The template context factory
-	 */
-	public ImageLinkFilter(Core core, TemplateContextFactory templateContextFactory) {
+	public ImageLinkFilter(Core core) {
 		this.core = core;
-		this.templateContextFactory = templateContextFactory;
+		templateContextFactory.addFilter("html", new HtmlFilter());
+		templateContextFactory.addFilter("css", new CssClassNameFilter());
 	}
 
 	/**