#include "AmShieldGenerator.h"
#include "EntityManager.h"
#include "DestroyableComponent.h"
#include "GameMessages.h"
#include "MovementAIComponent.h"
#include "BaseCombatAIComponent.h"
#include "SkillComponent.h"

void AmShieldGenerator::OnStartup(Entity* self) {
	self->SetProximityRadius(20, "shield");
	self->SetProximityRadius(21, "buffer");

	StartShield(self);
}

void AmShieldGenerator::OnProximityUpdate(Entity* self, Entity* entering, std::string name, std::string status) {
	auto* destroyableComponent = entering->GetComponent<DestroyableComponent>();

	if (status == "ENTER" && name == "shield") {
		if (destroyableComponent->HasFaction(4)) {
			EnemyEnteredShield(self, entering);
		}
	}

	if (name != "buffer" || !entering->IsPlayer()) {
		return;
	}

	auto entitiesInProximity = self->GetVar<std::vector<LWOOBJID>>(u"Players");

	if (status == "ENTER") {
		const auto& iter = std::find(entitiesInProximity.begin(), entitiesInProximity.end(), entering->GetObjectID());

		if (iter == entitiesInProximity.end()) {
			entitiesInProximity.push_back(entering->GetObjectID());
		}
	} else if (status == "LEAVE") {
		const auto& iter = std::find(entitiesInProximity.begin(), entitiesInProximity.end(), entering->GetObjectID());

		if (iter != entitiesInProximity.end()) {
			entitiesInProximity.erase(iter);
		}
	}

	self->SetVar<std::vector<LWOOBJID>>(u"Players", entitiesInProximity);
}

void AmShieldGenerator::OnDie(Entity* self, Entity* killer) {
	self->CancelAllTimers();

	auto* child = EntityManager::Instance()->GetEntity(self->GetVar<LWOOBJID>(u"Child"));

	if (child != nullptr) {
		child->Kill();
	}
}

void AmShieldGenerator::OnTimerDone(Entity* self, std::string timerName) {
	if (timerName == "BuffPlayers") {
		BuffPlayers(self);

		self->AddTimer("BuffPlayers", 3.0f);
	} else if (timerName == "PlayFX") {
		GameMessages::SendPlayFXEffect(self->GetObjectID(), 5351, u"generatorOn", "generatorOn");

		self->AddTimer("PlayFX", 1.5f);
	} else if (timerName == "RefreshEnemies") {
		auto enemiesInProximity = self->GetVar<std::vector<LWOOBJID>>(u"Enemies");

		for (const auto enemyID : enemiesInProximity) {
			auto* enemy = EntityManager::Instance()->GetEntity(enemyID);

			if (enemy != nullptr) {
				EnemyEnteredShield(self, enemy);
			}
		}

		self->AddTimer("RefreshEnemies", 1.5f);
	}
}

void AmShieldGenerator::StartShield(Entity* self) {
	self->AddTimer("PlayFX", 1.5f);
	self->AddTimer("BuffPlayers", 3.0f);
	self->AddTimer("RefreshEnemies", 1.5f);

	const auto myPos = self->GetPosition();
	const auto myRot = self->GetRotation();

	EntityInfo info{};
	info.lot = 13111;
	info.pos = myPos;
	info.rot = myRot;
	info.spawnerID = self->GetObjectID();

	auto* child = EntityManager::Instance()->CreateEntity(info);

	self->SetVar(u"Child", child->GetObjectID());

	BuffPlayers(self);
}

void AmShieldGenerator::BuffPlayers(Entity* self) {
	auto* skillComponent = self->GetComponent<SkillComponent>();

	if (skillComponent == nullptr) {
		return;
	}

	auto entitiesInProximity = self->GetVar<std::vector<LWOOBJID>>(u"Players");

	for (const auto playerID : entitiesInProximity) {
		auto* player = EntityManager::Instance()->GetEntity(playerID);

		if (player == nullptr) {
			return;
		}

		skillComponent->CalculateBehavior(1200, 27024, playerID, true);
	}
}

void AmShieldGenerator::EnemyEnteredShield(Entity* self, Entity* intruder) {
	auto* baseCombatAIComponent = intruder->GetComponent<BaseCombatAIComponent>();
	auto* movementAIComponent = intruder->GetComponent<MovementAIComponent>();

	if (baseCombatAIComponent == nullptr || movementAIComponent == nullptr) {
		return;
	}

	auto dir = intruder->GetRotation().GetForwardVector() * -1;
	dir.y += 15;
	dir.x *= 50;
	dir.z *= 50;

	// TODO: Figure out how todo knockback, I'll stun them for now

	if (NiPoint3::DistanceSquared(self->GetPosition(), movementAIComponent->GetCurrentPosition()) < 20 * 20) {
		baseCombatAIComponent->Stun(2.0f);
		movementAIComponent->SetDestination(baseCombatAIComponent->GetStartPosition());
	}

	baseCombatAIComponent->ClearThreat();
}