1 package net.pterodactylus.sone.template
3 import net.pterodactylus.sone.core.LinkedElement
4 import net.pterodactylus.util.template.Filter
5 import net.pterodactylus.util.template.TemplateContext
6 import net.pterodactylus.util.template.TemplateContextFactory
7 import net.pterodactylus.util.template.TemplateParser
8 import java.io.StringReader
9 import java.io.StringWriter
10 import javax.inject.Inject
13 * Renders all kinds of [LinkedElement]s.
15 class LinkedElementRenderFilter @Inject constructor(private val templateContextFactory: TemplateContextFactory): Filter {
18 private val loadedImageTemplate = """<%include linked/image.html>""".parse()
19 private val loadedHtmlPageTemplate = """<%include linked/html-page.html>""".parse()
20 private val notLoadedImageTemplate = """<%include linked/notLoaded.html>""".parse()
22 private fun String.parse() = StringReader(this).use { TemplateParser.parse(it) }!!
25 override fun format(templateContext: TemplateContext?, data: Any?, parameters: Map<String, Any?>?) =
27 data is LinkedElement && data.loading -> renderNotLoadedLinkedElement(data)
28 data is LinkedElement && data.properties["type"] == "image" -> renderLinkedImage(data)
29 data is LinkedElement && data.properties["type"] == "html" -> renderHtmlPage(data)
33 private fun renderLinkedImage(linkedElement: LinkedElement) =
35 val templateContext = templateContextFactory.createTemplateContext()
36 templateContext["link"] = linkedElement.link
37 it.also { loadedImageTemplate.render(templateContext, it) }
40 private fun renderHtmlPage(linkedElement: LinkedElement) =
42 val templateContext = templateContextFactory.createTemplateContext()
43 templateContext["link"] = linkedElement.link
44 templateContext["title"] = linkedElement.properties["title"] ?: "No title"
45 templateContext["description"] = linkedElement.properties["description"] ?: "No description"
46 it.also { loadedHtmlPageTemplate.render(templateContext, it) }
49 private fun renderNotLoadedLinkedElement(linkedElement: LinkedElement) =
51 val templateContext = templateContextFactory.createTemplateContext()
52 templateContext["link"] = linkedElement.link
53 it.also { notLoadedImageTemplate.render(templateContext, it) }