-- rButtonAura: core
-- zork, 2018

-----------------------------
-- Variables
-----------------------------

local A, L = ...

local tick = 0.1

local GetTime, UnitAura = GetTime, UnitAura
local numAuras = 0
local auras = {}
local spec = nil
local form = 0

-----------------------------
-- rButtonAura Global
-----------------------------

rButtonAura = {}
rButtonAura.addonName = A

-----------------------------
-- Functions
-----------------------------

--AddAura
function rButtonAura:AddAura(aura)
  table.insert(auras,aura)
end

--SetTick
function rButtonAura:SetTick(t)
  tick = t
end

local function ShowAura(aura)
  if aura.bar then
    aura.bar:Show()
  end
  if aura.border then
    aura.border:Show()
  end
end

local function HideAura(aura)
  if aura.bar then
    aura.bar:Hide()
  end
  if aura.border then
    aura.border:Hide()
  end
end

--UpdateAura
local function UpdateAura(aura)
  local name, icon, count, debuffType, duration, expires, caster = AuraUtil.FindAuraByName(aura.spellName, aura.unit, aura.filter)
  if name and caster == aura.caster then
    if aura.bar then
      local perc = (duration+GetTime()-expires)/duration
      local w = aura.bar.maxwidth-perc*aura.bar.maxwidth
      aura.bar:SetWidth(w)
    end
    ShowAura(aura)
  else
    HideAura(aura)
  end
end

--UpdateAuras
local function UpdateAuras()
  for i, aura in next, auras do
    if aura.enabled then
      UpdateAura(aura)
    end
  end
end

--Tick
local function Tick()
  UpdateAuras()
  C_Timer.After(tick, Tick)
end

local function UpdateSpells()
  spec = GetSpecialization()
  form = GetShapeshiftForm()
  for i, aura in next, auras do
    if aura.requireSpell and not IsPlayerSpell(aura.requireSpell) then
      HideAura(aura)
      aura.enabled = false
    elseif aura.spec and aura.form and (aura.spec ~= spec or aura.form ~= form) then
      HideAura(aura)
      aura.enabled = false
    elseif aura.spec and aura.spec ~= spec then
      HideAura(aura)
      aura.enabled = false
    elseif aura.form and aura.form ~= form then
      HideAura(aura)
      aura.enabled = false
    else
      aura.enabled = true
    end
  end
end

--Login
local function Login()
  numAuras = #auras
  if numAuras == 0 then return end
  local error = false
  for i, aura in next, auras do
    if type(aura.button) == "string" then aura.button = _G[aura.button] end
    local border = _G[aura.button:GetName().."Border"]
    if not border then
      print(A,aura.spellid,"border not found")
      error = true
      break
    else
      aura.border = border
    end
    local spellName = GetSpellInfo(aura.spellid)
    if not spellName then
      print(A,aura.spellid,"spell id not found")
      error = true
      break
    else
      aura.spellName = spellName
    end
    if aura.useBar then
      local a, b = aura.border:GetDrawLayer()
      aura.bar = aura.border:GetParent():CreateTexture(nil,a,nil,b)
      aura.bar:SetColorTexture(unpack(aura.barColor))
      aura.bar:SetBlendMode("ADD")
      aura.bar:SetPoint(unpack(aura.barPoint))
      aura.bar:SetSize(0,aura.barHeight)
      aura.bar:Hide()
      aura.bar.maxwidth = aura.border:GetParent():GetWidth()
    end
    if aura.useBorder then
      aura.border:SetVertexColor(unpack(aura.borderColor))
    end
    aura.enabled = false
  end
  if not error then
    --RegisterCallback SPELLS_CHANGE
    rLib:RegisterCallback("SPELLS_CHANGED", UpdateSpells)
    UpdateSpells()
    Tick()
  end
end

--RegisterCallback PLAYER_LOGIN
rLib:RegisterCallback("PLAYER_LOGIN", Login)