Add test for DI constructability of ViewPostPage

[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / pages / ViewPostPage.kt

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/pages/ViewPostPage.kt b/src/main/kotlin/net/pterodactylus/sone/web/pages/ViewPostPage.kt
index d2e163d..763fc99 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/pages/ViewPostPage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/ViewPostPage.kt
@@ -8,27 +8,28 @@ import net.pterodactylus.sone.web.page.FreenetRequest
 import net.pterodactylus.util.template.Template
 import net.pterodactylus.util.template.TemplateContext
 import java.net.URI
+import javax.inject.Inject
 
 /**
  * This page lets the user view a post and all its replies.
  */
-class ViewPostPage(template: Template, webInterface: WebInterface):
-               SoneTemplatePage("viewPost.html", template, "Page.ViewPost.Title", webInterface, false) {
+class ViewPostPage @Inject constructor(template: Template, webInterface: WebInterface):
+               SoneTemplatePage("viewPost.html", webInterface, template, "Page.ViewPost.Title") {
 
-       override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) {
-               templateContext["post"] = request.parameters["post"].let(webInterface.core::getPost).orNull()
-               templateContext["raw"] = request.parameters["raw"] == "true"
+       override fun handleRequest(freenetRequest: FreenetRequest, templateContext: TemplateContext) {
+               templateContext["post"] = freenetRequest.parameters["post"]?.let(webInterface.core::getPost)
+               templateContext["raw"] = freenetRequest.parameters["raw"] == "true"
        }
 
        override fun isLinkExcepted(link: URI?) = true
 
-       public override fun getPageTitle(request: FreenetRequest) =
-                       (request.parameters["post"].let(webInterface.core::getPost).let {
+       public override fun getPageTitle(freenetRequest: FreenetRequest) =
+                       (freenetRequest.parameters["post"]?.let(webInterface.core::getPost)?.let {
                                if (it.text.length > 20) {
                                        it.text.substring(0..19) + "…"
                                } else {
                                        it.text
                                } + " - ${SoneAccessor.getNiceName(it.sone)} - "
-                       } ?: "") + super.getPageTitle(request)
+                       } ?: "") + super.getPageTitle(freenetRequest)
 
 }