be5ae28 - Limit stance bar to paladin and druid atm

[wowui.git] / OmaAB / StanceBar.lua

diff --git a/OmaAB/StanceBar.lua b/OmaAB/StanceBar.lua
index 0fa0c1a..3428b8d 100644 (file)
--- a/OmaAB/StanceBar.lua
+++ b/OmaAB/StanceBar.lua
@@ -59,6 +59,10 @@ stancebar:SetScript("OnEvent", function(self, event)
     elseif event == "PLAYER_LOGIN" then
         -- TODO do class detection and only create necessary buttons
         stancebar:UnregisterAllEvents();
+        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