From: David ‘Bombe’ Roden <bombe@pterodactylus.net>
Date: Thu, 14 Oct 2010 10:15:31 +0000 (+0200)
Subject: Store replies in post.
X-Git-Tag: 0.1-RC1~410
X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=commitdiff_plain;h=7b415d1019795fa39d9c9727c3123e638a021ffa

Store replies in post.
---

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
 	//