tracking and tracing

src/main/java/carpet/utils/MobAI.java

@@ -0,0 +1,93 @@
+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;
+        }
+    }
+
+}