------------------------------------------------------------------------------------------------
--sTrinketWatch
------------------------------------------------------------------------------------------------

local addonName = ...
local sTrinketWatch = CreateFrame("Frame")
sTrinketWatch.icons = {}
sTrinketWatch.timers = {}
local bit_band = bit.band
local bit_bor = bit.bor
local ENEMY_PLAYER = bit_bor(COMBATLOG_OBJECT_REACTION_HOSTILE, COMBATLOG_OBJECT_TYPE_PLAYER)

local function hasFlag(flags, flag)
return bit_band(flags, flag) == flag
end

----------------------------------------
-- Create icons
----------------------------------------
function sTrinketWatch:CreateIcon(unit, point, anchor, relPoint, x, y)
if sTrinketWatch.icons[unit] then
return
end

local trinket = CreateFrame("Cooldown")
trinket:SetPoint(point, anchor, relPoint, x, y)
trinket:SetSize(20, 20)
trinket.icon = trinket:CreateTexture(nil, "BACKGROUND")
trinket.icon:SetAllPoints()
trinket.icon:SetTexture("Interface\\Icons\\inv_jewelry_trinketpvp_01")
trinket:Hide()
sTrinketWatch.icons[unit] = trinket
end

----------------------------------------
-- Add a timer
----------------------------------------
function sTrinketWatch:AddTimer(destGUID, duration)
sTrinketWatch.timers[destGUID] = {}
sTrinketWatch.timers[destGUID].start = GetTime()
sTrinketWatch.timers[destGUID].duration = duration

for unitID in pairs(sTrinketWatch.icons) do
if destGUID == UnitGUID(unitID) then
CooldownFrame_SetTimer(sTrinketWatch.icons[unitID], sTrinketWatch.timers[destGUID].start, sTrinketWatch.timers[destGUID].duration, 1)
end
end
end

----------------------------------------
-- Remove dead timers
----------------------------------------
function sTrinketWatch:RemoveTimers()
for GUID in pairs(sTrinketWatch.timers) do
if (sTrinketWatch.timers[GUID].start + sTrinketWatch.timers[GUID].duration) < GetTime() then
sTrinketWatch.timers[GUID] = nil
end
end
end

----------------------------------------
-- OnEvent Scripts
----------------------------------------
sTrinketWatch:SetScript("OnEvent", function(self, event, ...) return self[event](self, ...) end)

function sTrinketWatch:ADDON_LOADED(arg1)
if arg1 == addonName then
sTrinketWatch:CreateIcon("target", "BOTTOMRIGHT", TargetFrame, "TOP", 12, -16)
sTrinketWatch:CreateIcon("focus", "BOTTOMRIGHT", FocusFrame, "TOP", 16, -16)

sTrinketWatch:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
sTrinketWatch:RegisterEvent("PLAYER_TARGET_CHANGED")
sTrinketWatch:RegisterEvent("PLAYER_FOCUS_CHANGED")
end
end
sTrinketWatch:RegisterEvent("ADDON_LOADED")

function sTrinketWatch:COMBAT_LOG_EVENT_UNFILTERED(_, eventType, _, _, _, _, _, destGUID, _, destFlags, ...)
if eventType == "SPELL_AURA_APPLIED" and hasFlag(destFlags, ENEMY_PLAYER) then
local _, spellID = ...
if spellID == 59752 or spellID == 42292 then
sTrinketWatch:AddTimer(destGUID, 120)
elseif spellID == 7744 then
sTrinketWatch:AddTimer(destGUID, 30)
end
end
end

function sTrinketWatch:PLAYER_ENTERING_WORLD()
for _, icon in pairs(sTrinketWatch.icons) do
icon:SetCooldown(0, 0)
icon:Hide()
end

for player in pairs(sTrinketWatch.timers) do
player = nil
end
end
sTrinketWatch:RegisterEvent("PLAYER_ENTERING_WORLD")

function sTrinketWatch:PLAYER_TARGET_CHANGED()
sTrinketWatch:RemoveTimers()
if TargetFrame:IsVisible() then
local unitGUID = UnitGUID("target")
if sTrinketWatch.timers[unitGUID] then
CooldownFrame_SetTimer(sTrinketWatch.icons["target"], sTrinketWatch.timers[unitGUID].start, sTrinketWatch.timers[unitGUID].duration, 1)
return
end
end
sTrinketWatch.icons["target"]:Hide()
end

function sTrinketWatch:PLAYER_FOCUS_CHANGED()
sTrinketWatch:RemoveTimers()
if FocusFrame:IsVisible() then
local unitGUID = UnitGUID("focus")
if sTrinketWatch.timers[unitGUID] then
CooldownFrame_SetTimer(sTrinketWatch.icons["focus"], sTrinketWatch.timers[unitGUID].start, sTrinketWatch.timers[unitGUID].duration, 1)
return
end
end
sTrinketWatch.icons["focus"]:Hide()
end