From 40a41ee436a76dc4810915be1bb032c05b001fcb Mon Sep 17 00:00:00 2001
From: shagu <ericmauser@googlemail.com>
Date: Wed, 30 Dec 2020 13:05:19 +0100
Subject: [PATCH] ui-widgets: set script if no other script is found

---
 api/ui-widgets.lua | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/api/ui-widgets.lua b/api/ui-widgets.lua
index 326ac21a..ff90aecf 100644
--- a/api/ui-widgets.lua
+++ b/api/ui-widgets.lua
@@ -591,8 +591,20 @@ function pfUI.api.SetHighlight(frame, cr, cg, cb)
 
   if not frame.pfEnterLeave then
     if not frame.HookScript then frame.HookScript = HookScript end
-    frame:HookScript("OnEnter", SetHighlightEnter)
-    frame:HookScript("OnLeave", SetHighlightLeave)
+    local enter, leave = frame:GetScript("OnEnter"), frame:GetScript("OnLeave")
+
+    if enter then
+      frame:HookScript("OnEnter", SetHighlightEnter)
+    else
+      frame:SetScript("OnEnter", SetHighlightEnter)
+    end
+
+    if leave then
+      frame:HookScript("OnLeave", SetHighlightLeave)
+    else
+      frame:SetScript("OnLeave", SetHighlightLeave)
+    end
+
     frame.pfEnterLeave = true
   end
 end