From bc86a12c57e8e36952f1927cd63f35b7024e4972 Mon Sep 17 00:00:00 2001
From: Leymooo <mikimouse100@mail.ru>
Date: Wed, 12 Sep 2018 11:43:33 +0300
Subject: [PATCH] expand PreLoginComponentResult with force offline mode

---
 .../api/event/connection/PreLoginEvent.java   | 25 ++++++++++++++++---
 .../client/LoginSessionHandler.java           |  2 +-
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/api/src/main/java/com/velocitypowered/api/event/connection/PreLoginEvent.java b/api/src/main/java/com/velocitypowered/api/event/connection/PreLoginEvent.java
index ef8edfb9f..9d0620f8b 100644
--- a/api/src/main/java/com/velocitypowered/api/event/connection/PreLoginEvent.java
+++ b/api/src/main/java/com/velocitypowered/api/event/connection/PreLoginEvent.java
@@ -56,31 +56,41 @@ public class PreLoginEvent implements ResultedEvent<PreLoginEvent.PreLoginCompon
      */
     public static class PreLoginComponentResult extends ResultedEvent.ComponentResult {
         private static final PreLoginComponentResult ALLOWED = new PreLoginComponentResult((Component) null);
-        private static final PreLoginComponentResult FORCE_ONLINEMODE = new PreLoginComponentResult(true);
+        private static final PreLoginComponentResult FORCE_ONLINEMODE = new PreLoginComponentResult(true, false);
+        private static final PreLoginComponentResult FORCE_OFFLINEMODE = new PreLoginComponentResult(false, true);
 
         private final boolean onlineMode;
-
+        private final boolean forceOfflineMode;
         /**
          * Allows online mode to be enabled for the player connection, if Velocity is running in offline mode.
          * @param allowedOnlineMode if true, online mode will be used for the connection
          */
-        private PreLoginComponentResult(boolean allowedOnlineMode) {
+        private PreLoginComponentResult(boolean allowedOnlineMode, boolean forceOfflineMode) {
             super(true, null);
             this.onlineMode = allowedOnlineMode;
+            this.forceOfflineMode = forceOfflineMode;
         }
 
         private PreLoginComponentResult(@Nullable Component reason) {
             super(reason == null, reason);
             // Don't care about this
             this.onlineMode = false;
+            this.forceOfflineMode = false;
         }
 
         public boolean isOnlineModeAllowed() {
             return this.onlineMode;
         }
 
+        public boolean isForceOfflineMode() {
+            return forceOfflineMode;
+        }
+
         @Override
         public String toString() {
+            if (isForceOfflineMode()) {
+                return "allowed with force offline mode";
+            }
             if (isOnlineModeAllowed()) {
                 return "allowed with online mode";
             }
@@ -106,6 +116,15 @@ public class PreLoginEvent implements ResultedEvent<PreLoginEvent.PreLoginCompon
             return FORCE_ONLINEMODE;
         }
 
+        /**
+         * Returns a result indicating the connection will be allowed through the proxy, but the connection will be
+         * forced to use offline mode even when proxy running in online mode
+         * @return the result
+         */
+        public static PreLoginComponentResult forceOfflineMode() {
+            return FORCE_OFFLINEMODE;
+        }
+
         /**
          * Denies the login with the specified reason.
          * @param reason the reason for disallowing the connection
diff --git a/proxy/src/main/java/com/velocitypowered/proxy/connection/client/LoginSessionHandler.java b/proxy/src/main/java/com/velocitypowered/proxy/connection/client/LoginSessionHandler.java
index fed5cb580..63890c3f0 100644
--- a/proxy/src/main/java/com/velocitypowered/proxy/connection/client/LoginSessionHandler.java
+++ b/proxy/src/main/java/com/velocitypowered/proxy/connection/client/LoginSessionHandler.java
@@ -159,7 +159,7 @@ public class LoginSessionHandler implements MinecraftSessionHandler {
                         return;
                     }
 
-                    if (server.getConfiguration().isOnlineMode() || result.isOnlineModeAllowed()) {
+                    if (!result.isForceOfflineMode() && (server.getConfiguration().isOnlineMode() || result.isOnlineModeAllowed())) {
                         // Request encryption.
                         EncryptionRequest request = generateRequest();
                         this.verify = Arrays.copyOf(request.getVerifyToken(), 4);