Add “LikeReply” FCP command.

[Sone.git] / src / main / java / net / pterodactylus / sone / fcp / AbstractSoneCommand.java

diff --git a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
index 6473e4d..b183121 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
+++ b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
@@ -124,6 +124,32 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
        }
 
        /**
+        * Returns a reply whose ID is a parameter in the given simple field set.
+        *
+        * @param simpleFieldSet
+        *            The simple field set containing the ID of the reply
+        * @param parameterName
+        *            The name under which the reply ID is stored in the simple
+        *            field set
+        * @return The reply
+        * @throws FcpException
+        *             if there is no reply ID stored under the given parameter
+        *             name, or if the reply ID is invalid
+        */
+       protected Reply getReply(SimpleFieldSet simpleFieldSet, String parameterName) throws FcpException {
+               try {
+                       String replyId = simpleFieldSet.getString(parameterName);
+                       Reply reply = core.getReply(replyId, false);
+                       if (reply == null) {
+                               throw new FcpException("Could not load reply from “" + replyId + "”.");
+                       }
+                       return reply;
+               } catch (FSParseException fspe1) {
+                       throw new FcpException("Could not reply ID from “" + parameterName + "”.", fspe1);
+               }
+       }
+
+       /**
         * Creates a simple field set from the given collection of Sones.
         *
         * @param sones
@@ -172,6 +198,7 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
                }
                postBuilder.put(prefix + "Time", post.getTime());
                postBuilder.put(prefix + "Text", post.getText());
+               postBuilder.put(encodeLikes(core.getLikes(post), prefix + "Likes."));
 
                if (includeReplies) {
                        List<Reply> replies = core.getReplies(post);
@@ -236,4 +263,28 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
                return replyBuilder.get();
        }
 
+       /**
+        * Creates a simple field set from the given collection of Sones that like
+        * an element.
+        *
+        * @param likes
+        *            The liking Sones
+        * @param prefix
+        *            The prefix for the field names (may be empty but not
+        *            {@code null})
+        * @return The simple field set containing the likes
+        */
+       protected SimpleFieldSet encodeLikes(Collection<? extends Sone> likes, String prefix) {
+               SimpleFieldSetBuilder likesBuilder = new SimpleFieldSetBuilder();
+
+               int likeIndex = 0;
+               likesBuilder.put(prefix + "Count", likes.size());
+               for (Sone sone : likes) {
+                       String sonePrefix = prefix + likeIndex++ + ".";
+                       likesBuilder.put(sonePrefix + "ID", sone.getId());
+               }
+
+               return likesBuilder.get();
+       }
+
 }