package me.saif.betterenderchests.enderchest; import com.google.common.collect.Sets; import de.tr7zw.changeme.nbtapi.utils.MinecraftVersion; import me.saif.betterenderchests.VariableEnderChests; import me.saif.betterenderchests.data.DataManager; import me.saif.betterenderchests.data.SQLDataManager; import me.saif.betterenderchests.lang.MessageKey; import me.saif.betterenderchests.utils.Callback; import me.saif.betterenderchests.utils.CaselessString; import me.saif.betterenderchests.utils.Manager; import org.bukkit.Bukkit; import org.bukkit.Material; import org.bukkit.Sound; import org.bukkit.block.Block; import org.bukkit.entity.Entity; import org.bukkit.entity.HumanEntity; import org.bukkit.entity.Player; import org.bukkit.event.EventHandler; import org.bukkit.event.EventPriority; import org.bukkit.event.Listener; import org.bukkit.event.block.Action; import org.bukkit.event.inventory.InventoryCloseEvent; import org.bukkit.event.player.PlayerInteractEvent; import org.bukkit.event.player.PlayerJoinEvent; import org.bukkit.event.player.PlayerLoginEvent; import org.bukkit.event.player.PlayerQuitEvent; import org.bukkit.inventory.ItemStack; import java.util.*; import java.util.concurrent.ConcurrentHashMap; import java.util.function.Function; import java.util.stream.Collectors; public class EnderChestManager extends Manager implements Listener { private final DataManager dataManager; private final Map uuidEnderChestMap = new ConcurrentHashMap<>(); private final Map nameUUIDMap = new HashMap<>(); private final Map lastModifiedCache = new ConcurrentHashMap<>(); private final Map> uuidCallbackMap = new HashMap<>(); private final Map> nameCallbackMap = new HashMap<>(); private final Set snapshotsToSave = Sets.newConcurrentHashSet(); private final Map openFromBlocks = new HashMap<>(); private int defaultRows; private final boolean convert; private final boolean crossServerSync; private final Sound OPEN_SOUND; private final Sound CLOSE_SOUND; private final Set blacklist = new HashSet<>(); public EnderChestManager(VariableEnderChests plugin) { super(plugin); this.dataManager = getPlugin().getDataManager(); this.crossServerSync = getPlugin().getConfig().getBoolean("cross-server-sync", false); if (MinecraftVersion.getVersion() == MinecraftVersion.MC1_8_R3) { OPEN_SOUND = Sound.valueOf("CHEST_OPEN"); CLOSE_SOUND = Sound.valueOf("CHEST_CLOSE"); } else if (!MinecraftVersion.isNewerThan(MinecraftVersion.MC1_12_R1)) { OPEN_SOUND = Sound.valueOf("BLOCK_ENDERCHEST_OPEN"); CLOSE_SOUND = Sound.valueOf("BLOCK_ENDERCHEST_CLOSE"); } else { OPEN_SOUND = Sound.valueOf("BLOCK_ENDER_CHEST_OPEN"); CLOSE_SOUND = Sound.valueOf("BLOCK_ENDER_CHEST_CLOSE"); } this.convert = this.getPlugin().getConfig().getBoolean("convert-current-ender-chest", true); this.defaultRows = this.getPlugin().getConfig().getInt("default-rows", 3); if (this.defaultRows > 6) this.defaultRows = 6; else if (this.defaultRows < 0) this.defaultRows = 0; for (String s : this.getPlugin().getConfig().getStringList("blacklisted-items")) { Material material = Material.matchMaterial(s); if (material == null) { plugin.getLogger().warning("Blacklisted item by the name of \"" + s + "\" found. This is not a valid minecraft material. Ignoring..."); continue; } this.blacklist.add(material); } Bukkit.getScheduler().runTask(plugin, () -> { Set toGet = Bukkit.getOnlinePlayers().stream().map((Function) Entity::getUniqueId).collect(Collectors.toSet()); Map data = this.dataManager.loadEnderChestsByUUID(toGet); for (Player player : Bukkit.getOnlinePlayers()) { this.dataManager.saveNameAndUUID(player.getName(), player.getUniqueId()); EnderChestSnapshot snapshot = data.get(player.getUniqueId()); if (snapshot == null) { this.getPlugin().getLogger().info("Creating new enderchest for " + player.getName()); this.uuidEnderChestMap.put(player.getUniqueId(), createNew(player)); } else { this.getPlugin().getLogger().info("Loaded enderchest for " + player.getName()); EnderChest ec = new EnderChest(player.getUniqueId(), player.getName(), snapshot.getContents(), snapshot.getRows()); this.uuidEnderChestMap.put(player.getUniqueId(), ec); this.lastModifiedCache.put(player.getUniqueId(), getLastModifiedFromDB(player.getUniqueId())); } if (this.getPlugin().getDisabledWorlds().contains(player.getWorld().getName())) continue; if (player.getOpenInventory().getTopInventory().equals(player.getEnderChest())) { player.closeInventory(); int rows = this.getNumRows(player); if (rows == 0) { this.getPlugin().getMessenger().sendMessage(player, MessageKey.NO_ENDERCHEST_SELF); } this.openEnderChest(this.getEnderChest(player), player, this.getNumRows(player)); } } }); int saveInterval = crossServerSync ? 1200L : 6000L; Bukkit.getScheduler().scheduleSyncRepeatingTask(this.getPlugin(), () -> { Map toSave = new HashMap<>(); Set toRemove = new HashSet<>(); for (EnderChest enderChest : this.uuidEnderChestMap.values()) { toSave.put(enderChest.getUUID(), enderChest.snapshot()); if (Bukkit.getPlayer(enderChest.getUUID()) == null) { toRemove.add(enderChest.getUUID()); } } for (UUID uuid : toRemove) { EnderChest enderChest = this.uuidEnderChestMap.remove(uuid); this.lastModifiedCache.remove(uuid); this.getPlugin().getLogger().info("Saving data for " + enderChest.getName()); } Bukkit.getScheduler().runTaskAsynchronously(this.getPlugin(), () -> this.dataManager.saveEnderChestMultiple(toSave)); for (CaselessString minecraftName : this.nameCallbackMap.keySet()) { Player player = Bukkit.getPlayerExact(minecraftName.getOriginal()); if (player == null) this.nameCallbackMap.remove(minecraftName); } }, saveInterval, saveInterval); if (crossServerSync && dataManager instanceof SQLDataManager) { int syncInterval = getPlugin().getConfig().getInt("sync-interval", 60) * 20; Bukkit.getScheduler().scheduleSyncRepeatingTask(this.getPlugin(), () -> { for (Player player : Bukkit.getOnlinePlayers()) { checkAndReloadEnderChest(player); } }, syncInterval, syncInterval); } } private long getLastModifiedFromDB(UUID uuid) { if (dataManager instanceof SQLDataManager) { return ((SQLDataManager) dataManager).getLastModified(uuid); } return 0; } private void checkAndReloadEnderChest(Player player) { if (!crossServerSync || !(dataManager instanceof SQLDataManager)) return; UUID uuid = player.getUniqueId(); EnderChest enderChest = this.uuidEnderChestMap.get(uuid); if (enderChest == null || enderChest.hasViewers()) return; Bukkit.getScheduler().runTaskAsynchronously(this.getPlugin(), () -> { long dbLastModified = getLastModifiedFromDB(uuid); long cacheLastModified = lastModifiedCache.getOrDefault(uuid, 0L); if (dbLastModified > cacheLastModified) { EnderChestSnapshot snapshot = this.dataManager.loadEnderChest(uuid); if (snapshot != null) { Bukkit.getScheduler().runTask(this.getPlugin(), () -> { enderChest.setContents(snapshot.getContents()); enderChest.setRows(snapshot.getRows()); lastModifiedCache.put(uuid, dbLastModified); this.getPlugin().getLogger().info("Synced enderchest for " + player.getName() + " from database"); }); } } }); } private EnderChest loadData(String name, UUID uuid) { EnderChestSnapshot enderChestSnapshot = this.dataManager.loadEnderChest(uuid); if (enderChestSnapshot == null) { return null; } if (crossServerSync) { lastModifiedCache.put(uuid, getLastModifiedFromDB(uuid)); } return new EnderChest(uuid, name, enderChestSnapshot.getContents(), enderChestSnapshot.getRows()); } @EventHandler(priority = EventPriority.LOWEST) private void onJoin(PlayerJoinEvent event) { String name = event.getPlayer().getName(); UUID uuid = event.getPlayer().getUniqueId(); this.nameUUIDMap.put(new CaselessString(name), uuid); if (this.uuidEnderChestMap.containsKey(uuid)) { if (crossServerSync) { checkAndReloadEnderChest(event.getPlayer()); } return; } Bukkit.getScheduler().runTaskAsynchronously(this.getPlugin(), () -> { dataManager.saveNameAndUUID(name, uuid); EnderChest enderChest = loadData(name, uuid); if (enderChest == null) { this.getPlugin().getLogger().info("Creating new enderchest for " + name); } else { this.getPlugin().getLogger().info("Loaded enderchest for " + name); } Bukkit.getScheduler().runTask(getPlugin(), () -> uuidEnderChestMap.put(uuid, enderChest == null ? createNew(event.getPlayer()) : enderChest)); }); } @EventHandler(ignoreCancelled = true) private void onInventoryOpen(PlayerInteractEvent event) { if (event.getAction() != Action.RIGHT_CLICK_BLOCK) return; Player player = event.getPlayer(); if(this.getPlugin().getDisabledWorlds().contains(player.getWorld().getName())) return; if (event.getClickedBlock().getType() == Material.ENDER_CHEST) { event.setCancelled(true); if (crossServerSync) { checkAndReloadEnderChest(player); } int rows = this.getNumRows(player); if (rows == 0) { this.getPlugin().getMessenger().sendMessage(player, MessageKey.NO_ENDERCHEST_SELF); } else { EnderChest enderChest = this.getEnderChest(player); if (enderChest == null) { this.getPlugin().getLogger().severe("Enderchest for online player " + player.getName() + " could not be found."); return; } if (MinecraftVersion.isAtLeastVersion(MinecraftVersion.MC1_19_R1)) { org.bukkit.block.EnderChest chestBlock = ((org.bukkit.block.EnderChest) event.getClickedBlock().getState()); chestBlock.open(); chestBlock.update(); } event.getPlayer().playSound(event.getClickedBlock().getLocation(), OPEN_SOUND, 1, 1F); this.openEnderChest(enderChest, player, rows); this.openFromBlocks.put(player.getUniqueId(), event.getClickedBlock()); } } } @EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true) private void onInventoryClose(InventoryCloseEvent event) { if (this.openFromBlocks.containsKey(event.getPlayer().getUniqueId())) { event.getViewers().remove(event.getPlayer()); Block block = this.openFromBlocks.remove(event.getPlayer().getUniqueId()); if (MinecraftVersion.isAtLeastVersion(MinecraftVersion.MC1_19_R1) && block.getState() instanceof org.bukkit.block.EnderChest && event.getViewers().isEmpty()) { org.bukkit.block.EnderChest chestBlock = ((org.bukkit.block.EnderChest) block.getState()); chestBlock.close(); chestBlock.update(); } ((Player) event.getPlayer()).playSound(block.getLocation(), CLOSE_SOUND, 1, 1F); } if (crossServerSync && event.getInventory().getHolder() instanceof EnderChest) { EnderChest ec = (EnderChest) event.getInventory().getHolder(); UUID uuid = ec.getUUID(); Bukkit.getScheduler().runTaskAsynchronously(this.getPlugin(), () -> { Map toSave = new HashMap<>(); toSave.put(uuid, ec.snapshot()); this.dataManager.saveEnderChestMultiple(toSave); lastModifiedCache.put(uuid, System.currentTimeMillis()); }); } } @EventHandler private void onPlayerQuit(PlayerQuitEvent event) { this.openFromBlocks.remove(event.getPlayer().getUniqueId()); EnderChest enderChest = getEnderChest(event.getPlayer()); enderChest.getInventory().getViewers().remove(event.getPlayer()); event.getPlayer().closeInventory(); } public EnderChest getEnderChest(Player player) { EnderChest enderChest = this.uuidEnderChestMap.get(player.getUniqueId()); if (enderChest == null) { this.getPlugin().getLogger().warning(player.getName() + "'s enderchest could not be found! Loading it synchronously"); EnderChestSnapshot snapshot = this.dataManager.loadEnderChest(player.getUniqueId()); if (snapshot == null) { enderChest = createNew(player); } else { enderChest = new EnderChest(snapshot.getUuid(), snapshot.getName(), snapshot.getContents(), snapshot.getRows()); if (crossServerSync) { lastModifiedCache.put(player.getUniqueId(), getLastModifiedFromDB(player.getUniqueId())); } } this.uuidEnderChestMap.put(player.getUniqueId(), enderChest); this.nameUUIDMap.put(new CaselessString(player.getName()), player.getUniqueId()); } return enderChest; } public void openEnderChest(EnderChest chest, Player player, int rows) { chest.setRows(rows); this.openEnderChest(chest, player); } public void openEnderChest(EnderChest chest, Player player) { if (Bukkit.getPlayer(chest.getUUID()) != null) { chest.openInventory(player); } else { chest.openInventory(player); } } public void openEnderChest(Player player, int rows) { EnderChest chest = this.getEnderChest(player); chest.setRows(rows); this.openEnderChest(chest, player); } public void clearEnderChest(EnderChest enderChest) { enderChest.clearContents(); } public int getNumRows(Player player) { for (int i = 6; i > 0; i--) { if (player.hasPermission("enderchest.size." + i)) { return i; } } return defaultRows; } public void updateRows(Player player) { getEnderChest(player).setRows(getNumRows(player)); } public Callback getEnderChest(String name) { CaselessString mcName = new CaselessString(name); if (this.nameUUIDMap.containsKey(mcName) && this.uuidEnderChestMap.get(this.nameUUIDMap.get(mcName)) != null) return Callback.withResult(this.uuidEnderChestMap.get(this.nameUUIDMap.get(mcName))); if (this.nameCallbackMap.containsKey(mcName)) return nameCallbackMap.get(mcName); Callback callback = new Callback<>(); this.nameCallbackMap.put(mcName, callback); Bukkit.getScheduler().runTaskAsynchronously(this.getPlugin(), () -> { EnderChestSnapshot snapshot = this.dataManager.loadEnderChest(name); EnderChest enderChest = snapshot == null ? null : new EnderChest(snapshot.getUuid(), snapshot.getName(), snapshot.getContents(), snapshot.getRows()); if (enderChest != null && crossServerSync) { lastModifiedCache.put(snapshot.getUuid(), getLastModifiedFromDB(snapshot.getUuid())); } Bukkit.getScheduler().runTask(this.getPlugin(), () -> { this.nameCallbackMap.remove(mcName); if (enderChest != null) { this.uuidEnderChestMap.put(snapshot.getUuid(), enderChest); this.nameUUIDMap.put(mcName, snapshot.getUuid()); } callback.setResult(enderChest); }); }); return callback; } public Callback getEnderChest(UUID uuid) { if (this.uuidEnderChestMap.get(uuid) != null) return Callback.withResult(this.uuidEnderChestMap.get(uuid)); if (this.uuidCallbackMap.containsKey(uuid)) return uuidCallbackMap.get(uuid); Callback callback = new Callback<>(); this.uuidCallbackMap.put(uuid, callback); Bukkit.getScheduler().runTaskAsynchronously(this.getPlugin(), () -> { EnderChestSnapshot snapshot = this.dataManager.loadEnderChest(uuid); EnderChest enderChest = snapshot == null ? null : new EnderChest(snapshot.getUuid(), snapshot.getName(), snapshot.getContents(), snapshot.getRows()); if (enderChest != null && crossServerSync) { lastModifiedCache.put(uuid, getLastModifiedFromDB(uuid)); } Bukkit.getScheduler().runTask(this.getPlugin(), () -> { this.uuidCallbackMap.remove(uuid); this.uuidEnderChestMap.put(uuid, enderChest); callback.setResult(enderChest); }); }); return callback; } public void deleteEnderChest(UUID uuid) { EnderChest enderChest = this.uuidEnderChestMap.get(uuid); Player player = Bukkit.getPlayer(uuid); if (enderChest != null) { enderChest.clearContents(); if (player != null) { if (this.convert) enderChest.setContents(player.getEnderChest().getContents()); return; } this.uuidEnderChestMap.remove(enderChest.getUUID()); } lastModifiedCache.remove(uuid); this.dataManager.deleteEnderChest(uuid); } public void deleteEnderChest(String name) { CaselessString caselessString = new CaselessString(name); EnderChest enderChest = this.uuidEnderChestMap.get(this.nameUUIDMap.get(caselessString)); Player player = Bukkit.getPlayerExact(name); if (enderChest != null) { enderChest.clearContents(); if (player != null) { if (this.convert) enderChest.setContents(player.getEnderChest().getContents()); return; } this.uuidEnderChestMap.remove(enderChest.getUUID()); lastModifiedCache.remove(enderChest.getUUID()); } this.dataManager.deleteEnderChest(name); } public void finishUp() { for (EnderChest value : this.uuidEnderChestMap.values()) { List list = new ArrayList<>(value.getInventory().getViewers()); for (HumanEntity humanEntity : list) { humanEntity.closeInventory(); } } Map enderChestSnapshotMap = new HashMap<>(); for (EnderChest value : this.uuidEnderChestMap.values()) { enderChestSnapshotMap.put(value.getUUID(), value.snapshot()); this.getPlugin().getLogger().info("Saving data for " + value.getName()); } this.dataManager.saveEnderChestMultiple(enderChestSnapshotMap); for (EnderChestSnapshot enderChestSnapshot : this.snapshotsToSave) { this.dataManager.saveEnderChest(enderChestSnapshot.getUuid(), enderChestSnapshot); } this.snapshotsToSave.clear(); this.uuidCallbackMap.clear(); this.nameCallbackMap.clear(); this.uuidEnderChestMap.clear(); this.nameUUIDMap.clear(); this.lastModifiedCache.clear(); enderChestSnapshotMap.clear(); nameUUIDMap.clear(); } private EnderChest createNew(Player player) { return new EnderChest(player.getUniqueId(), player.getName(), this.convert ? player.getEnderChest().getContents() : new ItemStack[]{}); } public Set getBlacklist() { return blacklist; } }