CustomNPC-Plus/src/main/java/noppes/npcs/roles/JobHealer.java

package noppes.npcs.roles;

import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import noppes.npcs.NoppesUtilServer;
import noppes.npcs.entity.EntityNPCInterface;

import java.util.ArrayList;
import java.util.List;

public class JobHealer extends JobInterface{
	private long healTicks = 0;
	public int range = 5;
	public int speed = 5;

	public JobHealer(EntityNPCInterface npc) {
		super(npc);
	}

	@Override
	public NBTTagCompound writeToNBT(NBTTagCompound nbttagcompound) {
		nbttagcompound.setInteger("HealerRange", range);
		nbttagcompound.setInteger("HealerSpeed", speed);
		return nbttagcompound;
	}

	@Override
	public void readFromNBT(NBTTagCompound nbttagcompound) {
		range = nbttagcompound.getInteger("HealerRange");
		speed = nbttagcompound.getInteger("HealerSpeed");
	}
	private List<EntityLivingBase> toHeal = new ArrayList<EntityLivingBase>();
	
	//TODO heal food, heal potion effects, heal more types of entities besides just the player and npcs
	public boolean aiShouldExecute() {
		healTicks++;
		if (healTicks < speed * 10) 
			return false;
		
		for(Object plObj:  npc.worldObj.getEntitiesWithinAABB(EntityLivingBase.class, npc.boundingBox.expand(range, range/2, range))){
			EntityLivingBase entity = (EntityLivingBase) plObj;
			
			if(entity instanceof EntityPlayer){
				EntityPlayer player = (EntityPlayer) entity;
				if(player.getHealth() < player.getMaxHealth() && !npc.faction.isAggressiveToPlayer(player))
					toHeal.add(player);
			}
			if(entity instanceof EntityNPCInterface){
				EntityNPCInterface npc = (EntityNPCInterface) entity;
				if(npc.getHealth() < npc.getMaxHealth() && !this.npc.faction.isAggressiveToNpc(npc))
					toHeal.add(npc);
			}
			
		}

		healTicks = 0;
		return !toHeal.isEmpty();
	}

	public void aiStartExecuting() {
		for(EntityLivingBase entity : toHeal){
			float heal = entity.getMaxHealth() / 20; //heal 5% of max health
			heal(entity, heal > 0?heal:1);
		}
		toHeal.clear();
	}
	
	public void heal(EntityLivingBase entity, float amount){
		entity.heal(amount);
		NoppesUtilServer.spawnParticle(entity, "heal",entity.dimension);
	}
}