From e397ed310ed0bcdab1a5adbf38921b8e8a5bece4 Mon Sep 17 00:00:00 2001
From: EmosewaMC <39972741+EmosewaMC@users.noreply.github.com>
Date: Tue, 8 Feb 2022 02:02:05 -0800
Subject: [PATCH] Updated script for teams

Updated script to drop buffing items for all members in a team.
---
 dScripts/AgSurvivalBuffStation.cpp | 31 ++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/dScripts/AgSurvivalBuffStation.cpp b/dScripts/AgSurvivalBuffStation.cpp
index c893d68..01fe397 100644
--- a/dScripts/AgSurvivalBuffStation.cpp
+++ b/dScripts/AgSurvivalBuffStation.cpp
@@ -4,6 +4,7 @@
 #include "GameMessages.h"
 #include "SkillComponent.h"
 #include "dLogger.h"
+#include "TeamManager.h"
 
 void AgSurvivalBuffStation::OnRebuildComplete(Entity* self, Entity* target) {
     auto destroyableComponent = self->GetComponent<DestroyableComponent>();
@@ -20,12 +21,24 @@ void AgSurvivalBuffStation::OnRebuildComplete(Entity* self, Entity* target) {
     self->AddTimer("DropArmor", dropArmorTimer);
     self->AddTimer("DropLife", dropLifeTimer);
     self->AddTimer("Dropimagination", dropImaginationTimer);
-    self->SetVar<LWOOBJID>(u"PlayerId", target->GetObjectID());
+    // Since all survival players should be on the same team, we get the team.
+    auto team = TeamManager::Instance()->GetTeam(target->GetObjectID());
+
+    std::vector<LWOOBJID> builderTeam;
+    // Not on a team
+    if (team == nullptr) {
+        builderTeam.push_back(target->GetObjectID());
+        self->SetVar<std::vector<LWOOBJID>>(u"BuilderTeam", builderTeam);
+        return;
+    } 
+
+    for (auto memberID : team->members) {
+        builderTeam.push_back(memberID);
+    }
+    self->SetVar<std::vector<LWOOBJID>>(u"BuilderTeam", builderTeam);
 }
 
 void AgSurvivalBuffStation::OnTimerDone(Entity* self, std::string timerName) {
-    auto targetID = self->GetVar<LWOOBJID>(u"PlayerId");
-    auto target = EntityManager::Instance()->GetEntity(targetID);
     uint32_t powerupToDrop = lifePowerup;
     if (timerName == "DropArmor") {
         powerupToDrop = armorPowerup;
@@ -39,5 +52,15 @@ void AgSurvivalBuffStation::OnTimerDone(Entity* self, std::string timerName) {
         powerupToDrop = imaginationPowerup;
         self->AddTimer("Dropimagination", dropImaginationTimer);
     }
-    if (target != nullptr) GameMessages::SendDropClientLoot(target, self->GetObjectID(), powerupToDrop, 0, self->GetPosition());
+    auto team = self->GetVar<std::vector<LWOOBJID>>(u"BuilderTeam");
+    for (auto memberID : team) {
+        auto member = EntityManager::Instance()->GetEntity(memberID);
+        if (member != nullptr && !member->GetIsDead()) {
+            GameMessages::SendDropClientLoot(member, self->GetObjectID(), powerupToDrop, 0, self->GetPosition());
+        } else {
+            // If player left the team or left early erase them from the team variable.
+            team.erase(std::find(team.begin(), team.end(), memberID));
+            self->SetVar<std::vector<LWOOBJID>>(u"BuilderTeam", team);
+        }
+    }
 }