added AI component handlers (adjusting attributes based on added AI

behaviors)

Author: DevCybran

src/java/de/ntcomputer/minecraft/controllablemobs/implementation/ai/AIComponentHandlers.java

@@ -0,0 +1,58 @@
+package de.ntcomputer.minecraft.controllablemobs.implementation.ai;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import net.minecraft.server.v1_6_R3.PathfinderGoal;
+import net.minecraft.server.v1_6_R3.PathfinderGoalBreakDoor;
+import net.minecraft.server.v1_6_R3.PathfinderGoalDoorInteract;
+import net.minecraft.server.v1_6_R3.PathfinderGoalOpenDoor;
+import de.ntcomputer.minecraft.controllablemobs.api.ai.AIType;
+import de.ntcomputer.minecraft.controllablemobs.implementation.CraftControllableMob;
+import de.ntcomputer.minecraft.controllablemobs.implementation.nativeinterfaces.NativeInterfaces;
+
+public final class AIComponentHandlers {
+	private static final Map<Class<? extends PathfinderGoal>, AIComponentListener<?>> handlerMap = new HashMap<Class<? extends PathfinderGoal>, AIComponentListener<?>>();
+
+	private AIComponentHandlers() {
+		throw new AssertionError();
+	}
+	
+	private static <T extends PathfinderGoal> void add(Class<T> goalClass, AIComponentListener<? super T> listener) {
+		handlerMap.put(goalClass, listener);
+	}
+	
+	static {
+		AIComponentListener<PathfinderGoalDoorInteract> closedDoorListener = new AIComponentListener<PathfinderGoalDoorInteract>() {
+			@Override
+			public void onAdd(CraftControllableMob<?> mob, PathfinderGoalDoorInteract goal) {
+				NativeInterfaces.NAVIGATION.FIELD_USECLOSEDDOOR.set(mob.nmsEntity.getNavigation(), true);
+			}
+			@Override
+			public void onRemoved(CraftControllableMob<?> mob, PathfinderGoalDoorInteract goal) {
+				if(!mob.getAI().hasBehavior(AIType.ACTION_DOOROPEN, AIType.ACTION_DOORBREAK)) {
+					NativeInterfaces.NAVIGATION.FIELD_USECLOSEDDOOR.set(mob.nmsEntity.getNavigation(), false);
+				}
+			}
+		};
+		add(PathfinderGoalOpenDoor.class, closedDoorListener);
+		add(PathfinderGoalBreakDoor.class, closedDoorListener);
+	}
+	
+	@SuppressWarnings("unchecked")
+	static <T extends PathfinderGoal> void handleAdd(CraftControllableMob<?> mob, T goal) {
+		AIComponentListener<T> listener = (AIComponentListener<T>) handlerMap.get(goal.getClass());
+		if(listener!=null) {
+			listener.onAdd(mob, goal);
+		}
+	}
+	
+	@SuppressWarnings("unchecked")
+	static <T extends PathfinderGoal> void handleRemoved(CraftControllableMob<?> mob, T goal) {
+		AIComponentListener<T> listener = (AIComponentListener<T>) handlerMap.get(goal.getClass());
+		if(listener!=null) {
+			listener.onRemoved(mob, goal);
+		}
+	}
+
+}

src/java/de/ntcomputer/minecraft/controllablemobs/implementation/ai/AIComponentListener.java

@@ -0,0 +1,11 @@
+package de.ntcomputer.minecraft.controllablemobs.implementation.ai;
+
+import net.minecraft.server.v1_6_R3.PathfinderGoal;
+import de.ntcomputer.minecraft.controllablemobs.implementation.CraftControllableMob;
+
+public interface AIComponentListener<T extends PathfinderGoal> {
+
+	public void onAdd(CraftControllableMob<?> mob, T goal);
+	public void onRemoved(CraftControllableMob<?> mob, T goal);
+	
+}

src/java/de/ntcomputer/minecraft/controllablemobs/implementation/ai/AIController.java

@@ -154,6 +154,7 @@ <B extends AIBehavior<? super E>> CraftAIPart<E,B> add(B behavior) {
 	}
 
 	private void attach(CraftAIPart<E,?> part) {
+		AIComponentHandlers.handleAdd(mob, part.goal);
 		this.addGoal(part.priority, part.goal);
 		this.attachedParts.add(part);
 		this.goalPartMap.put(part.goal, part);
@@ -171,6 +172,7 @@ void unattach(CraftAIPart<E,?> part) {
 		this.attachedParts.remove(part);
 		this.goalPartMap.remove(part.goal);
 		part.setState(AIState.UNATTACHED);
+		AIComponentHandlers.handleRemoved(mob, part.goal);
 	}
 
 	void remove(Set<AIType> typeSet, boolean remove) {