diff --git a/compat/tbc.lua b/compat/tbc.lua index 07109dae..d4d18323 100644 --- a/compat/tbc.lua +++ b/compat/tbc.lua @@ -23,6 +23,7 @@ FRIENDS_NAME_LOCATION = "ButtonTextLocation" COOLDOWN_FRAME_TYPE = "Cooldown" LOOT_BUTTON_FRAME_TYPE = "Button" +LOOT_BUTTON_SLOT_FUNCTION = "SetID" PLAYER_BUFF_START_ID = 0 diff --git a/compat/vanilla.lua b/compat/vanilla.lua index 6c2748a6..61718ce1 100644 --- a/compat/vanilla.lua +++ b/compat/vanilla.lua @@ -18,6 +18,7 @@ FRIENDS_NAME_LOCATION = "ButtonTextNameLocation" COOLDOWN_FRAME_TYPE = "Model" LOOT_BUTTON_FRAME_TYPE = "LootButton" +LOOT_BUTTON_SLOT_FUNCTION = "SetSlot" PLAYER_BUFF_START_ID = -1 diff --git a/modules/loot.lua b/modules/loot.lua index 33ee891a..f1a7d679 100644 --- a/modules/loot.lua +++ b/modules/loot.lua @@ -470,11 +470,7 @@ pfUI:RegisterModule("loot", function () maxrarity = math.max(maxrarity, quality) maxwidth = math.max(maxwidth, slot.name:GetStringWidth()) - if slot.SetSlot then - slot:SetSlot(id) - else - slot:SetID(id) - end + slot[LOOT_BUTTON_SLOT_FUNCTION](slot, id) slot:Enable() slot:Show()