Hendrik Leppkes
133 lines
3.2 KiB
Lua
133 lines
3.2 KiB
Lua
--[[ $Id: StanceBar.lua 61678 2008-02-17 01:37:33Z nevcairiel $ ]]
|
|
|
|
-- register module
|
|
local PetBarMod = Bartender4:NewModule("PetBar")
|
|
|
|
-- fetch upvalues
|
|
local ActionBars = Bartender4:GetModule("ActionBars")
|
|
local ButtonBar = Bartender4.ButtonBar.prototype
|
|
|
|
-- create prototype information
|
|
local PetBar = setmetatable({}, {__index = ButtonBar})
|
|
local PetButtonPrototype = CreateFrame("CheckButton")
|
|
local PetButton_MT = {__index = PetButtonPrototype}
|
|
|
|
local defaults = { profile = Bartender4:Merge({
|
|
enabled = true,
|
|
scale = 1.5,
|
|
}, Bartender4.ButtonBar.defaults) }
|
|
|
|
function PetBarMod:OnInitialize()
|
|
self.db = Bartender4.db:RegisterNamespace("PetBar", defaults)
|
|
|
|
self:SetupOptions()
|
|
end
|
|
|
|
function PetBarMod:OnEnable()
|
|
if not self.bar then
|
|
self.bar = setmetatable(Bartender4.ButtonBar:Create("Pet", nil, self.db.profile), {__index = PetBar})
|
|
|
|
local buttons = {}
|
|
for i=1,10 do
|
|
buttons[i] = self:CreatePetButton(i)
|
|
end
|
|
self.bar.buttons = buttons
|
|
|
|
self.bar:ClearSetPoint("CENTER")
|
|
self.bar:ApplyConfig()
|
|
self.bar:SetScript("OnEvent", PetBar.OnEvent)
|
|
self.bar:SetAttribute("unit", "pet")
|
|
RegisterUnitWatch(self.bar, false)
|
|
end
|
|
end
|
|
|
|
function PetBarMod:OnDisable()
|
|
if not self.bar then return end
|
|
self.bar:UnregisterAllEvents()
|
|
self.bar:Hide()
|
|
end
|
|
|
|
function PetBarMod:CreatePetButton(id)
|
|
local name = "BT4PetButton" .. id
|
|
local button = setmetatable(CreateFrame("CheckButton", name, self.bar, "Bartender4PetButtonTemplate"), PetButton_MT)
|
|
button:SetAttribute("type", "pet")
|
|
button:SetAttribute("action", id)
|
|
button.id = id
|
|
button.flash = _G[name .. "Flash"]
|
|
button.cooldown = _G[name .. "Cooldown"]
|
|
button.icon = _G[name .. "Icon"]
|
|
button.autocastable = _G[name .. "AutoCastable"]
|
|
button.autocast = _G[name .. "AutoCast"]
|
|
|
|
return button
|
|
end
|
|
|
|
function PetBarMod:SetupOptions()
|
|
self.options = Bartender4.ButtonBar.prototype:GetOptionObject()
|
|
|
|
ActionBars.options.args["Pet"] = {
|
|
order = 30,
|
|
type = "group",
|
|
name = "Pet Bar",
|
|
desc = "Configure the Pet Bar",
|
|
childGroups = "tab",
|
|
}
|
|
ActionBars.options.args["Pet"].args = self.options.table
|
|
end
|
|
|
|
function PetBarMod:ApplyConfig()
|
|
self.bar:ApplyConfig(self.db.profile)
|
|
end
|
|
|
|
function PetButtonPrototype:Update()
|
|
local name, subtext, texture, isToken, isActive, autoCastAllowed, autoCastEnabled = GetPetActionInfo(self.id)
|
|
|
|
if not isToken then
|
|
self.icon:SetTexture(texture)
|
|
self.tooltipName = name;
|
|
else
|
|
self.icon:SetTexture(_G[texture])
|
|
self.tooltipName = _G[name]
|
|
end
|
|
|
|
self.isToken = isToken;
|
|
self.tooltipSubtext = subtext;
|
|
self:SetChecked(isActive and 1 or 0)
|
|
if autoCastAllowed then
|
|
self.autocastable:Show()
|
|
else
|
|
self.autocastable:Hide()
|
|
end
|
|
if autoCastEnabled then
|
|
self.autocast:Show()
|
|
else
|
|
self.autocast:Hide()
|
|
end
|
|
|
|
if texture then
|
|
if GetPetActionsUsable() then
|
|
SetDesaturation(self.icon, nil)
|
|
else
|
|
SetDesaturation(self.icon, 1)
|
|
end
|
|
self.icon:Show()
|
|
self:SetNormalTexture("Interface\\Buttons\\UI-Quickslot2")
|
|
else
|
|
self.icon:Hide()
|
|
self:SetNormalTexture("Interface\\Buttons\\UI-Quickslot")
|
|
end
|
|
end
|
|
|
|
function PetButtonPrototype:ClearSetPoint(...)
|
|
self:ClearAllPoints()
|
|
self:SetPoint(...)
|
|
end
|
|
|
|
|
|
function PetBar:ApplyConfig()
|
|
ButtonBar.ApplyConfig(self)
|
|
self:UpdateButtonLayout()
|
|
self:ForAll("Update")
|
|
self:ForAll("ApplyStyle", self.config.style)
|
|
end
|