#include "NsConcertChoiceBuildManager.h"
#include "EntityInfo.h"
#include "EntityManager.h"

const std::vector<Crate> NsConcertChoiceBuildManager::crates{
		{ "laser", 11203, 5.0, "Concert_Laser_QB_" },
		{ "rocket", 11204, 3.0, "Concert_Rocket_QB_" },
		{ "speaker", 11205, 5.0, "Concert_Speaker_QB_" },
		{ "spotlight", 11206, 5.0, "Concert_Spotlight_QB_" }
};

void NsConcertChoiceBuildManager::OnStartup(Entity* self) {
	NsConcertChoiceBuildManager::SpawnCrate(self);
}

void NsConcertChoiceBuildManager::SpawnCrate(Entity* self) {
	const auto spawnNumber = self->GetVar<uint32_t>(u"spawnNumber") % crates.size();
	const auto crate = crates[spawnNumber];

	const auto groups = self->GetGroups();
	if (groups.empty())
		return;

	// Groups are of the form CB_1, CB_2, etc.
	auto group = groups.at(0);
	const auto splitGroup = GeneralUtils::SplitString(group, '_');
	if (splitGroup.size() < 2)
		return;
	const auto groupNumber = std::stoi(splitGroup.at(1));

	EntityInfo info{};
	info.lot = crate.lot;
	info.pos = self->GetPosition();
	info.rot = self->GetRotation();
	info.spawnerID = self->GetObjectID();
	info.settings = {
		new LDFData<bool>(u"startsQBActivator", true),
		new LDFData<std::string>(u"grpNameQBShowBricks", crate.group + std::to_string(groupNumber)),
		new LDFData<std::u16string>(u"groupID", GeneralUtils::ASCIIToUTF16("Crate_" + group)),
		new LDFData<float>(u"crateTime", crate.time),
	};

	auto* spawnedCrate = Game::entityManager->CreateEntity(info);
	Game::entityManager->ConstructEntity(spawnedCrate);

	spawnedCrate->AddDieCallback([self]() {
		self->CancelAllTimers(); // Don't switch if the crate was smashed
		self->SetVar<LWOOBJID>(u"currentCrate", LWOOBJID_EMPTY);
		});

	self->SetVar<uint32_t>(u"spawnNumber", spawnNumber + 1);
	self->SetVar<float>(u"currentTimer", crate.time);
	self->SetVar<LWOOBJID>(u"currentCrate", spawnedCrate->GetObjectID());

	// Timer that rotates the crates
	self->AddCallbackTimer(crate.time, [self]() {
		auto crateID = self->GetVar<LWOOBJID>(u"currentCrate");
		if (crateID != LWOOBJID_EMPTY) {
			Game::entityManager->DestroyEntity(crateID);
			self->SetVar<LWOOBJID>(u"currentCrate", LWOOBJID_EMPTY);
		}

		SpawnCrate(self);
		});
}