local root = getroottable()

PopExt.robotTags         <- {}
PopExt.tankNames         <- {}
PopExt.tankNamesWildcard <- {}

popExtThinkFuncSet <- false
AddThinkToEnt(popExtEntity, null)

PrecacheModel("models/weapons/w_models/w_rocket.mdl")

function PopExt::AddRobotTag(tag, table) {
	if (!popExtThinkFuncSet) {
		AddThinkToEnt(popExtEntity, "PopulatorThink")
		popExtThinkFuncSet = true
	}
	PopExt.robotTags[tag] <- table
}

function PopExt::AddTankName(name, table) {
	if (!popExtThinkFuncSet) {
		AddThinkToEnt(popExtEntity, "PopulatorThink")
		popExtThinkFuncSet = true
	}

	name = name.tolower()
	local wildcard = name[name.len() - 1] == '*'
	if (wildcard) {
		name = name.slice(0, name.len() - 1)
		PopExt.tankNamesWildcard[name] <- table
	}
	else
		PopExt.tankNames[name] <- table
}

function PopExt::_PopIncrementTankIcon(icon) {
	local flags = MVM_CLASS_FLAG_NORMAL
	if (icon.isCrit) {
		flags = flags | MVM_CLASS_FLAG_ALWAYSCRIT
	}
	if (icon.isBoss) {
		flags = flags | MVM_CLASS_FLAG_MINIBOSS
	}
	if (icon.isSupport) {
		flags = flags | MVM_CLASS_FLAG_SUPPORT
	}
	if (icon.isSupportLimited) {
		flags = flags | MVM_CLASS_FLAG_SUPPORT_LIMITED
	}

	PopExt.DecrementWaveIconSpawnCount("tank", MVM_CLASS_FLAG_NORMAL | MVM_CLASS_FLAG_MINIBOSS | (icon.isSupport ? MVM_CLASS_FLAG_SUPPORT : 0) | (icon.isSupportLimited ? MVM_CLASS_FLAG_SUPPORT_LIMITED : 0), icon.count, false)
	PopExt.IncrementWaveIconSpawnCount(icon.name, flags, icon.count, false)
}

function PopExt::_PopIncrementIcon(icon) {
	local flags = MVM_CLASS_FLAG_NORMAL
	if (icon.isCrit) {
		flags = flags | MVM_CLASS_FLAG_ALWAYSCRIT
	}
	if (icon.isBoss) {
		flags = flags | MVM_CLASS_FLAG_MINIBOSS
	}
	if (icon.isSupport) {
		flags = flags | MVM_CLASS_FLAG_SUPPORT
	}
	if (icon.isSupportLimited) {
		flags = flags | MVM_CLASS_FLAG_SUPPORT_LIMITED
	}

	PopExt.IncrementWaveIconSpawnCount(icon.name, flags, icon.count, true)
}

function PopExt::AddCustomTankIcon(name, count, isCrit = false, isBoss = true, isSupport = false, isSupportLimited = false) {

	local icon = {
		name      = name
		count     = count
		isCrit    = isCrit
		isBoss    = isBoss
		isSupport = isSupport
		isSupportLimited = isSupportLimited
	}
	PopExtHooks.tankIcons.append(icon)
	PopExt._PopIncrementTankIcon(icon)
}

function PopExt::AddCustomIcon(name, count, isCrit = false, isBoss = false, isSupport = false, isSupportLimited = false) {

	local icon = {
		name      = name
		count     = count
		isCrit    = isCrit
		isBoss    = isBoss
		isSupport = isSupport
		isSupportLimited = isSupportLimited
	}
	PopExtHooks.icons.append(icon)
	PopExt._PopIncrementIcon(icon)
}

function PopExt::SetWaveIconsFunction(func) {
	PopExt.waveIconsFunction <- func
	func()
}

local resource = FindByClassname(null, "tf_objective_resource")

// Get wavebar spawn count of an icon with specified name and flags
function PopExt::GetWaveIconSpawnCount(name, flags) {

	local sizeArray = GetPropArraySize(resource, "m_nMannVsMachineWaveClassCounts")

	for (local a = 0; a < 2; a++) {

		local suffix = a == 0 ? "" : "2"

		for (local i = 0; i < sizeArray * 2; i++) {

			if (GetPropStringArray(resource, format("m_iszMannVsMachineWaveClassNames%s", suffix), i) == name && (flags == 0 || GetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), i) == flags)) {

				return GetPropIntArray(resource, format("m_nMannVsMachineWaveClassCounts%s", suffix), i)
			}
		}
	}
	return 0
}

