Add page that always requires a logged-in user

[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / pages / RescuePage.kt

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..c3fabdb 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/RescuePage.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
@@ -11,10 +12,10 @@ import net.pterodactylus.util.template.TemplateContext
  * 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) {
+               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 {