Use all available parameters for post equality.

[Sone.git] / src / main / java / net / pterodactylus / sone / data / Post.java

diff --git a/src/main/java/net/pterodactylus/sone/data/Post.java b/src/main/java/net/pterodactylus/sone/data/Post.java
index fc8a125..58c50e1 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/Post.java
+++ b/src/main/java/net/pterodactylus/sone/data/Post.java
@@ -17,6 +17,9 @@
 
 package net.pterodactylus.sone.data;
 
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.Set;
 import java.util.UUID;
 
 /**
@@ -39,6 +42,9 @@ public class Post {
        /** The text of the post. */
        private final String text;
 
+       /** The replies that have been loaded for this post. */
+       private final Set<Reply> replies = new HashSet<Reply>();
+
        /**
         * Creates a new post.
         *
@@ -124,6 +130,40 @@ public class Post {
                return text;
        }
 
+       /**
+        * Returns all replies to this post in unspecified order.
+        *
+        * @return All replies to this post
+        */
+       public Set<Reply> getReplies() {
+               return Collections.unmodifiableSet(replies);
+       }
+
+       /**
+        * Adds a reply to this post. The reply will not be added if its
+        * {@link Reply#getPost() post} is not equal to this post.
+        *
+        * @param reply
+        *            The reply to add
+        */
+       public void addReply(Reply reply) {
+               if (reply.getPost().equals(this)) {
+                       replies.add(reply);
+               }
+       }
+
+       /**
+        * Removes a reply from this post.
+        *
+        * @param reply
+        *            The reply to remove
+        */
+       public void removeReply(Reply reply) {
+               if (reply.getPost().equals(this)) {
+                       replies.remove(reply);
+               }
+       }
+
        //
        // OBJECT METHODS
        //
@@ -133,7 +173,7 @@ public class Post {
         */
        @Override
        public int hashCode() {
-               return id.hashCode();
+               return id.hashCode() ^ sone.hashCode() ^ (int) (time >>> 32) ^ (int) (time & 0xffffffff) ^ text.hashCode();
        }
 
        /**
@@ -144,7 +184,8 @@ public class Post {
                if (!(object instanceof Post)) {
                        return false;
                }
-               return ((Post) object).id.equals(id);
+               Post post = (Post) object;
+               return post.id.equals(id) && post.sone.equals(sone) && (post.time == time) && post.text.equals(text);
        }
 
 }