// Set wavebar spawn count of an icon with specified name and flags
// If count is set to 0, removes the icon from the wavebar
// Can be used to put custom icons on a wavebar
function PopExt::SetWaveIconSpawnCount(name, flags, count, changeMaxEnemyCount = true) {

	local sizeArray = GetPropArraySize(resource, "m_nMannVsMachineWaveClassCounts")

	for (local a = 0; a < 2; a++) {

		local suffix = a == 0 ? "" : "2"

		for (local i = 0; i < sizeArray; i++) {

			local nameSlot = GetPropStringArray(resource, "m_iszMannVsMachineWaveClassNames" + suffix, i)

			if (nameSlot == "" && count > 0) {

				SetPropStringArray(resource, format("m_iszMannVsMachineWaveClassNames%s", suffix), name, i)
				SetPropIntArray(resource, format("m_nMannVsMachineWaveClassCounts%s", suffix), count, i)
				SetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), flags, i)

				if (changeMaxEnemyCount && (flags & (MVM_CLASS_FLAG_NORMAL | MVM_CLASS_FLAG_MINIBOSS))) {

					SetPropInt(resource, "m_nMannVsMachineWaveEnemyCount", GetPropInt(resource, "m_nMannVsMachineWaveEnemyCount") + count)
				}
				return
			}

			if (nameSlot == name && (flags == 0 || GetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), i) == flags)) {

				local preCount = GetPropIntArray(resource, format("m_nMannVsMachineWaveClassCounts%s", suffix), i)
				SetPropIntArray(resource, format("m_nMannVsMachineWaveClassCounts%s", suffix), count, i)

				if (changeMaxEnemyCount && (flags & (MVM_CLASS_FLAG_NORMAL | MVM_CLASS_FLAG_MINIBOSS))) {

					SetPropInt(resource, "m_nMannVsMachineWaveEnemyCount", GetPropInt(resource, "m_nMannVsMachineWaveEnemyCount") + count - preCount)
				}
				if (count <= 0) {

					SetPropStringArray(resource, format("m_iszMannVsMachineWaveClassNames%s", suffix), "", i)
					SetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), 0, i)
					SetPropBoolArray(resource, format("m_bMannVsMachineWaveClassActive%s", suffix), false, i)
				}
				return
			}
		}
	}
}
// Increment wavebar spawn count of an icon with specified name and flags
// Can be used to put custom icons on a wavebar
function PopExt::IncrementWaveIconSpawnCount(name, flags, count = 1, changeMaxEnemyCount = true) {
	PopExt.SetWaveIconSpawnCount(name, flags, PopExt.GetWaveIconSpawnCount(name, flags) + count, changeMaxEnemyCount)
	return 0
}

// Increment wavebar spawn count of an icon with specified name and flags
// Use it to decrement the spawn count when the enemy is killed. Should not be used for support type icons
function PopExt::DecrementWaveIconSpawnCount(name, flags, count = 1, changeMaxEnemyCount = false) {

	local sizeArray = GetPropArraySize(resource, "m_nMannVsMachineWaveClassCounts")

	for (local a = 0; a < 2; a++) {

		local suffix = a == 0 ? "" : "2"

		for (local i = 0; i < sizeArray; i++) {

			local nameSlot = GetPropStringArray(resource, format("m_iszMannVsMachineWaveClassNames%s", suffix), i)

			if (nameSlot == name && (flags == 0 || GetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), i) == flags)) {

				local preCount = GetPropIntArray(resource, format("m_nMannVsMachineWaveClassCounts%s", suffix), i)

				SetPropIntArray(resource, format("m_nMannVsMachineWaveClassCounts%s", suffix), preCount - count > 0 ? preCount - count : 0, i)

				if (changeMaxEnemyCount && (flags & (MVM_CLASS_FLAG_NORMAL | MVM_CLASS_FLAG_MINIBOSS))) {

					SetPropInt(resource, "m_nMannVsMachineWaveEnemyCount", GetPropInt(resource, "m_nMannVsMachineWaveEnemyCount") - (count > preCount ? preCount : count))
				}

				if (preCount - count <= 0) {

					SetPropStringArray(resource, format("m_iszMannVsMachineWaveClassNames%s", suffix), "", i)
					SetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), 0, i)
					SetPropBoolArray(resource, format("m_bMannVsMachineWaveClassActive%s", suffix), false, i)
				}
				return
			}
		}
	}
	return 0
}

// Used for mission and support limited bots to display them on a wavebar during the wave, set by the game automatically when an enemy with this icon spawn
function PopExt::SetWaveIconActive(name, flags, active) {

	local sizeArray = GetPropArraySize(resource, "m_nMannVsMachineWaveClassCounts")

	for (local a = 0; a < 2; a++) {

		local suffix = a == 0 ? "" : "2"

		for (local i = 0; i < sizeArray; i++) {

			local nameSlot = GetPropStringArray(resource, format("m_iszMannVsMachineWaveClassNames%s", suffix), i)

			if (nameSlot == name && (flags == 0 || GetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), i) == flags)) {

				SetPropBoolArray(resource, format("m_bMannVsMachineWaveClassActive%s", suffix), active, i)
				return
			}
		}
	}
}

// Used for mission and support limited bots to display them on a wavebar during the wave, set by the game automatically when an enemy with this icon spawn
function PopExt::GetWaveIconActive(name, flags) {

	local sizeArray = GetPropArraySize(resource, "m_nMannVsMachineWaveClassCounts")

	for (local a = 0; a < 2; a++) {

		local suffix = a == 0 ? "" : "2"

		for (local i = 0; i < sizeArray; i++) {

			local nameSlot = GetPropStringArray(resource, format("m_iszMannVsMachineWaveClassNames%s", suffix), i)

			if (nameSlot == name && (flags == 0 || GetPropIntArray(resource, format("m_nMannVsMachineWaveClassFlags%s", suffix), i) == flags)) {

				return GetPropBoolArray(resource, format("m_bMannVsMachineWaveClassActive%s", suffix), i)
			}
		}
	}
	return false
}