Mark shown posts and replies known.

[Sone.git] / src / main / java / net / pterodactylus / sone / web / IndexPage.java

diff --git a/src/main/java/net/pterodactylus/sone/web/IndexPage.java b/src/main/java/net/pterodactylus/sone/web/IndexPage.java
index 0e72938..7562526 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/IndexPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/IndexPage.java
@@ -17,6 +17,13 @@
 
 package net.pterodactylus.sone.web;
 
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.Reply;
+import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.util.template.Template;
 
 /**
@@ -34,7 +41,7 @@ public class IndexPage extends SoneTemplatePage {
         *            The Sone web interface
         */
        public IndexPage(Template template, WebInterface webInterface) {
-               super("index.html", template, "Page.Index.Title", webInterface);
+               super("index.html", template, "Page.Index.Title", webInterface, true);
        }
 
        //
@@ -45,8 +52,34 @@ public class IndexPage extends SoneTemplatePage {
         * {@inheritDoc}
         */
        @Override
-       protected boolean requiresLogin() {
-               return true;
+       protected void processTemplate(Request request, Template template) throws RedirectException {
+               super.processTemplate(request, template);
+               Sone sone = getCurrentSone(request.getToadletContext());
+               List<Post> allPosts = new ArrayList<Post>();
+               allPosts.addAll(sone.getPosts());
+               for (String friendSoneId : sone.getFriends()) {
+                       if (!webInterface.getCore().hasSone(friendSoneId)) {
+                               continue;
+                       }
+                       allPosts.addAll(webInterface.getCore().getSone(friendSoneId).getPosts());
+               }
+               Collections.sort(allPosts, Post.TIME_COMPARATOR);
+               template.set("posts", allPosts);
+       }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       protected void postProcess(Request request, Template template) {
+               @SuppressWarnings("unchecked")
+               List<Post> posts = (List<Post>) template.get("posts");
+               for (Post post : posts) {
+                       webInterface.getCore().markPostKnown(post);
+                       for (Reply reply : webInterface.getCore().getReplies(post)) {
+                               webInterface.getCore().markReplyKnown(reply);
+                       }
+               }
        }
 
 }