Add page that always requires a logged-in user
[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / pages / UnlikePage.kt
index 42e552e..6bfabe2 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
@@ -11,13 +12,13 @@ import net.pterodactylus.util.template.TemplateContext
  * Page that lets the user unlike a [net.pterodactylus.sone.data.Post] or [net.pterodactylus.sone.data.Reply].
  */
 class UnlikePage(template: Template, webInterface: WebInterface):
-               SoneTemplatePage("unlike.html", template, "Page.Unlike.Title", webInterface, true) {
+               LoggedInPage("unlike.html", template, "Page.Unlike.Title", webInterface) {
 
-       override fun handleRequest(freenetRequest: FreenetRequest, templateContext: TemplateContext) {
+       override fun handleRequest(freenetRequest: FreenetRequest, currentSone: Sone, templateContext: TemplateContext) {
                if (freenetRequest.isPOST) {
                        when (freenetRequest.parameters["type"]) {
-                               "post" -> getCurrentSone(freenetRequest.toadletContext)!!.removeLikedPostId(freenetRequest.parameters["post"]!!)
-                               "reply" -> getCurrentSone(freenetRequest.toadletContext)!!.removeLikedReplyId(freenetRequest.parameters["reply"]!!)
+                               "post" -> currentSone.removeLikedPostId(freenetRequest.parameters["post"]!!)
+                               "reply" -> currentSone.removeLikedReplyId(freenetRequest.parameters["reply"]!!)
                        }
                        throw RedirectException(freenetRequest.parameters["returnPage", 256])
                }