Adding module for monitoring spellcasts in PvP
This commit is contained in:
pigmonkey
parent
cd8f501570
commit
0177b5934b
|
|
@ -27,5 +27,6 @@ modules\potions.lua
|
|||
modules\prat.lua
|
||||
modules\raid.lua
|
||||
modules\rbs.lua
|
||||
modules\spellcast.lua
|
||||
modules\staging.lua
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,327 @@
|
|||
local core = LibStub("AceAddon-3.0"):GetAddon("AllTheLittleThings")
|
||||
local mod = core:NewModule("Spellcast", "AceEvent-3.0", "AceTimer-3.0")
|
||||
local db
|
||||
|
||||
local defaults = {
|
||||
anchor = {
|
||||
x = 500,
|
||||
y = 500,
|
||||
locked = false,
|
||||
},
|
||||
|
||||
buttonSize = 40,
|
||||
textSize = 10,
|
||||
spacing = 5,
|
||||
}
|
||||
local options = {
|
||||
locked = {
|
||||
name = "Locked",
|
||||
type = 'toggle',
|
||||
order = 1,
|
||||
get = function() return db.anchor.locked end,
|
||||
set = function(_, val) db.anchor.locked = val mod:SetLocked(val) end,
|
||||
},
|
||||
}
|
||||
|
||||
local spellWatch = {
|
||||
-- Polymorph
|
||||
[118] = "Polymorph",
|
||||
[61305] = "Polymorph",
|
||||
[28272] = "Polymorph",
|
||||
[61721] = "Polymorph",
|
||||
[61780] = "Polymorph",
|
||||
[28271] = "Polymorph",
|
||||
-- Counterspell
|
||||
[2139] = "Counterspell",
|
||||
-- Deep Freeze
|
||||
[44572] = "Deep Freeze",
|
||||
-- Frostbolt
|
||||
[116] = "Frostbolt",
|
||||
}
|
||||
local auras = {
|
||||
["Polymorph"] = true,
|
||||
}
|
||||
local durations = {
|
||||
["Polymorph"] = 8,
|
||||
["Deep Freeze"] = 30,
|
||||
["Counterspell"] = 24,
|
||||
["Ice Lance"] = 10,
|
||||
}
|
||||
local validZones = {
|
||||
-- Battlegrounds
|
||||
["Alterac Valley"] = true,
|
||||
["Arathi Basin"] = true,
|
||||
["Eye of the Storm"] = true,
|
||||
["Isle of Conquest"] = true,
|
||||
["Strand of the Ancients"] = true,
|
||||
["The Battle for Gilneas"] = true,
|
||||
["Twin Peaks"] = true,
|
||||
["Warsong Gulch"] = true,
|
||||
|
||||
-- Arenas
|
||||
["Blade's Edge Arena"] = true,
|
||||
["Dalaran Arena"] = true,
|
||||
["Nagrand Arena"] = true,
|
||||
["Ruins of Lordaeron"] = true,
|
||||
["The Ring of Valor"] = true,
|
||||
|
||||
-- Testing
|
||||
-- ["Stormwind City"] = true,
|
||||
}
|
||||
local combatEvents = {
|
||||
["SPELL_AURA_APPLIED"] = true,
|
||||
["SPELL_AURA_REFRESHED"] = true,
|
||||
["SPELL_AURA_REMOVED"] = true,
|
||||
}
|
||||
local icons = {}
|
||||
local lastAvailable = {}
|
||||
local anchor, enabled, updateTimer
|
||||
|
||||
function mod:OnInitialize()
|
||||
self:RegisterOptions(options, defaults, function(d) db = d end)
|
||||
end
|
||||
|
||||
function mod:OnEnable()
|
||||
self:SetupIcons()
|
||||
self:RegisterEvent("ZONE_CHANGED_NEW_AREA")
|
||||
self:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
|
||||
self:RegisterEvent("UNIT_SPELLCAST_SUCCEEDED")
|
||||
|
||||
-- This doesn't always fire on load
|
||||
self:ZONE_CHANGED_NEW_AREA()
|
||||
end
|
||||
|
||||
function mod:SetLocked(val)
|
||||
if val then
|
||||
anchor:Hide()
|
||||
else
|
||||
anchor:Show()
|
||||
end
|
||||
end
|
||||
|
||||
function mod:ZONE_CHANGED_NEW_AREA()
|
||||
local before = enabled
|
||||
enabled = not not validZones[GetRealZoneText() or ""]
|
||||
|
||||
if before ~= enabled then
|
||||
-- wipe last table so that we don't have an accumulated time
|
||||
wipe(lastAvailable)
|
||||
self:UpdateDisplay()
|
||||
end
|
||||
|
||||
if not enabled then
|
||||
self:CancelTimer(updateTimer, true)
|
||||
updateTimer = nil
|
||||
end
|
||||
|
||||
if enabled and not updateTimer then
|
||||
updateTimer = self:ScheduleRepeatingTimer("UpdateDisplay", 0.1)
|
||||
end
|
||||
end
|
||||
|
||||
local pid = UnitGUID("player")
|
||||
function mod:COMBAT_LOG_EVENT_UNFILTERED(_, _, event, _, ...)
|
||||
if not enabled or not combatEvents[event] then return end
|
||||
|
||||
local srcGUID, dstGUID, spellId, _
|
||||
if select(4, GetBuildInfo()) > 40100 then
|
||||
srcGUID, _, _, _, dstGUID, _, _, _, spellId = ...
|
||||
else
|
||||
srcGUID, _, _, dstGUID, _, _, spellId = ...
|
||||
end
|
||||
|
||||
-- source is player
|
||||
if pid ~= srcGUID then return end
|
||||
|
||||
local category = spellWatch[spellId]
|
||||
if category then
|
||||
-- only handling auras here
|
||||
if auras[category] then
|
||||
if event == "SPELL_AURA_APPLIED" then
|
||||
local endTime = select(7, UnitDebuff(self:FindGUID(dstGUID), GetSpellInfo(spellId))) or (durations[category] or 0)+GetTime()
|
||||
self:SpellStart(category, endTime)
|
||||
elseif event == "SPELL_AURA_REMOVED" then
|
||||
self:SpellStop(category)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function mod:UNIT_SPELLCAST_SUCCEEDED(_, unit, _, _, _, spellId)
|
||||
if not enabled or UnitGUID(unit) ~= pid then return end
|
||||
|
||||
local category = spellWatch[spellId]
|
||||
if category then
|
||||
if not auras[category] then
|
||||
local duration = durations[category]
|
||||
if not duration then
|
||||
-- no duration means it has no cooldown, so we just start the timer
|
||||
self:SpellStop(category, true)
|
||||
else
|
||||
self:SpellStart(category, duration+GetTime())
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function mod:SpellStart(category, endTime)
|
||||
local icon = icons[category]
|
||||
local time = GetTime()
|
||||
if not icon or endTime<time then return end
|
||||
|
||||
lastAvailable[category] = nil
|
||||
local duration = endTime - time
|
||||
icon.cooldown = duration
|
||||
icon:SetCooldown(time, duration)
|
||||
icon:SetText("")
|
||||
end
|
||||
|
||||
function mod:SpellStop(category, nocd)
|
||||
local icon = icons[category]
|
||||
if not icon then return end
|
||||
|
||||
lastAvailable[category] = GetTime()
|
||||
icon.cooldown = 0
|
||||
if not nocd then
|
||||
icon:SetCooldown(0, 0)
|
||||
end
|
||||
icon:SetText("0:00")
|
||||
end
|
||||
|
||||
function mod:FindGUID(guid)
|
||||
if UnitGUID("target") == guid then
|
||||
return "target"
|
||||
elseif UnitGUID("focus") == guid then
|
||||
return "focus"
|
||||
else
|
||||
for i=1,5 do
|
||||
if UnitGUID("arena"..i) == guid then
|
||||
return "arena"..i
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- return a valid string so that UnitAura doesn't error
|
||||
return ""
|
||||
end
|
||||
|
||||
function mod:SetupIcons()
|
||||
-- return if we've already set up
|
||||
if anchor then return end
|
||||
|
||||
anchor = self:SetupAnchor()
|
||||
local last = anchor
|
||||
|
||||
-- each icon chains to the last
|
||||
for spellId, category in pairs(spellWatch) do
|
||||
if not icons[category] then
|
||||
last = self:SetupIcon(spellId, last)
|
||||
icons[category] = last
|
||||
end
|
||||
end
|
||||
|
||||
self:UpdateDisplay()
|
||||
end
|
||||
|
||||
function mod:SetupAnchor()
|
||||
local frame = CreateFrame("Frame", nil, UIParent)
|
||||
frame:SetFrameStrata("LOW")
|
||||
frame:SetSize(db.buttonSize, db.buttonSize)
|
||||
frame:SetMovable(true)
|
||||
frame:SetClampedToScreen(true)
|
||||
frame:RegisterForDrag("LeftButton")
|
||||
frame:EnableMouse(true)
|
||||
frame:SetScript("OnDragStart", function(self)
|
||||
if not db.anchor.locked then
|
||||
self:StartMoving()
|
||||
end
|
||||
end)
|
||||
frame:SetScript("OnDragStop", function(self)
|
||||
local scale = self:GetEffectiveScale()
|
||||
self:StopMovingOrSizing()
|
||||
db.anchor.x = self:GetLeft()*scale
|
||||
db.anchor.y = self:GetBottom()*scale
|
||||
end)
|
||||
|
||||
local texture = frame:CreateTexture()
|
||||
texture:SetAllPoints()
|
||||
texture:SetTexture(0, 0, 0, 0.6)
|
||||
|
||||
local scale = frame:GetEffectiveScale()
|
||||
frame:SetPoint("BOTTOMLEFT", UIParent, "BOTTOMLEFT", db.anchor.x/scale, db.anchor.y/scale)
|
||||
|
||||
if db.anchor.locked then
|
||||
frame:Hide()
|
||||
end
|
||||
|
||||
return frame
|
||||
end
|
||||
|
||||
function mod:SetupIcon(spellId, anchor)
|
||||
local frame = CreateFrame("Frame", nil, UIParent)
|
||||
frame:SetSize(db.buttonSize, db.buttonSize)
|
||||
frame:SetPoint("TOPLEFT", anchor, "BOTTOMLEFT", 0, -db.spacing)
|
||||
|
||||
local texture = frame:CreateTexture()
|
||||
texture:SetAllPoints()
|
||||
texture:SetTexture(select(3, GetSpellInfo(spellId)))
|
||||
|
||||
local clock = CreateFrame("Cooldown", nil, frame)
|
||||
clock:SetAllPoints()
|
||||
|
||||
local text = frame:CreateFontString(nil, nil, "GameFontNormal")
|
||||
text:SetSize(100, 20)
|
||||
-- text:SetText("0:00")
|
||||
text:SetPoint("BOTTOMLEFT", frame, "BOTTOMRIGHT", 5, 0)
|
||||
text:SetJustifyH("LEFT")
|
||||
|
||||
-- methods and members
|
||||
frame.cooldown = 0
|
||||
frame.spell = spellId
|
||||
frame.SetText = function(self, ...)
|
||||
text:SetText(...)
|
||||
end
|
||||
frame.SetFormattedText = function(self, ...)
|
||||
text:SetFormattedText(...)
|
||||
end
|
||||
frame.SetCooldown = function(self, ...)
|
||||
clock:SetCooldown(...)
|
||||
end
|
||||
|
||||
return frame
|
||||
end
|
||||
|
||||
function mod:UpdateDisplay()
|
||||
if not enabled then
|
||||
anchor:Hide()
|
||||
for _,frame in pairs(icons) do
|
||||
frame:Hide()
|
||||
end
|
||||
else
|
||||
if not db.anchor.locked then
|
||||
anchor:Show()
|
||||
end
|
||||
for category, frame in pairs(icons) do
|
||||
if not frame:IsVisible() then
|
||||
frame:Show()
|
||||
frame:SetText("")
|
||||
end
|
||||
|
||||
-- update text
|
||||
if lastAvailable[category] then
|
||||
local diff = GetTime()-lastAvailable[category]
|
||||
frame:SetFormattedText("%d:%02d", diff/60, diff%60)
|
||||
end
|
||||
|
||||
-- if a cooldown, check that it's still on cd
|
||||
if not auras[category] and frame.cooldown > 0 then
|
||||
local _, duration = GetSpellCooldown(frame.spell)
|
||||
-- within a global, we must not have it on cd
|
||||
if duration < 1.5 then
|
||||
self:SpellStop(category)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
Loading…
Reference in New Issue