import java.io.BufferedReader;
import java.io.Reader;
+import java.io.Writer;
+import java.util.Collection;
import java.util.Collections;
import java.util.List;
+import java.util.Set;
+import net.pterodactylus.irc.DccReceiver;
import net.pterodactylus.xdcc.core.Core;
import net.pterodactylus.xdcc.data.Bot;
import net.pterodactylus.xdcc.data.Pack;
import com.google.common.collect.Lists;
+import com.google.common.collect.Sets;
+import com.google.common.primitives.Ints;
import com.google.common.util.concurrent.AbstractExecutionThreadService;
/**
/** The reader to read commands from. */
private final BufferedReader reader;
+ /** The writer to write the results to. */
+ private final Writer writer;
+
/**
* Creates a new command reader.
*
* The core being controlled
* @param reader
* The reader to read commands from
+ * @param writer
+ * The write to write results to
*/
- public CommandReader(Core core, Reader reader) {
+ public CommandReader(Core core, Reader reader, Writer writer) {
this.core = core;
this.reader = new BufferedReader(reader);
+ this.writer = writer;
}
//
Collections.sort(lastResult);
int counter = 0;
for (Result result : lastResult) {
- System.out.println(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()));
+ writer.write(String.format("[%d] %s (%s) from %s (#%s) on %s\n", counter++, result.pack().name(), result.pack().size(), result.bot().name(), result.pack().id(), result.bot().network().name()));
+ }
+ writer.write("End of Search.\n");
+ } else if (words[0].equalsIgnoreCase("dcc")) {
+ int counter = 0;
+ for (DccReceiver dccReceiver : core.dccReceivers()) {
+ writer.write(String.format("[%d] %s (%s, %d%%%s)\n", counter++, dccReceiver.filename(), dccReceiver.size(), dccReceiver.progress() * 100 / dccReceiver.size(), dccReceiver.isRunning() ? "" : ", finished"));
+ }
+ writer.write("End of DCCs.\n");
+ } 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());
}
- System.out.println("End of Search.");
+ } else if (words[0].equalsIgnoreCase("stats")) {
+ int configuredChannelsCount = core.channels().size();
+ int joinedChannelsCount = core.joinedChannels().size();
+ int extraChannelsCount = core.extraChannels().size();
+ Collection<Bot> bots = core.bots();
+ Set<String> packNames = Sets.newHashSet();
+ int packsCount = 0;
+ for (Bot bot : bots) {
+ packsCount += bot.packs().size();
+ for (Pack pack : bot) {
+ packNames.add(pack.name());
+ }
+ }
+
+ writer.write(String.format("%d channels (%d joined, %d extra), %d bots offering %d packs (%d unique).\n", configuredChannelsCount, joinedChannelsCount, extraChannelsCount, bots.size(), packsCount, packNames.size()));
}
lastLine = line;
+ writer.flush();
}
}