Centralize disconnect handling

src/main/java/baritone/bot/UserManager.java

@@ -36,6 +36,8 @@ import net.minecraft.network.NetworkManager;
 import net.minecraft.network.handshake.client.C00Handshake;
 import net.minecraft.network.login.client.CPacketLoginStart;
 import net.minecraft.util.Session;
+import net.minecraft.util.text.ITextComponent;
+import net.minecraft.util.text.TextComponentString;
 
 import java.net.InetAddress;
 import java.net.UnknownHostException;
@@ -65,12 +67,12 @@ public final class UserManager implements IUserManager, Helper {
             public final void onTick(TickEvent event) {
                 if (event.getState() == EventState.PRE) {
                     if (event.getType() == TickEvent.Type.OUT) {
-                        UserManager.this.users.forEach(UserManager.this::disconnect);
+                        UserManager.this.users.forEach(user -> UserManager.this.disconnect(user, null));
                     }
 
                     UserManager.this.users.forEach(user -> {
                         if (!user.getNetworkManager().isChannelOpen()) {
-                            UserManager.this.disconnect(user);
+                            UserManager.this.disconnect(user, new TextComponentString("Channel Closed"));
                         }
                     });
 
@@ -154,17 +156,6 @@ public final class UserManager implements IUserManager, Helper {
         }
     }
 
-    /**
-     * Notifies the manager of an {@link IBaritoneUser} disconnect, and
-     * removes the {@link IBaritoneUser} from the list of users.
-     *
-     * @param user  The user that disconnected
-     * @param state The connection state at the time of disconnect
-     */
-    public final void notifyDisconnect(IBaritoneUser user, EnumConnectionState state) {
-        this.users.remove(user);
-    }
-
     /**
      * @return The bot world provider
      */
@@ -173,7 +164,7 @@ public final class UserManager implements IUserManager, Helper {
     }
 
     @Override
-    public final void disconnect(IBaritoneUser user) {
+    public final void disconnect(IBaritoneUser user, ITextComponent reason) {
         if (this.users.contains(user)) {
             if (user.getNetworkManager().isChannelOpen()) {
                 // It's probably fine to pass null to this, because the handlers aren't doing anything with it
@@ -181,6 +172,9 @@ public final class UserManager implements IUserManager, Helper {
                 user.getNetworkManager().closeChannel(null);
             }
             this.users.remove(user);
+            if (reason != null) {
+                logDirect(user.getSession().getUsername() + " Disconnected: " + reason.getUnformattedText());
+            }
         }
     }
 

src/main/java/baritone/bot/handler/BotNetHandlerLoginClient.java

@@ -66,6 +66,6 @@ public class BotNetHandlerLoginClient extends NetHandlerLoginClient {
     @Override
     public void onDisconnect(@Nonnull ITextComponent reason) {
         // It's important that we don't call the superclass method because that would mess up GUIs and make us upset
-        this.user.getManager().notifyDisconnect(this.user, EnumConnectionState.LOGIN);
+        this.user.getManager().disconnect(this.user, reason);
     }
 }

src/main/java/baritone/bot/handler/BotNetHandlerPlayClient.java

@@ -687,7 +687,7 @@ public class BotNetHandlerPlayClient extends NetHandlerPlayClient {
     public void onDisconnect(@Nonnull ITextComponent reason) {
         // TODO Maybe more world unloadinde
         this.world.removeEntity(this.player);
-        this.user.getManager().notifyDisconnect(this.user, EnumConnectionState.PLAY);
+        this.user.getManager().disconnect(this.user, reason);
     }
 
     @Nonnull