#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;
}