X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FPost.java;h=f7fbcd3ee11bda6b231d6af24f84df83bb9319d8;hb=8d298562a71682e0bbcdc311e741e00f85490a7d;hp=fc8a125805eca92f727f47d3ebbf28a71dad400f;hpb=f637fce46e6468595c70daee9a02f36bb37c57e3;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/data/Post.java b/src/main/java/net/pterodactylus/sone/data/Post.java
index fc8a125..f7fbcd3 100644
--- 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
 	//