X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FUnfollowSonePage.kt;h=11d456c2fdaf01686acaa81910268070623e1804;hp=3b299d2a6c3a53dac7556e0de6e5a5ae3370613d;hb=cd72add62ab407336b471d4b7cda8e33dd2df5c6;hpb=e041747229cbcb4acba7d76257bc50d76ff7a7a7

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..11d456c 100644
--- 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,25 @@
 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.sone.web.page.*
 import net.pterodactylus.util.template.Template
 import net.pterodactylus.util.template.TemplateContext
-import java.util.regex.Pattern
+import javax.inject.Inject
 
 /**
  * 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 @Inject constructor(template: Template, webInterface: WebInterface) :
+		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("[ ,]+"))
-						.forEach { webInterface.core.unfollowSone(currentSone, it) }
-			}
-			throw RedirectException(request.parameters["returnPage", 256])
+	override fun handleRequest(soneRequest: SoneRequest, currentSone: Sone, templateContext: TemplateContext) {
+		if (soneRequest.isPOST) {
+			soneRequest.parameters["sone"]!!.split(Regex("[ ,]+"))
+					.forEach { soneRequest.core.unfollowSone(currentSone, it) }
+			throw RedirectException(soneRequest.parameters["returnPage", 256])
 		}
 	}