Add clear old perms task

Winds-Studio · May 30, 2024 · b7d0d36 · b7d0d36
1 parent 29e1b0e
commit b7d0d36
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/src/main/java/cn/dreeam/caeruleum/CaeruleumCore.java b/src/main/java/cn/dreeam/caeruleum/CaeruleumCore.java
@@ -3,14 +3,19 @@
 import cn.dreeam.caeruleum.config.Config;
 import cn.dreeam.caeruleum.config.ConfigManager;
 import cn.dreeam.caeruleum.listener.LocaleChange;
+import cn.dreeam.caeruleum.utils.PermUtil;
 import net.luckperms.api.LuckPerms;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.bukkit.Bukkit;
+import org.bukkit.OfflinePlayer;
 import org.bukkit.plugin.PluginManager;
 import org.bukkit.plugin.RegisteredServiceProvider;
 import org.bukkit.plugin.java.JavaPlugin;
 
+import java.util.Arrays;
+import java.util.stream.Collectors;
+
 public class CaeruleumCore extends JavaPlugin {
 
     private static CaeruleumCore instance;
@@ -30,6 +35,7 @@ public void onEnable() {
         loadConfig();
         registerEvents();
         initHooks();
+        initTasks();
         //new Metrics(instance, 16810);
 
         LOGGER.info("CaeruleumCore {} enabled. By Dreeam.", instance.getDescription().getVersion());
@@ -51,6 +57,7 @@ private void loadConfig() {
     }
 
     private void initHooks() {
+        // Hook LuckPerms
         RegisteredServiceProvider<LuckPerms> luckPermsProvider = Bukkit.getServicesManager().getRegistration(LuckPerms.class);
         if (luckPermsProvider != null) {
             luckPermsAPI = luckPermsProvider.getProvider();
@@ -60,6 +67,23 @@ private void initHooks() {
         }
     }
 
+    private void initTasks() {
+        // Clear old perms task
+        if (!config.oldLangPermPrefixList().isEmpty()) {
+            Thread.startVirtualThread(
+                    () -> Arrays.stream(Bukkit.getServer().getOfflinePlayers())
+                    .filter(OfflinePlayer::hasPlayedBefore)
+                    .collect(Collectors.toSet())
+                    .forEach(
+                            p -> config.oldLangPermPrefixList()
+                                    .forEach(
+                                    oldLangPermPrefix -> PermUtil.clearLangPerm(p.getUniqueId(), oldLangPermPrefix)
+                            )
+                    )
+            );
+        }
+    }
+
     public static CaeruleumCore getInstance() {
         return instance;
     }

diff --git a/src/main/java/cn/dreeam/caeruleum/utils/PermUtil.java b/src/main/java/cn/dreeam/caeruleum/utils/PermUtil.java
@@ -48,8 +48,12 @@ public static void modifyLangPerm(UUID uuid, String langPerm) {
     }
 
     public static void clearLangPerm(UUID uuid) {
+        clearLangPerm(uuid, CaeruleumCore.config.langPermKeyPrefix());
+    }
+
+    public static void clearLangPerm(UUID uuid, String permPrefix) {
         CaeruleumCore.getLuckPermsAPI().getUserManager().modifyUser(uuid, user ->
-                user.data().clear(x -> x.getKey().startsWith(CaeruleumCore.config.langPermKeyPrefix()))
+                user.data().clear(x -> x.getKey().startsWith(permPrefix))
         );
     }
 }