X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fnotify%2FListNotification.java;h=c66d376f4836cad76d08f4d8b6878d16a612849d;hb=58eeba9b521b0a0094ac90a37fd88811c6a27376;hp=64c2ecbac0240a19f0bff32d1ef431e5fd7bf612;hpb=2889127f645d17c4ae0a650e935d1481d9cbd727;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/notify/ListNotification.java b/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
index 64c2ecb..c66d376 100644
--- 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);
 	}
 
 	//
@@ -56,6 +92,27 @@ public class ListNotification<T> extends TemplateNotification {
 	//
 
 	/**
+	 * Returns the current list of elements.
+	 *
+	 * @return The current list of elements
+	 */
+	public List<T> getElements() {
+		return new ArrayList<T>(elements);
+	}
+
+	/**
+	 * 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
@@ -84,6 +141,9 @@ public class ListNotification<T> extends TemplateNotification {
 	 */
 	public void remove(T element) {
 		elements.remove(element);
+		if (elements.isEmpty()) {
+			dismiss();
+		}
 		touch();
 	}