From 0fd903e0a02a2f4d8db8775534b0b8ba89182f38 Mon Sep 17 00:00:00 2001
From: Camotoy <20743703+Camotoy@users.noreply.github.com>
Date: Mon, 28 Feb 2022 10:24:27 -0500
Subject: [PATCH] Pending Microsoft Authentication changes for GeyserConnect

---
 .../PendingMicrosoftAuthentication.java       | 23 ++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/core/src/main/java/org/geysermc/geyser/session/PendingMicrosoftAuthentication.java b/core/src/main/java/org/geysermc/geyser/session/PendingMicrosoftAuthentication.java
index 044389d24..696d6b088 100644
--- a/core/src/main/java/org/geysermc/geyser/session/PendingMicrosoftAuthentication.java
+++ b/core/src/main/java/org/geysermc/geyser/session/PendingMicrosoftAuthentication.java
@@ -46,6 +46,10 @@ import java.util.concurrent.*;
  * It permits user to exit the server while they authorize Geyser to access their Microsoft account.
  */
 public class PendingMicrosoftAuthentication {
+    /**
+     * For GeyserConnect usage.
+     */
+    private boolean storeServerInformation = false;
     private final LoadingCache<String, AuthenticationTask> authentications;
 
     public PendingMicrosoftAuthentication(int timeoutSeconds) {
@@ -53,7 +57,8 @@ public class PendingMicrosoftAuthentication {
                 .build(new CacheLoader<>() {
                     @Override
                     public AuthenticationTask load(@NonNull String userKey) {
-                        return new AuthenticationTask(userKey, timeoutSeconds * 1000L);
+                        return storeServerInformation ? new ProxyAuthenticationTask(userKey, timeoutSeconds * 1000L)
+                                : new AuthenticationTask(userKey, timeoutSeconds * 1000L);
                     }
                 });
     }
@@ -67,6 +72,11 @@ public class PendingMicrosoftAuthentication {
         return authentications.get(userKey);
     }
 
+    @SuppressWarnings("unused") // GeyserConnect
+    public void setStoreServerInformation() {
+        storeServerInformation = true;
+    }
+
     public class AuthenticationTask {
         private static final Executor DELAYED_BY_ONE_SECOND = CompletableFuture.delayedExecutor(1, TimeUnit.SECONDS);
 
@@ -159,6 +169,17 @@ public class PendingMicrosoftAuthentication {
         }
     }
 
+    @Getter
+    @Setter
+    public final class ProxyAuthenticationTask extends AuthenticationTask {
+        private String server;
+        private int port;
+
+        private ProxyAuthenticationTask(String userKey, long timeoutMs) {
+            super(userKey, timeoutMs);
+        }
+    }
+
     /**
      * @see PendingMicrosoftAuthentication
      */