Wizardry/src/main/java/electroblob/wizardry/entity/living/EntitySpiritHorse.java

package electroblob.wizardry.entity.living;

import electroblob.wizardry.WizardData;
import electroblob.wizardry.Wizardry;
import electroblob.wizardry.item.ItemWand;
import electroblob.wizardry.registry.WizardrySounds;
import electroblob.wizardry.util.WizardryParticleType;
import electroblob.wizardry.util.WizardryUtilities;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.IEntityLivingData;
import net.minecraft.entity.SharedMonsterAttributes;
import net.minecraft.entity.passive.EntityAnimal;
import net.minecraft.entity.passive.EntityHorse;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.util.DamageSource;
import net.minecraft.util.EnumHand;
import net.minecraft.util.text.ITextComponent;
import net.minecraft.util.text.TextComponentTranslation;
import net.minecraft.util.text.translation.I18n;
import net.minecraft.world.DifficultyInstance;
import net.minecraft.world.World;

/**
 * Does not implement ISummonedCreature because it has different despawning rules and because EntityHorse already has an
 * owner system.
 */
@SuppressWarnings("deprecation") // It's what Entity does, so...
public class EntitySpiritHorse extends EntityHorse {

	private int idleTimer = 0;

	public EntitySpiritHorse(World par1World){
		super(par1World);
	}

	@Override
	public String getName(){
		if(this.hasCustomName()){
			return this.getCustomNameTag();
		}else{
			return I18n.translateToLocal("entity.wizardry.Spirit Horse.name");
		}
	}

	@Override
	public int getTotalArmorValue(){
		return 0;
	}

	@Override
	protected int getExperiencePoints(EntityPlayer p_70693_1_){
		return 0;
	}

	@Override
	protected Item getDropItem(){

		return null;
	}

	@Override
	protected void dropFewItems(boolean par1, int par2){
	}

	@Override
	protected void applyEntityAttributes(){
		super.applyEntityAttributes();
		this.getEntityAttribute(SharedMonsterAttributes.MAX_HEALTH).setBaseValue(24.0D);
	}

	@Override
	public void openGUI(EntityPlayer p_110199_1_){
	}

	@Override
	public boolean processInteract(EntityPlayer player, EnumHand hand){

		ItemStack itemstack = player.getHeldItem(hand);

		// Allows the owner (but not other players) to dispel the spirit horse using a wand (shift-clicking, because
		// clicking mounts the horse in this case).
		if(itemstack.getItem() instanceof ItemWand && this.getOwner() == player && player.isSneaking()){
			// Prevents accidental double clicking.
			if(this.ticksExisted > 20){
				for(int i = 0; i < 15; i++){
					Wizardry.proxy.spawnParticle(WizardryParticleType.SPARKLE, world,
							this.posX - this.width / 2 + this.rand.nextFloat() * width,
							this.posY + this.height * this.rand.nextFloat() + 0.2f,
							this.posZ - this.width / 2 + this.rand.nextFloat() * width, 0, 0, 0,
							48 + this.rand.nextInt(12), 0.8f, 0.8f, 1.0f);
				}
				this.setDead();
				if(WizardData.get(player) != null){
					WizardData.get(player).hasSpiritHorse = false;
				}
				this.playSound(WizardrySounds.SPELL_HEAL, 0.7F, rand.nextFloat() * 0.4F + 1.0F);
				// This is necessary to prevent the wand's spell being cast when performing this action.
				return true;
			}
			return false;
		}

		return super.processInteract(player, hand);
	}

	@Override
	public void onDeath(DamageSource par1DamageSource){

		super.onDeath(par1DamageSource);

		// Allows player to summon another spirit horse once this one has died.
		if(this.getOwner() instanceof EntityPlayer && WizardData.get((EntityPlayer)this.getOwner()) != null){
			WizardData.get((EntityPlayer)this.getOwner()).hasSpiritHorse = false;
		}
	}

	// I wrote this one!
	private EntityLivingBase getOwner(){

		// I think the DataManager stores any objects, so it now stores the UUID instead of its string representation.
		Entity owner = WizardryUtilities.getEntityByUUID(world, this.getOwnerUniqueId());

		if(owner instanceof EntityLivingBase){
			return (EntityLivingBase)owner;
		}else{
			return null;
		}
	}

	@Override
	public void onUpdate(){

		super.onUpdate();

		// Adds a dust particle effect
		if(this.world.isRemote){
			Wizardry.proxy.spawnParticle(WizardryParticleType.DUST, world,
					this.posX - this.width / 2 + this.rand.nextFloat() * width,
					this.posY + this.height * this.rand.nextFloat() + 0.2f,
					this.posZ - this.width / 2 + this.rand.nextFloat() * width, 0, 0, 0, 0, 0.8f, 0.8f, 1.0f);
		}

		// Spirit horse disappears a short time after being dismounted.
		if(!this.isBeingRidden()){
			this.idleTimer++;
		}else if(this.idleTimer > 0){
			this.idleTimer = 0;
		}

		if(this.idleTimer > 200){
			if(this.world.isRemote){
				for(int i = 0; i < 15; i++){
					Wizardry.proxy.spawnParticle(WizardryParticleType.SPARKLE, world,
							this.posX - this.width / 2 + this.rand.nextFloat() * width,
							this.posY + this.height * this.rand.nextFloat() + 0.2f,
							this.posZ - this.width / 2 + this.rand.nextFloat() * width, 0, 0, 0,
							48 + this.rand.nextInt(12), 0.8f, 0.8f, 1.0f);
				}
			}
			this.playSound(WizardrySounds.SPELL_HEAL, 0.7F, rand.nextFloat() * 0.4F + 1.0F);
			// Allows player to summon another spirit horse once this one has disappeared.
			if(this.getOwner() instanceof EntityPlayer && WizardData.get((EntityPlayer)this.getOwner()) != null){
				WizardData.get((EntityPlayer)this.getOwner()).hasSpiritHorse = false;
			}
			this.setDead();
		}
	}

	@Override
	public boolean canMateWith(EntityAnimal par1EntityAnimal){
		return false;
	}

	@Override
	public IEntityLivingData onInitialSpawn(DifficultyInstance difficulty, IEntityLivingData data){

		// Adds Particles on spawn. Due to client/server differences this cannot be done in the item.
		if(this.world.isRemote){
			for(int i = 0; i < 15; i++){
				Wizardry.proxy.spawnParticle(WizardryParticleType.SPARKLE, world,
						this.posX - this.width / 2 + this.rand.nextFloat() * width,
						this.posY + this.height * this.rand.nextFloat() + 0.2f,
						this.posZ - this.width / 2 + this.rand.nextFloat() * width, 0, 0, 0, 48 + this.rand.nextInt(12),
						0.8f, 0.8f, 1.0f);
			}
		}

		return super.onInitialSpawn(difficulty, data);
	}

	@Override
	public ITextComponent getDisplayName(){
		if(getOwner() != null){
			return new TextComponentTranslation(ISummonedCreature.NAMEPLATE_TRANSLATION_KEY, getOwner().getName(),
					new TextComponentTranslation("entity." + this.getEntityString() + ".name"));
		}else{
			return super.getDisplayName();
		}
	}

	@Override
	public boolean hasCustomName(){
		// If this returns true, the renderer will show the nameplate when looking directly at the entity
		return Wizardry.settings.showSummonedCreatureNames && getOwner() != null;
	}

}