Send an event if the connection is terminated for whatever reason.
[xudocci.git] / src / main / java / net / pterodactylus / irc / Connection.java
index ade43c9..10edc3c 100644 (file)
@@ -27,7 +27,9 @@ import java.io.InputStream;
 import java.io.InputStreamReader;
 import java.io.OutputStream;
 import java.io.UnsupportedEncodingException;
+import java.net.InetAddress;
 import java.net.Socket;
+import java.net.UnknownHostException;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
@@ -37,14 +39,20 @@ import java.util.logging.Logger;
 import javax.net.SocketFactory;
 
 import net.pterodactylus.irc.event.ChannelJoined;
+import net.pterodactylus.irc.event.ChannelLeft;
 import net.pterodactylus.irc.event.ChannelMessageReceived;
 import net.pterodactylus.irc.event.ChannelNicknames;
 import net.pterodactylus.irc.event.ChannelNotJoined;
 import net.pterodactylus.irc.event.ChannelNotJoined.Reason;
 import net.pterodactylus.irc.event.ChannelTopic;
+import net.pterodactylus.irc.event.ClientQuit;
+import net.pterodactylus.irc.event.ConnectionClosed;
 import net.pterodactylus.irc.event.ConnectionEstablished;
 import net.pterodactylus.irc.event.ConnectionFailed;
+import net.pterodactylus.irc.event.DccAcceptReceived;
+import net.pterodactylus.irc.event.DccSendReceived;
 import net.pterodactylus.irc.event.MotdReceived;
+import net.pterodactylus.irc.event.NicknameChanged;
 import net.pterodactylus.irc.event.NicknameInUseReceived;
 import net.pterodactylus.irc.event.NoNicknameGivenReceived;
 import net.pterodactylus.irc.event.PrivateMessageReceived;
@@ -56,6 +64,8 @@ import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
 import com.google.common.eventbus.EventBus;
 import com.google.common.io.Closeables;
+import com.google.common.primitives.Ints;
+import com.google.common.primitives.Longs;
 import com.google.common.util.concurrent.AbstractExecutionThreadService;
 import com.google.common.util.concurrent.Service;
 
@@ -196,6 +206,19 @@ public class Connection extends AbstractExecutionThreadService implements Servic
        //
 
        /**
+        * Checks whether the given source is the client represented by this
+        * connection.
+        *
+        * @param source
+        *              The source to check
+        * @return {@code true} if this connection represents the given source, {@code
+        *         false} otherwise
+        */
+       public boolean isSource(Source source) {
+               return source.nick().isPresent() && source.nick().get().equals(nickname);
+       }
+
+       /**
         * Joins the given channel.
         *
         * @param channel
@@ -222,6 +245,24 @@ public class Connection extends AbstractExecutionThreadService implements Servic
                connectionHandler.sendCommand("PRIVMSG", recipient, message);
        }
 
+       /**
+        * Sends a DCC RESUME request to the given recipient.
+        *
+        * @param recipient
+        *              The recipient of the request
+        * @param filename
+        *              The name of the file to resume
+        * @param port
+        *              The port number from the original DCC SEND request
+        * @param position
+        *              The position at which to resume the transfer
+        * @throws IOException
+        *              if an I/O error occurs
+        */
+       public void sendDccResume(String recipient, String filename, int port, long position) throws IOException {
+               connectionHandler.sendCommand("PRIVMSG", recipient, String.format("\u0001DCC RESUME %s %d %d\u0001", filename, port, position));
+       }
+
        //
        // ABSTRACTEXECUTIONTHREADSERVICE METHODS
        //
