package com.kodari.itemcap.overflow; import com.kodari.itemcap.ItemCapPlugin; import com.kodari.itemcap.config.CapConfig; import org.bukkit.configuration.ConfigurationSection; import org.bukkit.configuration.file.FileConfiguration; import org.bukkit.configuration.file.YamlConfiguration; import org.bukkit.entity.Player; import org.bukkit.inventory.ItemStack; import java.io.File; import java.io.IOException; import java.util.HashMap; import java.util.Map; import java.util.UUID; public class OverflowManager { private final ItemCapPlugin plugin; private final CapConfig capConfig; private final Map> overflowData = new HashMap<>(); private final Map withdrawSettings = new HashMap<>(); private File dataFile; public OverflowManager(ItemCapPlugin plugin, CapConfig capConfig) { this.plugin = plugin; this.capConfig = capConfig; this.dataFile = new File(plugin.getDataFolder(), "overflow_data.yml"); loadData(); } public int getOverflow(UUID uuid, TrackedItem item) { Map data = overflowData.get(uuid); if (data == null) return 0; return data.getOrDefault(item, 0); } public void addOverflow(UUID uuid, TrackedItem item, int amount) { overflowData.computeIfAbsent(uuid, k -> new HashMap<>()) .merge(item, amount, Integer::sum); saveData(); } public void removeOverflow(UUID uuid, TrackedItem item, int amount) { Map data = overflowData.get(uuid); if (data == null) return; int current = data.getOrDefault(item, 0); int newAmount = Math.max(0, current - amount); if (newAmount == 0) { data.remove(item); } else { data.put(item, newAmount); } saveData(); } public WithdrawAmount getWithdrawAmount(UUID uuid) { return withdrawSettings.getOrDefault(uuid, WithdrawAmount.ONE); } public void setWithdrawAmount(UUID uuid, WithdrawAmount amount) { withdrawSettings.put(uuid, amount); saveData(); } /** * Scans the player's inventory for tracked items exceeding their caps. * Moves excess to overflow and notifies the player only when something is moved. */ public void checkAndOverflow(Player player) { for (TrackedItem trackedItem : TrackedItem.values()) { int cap = capConfig.getCap(trackedItem); org.bukkit.Material mat = trackedItem.getMaterial(); int total = 0; for (ItemStack item : player.getInventory().getContents()) { if (item != null && item.getType() == mat) { total += item.getAmount(); } } if (total > cap) { int excess = total - cap; removeFromInventory(player, mat, excess); addOverflow(player.getUniqueId(), trackedItem, excess); player.sendMessage("§cExtra items went to overflow! The limit on this item is §f" + cap + "§c."); } } } private void removeFromInventory(Player player, org.bukkit.Material mat, int amount) { int remaining = amount; ItemStack[] contents = player.getInventory().getContents(); for (int i = 0; i < contents.length && remaining > 0; i++) { ItemStack item = contents[i]; if (item == null || item.getType() != mat) continue; if (item.getAmount() <= remaining) { remaining -= item.getAmount(); player.getInventory().setItem(i, null); } else { item.setAmount(item.getAmount() - remaining); remaining = 0; } } } private void loadData() { if (!dataFile.exists()) return; FileConfiguration data = YamlConfiguration.loadConfiguration(dataFile); ConfigurationSection overflowSection = data.getConfigurationSection("overflow"); if (overflowSection != null) { for (String uuidStr : overflowSection.getKeys(false)) { try { UUID uuid = UUID.fromString(uuidStr); ConfigurationSection playerSection = overflowSection.getConfigurationSection(uuidStr); if (playerSection == null) continue; Map playerData = new HashMap<>(); for (TrackedItem item : TrackedItem.values()) { int amount = playerSection.getInt(item.name(), 0); if (amount > 0) playerData.put(item, amount); } overflowData.put(uuid, playerData); } catch (IllegalArgumentException ignored) {} } } ConfigurationSection withdrawSection = data.getConfigurationSection("withdraw-amounts"); if (withdrawSection != null) { for (String uuidStr : withdrawSection.getKeys(false)) { try { UUID uuid = UUID.fromString(uuidStr); String val = withdrawSection.getString(uuidStr, "ONE"); withdrawSettings.put(uuid, WithdrawAmount.valueOf(val)); } catch (IllegalArgumentException ignored) {} } } } public void saveData() { FileConfiguration data = new YamlConfiguration(); for (Map.Entry> entry : overflowData.entrySet()) { String uuidPath = "overflow." + entry.getKey().toString(); for (Map.Entry itemEntry : entry.getValue().entrySet()) { data.set(uuidPath + "." + itemEntry.getKey().name(), itemEntry.getValue()); } } for (Map.Entry entry : withdrawSettings.entrySet()) { data.set("withdraw-amounts." + entry.getKey().toString(), entry.getValue().name()); } try { data.save(dataFile); } catch (IOException e) { plugin.getLogger().severe("Failed to save overflow data: " + e.getMessage()); } } }