X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FIndexPage.kt;h=0257cb1c7e1d1fb690d89b4d096d3337db5950f0;hb=74cbe983cd38da24847e6247ac595cb70d8e115b;hp=35f0f75e7690a9a143f73e46b4a52141cf446205;hpb=107af8767f5dce8c27fa4c91b3ff3bc951140cca;p=Sone.git

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/pages/IndexPage.kt b/src/main/kotlin/net/pterodactylus/sone/web/pages/IndexPage.kt
index 35f0f75..0257cb1 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/pages/IndexPage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/pages/IndexPage.kt
@@ -1,5 +1,6 @@
 package net.pterodactylus.sone.web.pages
 
+import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.notify.PostVisibilityFilter
 import net.pterodactylus.sone.utils.Pagination
 import net.pterodactylus.sone.utils.parameters
@@ -13,15 +14,12 @@ import net.pterodactylus.util.template.TemplateContext
  * of all friends of the current user.
  */
 class IndexPage(template: Template, webInterface: WebInterface, private val postVisibilityFilter: PostVisibilityFilter):
-		SoneTemplatePage("index.html", template, "Page.Index.Title", webInterface, true) {
+		LoggedInPage("index.html", template, "Page.Index.Title", webInterface) {
 
-	override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) {
-		getCurrentSone(request.toadletContext)!!.let { currentSone ->
+	override fun handleRequest(freenetRequest: FreenetRequest, currentSone: Sone, templateContext: TemplateContext) {
 			(currentSone.posts +
 					currentSone.friends
-							.map { webInterface.core.getSone(it) }
-							.filter { it.isPresent }
-							.map { it.get() }
+							.mapNotNull(webInterface.core::getSone)
 							.flatMap { it.posts } +
 					webInterface.core.getDirectedPosts(currentSone.id)
 					).distinct()
@@ -29,13 +27,12 @@ class IndexPage(template: Template, webInterface: WebInterface, private val post
 					.sortedByDescending { it.time }
 					.let { posts ->
 						Pagination(posts, webInterface.core.preferences.postsPerPage).apply {
-							page = request.parameters["page"]?.toIntOrNull() ?: 0
+							page = freenetRequest.parameters["page"]?.toIntOrNull() ?: 0
 						}.let { pagination ->
 							templateContext["pagination"] = pagination
 							templateContext["posts"] = pagination.items
 						}
 					}
-		}
 	}
 
 }