X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffcp%2FFcpInterface.java;h=01fae788ef6ead9139eb90a6f3e00a0cd12d166e;hb=6798ab08f87d387a7bddc0e3e41c08c8b6f4095d;hp=8841f638bd9ab88b0c6455c1eec76ad202304993;hpb=4ff19298a886f9aa09ef20309a833f9454284482;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/fcp/FcpInterface.java b/src/main/java/net/pterodactylus/sone/fcp/FcpInterface.java
index 8841f63..01fae78 100644
--- 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.Reply;
+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>());
 
@@ -51,9 +61,38 @@ 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));
+		commands.put("DeleteReply", new DeleteReplyCommand(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.
 	 *
@@ -69,31 +108,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 {
-			Reply 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);
 		}
 	}