Bring image-management up to speed.

[Sone.git] / src / main / java / net / pterodactylus / sone / notify / ListNotification.java

diff --git a/src/main/java/net/pterodactylus/sone/notify/ListNotification.java b/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
index 7a0d6f8..c66d376 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
+++ b/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
@@ -18,8 +18,9 @@
 package net.pterodactylus.sone.notify;
 
 import java.util.ArrayList;
-import java.util.Collections;
+import java.util.Collection;
 import java.util.List;
+import java.util.concurrent.CopyOnWriteArrayList;
 
 import net.pterodactylus.util.notify.TemplateNotification;
 import net.pterodactylus.util.template.Template;
@@ -33,8 +34,11 @@ import net.pterodactylus.util.template.Template;
  */
 public class ListNotification<T> extends TemplateNotification {
 
+       /** The key under which to store the elements in the template. */
+       private final String key;
+
        /** The list of new elements. */
-       private final List<T> elements = Collections.synchronizedList(new ArrayList<T>());
+       private final List<T> elements = new CopyOnWriteArrayList<T>();
 
        /**
         * Creates a new list notification.
@@ -47,8 +51,40 @@ public class ListNotification<T> extends TemplateNotification {
         *            The template to render
         */
        public ListNotification(String id, String key, Template template) {
-               super(id, template);
-               template.set(key, elements);
+               this(id, key, template, true);
+       }
+
+       /**
+        * Creates a new list notification.
+        *
+        * @param id
+        *            The ID of the notification
+        * @param key
+        *            The key under which to store the elements in the template
+        * @param template
+        *            The template to render
+        * @param dismissable
+        *            {@code true} if this notification should be dismissable by the
+        *            user, {@code false} otherwise
+        */
+       public ListNotification(String id, String key, Template template, boolean dismissable) {
+               super(id, System.currentTimeMillis(), System.currentTimeMillis(), dismissable, template);
+               this.key = key;
+               template.getInitialContext().set(key, elements);
+       }
+
+       /**
+        * Creates a new list notification that copies its ID and the template from
+        * the given list notification.
+        *
+        * @param listNotification
+        *            The list notification to copy
+        */
+       public ListNotification(ListNotification<T> listNotification) {
+               super(listNotification.getId(), listNotification.getCreatedTime(), listNotification.getLastUpdatedTime(), listNotification.isDismissable(), new Template());
+               this.key = listNotification.key;
+               getTemplate().add(listNotification.getTemplate());
+               getTemplate().getInitialContext().set(key, elements);
        }
 
        //
@@ -65,6 +101,18 @@ public class ListNotification<T> extends TemplateNotification {
        }
 
        /**
+        * Sets the elements to show in this notification.
+        *
+        * @param elements
+        *            The elements to show
+        */
+       public void setElements(Collection<? extends T> elements) {
+               this.elements.clear();
+               this.elements.addAll(elements);
+               touch();
+       }
+
+       /**
         * Returns whether there are any new elements.
         *
         * @return {@code true} if there are no new elements, {@code false} if there