Adding module for monitoring spellcasts in PvP

AllTheLittleThings.toc

@@ -27,5 +27,6 @@ modules\potions.lua
 modules\prat.lua
 modules\raid.lua
 modules\rbs.lua
+modules\spellcast.lua
 modules\staging.lua
 

modules/spellcast.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
+