X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FPost.java;h=a42cfaa9de617f2007f8792946d42097a505ac3f;hb=7b415d1019795fa39d9c9727c3123e638a021ffa;hp=fc8a125805eca92f727f47d3ebbf28a71dad400f;hpb=ed360e7532ffc1c1f2cc77fb41fed7cc0557dd77;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..a42cfaa 100644
--- a/src/main/java/net/pterodactylus/sone/data/Post.java
+++ b/src/main/java/net/pterodactylus/sone/data/Post.java
@@ -17,6 +17,10 @@
 
 package net.pterodactylus.sone.data;
 
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.Set;
 import java.util.UUID;
 
 /**
@@ -39,6 +43,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 +131,40 @@ public class Post {
 		return text;
 	}
 
+	/**
+	 * Returns all replies to this post in unspecified order.
+	 *
+	 * @return All replies to this post
+	 */
+	public Collection<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
 	//