Check if post belongs to this Sone before trying to delete it.

[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 33b97a0..a40b12e 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/Sone.java
+++ b/src/main/java/net/pterodactylus/sone/data/Sone.java
@@ -217,12 +217,13 @@ public class Sone {
        }
 
        /**
-        * Adds the given post to this Sone.
+        * Adds a post with the given text to this Sone.
         *
-        * @param post
-        *            The post to add
+        * @param text
+        *            The text to post
         */
-       public synchronized void addPost(Post post) {
+       public synchronized void addPost(String text) {
+               Post post = new Post(this, System.currentTimeMillis(), text);
                if (posts.add(post)) {
                        modificationCounter++;
                }
@@ -235,7 +236,7 @@ public class Sone {
         *            The post to remove
         */
        public synchronized void removePost(Post post) {
-               if (posts.remove(post)) {
+               if (post.getSone().equals(this) && posts.remove(post)) {
                        modificationCounter++;
                }
        }