From 397296ad231e220af96c82dd93831d4f9000459f Mon Sep 17 00:00:00 2001
From: muleyo <admin@muleyo.dev>
Date: Tue, 9 Jul 2024 12:33:45 +0200
Subject: [PATCH] EditMode PetActionBar/StanceBar Size fixes

---
 Modules/ActionBars/Load.xml   | 15 ++++++++-------
 Modules/ActionBars/_Small.lua | 15 ++++++++-------
 2 files changed, 16 insertions(+), 14 deletions(-)

diff --git a/Modules/ActionBars/Load.xml b/Modules/ActionBars/Load.xml
index 5f9d54e..3480fbb 100755
--- a/Modules/ActionBars/Load.xml
+++ b/Modules/ActionBars/Load.xml
@@ -1,13 +1,14 @@
 <Ui xmlns='http://www.blizzard.com/wow/ui/'>
+	<Script file = "_BFA.lua" />
+	<Script file = "_BFATransparent.lua" />
+	<Script file = "_Bindings.lua" />
 	<Script file = "_Buttons.lua"/>
+	<Script file = "_DefaultNoBg.lua" />
+	<Script file = "_DefaultPetActionBarFix.lua" />
+	<Script file = "_Flash.lua"/>
 	<Script file = "_Gryphones.lua"/>
+	<Script file = "_Menu.lua" />
+	<Script file = "_Mouseover.lua" />
 	<Script file = "_Range.lua"/>
-	<Script file = "_Flash.lua"/>
   	<Script file = "_Small.lua"/>
-	<Script file = "_DefaultNoBg.lua" />
-	<Script file = "_Mouseover.lua" />
-	<Script file = "_Bindings.lua" />
-	<Script file = "_BFA.lua" />
-	<Script file = "_BFATransparent.lua" />
-	<Script file = "_Menu.lua" />
 </Ui>
\ No newline at end of file
diff --git a/Modules/ActionBars/_Small.lua b/Modules/ActionBars/_Small.lua
index ec46435..2801022 100644
--- a/Modules/ActionBars/_Small.lua
+++ b/Modules/ActionBars/_Small.lua
@@ -58,12 +58,12 @@ function Module:OnEnable()
 		SUIMultiBarBottomRight:SetPoint("TOP", SUIMultiBarBottomLeft, "TOP", 0, 0)
 
 		SUIStanceBar = CreateFrame("Frame", "SUIStanceBar", UIParent)
-		SUIStanceBar:SetSize(18 * NUM_STANCE_SLOTS, size)
+		SUIStanceBar:SetSize(36 * GetNumShapeshiftForms(), size-10)
 		SUIStanceBar:SetPoint("TOP", SUIMultiBarBottomRight, "TOP", -45, 150)
 		SUIStanceBar:SetMovable(true)
 
 		SUIPetActionBar = CreateFrame("Frame", "SUIPetActionBar", UIParent)
-		SUIPetActionBar:SetSize(14 * 10, size)
+		SUIPetActionBar:SetSize(PetActionBarFrame:GetWidth()-140, size-10)
 		SUIPetActionBar:SetPoint("TOP", SUIMultiBarBottomRight, "TOP", -45, 150)
 		SUIPetActionBar:SetMovable(true)
 
@@ -85,10 +85,10 @@ function Module:OnEnable()
 			MultiCastActionBarFrame.SetPoint = function() end
 		end)
 
-		hooksecurefunc("StanceBar_Update", function()
-			StanceBarLeft:Hide()
-			StanceBarMiddle:Hide()
-			StanceBarRight:Hide()
+		C_Timer.After(0.1, function()
+			StanceBarLeft:SetAlpha(0)
+			StanceBarMiddle:SetAlpha(0)
+			StanceBarRight:SetAlpha(0)
 		end)
 
 		local function updatePositions()
@@ -136,7 +136,7 @@ function Module:OnEnable()
 				end
 			end
 
-			for i = 1, NUM_STANCE_SLOTS do
+			for i = 1, GetNumShapeshiftForms() do
 				local button = _G["StanceButton"..i]
 				local prevButton = _G["StanceButton"..i-1]
 
@@ -146,6 +146,7 @@ function Module:OnEnable()
 				if (i == 1) then
 					button:SetPoint("LEFT", SUIStanceBar, 0, 0)
 				else
+					if (i == 6) then return end
 					button:SetPoint("LEFT", prevButton, 37.5, 0)
 				end
 			end