X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffcp%2FFcpInterface.java;h=8854347e106b38f4100674dc68ebe61a18800cd7;hb=96e4cdd1c5a2dc719afbf1f1cc14a3c62e93f432;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..8854347 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,9 @@ import freenet.support.api.Bucket;
  */
 public class FcpInterface {
 
+	/** The logger. */
+	private static final Logger logger = Logging.getLogger(FcpInterface.class);
+
 	/** All available FCP commands. */
 	private final Map<String, Command> commands = Collections.synchronizedMap(new HashMap<String, Command>());
 
@@ -51,6 +58,8 @@ public class FcpInterface {
 	 */
 	public FcpInterface(Core core) {
 		commands.put("Version", new VersionCommand());
+		commands.put("GetLocalSones", new GetLocalSonesCommand(core));
+		commands.put("GetPosts", new GetPostsCommand(core));
 		commands.put("GetPostFeed", new GetPostFeedCommand(core));
 	}
 
@@ -70,30 +79,54 @@ public class FcpInterface {
 	 */
 	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? */
-			return;
-		}
 		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);
 		}
 	}