Show a message when a bot has left a channel.

[xudocci.git] / src / main / java / net / pterodactylus / xdcc / core / Core.java

diff --git a/src/main/java/net/pterodactylus/xdcc/core/Core.java b/src/main/java/net/pterodactylus/xdcc/core/Core.java
index 64fdbec..dce8da3 100644 (file)
--- a/src/main/java/net/pterodactylus/xdcc/core/Core.java
+++ b/src/main/java/net/pterodactylus/xdcc/core/Core.java
@@ -38,6 +38,7 @@ import net.pterodactylus.irc.event.ChannelLeft;
 import net.pterodactylus.irc.event.ChannelMessageReceived;
 import net.pterodactylus.irc.event.ClientQuit;
 import net.pterodactylus.irc.event.ConnectionEstablished;
+import net.pterodactylus.irc.event.DccAcceptReceived;
 import net.pterodactylus.irc.event.DccDownloadFailed;
 import net.pterodactylus.irc.event.DccDownloadFinished;
 import net.pterodactylus.irc.event.DccSendReceived;
@@ -49,6 +50,7 @@ import net.pterodactylus.xdcc.core.event.CoreStarted;
 import net.pterodactylus.xdcc.core.event.DownloadFailed;
 import net.pterodactylus.xdcc.core.event.DownloadFinished;
 import net.pterodactylus.xdcc.core.event.DownloadStarted;
+import net.pterodactylus.xdcc.core.event.GenericError;
 import net.pterodactylus.xdcc.core.event.GenericMessage;
 import net.pterodactylus.xdcc.core.event.MessageReceived;
 import net.pterodactylus.xdcc.data.Bot;
@@ -341,7 +343,10 @@ public class Core extends AbstractIdleService {
                        return;
                }
 
-               networkBots.remove(network.get(), channelLeft.client().nick().get());
+               Bot removedBot = networkBots.remove(network.get(), channelLeft.client().nick().get());
+               if (removedBot != null) {
+                       eventBus.post(new GenericMessage(String.format("Bot %s (%s) was removed, %d packs removed.", removedBot.name(), removedBot.network().name(), removedBot.packs().size())));
+               }
        }
 
        /**
@@ -357,7 +362,10 @@ public class Core extends AbstractIdleService {
                        return;
                }
 
-               networkBots.remove(network.get(), clientQuit.client().nick().get());
+               Bot removedBot = networkBots.remove(network.get(), clientQuit.client().nick().get());
+               if (removedBot != null) {
+                       eventBus.post(new GenericMessage(String.format("Bot %s (%s) was removed, %d packs removed.", removedBot.name(), removedBot.network().name(), removedBot.packs().size())));
+               }
        }
 
        /**
@@ -433,9 +441,38 @@ public class Core extends AbstractIdleService {
                        return;
                }
 
-               logger.info(String.format("Starting download of %s.", dccSendReceived.filename()));
+               /* check if the file already exists. */
+               File outputFile = new File(temporaryDirectory, dccSendReceived.filename());
+               if (outputFile.exists()) {
+                       long existingFileSize = outputFile.length();
+
+                       /* file already complete? */
+                       if ((dccSendReceived.filesize() > -1) && (existingFileSize >= dccSendReceived.filesize())) {
+                               /* file is apparently already complete. just move it. */
+                               if (outputFile.renameTo(new File(finalDirectory, download.pack().name()))) {
+                                       eventBus.post(new GenericMessage(String.format("File %s already downloaded.", download.pack().name())));
+                               } else {
+                                       eventBus.post(new GenericMessage(String.format("File %s already downloaded but not moved to %s.", download.pack().name(), finalDirectory)));
+                               }
+
+                               /* remove download. */
+                               downloads.remove(download);
+                               return;
+                       }
+
+                       /* file not complete yet, DCC resume it. */
+                       try {
+                               download.remoteAddress(dccSendReceived.inetAddress()).filesize(dccSendReceived.filesize());
+                               dccSendReceived.connection().sendDccResume(dccSendReceived.source().nick().get(), dccSendReceived.filename(), dccSendReceived.port(), existingFileSize);
+                       } catch (IOException ioe1) {
+                               eventBus.post(new GenericError(String.format("Could not send DCC RESUME %s to %s (%s).", dccSendReceived.filename(), dccSendReceived.source().nick().get(), ioe1.getMessage())));
+                       }
+
+                       return;
+               }
+
+               /* file does not exist, start the download. */
                try {
-                       File outputFile = new File(temporaryDirectory, dccSendReceived.filename());
                        OutputStream fileOutputStream = new FileOutputStream(outputFile);
                        DccReceiver dccReceiver = new DccReceiver(eventBus, dccSendReceived.inetAddress(), dccSendReceived.port(), dccSendReceived.filename(), dccSendReceived.filesize(), fileOutputStream);
                        download.filename(outputFile.getPath()).outputStream(fileOutputStream).dccReceiver(dccReceiver);
@@ -443,7 +480,39 @@ public class Core extends AbstractIdleService {
                        dccReceiver.start();
                        eventBus.post(new DownloadStarted(download));
                } catch (FileNotFoundException fnfe1) {
-                       logger.log(Level.WARNING, "Could not open file for download!", fnfe1);
+                       eventBus.post(new GenericError(String.format("Could not start download of %s from %s (%s).", dccSendReceived.filename(), dccSendReceived.source().nick().get(), fnfe1.getMessage())));
+               }
+       }
+
+       @Subscribe
+       public void dccAcceptReceived(DccAcceptReceived dccAcceptReceived) {
+               Optional<Network> network = getNetwork(dccAcceptReceived.connection());
+               if (!network.isPresent()) {
+                       return;
+               }
+
+               Download download = downloads.get(dccAcceptReceived.filename());
+               if (download == null) {
+                       /* unknown download, ignore. */
+                       return;
+               }
+
+               try {
+                       File outputFile = new File(temporaryDirectory, dccAcceptReceived.filename());
+                       if (outputFile.length() != dccAcceptReceived.position()) {
+                               eventBus.post(new GenericError(String.format("Download %s from %s does not start at the right position!")));
+                               logger.log(Level.WARNING, String.format("Download %s from %s: have %d bytes but wants to resume from %d!", dccAcceptReceived.filename(), dccAcceptReceived.source(), outputFile.length(), dccAcceptReceived.position()));
+
+                               downloads.remove(download);
+                               return;
+                       }
+                       OutputStream outputStream = new FileOutputStream(outputFile, true);
+                       DccReceiver dccReceiver = new DccReceiver(eventBus, download.remoteAddress(), dccAcceptReceived.port(), dccAcceptReceived.filename(), dccAcceptReceived.position(), download.filesize(), outputStream);
+                       download.filename(outputFile.getPath()).outputStream(outputStream).dccReceiver(dccReceiver);
+                       dccReceivers.add(dccReceiver);
+                       dccReceiver.start();
+                       eventBus.post(new DownloadStarted(download));
+               } catch (FileNotFoundException fnfe1) {
                }
        }