Support to change disconnect message when joining server.

OKTW-Network · Aug 12, 2023 · d49fe8f · d49fe8f
1 parent 258d0ee
commit d49fe8f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/main/java/one/oktw/ModConfig.java b/src/main/java/one/oktw/ModConfig.java
@@ -5,8 +5,18 @@ public class ModConfig {
     private boolean hackOnlineMode = true;
     private boolean hackEarlySend = false;
     private boolean hackMessageChain = true;
+    private String disconnectMessage = "This server requires you to connect with Velocity.";
     private String secret = "";
 
+    public String getAbortedMessage() {
+        String env = System.getenv("FABRIC_PROXY_MESSAGE");
+        if (env == null) {
+            return disconnectMessage;
+        } else {
+            return (env);
+        }
+    }
+
     public boolean getHackOnlineMode() {
         String env = System.getenv("FABRIC_PROXY_HACK_ONLINE_MODE");
         if (env == null) {

diff --git a/src/main/java/one/oktw/PacketHandler.java b/src/main/java/one/oktw/PacketHandler.java
@@ -25,7 +25,7 @@ class PacketHandler {
 
     void handleVelocityPacket(MinecraftServer server, ServerLoginNetworkHandler handler, boolean understood, PacketByteBuf buf, ServerLoginNetworking.LoginSynchronizer synchronizer, PacketSender ignored) {
         if (!understood) {
-            handler.disconnect(Text.of("This server requires you to connect with Velocity."));
+            handler.disconnect(Text.of(config.getAbortedMessage()));
             return;
         }