1 package net.pterodactylus.sone.web.pages
3 import freenet.support.api.Bucket
4 import net.pterodactylus.sone.text.TextFilter
5 import net.pterodactylus.sone.utils.emptyToNull
6 import net.pterodactylus.sone.utils.headers
7 import net.pterodactylus.sone.utils.isPOST
8 import net.pterodactylus.sone.utils.parameters
9 import net.pterodactylus.sone.utils.use
10 import net.pterodactylus.sone.web.WebInterface
11 import net.pterodactylus.sone.web.page.FreenetRequest
12 import net.pterodactylus.util.template.Template
13 import net.pterodactylus.util.template.TemplateContext
14 import java.awt.image.BufferedImage
15 import java.io.ByteArrayInputStream
16 import java.io.ByteArrayOutputStream
17 import javax.imageio.ImageIO
20 * Page implementation that lets the user upload an image.
22 class UploadImagePage(template: Template, webInterface: WebInterface):
23 SoneTemplatePage("uploadImage.html", template, "Page.UploadImage.Title", webInterface, true) {
25 override fun handleRequest(freenetRequest: FreenetRequest, templateContext: TemplateContext) {
26 if (freenetRequest.isPOST) {
27 val parentAlbum = freenetRequest.parameters["parent"]!!.let(webInterface.core::getAlbum) ?: throw RedirectException("noPermission.html")
28 if (parentAlbum.sone != getCurrentSone(freenetRequest.toadletContext)) {
29 throw RedirectException("noPermission.html")
31 val title = freenetRequest.parameters["title", 200].emptyToNull ?: throw RedirectException("emptyImageTitle.html")
33 val uploadedFile = freenetRequest.httpRequest.getUploadedFile("image")
34 val bytes = uploadedFile.data.use { it.toByteArray() }
35 val bufferedImage = bytes.toImage()
36 if (bufferedImage == null) {
37 templateContext["messages"] = webInterface.l10n.getString("Page.UploadImage.Error.InvalidImage")
41 val temporaryImage = webInterface.core.createTemporaryImage(bytes.mimeType, bytes)
42 webInterface.core.createImage(getCurrentSone(freenetRequest.toadletContext), parentAlbum, temporaryImage).modify().apply {
43 setWidth(bufferedImage.width)
44 setHeight(bufferedImage.height)
46 setDescription(TextFilter.filter(freenetRequest.headers["Host"], freenetRequest.parameters["description", 4000]))
48 throw RedirectException("imageBrowser.html?album=${parentAlbum.id}")
52 private fun Bucket.toByteArray(): ByteArray = ByteArrayOutputStream(size().toInt()).use { outputStream ->
53 inputStream.copyTo(outputStream)
54 outputStream.toByteArray()
57 private fun ByteArray.toImage(): BufferedImage? = ByteArrayInputStream(this).use {
61 private val ByteArray.mimeType get() = ByteArrayInputStream(this).use {
62 ImageIO.createImageInputStream(it).use {
63 ImageIO.getImageReaders(it).asSequence()
64 .firstOrNull()?.originatingProvider?.mimeTypes?.firstOrNull()
71 private const val UNKNOWN_MIME_TYPE = "application/octet-stream"