2 * Sone - ReplyGroupFilter.java - Copyright © 2010–2013 David Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sone.template;
20 import java.util.HashMap;
21 import java.util.HashSet;
22 import java.util.List;
24 import java.util.Map.Entry;
27 import net.pterodactylus.sone.data.Post;
28 import net.pterodactylus.sone.data.PostReply;
29 import net.pterodactylus.sone.data.Sone;
30 import net.pterodactylus.util.template.Filter;
31 import net.pterodactylus.util.template.TemplateContext;
33 import com.google.common.base.Optional;
36 * {@link Filter} implementation that groups replies by the post the are in
37 * reply to, returning a map with the post as key and the list of replies as
40 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
42 public class ReplyGroupFilter implements Filter {
45 public Object format(TemplateContext templateContext, Object data, Map<String, Object> parameters) {
46 @SuppressWarnings("unchecked")
47 List<PostReply> allReplies = (List<PostReply>) data;
48 Map<Post, Set<Sone>> postSones = new HashMap<Post, Set<Sone>>();
49 Map<Post, Set<PostReply>> postReplies = new HashMap<Post, Set<PostReply>>();
50 for (PostReply reply : allReplies) {
52 * All replies from a new-reply notification have posts,
53 * ListNotificationFilters takes care of that.
55 Optional<Post> post = reply.getPost();
56 Set<Sone> sones = postSones.get(post.get());
58 sones = new HashSet<Sone>();
59 postSones.put(post.get(), sones);
61 sones.add(reply.getSone());
62 Set<PostReply> replies = postReplies.get(post.get());
63 if (replies == null) {
64 replies = new HashSet<PostReply>();
65 postReplies.put(post.get(), replies);
69 Map<Post, Map<String, Set<?>>> result = new HashMap<Post, Map<String, Set<?>>>();
70 for (Entry<Post, Set<Sone>> postEntry : postSones.entrySet()) {
71 if (result.containsKey(postEntry.getKey())) {
74 Map<String, Set<?>> postResult = new HashMap<String, Set<?>>();
75 postResult.put("sones", postEntry.getValue());
76 postResult.put("replies", postReplies.get(postEntry.getKey()));
77 result.put(postEntry.getKey(), postResult);