src/main/kotlin/net/pterodactylus/sone/web/pages/SoneTemplatePage.kt

   1 package net.pterodactylus.sone.web.pages
   2
   3 import freenet.clients.http.*
   4 import net.pterodactylus.sone.data.*
   5 import net.pterodactylus.sone.main.*
   6 import net.pterodactylus.sone.utils.*
   7 import net.pterodactylus.sone.web.*
   8 import net.pterodactylus.sone.web.page.*
   9 import net.pterodactylus.util.notify.*
  10 import net.pterodactylus.util.template.*
  11 import net.pterodactylus.util.web.*
  12 import java.net.*
  13
  14 /**
  15  * Base page for the Sone web interface.
  16  */
  17 open class SoneTemplatePage @JvmOverloads constructor(
  18                 private val webInterface: WebInterface,
  19                 loaders: Loaders,
  20                 templateRenderer: TemplateRenderer,
  21                 private val pageTitleKey: String? = null,
  22                 private val requiresLogin: Boolean = false,
  23                 private val pageTitle: (FreenetRequest) -> String = { pageTitleKey?.let(webInterface.l10n::getString) ?: "" }
  24 ) : FreenetTemplatePage(templateRenderer, loaders, "noPermission.html") {
  25
  26         private val core = webInterface.core
  27         private val sessionProvider: SessionProvider = webInterface
  28
  29         protected fun getCurrentSone(toadletContext: ToadletContext, createSession: Boolean = true) =
  30                         sessionProvider.getCurrentSone(toadletContext, createSession)
  31
  32         protected fun setCurrentSone(toadletContext: ToadletContext, sone: Sone?) =
  33                         sessionProvider.setCurrentSone(toadletContext, sone)
  34
  35         fun requiresLogin() = requiresLogin
  36
  37         override fun getPageTitle(request: FreenetRequest) = getPageTitle(request.toSoneRequest(core, webInterface))
  38
  39         open fun getPageTitle(soneRequest: SoneRequest) = pageTitle(soneRequest)
  40
  41         override val styleSheets = listOf("css/sone.css")
  42
  43         override val shortcutIcon = "images/icon.png"
  44
  45         override fun getAdditionalLinkNodes(request: FreenetRequest) =
  46                         listOf(mapOf(
  47                                         "rel" to "search",
  48                                         "type" to "application/opensearchdescription+xml",
  49                                         "title" to "Sone",
  50                                         "href" to "http://${request.httpRequest.getHeader("host")}/Sone/OpenSearch.xml"
  51                         ))
  52
  53         final override fun processTemplate(request: FreenetRequest, templateContext: TemplateContext) {
  54                 super.processTemplate(request, templateContext)
  55                 templateContext["preferences"] = core.preferences
  56                 templateContext["currentSone"] = getCurrentSone(request.toadletContext)
  57                 templateContext["localSones"] = core.localSones
  58                 templateContext["request"] = request
  59                 templateContext["currentVersion"] = SonePlugin.getPluginVersion()
  60                 templateContext["hasLatestVersion"] = core.updateChecker.hasLatestVersion()
  61                 templateContext["latestEdition"] = core.updateChecker.latestEdition
  62                 templateContext["latestVersion"] = core.updateChecker.latestVersion
  63                 templateContext["latestVersionTime"] = core.updateChecker.latestVersionDate
  64                 webInterface.getNotifications(getCurrentSone(request.toadletContext)).sortedBy(Notification::getCreatedTime).run {
  65                         templateContext["notifications"] = this
  66                         templateContext["notificationHash"] = this.hashCode()
  67                 }
  68                 handleRequest(request, templateContext)
  69         }
  70
  71         open fun handleRequest(freenetRequest: FreenetRequest, templateContext: TemplateContext) {
  72                 handleRequest(freenetRequest.toSoneRequest(core, webInterface), templateContext)
  73         }
  74
  75         open fun handleRequest(soneRequest: SoneRequest, templateContext: TemplateContext) {
  76         }
  77
  78         override fun getRedirectTarget(request: FreenetRequest): String? {
  79                 if (requiresLogin && getCurrentSone(request.toadletContext) == null) {
  80                         val parameters = request.httpRequest.parameterNames
  81                                         .flatMap { name -> request.httpRequest.getMultipleParam(name).map { name to it } }
  82                                         .joinToString("&") { "${it.first.urlEncode}=${it.second.urlEncode}" }
  83                                         .emptyToNull
  84                         return "login.html?target=${request.httpRequest.path}${parameters?.let { ("?" + it).urlEncode } ?: ""}"
  85                 }
  86                 return null
  87         }
  88
  89         private val String.urlEncode: String get() = URLEncoder.encode(this, "UTF-8")
  90
  91         override fun isEnabled(toadletContext: ToadletContext) =
  92                         isEnabled(SoneRequest(toadletContext.uri, Method.GET, HTTPRequestImpl(toadletContext.uri, "GET"), toadletContext, webInterface.l10n, webInterface.sessionManager, core, webInterface))
  93
  94         open fun isEnabled(soneRequest: SoneRequest) = when {
  95                 requiresLogin && getCurrentSone(soneRequest.toadletContext) == null -> false
  96                 core.preferences.requireFullAccess && !soneRequest.toadletContext.isAllowedFullAccess -> false
  97                 else -> true
  98         }
  99
 100 }