mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-11-23 05:57:20 +00:00
52 lines
1.5 KiB
C++
52 lines
1.5 KiB
C++
#include "NjRailPostServer.h"
|
|
#include "QuickBuildComponent.h"
|
|
#include "EntityManager.h"
|
|
|
|
void NjRailPostServer::OnStartup(Entity* self) {
|
|
auto* quickBuildComponent = self->GetComponent<QuickBuildComponent>();
|
|
if (quickBuildComponent != nullptr) {
|
|
self->SetNetworkVar<bool>(NetworkNotActiveVariable, true);
|
|
}
|
|
}
|
|
|
|
void NjRailPostServer::OnNotifyObject(Entity* self, Entity* sender, const std::string& name, int32_t param1,
|
|
int32_t param2) {
|
|
if (name == "PostRebuilt") {
|
|
self->SetNetworkVar<bool>(NetworkNotActiveVariable, false);
|
|
} else if (name == "PostDied") {
|
|
self->SetNetworkVar<bool>(NetworkNotActiveVariable, true);
|
|
}
|
|
}
|
|
|
|
void NjRailPostServer::OnQuickBuildNotifyState(Entity* self, eQuickBuildState state) {
|
|
if (state == eQuickBuildState::COMPLETED) {
|
|
auto* relatedRail = GetRelatedRail(self);
|
|
if (relatedRail == nullptr)
|
|
return;
|
|
|
|
relatedRail->NotifyObject(self, "PostRebuilt");
|
|
|
|
if (self->GetVar<bool>(NotActiveVariable))
|
|
return;
|
|
|
|
self->SetNetworkVar(NetworkNotActiveVariable, false);
|
|
} else if (state == eQuickBuildState::RESETTING) {
|
|
auto* relatedRail = GetRelatedRail(self);
|
|
if (relatedRail == nullptr)
|
|
return;
|
|
|
|
relatedRail->NotifyObject(self, "PostDied");
|
|
}
|
|
}
|
|
|
|
Entity* NjRailPostServer::GetRelatedRail(Entity* self) {
|
|
const auto& railGroup = self->GetVar<std::u16string>(RailGroupVariable);
|
|
if (!railGroup.empty()) {
|
|
for (auto* entity : Game::entityManager->GetEntitiesInGroup(GeneralUtils::UTF16ToWTF8(railGroup))) {
|
|
return entity;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|