X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FPostAccessor.java;h=d3af8975d06964a038526bb5348f3dfaac04f2c2;hb=8871424239966323798a4ebd1c7d2c27ba830f1e;hp=9284f0dfcdb03f759ed0b29c45a6b9e76281d51c;hpb=850ac7a9ce6d9051d8f85919a1cd5a4beb57950f;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 9284f0d..d3af897 100644 --- a/src/main/java/net/pterodactylus/sone/template/PostAccessor.java +++ b/src/main/java/net/pterodactylus/sone/template/PostAccessor.java @@ -19,6 +19,7 @@ package net.pterodactylus.sone.template; import net.pterodactylus.sone.core.Core; import net.pterodactylus.sone.data.Post; +import net.pterodactylus.sone.data.Sone; import net.pterodactylus.util.template.DataProvider; import net.pterodactylus.util.template.ReflectionAccessor; @@ -51,8 +52,16 @@ public class PostAccessor extends ReflectionAccessor { */ @Override public Object get(DataProvider dataProvider, Object object, String member) { + Post post = (Post) object; if ("replies".equals(member)) { - return core.getReplies((Post) object); + return core.getReplies(post); + } else if (member.equals("likes")) { + return core.getLikes(post); + } else if (member.equals("liked")) { + Sone currentSone = (Sone) dataProvider.get("currentSone"); + return (currentSone != null) && (currentSone.isLikedPostId(post.getId())); + } else if (member.equals("new")) { + return core.isNewPost(post.getId(), false); } return super.get(dataProvider, object, member); }