X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FPostAccessor.kt;h=42239aa43bfea611995abe72e5d8338e3389a15e;hp=5222b945ab5ba51d44dc19ee8fcc4310fd501f16;hb=0ace0221fc20ef93457b8c62c6ac12286c1aa12c;hpb=d9cffbf70e8504180328964511303c1d70207b84

diff --git a/src/main/kotlin/net/pterodactylus/sone/template/PostAccessor.kt b/src/main/kotlin/net/pterodactylus/sone/template/PostAccessor.kt
index 5222b94..42239aa 100644
--- a/src/main/kotlin/net/pterodactylus/sone/template/PostAccessor.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/template/PostAccessor.kt
@@ -1,5 +1,5 @@
 /*
- * Sone - PostAccessor.java - Copyright © 2010–2019 David Roden
+ * Sone - PostAccessor.kt - Copyright © 2010–2020 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -14,10 +14,12 @@
  * You should have received a copy of the GNU General Public License
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
+
 package net.pterodactylus.sone.template
 
 import net.pterodactylus.sone.core.*
 import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.utils.*
 import net.pterodactylus.util.template.*
 
 /**
@@ -39,9 +41,8 @@ class PostAccessor(private val core: Core) : ReflectionAccessor() {
 					"liked" -> templateContext.currentSone?.isLikedPostId(post.id) ?: false
 					"new" -> !post.isKnown
 					"bookmarked" -> core.isBookmarked(post)
-					"replySone" -> core.getReplies(post)
-							.lastOrNull { it.sone.isLocal }
-							?.sone
+					"replySone" -> core.getReplies(post).lastOrNull { it.sone.isLocal }?.sone
+							?: post.recipient.let { it.takeIf { it.isLocal } }
 							?: post.sone.takeIf { it.isLocal }
 							?: templateContext.currentSone
 					else -> super.get(templateContext, `object`, member)
@@ -50,5 +51,5 @@ class PostAccessor(private val core: Core) : ReflectionAccessor() {
 
 }
 
-private fun Core.getReplies(post: Post) = getReplies(post.id).filter { Reply.FUTURE_REPLY_FILTER.apply(it) }
+private fun Core.getReplies(post: Post) = getReplies(post.id).filter(noFutureReply)
 private val TemplateContext?.currentSone: Sone? get() = this?.get("currentSone") as? Sone