----------------------------------------------------
-- Interrupt Bar by Kollektiv
-- Modifications by Veliace/Romiculum
----------------------------------------------------

InterruptBarDB = InterruptBarDB or { scale = 1, hidden = false, lock = false, }
local abilities = {}
local order
local band = bit.band

local spellids = {[6552] = 10, [2139] = 24, [19647] = 24, [16979] = 15, [1766] = 10, [47528] = 10, [72] = 12, [18499] = 30, [48020] = 30, [57994] = 6, [15487] = 45, [8643] = 20, [61384] = 20, [49802] = 10, [44572] = 30, [10890] = 27, [34490] = 20, [51514] = 45}
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
-----------------------------------------------------
-- 6552  Pummel
-- 2139  Counterspell
-- 19647 Spell Lock
-- 16979 Feral Charge
-- 1766  Kick
-- 47528 Mind Freeze
-- 72 Shield Bash
-- 18499 Berserker Rage
-- 48020 Demonic Circle: Teleport
-- 57994 Wind Shear
-- 15487 Silence
-- 8643 Kidney shot
-- 61384 Typhoon
-- 49802 Maim
-- 44572 Deep Freeze
-- 10890 Fear
-- 34490 Silence shot
-- 51514 Hex
-----------------------------------------------------

local order = {6552, 2139, 19647, 1766, 47528, 16979, 72, 18499, 48020, 57994, 15487, 8643, 61384, 49802, 44572, 10890, 34490, 51514}

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

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_AddIcons()
   local x = -45
    local y = 0
   local r = 0
   for _,ability in ipairs(order) do
      local btn = CreateFrame("Frame",nil,bar)
      btn:SetWidth(30)
      btn:SetHeight(30)
        if x < (InterruptBarDB.columns * 30 - 45) then
      btn:SetPoint("CENTER",bar,"CENTER",x,y)
        else
      x = x-(InterruptBarDB.columns * 30)
      y = y-30
            btn:SetPoint("Center",bar,"Center",x,y)

        end
      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

      x = x + 30
   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 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 InterruptBarDB.hidden then ref:Hide() end
   if activetimers[ability] then activetimers[ability] = nil 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 InterruptBarDB.hidden then
      ref:Show()
   end
   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)
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 == 80964 then spellID = 80965
      end
      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_Test()
   for _,ability in ipairs(order) do
      InterruptBar_StartTimer(bar[ability],ability)
   end
end

local cmdfuncs = {
   scale = function(v) InterruptBarDB.scale = v; InterruptBar_UpdateBar() end,
   hidden = function() InterruptBarDB.hidden = not InterruptBarDB.hidden; InterruptBar_UpdateBar() end,
   lock = function() InterruptBarDB.lock = not InterruptBarDB.lock; InterruptBar_UpdateBar() end,
   test = function() InterruptBar_Test() end,
   columns = function(v) InterruptBarDB.columns = v;
ChatFrame1:AddMessage("Type '/console reloadui' to update the interface of InterruptBar.",0,1,0)     
  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 ",0,1,0)     
     ChatFrame1:AddMessage("-- scale  | value: " .. InterruptBarDB.scale,0,1,0)
     ChatFrame1:AddMessage("-- hidden (toggle) | value: " .. tostring(InterruptBarDB.hidden),0,1,0)
     ChatFrame1:AddMessage("-- lock (toggle) | value: " .. tostring(InterruptBarDB.lock),0,1,0)
     ChatFrame1:AddMessage("-- test (execute)",0,1,0)
   ChatFrame1:AddMessage("-- columns  | value: " .. InterruptBarDB.columns,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.columns then InterruptBarDB.columns = 5 end
   if not InterruptBarDB.hidden then InterruptBarDB.hidden = 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; update by Veliace. 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")