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

diff --git a/src/main/java/net/pterodactylus/sone/notify/ListNotification.java b/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
index 64c2ecb..11b0829 100644
--- a/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
+++ b/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
@@ -1,5 +1,5 @@
 /*
- * Sone - ListNotification.java - Copyright © 2010 David Roden
+ * Sone - ListNotification.java - Copyright © 2010–2016 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
@@ -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;
@@ -29,12 +30,14 @@ import net.pterodactylus.util.template.Template;
  *
  * @param <T>
  *            The type of the items
- * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 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<>();
 
 	/**
 	 * Creates a new list notification.
@@ -47,8 +50,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 +91,28 @@ 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<>(elements);
+	}
+
+	/**
+	 * Sets the elements to show in this notification. This method will not call
+	 * {@link #touch()}.
+	 *
+	 * @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
@@ -83,7 +140,12 @@ public class ListNotification<T> extends TemplateNotification {
 	 *            The element to remove
 	 */
 	public void remove(T element) {
-		elements.remove(element);
+		while (elements.remove(element)) {
+			/* do nothing, just remove all instances of the element. */
+		}
+		if (elements.isEmpty()) {
+			dismiss();
+		}
 		touch();
 	}
 
@@ -100,4 +162,38 @@ public class ListNotification<T> extends TemplateNotification {
 		elements.clear();
 	}
 
+	//
+	// OBJECT METHODS
+	//
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public int hashCode() {
+		int hashCode = super.hashCode();
+		for (T element : elements) {
+			hashCode ^= element.hashCode();
+		}
+		return hashCode;
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public boolean equals(Object object) {
+		if (!(object instanceof ListNotification)) {
+			return false;
+		}
+		ListNotification<?> listNotification = (ListNotification<?>) object;
+		if (!super.equals(listNotification)) {
+			return false;
+		}
+		if (!key.equals(listNotification.key)) {
+			return false;
+		}
+		return elements.equals(listNotification.elements);
+	}
+
 }