Store all replies of a Sone in the Sone.

diff --git a/src/main/java/net/pterodactylus/sone/data/Sone.java b/src/main/java/net/pterodactylus/sone/data/Sone.java
index dbe1b35..6c50462 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/Sone.java
+++ b/src/main/java/net/pterodactylus/sone/data/Sone.java
@@ -58,6 +58,9 @@ public class Sone {
        /** All posts. */
        private final List<Post> posts = new ArrayList<Post>();
 
+       /** All replies. */
+       private final Set<Reply> replies = new HashSet<Reply>();
+
        /** Modification count. */
        private volatile long modificationCounter = 0;
 
@@ -242,6 +245,40 @@ public class Sone {
        }
 
        /**
+        * Returns all replies this Sone made.
+        *
+        * @return All replies this Sone made
+        */
+       public Set<Reply> getReplies() {
+               return Collections.unmodifiableSet(replies);
+       }
+
+       /**
+        * Adds a reply to this Sone. If the given reply was not made by this Sone,
+        * nothing is added to this Sone.
+        *
+        * @param reply
+        *            The reply to add
+        */
+       public synchronized void addReply(Reply reply) {
+               if (reply.getSone().equals(this) && replies.add(reply)) {
+                       modificationCounter++;
+               }
+       }
+
+       /**
+        * Removes a reply from this Sone.
+        *
+        * @param reply
+        *            The reply to remove
+        */
+       public synchronized void removeReply(Reply reply) {
+               if (reply.getSone().equals(this) && replies.remove(reply)) {
+                       modificationCounter++;
+               }
+       }
+
+       /**
         * Returns the modification counter.
         *
         * @return The modification counter