Add “DeletePost” FCP command.

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

diff --git a/src/main/java/net/pterodactylus/sone/fcp/FcpInterface.java b/src/main/java/net/pterodactylus/sone/fcp/FcpInterface.java
index 3a64bbf..bcd70a2 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/fcp/FcpInterface.java
+++ b/src/main/java/net/pterodactylus/sone/fcp/FcpInterface.java
@@ -20,12 +20,16 @@ package net.pterodactylus.sone.fcp;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
 import net.pterodactylus.sone.core.Core;
 import net.pterodactylus.sone.freenet.fcp.Command;
-import net.pterodactylus.sone.freenet.fcp.FcpException;
 import net.pterodactylus.sone.freenet.fcp.Command.AccessType;
+import net.pterodactylus.sone.freenet.fcp.Command.ErrorResponse;
 import net.pterodactylus.sone.freenet.fcp.Command.Response;
+import net.pterodactylus.sone.freenet.fcp.FcpException;
+import net.pterodactylus.util.logging.Logging;
 import freenet.pluginmanager.FredPluginFCP;
 import freenet.pluginmanager.PluginNotFoundException;
 import freenet.pluginmanager.PluginReplySender;
@@ -40,6 +44,12 @@ import freenet.support.api.Bucket;
  */
 public class FcpInterface {
 
+       /** The logger. */
+       private static final Logger logger = Logging.getLogger(FcpInterface.class);
+
+       /** Whether the FCP interface is currently active. */
+       private volatile boolean active;
+
        /** All available FCP commands. */
        private final Map<String, Command> commands = Collections.synchronizedMap(new HashMap<String, Command>());
 
@@ -52,9 +62,36 @@ public class FcpInterface {
        public FcpInterface(Core core) {
                commands.put("Version", new VersionCommand());
                commands.put("GetLocalSones", new GetLocalSonesCommand(core));
+               commands.put("GetPost", new GetPostCommand(core));
+               commands.put("GetPosts", new GetPostsCommand(core));
                commands.put("GetPostFeed", new GetPostFeedCommand(core));
+               commands.put("LikePost", new LikePostCommand(core));
+               commands.put("LikeReply", new LikeReplyCommand(core));
+               commands.put("CreatePost", new CreatePostCommand(core));
+               commands.put("CreateReply", new CreateReplyCommand(core));
+               commands.put("DeletePost", new DeletePostCommand(core));
        }
 
+       //
+       // ACCESSORS
+       //
+
+       /**
+        * Sets whether the FCP interface should handle requests. If {@code active}
+        * is {@code false}, all requests are answered with an error.
+        *
+        * @param active
+        *            {@code true} to activate the FCP interface, {@code false} to
+        *            deactivate the FCP interface
+        */
+       public void setActive(boolean active) {
+               this.active = active;
+       }
+
+       //
+       // ACTIONS
+       //
+
        /**
         * Handles a plugin FCP request.
         *
@@ -70,31 +107,63 @@ public class FcpInterface {
         *            {@link FredPluginFCP#ACCESS_FCP_RESTRICTED}
         */
        public void handle(PluginReplySender pluginReplySender, SimpleFieldSet parameters, Bucket data, int accessType) {
-               Command command = commands.get(parameters.get("Message"));
-               if (command == null) {
-                       /* TODO - return error? */
-                       return;
-               }
-               String identifier = parameters.get("Identifier");
-               if ((identifier == null) || (identifier.length() == 0)) {
-                       /* TODO - return error? */
+               if (!active) {
+                       try {
+                               sendReply(pluginReplySender, null, new ErrorResponse(400, "FCP Interface deactivated"));
+                       } catch (PluginNotFoundException pnfe1) {
+                               logger.log(Level.FINE, "Could not set error to plugin.", pnfe1);
+                       }
                        return;
                }
+               Command command = commands.get(parameters.get("Message"));
                try {
-                       Response reply = command.execute(parameters, data, AccessType.values()[accessType]);
-                       SimpleFieldSet replyParameters = reply.getReplyParameters();
-                       replyParameters.putOverwrite("Identifier", identifier);
-                       if (reply.hasData()) {
-                               pluginReplySender.send(replyParameters, reply.getData());
-                       } else if (reply.hasBucket()) {
-                               pluginReplySender.send(replyParameters, reply.getBucket());
-                       } else {
-                               pluginReplySender.send(replyParameters);
+                       if (command == null) {
+                               sendReply(pluginReplySender, null, new ErrorResponse("Unrecognized Message: " + parameters.get("Message")));
+                               return;
+                       }
+                       String identifier = parameters.get("Identifier");
+                       if ((identifier == null) || (identifier.length() == 0)) {
+                               sendReply(pluginReplySender, null, new ErrorResponse("Missing Identifier."));
+                               return;
+                       }
+                       try {
+                               Response response = command.execute(parameters, data, AccessType.values()[accessType]);
+                               sendReply(pluginReplySender, identifier, response);
+                       } catch (FcpException fe1) {
+                               sendReply(pluginReplySender, identifier, new ErrorResponse("Error executing command: " + fe1.getMessage()));
                        }
-               } catch (FcpException fe1) {
-                       /* TODO - log, report */
                } catch (PluginNotFoundException pnfe1) {
-                       /* TODO - log */
+                       logger.log(Level.WARNING, "Could not find destination plugin: " + pluginReplySender);
+               }
+       }
+
+       //
+       // PRIVATE METHODS
+       //
+
+       /**
+        * Sends the given response to the given plugin.
+        *
+        * @param pluginReplySender
+        *            The reply sender
+        * @param identifier
+        *            The identifier (may be {@code null})
+        * @param response
+        *            The response to send
+        * @throws PluginNotFoundException
+        *             if the plugin can not be found
+        */
+       private void sendReply(PluginReplySender pluginReplySender, String identifier, Response response) throws PluginNotFoundException {
+               SimpleFieldSet replyParameters = response.getReplyParameters();
+               if (identifier != null) {
+                       replyParameters.putOverwrite("Identifier", identifier);
+               }
+               if (response.hasData()) {
+                       pluginReplySender.send(replyParameters, response.getData());
+               } else if (response.hasBucket()) {
+                       pluginReplySender.send(replyParameters, response.getBucket());
+               } else {
+                       pluginReplySender.send(replyParameters);
                }
        }