InterruptBar.lua

----------------------------------------------------
-- Interrupt Bar by Kollektiv
----------------------------------------------------

InterruptBarDB = InterruptBarDB or { scale = 1, reuse = true, hidden = false, lock = false, }
local abilities = {}
local order
local band = bit.band
local spellids = {[19503] = 30, [61490] = 25, [100] = 20, [6552] = 10, [8643] = 20, [2139] = 24, [19647] = 24, [23920] = 10, [72] = 12, [15487] = 45, [8177] = 13, [57994] = 5, [1766] = 10, [31224] = 60, [10890] = 23, [10308] = 40, [30283] = 20, [36554] = 20, [47528] = 10, [2094] = 120, [47860] = 120, [18708] = 180, [48020] = 30, [47585] = 75, [64044] = 120, [33206] = 144, [44572] = 30, [49203] = 60, [49206] = 180, [47476] = 120, [49576] = 25, [34600] = 28, [34490] = 20, [14311] = 28, [60192] = 28, [53271] = 60, [53480] = 60, [53548] = 40, [23989] = 180, [22812] = 60, [50334] = 180, [1022] = 180, [20066] = 60, [31884] = 120, [51514] = 45, [59159] = 35}
for spellid,time in pairs(spellids) do
	local name,_,spellicon = GetSpellInfo(spellid)	
	abilities[name] = { icon = spellicon, duration = time }
end

-----------------------------------------------------
-- Edit this table to change the order
-----------------------------------------------------
-- 19503 Scatter Shot
-- 100 Charge
-- 61490 Intercept
-- 6552 Pummel
-- 8643 Kidney Shot
-- 2139 Counterspell
-- 19647 Spell Lock
-- 23920 Spell Reflect
-- 72 Shield Bash
-- 15487 Silence
-- 8177 Grounding Totem
-- 57994 Wind Snear
-- 1766 Kick
-- 31224 Cloak of Shadow
-- 10890 Physchic Scream
-- 10308 Hammer of Justice
-- 47528 Mind Freeze
-- 30283 Shadowfury
-- 36554 Shadowstep
-- 2094 Blind
-- 47860 Death Coil
-- 18708 Fel Domination
-- 48020 Demonic Circle: Teleport
-- 47585 Dispersion
-- 64044 Psychic Horror
-- 33206 Pain Suppression
-- 44572 Deep Freeze
-- 49203 Hungering Cold
-- 49206 Summon Gargoyle
-- 47476 Strangulate
-- 49576 Death Grip
-- 34600 Snake Trap
-- 34490 Silencing Shot
-- 14311 Freezing Trap
-- 60192 Freezing Arrow
-- 53271 Master's Call
-- 53480 Roar of Sacrifice
-- 53548 Pin
-- 23989 Readiness
-- 22812 Barkskin
-- 50334 Berserk
-- 1022 Hand of Protection
-- 20066 Repentance
-- 31884 Avenging Wrath
-- 51514 Hex
-- 59159 Thunderstorm
-----------------------------------------------------

local order = {100, 61490, 6552, 72, 1766, 47528, 2139, 19647, 15487, 30283, 57994, 8177, 23920, 8643, 10308, 10890, 31224, 19503, 36554, 2094, 47860, 18708, 48020, 47585, 64044, 33206, 44572, 49203, 49206, 47476, 49576, 34600, 34490, 14311, 60192, 53271, 53480, 53548, 23989, 22812, 50334, 1022, 20066, 31884, 51514, 59159}

-----------------------------------------------------
-----------------------------------------------------

for k,v in ipairs(order) do order[k] = GetSpellInfo(v) end

local frame
local bar

local GetTime = GetTime
local ipairs = ipairs
local pairs = pairs
local select = select
local floor = floor
local band = bit.band
local GetSpellInfo = GetSpellInfo

local GROUP_UNITS = bit.bor(0x00000010, 0x00000400)

local activetimers = {}

local size = 0
local function getsize()
	size = 0
	for k in pairs(activetimers) do
		size = size + 1
	end
end

local function InterruptBar_UpdateIconPos()
	local x = -45
	local y = 0

	local i = 1
	for curAbility,ability in ipairs(order) do
		local btn = bar[ability]
		if activetimers[ability] then
			btn:SetPoint("CENTER",bar,"CENTER",x,y)
			btn:Show()
			--
			--	Originally i < AbilityCount / 2
			--	Which was counted early on when adding 
			--  	The abilities to a table
			--	
			--  	This method ensures we only use the number of
			--	Abilities that are going to be displayed
			--  	Given in the order table.
			--
			if i < #(order) / 2 then
				x = x + 30
			else
				if i > #(order) / 2 then
					x = x + 30
				else
					y = y - 30
					x = -45
				end
			end

			-- V: maintain a different counter than curAbility
			i = i + 1
		else
			btn:Hide()
		end
	end
end

