minor bugfixes

src/main/java/com/actualplayer/rememberme/handlers/FileHandler.java

@@ -25,7 +25,7 @@ public class FileHandler implements IRememberMeHandler {
     public CompletableFuture<String> getLastServerName(UUID uuid) {
         try {
             File userFile = FileUtils.getOrCreate(rememberMe.getDataFolderPath().resolve("data"), uuid.toString() + ".yml");
-            UserServer userServer = YamlUtils.readFile(userFile, UserServer.class);
+            UserServer userServer = YamlUtils.readFile(userFile, com.actualplayer.rememberme.models.UserServer.class);
 
             if(userServer == null) return null;
 

src/main/java/com/actualplayer/rememberme/handlers/LuckPermsHandler.java

@@ -9,10 +9,12 @@ import net.luckperms.api.model.user.User;
 import net.luckperms.api.model.user.UserManager;
 import net.luckperms.api.node.Node;
 import net.luckperms.api.node.NodeType;
+import net.luckperms.api.node.types.MetaNode;
 import net.luckperms.api.query.QueryMode;
 import net.luckperms.api.query.QueryOptions;
 import org.checkerframework.checker.nullness.Opt;
 
+import java.util.Collection;
 import java.util.Optional;
 import java.util.UUID;
 import java.util.concurrent.CompletableFuture;
@@ -28,14 +30,19 @@ public class LuckPermsHandler implements IRememberMeHandler {
     @Override
     public CompletableFuture<String> getLastServerName(UUID uuid) {
         UserManager userManager = api.getUserManager();
-        CompletableFuture<User> userFuture = userManager.loadUser(uuid);
+        return userManager.loadUser(uuid).thenApply(user -> {
-        return userFuture.thenApply(user -> {
             if (user != null) {
-                ContextManager cm = api.getContextManager();
+                MetaNode lastServerNode = user.getNodes().stream()
-                ImmutableContextSet context = cm.getContext(user).orElse(cm.getStaticContext());
+                    .filter(NodeType.META::matches)
+                    .map(NodeType.META::cast)
+                    .filter(n -> n.getMetaKey().equals("last-server"))
+                    .findFirst().orElse(null);
 
-                CachedMetaData metaData = user.getCachedData().getMetaData(QueryOptions.builder(QueryMode.CONTEXTUAL).context(context).build());
+                if (lastServerNode != null) {
-                return metaData.getMetaValue("last-server");
+                    return lastServerNode.getMetaValue();
+                }
+
+                return null;
             }
 
             return null;
@@ -44,15 +51,22 @@ public class LuckPermsHandler implements IRememberMeHandler {
 
     @Override
     public void setLastServerName(UUID uuid, String serverName) {
-        User user = api.getUserManager().getUser(uuid);
+        api.getUserManager().loadUser(uuid).thenAcceptAsync(user -> {
             if (user != null) {
-            // Remove last server
+                // Find last server
-            user.getNodes().removeIf(n -> n.getType() == NodeType.META && n.getKey().contains("last-server"));
+                MetaNode serverNode = user.getNodes().stream()
+                        .filter(NodeType.META::matches)
+                        .map(NodeType.META::cast)
+                        .filter(n -> n.getMetaKey().equals("last-server"))
+                        .findFirst().orElse(MetaNode.builder("last-server", serverName).build());
 
-            // Add current server as last server
+                user.data().remove(serverNode);
-            Node node = api.getNodeBuilderRegistry().forMeta().key("last-server").value(serverName).build();
+                serverNode = serverNode.toBuilder().value(serverName).build();
-            user.getNodes().add(node);
+                user.data().add(serverNode);
+
+                // Save changes
                 api.getUserManager().saveUser(user);
             }
+        });
     }
 }

src/main/java/com/actualplayer/rememberme/models/UserServer.java

@@ -1,6 +1,5 @@
 package com.actualplayer.rememberme.models;
 
-
 import lombok.Data;
 
 @Data