From e125716f34a5136d0ca0ce56edc4658818ec48a0 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@freenetproject.org>
Date: Fri, 21 Aug 2015 19:15:31 +0200
Subject: [PATCH] Add method to load plugin from file

---
 .../fcp/quelaton/LoadPluginCommand.java            |  1 +
 .../fcp/quelaton/LoadPluginCommandImpl.java        |  7 +++++
 .../fcp/quelaton/DefaultFcpClientTest.java         | 33 ++++++++++++++++++++++
 3 files changed, 41 insertions(+)

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommand.java b/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommand.java
index 0432f14..959f559 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommand.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommand.java
@@ -14,5 +14,6 @@ public interface LoadPluginCommand {
 	LoadPluginCommand addToConfig();
 	Executable<Optional<PluginInfo>> officialFromFreenet(String pluginIdentifier);
 	Executable<Optional<PluginInfo>> officialFromHttps(String pluginIdentifier);
+	Executable<Optional<PluginInfo>> fromFile(String filename);
 
 }
diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommandImpl.java
index eccb10f..209e394 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/LoadPluginCommandImpl.java
@@ -56,6 +56,13 @@ public class LoadPluginCommandImpl implements LoadPluginCommand {
 		return this::execute;
 	}
 
+	@Override
+	public Executable<Optional<PluginInfo>> fromFile(String filename) {
+		loadPlugin.setUrlType(UrlType.FILE);
+		loadPlugin.setPluginUrl(filename);
+		return this::execute;
+	}
+
 	private ListenableFuture<Optional<PluginInfo>> execute() {
 		return threadPool.submit(this::executeDialog);
 	}
diff --git a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
index f54298f..5d70ff2 100644
--- a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
@@ -2093,6 +2093,39 @@ public class DefaultFcpClientTest {
 	}
 
 	@Test
+	public void defaultFcpClientCanLoadPluginFromFile() throws ExecutionException, InterruptedException,
+	IOException {
+		Future<Optional<PluginInfo>> pluginInfo = fcpClient.loadPlugin().fromFile("/path/to/plugin.jar").execute();
+		connectNode();
+		List<String> lines = fcpServer.collectUntil(is("EndMessage"));
+		String identifier = extractIdentifier(lines);
+		assertThat(lines, matchesFcpMessage(
+			"LoadPlugin",
+			"Identifier=" + identifier,
+			"PluginURL=/path/to/plugin.jar",
+			"URLType=file",
+			"EndMessage"
+		));
+		fcpServer.writeLine(
+			"PluginInfo",
+			"Identifier=" + identifier,
+			"PluginName=superPlugin",
+			"IsTalkable=true",
+			"LongVersion=1.2.3",
+			"Version=42",
+			"OriginUri=superPlugin",
+			"Started=true",
+			"EndMessage"
+		);
+		assertThat(pluginInfo.get().get().getPluginName(), is("superPlugin"));
+		assertThat(pluginInfo.get().get().getOriginalURI(), is("superPlugin"));
+		assertThat(pluginInfo.get().get().isTalkable(), is(true));
+		assertThat(pluginInfo.get().get().getVersion(), is("42"));
+		assertThat(pluginInfo.get().get().getLongVersion(), is("1.2.3"));
+		assertThat(pluginInfo.get().get().isStarted(), is(true));
+	}
+
+	@Test
 	public void failedLoadingPluginIsRecognized() throws ExecutionException, InterruptedException,
 	IOException {
 		Future<Optional<PluginInfo>> pluginInfo = fcpClient.loadPlugin().officialFromFreenet("superPlugin").execute();
-- 
2.7.4