TraitEnderference.java

package slimeknights.tconstruct.tools.traits;

import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.monster.EntityEnderman;
import net.minecraft.item.ItemStack;
import net.minecraft.potion.PotionEffect;
import net.minecraft.util.text.TextFormatting;
import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.event.entity.living.EnderTeleportEvent;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;

import slimeknights.tconstruct.library.Util;
import slimeknights.tconstruct.library.potion.TinkerPotion;
import slimeknights.tconstruct.library.traits.AbstractTrait;

public class TraitEnderference extends AbstractTrait {

  public static TinkerPotion Enderference = new TinkerPotion(Util.getResource("enderference"), true, false, 0x21985f);

  public TraitEnderference() {
    super("enderference", TextFormatting.DARK_AQUA);

    MinecraftForge.EVENT_BUS.register(this);
  }

  @Override
  public void onHit(ItemStack tool, EntityLivingBase player, EntityLivingBase target, float damage, boolean isCritical) {
    if(target instanceof EntityEnderman) {
      PotionEffect effect = new PotionEffect(Enderference, 100, 1, false, true);
      target.addPotionEffect(effect);
    }
  }

  @Override
  public void afterHit(ItemStack tool, EntityLivingBase player, EntityLivingBase target, float damageDealt, boolean wasCritical, boolean wasHit) {
    if(!wasHit) {
      target.removePotionEffect(Enderference);
    }
  }

  @SubscribeEvent
  public void onEnderTeleport(EnderTeleportEvent event) {
    if(Enderference.getLevel(event.getEntityLiving()) > 0) {
      event.setCanceled(true);
    }
  }
}