From fb8fbbc8c438d23478dff2d7b67d4f37944701be Mon Sep 17 00:00:00 2001
From: shagu <ericmauser@googlemail.com>
Date: Sun, 4 Nov 2018 13:11:31 +0100
Subject: [PATCH] loot: move client abstraction into compat files

---
 compat/tbc.lua     | 1 +
 compat/vanilla.lua | 1 +
 modules/loot.lua   | 6 +-----
 3 files changed, 3 insertions(+), 5 deletions(-)

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()