+++ /dev/null
-/*
- * Sone - PageToadlet.kt - Copyright © 2010–2020 David Roden
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- */
-
-package net.pterodactylus.sone.web.page
-
-import freenet.client.HighLevelSimpleClient
-import freenet.clients.http.LinkEnabledCallback
-import freenet.clients.http.LinkFilterExceptedToadlet
-import freenet.clients.http.SessionManager
-import freenet.clients.http.Toadlet
-import freenet.clients.http.ToadletContext
-import freenet.support.MultiValueTable
-import freenet.support.api.HTTPRequest
-import net.pterodactylus.sone.utils.use
-import net.pterodactylus.util.web.Method
-import net.pterodactylus.util.web.Page
-import net.pterodactylus.util.web.Response
-import java.net.URI
-
-/**
- * [Toadlet] implementation that is wrapped around a [Page].
- */
-class PageToadlet(
- highLevelSimpleClient: HighLevelSimpleClient,
- private val sessionManager: SessionManager,
- val menuName: String?,
- private val page: Page<FreenetRequest>,
- private val pathPrefix: String
-) : Toadlet(highLevelSimpleClient), LinkEnabledCallback, LinkFilterExceptedToadlet {
-
- override fun path() = pathPrefix + page.path
-
- override fun handleMethodGET(uri: URI, httpRequest: HTTPRequest, toadletContext: ToadletContext) =
- handleRequest(FreenetRequest(uri, Method.GET, httpRequest, toadletContext, sessionManager))
-
- fun handleMethodPOST(uri: URI?, httpRequest: HTTPRequest?, toadletContext: ToadletContext?) =
- handleRequest(FreenetRequest(uri!!, Method.POST, httpRequest!!, toadletContext!!, sessionManager))
-
- private fun handleRequest(pageRequest: FreenetRequest) {
- pageRequest.toadletContext.bucketFactory.makeBucket(-1).use { pageBucket ->
- pageBucket.outputStream.use { pageBucketOutputStream ->
- val pageResponse = page.handleRequest(pageRequest, Response(pageBucketOutputStream))
- // according to the javadoc, headers is allowed to return null but that’s stupid and it doesn’t do that.
- val headers = pageResponse.headers.fold(MultiValueTable<String, String>()) { headers, header ->
- headers.apply {
- header.forEach { put(header.name, it) }
- }
- }
- with(pageResponse) {
- writeReply(pageRequest.toadletContext, statusCode, contentType, statusText, headers, pageBucket)
- }
- }
- }
- }
-
- override fun isEnabled(toadletContext: ToadletContext) =
- if (page is LinkEnabledCallback) {
- page.isEnabled(toadletContext)
- } else
- true
-
- override fun isLinkExcepted(link: URI) =
- page is FreenetPage && page.isLinkExcepted(link)
-
- override fun toString() = "${javaClass.name}[path=${path()},page=$page]"
-
-}