X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpage%2FPageToadlet.kt;fp=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpage%2FPageToadlet.kt;h=a1cddfe4b30b6d464cc52b671fc16ada872612c8;hp=0000000000000000000000000000000000000000;hb=9235105b5767441ce15d7ede26d6ef963a38e51d;hpb=ef88dd2add0bad9d08dfecbb49bd7577bfafbb50

diff --git a/src/main/java/net/pterodactylus/sone/web/page/PageToadlet.kt b/src/main/java/net/pterodactylus/sone/web/page/PageToadlet.kt
new file mode 100644
index 0000000..a1cddfe
--- /dev/null
+++ b/src/main/java/net/pterodactylus/sone/web/page/PageToadlet.kt
@@ -0,0 +1,81 @@
+/*
+ * 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]"
+
+}