package com.example.powerarmor.effects; import com.example.powerarmor.PowerArmorPlugin; import com.example.powerarmor.armor.ArmorManager; import com.example.powerarmor.armor.ArmorType; import org.bukkit.entity.Player; import org.bukkit.inventory.ItemStack; import org.bukkit.potion.PotionEffect; import org.bukkit.potion.PotionEffectType; import org.bukkit.scheduler.BukkitRunnable; import java.util.HashMap; import java.util.HashSet; import java.util.Map; import java.util.Set; public class EffectManager { private final PowerArmorPlugin plugin; private final ArmorManager armorManager; private final Map> activeArmor = new HashMap<>(); public EffectManager(PowerArmorPlugin plugin, ArmorManager armorManager) { this.plugin = plugin; this.armorManager = armorManager; startEffectTask(); } private void startEffectTask() { new BukkitRunnable() { @Override public void run() { for (Player player : plugin.getServer().getOnlinePlayers()) { updatePlayerEffects(player); } } }.runTaskTimer(plugin, 0L, 20L); } private void updatePlayerEffects(Player player) { Set currentArmor = getEquippedArmorTypes(player); Set previousArmor = activeArmor.getOrDefault(player, new HashSet<>()); if (!currentArmor.equals(previousArmor)) { removeOldEffects(player, previousArmor, currentArmor); } applyArmorEffects(player, currentArmor); if (currentArmor.isEmpty()) { activeArmor.remove(player); } else { activeArmor.put(player, currentArmor); } } private Set getEquippedArmorTypes(Player player) { Set types = new HashSet<>(); ItemStack[] armor = player.getInventory().getArmorContents(); for (ItemStack piece : armor) { ArmorType type = armorManager.getArmorType(piece); if (type != null) { types.add(type); } } return types; } private void applyArmorEffects(Player player, Set armorTypes) { Map effectMap = new HashMap<>(); for (ArmorType type : armorTypes) { for (ArmorType.ArmorEffect effect : type.getEffects()) { PotionEffectType potionType = effect.getPotionEffectType(); if (potionType != null) { int currentLevel = effectMap.getOrDefault(potionType, -1); effectMap.put(potionType, Math.max(currentLevel, effect.getAmplifier())); } } } for (Map.Entry entry : effectMap.entrySet()) { PotionEffect potionEffect = new PotionEffect( entry.getKey(), 60, entry.getValue(), true, false, true ); player.addPotionEffect(potionEffect); } } private void removeOldEffects(Player player, Set oldArmor, Set newArmor) { Set oldEffects = new HashSet<>(); Set newEffects = new HashSet<>(); for (ArmorType type : oldArmor) { for (ArmorType.ArmorEffect effect : type.getEffects()) { PotionEffectType potionType = effect.getPotionEffectType(); if (potionType != null) { oldEffects.add(potionType); } } } for (ArmorType type : newArmor) { for (ArmorType.ArmorEffect effect : type.getEffects()) { PotionEffectType potionType = effect.getPotionEffectType(); if (potionType != null) { newEffects.add(potionType); } } } for (PotionEffectType effect : oldEffects) { if (!newEffects.contains(effect)) { player.removePotionEffect(effect); } } } public void cleanup() { for (Player player : activeArmor.keySet()) { Set armor = activeArmor.get(player); removeOldEffects(player, armor, new HashSet<>()); } activeArmor.clear(); } }