src/main/java/net/pterodactylus/irc/connection/ConnectionEstablishHandler.java

   1 package net.pterodactylus.irc.connection;
   2
   3 import static java.util.Arrays.asList;
   4
   5 import net.pterodactylus.irc.Connection;
   6 import net.pterodactylus.irc.Reply;
   7 import net.pterodactylus.irc.event.ConnectionEstablished;
   8
   9 import com.google.common.eventbus.EventBus;
  10
  11 /**
  12  * Handles the necessary commands that establish a connection.
  13  *
  14  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  15  */
  16 public class ConnectionEstablishHandler implements Handler {
  17
  18         private static final int FIRST_COMMAND_RECEIVED = 0x01;
  19         private static final int SECOND_COMMAND_RECEIVED = 0x02;
  20         private static final int THIRD_COMMAND_RECEIVED = 0x04;
  21         private static final int FOURTH_COMMAND_RECEIVED = 0x08;
  22         private static final int ALL_COMMANDS_RECEIVED = 0x0f;
  23
  24         private final EventBus eventBus;
  25         private final Connection connection;
  26         private int connectionStatus;
  27         private int oldConnectionStatus;
  28
  29         public ConnectionEstablishHandler(EventBus eventBus,
  30                         Connection connection) {
  31                 this.eventBus = eventBus;
  32                 this.connection = connection;
  33         }
  34
  35         @Override
  36         public boolean willHandle(Reply reply) {
  37                 return asList("001", "002", "003", "004").contains(reply.command());
  38         }
  39
  40         @Override
  41         public void handleReply(Reply reply) {
  42                 String command = reply.command();
  43                 /* replies 001-004 don’t hold information but they
  44                  * have to be sent on a successful connection. */
  45                 if (command.equals("001")) {
  46                         connectionStatus |= FIRST_COMMAND_RECEIVED;
  47                 } else if (command.equals("002")) {
  48                         connectionStatus |= SECOND_COMMAND_RECEIVED;
  49                 } else if (command.equals("003")) {
  50                         connectionStatus |= THIRD_COMMAND_RECEIVED;
  51                 } else if (command.equals("004")) {
  52                         connectionStatus |= FOURTH_COMMAND_RECEIVED;
  53                 }
  54
  55                 if (hasConnectionStatusChanged()
  56                                 && isFinalConnectionStatusReached()) {
  57                         eventBus.post(new ConnectionEstablished(connection));
  58                 }
  59                 oldConnectionStatus = connectionStatus;
  60         }
  61
  62         private boolean hasConnectionStatusChanged() {
  63                 return connectionStatus != oldConnectionStatus;
  64         }
  65
  66         private boolean isFinalConnectionStatusReached() {
  67                 return connectionStatus == ALL_COMMANDS_RECEIVED;
  68         }
  69
  70 }