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;
*/
@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);
}