Show loading animation while loading elements

[Sone.git] / src / main / kotlin / net / pterodactylus / sone / template / LinkedElementsFilter.kt

diff --git a/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementsFilter.kt b/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementsFilter.kt
new file mode 100644 (file)
index 0000000..f95d6c0
--- /dev/null
+++ b/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementsFilter.kt
@@ -0,0 +1,23 @@
+package net.pterodactylus.sone.template
+
+import net.pterodactylus.sone.core.ElementLoader
+import net.pterodactylus.sone.core.LinkedElement
+import net.pterodactylus.sone.text.FreenetLinkPart
+import net.pterodactylus.sone.text.Part
+import net.pterodactylus.util.template.Filter
+import net.pterodactylus.util.template.TemplateContext
+
+/**
+ * Filter that takes a number of pre-rendered [Part]s and replaces all identified links to freenet elements
+ * with [LinkedElement]s.
+ */
+class LinkedElementsFilter(private val elementLoader: ElementLoader) : Filter {
+
+       @Suppress("UNCHECKED_CAST")
+       override fun format(templateContext: TemplateContext?, data: Any?, parameters: MutableMap<String, Any?>?) =
+                       (data as? Iterable<Part>)
+                                       ?.filterIsInstance<FreenetLinkPart>()
+                                       ?.map { elementLoader.loadElement(it.link) }
+                                       ?: listOf<LinkedElement>()
+
+}