Don’t notify for replies on posts we don’t have.

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

diff --git a/src/main/java/net/pterodactylus/sone/web/WebInterface.java b/src/main/java/net/pterodactylus/sone/web/WebInterface.java
index 0002e2c..2f3f575 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/WebInterface.java
+++ b/src/main/java/net/pterodactylus/sone/web/WebInterface.java
@@ -37,6 +37,8 @@ import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.freenet.L10nFilter;
 import net.pterodactylus.sone.freenet.wot.Identity;
 import net.pterodactylus.sone.main.SonePlugin;
+import net.pterodactylus.sone.notify.NewPostNotification;
+import net.pterodactylus.sone.notify.NewReplyNotification;
 import net.pterodactylus.sone.notify.NewSoneNotification;
 import net.pterodactylus.sone.template.CollectionAccessor;
 import net.pterodactylus.sone.template.CssClassNameFilter;
@@ -110,6 +112,12 @@ public class WebInterface implements CoreListener {
        /** The “new Sone” notification. */
        private final NewSoneNotification newSoneNotification;
 
+       /** The “new post” notification. */
+       private final NewPostNotification newPostNotification;
+
+       /** The “new reply” notification. */
+       private final NewReplyNotification newReplyNotification;
+
        /**
         * Creates a new web interface.
         *
@@ -143,6 +151,12 @@ public class WebInterface implements CoreListener {
                /* create notifications. */
                Template newSoneNotificationTemplate = templateFactory.createTemplate(createReader("/templates/notify/newSoneNotification.html"));
                newSoneNotification = new NewSoneNotification(newSoneNotificationTemplate);
+
+               Template newPostNotificationTemplate = templateFactory.createTemplate(createReader("/templates/notify/newPostNotification.html"));
+               newPostNotification = new NewPostNotification(newPostNotificationTemplate);
+
+               Template newReplyNotificationTemplate = templateFactory.createTemplate(createReader("/templates/notify/newReplyNotification.html"));
+               newReplyNotification = new NewReplyNotification(newReplyNotificationTemplate);
        }
 
        //
@@ -359,7 +373,8 @@ public class WebInterface implements CoreListener {
         */
        @Override
        public void newPostFound(Post post) {
-               /* TODO */
+               newPostNotification.addPost(post);
+               notificationManager.addNotification(newPostNotification);
        }
 
        /**
@@ -367,7 +382,11 @@ public class WebInterface implements CoreListener {
         */
        @Override
        public void newReplyFound(Reply reply) {
-               /* TODO */
+               if (reply.getPost().getSone() == null) {
+                       return;
+               }
+               newReplyNotification.addReply(reply);
+               notificationManager.addNotification(newReplyNotification);
        }
 
        /**
@@ -386,7 +405,10 @@ public class WebInterface implements CoreListener {
         */
        @Override
        public void markPostKnown(Post post) {
-               /* TODO */
+               newPostNotification.removePost(post);
+               if (newPostNotification.isEmpty()) {
+                       newPostNotification.dismiss();
+               }
        }
 
        /**
@@ -394,7 +416,10 @@ public class WebInterface implements CoreListener {
         */
        @Override
        public void markReplyKnown(Reply reply) {
-               /* TODO */
+               newReplyNotification.removeReply(reply);
+               if (newReplyNotification.isEmpty()) {
+                       newReplyNotification.dismiss();
+               }
        }
 
        /**