local ldb, ae = LibStub:GetLibrary("LibDataBroker-1.1"), LibStub("AceEvent-3.0")
Cork = {db = {}, dbpc = {}, petmappings = {player = "pet"}, defaultspc = {castonpets = false, multithreshold = 2, tooltiplimit = 10, raid_thresh = 5}, corks = {}, petunits = {pet = true}, keyblist = {CorkIt = true, type = true, Scan = true, Init = true, configframe = true, RaidLine = true}}
local corks = Cork.corks
local defaults = {point = "TOP", x = 0, y = -100, showanchor = true, showunit = false, bindwheel = false}
local tooltip, anchor
for i=1,4 do Cork.petmappings["party"..i], Cork.petunits["partypet"..i] = "partypet"..i, true end
for i=1,40 do Cork.petmappings["raid"..i], Cork.petunits["raidpet"..i] = "raidpet"..i, true end
----------------------------
-- Localization --
----------------------------
Cork.classnames = {
["WARLOCK"] = "Warlock",
["WARRIOR"] = "Warrior",
["HUNTER"] = "Hunter",
["MAGE"] = "Mage",
["PRIEST"] = "Priest",
["DRUID"] = "Druid",
["PALADIN"] = "Paladin",
["SHAMAN"] = "Shaman",
["ROGUE"] = "Rogue",
["DEATHKNIGHT"] = "Death Knight",
}
Cork.colors = {}
for token in pairs(Cork.classnames) do
local c = RAID_CLASS_COLORS[token]
Cork.colors[token] = string.format("%02x%02x%02x", c.r*255, c.g*255, c.b*255)
end
------------------------------
-- Initialization --
------------------------------
local meta = {__index = Cork.defaultspc}
ae.RegisterEvent("Cork", "ADDON_LOADED", function(event, addon)
if addon:lower() ~= "cork" then return end
CorkDB = setmetatable(CorkDB or {}, {__index = defaults})
CorkDBPC = CorkDBPC or {{},{}}
if not CorkDBPC[1] then CorkDBPC = {CorkDBPC, {}} end
Cork.db = CorkDB
anchor:SetPoint(Cork.db.point, Cork.db.x, Cork.db.y)
if not Cork.db.showanchor then anchor:Hide() end
LibStub("tekKonfig-AboutPanel").new("Cork", "Cork")
ae.UnregisterEvent("Cork", "ADDON_LOADED")
end)
ae.RegisterEvent("Cork", "PLAYER_LOGIN", function()
local lastgroup = GetActiveTalentGroup()
Cork.dbpc = setmetatable(CorkDBPC[lastgroup], meta)
for name,dataobj in pairs(corks) do if dataobj.Init then dataobj:Init() end end
for name,dataobj in pairs(corks) do dataobj:Scan() end
ae.RegisterEvent("Cork", "ZONE_CHANGED_NEW_AREA", Cork.Update)
ae.RegisterEvent("Cork", "PLAYER_TALENT_UPDATE", function()
if lastgroup == GetActiveTalentGroup() then return end
lastgroup = GetActiveTalentGroup()
for i,v in pairs(Cork.defaultspc) do if Cork.dbpc[i] == v then Cork.dbpc[i] = nil end end
Cork.dbpc = setmetatable(CorkDBPC[lastgroup], meta)
for name,dataobj in pairs(corks) do if dataobj.Init then dataobj:Init() end end
for name,dataobj in pairs(corks) do dataobj:Scan() end
end)
ae.UnregisterEvent("Cork", "PLAYER_LOGIN")
end)
ae.RegisterEvent("Cork", "PLAYER_LOGOUT", function()
for i,v in pairs(defaults) do if Cork.db[i] == v then Cork.db[i] = nil end end
for i,v in pairs(Cork.defaultspc) do if Cork.dbpc[i] == v then Cork.dbpc[i] = nil end end
end)
local onTaxi = nil
ae.RegisterEvent("Cork BeginTaxi", "PLAYER_CONTROL_LOST", function()
onTaxi = true
Cork.Update()
end)
ae.RegisterEvent("Cork EndTaxi", "PLAYER_CONTROL_GAINED", function()
onTaxi = nil
Cork.Update()
end)
------------------------------
-- Tooltip anchor --
------------------------------
anchor = CreateFrame("Button", nil, UIParent)
anchor:SetHeight(24)
Cork.anchor = anchor
anchor:SetBackdrop({bgFile = "Interface\\Tooltips\\UI-Tooltip-Background", edgeFile = "Interface\\Tooltips\\UI-Tooltip-Border", edgeSize = 16, insets = {left = 5, right = 5, top = 5, bottom = 5}, tile = true, tileSize = 16})
anchor:SetBackdropColor(0.09, 0.09, 0.19, 0.5)
anchor:SetBackdropBorderColor(0.5, 0.5, 0.5, 1)
local text = anchor:CreateFontString(nil, nil, "GameFontNormalSmall")
text:SetPoint("CENTER")
text:SetText("Cork")
anchor:SetWidth(text:GetStringWidth() + 8)
anchor:SetMovable(true)
anchor:RegisterForDrag("LeftButton")
anchor:SetScript("OnClick", function(self) InterfaceOptionsFrame_OpenToCategory(Cork.config) end)
anchor:SetScript("OnDragStart", function(self)
tooltip:Hide()
self:StartMoving()
end)
anchor:SetScript("OnDragStop", function(self)
self:StopMovingOrSizing()
Cork.db.point, Cork.db.x, Cork.db.y = "BOTTOMLEFT", self:GetLeft(), self:GetBottom()
Cork.Update()
end)
-----------------------
-- Tooltip --
-----------------------
tooltip = CreateFrame("GameTooltip", "Corkboard", UIParent, "GameTooltipTemplate")
tooltip:SetFrameStrata("MEDIUM")
CorkboardTextLeft1:SetFontObject(GameTooltipText)
CorkboardTextRight1:SetFontObject(GameTooltipText)
local function GetTipAnchor(frame)
local x,y = frame:GetCenter()
if not x or not y then return "TOPLEFT", frame, "BOTTOMLEFT" end
local hhalf = (x > UIParent:GetWidth()*2/3) and "RIGHT" or (x < UIParent:GetWidth()/3) and "LEFT" or ""
local vhalf = (y > UIParent:GetHeight()/2) and "TOP" or "BOTTOM"
return vhalf..hhalf, frame, (vhalf == "TOP" and "BOTTOM" or "TOP")..hhalf
end
local raidunits = {player = true}
for i=1,4 do raidunits["party"..i] = true end
for i=1,40 do raidunits["raid"..i] = true end
function Cork.Update(event, name, attr, value, dataobj)
if Cork.keyblist[attr] then return end
tooltip:ClearLines()
tooltip:SetOwner(anchor, "ANCHOR_NONE")
tooltip:SetPoint(GetTipAnchor(anchor))
if Cork.db.showbg or (GetZoneText() ~= "Wintergrasp" and select(2, IsInInstance()) ~= "pvp") then
local count = 0
for name,dataobj in pairs(corks) do
local inneed, numr = 0, GetNumRaidMembers()
for i=1,numr do if dataobj.RaidLine and dataobj["raid"..i] then inneed = inneed + 1 end end
if dataobj.RaidLine and numr > 0 and dataobj["player"] then inneed = inneed + 1 end
if inneed > 0 and count < Cork.dbpc.tooltiplimit then
if Cork.db.showunit then tooltip:AddDoubleLine(string.format(dataobj.RaidLine, inneed), "raid") else tooltip:AddLine(string.format(dataobj.RaidLine, inneed)) end
count = count + 1
end
for i,v in ldb:pairs(dataobj) do
if not Cork.keyblist[i] and (inneed == 0 or not raidunits[i]) and count < Cork.dbpc.tooltiplimit then
if Cork.db.showunit then tooltip:AddDoubleLine(v, i) else tooltip:AddLine(v) end
count = count + 1
end
end
end
end
if tooltip:NumLines() > 0 and not onTaxi then tooltip:Show() end
end
-------------------------
-- LDB stuff --
-------------------------
local function NewDataobject(event, name, dataobj)
if dataobj.type ~= "cork" then return end
corks[name] = dataobj
ldb.RegisterCallback("Corker", "LibDataBroker_AttributeChanged_"..name, Cork.Update)
end
ldb.RegisterCallback("Corker", "LibDataBroker_DataObjectCreated", NewDataobject)
----------------------------
-- Secure frame --
----------------------------
local secureframe = CreateFrame("Button", "CorkFrame", UIParent, "SecureActionButtonTemplate")
secureframe.SetManyAttributes = function(self, ...)
for i=1,select("#", ...),2 do
local att,val = select(i, ...)
if not att then return end
self:SetAttribute(att,val)
end
return true
end
secureframe:SetScript("PreClick", function(self)
if InCombatLockdown() or IsStealthed() then return end
for name,dataobj in pairs(corks) do if dataobj.CorkIt and dataobj.player and dataobj:CorkIt(self) then return end end
for name,dataobj in pairs(corks) do if dataobj.CorkIt and not dataobj.player and dataobj:CorkIt(self, true) then return end end
for name,dataobj in pairs(corks) do if dataobj.CorkIt and not dataobj.player and dataobj:CorkIt(self) then return end end
end)
secureframe:SetScript("PostClick", function()
if InCombatLockdown() then return end
secureframe:SetManyAttributes("type1", ATTRIBUTE_NOOP, "bag1", nil, "slot1", nil, "item1", nil, "spell", nil, "unit", nil, "macrotext1", nil)
end)
--------------------------------
-- Shared functions --
--------------------------------
function Cork.IsSpellInRange(spell, unit)
return IsSpellInRange(spell, unit) == 1
end
function Cork.SpellCastableOnUnit(spell, unit)
if Cork.keyblist[i] then return end
return UnitExists(unit) and UnitCanAssist("player", unit) and not UnitIsDeadOrGhost(unit) and Cork.IsSpellInRange(spell, unit)
end
function Cork.IconLine(icon, text, token)
return "|T"..icon..":24|t ".. (token and ("|cff".. Cork.colors[token]) or "").. text
end
) is private.