local function InterruptBar_AddIcons()
	--local curAbility = 0
	for curAbility,ability in ipairs(order) do
		local btn = CreateFrame("Frame",nil,bar)
		btn:SetWidth(30)
		btn:SetHeight(30)
		btn:SetFrameStrata("LOW")
		
		local cd = CreateFrame("Cooldown",nil,btn)
		cd.noomnicc = true
		cd.noCooldownCount = true
		cd:SetAllPoints(true)
		cd:SetFrameStrata("MEDIUM")
		cd:Hide()
		
		local texture = btn:CreateTexture(nil,"BACKGROUND")
		texture:SetAllPoints(true)
		texture:SetTexture(abilities[ability].icon)
		texture:SetTexCoord(0.07,0.9,0.07,0.90)
	
		local text = cd:CreateFontString(nil,"ARTWORK")
		text:SetFont(STANDARD_TEXT_FONT,18,"OUTLINE")
		text:SetTextColor(1,1,0,1)
		text:SetPoint("LEFT",btn,"LEFT",2,0)
		
		btn.texture = texture
		btn.text = text
		btn.duration = abilities[ability].duration
		btn.cd = cd
		
		bar[ability] = btn
	end
end

local function InterruptBar_SavePosition()
	local point, _, relativePoint, xOfs, yOfs = bar:GetPoint()
	if not InterruptBarDB.Position then 
		InterruptBarDB.Position = {}
	end
	InterruptBarDB.Position.point = point
	InterruptBarDB.Position.relativePoint = relativePoint
	InterruptBarDB.Position.xOfs = xOfs
	InterruptBarDB.Position.yOfs = yOfs
end

local function InterruptBar_LoadPosition()
	if InterruptBarDB.Position then
		bar:SetPoint(InterruptBarDB.Position.point,UIParent,InterruptBarDB.Position.relativePoint,InterruptBarDB.Position.xOfs,InterruptBarDB.Position.yOfs)
	else
		bar:SetPoint("CENTER", UIParent, "CENTER")
	end
end

local function InterruptBar_UpdateBar()
	bar:SetScale(InterruptBarDB.scale)
	if InterruptBarDB.hidden or InterruptBarDB.reuse then
		for _,v in ipairs(order) do bar[v]:Hide() end
	else
		for _,v in ipairs(order) do bar[v]:Show() end
	end
	if InterruptBarDB.lock then
		bar:EnableMouse(false)
	else
		bar:EnableMouse(true)
	end
end

local function InterruptBar_CreateBar()
	bar = CreateFrame("Frame", nil, UIParent)
	bar:SetMovable(true)
	bar:SetWidth(120)
	bar:SetHeight(30)
	bar:SetClampedToScreen(true) 
	bar:SetScript("OnMouseDown",function(self,button) if button == "LeftButton" then self:StartMoving() end end)
	bar:SetScript("OnMouseUp",function(self,button) if button == "LeftButton" then self:StopMovingOrSizing() InterruptBar_SavePosition() end end)
	bar:Show()
	
	InterruptBar_AddIcons()
	InterruptBar_UpdateBar()
	InterruptBar_LoadPosition()
end

local function InterruptBar_UpdateText(text,cooldown)
	if cooldown < 10 then 
		if cooldown <= 0.5 then
			text:SetText("")
		else
			text:SetFormattedText(" %d",cooldown)
		end
	else
		text:SetFormattedText("%d",cooldown)
	end
	if cooldown < 6 then 
		text:SetTextColor(1,0,0,1)
	else 
		text:SetTextColor(1,1,0,1) 
	end
end

local function InterruptBar_StopAbility(ref,ability)
	if activetimers[ability] then activetimers[ability] = nil end

	-- V: only update icon pos after the timer was cleared
	if InterruptBarDB.reuse then
		InterruptBar_UpdateIconPos()
	elseif InterruptBarDB.hidden then
		ref:Hide()
	end

	ref.text:SetText("")
	ref.cd:Hide()
end

local time = 0
local function InterruptBar_OnUpdate(self, elapsed)
	time = time + elapsed
	if time > 0.25 then
		getsize()
		for ability,ref in pairs(activetimers) do
			ref.cooldown = ref.start + ref.duration - GetTime()
			if ref.cooldown <= 0 then
				InterruptBar_StopAbility(ref,ability)
			else 
				InterruptBar_UpdateText(ref.text,floor(ref.cooldown+0.5))
			end
		end
		if size == 0 then frame:SetScript("OnUpdate",nil) end
		time = time - 0.25
	end
end

local function InterruptBar_StartTimer(ref,ability)
	if not activetimers[ability] then
		local duration
		activetimers[ability] = ref
		ref.cd:Show()
		ref.cd:SetCooldown(GetTime()-0.40,ref.duration)
		ref.start = GetTime()
		InterruptBar_UpdateText(ref.text,ref.duration)
	end
	frame:SetScript("OnUpdate",InterruptBar_OnUpdate)

	if InterruptBarDB.reuse then
		InterruptBar_UpdateIconPos()
	elseif InterruptBarDB.hidden then
		ref:Show()
	end

end

