package net.pterodactylus.sone.web.pages import net.pterodactylus.sone.data.Album import net.pterodactylus.sone.data.Sone import net.pterodactylus.sone.utils.Pagination import net.pterodactylus.sone.utils.parameters 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 java.net.URI /** * The image browser page is the entry page for the image management. */ class ImageBrowserPage(template: Template, webInterface: WebInterface): SoneTemplatePage("imageBrowser.html", template, "Page.ImageBrowser.Title", webInterface, true) { override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) { if ("album" in request.parameters) { templateContext["albumRequested"] = true templateContext["album"] = webInterface.core.getAlbum(request.parameters["album"]!!) templateContext["page"] = request.parameters["page"] } else if ("image" in request.parameters) { templateContext["imageRequested"] = true templateContext["image"] = webInterface.core.getImage(request.parameters["image"]) } else if (request.parameters["mode"] == "gallery") { templateContext["galleryRequested"] = true webInterface.core.sones .map(Sone::getRootAlbum) .flatMap(Album::getAlbums) .flatMap { Album.FLATTENER.apply(it)!! } .filterNot(Album::isEmpty) .sortedBy(Album::getTitle) .also { albums -> Pagination(albums, webInterface.core.preferences.imagesPerPage).apply { page = request.parameters["page"]?.toIntOrNull() ?: 0 }.also { pagination -> templateContext["albumPagination"] = pagination templateContext["albums"] = pagination.items } } } else { templateContext["soneRequested"] = true templateContext["sone"] = webInterface.core.getSone(request.httpRequest.getParam("sone")).orNull() ?: getCurrentSone(request.toadletContext) } } override fun isLinkExcepted(link: URI?) = true }