--[[
    === About ===
    Raid debuff mod for oUF

    === License ===
--]]

local _, ns = ...
local oUF = ns.oUF or oUF
assert(oUF, 'oUF RaidDebuffs: unable to locate oUF')

local DispelColor = {
    ['Magic']   = {.2, .6, 1},
    ['Curse']   = {.6, 0, 1},
    ['Disease'] = {.6, .4, 0},
    ['Poison']  = {0, .6, 0},
    ['none'] = {0, 0, 0},
}

local DispelPriority = {
    ['Magic']   = 4,
    ['Curse']   = 3,
    ['Disease'] = 2,
    ['Poison']  = 1,
}

local DispelFilter = ({
    PIREST = {
        Magic = true,
        Disease = true,
    },
    SHAMAN = {
        --Magic = true,
        Curse = true,
    },
    PALADIN = {
        --Magic = false,
        Poison = true,
        Disease = true,
    },
    MAGE = {
        Curse = true,
    },
    DRUID = {
        --Magic = true,
        Curse = true,
        Poison = true,
    },
})[select(2, UnitClass'player')]

local function formatTime(s)
    if s > 60 then
        return format('%dm', s/60), s%60
    else
        return format('%d', s), s - floor(s)
    end
end

local function OnUpdate(self, elps)
    self.nextUpdate = self.nextUpdate - elps
    if self.nextUpdate > 0 then return end

    local timeLeft = self.endTime - GetTime()
    if timeLeft > 0 then
        local text, nextUpdate = formatTime(timeLeft)
        self.time:SetText(text)
        self.nextUpdate = nextUpdate
    else
        self:SetScript('OnUpdate', nil)
        self.time:Hide()
    end
end

local function UpdateDebuff(self, name, icon, count, debuffType, duration, endTime)
    local rd = self.RaidDebuffs
    if name then
        rd.icon:SetTexture(icon)
        rd.icon:Show()

        if rd.count then
            if count and (count > 0) then
                rd.count:SetText(count)
                rd.count:Show()
            else
                rd.count:Hide()
            end
        end

        if rd.time then
            if duration and (duration > 0) then
                rd.endTime = endTime
                rd.nextUpdate = 0
                rd:SetScript('OnUpdate', OnUpdate)
                rd.time:Show()
            else
                rd:SetScript('OnUpdate', nil)
                rd.time:Hide()
            end
        end

        if rd.cd then
            if duration and (duration > 0) then
                rd.cd:SetCooldown(endTime - duration, duration)
                rd.cd:Show()
            else
                rd.cd:Hide()
            end
        end

        if(rd.SetBackdropColor) then
            local dispelColor = rd.DispelColor or DispelColor
            local c = dispelColor[debuffType] or dispelColor.none
            rd:SetBackdropColor(unpack(c or DispelColor.none))
        end

        rd:Show()
    else
        rd:Hide()
    end
end

local function Update(self, event, unit)
    if unit ~= self.unit then return end
    local rd = self.RaidDebuffs
    local _name, _icon, _count, _dtype, _duration, _endTime
    local _priority = 0
    local i = 0
    while(true) do
        i = i + 1
        local name, rank, icon, count, debuffType, duration, expirationTime, unitCaster, isStealable, shouldConsolidate, spellId = UnitAura(unit, i, rd.Filter or 'HARMFUL')
        if (not name) then break end
        local priority

        if rd.ShowDispelableDebuff and debuffType then
            local dispelPriority = rd.DispelPriority or DispelPriority
            if rd.FilterDispelableDebuff then
                priority = (rd.DispelFilter or DispelFilter)[debuffType] and DispelPriority[debuffType]
            else
                priority = DispelPriority[debuffType]
            end

            if priority and (priority > _priority) then
                _priority, _name, _icon, _count, _dtype, _duration, _endTime = priority, name, icon, count, debuffType, duration, expirationTime
            end
        end

        priority = rd.Debuffs and rd.Debuffs[rd.MatchBySpellName and name or spellId]
        if priority and (priority > _priority) then
            _priority, _name, _icon, _count, _dtype, _duration, _endTime = priority, name, icon, count, debuffType, duration, expirationTime
        end
    end

    UpdateDebuff(self, _name, _icon, _count, _dtype, _duration, _endTime)
end

local f

local function searchFor(spell, i)
    local spellName = GetSpellInfo(spell)
    local found
    for j = 1, GetNumSpellTabs() do
        for k = 1, GetNumTalents(j) do
            local talentName, _, _, _, rank = GetTalentInfo(j, k)
            if(talentName and talentName == spellName) then
                return true
            end
        end
    end
end

local function spellCheck()
    local _, class = UnitClass'player'
    if(class == 'PALADIN') then
        -- http://www.wowhead.com/spell=53551
        -- Sacred Cleansing
        DispelFilter.Magic = searchFor(53551)
    elseif(class == 'SHAMAN') then
        -- http://www.wowhead.com/spell=77130
        -- Improved Cleanse Spirit
        DispelFilter.Magic = searchFor(77130)
    elseif(class == 'DRUID') then
        -- http://www.wowhead.com/spell=88423
        -- Nature's Cure
        DispelFilter.Magic = searchFor(88423)
    end
end


local function Enable(self)
    if self.RaidDebuffs then
        self:RegisterEvent('UNIT_AURA', Update)
        self.RaidDebuffs.ForceUpdate = Update

        if(not f and not self.RaidDebuffs.DispelFilter) then
            f = CreateFrame'Frame'
            f:SetScript('OnEvent', spellCheck)
            f:RegisterEvent('PLAYER_TALENT_UPDATE')
            f:RegisterEvent('CHARACTER_POINTS_CHANGED')
            spellCheck()
        end

        return true
    end
end

local function Disable(self)
    if self.RaidDebuffs then
        self:UnregisterEvent('UNIT_AURA', Update)
        self.RaidDebuffs:Hide()
        self.RaidDebuffs.ForceUpdate = nil
    end
end

oUF:AddElement('RaidDebuffs', Update, Enable, Disable)