X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fxdcc%2Fui%2Fstdin%2FCommandReader.java;h=abef80e5c57b502a28c785927e6439a5349218fc;hb=55c7c1e4b98b06091b5053af4a9157e4a3428b32;hp=2ef14d7224098db6594f6aa2cc9c3c8e8a66e4a2;hpb=d3a8f3edb125265c6feb8b0ccfaa035497c07f15;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 2ef14d7..abef80e 100644
--- a/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java
+++ b/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java
@@ -21,10 +21,13 @@ import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.Reader;
 import java.io.Writer;
+import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
+import java.util.Comparator;
 import java.util.List;
 import java.util.Set;
+import java.util.regex.Pattern;
 
 import net.pterodactylus.irc.Connection;
 import net.pterodactylus.irc.DccReceiver;
@@ -39,6 +42,7 @@ import net.pterodactylus.xdcc.data.Bot;
 import net.pterodactylus.xdcc.data.Download;
 import net.pterodactylus.xdcc.data.Pack;
 
+import com.google.common.base.Predicate;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Sets;
 import com.google.common.eventbus.Subscribe;
@@ -169,9 +173,15 @@ public class CommandReader extends AbstractExecutionThreadService {
 				}
 				writeLine("End of connections.");
 			} else if (words[0].equalsIgnoreCase("disconnect")) {
-				Integer index = Ints.tryParse(words[1]);
-				if ((index != null) && (index < lastConnections.size())) {
-					core.closeConnection(lastConnections.get(index));
+				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));
+					}
 				}
 			}
 
@@ -302,6 +312,74 @@ public class CommandReader extends AbstractExecutionThreadService {
 	/** Container for result information. */
 	private static class Result implements Comparable<Result> {
 
+		/** {@link Predicate} that matches {@link Result}s that contain an archive. */
+		private static final Predicate<Result> isArchive = new Predicate<Result>() {
+
+			/** All suffixes that are recognized as archives. */
+			private final List<String> archiveSuffixes = Arrays.asList("rar", "tar", "zip", "tar.gz", "tar.bz2", "tar.lzma", "7z");
+
+			@Override
+			public boolean apply(Result result) {
+				for (String suffix : archiveSuffixes) {
+					if (result.pack().name().toLowerCase().endsWith(suffix)) {
+						return true;
+					}
+				}
+				return false;
+			}
+		};
+
+		/**
+		 * {@link Comparator} for {@link Result}s that sorts archives (as per {@link
+		 * #isArchive} to the back of the list.
+		 */
+		private static final Comparator<Result> packArchiveComparator = new Comparator<Result>() {
+			@Override
+			public int compare(Result leftResult, Result rightResult) {
+				if (isArchive.apply(leftResult) && !isArchive.apply(rightResult)) {
+					return 1;
+				}
+				if (!isArchive.apply(leftResult) && isArchive.apply(rightResult)) {
+					return -1;
+				}
+				return 0;
+			}
+		};
+
+		/**
+		 * {@link Comparator} for bot nicknames. It comprises different strategies:
+		 * one name pattern is preferred (and thus listed first), one pattern is
+		 * disliked (and thus listed last), the rest is sorted alphabetically.
+		 */
+		private static final Comparator<String> botNameComparator = new Comparator<String>() {
+
+			/** Regular expression pattern for preferred names. */
+			private final Pattern preferredNames = Pattern.compile("(?i)[^\\w]EUR?[^\\w]");
+
+			/** Regular expression pattern for disliked names. */
+			private final Pattern dislikedNames = Pattern.compile("(?i)[^\\w]USA?[^\\w]");
+
+			@Override
+			public int compare(String leftBotName, String rightBotName) {
+				/* preferred names to the front! */
+				if (preferredNames.matcher(leftBotName).find() && !preferredNames.matcher(rightBotName).find()) {
+					return -1;
+				}
+				if (preferredNames.matcher(rightBotName).find() && !preferredNames.matcher(leftBotName).find()) {
+					return 1;
+				}
+				/* disliked names to the back. */
+				if (dislikedNames.matcher(leftBotName).find() && !dislikedNames.matcher(rightBotName).find()) {
+					return 1;
+				}
+				if (dislikedNames.matcher(rightBotName).find() && !dislikedNames.matcher(leftBotName).find()) {
+					return -1;
+				}
+				/* rest is sorted by name. */
+				return leftBotName.compareToIgnoreCase(rightBotName);
+			}
+		};
+
 		/** The bot carrying the pack. */
 		private final Bot bot;
 
@@ -349,7 +427,14 @@ public class CommandReader extends AbstractExecutionThreadService {
 
 		@Override
 		public int compareTo(Result result) {
-			return pack().name().compareToIgnoreCase(result.pack().name());
+			if (isArchive.apply(this) && !isArchive.apply(result)) {
+				return 1;
+			}
+			if (!isArchive.apply(this) && isArchive.apply(result)) {
+				return -1;
+			}
+			/* sort by bot name. */
+			return botNameComparator.compare(bot().name(), result.bot().name());
 		}
 
 	}