X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FUnfollowSonePage.kt;h=2521ef08f2a09c117449334767185ce9235b5315;hb=9bb01bc8fe281e6748ebd744ade8945bee841213;hp=cc411e229e3bef3ba4d31f95cf861a5785aa9434;hpb=ffd92ca2374c0b2218e583d02e0bdd24b8c110ae;p=Sone.git

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