used correct server event for redirect

ActualPlayer · Jul 13, 2020 · 59495d0 · 59495d0
1 parent 56d8254
commit 59495d0
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/main/java/com/actualplayer/rememberme/RememberMe.java b/src/main/java/com/actualplayer/rememberme/RememberMe.java
@@ -5,6 +5,7 @@
 import com.google.inject.name.Named;
 import com.velocitypowered.api.event.Subscribe;
 import com.velocitypowered.api.event.connection.LoginEvent;
+import com.velocitypowered.api.event.player.PlayerChooseInitialServerEvent;
 import com.velocitypowered.api.event.player.ServerConnectedEvent;
 import com.velocitypowered.api.event.proxy.ProxyInitializeEvent;
 import com.velocitypowered.api.plugin.Dependency;
@@ -65,14 +66,12 @@ public void onProxyInitialization(ProxyInitializeEvent event) {
     }
 
     @Subscribe
-    public void onLoginEvent(LoginEvent loginEvent) {
+    public void onServerChooseEvent(PlayerChooseInitialServerEvent chooseServerEvent) {
         // Ignore plugin when user has notransfer permission
-        if (!loginEvent.getPlayer().hasPermission("rememberme.notransfer")) {
-            handler.getLastServerName(loginEvent.getPlayer().getUniqueId()).thenAcceptAsync(lastServerName -> {
+        if (!chooseServerEvent.getPlayer().hasPermission("rememberme.notransfer")) {
+            handler.getLastServerName(chooseServerEvent.getPlayer().getUniqueId()).thenAcceptAsync(lastServerName -> {
                 if (lastServerName != null) {
-                    getServer().getServer(lastServerName).ifPresent(lastServer -> {
-                        loginEvent.getPlayer().createConnectionRequest(lastServer).connectWithIndication();
-                    });
+                    getServer().getServer(lastServerName).ifPresent(chooseServerEvent::setInitialServer);
                 }
             }).join();
         }