--[[ $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