a7dd68b - priest and mage click action updates

[wowui.git] / OmaAB / StanceBar.lua

diff --git a/OmaAB/StanceBar.lua b/OmaAB/StanceBar.lua
index afb5cc6..3428b8d 100644 (file)
--- a/OmaAB/StanceBar.lua
+++ b/OmaAB/StanceBar.lua
@@ -30,7 +30,7 @@ end
 function update(bar)
     local n = GetNumShapeshiftForms();
     --for i=1,NUM_STANCE_SLOTS do
 function update(bar)
     local n = GetNumShapeshiftForms();
     --for i=1,NUM_STANCE_SLOTS do

-    for i=1,3 do
+    for i=1,4 do

         local button = bar[i];
         if i <= n then
             local tex, active, castable = GetShapeshiftFormInfo(i);
         local button = bar[i];
         if i <= n then
             local tex, active, castable = GetShapeshiftFormInfo(i);


@@ -46,9 +46,9 @@ function update(bar)
             else
                 icon:SetVertexColor(0.4, 0.4, 0.4);
             end
             else
                 icon:SetVertexColor(0.4, 0.4, 0.4);
             end

-            button:Show();
+            --button:Show();

         else
         else

-            button:Hide();
+            --button:Hide();

         end
     end
 end
         end
     end
 end


@@ -59,12 +59,16 @@ stancebar:SetScript("OnEvent", function(self, event)
     elseif event == "PLAYER_LOGIN" then
         -- TODO do class detection and only create necessary buttons
         stancebar:UnregisterAllEvents();
     elseif event == "PLAYER_LOGIN" then
         -- TODO do class detection and only create necessary buttons
         stancebar:UnregisterAllEvents();

-        for i=1,3 do
+        local _, class = UnitClass("player");
+        if class ~= "PALADIN" and class ~= "DRUID" then
+            return
+        end
+        for i=1,4 do

             create(i, self, self[i-1]);
         end
         update(self);
         _G["BINDING_HEADER_OMAABSTANCE"] = "Stance Bar";
             create(i, self, self[i-1]);
         end
         update(self);
         _G["BINDING_HEADER_OMAABSTANCE"] = "Stance Bar";

-        for i = 1,3 do
+        for i = 1,4 do

             _G[format("BINDING_NAME_CLICK OmaBTStance%d:LeftButton", i)] = format("Stance Bar Button %d", i);
         end
         stancebar:RegisterEvent("UPDATE_SHAPESHIFT_COOLDOWN");
             _G[format("BINDING_NAME_CLICK OmaBTStance%d:LeftButton", i)] = format("Stance Bar Button %d", i);
         end
         stancebar:RegisterEvent("UPDATE_SHAPESHIFT_COOLDOWN");