ClassTargetSelectors.lua

---@class AddonNamespace
local addon = select(2, ...)

local ClassTargetSelectors = {}
addon.ClassTargetSelectors = ClassTargetSelectors

local TargetSelector = addon.TargetSelector
local TargetSelectionFilter = addon.TargetSelectionFilter

---@type fun(method: string): TargetSelector
local getTankSelector
do
	local tankSelectionFactories = {
		tankRoleOnly = function()
			return TargetSelector.Sort(TargetSelector.PartyOrRaid(TargetSelectionFilter.Role("TANK")))
		end,
		tanksAndMainTanks = function()
			return TargetSelector.Sort(TargetSelector.PartyOrRaid(
				TargetSelectionFilter.Any({
					TargetSelectionFilter.MainTank(),
					TargetSelectionFilter.Role("TANK"),
				})
			))
		end,
		prioritizeMainTanks = function()
			return TargetSelector.Chain({
				TargetSelector.Sort(TargetSelector.PartyOrRaid(TargetSelectionFilter.MainTank())),
				TargetSelector.Sort(TargetSelector.PartyOrRaid(TargetSelectionFilter.Role("TANK"))),
			})
		end,
		mainTanksOnly = function()
			return TargetSelector.Sort(TargetSelector.PartyOrRaid(TargetSelectionFilter.MainTank()))
		end,
	}

	getTankSelector = function(method)
		return (tankSelectionFactories[method] or tankSelectionFactories.tankRoleOnly)()
	end
end

---@param selector TargetSelector
---@return TargetSelector
local function chainWithFocus(selector)
	if addon.db.profile.prioritizeFocus then
		return TargetSelector.Chain({
			TargetSelector.Focus(),
			selector,
		})
	end
	return selector
end

function ClassTargetSelectors.HUNTER()
	return chainWithFocus(TargetSelector.Chain({
		getTankSelector(addon.db.profile.tankSelectionMethod),
		TargetSelector.Pet()
	}))
end

function ClassTargetSelectors.ROGUE()
	return chainWithFocus(getTankSelector(addon.db.profile.tankSelectionMethod))
end

function ClassTargetSelectors.PALADIN()
	return chainWithFocus(getTankSelector(addon.db.profile.tankSelectionMethod))
end

function ClassTargetSelectors.EVOKER()
	return chainWithFocus(TargetSelector.Chain({
		getTankSelector(addon.db.profile.tankSelectionMethod),
		TargetSelector.Player(),
	}))
end

function ClassTargetSelectors.DRUID()
	return chainWithFocus(
		TargetSelector.Sort(TargetSelector.PartyOrRaid(TargetSelectionFilter.Role("HEALER")))
	)
end