/*
- * Sone - PostAccessor.java - Copyright © 2010 David Roden
+ * Sone - PostAccessor.java - Copyright © 2010–2013 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
import net.pterodactylus.sone.core.Core;
import net.pterodactylus.sone.data.Post;
-import net.pterodactylus.util.template.DataProvider;
+import net.pterodactylus.sone.data.Reply;
+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:
* {@inheritDoc}
*/
@Override
- public Object get(DataProvider dataProvider, Object object, String member) {
+ public Object get(TemplateContext templateContext, Object object, String member) {
+ Post post = (Post) object;
if ("replies".equals(member)) {
- return core.getReplies((Post) object);
+ return Collections2.filter(core.getReplies(post), Reply.FUTURE_REPLY_FILTER);
+ } else if (member.equals("likes")) {
+ return core.getLikes(post);
+ } else if (member.equals("liked")) {
+ Sone currentSone = (Sone) templateContext.get("currentSone");
+ return (currentSone != null) && (currentSone.isLikedPostId(post.getId()));
+ } else if (member.equals("new")) {
+ return !post.isKnown();
+ } else if (member.equals("bookmarked")) {
+ return core.isBookmarked(post);
+ } else if (member.equals("loaded")) {
+ return post.getSone() != null;
}
- return super.get(dataProvider, object, member);
+ return super.get(templateContext, object, member);
}
}