Simplify some code

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

diff --git a/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt b/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt
index e15f68b..a527a1d 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilter.kt
@@ -34,8 +34,7 @@ class LinkedElementRenderFilter @Inject constructor(private val templateContextF
                        StringWriter().use {
                                val templateContext = templateContextFactory.createTemplateContext()
                                templateContext["link"] = linkedElement.link
-                               loadedImageTemplate.render(templateContext, it)
-                               it
+                               it.also { loadedImageTemplate.render(templateContext, it) }
                        }.toString()
 
        private fun renderHtmlPage(linkedElement: LinkedElement) =
@@ -51,8 +50,7 @@ class LinkedElementRenderFilter @Inject constructor(private val templateContextF
                        StringWriter().use {
                                val templateContext = templateContextFactory.createTemplateContext()
                                templateContext["link"] = linkedElement.link
-                               notLoadedImageTemplate.render(templateContext, it)
-                               it
+                               it.also { notLoadedImageTemplate.render(templateContext, it) }
                        }.toString()
 
 }