package de.hysky.skyblocker.skyblock.dungeon;

import de.hysky.skyblocker.utils.Utils;
import de.hysky.skyblocker.utils.render.culling.OcclusionCulling;
import net.minecraft.entity.Entity;
import net.minecraft.entity.decoration.ArmorStandEntity;
import net.minecraft.entity.passive.BatEntity;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.predicate.entity.EntityPredicates;
import net.minecraft.util.math.Box;
import net.minecraft.world.World;

import java.util.List;

public class StarredMobGlow {

	public static boolean shouldMobGlow(Entity entity) {
		Box box = entity.getBoundingBox();

		if (Utils.isInDungeons() && !entity.isInvisible() && OcclusionCulling.isVisible(box.minX, box.minY, box.minZ, box.maxX, box.maxY, box.maxZ)) {
			// Minibosses
			if (entity instanceof PlayerEntity) {
				switch (entity.getName().getString()) {
					case "Lost Adventurer", "Shadow Assassin", "Diamond Guy" -> {
						return true;
					}
					case "Arcade Livid", "Crossed Livid", "Doctor Livid", "Frog Livid", "Hockey Livid",
					"Purple Livid", "Scream Livid", "Smile Livid", "Vendetta Livid" -> {
						List<ArmorStandEntity> armorStands = getArmorStands(entity.getWorld(), box);

						if (!armorStands.isEmpty() && LividColor.shouldGlow(armorStands.get(0))) {
							return true;
						}
					}
				}
			}

			// Regular Mobs
			if (!(entity instanceof ArmorStandEntity)) {
				List<ArmorStandEntity> armorStands = getArmorStands(entity.getWorld(), box);

				if (!armorStands.isEmpty() && armorStands.get(0).getName().getString().contains("✯")) return true;
			}

			// Bats
			return entity instanceof BatEntity;
		}

		return false;
	}

	private static List<ArmorStandEntity> getArmorStands(World world, Box box) {
        return world.getEntitiesByClass(ArmorStandEntity.class, box.expand(0, 2, 0), EntityPredicates.NOT_MOUNTED);
	}

	public static int getGlowColor(Entity entity) {
		String name = entity.getName().getString();

		if (entity instanceof PlayerEntity) {
			return switch (name) {
				case "Lost Adventurer" -> 0xfee15c;
				case "Shadow Assassin" -> 0x5b2cb2;
				case "Diamond Guy" -> 0x57c2f7;
				case "Arcade Livid", "Crossed Livid", "Doctor Livid", "Frog Livid", "Hockey Livid",
				"Purple Livid", "Scream Livid", "Smile Livid", "Vendetta Livid" -> LividColor.getGlowColor(name);
				default -> 0xf57738;
			};
		}

		return 0xf57738;
	}
}