package carpet.utils;

import carpet.CarpetServer;
import com.google.common.collect.Sets;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityType;
import net.minecraft.server.world.ServerWorld;
import net.minecraft.util.registry.Registry;

import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

public class MobAI
{
    private static Map<EntityType,Set<TrackingType>> aiTrackers = new HashMap<>();

    public static void resetTrackers()
    {
        aiTrackers.clear();
    }

    public static boolean isTracking(Entity e, TrackingType type)
    {
        Set<TrackingType> currentTrackers = aiTrackers.get(e.getType());
        if (currentTrackers == null)
            return false;
        return currentTrackers.contains(type);
    }

    public static void clearTracking(EntityType etype)
    {
        aiTrackers.remove(etype);
        for(ServerWorld world : CarpetServer.minecraft_server.getWorlds() )
        {
            for (Entity e: world.getEntities(etype, Entity::hasCustomName))
            {
                e.setCustomNameVisible(false);
                e.setCustomName(null);
            }
        }
    }

    public static void startTracking(EntityType e, TrackingType type)
    {
        aiTrackers.putIfAbsent(e,Sets.newHashSet());
        aiTrackers.get(e).add(type);
    }

    public static List<String> availbleTypes()
    {
        Set<EntityType> types = new HashSet<>();
        for (TrackingType type: TrackingType.values())
        {
            types.addAll(type.types);
        }
        return types.stream().map(t -> Registry.ENTITY_TYPE.getId(t).getPath()).collect(Collectors.toList());
    }

    public static List<String> availableFor(EntityType<?> entityType)
    {
        Set<TrackingType> availableOptions = new HashSet<>();
        for (TrackingType type: TrackingType.values())
            for (EntityType etype: type.types)
                if (etype == entityType)
                    availableOptions.add(type);
        return availableOptions.stream().map(t -> t.name).collect(Collectors.toList());
    }

    public enum TrackingType
    {
        IRON_GOLEM_SPAWNING("iron_golem_spawning", Sets.newHashSet(EntityType.VILLAGER));
        public Set<EntityType> types;
        public String name;
        TrackingType(String name, Set<EntityType> applicableTypes)
        {
            this.name = name;
            types = applicableTypes;
        }

        public static TrackingType byName(String aspect)
        {
            for (TrackingType type: values())
                if (type.name.equalsIgnoreCase(aspect))
                    return type;
            return null;
        }
    }

}