Move post creation back to CreatePostPage.

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

diff --git a/src/main/java/net/pterodactylus/sone/data/Sone.java b/src/main/java/net/pterodactylus/sone/data/Sone.java
index b70dbdb..dbe1b35 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/Sone.java
+++ b/src/main/java/net/pterodactylus/sone/data/Sone.java
@@ -17,8 +17,10 @@
 
 package net.pterodactylus.sone.data;
 
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
 import java.util.UUID;
 
@@ -53,6 +55,9 @@ public class Sone {
        /** All friend Sones. */
        private final Set<Sone> friendSones = new HashSet<Sone>();
 
+       /** All posts. */
+       private final List<Post> posts = new ArrayList<Post>();
+
        /** Modification count. */
        private volatile long modificationCounter = 0;
 
@@ -203,6 +208,40 @@ public class Sone {
        }
 
        /**
+        * Returns the list of posts of this Sone.
+        *
+        * @return All posts of this Sone
+        */
+       public List<Post> getPosts() {
+               return Collections.unmodifiableList(posts);
+       }
+
+       /**
+        * Adds the given post to this Sone. The post will not be added if its
+        * {@link Post#getSone() Sone} is not this Sone.
+        *
+        * @param post
+        *            The post to add
+        */
+       public synchronized void addPost(Post post) {
+               if (post.getSone().equals(this) && posts.add(post)) {
+                       modificationCounter++;
+               }
+       }
+
+       /**
+        * Removes the given post from this Sone.
+        *
+        * @param post
+        *            The post to remove
+        */
+       public synchronized void removePost(Post post) {
+               if (post.getSone().equals(this) && posts.remove(post)) {
+                       modificationCounter++;
+               }
+       }
+
+       /**
         * Returns the modification counter.
         *
         * @return The modification counter