src/main/kotlin/net/pterodactylus/sone/web/pages/ImageBrowserPage.kt

   1 package net.pterodactylus.sone.web.pages
   2
   3 import net.pterodactylus.sone.data.Album
   4 import net.pterodactylus.sone.data.Sone
   5 import net.pterodactylus.sone.utils.paginate
   6 import net.pterodactylus.sone.utils.parameters
   7 import net.pterodactylus.sone.web.WebInterface
   8 import net.pterodactylus.sone.web.page.FreenetRequest
   9 import net.pterodactylus.util.template.Template
  10 import net.pterodactylus.util.template.TemplateContext
  11 import java.net.URI
  12 import javax.inject.Inject
  13
  14 /**
  15  * The image browser page is the entry page for the image management.
  16  */
  17 class ImageBrowserPage @Inject constructor(template: Template, webInterface: WebInterface):
  18                 LoggedInPage("imageBrowser.html", template, "Page.ImageBrowser.Title", webInterface) {
  19
  20         override fun handleRequest(freenetRequest: FreenetRequest, currentSone: Sone, templateContext: TemplateContext) {
  21                 if ("album" in freenetRequest.parameters) {
  22                         templateContext["albumRequested"] = true
  23                         templateContext["album"] = webInterface.core.getAlbum(freenetRequest.parameters["album"]!!)
  24                         templateContext["page"] = freenetRequest.parameters["page"]
  25                 } else if ("image" in freenetRequest.parameters) {
  26                         templateContext["imageRequested"] = true
  27                         templateContext["image"] = webInterface.core.getImage(freenetRequest.parameters["image"])
  28                 } else if (freenetRequest.parameters["mode"] == "gallery") {
  29                         templateContext["galleryRequested"] = true
  30                         webInterface.core.sones
  31                                         .map(Sone::getRootAlbum)
  32                                         .flatMap(Album::getAlbums)
  33                                         .flatMap { Album.FLATTENER.apply(it)!! }
  34                                         .filterNot(Album::isEmpty)
  35                                         .sortedBy(Album::getTitle)
  36                                         .also { albums ->
  37                                                 albums.paginate(webInterface.core.preferences.imagesPerPage)
  38                                                                 .turnTo(freenetRequest.parameters["page"]?.toIntOrNull() ?: 0)
  39                                                                 .also { pagination ->
  40                                                         templateContext["albumPagination"] = pagination
  41                                                         templateContext["albums"] = pagination.items
  42                                                 }
  43                                         }
  44                 } else {
  45                         templateContext["soneRequested"] = true
  46                         templateContext["sone"] = webInterface.core.getSone(freenetRequest.httpRequest.getParam("sone")) ?: currentSone
  47                 }
  48         }
  49
  50         override fun isLinkExcepted(link: URI) = true
  51
  52 }