X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fxdcc%2Fui%2Fstdin%2FCommandReader.java;h=eaaef2389c127b747548799928e681b3b2431710;hb=52be4d6d0bf21502d301814ba88735fd72a3f284;hp=a530a1d28935d9e2a397b3259a8c1d452a11b210;hpb=6df6da151ec3804a8f3cc8ac65277842358d7e38;p=xudocci.git

diff --git a/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java b/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java
index a530a1d..eaaef23 100644
--- a/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java
+++ b/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java
@@ -26,6 +26,7 @@ import java.util.Collections;
 import java.util.List;
 import java.util.Set;
 
+import net.pterodactylus.irc.Connection;
 import net.pterodactylus.irc.DccReceiver;
 import net.pterodactylus.irc.util.MessageCleaner;
 import net.pterodactylus.xdcc.core.Core;
@@ -85,6 +86,7 @@ public class CommandReader extends AbstractExecutionThreadService {
 		String lastLine = "";
 		String line;
 		final List<Result> lastResult = Lists.newArrayList();
+		final List<Connection> lastConnections = Lists.newArrayList();
 		while ((line = reader.readLine()) != null) {
 			if (line.equals("")) {
 				line = lastLine;
@@ -92,8 +94,8 @@ public class CommandReader extends AbstractExecutionThreadService {
 			String[] words = line.split(" +");
 			if (words[0].equalsIgnoreCase("search")) {
 				lastResult.clear();
-				for (Bot bot : core.bots()) {
-					for (Pack pack : bot) {
+				for (Bot bot : Lists.newArrayList(core.bots())) {
+					for (Pack pack : Lists.newArrayList(bot)) {
 						boolean found = true;
 						for (int wordIndex = 1; wordIndex < words.length; ++wordIndex) {
 							if (words[wordIndex].startsWith("-") && pack.name().toLowerCase().contains(words[wordIndex].toLowerCase().substring(1))) {
@@ -118,8 +120,14 @@ public class CommandReader extends AbstractExecutionThreadService {
 				writeLine("End of Search.");
 			} else if (words[0].equalsIgnoreCase("dcc")) {
 				int counter = 0;
-				for (DccReceiver dccReceiver : core.dccReceivers()) {
-					writer.write(String.format("[%d] %s (%s, ", counter++, dccReceiver.filename(), f(dccReceiver.size())));
+				for (Download download : core.downloads()) {
+					DccReceiver dccReceiver = download.dccReceiver();
+					if (dccReceiver == null) {
+						/* download has not even started. */
+						writer.write(String.format("[%d] %s requested from %s (not started yet)\n", counter++, download.pack().name(), download.bot().name()));
+						continue;
+					}
+					writer.write(String.format("[%d] %s from %s (%s, ", counter++, dccReceiver.filename(), download.bot().name(), f(dccReceiver.size())));
 					if (dccReceiver.isRunning()) {
 						writer.write(String.format("%.1f%%, %s", dccReceiver.progress() * 100.0 / dccReceiver.size(), f(dccReceiver.currentRate())));
 					} else {
@@ -152,6 +160,25 @@ public class CommandReader extends AbstractExecutionThreadService {
 				}
 
 				writeLine(String.format("%d channels (%d joined, %d extra), %d bots offering %d packs (%d unique).", configuredChannelsCount, joinedChannelsCount, extraChannelsCount, bots.size(), packsCount, packNames.size()));
+			} else if (words[0].equalsIgnoreCase("connections")) {
+				lastConnections.clear();
+				int counter = 0;
+				for (Connection connection : core.connections()) {
+					lastConnections.add(connection);
+					writer.write(String.format("[%d] %s:%d, %s/s\n", counter++, connection.hostname(), connection.port(), f(connection.getInputRate())));
+				}
+				writeLine("End of connections.");
+			} else if (words[0].equalsIgnoreCase("disconnect")) {
+				if ((words.length == 1) || ("all".equals(words[1]))) {
+					for (Connection connection : lastConnections) {
+						core.closeConnection(connection);
+					}
+				} else {
+					Integer index = Ints.tryParse(words[1]);
+					if ((index != null) && (index < lastConnections.size())) {
+						core.closeConnection(lastConnections.get(index));
+					}
+				}
 			}
 
 			lastLine = line;