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;
        }
    }
}