X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FLinkedElementRenderFilter.kt;h=6644ad8c28ac6854c93c97fe506d2db59dc8d06c;hp=6255f9dbf2ed7f24b0d4f7e20b46361786020ed8;hb=b91ce252d94a8876097b939e129dc33264cef2f5;hpb=1cd71a1c2a94f889d63d6ec3168d76da0a8b116b

diff --git a/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt b/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt
index 6255f9d..6644ad8 100644
--- a/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt
@@ -1,7 +1,6 @@
 package net.pterodactylus.sone.template
 
 import net.pterodactylus.sone.core.LinkedElement
-import net.pterodactylus.sone.core.LinkedImage
 import net.pterodactylus.util.template.Filter
 import net.pterodactylus.util.template.TemplateContext
 import net.pterodactylus.util.template.TemplateContextFactory
@@ -24,14 +23,14 @@ class LinkedElementRenderFilter(private val templateContextFactory: TemplateCont
 	override fun format(templateContext: TemplateContext?, data: Any?, parameters: Map<String, Any?>?) =
 			when {
 				data is LinkedElement && data.loading -> renderNotLoadedLinkedElement(data)
-				data is LinkedImage -> renderLinkedImage(data)
+				data is LinkedElement -> renderLinkedImage(data)
 				else -> null
 			}
 
-	private fun renderLinkedImage(linkedImage: LinkedImage) =
+	private fun renderLinkedImage(linkedElement: LinkedElement) =
 			StringWriter().use {
 				val templateContext = templateContextFactory.createTemplateContext()
-				templateContext["link"] = linkedImage.link
+				templateContext["link"] = linkedElement.link
 				loadedImageTemplate.render(templateContext, it)
 				it
 			}.toString()