Add page that always requires a logged-in user

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

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/pages/UnfollowSonePage.kt b/src/main/kotlin/net/pterodactylus/sone/web/pages/UnfollowSonePage.kt
index 3b299d2..ef4eab9 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/pages/UnfollowSonePage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/UnfollowSonePage.kt
@@ -1,26 +1,24 @@
 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
 import net.pterodactylus.sone.web.page.FreenetRequest
 import net.pterodactylus.util.template.Template
 import net.pterodactylus.util.template.TemplateContext
-import java.util.regex.Pattern
 
 /**
  * This page lets the user unfollow another Sone.
  */
 class UnfollowSonePage(template: Template, webInterface: WebInterface):
-               SoneTemplatePage("unfollowSone.html", template, "Page.UnfollowSone.Title", webInterface, true) {
+               LoggedInPage("unfollowSone.html", template, "Page.UnfollowSone.Title", webInterface) {
 
-       override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) {
-               if (request.isPOST) {
-                       getCurrentSone(request.toadletContext)!!.also { currentSone ->
-                               request.parameters["sone"]!!.split(Regex("[ ,]+"))
+       override fun handleRequest(freenetRequest: FreenetRequest, currentSone: Sone, templateContext: TemplateContext) {
+               if (freenetRequest.isPOST) {
+                               freenetRequest.parameters["sone"]!!.split(Regex("[ ,]+"))
                                                .forEach { webInterface.core.unfollowSone(currentSone, it) }
-                       }
-                       throw RedirectException(request.parameters["returnPage", 256])
+                       throw RedirectException(freenetRequest.parameters["returnPage", 256])
                }
        }