Refactor notification filtering
[Sone.git] / src / main / java / net / pterodactylus / sone / notify / ListNotificationFilters.java
index beeb81e..0a2a518 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Sone - ListNotificationFilters.java - Copyright © 2010 David Roden
+ * Sone - ListNotificationFilters.java - Copyright © 2010–2015 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
 
 package net.pterodactylus.sone.notify;
 
+import static com.google.common.collect.FluentIterable.from;
+
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
+import javax.annotation.Nonnull;
+import javax.inject.Inject;
+import javax.inject.Singleton;
 
 import net.pterodactylus.sone.data.Post;
 import net.pterodactylus.sone.data.PostReply;
-import net.pterodactylus.sone.data.Reply;
 import net.pterodactylus.sone.data.Sone;
-import net.pterodactylus.sone.freenet.wot.OwnIdentity;
-import net.pterodactylus.sone.freenet.wot.Trust;
 import net.pterodactylus.util.notify.Notification;
-import net.pterodactylus.util.validation.Validation;
+
+import com.google.common.base.Optional;
 
 /**
  * Filter for {@link ListNotification}s.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
+@Singleton
 public class ListNotificationFilters {
 
+       private final PostVisibilityFilter postVisibilityFilter;
+       private final ReplyVisibilityFilter replyVisibilityFilter;
+
+       @Inject
+       public ListNotificationFilters(@Nonnull PostVisibilityFilter postVisibilityFilter, @Nonnull ReplyVisibilityFilter replyVisibilityFilter) {
+               this.postVisibilityFilter = postVisibilityFilter;
+               this.replyVisibilityFilter = replyVisibilityFilter;
+       }
+
        /**
         * Filters new-post and new-reply notifications in the given list of
         * notifications. If {@code currentSone} is <code>null</code>, new-post and
@@ -46,29 +59,47 @@ public class ListNotificationFilters {
         * itself will be retained in the notifications.
         *
         * @param notifications
-        *            The notifications to filter
+        *              The notifications to filter
         * @param currentSone
-        *            The current Sone, or {@code null} if not logged in
+        *              The current Sone, or {@code null} if not logged in
         * @return The filtered notifications
         */
        @SuppressWarnings("unchecked")
