package de.hysky.skyblocker.skyblock.crimson.dojo;

import de.hysky.skyblocker.utils.render.RenderHelper;
import it.unimi.dsi.fastutil.objects.Object2LongOpenHashMap;
import net.fabricmc.fabric.api.client.rendering.v1.WorldRenderContext;
import net.minecraft.entity.Entity;
import net.minecraft.entity.mob.ZombieEntity;
import net.minecraft.text.MutableText;
import net.minecraft.text.Text;
import net.minecraft.util.Formatting;
import net.minecraft.util.math.Vec3d;

import java.awt.*;
import java.text.DecimalFormat;
import java.util.Map;

public class ForceTestHelper {

    private static final DecimalFormat FORMATTER = new DecimalFormat("0.0");
    private static final int ZOMBIE_LIFE_TIME = 10100;

    private static final Object2LongOpenHashMap<ZombieEntity> zombies = new Object2LongOpenHashMap<>();

    protected static void reset() {
        zombies.clear();
    }

    /**
     * If a zombie value is negative make it glow
     *
     * @param name zombies value
     * @return if the zombie should glow
     */
    protected static boolean shouldGlow(String name) {
        return name.contains("-");
    }

    protected static int getColor() {
        return Color.RED.getRGB();
    }

    protected static void onEntitySpawn(Entity entity) {
        if (entity instanceof ZombieEntity zombie) {
            zombies.put(zombie, System.currentTimeMillis() + ZOMBIE_LIFE_TIME);
        }
    }

    protected static void onEntityAttacked(Entity entity) {
        if (entity instanceof ZombieEntity zombie) {
            if (zombies.containsKey(zombie)) {
                zombies.put(zombie, System.currentTimeMillis() + ZOMBIE_LIFE_TIME); //timer is reset when they are hit
            }
        }
    }

    protected static void onEntityDespawn(Entity entity) {
        if (entity instanceof ZombieEntity zombie) {
            zombies.removeLong(zombie);
        }
    }

    protected static void render(WorldRenderContext context) {
        //render times
        long currentTime = System.currentTimeMillis();
        for (Map.Entry<ZombieEntity, Long> zombie : zombies.object2LongEntrySet()) {
            float secondsTime = Math.max((zombie.getValue() - currentTime) / 1000f, 0);

            MutableText text = Text.literal(FORMATTER.format(secondsTime));
            if (secondsTime > 1) {
                text = text.formatted(Formatting.GREEN);
            } else if (secondsTime > 0) {
                text = text.formatted(Formatting.YELLOW);
            } else {
                text = text.formatted(Formatting.RED);
            }

            Vec3d labelPos = zombie.getKey().getCameraPosVec(context.tickCounter().getTickDelta(false));
            RenderHelper.renderText(context, text, labelPos, 1.5f, true);
        }
    }
}