1 files changed, 98 insertions, 0 deletions
diff --git a/src/main/java/com/thatgravyboat/skyblockhud/handlers/CooldownHandler.java b/src/main/java/com/thatgravyboat/skyblockhud/handlers/CooldownHandler.java
new file mode 100644
index 0000000..d7d3a41
--- /dev/null
+++ b/src/main/java/com/thatgravyboat/skyblockhud/handlers/CooldownHandler.java
@@ -0,0 +1,98 @@
+package com.thatgravyboat.skyblockhud.handlers;
+
+import com.thatgravyboat.skyblockhud.SkyblockHud;
+import com.thatgravyboat.skyblockhud.api.item.IAbility;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.nbt.NBTTagList;
+import net.minecraftforge.event.entity.player.PlayerInteractEvent;
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
+import net.minecraftforge.fml.common.gameevent.TickEvent;
+
+public class CooldownHandler {
+
+    private static final Pattern ABILITY_REGEX = Pattern.compile("\u00A76Ability: (.*) \u00A7e\u00A7lRIGHT CLICK .* \u00A78Cooldown: \u00A7a(\\d*)s");
+
+    private static final Map<String, Cooldown> COOLDOWNS = new HashMap<>();
+
+    public static Matcher getAbility(NBTTagCompound nbt) {
+        if (nbt != null && nbt.hasKey("ExtraAttributes") && nbt.getCompoundTag("ExtraAttributes").hasKey("uuid") && nbt.hasKey("display")){
+            NBTTagCompound display = nbt.getCompoundTag("display");
+            if (display != null && display.hasKey("Lore")){
+                NBTTagList lore = display.getTagList("Lore", 8);
+                List<String> loreList = new ArrayList<>();
+                for (int i = 0; i < lore.tagCount(); i++) {
+                    String loreLine = lore.getStringTagAt(i).trim();
+                    if (!loreLine.isEmpty()) loreList.add(loreLine);
+                }
+                Matcher abilityMatcher = ABILITY_REGEX.matcher(String.join(" ", loreList));
+                if (abilityMatcher.find()){
+                    return abilityMatcher;
+                }
+            }
+        }
+        return null;
+    }
+
+    private static void addCooldown(IAbility ability) {
+        COOLDOWNS.putIfAbsent(ability.getAbility(), new Cooldown(ability.getAbilityTime()*20));
+    }
+
+    @SubscribeEvent
+    public void tick(TickEvent.ClientTickEvent event){
+        if (SkyblockHud.config.misc.hideItemCooldowns) return;
+        if (event.phase.equals(TickEvent.Phase.END)) {
+            COOLDOWNS.values().forEach(Cooldown::tick);
+            COOLDOWNS.entrySet().removeIf(entry -> entry.getValue().isOver());
+        }
+    }
+
+    @SubscribeEvent
+    public void onPlayerInteract(PlayerInteractEvent event) {
+        if (SkyblockHud.config.misc.hideItemCooldowns) return;
+        if (event.action.equals(PlayerInteractEvent.Action.RIGHT_CLICK_AIR) || event.action.equals(PlayerInteractEvent.Action.RIGHT_CLICK_BLOCK)){
+            if (event.entityPlayer.getHeldItem() != null) {
+                IAbility ability = (IAbility)((Object) event.entityPlayer.getHeldItem());
+                if (ability.getAbility() != null) {
+                    addCooldown(ability);
+                }
+            }
+        }
+    }
+
+    public static float getAbilityTime(ItemStack stack){
+        IAbility ability = (IAbility)((Object) stack);
+        if (ability.getAbility() != null) {
+            return COOLDOWNS.containsKey(ability.getAbility()) ? COOLDOWNS.get(ability.getAbility()).getTime() : -1f;
+        }
+        return -1f;
+    }
+
+    private static class Cooldown {
+        public int current;
+        public final int end;
+
+        Cooldown(int end){
+            this.end = end;
+        }
+
+        public boolean isOver() {
+            return current >= end;
+        }
+
+        public void tick() {
+            current++;
+        }
+
+        public float getTime() {
+            return current/(float)end;
+        }
+    }
+
+}