X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FFollowSonePage.kt;h=4cba65add5b107b5f81d77b9c042eec2df35d320;hp=781e07fa6346a6d47592190ddaa0754a810b7c5d;hb=c389ee8f5e6eacae70438c3f364e8674c09d1959;hpb=3cb0e2c78bb37f2d70ecde92ad4fcf642de788a3

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/pages/FollowSonePage.kt b/src/main/kotlin/net/pterodactylus/sone/web/pages/FollowSonePage.kt
index 781e07f..4cba65a 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/pages/FollowSonePage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/FollowSonePage.kt
@@ -1,31 +1,31 @@
 package net.pterodactylus.sone.web.pages
 
-import net.pterodactylus.sone.utils.isPOST
-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 net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.main.*
+import net.pterodactylus.sone.utils.*
+import net.pterodactylus.sone.web.*
+import net.pterodactylus.sone.web.page.*
+import net.pterodactylus.util.template.*
+import javax.inject.*
 
 /**
  * This page lets the user follow another Sone.
  */
-class FollowSonePage(template: Template, webInterface: WebInterface):
-		SoneTemplatePage("followSone.html", template, "Page.FollowSone.Title", webInterface, true) {
+@ToadletPath("followSone.html")
+class FollowSonePage @Inject constructor(webInterface: WebInterface, loaders: Loaders, templateRenderer: TemplateRenderer) :
+		LoggedInPage("followSone.html", "Page.FollowSone.Title", webInterface, loaders, templateRenderer) {
 
-	override fun handleRequest(freenetRequest: FreenetRequest, templateContext: TemplateContext) {
-		if (freenetRequest.isPOST) {
-			freenetRequest.httpRequest.getPartAsStringFailsafe("sone", 1200).split(Regex("[ ,]+"))
-					.map { it to webInterface.core.getSone(it) }
-					.filter { it.second.isPresent }
-					.map { it.first to it.second.get() }
+	override fun handleRequest(soneRequest: SoneRequest, currentSone: Sone, templateContext: TemplateContext) {
+		if (soneRequest.isPOST) {
+			soneRequest.httpRequest.getPartAsStringFailsafe("sone", 1200).split(Regex("[ ,]+"))
+					.map { it to soneRequest.core.getSone(it) }
+					.filterNot { it.second == null }
 					.forEach { sone ->
-						webInterface.core.followSone(freenetRequest.currentSone, sone.first)
-						webInterface.core.markSoneKnown(sone.second)
+						soneRequest.core.followSone(currentSone, sone.first)
+						soneRequest.core.markSoneKnown(sone.second)
 					}
-			throw RedirectException(freenetRequest.httpRequest.getPartAsStringFailsafe("returnPage", 256))
+			throw RedirectException(soneRequest.httpRequest.getPartAsStringFailsafe("returnPage", 256))
 		}
 	}
 
-	private val FreenetRequest.currentSone get() = sessionProvider.getCurrentSone(toadletContext)
-
 }