local function InterruptBar_COMBAT_LOG_EVENT_UNFILTERED(...)
	local spellID, ability, useSecondDuration
	return function(_, eventtype, _, srcName, srcFlags, _, dstName, dstFlags, id)
		if (band(srcFlags, 0x00000040) == 0x00000040 and eventtype == "SPELL_CAST_SUCCESS") then 
			spellID = id
		else
			return
		end
		useSecondDuration = false
		if spellID == 49376 then spellID = 16979; useSecondDuration = true end -- Feral Charge - Cat -> Feral Charge - Bear
		ability = GetSpellInfo(spellID)
		if abilities[ability] then			
			if useSecondDuration and spellID == 16979 then
				bar[ability].duration = 30
			elseif spellID == 16979 then
				bar[ability].duration = 15
			end
			InterruptBar_StartTimer(bar[ability],ability)
		end
	end
end

InterruptBar_COMBAT_LOG_EVENT_UNFILTERED = InterruptBar_COMBAT_LOG_EVENT_UNFILTERED()

local function InterruptBar_ResetAllTimers()
	for _,ability in ipairs(order) do
		InterruptBar_StopAbility(bar[ability])
	end
	active = 0
end

local function InterruptBar_PLAYER_ENTERING_WORLD(self)
	InterruptBar_ResetAllTimers()
end

local function InterruptBar_Reset()
	InterruptBarDB = { scale = 1, hidden = false, reuse = true, lock = false }
	InterruptBar_UpdateBar()
	InterruptBar_LoadPosition()
end

local function InterruptBar_Test()
	for _,ability in ipairs(order) do
		InterruptBar_StartTimer(bar[ability],ability)
	end
	if InterruptBarDB.reuse then
		InterruptBar_UpdateIconPos()
	end
end

local cmdfuncs = {
	scale = function(v) InterruptBarDB.scale = v; InterruptBar_UpdateBar() end,
	hidden = function() InterruptBarDB.hidden = not InterruptBarDB.hidden; InterruptBar_UpdateBar() end,
	reuse = function() InterruptBarDB.reuse = not InterruptBarDB.reuse; InterruptBar_UpdateBar() end,
	lock = function() InterruptBarDB.lock = not InterruptBarDB.lock; InterruptBar_UpdateBar() end,
	reset = function() InterruptBar_Reset() end,
	test = function() InterruptBar_Test() end,
}

local cmdtbl = {}
function InterruptBar_Command(cmd)
	for k in ipairs(cmdtbl) do
		cmdtbl[k] = nil
	end
	for v in gmatch(cmd, "[^ ]+") do
  	tinsert(cmdtbl, v)
  end
  local cb = cmdfuncs[cmdtbl[1]] 
  if cb then
  	local s = tonumber(cmdtbl[2])
  	cb(s)
  else
  	ChatFrame1:AddMessage("InterruptBar Options | /ib <option>",0,1,0)  	
  	ChatFrame1:AddMessage("-- scale <number> | value: " .. InterruptBarDB.scale,0,1,0)
  	ChatFrame1:AddMessage("-- hidden (toggle) | value: " .. tostring(InterruptBarDB.hidden),0,1,0)
  	ChatFrame1:AddMessage("-- reuse (toggle) | value: " .. tostring(InterruptBarDB.reuse),0,1,0)
  	ChatFrame1:AddMessage("-- lock (toggle) | value: " .. tostring(InterruptBarDB.lock),0,1,0)
  	ChatFrame1:AddMessage("-- test (execute)",0,1,0)
  	ChatFrame1:AddMessage("-- reset (execute)",0,1,0)
  end
end

local function InterruptBar_OnLoad(self)
	self:RegisterEvent("PLAYER_ENTERING_WORLD")
	self:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
	if not InterruptBarDB.scale then InterruptBarDB.scale = 1 end
	if not InterruptBarDB.hidden then InterruptBarDB.hidden = false end
	if not InterruptBarDB.reuse then InterruptBarDB.reuse = false end
	if not InterruptBarDB.lock then InterruptBarDB.lock = false end
	InterruptBar_CreateBar()
	
	SlashCmdList["InterruptBar"] = InterruptBar_Command
	SLASH_InterruptBar1 = "/ib"
	
--	ChatFrame1:AddMessage("Interrupt Bar by Kollektiv and Vendethiel. Type /ib for options.",0,1,0)
end

local eventhandler = {
	["VARIABLES_LOADED"] = function(self) InterruptBar_OnLoad(self) end,
	["PLAYER_ENTERING_WORLD"] = function(self) InterruptBar_PLAYER_ENTERING_WORLD(self) end,
	["COMBAT_LOG_EVENT_UNFILTERED"] = function(self,...) InterruptBar_COMBAT_LOG_EVENT_UNFILTERED(...) end,
}

local function InterruptBar_OnEvent(self,event,...)
	eventhandler[event](self,...)
end

frame = CreateFrame("Frame",nil,UIParent)
frame:SetScript("OnEvent",InterruptBar_OnEvent)
frame:RegisterEvent("VARIABLES_LOADED")