0.5 Maybe fixes an issue. New SyncResult class

Velocity

@@ -0,0 +1 @@
+Subproject commit 5fe3663d5102c98df1128b06f002e0c5b212f61f

build.gradle

@@ -4,7 +4,7 @@ plugins {
 }
 
 group 'org.adde0109'
-version '0.4.0'
+version '0.5.0'
 
 repositories {
     maven {

src/main/java/org/adde0109/ambassador/Ambassador.java

@@ -23,7 +23,7 @@ import org.slf4j.Logger;
 import java.nio.file.Path;
 import java.util.*;
 
-@Plugin(id = "ambassador", name = "Ambassador", version = "0.4.0", authors = {"adde0109"})
+@Plugin(id = "ambassador", name = "Ambassador", version = "0.5.0", authors = {"adde0109"})
 public class Ambassador {
 
   public ProxyServer server;
@@ -82,7 +82,7 @@ public class Ambassador {
       //Forge client
       ForgeConnection forgeConnection = forgeHandshakeHandler.getForgeConnection(event.getPlayer()).get();
       if (forgeConnection.isForced()) {
-        event.setInitialServer(forgeConnection.getSyncedServer().get());
+        event.setInitialServer(forgeConnection.getSyncResult().get().getSyncedServer());
       }
       forgeConnection.setForced(config.getForced(forgeConnection.getConnection().getProtocolVersion().getProtocol()));
     }

src/main/java/org/adde0109/ambassador/forge/ForgeConnection.java

@@ -18,13 +18,11 @@ public class ForgeConnection {
   private final Logger logger;
   private final LoginPhaseConnection connection;
 
-  private Optional<byte[]> recivedClientModlist = Optional.empty();
-  private static byte[] recivedClientACK;
   private boolean ignoreSyncExepction = false;
 
-  private Optional<ForgeHandshakeUtils.CachedServerHandshake> transmittedHandshake = Optional.empty();
   private boolean forced = false;
-  private Optional<RegisteredServer> syncedTo = Optional.empty();
+
+  private SyncResult syncResult;
 
 
   public ForgeConnection(LoginPhaseConnection connection, Logger logger) {
@@ -47,34 +45,35 @@ public class ForgeConnection {
 
 
 
-  public CompletableFuture<Boolean> sync(ForgeServerConnection forgeServerConnection) {
+  public void startSync(ForgeServerConnection forgeServerConnection, Continuation continuation) {
-    CompletableFuture<Boolean> future = new CompletableFuture<>();
     forgeServerConnection.getHandshake().whenComplete((msg,ex) -> {
       if (ex != null) {
-        future.complete(false);
+        continuation.resume();
         logger.warn("Sync Exception: " + ex);
       } else {
+        CompletableFuture<byte[]> clientModListFuture = new CompletableFuture<>();
         //This gets also sent to vanilla
         sendModlist(msg.modListPacket).thenAccept((response) -> {
           if (!ignoreSyncExepction && response == null) {
             logger.warn("Sync Exception: Client responded with an empty body.");
           }
-          recivedClientModlist = Optional.ofNullable(response);
+          clientModListFuture.complete(response);
         });
+        syncResult = new SyncResult(msg,clientModListFuture, forgeServerConnection.getServer());
         //This gets also sent to vanilla
         sendOther(msg.otherPackets).thenAccept((response) -> {
           if (!ignoreSyncExepction && response == null) {
             logger.warn("Sync Exception: Client responded with an empty body.");
           }
           //TODO: Generate the ACK packet ourself.
-          ForgeConnection.recivedClientACK = (response == null) ? ForgeConnection.recivedClientACK : response;
+          if (response != null && SyncResult.recivedClientACK == null) {
-          transmittedHandshake = Optional.of(msg);
+            SyncResult.recivedClientACK = response;
-          syncedTo = Optional.of(forgeServerConnection.getServer());
+          }
+          syncResult.complete(syncResult);
         });
-        future.complete(true);
+        continuation.resume();
       }
     });
-    return future;
   }
 
   private CompletableFuture<byte[]> sendModlist(byte[] modListPacket) {
@@ -95,6 +94,11 @@ public class ForgeConnection {
   }
 
   public void handleServerHandshakePacket(ServerLoginPluginMessageEvent event, Continuation continuation) {
+    if (getSyncResult().isEmpty()) {
+      continuation.resumeWithException(new Exception("Client isn't synced. This should have been caught" +
+              " during serverPreConnect"));
+      return;
+    }
     ByteArrayDataInput data = event.contentsAsDataStream();
     if (data.skipBytes(14) != 14) {  //Channel Identifier
       continuation.resumeWithException(new EOFException());
@@ -104,41 +108,27 @@ public class ForgeConnection {
     int packetID = ForgeHandshakeUtils.readVarInt(data);
 
     if (packetID == 1) {
-      if (getRecivedClientModlist().isPresent()) {
+      getSyncResult().get().getRecivedClientModlist().whenComplete((msg,ex) -> {
-        event.setResult(ServerLoginPluginMessageEvent.ResponseResult.reply(getRecivedClientModlist().get()));
+        event.setResult(ServerLoginPluginMessageEvent.ResponseResult.reply(msg));
-      } else {
+        continuation.resume();
-        continuation.resumeWithException(new Exception("Client isn't synced. This should have been caught" +
+      });
-                " during serverPreConnect"));
-        return;
-      }
     } else {
-      if (getRecivedClientACK() != null) {
+      if (SyncResult.recivedClientACK != null) {
-        event.setResult(ServerLoginPluginMessageEvent.ResponseResult.reply(getRecivedClientACK()));
+        event.setResult(ServerLoginPluginMessageEvent.ResponseResult.reply(SyncResult.recivedClientACK));
       } else {
-        continuation.resumeWithException(new Exception("No response available."));
+        continuation.resumeWithException(new Exception("No ACK response packet available."));
         return;
       }
+      continuation.resume();
     }
-    continuation.resume();
 }
 
   public LoginPhaseConnection getConnection() {
     return connection;
   }
 
-  public Optional<ForgeHandshakeUtils.CachedServerHandshake> getTransmittedHandshake() {
+  public Optional<SyncResult> getSyncResult() {
-    return transmittedHandshake;
+    return Optional.ofNullable(syncResult);
-  }
-
-  public Optional<byte[]> getRecivedClientModlist() {
-    return recivedClientModlist;
-  }
-
-  public static byte[] getRecivedClientACK() {
-    return recivedClientACK;
-  }
-  public Optional<RegisteredServer> getSyncedServer() {
-    return syncedTo;
   }
   public void setForced(boolean forced) {
     this.forced = forced;
@@ -146,4 +136,28 @@ public class ForgeConnection {
   public boolean isForced() {
     return forced;
   }
+  public static class SyncResult extends CompletableFuture<SyncResult> {
+    private final ForgeHandshakeUtils.CachedServerHandshake transmittedHandshake;
+    private final CompletableFuture<byte[]> recivedClientModlist;
+    private static byte[] recivedClientACK;
+    private final RegisteredServer syncedTo;
+
+    SyncResult(ForgeHandshakeUtils.CachedServerHandshake transmittedHandshake, CompletableFuture<byte[]> recivedClientModlist, RegisteredServer syncedTo) {
+      this.transmittedHandshake = transmittedHandshake;
+      this.recivedClientModlist = recivedClientModlist;
+      this.syncedTo = syncedTo;
+    }
+
+    public ForgeHandshakeUtils.CachedServerHandshake getTransmittedHandshake() {
+      return transmittedHandshake;
+    }
+
+    public CompletableFuture<byte[]> getRecivedClientModlist() {
+      return recivedClientModlist;
+    }
+
+    public RegisteredServer getSyncedServer() {
+      return syncedTo;
+    }
+  }
 }

src/main/java/org/adde0109/ambassador/forge/ForgeHandshakeHandler.java

@@ -52,9 +52,7 @@ public class ForgeHandshakeHandler {
         });
 
     if (ambassador.forgeServerSwitchHandler.reSyncMap.containsKey(event.getUsername())) {
-      forgeConnection.sync(ambassador.forgeServerSwitchHandler.reSyncMap.remove(event.getUsername())).thenAccept((done) -> {
+      forgeConnection.startSync(ambassador.forgeServerSwitchHandler.reSyncMap.remove(event.getUsername()),continuation);
-        continuation.resume();
-      });
       forgeConnection.setForced(true);
     } else if (defaultServer != null) {
       //If a connection does not already exist, create one.
@@ -62,9 +60,7 @@ public class ForgeHandshakeHandler {
         forgeServerConnectionMap.put(defaultServer, new ForgeServerConnection(defaultServer));
       }
       //Forge Handshake
-      forgeConnection.sync(forgeServerConnectionMap.get(defaultServer)).thenAccept((done) -> {
+      forgeConnection.startSync(forgeServerConnectionMap.get(defaultServer),continuation);
-        continuation.resume();
-      });
       forgeConnection.setForced(ambassador.config.getForced(forgeConnection.getConnection().getProtocolVersion().getProtocol()));
     } else {
       continuation.resume();

src/main/java/org/adde0109/ambassador/forge/ForgeServerSwitchHandler.java

@@ -34,8 +34,9 @@ public class ForgeServerSwitchHandler {
       return;
     }
     Optional<ForgeServerConnection> forgeServerConnectionOptional = ambassador.forgeHandshakeHandler.getForgeServerConnection(event.getOriginalServer());
-    Optional<ForgeConnection> forgeConnection = ambassador.forgeHandshakeHandler.getForgeConnection(event.getPlayer());
+    Optional<ForgeConnection> forgeConnectionOptional = ambassador.forgeHandshakeHandler.getForgeConnection(event.getPlayer());
-    if (forgeConnection.isPresent()) {
+    if (forgeConnectionOptional.isPresent()) {
+      ForgeConnection forgeConnection = forgeConnectionOptional.get();
       ForgeServerConnection forgeServerConnection = forgeServerConnectionOptional.orElseGet(() -> new ForgeServerConnection(event.getOriginalServer()));
       forgeServerConnection.getHandshake().whenComplete((msg, ex) -> {
         if (ex != null) {
@@ -56,7 +57,7 @@ public class ForgeServerSwitchHandler {
           event.getPlayer().setGameProfileProperties(properties);
 
           if (ambassador.config.reSyncOptionForge() != AmbassadorConfig.reSyncOption.NEVER) {
-            if (forgeConnection.get().getTransmittedHandshake().isEmpty() || !msg.equals(forgeConnection.get().getTransmittedHandshake().get())) {
+            if (forgeConnection.getSyncResult().isEmpty() || !msg.equals(forgeConnection.getSyncResult().get().getTransmittedHandshake())) {
               event.setResult(ServerPreConnectEvent.ServerResult.denied());
               reSync(event.getPlayer(),forgeServerConnection);
             }