128 lines
4.1 KiB
Lua
128 lines
4.1 KiB
Lua
local _G = ShaguTweaks.GetGlobalEnv()
|
|
local T = ShaguTweaks.T
|
|
local API = ShaguTweaks.API
|
|
|
|
local module = ShaguTweaks:register({
|
|
title = T["Reagent Counter"],
|
|
description = T["Shows a reagent counter on action buttons."],
|
|
expansions = { ["vanilla"] = true, ["tbc"] = false },
|
|
maintainer = "@shagu (GitHub)",
|
|
category = T["Action Bar"],
|
|
enabled = nil,
|
|
})
|
|
|
|
module.enable = function(self)
|
|
local reagent_slots = { }
|
|
local reagent_counts = { }
|
|
local bars = { "Action", "BonusAction", "MultiBarBottomLeft", "MultiBarBottomRight", "MultiBarLeft", "MultiBarRight" }
|
|
|
|
local reagentcounter = CreateFrame("Frame", "ShaguTweaksReagentCount", UIParent)
|
|
reagentcounter:RegisterEvent("PLAYER_ENTERING_WORLD")
|
|
reagentcounter:RegisterEvent("ACTIONBAR_SLOT_CHANGED")
|
|
reagentcounter:RegisterEvent("ACTIONBAR_PAGE_CHANGED")
|
|
reagentcounter:RegisterEvent("UPDATE_BONUS_ACTIONBAR")
|
|
reagentcounter:RegisterEvent("UPDATE_MACROS")
|
|
reagentcounter:RegisterEvent("BAG_UPDATE")
|
|
|
|
-- Hidden frames still receive events, but do not execute OnUpdate.
|
|
-- Wake for one frame only when work is pending.
|
|
reagentcounter:Hide()
|
|
reagentcounter:SetScript("OnEvent", function()
|
|
this.pending = true
|
|
if event ~= "BAG_UPDATE" then
|
|
this.rescan = true
|
|
end
|
|
this:Show()
|
|
end)
|
|
|
|
reagentcounter:SetScript("OnUpdate", function()
|
|
if not this.pending then
|
|
this:Hide()
|
|
return
|
|
end
|
|
|
|
-- Rebuild the action-slot -> reagent mapping only when the action layout
|
|
-- changes. ClassicAPI resolves the spell/macro directly, so no tooltip
|
|
-- scan or localized reagent-name parsing is required.
|
|
if this.rescan then
|
|
reagent_counts = {}
|
|
for slot = 1, 120 do
|
|
reagentcounter.ScanSlot(slot)
|
|
end
|
|
end
|
|
|
|
-- Count only reagents referenced by active actions. ClassicAPI reads the
|
|
-- inventory directly, avoiding a Lua bag scan for every reagent.
|
|
for itemID in pairs(reagent_counts) do
|
|
reagent_counts[itemID] = API.GetItemCount(itemID, false, false) or 0
|
|
end
|
|
|
|
-- update all actionbar buttons
|
|
for _, prefix in ipairs(bars) do
|
|
for i = 1, NUM_ACTIONBAR_BUTTONS do
|
|
local button = _G[prefix .. "Button" .. i]
|
|
if button then
|
|
local text = _G[button:GetName().."Count"]
|
|
local slot = ActionButton_GetPagedID(button)
|
|
|
|
if text then
|
|
local itemID = reagent_slots[slot]
|
|
if itemID then
|
|
text:SetText(reagent_counts[itemID] or 0)
|
|
elseif not IsConsumableAction(slot) then
|
|
text:SetText()
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
this.pending = nil
|
|
this.rescan = nil
|
|
this:Hide()
|
|
end)
|
|
|
|
reagentcounter.ScanSlot = function(slot)
|
|
reagent_slots[slot] = nil
|
|
if not HasAction(slot) then return end
|
|
|
|
local actionType, actionID = API.GetActionInfo(slot)
|
|
local spellID
|
|
|
|
if actionType == "spell" then
|
|
spellID = actionID
|
|
elseif actionType == "macro" and actionID then
|
|
local _, _, macroSpellID = API.GetMacroSpell(actionID)
|
|
spellID = macroSpellID
|
|
end
|
|
|
|
if not spellID then return end
|
|
|
|
local reagents = API.GetSpellReagents(spellID)
|
|
local reagent = reagents and reagents[1]
|
|
if not reagent or not reagent.itemID then return end
|
|
|
|
-- The action button only has one count field. Preserve the historical
|
|
-- behavior by showing the first reagent's owned stack count.
|
|
reagent_slots[slot] = reagent.itemID
|
|
reagent_counts[reagent.itemID] = reagent_counts[reagent.itemID] or 0
|
|
end
|
|
|
|
-- The stock UI clears the Count fontstring on non-consumable actions whenever
|
|
-- ActionButton_UpdateCount() runs. Re-apply our reagent value afterwards so
|
|
-- login/reload refreshes cannot erase it a moment later.
|
|
ShaguTweaks.hooksecurefunc("ActionButton_UpdateCount", function()
|
|
local button = this
|
|
if not button or not button.GetName then return end
|
|
|
|
local slot = ActionButton_GetPagedID(button)
|
|
local itemID = slot and reagent_slots[slot]
|
|
if not itemID then return end
|
|
|
|
local text = _G[button:GetName().."Count"]
|
|
if text then
|
|
text:SetText(reagent_counts[itemID] or 0)
|
|
end
|
|
end)
|
|
end
|