InternalCooldowns.lua

local _, ICD = ...

ICD.InternalCooldowns = {}
local lib = ICD.InternalCooldowns

local substr = _G.string.sub
local playerGUID = UnitGUID("player")
local GetTime, unpack, tonumber = _G.GetTime, _G.unpack, _G.tonumber
local type, ipairs = _G.type, _G.ipairs
local CombatLogGetCurrentEventInfo = CombatLogGetCurrentEventInfo
local bt4 = IsAddOnLoaded("Bartender4")

lib.spellToItem = lib.spellToItem or {}
lib.cooldownStartTimes = lib.cooldownStartTimes or {}
lib.cooldownDurations = lib.cooldownDurations or {}
lib.cooldowns = lib.cooldowns or {}

if not lib.eventFrame then
    lib.eventFrame = CreateFrame("Frame")
    lib.eventFrame:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
    lib.eventFrame:RegisterEvent("PLAYER_LOGIN")
    lib.eventFrame:SetScript("OnEvent", function(frame, event, ...)
        frame.lib[event](frame.lib, event, ...)
    end)
end
lib.eventFrame.lib = lib

local INVALID_EVENTS = {
    SPELL_DISPEL = true,
    SPELL_DISPEL_FAILED = true,
    SPELL_STOLEN = true,
    SPELL_AURA_REMOVED = true,
    SPELL_AURA_REMOVED_DOSE = true,
    SPELL_AURA_BROKEN = true,
    SPELL_AURA_BROKEN_SPELL = true,
    SPELL_CAST_FAILED = true,
}

local function isEquipped(itemID)
    for slot = 0, 19 do
        if GetInventoryItemID("player", slot) == itemID then
            return true
        end
    end

    return false
end

function lib:COMBAT_LOG_EVENT_UNFILTERED()
    local _, event, _, sourceGUID, _, _, _, destGUID, _, _, _, spellID = CombatLogGetCurrentEventInfo()

    if ((destGUID == playerGUID and (sourceGUID == nil or sourceGUID == destGUID)) or sourceGUID == playerGUID) and not INVALID_EVENTS[event] and substr(event, 0, 6) == "SPELL_" then
        local itemID = lib.spellToItem[spellID]
        if itemID then
            if type(itemID) == "table" then
                for _, v in ipairs(itemID) do
                    if isEquipped(v) then
                        self:SetCooldownFor(v, spellID)
                    end
                end
                return
            else
                if isEquipped(itemID) then
                    self:SetCooldownFor(itemID, spellID)
                end
                return
            end
        end
    end
end

function lib:SetCooldownFor(itemID, spellID)
    local duration = lib.cooldowns[spellID] or 45
    lib.cooldownStartTimes[itemID] = GetTime()
    lib.cooldownDurations[itemID] = duration
end

local function cooldownReturn(itemID)
    if not itemID or not lib.cooldownStartTimes[itemID] or not lib.cooldownDurations[itemID] then
        return nil
    end

    local startTime = lib.cooldownStartTimes[itemID]
    local duration = lib.cooldownDurations[itemID]
    if GetTime() > startTime + duration then
        return 0, 0, 0
    else
        return startTime, duration, 1
    end
end

local function TMGetInventoryItemCooldown(maybeGlobal)
    local itemID1 = GetInventoryItemID("player", 13)
    local itemID2 = GetInventoryItemID("player", 14)
    if itemID1 or itemID2 then
        local start1, duration1, enable1 = GetInventoryItemCooldown("player", 13)
        local start2, duration2, enable2 = GetInventoryItemCooldown("player", 14)
        local start3, duration3, running3 = cooldownReturn(itemID1)
        local start4, duration4, running4 = cooldownReturn(itemID2)
        if start3 then
            start1 = start3
            duration1 = duration3
            enable1 = running3
        end
        if start4 then
            start2 = start4
            duration2 = duration4
            enable2 = running4
        end
        CooldownFrame_Set(TrinketMenu_Trinket0Cooldown, start1, duration1, enable1)
        CooldownFrame_Set(TrinketMenu_Trinket1Cooldown, start2, duration2, enable2)
    end

    if not maybeGlobal then
        TrinketMenu.WriteWornCooldowns()
    end
end

local something -- overflow prevention flag
local function GetActionCooldown(self, start, duration, enable)
    local parent, slot = self:GetParent()
    if bt4 then
        slot = parent._state_action
    else
        slot = parent.action
    end
    if not slot or something then return end
    something = true

    local actionType, actionID, actionSubType = GetActionInfo(slot)
    if actionType == "item" then
        local start, duration, running = cooldownReturn(actionID)
        if start then
            CooldownFrame_Set(self, start, duration, running, false)
        end
    elseif actionType == "macro" then
        local _, tex = GetMacroInfo(actionID)
        local itemID1 = GetInventoryItemID("player", 13)
        local itemID2 = GetInventoryItemID("player", 14)
        if itemID1 and GetItemIcon(itemID1) == tex then
            local start, duration, running = cooldownReturn(itemID1)
            if start then
                CooldownFrame_Set(self, start, duration, running, false)
            end
        elseif itemID2 and GetItemIcon(itemID2) == tex then
            local start, duration, running = cooldownReturn(itemID2)
            if start then
                CooldownFrame_Set(self, start, duration, running, false)
            end
        end
    end
    something = false
end

local function PaperDollCD(self)
    local start, duration, enable = GetInventoryItemCooldown("player", self:GetID())

    if not enable or enable == 0 then
        local itemID = GetInventoryItemID("player", self:GetID())

        if itemID then
            local start, duration, running = cooldownReturn(itemID)

            local cooldown = _G[self:GetName() .. "Cooldown"]
            if start then
                CooldownFrame_Set(cooldown, start, duration, running)
            end
        end
    end
end

function lib:PLAYER_LOGIN()
    playerGUID = UnitGUID("player")

    hooksecurefunc("CooldownFrame_Set", GetActionCooldown)
    hooksecurefunc("PaperDollItemSlotButton_Update", PaperDollCD)

    if IsAddOnLoaded("TrinketMenu") then
        TrinketMenu.UpdateWornCooldowns = TMGetInventoryItemCooldown
    end
end