added onAdd and onRemoved behavior methods

Author: DevCybran

src/java/de/ntcomputer/minecraft/controllablemobs/api/ai/behaviors/AIBehavior.java

@@ -35,4 +35,10 @@ public final int getPriority(int lastBehaviorPriority) {
 
 	public abstract PathfinderGoal createPathfinderGoal(CraftControllableMob<? extends E> mob);
 
+	public void onAdd(CraftControllableMob<? extends E> mob) {
+	}
+	
+	public void onRemoved(CraftControllableMob<? extends E> mob) {
+	}
+	
 }

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) {
+		if(part.behavior!=null) part.behavior.onAdd(this.mob);
 		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);
+		if(part.behavior!=null) part.behavior.onRemoved(this.mob);
 	}
 
 	void remove(Set<AIType> typeSet, boolean remove) {

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

@@ -11,7 +11,7 @@
 import de.ntcomputer.minecraft.controllablemobs.api.ai.behaviors.AIBehavior;
 
 public class CraftAIPart<E extends LivingEntity, B extends AIBehavior<? super E>> implements AIPart<E,B> {
-	private final B behavior;
+	final B behavior;
 	final int priority;
 	private final AIType type;
 	private final AIController<E> controller;