X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FPostAccessor.java;h=26a05db673e4105e6d53efbad994314e5dfb7bcd;hb=9411a2829261e28c88709228b07e1784e7333bd6;hp=d593b516d2f9f5da8f206331227e40b8ab249fad;hpb=6e9a43ccd93ae125720547c0fe421dc81a54ba90;p=Sone.git diff --git a/src/main/java/net/pterodactylus/sone/template/PostAccessor.java b/src/main/java/net/pterodactylus/sone/template/PostAccessor.java index d593b51..26a05db 100644 --- a/src/main/java/net/pterodactylus/sone/template/PostAccessor.java +++ b/src/main/java/net/pterodactylus/sone/template/PostAccessor.java @@ -17,6 +17,8 @@ package net.pterodactylus.sone.template; +import static com.google.common.collect.FluentIterable.from; + import net.pterodactylus.sone.core.Core; import net.pterodactylus.sone.data.Post; import net.pterodactylus.sone.data.Reply; @@ -24,8 +26,6 @@ import net.pterodactylus.sone.data.Sone; import net.pterodactylus.util.template.ReflectionAccessor; import net.pterodactylus.util.template.TemplateContext; -import com.google.common.collect.Collections2; - /** * Accessor for {@link Post} objects that adds additional properties: *
@@ -50,19 +50,16 @@ public class PostAccessor extends ReflectionAccessor { this.core = core; } - /** - * {@inheritDoc} - */ @Override public Object get(TemplateContext templateContext, Object object, String member) { Post post = (Post) object; if ("replies".equals(member)) { - return Collections2.filter(core.getReplies(post.getId()), Reply.FUTURE_REPLY_FILTER); + return from(post.getReplies()).filter(Reply.FUTURE_REPLY_FILTER).toList(); } else if (member.equals("likes")) { - return core.getLikes(post); + return post.getLikes(); } else if (member.equals("liked")) { Sone currentSone = (Sone) templateContext.get("currentSone"); - return (currentSone != null) && (currentSone.isLikedPostId(post.getId())); + return (currentSone != null) && post.isLiked(currentSone); } else if (member.equals("new")) { return !post.isKnown(); } else if (member.equals("bookmarked")) {