X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FViewSonePage.java;h=3f4d92d730ec47ca7087b856aa8cb55d263523e4;hb=ece9119924c20b12ef1c2f0908a92dc8323f8884;hp=8b88f2e3bf4bae9be6c8f6fab614e2bdd7c1ad64;hpb=728eb9dfab6b9284dbb69701fa8ba4a9b072e7ca;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/web/ViewSonePage.java b/src/main/java/net/pterodactylus/sone/web/ViewSonePage.java
index 8b88f2e..3f4d92d 100644
--- a/src/main/java/net/pterodactylus/sone/web/ViewSonePage.java
+++ b/src/main/java/net/pterodactylus/sone/web/ViewSonePage.java
@@ -17,8 +17,21 @@
 
 package net.pterodactylus.sone.web;
 
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Comparator;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
+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;
 
 /**
  * Lets the user browser another Sone.
@@ -47,11 +60,56 @@ public class ViewSonePage extends SoneTemplatePage {
 	 * {@inheritDoc}
 	 */
 	@Override
-	protected void processTemplate(Request request, Template template) throws RedirectException {
-		super.processTemplate(request, template);
+	protected void processTemplate(Request request, TemplateContext templateContext) throws RedirectException {
+		super.processTemplate(request, templateContext);
 		String soneId = request.getHttpRequest().getParam("sone");
 		Sone sone = webInterface.getCore().getSone(soneId, false);
-		template.set("sone", sone);
+		templateContext.set("sone", sone);
+		Set<Reply> replies = sone.getReplies();
+		final Map<Post, List<Reply>> repliedPosts = new HashMap<Post, List<Reply>>();
+		for (Reply reply : replies) {
+			Post post = reply.getPost();
+			if (repliedPosts.containsKey(post) || sone.equals(post.getSone())) {
+				continue;
+			}
+			repliedPosts.put(post, webInterface.getCore().getReplies(post));
+		}
+		List<Post> posts = new ArrayList<Post>(repliedPosts.keySet());
+		Collections.sort(posts, new Comparator<Post>() {
+
+			@Override
+			public int compare(Post leftPost, Post rightPost) {
+				return (int) Math.min(Integer.MAX_VALUE, Math.max(Integer.MIN_VALUE, repliedPosts.get(rightPost).get(0).getTime() - repliedPosts.get(leftPost).get(0).getTime()));
+			}
+
+		});
+
+		Pagination<Post> repliedPostPagination = new Pagination<Post>(posts, 10).setPage(Numbers.safeParseInteger(request.getHttpRequest().getParam("repliedPostPage"), 0));
+		templateContext.set("repliedPostPagination", repliedPostPagination);
+		templateContext.set("repliedPosts", repliedPostPagination.getItems());
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	@SuppressWarnings("unchecked")
+	protected void postProcess(Request request, TemplateContext templateContext) {
+		Sone sone = (Sone) templateContext.get("sone");
+		if (sone == null) {
+			return;
+		}
+		webInterface.getCore().markSoneKnown(sone);
+		List<Post> posts = sone.getPosts();
+		posts.addAll((List<Post>) templateContext.get("repliedPosts"));
+		for (Post post : posts) {
+			if (post.getSone() != null) {
+				webInterface.getCore().markPostKnown(post);
+			}
+			for (Reply reply : webInterface.getCore().getReplies(post)) {
+				webInterface.getCore().markReplyKnown(reply);
+			}
+		}
 	}
 
 }