X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FUploadImagePage.kt;fp=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FUploadImagePage.kt;h=774a17940f4bf8e34f10695cbaab42c2f2c28e4a;hb=51ab1ab2da27d02a884295cea31c4f6d30b5211b;hp=0000000000000000000000000000000000000000;hpb=c3a5dfacd98ca8190b266ac0652cbba93423a7c7;p=Sone.git

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/pages/UploadImagePage.kt b/src/main/kotlin/net/pterodactylus/sone/web/pages/UploadImagePage.kt
new file mode 100644
index 0000000..774a179
--- /dev/null
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/UploadImagePage.kt
@@ -0,0 +1,72 @@
+package net.pterodactylus.sone.web.pages
+
+import freenet.support.api.Bucket
+import net.pterodactylus.sone.text.TextFilter
+import net.pterodactylus.sone.utils.emptyToNull
+import net.pterodactylus.sone.utils.headers
+import net.pterodactylus.sone.utils.isPOST
+import net.pterodactylus.sone.utils.parameters
+import net.pterodactylus.sone.utils.use
+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 org.bouncycastle.asn1.x500.style.RFC4519Style.title
+import java.awt.image.BufferedImage
+import java.io.ByteArrayInputStream
+import java.io.ByteArrayOutputStream
+import javax.imageio.ImageIO
+
+/**
+ * Page implementation that lets the user upload an image.
+ */
+class UploadImagePage(template: Template, webInterface: WebInterface):
+		SoneTemplatePage("uploadImage.html", template, "Page.UploadImage.Title", webInterface, true) {
+
+	override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) {
+		if (request.isPOST) {
+			val parentAlbum = request.parameters["parent"]!!.let(webInterface.core::getAlbum) ?: throw RedirectException("noPermission.html")
+			if (parentAlbum.sone != getCurrentSone(request.toadletContext)) {
+				throw RedirectException("noPermission.html")
+			}
+			val title = request.parameters["title", 200].emptyToNull ?: throw RedirectException("emptyImageTitle.html")
+
+			val uploadedFile = request.httpRequest.getUploadedFile("image")
+			val bytes = uploadedFile.data.use { it.toByteArray() }
+			val bufferedImage = bytes.toImage()
+			if (bufferedImage == null) {
+				templateContext["messages"] = webInterface.l10n.getString("Page.UploadImage.Error.InvalidImage")
+				return
+			}
+
+			val temporaryImage = webInterface.core.createTemporaryImage(bytes.mimeType, bytes)
+			webInterface.core.createImage(getCurrentSone(request.toadletContext), parentAlbum, temporaryImage).modify().apply {
+				setWidth(bufferedImage.width)
+				setHeight(bufferedImage.height)
+				setTitle(title)
+				setDescription(TextFilter.filter(request.headers["Host"], request.parameters["description", 4000]))
+			}.update()
+			throw RedirectException("imageBrowser.html?album=${parentAlbum.id}")
+		}
+	}
+
+	private fun Bucket.toByteArray(): ByteArray = ByteArrayOutputStream(size().toInt()).use { outputStream ->
+		inputStream.copyTo(outputStream)
+		outputStream.toByteArray()
+	}
+
+	private fun ByteArray.toImage(): BufferedImage? = ByteArrayInputStream(this).use {
+		ImageIO.read(it)
+	}
+
+	private val ByteArray.mimeType get() = ByteArrayInputStream(this).use {
+		ImageIO.createImageInputStream(it).use {
+			ImageIO.getImageReaders(it).asSequence()
+					.firstOrNull()?.originatingProvider?.mimeTypes?.firstOrNull()
+					?: UNKNOWN_MIME_TYPE
+		}
+	}
+
+}
+
+private const val UNKNOWN_MIME_TYPE = "application/octet-stream"