From f04bc8378a12e29cb9ca8e2dfe0fc567866236ca Mon Sep 17 00:00:00 2001
From: Louis Hobbs <bazzadobs@live.co.uk>
Date: Wed, 23 Aug 2023 00:19:30 +0100
Subject: [PATCH] Add functionality to TaskAcquireAttackToken

---
 Source/Nakatomi/TaskAcquireAttackToken.cpp | 15 ++++++++++++++-
 Source/Nakatomi/TaskAcquireAttackToken.h   |  1 +
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/Source/Nakatomi/TaskAcquireAttackToken.cpp b/Source/Nakatomi/TaskAcquireAttackToken.cpp
index ff960f7..6452493 100644
--- a/Source/Nakatomi/TaskAcquireAttackToken.cpp
+++ b/Source/Nakatomi/TaskAcquireAttackToken.cpp
@@ -2,8 +2,21 @@
 
 
 #include "TaskAcquireAttackToken.h"
+#include "EnemyAIController.h"
 
 EBTNodeResult::Type UTaskAcquireAttackToken::ExecuteTask(UBehaviorTreeComponent& owner, uint8* memory)
 {
-	return EBTNodeResult::Succeeded;
+	auto enemyController = Cast<AEnemyAIController>(owner.GetAIOwner());
+
+	if (enemyController->TryObtainAttackToken())
+	{
+		return EBTNodeResult::Succeeded;
+	}
+	
+	return EBTNodeResult::Failed;
+}
+
+EBTNodeResult::Type UTaskAcquireAttackToken::AbortTask(UBehaviorTreeComponent& owner, uint8* memory)
+{
+	return EBTNodeResult::Aborted;
 }
diff --git a/Source/Nakatomi/TaskAcquireAttackToken.h b/Source/Nakatomi/TaskAcquireAttackToken.h
index d59208a..c6435f8 100644
--- a/Source/Nakatomi/TaskAcquireAttackToken.h
+++ b/Source/Nakatomi/TaskAcquireAttackToken.h
@@ -18,4 +18,5 @@ public:
 
 	virtual EBTNodeResult::Type ExecuteTask(UBehaviorTreeComponent& owner, uint8* memory) override;
 	
+	virtual EBTNodeResult::Type AbortTask(UBehaviorTreeComponent& owner, uint8* memory) override;
 };