#include "PullToPointBehavior.h"

#include "BehaviorContext.h"
#include "BehaviorBranchContext.h"
#include "EntityManager.h"
#include "MovementAIComponent.h"

void PullToPointBehavior::Handle(BehaviorContext* context, RakNet::BitStream* bitStream, BehaviorBranchContext branch)
{
	auto* entity = EntityManager::Instance()->GetEntity(context->originator);
	
	auto* target = EntityManager::Instance()->GetEntity(branch.target);

	if (entity == nullptr || target == nullptr)
	{
		return;
	}
	
	auto* movement = target->GetComponent<MovementAIComponent>();

	if (movement == nullptr)
	{
		return;
	}

	const auto position = branch.isProjectile ? branch.referencePosition : entity->GetPosition();

	movement->PullToPoint(position);
}

void PullToPointBehavior::Calculate(BehaviorContext* context, RakNet::BitStream* bitStream, BehaviorBranchContext branch)
{
	Handle(context, bitStream, branch);
}

void PullToPointBehavior::Load()
{
}