DarkflameServer/dScripts/ai/GF/PetDigBuild.cpp

50 lines
1.6 KiB
C++
Raw Normal View History

#include "PetDigBuild.h"
#include "EntityManager.h"
#include "MissionComponent.h"
void PetDigBuild::OnRebuildComplete(Entity* self, Entity* target) {
2022-07-28 13:39:57 +00:00
auto flagNumber = self->GetVar<std::u16string>(u"flagNum");
2022-07-28 13:39:57 +00:00
EntityInfo info{};
auto pos = self->GetPosition();
pos.SetY(pos.GetY() + 0.5f);
info.pos = pos;
info.rot = self->GetRotation();
info.spawnerID = self->GetSpawnerID();
info.settings = {
new LDFData<LWOOBJID>(u"builder", target->GetObjectID()),
new LDFData<LWOOBJID>(u"X", self->GetObjectID())
};
2022-07-28 13:39:57 +00:00
if (!flagNumber.empty()) {
info.lot = 7410; // Normal GF treasure
info.settings.push_back(new LDFData<std::u16string>(u"groupID", u"Flag" + flagNumber));
} else {
auto* missionComponent = target->GetComponent<MissionComponent>();
if (missionComponent != nullptr && missionComponent->GetMissionState(746) == MissionState::MISSION_STATE_ACTIVE) {
info.lot = 9307; // Special Captain Jack treasure that drops a mission item
} else {
info.lot = 3495; // Normal AG treasure
}
}
2022-07-28 13:39:57 +00:00
auto* treasure = EntityManager::Instance()->CreateEntity(info);
EntityManager::Instance()->ConstructEntity(treasure);
self->SetVar<LWOOBJID>(u"chestObj", treasure->GetObjectID());
}
void PetDigBuild::OnDie(Entity* self, Entity* killer) {
2022-07-28 13:39:57 +00:00
auto treasureID = self->GetVar<LWOOBJID>(u"chestObj");
if (treasureID == LWOOBJID_EMPTY)
return;
2022-07-28 13:39:57 +00:00
auto treasure = EntityManager::Instance()->GetEntity(treasureID);
if (treasure == nullptr)
return;
2022-07-28 13:39:57 +00:00
// If the quick build expired and the treasure was not collected, hide the treasure
if (!treasure->GetIsDead()) {
treasure->Smash(self->GetObjectID(), SILENT);
}
}