#784 Perform bypass purge permission check with OfflinePlayer objects

- Move permission check inside PurgeTask to perform it with OfflinePlayer objects instead of lowercase names - Move purge members into child "purge" package - Unify online and offline default permission behavior in DefaultPermission
2016-07-17 11:54:22 +02:00
parent b439a0391c
commit 2a4cda0709
18 changed files with 102 additions and 149 deletions
@@ -0,0 +1,132 @@
+package fr.xephi.authme.task.purge;
+
+import fr.xephi.authme.ConsoleLogger;
+import fr.xephi.authme.permission.PermissionsManager;
+import fr.xephi.authme.permission.PlayerStatePermission;
+import org.bukkit.Bukkit;
+import org.bukkit.ChatColor;
+import org.bukkit.OfflinePlayer;
+import org.bukkit.command.CommandSender;
+import org.bukkit.entity.Player;
+import org.bukkit.scheduler.BukkitRunnable;
+
+import java.util.HashSet;
+import java.util.Set;
+import java.util.UUID;
+
+public class PurgeTask extends BukkitRunnable {
+
+    //how many players we should check for each tick
+    private static final int INTERVAL_CHECK = 5;
+
+    private final PurgeService purgeService;
+    private final PermissionsManager permissionsManager;
+    private final UUID sender;
+    private final Set<String> toPurge;
+
+    private final OfflinePlayer[] offlinePlayers;
+    private final int totalPurgeCount;
+
+    private int currentPage = 0;
+
+    /**
+     * Constructor.
+     *
+     * @param service the purge service
+     * @param permissionsManager the permissions manager
+     * @param sender the sender who initiated the purge, or null
+     * @param toPurge lowercase names to purge
+     * @param offlinePlayers offline players to map to the names
+     */
+    public PurgeTask(PurgeService service, PermissionsManager permissionsManager, CommandSender sender,
+                     Set<String> toPurge, OfflinePlayer[] offlinePlayers) {
+        this.purgeService = service;
+        this.permissionsManager = permissionsManager;
+
+        if (sender instanceof Player) {
+            this.sender = ((Player) sender).getUniqueId();
+        } else {
+            this.sender = null;
+        }
+
+        this.toPurge = toPurge;
+        this.totalPurgeCount = toPurge.size();
+        this.offlinePlayers = offlinePlayers;
+    }
+
+    @Override
+    public void run() {
+        if (toPurge.isEmpty()) {
+            //everything was removed
+            finish();
+            return;
+        }
+
+        Set<OfflinePlayer> playerPortion = new HashSet<>(INTERVAL_CHECK);
+        Set<String> namePortion = new HashSet<>(INTERVAL_CHECK);
+        for (int i = 0; i < INTERVAL_CHECK; i++) {
+            int nextPosition = (currentPage * INTERVAL_CHECK) + i;
+            if (offlinePlayers.length <= nextPosition) {
+                //no more offline players on this page
+                break;
+            }
+
+            OfflinePlayer offlinePlayer = offlinePlayers[nextPosition];
+            if (toPurge.remove(offlinePlayer.getName().toLowerCase())) {
+                if (!permissionsManager.hasPermissionOffline(offlinePlayer, PlayerStatePermission.BYPASS_PURGE)) {
+                    playerPortion.add(offlinePlayer);
+                    namePortion.add(offlinePlayer.getName());
+                }
+            }
+        }
+
+        if (!toPurge.isEmpty() && playerPortion.isEmpty()) {
+            ConsoleLogger.info("Finished lookup up offlinePlayers. Begin looking purging player names only");
+
+            //we went through all offlineplayers but there are still names remaining
+            for (String name : toPurge) {
+                if (!permissionsManager.hasPermissionOffline(name, PlayerStatePermission.BYPASS_PURGE)) {
+                    namePortion.add(name);
+                }
+            }
+            toPurge.clear();
+        }
+
+        currentPage++;
+        purgeData(playerPortion, namePortion);
+        if (currentPage % 20 == 0) {
+            int completed = totalPurgeCount - toPurge.size();
+            sendMessage("[AuthMe] Purge progress " + completed + '/' + totalPurgeCount);
+        }
+    }
+
+    private void purgeData(Set<OfflinePlayer> playerPortion, Set<String> namePortion) {
+        // Purge other data
+        purgeService.purgeEssentials(playerPortion);
+        purgeService.purgeDat(playerPortion);
+        purgeService.purgeLimitedCreative(namePortion);
+        purgeService.purgeAntiXray(namePortion);
+        purgeService.purgePermissions(playerPortion);
+    }
+
+    private void finish() {
+        cancel();
+
+        // Show a status message
+        sendMessage(ChatColor.GREEN + "[AuthMe] Database has been purged correctly");
+
+        ConsoleLogger.info("Purge Finished!");
+        purgeService.setPurging(false);
+    }
+
+    private void sendMessage(String message) {
+        if (sender == null) {
+            Bukkit.getConsoleSender().sendMessage(message);
+        } else {
+            Player player = Bukkit.getPlayer(sender);
+            if (player != null) {
+                player.sendMessage(message);
+            }
+        }
+    }
+}