X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FPostAccessor.java;h=2c0b59efaec0ec55ab648e3bbd082429a5d6b4f1;hp=9284f0dfcdb03f759ed0b29c45a6b9e76281d51c;hb=c8f518f638aa16cf37145561b709d28252de213a;hpb=04b208b68bfb8c59647067b4a3b29ddd674455ac diff --git a/src/main/java/net/pterodactylus/sone/template/PostAccessor.java b/src/main/java/net/pterodactylus/sone/template/PostAccessor.java index 9284f0d..2c0b59e 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,14 @@ 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.getData("currentSone"); + return (currentSone != null) && (currentSone.isLikedPostId(post.getId())); } return super.get(dataProvider, object, member); }