print disconnect message of InitialHandlers

SpigotMC · Dec 20, 2023 · 46fddc4 · 46fddc4
1 parent 8537709
commit 46fddc4
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/proxy/src/main/java/net/md_5/bungee/netty/HandlerBoss.java b/proxy/src/main/java/net/md_5/bungee/netty/HandlerBoss.java
@@ -60,9 +60,22 @@ public void channelInactive(ChannelHandlerContext ctx) throws Exception
             channel.markClosed();
             handler.disconnected( channel );
 
-            if ( !( handler instanceof InitialHandler || handler instanceof PingHandler ) )
+            if ( !( handler instanceof PingHandler ) )
             {
-                ProxyServer.getInstance().getLogger().log( Level.INFO, "{0} has disconnected", handler );
+                if ( handler instanceof InitialHandler )
+                {
+                    InitialHandler initialHandler = (InitialHandler) handler;
+                    if ( initialHandler.getDisconnectMessage() != null )
+                    {
+                        ProxyServer.getInstance().getLogger().log( Level.INFO, "{0} has disconnected: {1}", new Object[]
+                        {
+                            handler, initialHandler.getDisconnectMessage().toPlainText()
+                        } );
+                    }
+                } else
+                {
+                    ProxyServer.getInstance().getLogger().log( Level.INFO, "{0} has disconnected", handler );
+                }
             }
         }
     }