local T = ShaguTweaks.T

local module = ShaguTweaks:register({
    title = T["Improved Castbar"],
    description = T["Adds a spell icon and remaining cast time to the cast bar."],
    expansions = { ["vanilla"] = true },
    category = T["Action Bar"],
    enabled = nil,
})

module.enable = function(self)
    local _G = ShaguTweaks.GetGlobalEnv()
    local API = ShaguTweaks.API

    local classicEvents = {
        "UNIT_SPELLCAST_START",
        "UNIT_SPELLCAST_STOP",
        "UNIT_SPELLCAST_INTERRUPTED",
        "UNIT_SPELLCAST_FAILED",
        "UNIT_SPELLCAST_DELAYED",
        "UNIT_SPELLCAST_CHANNEL_START",
        "UNIT_SPELLCAST_CHANNEL_STOP",
        "UNIT_SPELLCAST_CHANNEL_UPDATE",
    }

    -- ClassicAPI is a required dependency of this fork. Keep this module on
    -- its native cast/query path instead of restoring the old 1.12 fallbacks.
    if not API or not API.casts or not API.eventutils
        or not _G.C_EventUtils
        or type(_G.C_EventUtils.IsEventValid) ~= "function" then
        return
    end

    for index = 1, table.getn(classicEvents) do
        if not _G.C_EventUtils.IsEventValid(classicEvents[index]) then
            return
        end
    end

    local castbar = CreateFrame("FRAME", "STImprovedCastbar", CastingBarFrame)
    castbar:Hide()

    castbar.texture = CreateFrame("Frame", nil, castbar)
    castbar.texture:SetPoint("RIGHT", CastingBarFrame, "LEFT", -10, 2)
    castbar.texture:SetWidth(28)
    castbar.texture:SetHeight(28)

    castbar.texture.icon = castbar.texture:CreateTexture(nil, "BACKGROUND")
    castbar.texture.icon:SetPoint("CENTER", 0, 0)
    castbar.texture.icon:SetWidth(24)
    castbar.texture.icon:SetHeight(24)
    castbar.texture:SetBackdrop({
        edgeFile = "Interface\\Tooltips\\UI-Tooltip-Border",
        tile = true, tileSize = 8, edgeSize = 12,
        insets = { left = 2, right = 2, top = 2, bottom = 2 }
    })

    if ShaguTweaks.DarkMode then
        castbar.texture:SetBackdropBorderColor(.3, .3, .3, .9)
    end

    castbar.spellText = castbar:CreateFontString(nil, "HIGH", "GameFontWhite")
    castbar.spellText:SetPoint("CENTER", CastingBarFrame, "CENTER", 0, 3)
    local font, size, opts = castbar.spellText:GetFont()
    castbar.spellText:SetFont(font, size, "THINOUTLINE")

    castbar.timerText = castbar:CreateFontString(nil, "HIGH", "GameFontWhite")
    castbar.timerText:SetPoint("RIGHT", CastingBarFrame, "RIGHT", -5, 3)
    castbar.timerText:SetFont(font, size, "THINOUTLINE")

    CastingBarText:Hide()

    local function QueryPlayerCast()
        local cast, nameSubtext, text, texture, startTime, endTime, isTradeSkill = API.GetCastInfo("player")
        if cast then
            return cast, nameSubtext, text, texture, startTime, endTime, isTradeSkill, false
        end

        local channel
        channel, nameSubtext, text, texture, startTime, endTime, isTradeSkill = API.GetChannelInfo("player")
        if channel then
            return channel, nameSubtext, text, texture, startTime, endTime, isTradeSkill, true
        end
    end

    local timerFormat = "%.1f"..T["s"]

    local function UpdateTimer()
        local alpha = CastingBarFrame:GetAlpha()
        if castbar.lastAlpha ~= alpha then
            castbar.lastAlpha = alpha
            castbar:SetAlpha(alpha)
        end

        local startTime = castbar.startTime
        local endTime = castbar.endTime
        if not startTime or not endTime or endTime <= startTime then
            if alpha == 0 then
                castbar:Hide()
            end
            return
        end

        local now = GetTime()
        local max = endTime - startTime
        local cur
        if castbar.isChannel then
            cur = endTime - now
        else
            cur = now - startTime
        end

        cur = cur > max and max or cur
        cur = cur < 0 and 0 or cur
        local rem = castbar.isChannel and cur or (max - cur)
        rem = rem < 0 and 0 or rem

        local timerText = string.format(timerFormat, rem)
        if castbar.lastTimerText ~= timerText then
            castbar.lastTimerText = timerText
            castbar.timerText:SetText(timerText)
        end

        if now >= endTime then
            castbar.startTime = nil
            castbar.endTime = nil
        end
    end

    local function RefreshCast()
        local cast, nameSubtext, text, texture, startTime, endTime, isTradeSkill, isChannel = QueryPlayerCast()

        if not cast or not startTime or not endTime or endTime <= startTime then
            -- Preserve Blizzard's normal fade-out: keep our last icon/text
            -- attached until the native castbar has finished fading.
            castbar.startTime = nil
            castbar.endTime = nil
            castbar.isChannel = nil
            if CastingBarFrame:GetAlpha() == 0 then
                castbar:Hide()
            end
            return
        end

        castbar.startTime = startTime / 1000
        castbar.endTime = endTime / 1000
        castbar.isChannel = isChannel
        castbar.lastTimerText = nil
        castbar.spellText:SetText(cast)

        if texture then
            castbar.texture.icon:SetTexture(texture)
            castbar.texture.icon:Show()
        else
            castbar.texture.icon:Hide()
        end

        castbar.elapsed = 0
        castbar:Show()
        UpdateTimer()
    end

    -- The cast/channel query runs only on ClassicAPI state-change events.
    -- While visible, OnUpdate only animates the remaining-time text and alpha.
    castbar.elapsed = 0
    castbar:SetScript("OnUpdate", function()
        this.elapsed = (this.elapsed or 0) + arg1
        if this.elapsed < .05 then return end
        this.elapsed = 0
        UpdateTimer()
    end)

    local events = CreateFrame("Frame", nil, UIParent)
    events:RegisterEvent("PLAYER_ENTERING_WORLD")

    for index = 1, table.getn(classicEvents) do
        events:RegisterEvent(classicEvents[index])
    end

    events:SetScript("OnEvent", function()
        if event == "PLAYER_ENTERING_WORLD" then
            RefreshCast()
            return
        end

        if arg1 ~= "player" then return end
        RefreshCast()
    end)
end