add bungeecord-proxy dependency

Author: Kamesuta

pom.xml

@@ -64,5 +64,11 @@
       <version>1.21-R0.1-SNAPSHOT</version>
       <scope>provided</scope>
     </dependency>
+    <dependency>
+      <groupId>net.md-5</groupId>
+      <artifactId>bungeecord-proxy</artifactId>
+      <version>1.20-R0.3-SNAPSHOT</version>
+      <scope>provided</scope>
+    </dependency>
   </dependencies>
 </project>

src/main/java/com/kamesuta/bungeeviaproxy/BungeeViaProxy.java

@@ -7,9 +7,10 @@
 import net.md_5.bungee.api.event.ServerConnectedEvent;
 import net.md_5.bungee.api.plugin.Listener;
 import net.md_5.bungee.api.plugin.Plugin;
+import net.md_5.bungee.connection.InitialHandler;
 import net.md_5.bungee.event.EventHandler;
+import net.md_5.bungee.protocol.packet.Handshake;
 
-import java.net.InetSocketAddress;
 import java.net.SocketAddress;
 import java.util.logging.LogManager;
 import java.util.logging.Logger;
@@ -55,8 +56,9 @@ public void onServerConnect(ServerConnectEvent event) {
         // To address this, any address containing ".viaproxy." is passed directly to the target server.
         if (host.contains(IDENTIFIER)) {
             PendingConnection con = event.getPlayer().getPendingConnection();
-            if (ReflectionUtil.isInitialHandler(con)) {
-                ReflectionUtil.setInitialHandlerHandshakeHost(con, host);
+            if (con instanceof InitialHandler) {
+                Handshake handshake = ((InitialHandler) con).getHandshake();
+                handshake.setHost(host);
             }
         }
     }
@@ -66,8 +68,9 @@ public void onServerConnected(ServerConnectedEvent event) {
         // Reverse the changes made in onServerConnect after the connection is established
         PendingConnection con = event.getPlayer().getPendingConnection();
         String host = con.getVirtualHost().getHostString();
-        if (host.contains(IDENTIFIER) && ReflectionUtil.isInitialHandler(con)) {
-            ReflectionUtil.setInitialHandlerHandshakeHost(con, host);
+        if (host.contains(IDENTIFIER) && con instanceof InitialHandler) {
+            Handshake handshake = ((InitialHandler) con).getHandshake();
+            handshake.setHost(host);
         }
     }
 

src/main/java/com/kamesuta/bungeeviaproxy/ReflectionUtil.java

@@ -8,18 +8,12 @@
 
 public final class ReflectionUtil {
 
-    public static Field initialHandlerHandshakeField, handshakeHostField, bungeeServerInfoSocketAddressField;
+    public static Field bungeeServerInfoSocketAddressField;
 
-    public static Class<?> initialHandlerClass, handshakeClass, bungeeServerInfoClass;
+    public static Class<?> bungeeServerInfoClass;
 
     static {
         try {
-            initialHandlerClass = Class.forName("net.md_5.bungee.connection.InitialHandler");
-            initialHandlerHandshakeField = initialHandlerClass.getDeclaredField("handshake");
-            initialHandlerHandshakeField.setAccessible(true);
-            handshakeClass = Class.forName("net.md_5.bungee.protocol.packet.Handshake");
-            handshakeHostField = handshakeClass.getDeclaredField("host");
-            handshakeHostField.setAccessible(true);
             bungeeServerInfoClass = Class.forName("net.md_5.bungee.BungeeServerInfo");
             bungeeServerInfoSocketAddressField = bungeeServerInfoClass.getDeclaredField("socketAddress");
             bungeeServerInfoSocketAddressField.setAccessible(true);
@@ -31,19 +25,6 @@ public final class ReflectionUtil {
     private ReflectionUtil() {
     }
 
-    public static boolean isInitialHandler(Object obj) {
-        return initialHandlerClass.isAssignableFrom(obj.getClass());
-    }
-
-    public static void setInitialHandlerHandshakeHost(Object obj, String host) {
-        try {
-            Object handshake = initialHandlerHandshakeField.get(obj);
-            handshakeHostField.set(handshake, host);
-        } catch (Exception e) {
-            logger.log(java.util.logging.Level.SEVERE, "Failed to set handshake host", e);
-        }
-    }
-
     public static boolean isBungeeSeverInfo(Object obj) {
         return bungeeServerInfoClass.isAssignableFrom(obj.getClass());
     }