X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fnotify%2FListNotification.java;h=039639eb176ee92e96b63d507067d9fca4f1f516;hp=2217b70f75939f2782784336841d8b49d0e87a55;hb=62573c314957b1851f4fbe693b8746686caa940a;hpb=5dfc6973e3f089ddc33327ad42fe25142150c206

diff --git a/src/main/java/net/pterodactylus/sone/notify/ListNotification.java b/src/main/java/net/pterodactylus/sone/notify/ListNotification.java
index 2217b70..039639e 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
@@ -30,7 +30,6 @@ 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 {
 
@@ -51,7 +50,24 @@ public class ListNotification<T> extends TemplateNotification {
 	 *            The template to render
 	 */
 	public ListNotification(String id, String key, Template template) {
-		super(id, template);
+		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);
 	}
@@ -64,7 +80,7 @@ public class ListNotification<T> extends TemplateNotification {
 	 *            The list notification to copy
 	 */
 	public ListNotification(ListNotification<T> listNotification) {
-		super(listNotification.getId(), new Template());
+		super(listNotification.getId(), listNotification.getCreatedTime(), listNotification.getLastUpdatedTime(), listNotification.isDismissable(), new Template());
 		this.key = listNotification.key;
 		getTemplate().add(listNotification.getTemplate());
 		getTemplate().getInitialContext().set(key, elements);
@@ -84,6 +100,19 @@ public class ListNotification<T> extends TemplateNotification {
 	}
 
 	/**
+	 * 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
@@ -111,7 +140,9 @@ 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();
 		}
@@ -131,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);
+	}
+
 }