From 6090cca7d3eedfc2c8e41c8ff162f4705519705c Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@freenetproject.org>
Date: Fri, 16 Oct 2015 08:30:50 +0200
Subject: [PATCH] Send generated URIs on ClientPutDiskDir to registered
 consumers

---
 .../pterodactylus/fcp/quelaton/ClientPutDiskDirCommand.java |  1 +
 .../fcp/quelaton/ClientPutDiskDirCommandImpl.java           | 13 +++++++++++++
 .../pterodactylus/fcp/quelaton/DefaultFcpClientTest.java    | 12 ++++++++++++
 3 files changed, 26 insertions(+)

diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommand.java b/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommand.java
index dad6750..1bcd116 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommand.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommand.java
@@ -15,6 +15,7 @@ import net.pterodactylus.fcp.RequestProgress;
 public interface ClientPutDiskDirCommand {
 
 	ClientPutDiskDirCommand onProgress(Consumer<RequestProgress> requestProgressConsumer);
+	ClientPutDiskDirCommand onKeyGenerated(Consumer<String> keyGenerated);
 
 	WithUri fromDirectory(File directory);
 
diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommandImpl.java b/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommandImpl.java
index 2cf10c7..d223da2 100644
--- a/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommandImpl.java
+++ b/src/main/java/net/pterodactylus/fcp/quelaton/ClientPutDiskDirCommandImpl.java
@@ -18,6 +18,7 @@ import net.pterodactylus.fcp.ProtocolError;
 import net.pterodactylus.fcp.PutSuccessful;
 import net.pterodactylus.fcp.RequestProgress;
 import net.pterodactylus.fcp.SimpleProgress;
+import net.pterodactylus.fcp.URIGenerated;
 import net.pterodactylus.fcp.Verbosity;
 
 import com.google.common.util.concurrent.ListeningExecutorService;
@@ -36,6 +37,7 @@ public class ClientPutDiskDirCommandImpl implements ClientPutDiskDirCommand {
 	private final AtomicReference<String> directory = new AtomicReference<>();
 	private final AtomicReference<String> uri = new AtomicReference<>();
 	private final List<Consumer<RequestProgress>> requestProgressConsumers = new CopyOnWriteArrayList<>();
+	private final List<Consumer<String>> keyGeneratedConsumers = new CopyOnWriteArrayList<>();
 
 	public ClientPutDiskDirCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier, Supplier<String> identifierGenerator) {
 		this.threadPool = MoreExecutors.listeningDecorator(threadPool);
@@ -50,6 +52,12 @@ public class ClientPutDiskDirCommandImpl implements ClientPutDiskDirCommand {
 	}
 
 	@Override
+	public ClientPutDiskDirCommand onKeyGenerated(Consumer<String> keyGenerated) {
+		keyGeneratedConsumers.add(Objects.requireNonNull(keyGenerated));
+		return this;
+	}
+
+	@Override
 	public WithUri fromDirectory(File directory) {
 		this.directory.set(Objects.requireNonNull(directory).getPath());
 		return this::uri;
@@ -97,6 +105,11 @@ public class ClientPutDiskDirCommandImpl implements ClientPutDiskDirCommand {
 		}
 
 		@Override
+		protected void consumeURIGenerated(URIGenerated uriGenerated) {
+			keyGeneratedConsumers.forEach(consumer -> consumer.accept(uriGenerated.getURI()));
+		}
+
+		@Override
 		protected void consumeProtocolError(ProtocolError protocolError) {
 			finish();
 		}
diff --git a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
index 77573c0..0a39451 100644
--- a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
@@ -1706,6 +1706,18 @@ public class DefaultFcpClientTest {
 				));
 			}
 
+			@Test
+			public void generatedUriIsSentToConsumerCorrectly() throws InterruptedException, ExecutionException, IOException {
+				List<String> generatedKeys = new ArrayList<>();
+				Future<Optional<Key>> key = fcpClient.clientPutDiskDir().onKeyGenerated(generatedKeys::add)
+					.fromDirectory(new File("")).uri("CHK@").execute();
+				connectAndAssert(this::matchesClientPutDiskDir);
+				replyWithGeneratedUri();
+				replyWithPutSuccessful(identifier);
+				assertThat(key.get().get().getKey(), is("KSK@foo.txt"));
+				assertThat(generatedKeys, contains("KSK@foo.txt"));
+			}
+
 			private Matcher<List<String>> matchesClientPutDiskDir(String... additionalLines) {
 				List<String> lines = new ArrayList<>(Arrays.asList("Identifier=" + identifier, "URI=CHK@", "Filename=" + new File("").getPath()));
 				Arrays.asList(additionalLines).forEach(lines::add);
-- 
2.7.4