Clean up ClientPlaySessionHandler#handle(PluginMessage)

proxy/src/main/java/com/velocitypowered/proxy/connection/client/ClientPlaySessionHandler.java

@@ -170,41 +170,32 @@ public class ClientPlaySessionHandler implements MinecraftSessionHandler {
   public boolean handle(PluginMessage packet) {
     VelocityServerConnection serverConn = player.getConnectedServer();
     MinecraftConnection backendConn = serverConn != null ? serverConn.getConnection() : null;
-    if (serverConn != null && backendConn != null) {
+    if (serverConn == null || backendConn == null) {
+      return true;
+    }
+
     if (backendConn.getState() != StateRegistry.PLAY) {
       logger.warn("A plugin message was received while the backend server was not "
           + "ready. Channel: {}. Packet discarded.", packet.getChannel());
-      } else if (PluginMessageUtil.isRegister(packet)) {
+      return true;
-        player.getKnownChannels().addAll(PluginMessageUtil.getChannels(packet));
+    }
-        backendConn.write(packet.retain());
+
-      } else if (PluginMessageUtil.isUnregister(packet)) {
+    if (this.tryHandleVanillaPluginMessageChannel(packet, backendConn)) {
-        player.getKnownChannels().removeAll(PluginMessageUtil.getChannels(packet));
+      return true;
-        backendConn.write(packet.retain());
+    }
-      } else if (PluginMessageUtil.isMcBrand(packet)) {
+
-        backendConn.write(PluginMessageUtil
-            .rewriteMinecraftBrand(packet, server.getVersion(), player.getProtocolVersion()));
-      } else {
     if (serverConn.getPhase() == BackendConnectionPhases.IN_TRANSITION) {
       // We must bypass the currently-connected server when forwarding Forge packets.
       VelocityServerConnection inFlight = player.getConnectionInFlight();
       if (inFlight != null) {
-            player.getPhase().handle(player, packet, inFlight);
+        if (player.getPhase().handle(player, packet, inFlight)) {
+          return true;
         }
+      }
+    } else if (!this.tryHandleForgeMessage(packet, serverConn)) {
       return true;
     }
 
-        if (!player.getPhase().handle(player, packet, serverConn)) {
-          if (!player.getPhase().consideredComplete() || !serverConn.getPhase()
-              .consideredComplete()) {
-            // The client is trying to send messages too early. This is primarily caused by mods,
-            // but further aggravated by Velocity. To work around these issues, we will queue any
-            // non-FML handshake messages to be sent once the FML handshake has completed or the
-            // JoinGame packet has been received by the proxy, whichever comes first.
-            //
-            // We also need to make sure to retain these packets so they can be flushed
-            // appropriately.
-            loginPluginMessages.add(packet.retain());
-          } else {
     ChannelIdentifier id = server.getChannelRegistrar().getFromId(packet.getChannel());
     if (id == null) {
       backendConn.write(packet.retain());
@@ -218,11 +209,6 @@ public class ClientPlaySessionHandler implements MinecraftSessionHandler {
         backendConn.write(message);
       }, backendConn.eventLoop());
     }
-          }
-        }
-      }
-    }
-
     return true;
   }
 
@@ -293,6 +279,44 @@ public class ClientPlaySessionHandler implements MinecraftSessionHandler {
     }
   }
 
+  private boolean tryHandleVanillaPluginMessageChannel(PluginMessage packet,
+      MinecraftConnection backendConn) {
+    if (PluginMessageUtil.isRegister(packet)) {
+      player.getKnownChannels().addAll(PluginMessageUtil.getChannels(packet));
+      backendConn.write(packet.retain());
+      return true;
+    } else if (PluginMessageUtil.isUnregister(packet)) {
+      player.getKnownChannels().removeAll(PluginMessageUtil.getChannels(packet));
+      backendConn.write(packet.retain());
+      return true;
+    } else if (PluginMessageUtil.isMcBrand(packet)) {
+      backendConn.write(PluginMessageUtil.rewriteMinecraftBrand(packet, server.getVersion(),
+          player.getProtocolVersion()));
+      return true;
+    }
+    return false;
+  }
+
+  private boolean tryHandleForgeMessage(PluginMessage packet, VelocityServerConnection serverConn) {
+    if (player.getPhase().handle(player, packet, serverConn)) {
+      return true;
+    }
+
+    if (!player.getPhase().consideredComplete() || !serverConn.getPhase().consideredComplete()) {
+      // The client is trying to send messages too early. This is primarily caused by mods,
+      // but further aggravated by Velocity. To work around these issues, we will queue any
+      // non-FML handshake messages to be sent once the FML handshake has completed or the
+      // JoinGame packet has been received by the proxy, whichever comes first.
+      //
+      // We also need to make sure to retain these packets so they can be flushed
+      // appropriately.
+      loginPluginMessages.add(packet.retain());
+      return true;
+    } else {
+      return false;
+    }
+  }
+
   /**
    * Handles the {@code JoinGame} packet. This function is responsible for handling the client-side
    * switching servers in Velocity.