-       public static List<Notification> filterNotifications(Collection<? extends Notification> notifications, Sone currentSone) {
+       public List<Notification> filterNotifications(Collection<? extends Notification> notifications, Sone currentSone) {
                List<Notification> filteredNotifications = new ArrayList<Notification>();
                for (Notification notification : notifications) {
-                       if (notification.getId().equals("new-post-notification")) {
-                               ListNotification<Post> filteredNotification = filterNewPostNotification((ListNotification<Post>) notification, currentSone, true);
-                               if (filteredNotification != null) {
-                                       filteredNotifications.add(filteredNotification);
+                       if (notification.getId().equals("new-sone-notification")) {
+                               if ((currentSone != null) && !currentSone.getOptions().isShowNewSoneNotifications()) {
+                                       continue;
+                               }
+                               filteredNotifications.add(notification);
+                       } else if (notification.getId().equals("new-post-notification")) {
+                               if (currentSone == null) {
+                                       continue;
+                               }
+                               if (!currentSone.getOptions().isShowNewPostNotifications()) {
+                                       continue;
+                               }
+                               Optional<ListNotification<Post>> filteredNotification = filterNewPostNotification((ListNotification<Post>) notification, currentSone);
+                               if (filteredNotification.isPresent()) {
+                                       filteredNotifications.add(filteredNotification.get());
                                }
                        } else if (notification.getId().equals("new-reply-notification")) {
-                               ListNotification<PostReply> filteredNotification = filterNewReplyNotification((ListNotification<PostReply>) notification, currentSone);
-                               if (filteredNotification != null) {
-                                       filteredNotifications.add(filteredNotification);
+                               if (currentSone == null) {
+                                       continue;
+                               }
+                               if (!currentSone.getOptions().isShowNewReplyNotifications()) {
+                                       continue;
+                               }
+                               Optional<ListNotification<PostReply>> filteredNotification =
+                                               filterNewReplyNotification((ListNotification<PostReply>) notification, currentSone);
+                               if (filteredNotification.isPresent()) {
+                                       filteredNotifications.add(filteredNotification.get());
                                }
                        } else if (notification.getId().equals("mention-notification")) {
-                               ListNotification<Post> filteredNotification = filterNewPostNotification((ListNotification<Post>) notification, null, false);
-                               if (filteredNotification != null) {
-                                       filteredNotifications.add(filteredNotification);
+                               Optional<ListNotification<Post>> filteredNotification = filterNewPostNotification((ListNotification<Post>) notification, null);
+                               if (filteredNotification.isPresent()) {
+                                       filteredNotifications.add(filteredNotification.get());
                                }
                        } else {
                                filteredNotifications.add(notification);
@@ -85,35 +116,26 @@ public class ListNotificationFilters {
         * other posts are removed.
         *
         * @param newPostNotification
-        *            The new-post notification
+        *              The new-post notification
         * @param currentSone
-        *            The current Sone, or {@code null} if not logged in
-        * @param soneRequired
-        *            Whether a non-{@code null} Sone in {@code currentSone} is
-        *            required
+        *              The current Sone, or {@code null} if not logged in
         * @return The filtered new-post notification, or {@code null} if the
-        *         notification should be removed
+        * notification should be removed
         */
-       public static ListNotification<Post> filterNewPostNotification(ListNotification<Post> newPostNotification, Sone currentSone, boolean soneRequired) {
-               if (soneRequired && (currentSone == null)) {
-                       return null;
-               }
-               List<Post> newPosts = new ArrayList<Post>();
-               for (Post post : newPostNotification.getElements()) {
-                       if (isPostVisible(currentSone, post)) {
-                               newPosts.add(post);
-                       }
-               }
+       @Nonnull
+       private Optional<ListNotification<Post>> filterNewPostNotification(@Nonnull ListNotification<Post> newPostNotification,
+                       @Nonnull Sone currentSone) {
+               List<Post> newPosts = from(newPostNotification.getElements()).filter(postVisibilityFilter.isVisible(currentSone)).toList();
                if (newPosts.isEmpty()) {
-                       return null;
+                       return Optional.absent();
                }
                if (newPosts.size() == newPostNotification.getElements().size()) {
-                       return newPostNotification;
+                       return Optional.of(newPostNotification);
                }
                ListNotification<Post> filteredNotification = new ListNotification<Post>(newPostNotification);
                filteredNotification.setElements(newPosts);
                filteredNotification.setLastUpdateTime(newPostNotification.getLastUpdatedTime());
-               return filteredNotification;
+               return Optional.of(filteredNotification);
        }
 
        /**
@@ -124,133 +146,25 @@ public class ListNotificationFilters {
         * replies are removed.
         *
         * @param newReplyNotification
-        *            The new-reply notification
+        *              The new-reply notification
         * @param currentSone
-        *            The current Sone, or {@code null} if not logged in
+        *              The current Sone, or {@code null} if not logged in
         * @return The filtered new-reply notification, or {@code null} if the
-        *         notification should be removed
+        * notification should be removed
         */
-       public static ListNotification<PostReply> filterNewReplyNotification(ListNotification<PostReply> newReplyNotification, Sone currentSone) {
-               if (currentSone == null) {
-                       return null;
-               }
-               List<PostReply> newReplies = new ArrayList<PostReply>();
-               for (PostReply reply : newReplyNotification.getElements()) {
-                       if (isReplyVisible(currentSone, reply)) {
-                               newReplies.add(reply);
-                       }
-               }
+       private Optional<ListNotification<PostReply>> filterNewReplyNotification(ListNotification<PostReply> newReplyNotification,
+                       @Nonnull Sone currentSone) {
+               List<PostReply> newReplies = from(newReplyNotification.getElements()).filter(replyVisibilityFilter.isVisible(currentSone)).toList();
                if (newReplies.isEmpty()) {
-                       return null;
+                       return Optional.absent();
                }
                if (newReplies.size() == newReplyNotification.getElements().size()) {
-                       return newReplyNotification;
+                       return Optional.of(newReplyNotification);
                }
                ListNotification<PostReply> filteredNotification = new ListNotification<PostReply>(newReplyNotification);
                filteredNotification.setElements(newReplies);
                filteredNotification.setLastUpdateTime(newReplyNotification.getLastUpdatedTime());
-               return filteredNotification;
-       }
-
-       /**
-        * Checks whether a post is visible to the given Sone. A post is not
-        * considered visible if one of the following statements is true:
-        * <ul>
-        * <li>The post does not have a Sone.</li>
-        * <li>The post’s {@link Post#getTime() time} is in the future.</li>
-        * </ul>
-        * <p>
-        * If {@code post} is not {@code null} more checks are performed, and the
-        * post will be invisible if:
-        * </p>
-        * <ul>
-        * <li>The Sone of the post is not the given Sone, the given Sone does not
-        * follow the post’s Sone, and the given Sone is not the recipient of the
-        * post.</li>
-        * <li>The trust relationship between the two Sones can not be retrieved.</li>
-        * <li>The given Sone has explicitely assigned negative trust to the post’s
-        * Sone.</li>
-        * <li>The given Sone has not explicitely assigned negative trust to the
-        * post’s Sone but the implicit trust is negative.</li>
-        * </ul>
-        * If none of these statements is true the post is considered visible.
-        *
-        * @param sone
-        *            The Sone that checks for a post’s visibility (may be
-        *            {@code null} to skip Sone-specific checks, such as trust)
-        * @param post
-        *            The post to check for visibility
-        * @return {@code true} if the post is considered visible, {@code false}
-        *         otherwise
-        */
-       public static boolean isPostVisible(Sone sone, Post post) {
-               Validation.begin().isNotNull("Post", post).check();
-               Sone postSone = post.getSone();
-               if (postSone == null) {
-                       return false;
-               }
-               if (sone != null) {
-                       Trust trust = postSone.getIdentity().getTrust((OwnIdentity) sone.getIdentity());
-                       if (trust != null) {
-                               if ((trust.getExplicit() != null) && (trust.getExplicit() < 0)) {
-                                       return false;
-                               }
-                               if ((trust.getExplicit() == null) && (trust.getImplicit() != null) && (trust.getImplicit() < 0)) {
-                                       return false;
-                               }
-                       } else {
-                               return false;
-                       }
-                       if ((!postSone.equals(sone)) && !sone.hasFriend(postSone.getId()) && !sone.equals(post.getRecipient())) {
-                               return false;
-                       }
-               }
-               if (post.getTime() > System.currentTimeMillis()) {
-                       return false;
-               }
-               return true;
-       }
-
-       /**
-        * Checks whether a reply is visible to the given Sone. A reply is not
-        * considered visible if one of the following statements is true:
-        * <ul>
-        * <li>The reply does not have a post.</li>
-        * <li>The reply’s post does not have a Sone.</li>
-        * <li>The Sone of the reply’s post is not the given Sone, the given Sone
-        * does not follow the reply’s post’s Sone, and the given Sone is not the
-        * recipient of the reply’s post.</li>
-        * <li>The trust relationship between the two Sones can not be retrieved.</li>
-        * <li>The given Sone has explicitely assigned negative trust to the post’s
-        * Sone.</li>
-        * <li>The given Sone has not explicitely assigned negative trust to the
-        * reply’s post’s Sone but the implicit trust is negative.</li>
-        * <li>The reply’s post’s {@link Post#getTime() time} is in the future.</li>
-        * <li>The reply’s {@link Reply#getTime() time} is in the future.</li>
-        * </ul>
-        * If none of these statements is true the reply is considered visible.
-        *
-        * @param sone
-        *            The Sone that checks for a post’s visibility (may be
-        *            {@code null} to skip Sone-specific checks, such as trust)
-        * @param reply
-        *            The reply to check for visibility
-        * @return {@code true} if the reply is considered visible, {@code false}
-        *         otherwise
-        */
-       public static boolean isReplyVisible(Sone sone, PostReply reply) {
-               Validation.begin().isNotNull("Reply", reply).check();
-               Post post = reply.getPost();
-               if (post == null) {
-                       return false;
-               }
-               if (!isPostVisible(sone, post)) {
-                       return false;
-               }
-               if (reply.getTime() > System.currentTimeMillis()) {
-                       return false;
-               }
-               return true;
+               return Optional.of(filteredNotification);
        }
 
 }