From 5c5094e0a1eace0ff43d032b550d872ea2e45050 Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Tue, 13 Aug 2013 20:42:03 +0200 Subject: [PATCH] Store last displayed elements in a custom state object. --- .../pterodactylus/xdcc/ui/stdin/CommandReader.java | 142 ++++++++++++++++++--- 1 file changed, 126 insertions(+), 16 deletions(-) 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 c278ea5..68b6b86 100644 --- a/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java +++ b/src/main/java/net/pterodactylus/xdcc/ui/stdin/CommandReader.java @@ -17,6 +17,7 @@ package net.pterodactylus.xdcc.ui.stdin; +import static com.google.common.collect.Lists.newArrayList; import static net.pterodactylus.xdcc.data.Download.BY_NAME; import static net.pterodactylus.xdcc.data.Download.BY_RUNNING; @@ -95,18 +96,16 @@ public class CommandReader extends AbstractExecutionThreadService { protected void run() throws Exception { String lastLine = ""; String line; - final List lastResult = Lists.newArrayList(); - final List downloads = Lists.newArrayList(); - final List lastConnections = Lists.newArrayList(); + State state = new State(); while ((line = reader.readLine()) != null) { if (line.equals("")) { line = lastLine; } String[] words = line.split(" +"); if (words[0].equalsIgnoreCase("search")) { - lastResult.clear(); - for (Bot bot : Lists.newArrayList(core.bots())) { - for (Pack pack : Lists.newArrayList(bot)) { + List lastResult = newArrayList(); + for (Bot bot : newArrayList(core.bots())) { + for (Pack pack : 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))) { @@ -129,10 +128,10 @@ public class CommandReader extends AbstractExecutionThreadService { writeLine(String.format("[%d] %s (%s) from %s (#%s) on %s", counter++, result.pack().name(), result.pack().size(), result.bot().name(), result.pack().id(), result.bot().network().name())); } writeLine("End of Search."); + state = state.setLastResults(lastResult); } else if (words[0].equalsIgnoreCase("dcc")) { int counter = 0; - downloads.clear(); - downloads.addAll(FluentIterable.from(core.downloads()).toSortedList(Ordering.from(BY_NAME).compound(BY_RUNNING))); + List downloads = newArrayList(FluentIterable.from(core.downloads()).toSortedList(Ordering.from(BY_NAME).compound(BY_RUNNING))); for (Download download : downloads) { DccReceiver dccReceiver = download.dccReceiver(); if (dccReceiver == null) { @@ -153,15 +152,16 @@ public class CommandReader extends AbstractExecutionThreadService { writer.write(")\n"); } writeLine("End of DCCs."); + state = state.setLastDownloads(downloads); } else if (words[0].equalsIgnoreCase("get")) { Integer index = Ints.tryParse(words[1]); - if ((index != null) && (index < lastResult.size())) { - core.fetch(lastResult.get(index).bot(), lastResult.get(index).pack()); + if ((index != null) && (index < state.getLastResults().size())) { + core.fetch(state.getLastResults().get(index).bot(), state.getLastResults().get(index).pack()); } } else if (words[0].equalsIgnoreCase("cancel")) { Integer index = Ints.tryParse(words[1]); - if ((index != null) && (index < downloads.size())) { - core.cancelDownload(downloads.get(index).bot(), downloads.get(index).pack()); + if ((index != null) && (index < state.getLastDownloads().size())) { + core.cancelDownload(state.getLastDownloads().get(index).bot(), state.getLastDownloads().get(index).pack()); } } else if (words[0].equalsIgnoreCase("stats")) { int configuredChannelsCount = core.channels().size(); @@ -179,22 +179,23 @@ 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(); + List lastConnections = newArrayList(); 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."); + state = state.setLastConnections(lastConnections); } else if (words[0].equalsIgnoreCase("disconnect")) { if ((words.length == 1) || ("all".equals(words[1]))) { - for (Connection connection : lastConnections) { + for (Connection connection : state.getLastConnections()) { core.closeConnection(connection); } } else { Integer index = Ints.tryParse(words[1]); - if ((index != null) && (index < lastConnections.size())) { - core.closeConnection(lastConnections.get(index)); + if ((index != null) && (index < state.getLastConnections().size())) { + core.closeConnection(state.getLastConnections().get(index)); } } } @@ -480,4 +481,113 @@ public class CommandReader extends AbstractExecutionThreadService { } + /** + * Container for the current state of the command reader. + * + * @author David ‘Bombe’ Roden + */ + private static class State { + + /** The last connections displayed. */ + private final List lastConnections; + + /** The last results displayed. */ + private final List lastResults; + + /** The last downloads displayed. */ + private final List lastDownloads; + + public State() { + this(Lists.newArrayList(), Lists.newArrayList(), Lists.newArrayList()); + } + + /** + * Creates a new state. + * + * @param lastConnections + * The last connections + * @param lastResults + * The last results + * @param lastDownloads + * The last downloads + */ + private State(List lastConnections, List lastResults, List lastDownloads) { + this.lastConnections = lastConnections; + this.lastResults = lastResults; + this.lastDownloads = lastDownloads; + } + + // + // ACCESSORS + // + + /** + * Returns the last connections displayed. + * + * @return The last connections displayed + */ + public List getLastConnections() { + return lastConnections; + } + + /** + * Returns the last results displayed. + * + * @return The last results displayed + */ + public List getLastResults() { + return lastResults; + } + + /** + * Returns the last downloads displayed. + * + * @return The last downloads displayed + */ + public List getLastDownloads() { + return lastDownloads; + } + + // + // MUTATORS + // + + /** + * Returns a new state with the given last connections and the last downloads + * and results of this state. + * + * @param lastConnections + * The new last connections displayed + * @return The new state + */ + public State setLastConnections(List lastConnections) { + return new State(lastConnections, lastResults, lastDownloads); + } + + /** + * Returns a new state with the given last results and the last downloads and + * connections of this state. + * + * @param lastResults + * The new last results displayed + * @return The new state + */ + public State setLastResults(List lastResults) { + return new State(lastConnections, lastResults, lastDownloads); + } + + /** + * Returns a new state with the given last downloads and the last connections + * and results of this state. + * + * @param lastDownloads + * The new last downloads displayed + * @return The new state + */ + public State setLastDownloads(List lastDownloads) { + return new State(lastConnections, lastResults, lastDownloads); + } + + } + } -- 2.7.4