X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FRescuePage.kt;h=367d6fb68dc2bb8fed1e6a2ccd75d55a52f23f46;hp=b882ee6e29906ceb0b5bb2da2345fd9efbf7cc16;hb=c28013c8a4bcb9776a1e2d82ffd6c4c8297ffb62;hpb=2000b0561c6a7e4b1fbc6e452b89c9504b3b5d33

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.kt b/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.kt
index b882ee6..367d6fb 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.kt
@@ -1,28 +1,30 @@
 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 javax.inject.Inject
 
 /**
  * Page that lets the user control the rescue mode for a Sone.
  */
-class RescuePage(template: Template, webInterface: WebInterface):
-		SoneTemplatePage("rescue.html", template, "Page.Rescue.Title", webInterface, true) {
+class RescuePage @Inject constructor(template: Template, webInterface: WebInterface):
+		LoggedInPage("rescue.html", template, "Page.Rescue.Title", webInterface) {
 
-	override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) {
-		val soneRescuer = webInterface.core.getSoneRescuer(getCurrentSone(request.toadletContext)!!)
+	override fun handleRequest(soneRequest: SoneRequest, currentSone: Sone, templateContext: TemplateContext) {
+		val soneRescuer = soneRequest.core.getSoneRescuer(currentSone)
 		templateContext["soneRescuer"] = soneRescuer
-		if (request.isPOST) {
-			request.parameters["edition", 9]?.toIntOrNull()?.also {
+		if (soneRequest.isPOST) {
+			soneRequest.parameters["edition", 9]?.toIntOrNull()?.also {
 				if (it > -1) {
 					soneRescuer.setEdition(it.toLong())
 				}
 			}
-			if (request.parameters["fetch", 8] == "true") {
+			if (soneRequest.parameters["fetch", 8] == "true") {
 				soneRescuer.startNextFetch()
 			}
 			throw RedirectException("rescue.html")