#include "stdafx.h"
#include "com.mojang.nbt.h"
#include "net.minecraft.world.h"
#include "net.minecraft.world.level.h"
#include "net.minecraft.world.phys.h"
#include "net.minecraft.world.entity.ai.attributes.h"
#include "net.minecraft.world.entity.player.h"
#include "net.minecraft.world.entity.monster.h"
#include "net.minecraft.world.item.h"
#include "net.minecraft.world.item.enchantment.h"
#include "net.minecraft.world.entity.item.h"
#include "net.minecraft.world.damagesource.h"
#include "PigZombie.h"
#include "..\Minecraft.Client\Textures.h"
#include "SoundTypes.h"

AttributeModifier *PigZombie::SPEED_MODIFIER_ATTACKING = (new AttributeModifier(eModifierId_MOB_PIG_ATTACKSPEED, 0.45, AttributeModifier::OPERATION_ADDITION))->setSerialize(false);

void PigZombie::_init()
{
	registerAttributes();

	angerTime = 0;
	playAngrySoundIn = 0;
	lastAttackTarget = nullptr;
}

PigZombie::PigZombie(Level *level) : Zombie( level )
{
	_init();

	fireImmune = true;
}

void PigZombie::registerAttributes()
{
	Zombie::registerAttributes();

	getAttribute(SPAWN_REINFORCEMENTS_CHANCE)->setBaseValue(0);
	getAttribute(SharedMonsterAttributes::MOVEMENT_SPEED)->setBaseValue(0.5f);
	getAttribute(SharedMonsterAttributes::ATTACK_DAMAGE)->setBaseValue(5);
}

bool PigZombie::useNewAi()
{
	return false;
}

void PigZombie::tick()
{
	if (lastAttackTarget != attackTarget && !level->isClientSide)
	{
		AttributeInstance *speed = getAttribute(SharedMonsterAttributes::MOVEMENT_SPEED);
		speed->removeModifier(SPEED_MODIFIER_ATTACKING);

		if (attackTarget != NULL)
		{
			speed->addModifier(new AttributeModifier(*SPEED_MODIFIER_ATTACKING));
		}
	}
	lastAttackTarget = attackTarget;

	if (playAngrySoundIn > 0)
	{
		if (--playAngrySoundIn == 0)
		{
			playSound(eSoundType_MOB_ZOMBIEPIG_ZPIGANGRY, getSoundVolume() * 2, ((random->nextFloat() - random->nextFloat()) * 0.2f + 1.0f) * 1.8f);
		}
	}
	Zombie::tick();
}

bool PigZombie::canSpawn() 
{
	return level->difficulty > Difficulty::PEACEFUL && level->isUnobstructed(bb) && level->getCubes(shared_from_this(), bb)->empty() && !level->containsAnyLiquid(bb);
}

void PigZombie::addAdditonalSaveData(CompoundTag *tag)
{
	Zombie::addAdditonalSaveData(tag);
	tag->putShort(L"Anger", (short) angerTime);
}

void PigZombie::readAdditionalSaveData(CompoundTag *tag)
{
	Zombie::readAdditionalSaveData(tag);
	angerTime = tag->getShort(L"Anger");
}

shared_ptr<Entity> PigZombie::findAttackTarget()
{
#ifndef _FINAL_BUILD
#ifdef _DEBUG_MENUS_ENABLED
	if(app.GetMobsDontAttackEnabled())
	{
		return shared_ptr<Player>();
	}
#endif
#endif

	if (angerTime == 0) return nullptr;
	return Zombie::findAttackTarget();
}

bool PigZombie::hurt(DamageSource *source, float dmg)
{
	shared_ptr<Entity> sourceEntity = source->getEntity();
	if ( sourceEntity != NULL && sourceEntity->instanceof(eTYPE_PLAYER) )
	{
		vector<shared_ptr<Entity> > *nearby = level->getEntities( shared_from_this(), bb->grow(32, 32, 32));		
		AUTO_VAR(itEnd, nearby->end());
		for (AUTO_VAR(it, nearby->begin()); it != itEnd; it++)
		{
			shared_ptr<Entity> e = *it; //nearby->at(i);
			if ( e->instanceof(eTYPE_PIGZOMBIE) )
			{
				shared_ptr<PigZombie> pigZombie = dynamic_pointer_cast<PigZombie>(e);
				pigZombie->alert(sourceEntity);
			}
		}
		alert(sourceEntity);
	}
	return Zombie::hurt(source, dmg);
}

void PigZombie::alert(shared_ptr<Entity> target)
{
	attackTarget = target;
	angerTime = 20 * 20 + random->nextInt(20 * 20);
	playAngrySoundIn = random->nextInt(20 * 2);
}

int PigZombie::getAmbientSound()
{
	return eSoundType_MOB_ZOMBIEPIG_AMBIENT;
}

int PigZombie::getHurtSound()
{
	return eSoundType_MOB_ZOMBIEPIG_HURT;
}

int PigZombie::getDeathSound()
{
	return eSoundType_MOB_ZOMBIEPIG_DEATH;
}

void PigZombie::dropDeathLoot(bool wasKilledByPlayer, int playerBonusLevel)
{
	int count = random->nextInt(2 + playerBonusLevel);
	for (int i = 0; i < count; i++)
	{
		spawnAtLocation(Item::rotten_flesh_Id, 1);
	}
	count = random->nextInt(2 + playerBonusLevel);
	for (int i = 0; i < count; i++)
	{
		spawnAtLocation(Item::goldNugget_Id, 1);
	}
}

bool PigZombie::mobInteract(shared_ptr<Player> player)
{
	return false;
}

void PigZombie::dropRareDeathLoot(int rareLootLevel)
{
	spawnAtLocation(Item::goldIngot_Id, 1);
}

int PigZombie::getDeathLoot()
{
	return Item::rotten_flesh_Id;
}

void PigZombie::populateDefaultEquipmentSlots()
{
	setEquippedSlot(SLOT_WEAPON, shared_ptr<ItemInstance>( new ItemInstance(Item::sword_gold)) );
}

MobGroupData *PigZombie::finalizeMobSpawn(MobGroupData *groupData, int extraData /*= 0*/) // 4J Added extraData param
{
	Zombie::finalizeMobSpawn(groupData);
	setVillager(false);
	return groupData;
}