X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FRescuePage.kt;h=1061a1089a3da9c9be3567324ded8c6820013d47;hb=1557106d0e373f7efc15faed65f69eb567613a45;hp=dbcf59f9632d748da8338b7af65da4d07afe100a;hpb=3cb0e2c78bb37f2d70ecde92ad4fcf642de788a3;p=Sone.git

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 dbcf59f..1061a10 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.kt
@@ -1,20 +1,22 @@
 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 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(freenetRequest: FreenetRequest, templateContext: TemplateContext) {
-		val soneRescuer = webInterface.core.getSoneRescuer(getCurrentSone(freenetRequest.toadletContext)!!)
+	override fun handleRequest(freenetRequest: FreenetRequest, currentSone: Sone, templateContext: TemplateContext) {
+		val soneRescuer = webInterface.core.getSoneRescuer(currentSone)
 		templateContext["soneRescuer"] = soneRescuer
 		if (freenetRequest.isPOST) {
 			freenetRequest.parameters["edition", 9]?.toIntOrNull()?.also {