X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffcp%2FAbstractSoneCommand.java;h=ba3bc88fc5bdcc3df830ab82eca060fe96e5e59f;hb=ef7bc5f24ecedb78122d0a314bb20b7eacea6e43;hp=a8d851446fa4d29e4a86e3dc298388a1d747b2f4;hpb=698fdb0cf7923e58d0fb5dc3695b2e1b811c0816;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
index a8d8514..ba3bc88 100644
--- a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
+++ b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
@@ -21,12 +21,14 @@ import java.util.Collection;
 
 import net.pterodactylus.sone.core.Core;
 import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.Reply;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.freenet.SimpleFieldSetBuilder;
 import net.pterodactylus.sone.freenet.fcp.AbstractCommand;
 import net.pterodactylus.sone.freenet.fcp.Command;
 import net.pterodactylus.sone.freenet.fcp.FcpException;
 import net.pterodactylus.sone.template.SoneAccessor;
+import net.pterodactylus.util.filter.Filters;
 import freenet.node.FSParseException;
 import freenet.support.SimpleFieldSet;
 
@@ -95,6 +97,32 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
 	}
 
 	/**
+	 * Returns a post whose ID is a parameter in the given simple field set.
+	 *
+	 * @param simpleFieldSet
+	 *            The simple field set containing the ID of the post
+	 * @param parameterName
+	 *            The name under which the post ID is stored in the simple field
+	 *            set
+	 * @return The post
+	 * @throws FcpException
+	 *             if there is no post ID stored under the given parameter name,
+	 *             or if the post ID is invalid
+	 */
+	protected Post getPost(SimpleFieldSet simpleFieldSet, String parameterName) throws FcpException {
+		try {
+			String postId = simpleFieldSet.getString(parameterName);
+			Post post = core.getPost(postId, false);
+			if (post == null) {
+				throw new FcpException("Could not load post from “" + postId + "”.");
+			}
+			return post;
+		} catch (FSParseException fspe1) {
+			throw new FcpException("Could not post ID from “" + parameterName + "”.", fspe1);
+		}
+	}
+
+	/**
 	 * Creates a simple field set from the given collection of Sones.
 	 *
 	 * @param sones
@@ -122,9 +150,12 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
 	 *
 	 * @param posts
 	 *            The posts to encode
+	 * @param includeReplies
+	 *            {@code true} to include the replies, {@code false} to not
+	 *            include the replies
 	 * @return The simple field set containing the posts
 	 */
-	public SimpleFieldSet encodePosts(Collection<? extends Post> posts) {
+	public SimpleFieldSet encodePosts(Collection<? extends Post> posts, boolean includeReplies) {
 		SimpleFieldSetBuilder postBuilder = new SimpleFieldSetBuilder();
 
 		int postIndex = 0;
@@ -138,9 +169,38 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
 			}
 			postBuilder.put(postPrefix + ".Time", post.getTime());
 			postBuilder.put(postPrefix + ".Text", post.getText());
+			if (includeReplies) {
+				postBuilder.put(encodeReplies(Filters.filteredList(core.getReplies(post), Reply.FUTURE_REPLIES_FILTER), postPrefix + "."));
+			}
 		}
 
 		return postBuilder.get();
 	}
 
+	/**
+	 * Creates a simple field set from the given collection of replies.
+	 *
+	 * @param replies
+	 *            The replies to encode
+	 * @param prefix
+	 *            The prefix for the field names (may be empty, but not
+	 *            {@code null})
+	 * @return The simple field set containing the replies
+	 */
+	public SimpleFieldSet encodeReplies(Collection<? extends Reply> replies, String prefix) {
+		SimpleFieldSetBuilder replyBuilder = new SimpleFieldSetBuilder();
+
+		int replyIndex = 0;
+		replyBuilder.put(prefix + "Replies.Count", replies.size());
+		for (Reply reply : replies) {
+			String replyPrefix = prefix + "Replies." + replyIndex++;
+			replyBuilder.put(replyPrefix + ".ID", reply.getId());
+			replyBuilder.put(replyPrefix + ".Sone", reply.getSone().getId());
+			replyBuilder.put(replyPrefix + ".Time", reply.getTime());
+			replyBuilder.put(replyPrefix + ".Text", reply.getText());
+		}
+
+		return replyBuilder.get();
+	}
+
 }