import net.pterodactylus.sone.data.Post;
import net.pterodactylus.sone.data.Reply;
import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.util.collection.Pagination;
+import net.pterodactylus.util.number.Numbers;
import net.pterodactylus.util.template.Template;
+import net.pterodactylus.util.template.TemplateContext;
/**
* The index page shows the main page of Sone. This page will contain the posts
* {@inheritDoc}
*/
@Override
- protected void processTemplate(Request request, Template template) throws RedirectException {
- super.processTemplate(request, template);
- Sone sone = getCurrentSone(request.getToadletContext());
+ protected void processTemplate(Request request, TemplateContext templateContext) throws RedirectException {
+ super.processTemplate(request, templateContext);
+ Sone currentSone = getCurrentSone(request.getToadletContext());
List<Post> allPosts = new ArrayList<Post>();
- allPosts.addAll(sone.getPosts());
- for (String friendSoneId : sone.getFriends()) {
+ allPosts.addAll(currentSone.getPosts());
+ for (String friendSoneId : currentSone.getFriends()) {
if (!webInterface.getCore().hasSone(friendSoneId)) {
continue;
}
allPosts.addAll(webInterface.getCore().getSone(friendSoneId).getPosts());
}
+ for (Sone sone : webInterface.getCore().getSones()) {
+ for (Post post : sone.getPosts()) {
+ if (currentSone.equals(post.getRecipient()) && !allPosts.contains(post)) {
+ allPosts.add(post);
+ }
+ }
+ }
Collections.sort(allPosts, Post.TIME_COMPARATOR);
- template.set("posts", allPosts);
- }
+ Pagination<Post> pagination = new Pagination<Post>(allPosts, 25).setPage(Numbers.safeParseInteger(request.getHttpRequest().getParam("page"), 0));
+ templateContext.set("pagination", pagination);
+ templateContext.set("posts", pagination.getItems());
- /**
- * {@inheritDoc}
- */
- @Override
- protected void postProcess(Request request, Template template) {
- @SuppressWarnings("unchecked")
- List<Post> posts = (List<Post>) template.get("posts");
- for (Post post : posts) {
+ /* mark it all as known. */
+ for (Post post : pagination.getItems()) {
webInterface.getCore().markPostKnown(post);
for (Reply reply : webInterface.getCore().getReplies(post)) {
webInterface.getCore().markReplyKnown(reply);
}
}
+
}
}