Add page that always requires a logged-in user
[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / pages / LikePage.kt
index 9f6191e..7d6af18 100644 (file)
@@ -1,5 +1,6 @@
 package net.pterodactylus.sone.web.pages
 
+import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.utils.isPOST
 import net.pterodactylus.sone.utils.parameters
 import net.pterodactylus.sone.web.WebInterface
@@ -10,20 +11,18 @@ import net.pterodactylus.util.template.TemplateContext
 /**
  * Page that lets the user like [net.pterodactylus.sone.data.Post]s and [net.pterodactylus.sone.data.Reply]s.
  */
-class LikePage(template: Template, webInterface: WebInterface)
-       : SoneTemplatePage("like.html", template, "Page.Like.Title", webInterface, true) {
+class LikePage(template: Template, webInterface: WebInterface) :
+       LoggedInPage("like.html", template, "Page.Like.Title", webInterface) {
 
-       override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) {
-               if (request.isPOST) {
-                       getCurrentSone(request.toadletContext)!!.let { currentSone ->
-                               request.parameters["type", 16]?.also { type ->
+       override fun handleRequest(freenetRequest: FreenetRequest, currentSone: Sone, templateContext: TemplateContext) {
+               if (freenetRequest.isPOST) {
+                               freenetRequest.parameters["type", 16]?.also { type ->
                                        when(type) {
-                                               "post" -> currentSone.addLikedPostId(request.parameters["post", 36]!!)
-                                               "reply" -> currentSone.addLikedReplyId(request.parameters["reply", 36]!!)
+                                               "post" -> currentSone.addLikedPostId(freenetRequest.parameters["post", 36]!!)
+                                               "reply" -> currentSone.addLikedReplyId(freenetRequest.parameters["reply", 36]!!)
                                        }
                                }
-                               throw RedirectException(request.parameters["returnPage", 256]!!)
-                       }
+                               throw RedirectException(freenetRequest.parameters["returnPage", 256]!!)
                }
        }