X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FSone.java;h=6c50462afad246325047aec5f043c225b4e958d3;hb=8bc55daff94063bc5cb6b1d240e408aac836f82b;hp=b70dbdbc0bc3e5586d65eafbe6c1e2c628bc1179;hpb=50f89617d6c0b65e4ddce2daf65e5be7a075e5ea;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/data/Sone.java b/src/main/java/net/pterodactylus/sone/data/Sone.java
index b70dbdb..6c50462 100644
--- a/src/main/java/net/pterodactylus/sone/data/Sone.java
+++ b/src/main/java/net/pterodactylus/sone/data/Sone.java
@@ -17,8 +17,10 @@
 
 package net.pterodactylus.sone.data;
 
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
 import java.util.UUID;
 
@@ -53,6 +55,12 @@ public class Sone {
 	/** All friend Sones. */
 	private final Set<Sone> friendSones = new HashSet<Sone>();
 
+	/** All posts. */
+	private final List<Post> posts = new ArrayList<Post>();
+
+	/** All replies. */
+	private final Set<Reply> replies = new HashSet<Reply>();
+
 	/** Modification count. */
 	private volatile long modificationCounter = 0;
 
@@ -203,6 +211,74 @@ public class Sone {
 	}
 
 	/**
+	 * Returns the list of posts of this Sone.
+	 *
+	 * @return All posts of this Sone
+	 */
+	public List<Post> getPosts() {
+		return Collections.unmodifiableList(posts);
+	}
+
+	/**
+	 * Adds the given post to this Sone. The post will not be added if its
+	 * {@link Post#getSone() Sone} is not this Sone.
+	 *
+	 * @param post
+	 *            The post to add
+	 */
+	public synchronized void addPost(Post post) {
+		if (post.getSone().equals(this) && posts.add(post)) {
+			modificationCounter++;
+		}
+	}
+
+	/**
+	 * Removes the given post from this Sone.
+	 *
+	 * @param post
+	 *            The post to remove
+	 */
+	public synchronized void removePost(Post post) {
+		if (post.getSone().equals(this) && posts.remove(post)) {
+			modificationCounter++;
+		}
+	}
+
+	/**
+	 * Returns all replies this Sone made.
+	 *
+	 * @return All replies this Sone made
+	 */
+	public Set<Reply> getReplies() {
+		return Collections.unmodifiableSet(replies);
+	}
+
+	/**
+	 * Adds a reply to this Sone. If the given reply was not made by this Sone,
+	 * nothing is added to this Sone.
+	 *
+	 * @param reply
+	 *            The reply to add
+	 */
+	public synchronized void addReply(Reply reply) {
+		if (reply.getSone().equals(this) && replies.add(reply)) {
+			modificationCounter++;
+		}
+	}
+
+	/**
+	 * Removes a reply from this Sone.
+	 *
+	 * @param reply
+	 *            The reply to remove
+	 */
+	public synchronized void removeReply(Reply reply) {
+		if (reply.getSone().equals(this) && replies.remove(reply)) {
+			modificationCounter++;
+		}
+	}
+
+	/**
 	 * Returns the modification counter.
 	 *
 	 * @return The modification counter