initial commit
This commit is contained in:
Gitea
270
Modules/ActionBars/StanceBar.lua
Normal file
270
Modules/ActionBars/StanceBar.lua
Normal file
@@ -0,0 +1,270 @@
|
||||
local E, L, V, P, G = unpack(select(2, ...)); --Import: Engine, Locales, PrivateDB, ProfileDB, GlobalDB
|
||||
local AB = E:GetModule('ActionBars')
|
||||
|
||||
local _G = _G
|
||||
local gsub = gsub
|
||||
local format, ipairs = format, ipairs
|
||||
local CooldownFrame_Set = CooldownFrame_Set
|
||||
local CreateFrame = CreateFrame
|
||||
local GetBindingKey = GetBindingKey
|
||||
local GetNumShapeshiftForms = GetNumShapeshiftForms
|
||||
local GetShapeshiftForm = GetShapeshiftForm
|
||||
local GetShapeshiftFormCooldown = GetShapeshiftFormCooldown
|
||||
local GetShapeshiftFormInfo = GetShapeshiftFormInfo
|
||||
local GetSpellInfo = GetSpellInfo
|
||||
local InCombatLockdown = InCombatLockdown
|
||||
local RegisterStateDriver = RegisterStateDriver
|
||||
local NUM_STANCE_SLOTS = NUM_STANCE_SLOTS
|
||||
|
||||
local Masque = E.Masque
|
||||
local MasqueGroup = Masque and Masque:Group('ElvUI', 'Stance Bar')
|
||||
local WispSplode = [[Interface\Icons\Spell_Nature_WispSplode]]
|
||||
local bar = CreateFrame('Frame', 'ElvUI_StanceBar', E.UIParent, 'SecureHandlerStateTemplate')
|
||||
bar.buttons = {}
|
||||
|
||||
function AB:UPDATE_SHAPESHIFT_COOLDOWN()
|
||||
local numForms = GetNumShapeshiftForms()
|
||||
local start, duration, enable, cooldown
|
||||
for i = 1, NUM_STANCE_SLOTS do
|
||||
if i <= numForms then
|
||||
cooldown = _G['ElvUI_StanceBarButton'..i..'Cooldown']
|
||||
start, duration, enable = GetShapeshiftFormCooldown(i)
|
||||
CooldownFrame_Set(cooldown, start, duration, enable)
|
||||
cooldown:SetDrawBling(cooldown:GetEffectiveAlpha() > 0.5) --Cooldown Bling Fix
|
||||
end
|
||||
end
|
||||
|
||||
AB:StyleShapeShift('UPDATE_SHAPESHIFT_COOLDOWN')
|
||||
end
|
||||
|
||||
function AB:StyleShapeShift()
|
||||
local numForms = GetNumShapeshiftForms()
|
||||
local stance = GetShapeshiftForm()
|
||||
local darkenInactive = AB.db.stanceBar.style == 'darkenInactive'
|
||||
|
||||
for i = 1, NUM_STANCE_SLOTS do
|
||||
local buttonName = 'ElvUI_StanceBarButton'..i
|
||||
local button = _G[buttonName]
|
||||
local cooldown = _G[buttonName..'Cooldown']
|
||||
|
||||
if i <= numForms then
|
||||
local texture, isActive, isCastable, spellID, _ = GetShapeshiftFormInfo(i)
|
||||
|
||||
if darkenInactive then
|
||||
_, _, texture = GetSpellInfo(spellID)
|
||||
end
|
||||
|
||||
if not texture then texture = WispSplode end
|
||||
|
||||
button.icon:SetTexture(texture)
|
||||
|
||||
if not button.useMasque then
|
||||
cooldown:SetAlpha(1)
|
||||
|
||||
if isActive then
|
||||
_G.StanceBarFrame.lastSelected = button:GetID()
|
||||
if numForms == 1 then
|
||||
button.checked:SetColorTexture(1, 1, 1, 0.5)
|
||||
button:SetChecked(true)
|
||||
else
|
||||
button.checked:SetColorTexture(1, 1, 1, 0.5)
|
||||
button:SetChecked(not darkenInactive)
|
||||
end
|
||||
else
|
||||
if numForms == 1 or stance == 0 then
|
||||
button:SetChecked(false)
|
||||
else
|
||||
button:SetChecked(darkenInactive)
|
||||
button.checked:SetAlpha(1)
|
||||
if darkenInactive then
|
||||
button.checked:SetColorTexture(0, 0, 0, 0.5)
|
||||
else
|
||||
button.checked:SetColorTexture(1, 1, 1, 0.5)
|
||||
end
|
||||
end
|
||||
end
|
||||
else
|
||||
if isActive then
|
||||
button:SetChecked(true)
|
||||
else
|
||||
button:SetChecked(false)
|
||||
end
|
||||
end
|
||||
|
||||
if isCastable then
|
||||
button.icon:SetVertexColor(1.0, 1.0, 1.0)
|
||||
else
|
||||
button.icon:SetVertexColor(0.4, 0.4, 0.4)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function AB:PositionAndSizeBarShapeShift()
|
||||
local db = AB.db.stanceBar
|
||||
|
||||
local buttonSpacing = db.buttonspacing
|
||||
local backdropSpacing = db.backdropSpacing
|
||||
local buttonsPerRow = db.buttonsPerRow
|
||||
local numButtons = db.buttons
|
||||
local point = db.point
|
||||
local visibility = db.visibility
|
||||
|
||||
bar.db = db
|
||||
bar.mouseover = db.mouseover
|
||||
|
||||
if bar.LastButton then
|
||||
if numButtons > bar.LastButton then numButtons = bar.LastButton end
|
||||
if buttonsPerRow > bar.LastButton then buttonsPerRow = bar.LastButton end
|
||||
end
|
||||
if numButtons < buttonsPerRow then
|
||||
buttonsPerRow = numButtons
|
||||
end
|
||||
|
||||
bar:SetParent(db.inheritGlobalFade and AB.fadeParent or E.UIParent)
|
||||
bar:EnableMouse(not db.clickThrough)
|
||||
bar:SetAlpha(bar.mouseover and 0 or db.alpha)
|
||||
AB:FadeBarBlings(bar, bar.mouseover and 0 or db.alpha)
|
||||
|
||||
bar.backdrop:SetShown(db.backdrop)
|
||||
bar.backdrop:ClearAllPoints()
|
||||
|
||||
AB:MoverMagic(bar)
|
||||
|
||||
local _, horizontal, anchorUp, anchorLeft = AB:GetGrowth(point)
|
||||
local button, lastButton, lastColumnButton, anchorRowButton, lastShownButton
|
||||
local useMasque = MasqueGroup and E.private.actionbar.masque.stanceBar
|
||||
|
||||
for i = 1, NUM_STANCE_SLOTS do
|
||||
button = _G['ElvUI_StanceBarButton'..i]
|
||||
lastButton = _G['ElvUI_StanceBarButton'..i-1]
|
||||
lastColumnButton = _G['ElvUI_StanceBarButton'..i-buttonsPerRow]
|
||||
button.db = db
|
||||
|
||||
if i == 1 or i == buttonsPerRow then
|
||||
anchorRowButton = button
|
||||
end
|
||||
|
||||
if i > numButtons then
|
||||
button:SetAlpha(0)
|
||||
button.handleBackdrop = nil
|
||||
else
|
||||
button:SetAlpha(db.alpha)
|
||||
lastShownButton = button
|
||||
button.handleBackdrop = true -- keep over HandleButton
|
||||
end
|
||||
|
||||
AB:HandleButton(bar, button, i, lastButton, lastColumnButton)
|
||||
|
||||
if useMasque then
|
||||
MasqueGroup:AddButton(bar.buttons[i])
|
||||
end
|
||||
|
||||
if not button.FlyoutUpdateFunc then
|
||||
AB:StyleButton(button, nil, useMasque, true)
|
||||
|
||||
if not useMasque then
|
||||
if db.style == 'darkenInactive' then
|
||||
button.checked:SetBlendMode('BLEND')
|
||||
else
|
||||
button.checked:SetBlendMode('ADD')
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
AB:HandleBackdropMultiplier(bar, backdropSpacing, buttonSpacing, db.widthMult, db.heightMult, anchorUp, anchorLeft, horizontal, lastShownButton, anchorRowButton)
|
||||
AB:HandleBackdropMover(bar, backdropSpacing)
|
||||
|
||||
if db.enabled then
|
||||
visibility = gsub(visibility,'[\n\r]','')
|
||||
|
||||
RegisterStateDriver(bar, 'visibility', (GetNumShapeshiftForms() == 0 and 'hide') or visibility)
|
||||
E:EnableMover(bar.mover:GetName())
|
||||
else
|
||||
RegisterStateDriver(bar, 'visibility', 'hide')
|
||||
E:DisableMover(bar.mover:GetName())
|
||||
end
|
||||
|
||||
if useMasque then
|
||||
MasqueGroup:ReSkin()
|
||||
|
||||
for _, btn in ipairs(bar.buttons) do
|
||||
AB:TrimIcon(btn, true)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function AB:AdjustMaxStanceButtons(event)
|
||||
if InCombatLockdown() then
|
||||
AB.NeedsAdjustMaxStanceButtons = event or true
|
||||
AB:RegisterEvent('PLAYER_REGEN_ENABLED')
|
||||
return
|
||||
end
|
||||
|
||||
for _, button in ipairs(bar.buttons) do
|
||||
button:Hide()
|
||||
end
|
||||
|
||||
local numButtons = GetNumShapeshiftForms()
|
||||
for i = 1, NUM_STANCE_SLOTS do
|
||||
if not bar.buttons[i] then
|
||||
bar.buttons[i] = CreateFrame('CheckButton', format(bar:GetName()..'Button%d', i), bar, 'StanceButtonTemplate')
|
||||
bar.buttons[i]:SetID(i)
|
||||
|
||||
AB:HookScript(bar.buttons[i], 'OnEnter', 'Button_OnEnter')
|
||||
AB:HookScript(bar.buttons[i], 'OnLeave', 'Button_OnLeave')
|
||||
end
|
||||
|
||||
local blizz = _G[format('StanceButton%d', i)]
|
||||
if blizz and blizz.commandName then
|
||||
bar.buttons[i].commandName = blizz.commandName
|
||||
end
|
||||
|
||||
if i <= numButtons then
|
||||
bar.buttons[i]:Show()
|
||||
bar.LastButton = i
|
||||
end
|
||||
end
|
||||
|
||||
AB:PositionAndSizeBarShapeShift()
|
||||
|
||||
-- sometimes after combat lock down `event` may be true because of passing it back with `AB.NeedsAdjustMaxStanceButtons`
|
||||
if event == 'UPDATE_SHAPESHIFT_FORMS' or event == 'PLAYER_ENTERING_WORLD' then
|
||||
AB:StyleShapeShift()
|
||||
end
|
||||
end
|
||||
|
||||
function AB:UpdateStanceBindings()
|
||||
for i = 1, NUM_STANCE_SLOTS do
|
||||
local button = _G['ElvUI_StanceBarButton'..i]
|
||||
if not button then break end
|
||||
|
||||
if AB.db.hotkeytext then
|
||||
button.HotKey:Show()
|
||||
button.HotKey:SetText(GetBindingKey('SHAPESHIFTBUTTON'..i))
|
||||
|
||||
AB:FixKeybindText(button)
|
||||
else
|
||||
button.HotKey:Hide()
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function AB:CreateBarShapeShift()
|
||||
bar:CreateBackdrop(AB.db.transparent and 'Transparent')
|
||||
bar.backdrop:SetFrameLevel(0)
|
||||
|
||||
bar:Point('TOPLEFT', E.UIParent, 'BOTTOMLEFT', 4, -769)
|
||||
|
||||
AB:HookScript(bar, 'OnEnter', 'Bar_OnEnter')
|
||||
AB:HookScript(bar, 'OnLeave', 'Bar_OnLeave')
|
||||
|
||||
AB:RegisterEvent('UPDATE_SHAPESHIFT_COOLDOWN')
|
||||
AB:RegisterEvent('UPDATE_SHAPESHIFT_FORMS', 'AdjustMaxStanceButtons')
|
||||
AB:RegisterEvent('UPDATE_SHAPESHIFT_FORM', 'StyleShapeShift')
|
||||
AB:RegisterEvent('UPDATE_SHAPESHIFT_USABLE', 'StyleShapeShift')
|
||||
AB:RegisterEvent('ACTIONBAR_PAGE_CHANGED', 'StyleShapeShift')
|
||||
|
||||
E:CreateMover(bar, 'ShiftAB', L["Stance Bar"], nil, -3, nil, 'ALL,ACTIONBARS', nil, 'actionbar,stanceBar', true)
|
||||
end
|
||||
Reference in New Issue
Block a user