package net.pterodactylus.sone.web.pages

import net.pterodactylus.sone.data.Album
import net.pterodactylus.sone.data.Sone
import net.pterodactylus.sone.main.*
import net.pterodactylus.sone.utils.paginate
import net.pterodactylus.sone.utils.parameters
import net.pterodactylus.sone.web.WebInterface
import net.pterodactylus.sone.web.page.*
import net.pterodactylus.util.template.Template
import net.pterodactylus.util.template.TemplateContext
import java.net.URI
import javax.inject.Inject

/**
 * The image browser page is the entry page for the image management.
 */
@MenuName("ImageBrowser")
class ImageBrowserPage @Inject constructor(template: Template, webInterface: WebInterface, loaders: Loaders):
		LoggedInPage("imageBrowser.html", template, "Page.ImageBrowser.Title", webInterface, loaders) {

	override fun handleRequest(soneRequest: SoneRequest, currentSone: Sone, templateContext: TemplateContext) {
		if ("album" in soneRequest.parameters) {
			templateContext["albumRequested"] = true
			templateContext["album"] = soneRequest.core.getAlbum(soneRequest.parameters["album"]!!)
			templateContext["page"] = soneRequest.parameters["page"]
		} else if ("image" in soneRequest.parameters) {
			templateContext["imageRequested"] = true
			templateContext["image"] = soneRequest.core.getImage(soneRequest.parameters["image"])
		} else if (soneRequest.parameters["mode"] == "gallery") {
			templateContext["galleryRequested"] = true
			soneRequest.core.sones
					.map(Sone::getRootAlbum)
					.flatMap(Album::getAlbums)
					.flatMap { Album.FLATTENER.apply(it)!! }
					.filterNot(Album::isEmpty)
					.sortedBy(Album::getTitle)
					.also { albums ->
						albums.paginate(soneRequest.core.preferences.imagesPerPage)
								.turnTo(soneRequest.parameters["page"]?.toIntOrNull() ?: 0)
								.also { pagination ->
							templateContext["albumPagination"] = pagination
							templateContext["albums"] = pagination.items
						}
					}
		} else {
			templateContext["soneRequested"] = true
			templateContext["sone"] = soneRequest.core.getSone(soneRequest.httpRequest.getParam("sone")) ?: currentSone
		}
	}

	override fun isLinkExcepted(link: URI) = true

}