@@ -276,7 +317,31 @@ public class Connection extends AbstractExecutionThreadService implements Servic
                                if (command.equalsIgnoreCase("PRIVMSG")) {
                                        String recipient = parameters.get(0);
                                        String message = parameters.get(1);
-                                       if (!channelTypes.contains(recipient.charAt(0))) {
+                                       if (message.startsWith("\u0001") && message.endsWith("\u0001")) {
+                                               /* CTCP! */
+                                               String[] messageWords = message.substring(1, message.length() - 1).split(" +");
+                                               String ctcpCommand = messageWords[0];
+                                               if (ctcpCommand.equalsIgnoreCase("DCC")) {
+                                                       if (messageWords[1].equalsIgnoreCase("SEND")) {
+                                                               Optional<InetAddress> inetAddress = parseInetAddress(messageWords[3]);
+                                                               Optional<Integer> port = Optional.fromNullable(Ints.tryParse(messageWords[4]));
+                                                               long fileSize = Optional.fromNullable(Longs.tryParse(messageWords[5])).or(-1L);
+                                                               if (inetAddress.isPresent() && port.isPresent()) {
+                                                                       eventBus.post(new DccSendReceived(this, reply.source().get(), messageWords[2], inetAddress.get(), port.get(), fileSize));
+                                                               } else {
+                                                                       logger.warning(String.format("Received malformed DCC SEND: “%s”", message));
+                                                               }
+                                                       } else if (messageWords[1].equalsIgnoreCase("ACCEPT")) {
+                                                               Optional<Integer> port = Optional.fromNullable(Ints.tryParse(messageWords[3]));
+                                                               long position = Optional.fromNullable(Longs.tryParse(messageWords[4])).or(-1L);
+                                                               if (port.isPresent()) {
+                                                                       eventBus.post(new DccAcceptReceived(this, reply.source().get(), messageWords[2], port.get(), position));
+                                                               } else {
+                                                                       logger.warning(String.format("Received malformed DCC ACCEPT: “%s”", message));
+                                                               }
+                                                       }
+                                               }
+                                       } else if (!channelTypes.contains(recipient.charAt(0))) {
                                                eventBus.post(new PrivateMessageReceived(this, reply.source().get(), message));
                                        } else {
                                                eventBus.post(new ChannelMessageReceived(this, recipient, reply.source().get(), message));
@@ -336,6 +401,10 @@ public class Connection extends AbstractExecutionThreadService implements Servic
                                                eventBus.post(new NicknameInUseReceived(this, reply));
                                        }
 
+                               /* client stuff. */
+                               } else if (command.equalsIgnoreCase("NICK")) {
+                                       eventBus.post(new NicknameChanged(this, reply.source().get(), parameters.get(0)));
+
                                /* channel stuff. */
                                } else if (command.equalsIgnoreCase("JOIN")) {
                                        eventBus.post(new ChannelJoined(this, parameters.get(0), reply.source().get()));
@@ -354,6 +423,10 @@ public class Connection extends AbstractExecutionThreadService implements Servic
                                } else if (command.equals("366")) {
                                        eventBus.post(new ChannelNicknames(this, parameters.get(1), nicks));
                                        nicks.clear();
+                               } else if (command.equalsIgnoreCase("PART")) {
+                                       eventBus.post(new ChannelLeft(this, parameters.get(0), reply.source().get(), getOptional(parameters, 1)));
+                               } else if (command.equalsIgnoreCase("QUIT")) {
+                                       eventBus.post(new ClientQuit(this, reply.source().get(), parameters.get(0)));
 
                                /* common channel join errors. */
                                } else if (command.equals("474")) {
@@ -378,8 +451,10 @@ public class Connection extends AbstractExecutionThreadService implements Servic
                                }
                                oldConnectionStatus = connectionStatus;
                        }
+                       eventBus.post(new ConnectionClosed(this));
                } catch (IOException ioe1) {
                        logger.log(Level.WARNING, "I/O error", ioe1);
+                       eventBus.post(new ConnectionClosed(this, ioe1));
                } finally {
                        logger.info("Closing Connection.");
                        try {
@@ -415,6 +490,32 @@ public class Connection extends AbstractExecutionThreadService implements Servic
                return Optional.absent();
        }
 
+       /**
+        * Parses the given {@code ip} and returns an {@link InetAddress} from it.
+        *
+        * @param ip
+        *              The IP to parse
+        * @return The parsed inet address, or {@link Optional#absent()} if no inet
+        *         address could be parsed
+        */
+       private Optional<InetAddress> parseInetAddress(String ip) {
+               Long ipNumber = Longs.tryParse(ip);
+               if (ipNumber == null) {
+                       return Optional.absent();
+               }
+
+               StringBuilder hostname = new StringBuilder(15);
+               hostname.append((ipNumber >>> 24) & 0xff).append('.');
+               hostname.append((ipNumber >>> 16) & 0xff).append('.');
+               hostname.append((ipNumber >>> 8) & 0xff).append('.');
+               hostname.append(ipNumber & 0xff);
+               try {
+                       return Optional.of(InetAddress.getByName(hostname.toString()));
+               } catch (UnknownHostException uhe1) {
+                       return Optional.absent();
+               }
+       }
+
        /** Handles input and output for the connection. */
        private class ConnectionHandler implements Closeable {