package com.deadlysins.component; import net.minecraft.nbt.NbtCompound; import java.util.HashMap; import java.util.Map; public class SinAbilityComponentImpl implements SinAbilityComponent { private final Map cooldowns = new HashMap<>(); private final Map activeAbilities = new HashMap<>(); private final Map activeDurations = new HashMap<>(); @Override public int getCooldown(String abilityId) { return cooldowns.getOrDefault(abilityId, 0); } @Override public void setCooldown(String abilityId, int ticks) { if (ticks <= 0) { cooldowns.remove(abilityId); } else { cooldowns.put(abilityId, ticks); } } @Override public boolean isOnCooldown(String abilityId) { return getCooldown(abilityId) > 0; } @Override public boolean isAbilityActive(String abilityId) { return activeAbilities.getOrDefault(abilityId, false); } @Override public void setAbilityActive(String abilityId, boolean active) { if (active) { activeAbilities.put(abilityId, true); } else { activeAbilities.remove(abilityId); } } @Override public int getActiveDuration(String abilityId) { return activeDurations.getOrDefault(abilityId, 0); } @Override public void setActiveDuration(String abilityId, int ticks) { if (ticks <= 0) { activeDurations.remove(abilityId); setAbilityActive(abilityId, false); } else { activeDurations.put(abilityId, ticks); } } @Override public void tick() { // Tick down cooldowns cooldowns.entrySet().removeIf(entry -> { entry.setValue(entry.getValue() - 1); return entry.getValue() <= 0; }); // Tick down active durations activeDurations.entrySet().removeIf(entry -> { entry.setValue(entry.getValue() - 1); if (entry.getValue() <= 0) { setAbilityActive(entry.getKey(), false); return true; } return false; }); } @Override public void readFromNbt(NbtCompound tag) { cooldowns.clear(); activeAbilities.clear(); activeDurations.clear(); NbtCompound cooldownsTag = tag.getCompoundOrEmpty("Cooldowns"); for (String key : cooldownsTag.getKeys()) { cooldowns.put(key, cooldownsTag.getInt(key, 0)); } NbtCompound activeTag = tag.getCompoundOrEmpty("ActiveAbilities"); for (String key : activeTag.getKeys()) { activeAbilities.put(key, activeTag.getBoolean(key)); } NbtCompound durationTag = tag.getCompoundOrEmpty("ActiveDurations"); for (String key : durationTag.getKeys()) { activeDurations.put(key, durationTag.getInt(key, 0)); } } @Override public void writeToNbt(NbtCompound tag) { NbtCompound cooldownsTag = new NbtCompound(); cooldowns.forEach(cooldownsTag::putInt); tag.put("Cooldowns", cooldownsTag); NbtCompound activeTag = new NbtCompound(); activeAbilities.forEach(activeTag::putBoolean); tag.put("ActiveAbilities", activeTag); NbtCompound durationTag = new NbtCompound(); activeDurations.forEach(durationTag::putInt); tag.put("ActiveDurations", durationTag); } }