#include "ForceMovementBehavior.h"
#include "BehaviorBranchContext.h"
#include "BehaviorContext.h"
#include "ControllablePhysicsComponent.h"
#include "EntityManager.h"

void ForceMovementBehavior::Handle(BehaviorContext* context, RakNet::BitStream* bitStream, const BehaviorBranchContext branch) {
	if (this->m_hitAction->m_templateId == BehaviorTemplates::BEHAVIOR_EMPTY && this->m_hitEnemyAction->m_templateId == BehaviorTemplates::BEHAVIOR_EMPTY && this->m_hitFactionAction->m_templateId == BehaviorTemplates::BEHAVIOR_EMPTY) {
		return;
	}

	uint32_t handle;
	bitStream->Read(handle);
	context->RegisterSyncBehavior(handle, this, branch);
}

void ForceMovementBehavior::Sync(BehaviorContext* context, RakNet::BitStream* bitStream, BehaviorBranchContext branch) {
	uint32_t next;
	bitStream->Read(next);

	LWOOBJID target;
	bitStream->Read(target);

	branch.target = target;
	auto* behavior = CreateBehavior(next);
	behavior->Handle(context, bitStream, branch);
}

void ForceMovementBehavior::Calculate(BehaviorContext* context, RakNet::BitStream* bitStream, BehaviorBranchContext branch) {
	if (this->m_hitAction->m_templateId == BehaviorTemplates::BEHAVIOR_EMPTY && this->m_hitEnemyAction->m_templateId == BehaviorTemplates::BEHAVIOR_EMPTY && this->m_hitFactionAction->m_templateId == BehaviorTemplates::BEHAVIOR_EMPTY) {
		return;
	}

	auto* casterEntity = EntityManager::Instance()->GetEntity(context->caster);
	if (casterEntity != nullptr) {
		auto* controllablePhysicsComponent = casterEntity->GetComponent<ControllablePhysicsComponent>();
		if (controllablePhysicsComponent != nullptr) {

			if (m_Forward == 1) {
				controllablePhysicsComponent->SetVelocity(controllablePhysicsComponent->GetRotation().GetForwardVector() * 25);
			}

			EntityManager::Instance()->SerializeEntity(casterEntity);
		}
	}

	const auto skillHandle = context->GetUniqueSkillId();
	bitStream->Write(skillHandle);

	context->SyncCalculation(skillHandle, this->m_Duration, this, branch);
}

void ForceMovementBehavior::Load() {
	this->m_hitAction = GetAction("hit_action");
	this->m_hitEnemyAction = GetAction("hit_action_enemy");
	this->m_hitFactionAction = GetAction("hit_action_faction");
	this->m_Duration = GetFloat("duration");
	this->m_Forward = GetFloat("forward");
	this->m_Left = GetFloat("left");
	this->m_Yaw = GetFloat("yaw");
}

void ForceMovementBehavior::SyncCalculation(BehaviorContext* context, RakNet::BitStream* bitStream, BehaviorBranchContext branch) {
	auto* casterEntity = EntityManager::Instance()->GetEntity(context->caster);
	if (casterEntity != nullptr) {
		auto* controllablePhysicsComponent = casterEntity->GetComponent<ControllablePhysicsComponent>();
		if (controllablePhysicsComponent != nullptr) {

			controllablePhysicsComponent->SetPosition(controllablePhysicsComponent->GetPosition() + controllablePhysicsComponent->GetVelocity() * m_Duration);
			controllablePhysicsComponent->SetVelocity({});

			EntityManager::Instance()->SerializeEntity(casterEntity);
		}
	}

	this->m_hitAction->Calculate(context, bitStream, branch);
	this->m_hitEnemyAction->Calculate(context, bitStream, branch);
	this->m_hitEnemyAction->Calculate(context, bitStream, branch);
}