src/main/java/com/legacy/aether/entities/ai/SheepuffAIEatAetherGrass.java

package com.legacy.aether.entities.ai;

import net.minecraft.block.Block;
import net.minecraft.entity.ai.EntityAIBase;
import net.minecraft.world.World;

import com.legacy.aether.blocks.BlocksAether;
import com.legacy.aether.entities.passive.EntitySheepuff;

public class SheepuffAIEatAetherGrass extends EntityAIBase
{

    private EntitySheepuff sheepuff;

    private World entityWorld;

    int eatingGrassTimer;

    public SheepuffAIEatAetherGrass(EntitySheepuff sheepuff)
    {
        this.sheepuff = sheepuff;
        this.entityWorld = sheepuff.worldObj;
        this.setMutexBits(7);
    }

    public boolean shouldExecute()
    {
        if (this.sheepuff.getRNG().nextInt(1000) != 0)
        {
            return false;
        }
        else
        {
            return this.entityWorld.getBlock((int) this.sheepuff.posX, (int) this.sheepuff.posY - 1, (int) this.sheepuff.posZ) == BlocksAether.aether_grass;
        }
    }

    public void startExecuting()
    {
        this.eatingGrassTimer = 40;
        this.entityWorld.setEntityState(this.sheepuff, (byte)10);
        this.sheepuff.getNavigator().clearPathEntity();
    }

    public void resetTask()
    {
        this.eatingGrassTimer = 0;
    }

    public boolean continueExecuting()
    {
        return this.eatingGrassTimer > 0;
    }

    public int getEatingGrassTimer()
    {
        return this.eatingGrassTimer;
    }

    public void updateTask()
    {
        this.eatingGrassTimer = Math.max(0, this.eatingGrassTimer - 1);

        if (this.eatingGrassTimer == 4)
        {

            if (this.entityWorld.getBlock((int) this.sheepuff.posX, (int) this.sheepuff.posY - 1, (int) this.sheepuff.posZ) == BlocksAether.aether_grass)
            {
                if (this.entityWorld.getGameRules().getGameRuleBooleanValue("mobGriefing"))
                {
                    this.entityWorld.playAuxSFX(2001, (int) this.sheepuff.posX, (int) this.sheepuff.posY - 1, (int) this.sheepuff.posZ, Block.getIdFromBlock(BlocksAether.aether_grass));
                    this.entityWorld.setBlock((int) this.sheepuff.posX, (int) this.sheepuff.posY - 1, (int) this.sheepuff.posZ, BlocksAether.aether_dirt);
                }

                this.sheepuff.eatGrassBonus();
            }
        }
    }

}