Fix new tooltip code for classic clients

SharpWoW · Dec 2, 2022 · 6f726fa · 6f726fa
1 parent 118703c
commit 6f726fa
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/.luacheckrc b/.luacheckrc
@@ -67,7 +67,9 @@ stds.wow = {
     "UnitName",
     "UnitXP",
     "UnitXPMax",
-    "wipe"
+    "wipe",
+    "WOW_PROJECT_ID",
+    "WOW_PROJECT_MAINLINE"
   }
 }
 

diff --git a/KillTrack.lua b/KillTrack.lua
@@ -275,7 +275,7 @@ function KT.Events.ENCOUNTER_END(self, _, _, _, size)
     end
 end
 
-TooltipDataProcessor.AddTooltipPostCall(Enum.TooltipDataType.Unit, function(self)
+local function tooltip_enhancer(self)
     if not KT.Global.TOOLTIP then return end
     local _, unit = self:GetUnit()
     if not unit then return end
@@ -296,7 +296,13 @@ TooltipDataProcessor.AddTooltipPostCall(Enum.TooltipDataType.Unit, function(self
         self:AddLine(("ID = %d"):format(id))
     end
     self:Show()
-end)
+end
+
+if WOW_PROJECT_ID == WOW_PROJECT_MAINLINE then
+    TooltipDataProcessor.AddTooltipPostCall(Enum.TooltipDataType.Unit, tooltip_enhancer)
+else
+    GameTooltip:HookScript("OnTooltipSetUnit", tooltip_enhancer)
+end
 
 function KT:ToggleLoadMessage()
     self.Global.LOAD_MESSAGE = not self.Global.LOAD_MESSAGE