Store the originating Sone in the post.

diff --git a/src/main/java/net/pterodactylus/sone/data/Post.java b/src/main/java/net/pterodactylus/sone/data/Post.java
index 3c02a1c..456b205 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/Post.java
+++ b/src/main/java/net/pterodactylus/sone/data/Post.java
@@ -25,6 +25,9 @@ package net.pterodactylus.sone.data;
  */
 public class Post {
 
+       /** The Sone this post belongs to. */
+       private final Sone sone;
+
        /** The time of the post (in milliseconds since Jan 1, 1970 UTC). */
        private final long time;
 
@@ -34,17 +37,29 @@ public class Post {
        /**
         * Creates a new post.
         *
+        * @param sone
+        *            The Sone this post belongs to
         * @param time
         *            The time of the post (in milliseconds since Jan 1, 1970 UTC)
         * @param text
         *            The text of the post
         */
-       public Post(long time, String text) {
+       public Post(Sone sone, long time, String text) {
+               this.sone = sone;
                this.time = time;
                this.text = text;
        }
 
        /**
+        * Returns the Sone this post belongs to.
+        *
+        * @return The Sone of this post
+        */
+       public Sone getSone() {
+               return sone;
+       }
+
+       /**
         * Returns the time of the post.
         *
         * @return The time of the post (in milliseconds since Jan 1, 1970 UTC)

diff --git a/src/main/java/net/pterodactylus/sone/data/Sone.java b/src/main/java/net/pterodactylus/sone/data/Sone.java
index 33b97a0..27c7852 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++;
                }

diff --git a/src/main/java/net/pterodactylus/sone/web/CreatePostPage.java b/src/main/java/net/pterodactylus/sone/web/CreatePostPage.java
index b6b291e..44e0d21 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/CreatePostPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/CreatePostPage.java
@@ -51,8 +51,7 @@ public class CreatePostPage extends SoneTemplatePage {
                super.processTemplate(request, template);
                String text = request.getHttpRequest().getPartAsStringFailsafe("text", 65536).trim();
                if (text.length() != 0) {
-                       Post post = new Post(System.currentTimeMillis(), text);
-                       getCurrentSone(request.getToadletContext()).addPost(post);
+                       getCurrentSone(request.getToadletContext()).addPost(text);
                        throw new RedirectException("index.html");
                }
                template.set("errorTextEmpty", true);