From d53f89c6d42b46f41fc827a78775f21be3f75ef8 Mon Sep 17 00:00:00 2001
From: Sumsebrum <kyle@xicon.de>
Date: Thu, 29 Jul 2021 14:06:15 +0200
Subject: [PATCH] hide blizzard arena pets as well

---
 Frame.lua  | 11 +++++++++++
 Gladdy.lua | 46 +++++++++++++++++++++++++---------------------
 2 files changed, 36 insertions(+), 21 deletions(-)

diff --git a/Frame.lua b/Frame.lua
index e4a9fd0..5ed139a 100644
--- a/Frame.lua
+++ b/Frame.lua
@@ -275,6 +275,17 @@ function Gladdy:UpdateFrame()
     for _, v in self:IterModules() do
         self:Call(v, "UpdateFrameOnce")
     end
+    if Gladdy.db.hideBlizzard == "always" then
+        Gladdy:BlizzArenaSetAlpha(0)
+    elseif Gladdy.db.hideBlizzard == "arena" then
+        if Gladdy.curBracket then
+            Gladdy:BlizzArenaSetAlpha(0)
+        else
+            Gladdy:BlizzArenaSetAlpha(1)
+        end
+    elseif Gladdy.db.hideBlizzard == "never" then
+        Gladdy:BlizzArenaSetAlpha(1)
+    end
 end
 
 function Gladdy:HideFrame()
diff --git a/Gladdy.lua b/Gladdy.lua
index 252c270..32b0e1f 100644
--- a/Gladdy.lua
+++ b/Gladdy.lua
@@ -229,13 +229,7 @@ function Gladdy:OnInitialize()
     end
     self:DeleteUnknownOptions(self.db, self.defaults.profile)
     if Gladdy.db.hideBlizzard == "always" then
-        if IsAddOnLoaded("Blizzard_ArenaUI") then
-            ArenaEnemyFrame1:SetAlpha(0)
-            ArenaEnemyFrame2:SetAlpha(0)
-            ArenaEnemyFrame3:SetAlpha(0)
-            ArenaEnemyFrame4:SetAlpha(0)
-            ArenaEnemyFrame5:SetAlpha(0)
-        end
+        Gladdy:BlizzArenaSetAlpha(0)
     end
 end
 
@@ -391,13 +385,7 @@ function Gladdy:Reset()
         self:ResetUnit(unit)
     end
     if Gladdy.db.hideBlizzard == "never" or Gladdy.db.hideBlizzard == "arena" then
-        if IsAddOnLoaded("Blizzard_ArenaUI") then
-            ArenaEnemyFrame1:SetAlpha(1)
-            ArenaEnemyFrame2:SetAlpha(1)
-            ArenaEnemyFrame3:SetAlpha(1)
-            ArenaEnemyFrame4:SetAlpha(1)
-            ArenaEnemyFrame5:SetAlpha(1)
-        end
+        Gladdy:BlizzArenaSetAlpha(1)
     end
 end
 
@@ -462,12 +450,28 @@ function Gladdy:JoinedArena()
         self.buttons["arena" .. i]:SetAlpha(1)
     end
     if Gladdy.db.hideBlizzard == "arena" or Gladdy.db.hideBlizzard == "always" then
-        if IsAddOnLoaded("Blizzard_ArenaUI") then
-            ArenaEnemyFrame1:SetAlpha(0)
-            ArenaEnemyFrame2:SetAlpha(0)
-            ArenaEnemyFrame3:SetAlpha(0)
-            ArenaEnemyFrame4:SetAlpha(0)
-            ArenaEnemyFrame5:SetAlpha(0)
-        end
+        Gladdy:BlizzArenaSetAlpha(0)
     end
 end
+
+---------------------------
+
+-- BLIZZARD FRAMES
+
+---------------------------
+
+function Gladdy:BlizzArenaSetAlpha(alpha)
+    if IsAddOnLoaded("Blizzard_ArenaUI") then
+        ArenaEnemyFrames:SetAlpha(alpha)
+        ArenaEnemyFrame1:SetAlpha(alpha)
+        ArenaEnemyFrame1PetFrame:SetAlpha(alpha)
+        ArenaEnemyFrame2:SetAlpha(alpha)
+        ArenaEnemyFrame2PetFrame:SetAlpha(alpha)
+        ArenaEnemyFrame3:SetAlpha(alpha)
+        ArenaEnemyFrame3PetFrame:SetAlpha(alpha)
+        ArenaEnemyFrame4:SetAlpha(alpha)
+        ArenaEnemyFrame4PetFrame:SetAlpha(alpha)
+        ArenaEnemyFrame5:SetAlpha(alpha)
+        ArenaEnemyFrame5PetFrame:SetAlpha(alpha)
+    end
+end
\ No newline at end of file