AnimalEvents.java

package joshie.harvest.animals;

import joshie.harvest.animals.item.ItemAnimalTool.Tool;
import joshie.harvest.animals.tracker.AnimalTrackerServer;
import joshie.harvest.api.HFApi;
import joshie.harvest.api.animals.AnimalAction;
import joshie.harvest.api.animals.AnimalStats;
import joshie.harvest.api.animals.AnimalTest;
import joshie.harvest.core.HFTrackers;
import joshie.harvest.core.entity.EntityBasket;
import joshie.harvest.core.helpers.EntityHelper;
import joshie.harvest.core.util.annotations.HFEvents;
import net.minecraft.entity.Entity;
import net.minecraft.entity.passive.EntityAnimal;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.init.Items;
import net.minecraft.init.MobEffects;
import net.minecraft.item.ItemStack;
import net.minecraft.potion.PotionEffect;
import net.minecraftforge.event.entity.EntityJoinWorldEvent;
import net.minecraftforge.event.entity.living.LivingAttackEvent;
import net.minecraftforge.event.entity.living.LivingDeathEvent;
import net.minecraftforge.event.entity.player.AttackEntityEvent;
import net.minecraftforge.event.entity.player.PlayerInteractEvent;
import net.minecraftforge.event.entity.player.PlayerInteractEvent.EntityInteract;
import net.minecraftforge.fml.common.eventhandler.EventPriority;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
import net.minecraftforge.fml.common.gameevent.PlayerEvent.PlayerLoggedOutEvent;

import static joshie.harvest.core.handlers.BasketHandler.forbidsDrop;
import static joshie.harvest.core.helpers.InventoryHelper.ITEM;
import static joshie.harvest.core.helpers.InventoryHelper.ITEM_STACK;

@HFEvents
@SuppressWarnings("unused")
public class AnimalEvents {
    @SubscribeEvent(priority = EventPriority.HIGHEST)
    @SuppressWarnings("ConstantConditions")
    public void onEntityLoaded(EntityJoinWorldEvent event) {
        Entity entity = event.getEntity();
        AnimalStats stats = EntityHelper.getStats(entity);
        if (stats != null && entity instanceof EntityAnimal) {
            if (!entity.worldObj.isRemote) {
                stats.setEntity((EntityAnimal)entity);
                HFTrackers.<AnimalTrackerServer>getAnimalTracker(event.getWorld()).add(stats);
            }
        }
    }

    @SubscribeEvent
    @SuppressWarnings("ConstantConditions")
    public void onEntityDeath(LivingDeathEvent event) {
        AnimalStats stats = EntityHelper.getStats(event.getEntityLiving());
        if (stats != null && !event.getEntity().worldObj.isRemote) {
            HFTrackers.<AnimalTrackerServer>getAnimalTracker(event.getEntityLiving().worldObj).onDeath(stats);
        }
    }

    @SubscribeEvent
    public void onEntityInteract(EntityInteract event) {
        AnimalStats stats = EntityHelper.getStats(event.getTarget());
        ItemStack stack = event.getItemStack();
        if (stats != null && stack != null) {
            if (HFApi.animals.canEat(stack, stats.getType().getFoodTypes()) && stats.performAction(event.getWorld(), stack, AnimalAction.FEED)) {
                stack.splitStack(1);
                event.setCanceled(true);
            }
        }
    }

    @SubscribeEvent
    public void onEntityAttackedByPlayer(AttackEntityEvent event) {
        AnimalStats stats = EntityHelper.getStats(event.getTarget());
        if (stats != null) {
            stats.affectHappiness(stats.getType().getRelationshipBonus(AnimalAction.HURT));
        }
    }

    @SubscribeEvent
    public void onEntityAttacked(LivingAttackEvent event) {
        AnimalStats stats = EntityHelper.getStats(event.getEntityLiving());
        if (stats != null) {
            event.getEntityLiving().addPotionEffect(new PotionEffect(MobEffects.REGENERATION, 200, 0, true, false));
        }
    }

    /* When right clicking poultry, will throw any poultry on your head **/
    @HFEvents
    public static class PickupPoultry {
        public static boolean register() { return HFAnimals.PICKUP_POULTRY; }
        private ItemStack[] stacks;

        private ItemStack[] getStacks() {
            if (stacks != null) return stacks;
            stacks = new ItemStack[] {
                    HFAnimals.TOOLS.getStackFromEnum(Tool.CHICKEN_FEED),
                    HFAnimals.TOOLS.getStackFromEnum(Tool.MEDICINE)
            };

            return stacks;
        }

        private boolean isHolding(ItemStack stack) {
            return ITEM_STACK.matchesAny(stack, getStacks()) || ITEM.matchesAny(stack, HFAnimals.TREATS, Items.NAME_TAG);
        }

        boolean blocksPickup(EntityPlayer player) {
            return isHolding(player.getHeldItemMainhand()) || isHolding(player.getHeldItemOffhand());
        }

        @SubscribeEvent
        public void onEntityInteract(PlayerInteractEvent.EntityInteract event) {
            EntityPlayer player = event.getEntityPlayer();
            if (!player.isBeingRidden() && !blocksPickup(player)) {
                Entity entity = event.getTarget();
                AnimalStats stats = EntityHelper.getStats(entity);
                if (stats != null && stats.performTest(AnimalTest.CAN_CARRY)) {
                    entity.setEntityInvulnerable(true);
                    entity.startRiding(player, true);
                }
            }
        }

        @SubscribeEvent
        public void onPlayerLoggedOut(PlayerLoggedOutEvent event) {
            event.player.getPassengers().stream().filter(entity -> entity instanceof EntityBasket).forEach(entity -> {
                AnimalStats stats = EntityHelper.getStats(entity);
                entity.dismountRidingEntity();
                entity.rotationPitch = event.player.rotationPitch;
                entity.rotationYaw = event.player.rotationYaw;
                entity.moveRelative(0F, 0.1F, 1.05F);
                entity.setEntityInvulnerable(false);
            });
        }

        @SubscribeEvent
        @SuppressWarnings("ConstantConditions")
        public void onRightClickGround(PlayerInteractEvent.RightClickBlock event) {
            EntityPlayer player = event.getEntityPlayer();
            if (!forbidsDrop(event.getWorld().getBlockState(event.getPos()).getBlock())) {
                for (Entity entity : player.getPassengers()) {
                    AnimalStats stats = EntityHelper.getStats(entity);
                    if (stats != null && stats.performTest(AnimalTest.CAN_CARRY)) {
                        entity.dismountRidingEntity();
                        entity.rotationPitch = player.rotationPitch;
                        entity.rotationYaw = player.rotationYaw;
                        entity.moveRelative(0F, 0.1F, 1.05F);
                        entity.setEntityInvulnerable(false);
                        stats.performAction(player.worldObj, null, AnimalAction.DISMOUNT);
                    }
                }
            }
        }
    }
}