Broker_Essentials/EquipmentSets.lua at master · Lombra/Broker_Essentials · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
local _, core = ...

local function onClick(self, set)
	EquipmentManager_EquipSet(set)
end

local dropdown = core:CreateDropdown("Menu")
dropdown.initialize = function(self)
	for i, id in ipairs(C_EquipmentSet.GetEquipmentSetIDs()) do
		local name, icon, setID, isEquipped, numItems, numEquipped, numInventory, numMissing, numIgnored = C_EquipmentSet.GetEquipmentSetInfo(id)
		local info = UIDropDownMenu_CreateInfo()
		info.text = name
		info.func = onClick
		info.arg1 = id
		info.isNotRadio = true
		info.checked = isEquipped
		self:AddButton(info)
	end
end

local module = core:NewModule("EquipmentSets", {
	type = "data source",
	label = "Equipment sets",
	icon = [[Interface\PaperDollInfoFrame\UI-GearManager-Button]],
	OnClick = function(self)
		dropdown:Toggle(nil, self)
	end,
	-- OnTooltipShow = function(self)
		-- GetEquipmentSetLocations("name")
	-- end,
})

function module:OnInitialize()
	self:RegisterEvent("EQUIPMENT_SETS_CHANGED", "Update")
	self:RegisterEvent("PLAYER_EQUIPMENT_CHANGED", "Update")
	self:RegisterEvent("BAG_UPDATE_DELAYED", "Update")
end

function module:Update()
	local equipped
	local mostEquipped = 0
	for i, id in ipairs(C_EquipmentSet.GetEquipmentSetIDs()) do
		local name, icon, setID, isEquipped, numItems, numEquipped, numInventory, numMissing, numIgnored = C_EquipmentSet.GetEquipmentSetInfo(id)
		if numEquipped > mostEquipped then
			equipped = id
			mostEquipped = numEquipped
		end
	end
	if equipped then
		local name, icon, setID, isEquipped, numItems, numEquipped, numInventory, numMissing, numIgnored = C_EquipmentSet.GetEquipmentSetInfo(equipped)
		if isEquipped then
			self.text = name
		else
			self.text = format("%s (|cffff2020-%d|r)", name, numItems - numEquipped)
		end
	else
		self.text = "N/A"
	end
end