local addonName, ns = ...
local W = ns.Widgets
local L = ns.LayoutUtil

local function IsSecret(value)
    return issecretvalue and issecretvalue(value) or false
end

local VIGOR_SPELL = 372608
local SECOND_WIND_SPELL = 425782
local WHIRLING_SURGE_SPELL = 361584

local SPEED_RECIPROCAL = 0.01176
local SPEED_DISPLAY_FACTOR = 14.285
local THRILL_THRESHOLD = 6.003
local GROUND_SKIM_DURATION = 8.28
local THROTTLE = 0.0333
local NUM_CHARGES = 6
local BAR_TEXTURE = [[Interface\Buttons\WHITE8x8]]
local THRILL_SPEED_THRESHOLD = 0.60

local COLOR_PRESETS = {
    Classic = {
        charge     = { r = 0.01, g = 0.56, b = 0.91 },
        thrill     = { r = 1.00, g = 0.66, b = 0.00 },
        groundSkim = { r = 1.00, g = 0.80, b = 0.20 },
        lowSpeed   = { r = 0.00, g = 0.49, b = 0.79 },
        secondWind = { r = 0.00, g = 0.49, b = 0.79 },
        background = { r = 0.12, g = 0.12, b = 0.12 },
        border     = { r = 0.00, g = 0.00, b = 0.00 },
    },
}

local DEFAULTS = {
    barWidth = 36, speedHeight = 14, chargeHeight = 14, gap = 0,
    showSpeedText = true, swapPosition = false, hideWhenGroundedFull = false,
    showSecondWind = true, showWhirlingSurge = true, colorPreset = "Classic",
    unlocked = false, point = "BOTTOM", posX = 0, posY = 200,
    barStyle = "Solid",
    speedColorR = 0.00, speedColorG = 0.49, speedColorB = 0.79,
    thrillColorR = 1.00, thrillColorG = 0.66, thrillColorB = 0.00,
    chargeColorR = 0.01, chargeColorG = 0.56, chargeColorB = 0.91,
    speedFont = "Naowh",
    speedFontSize = 12, speedTextOffsetX = 0, speedTextOffsetY = 0,
    surgeIconSize = 0, surgeAnchor = "RIGHT",
    surgeOffsetX = 6, surgeOffsetY = 0,
    surgeIconFont = "Naowh", surgeIconFontSize = 11,
    anchorFrame = "UIParent", anchorTo = "BOTTOM",
    matchAnchorWidth = false,
    bgColorR = 0.12, bgColorG = 0.12, bgColorB = 0.12, bgAlpha = 0.8,
    borderColorR = 0, borderColorG = 0, borderColorB = 0, borderAlpha = 1, borderSize = 1,
    iconBorderColorR = 0, iconBorderColorG = 0, iconBorderColorB = 0, iconBorderAlpha = 1, iconBorderSize = 1,
    hideCdmWhileMounted = false,
    hideBcmWhileMounted = false,
    showThrillTick = false,
}

local function Get(key)
    local db = NaowhQOL.dragonriding
    if db and db[key] ~= nil then return db[key] end
    return DEFAULTS[key]
end

local prevSpeed = 0
local elapsed = 0
local lastColorState = nil
local uiBuilt = false
local stashedPosition = nil

local mainFrame
local speedBar, speedText, speedTextFrame, thrillTick
local chargeBars = {}
local chargeDividers = {}
local secondWindBars = {}
local surgeFrame, surgeCooldown, surgeTextFrame, surgeCooldownText, surgeBorder
local surgeCdStart, surgeCdDuration = 0, 0
local lastSurgeDisplayText = ""
local eventFrame
local pendingCdmShow = false
local pendingCdmHide = false
local pendingBcmShow = false
local pendingBcmHide = false
local isCasting = false

local function IsEnabled()
    return NaowhQOL.dragonriding and NaowhQOL.dragonriding.enabled
end

local eventsRegistered = false
local DYNAMIC_EVENTS = {
    "ACTIONBAR_UPDATE_COOLDOWN",
    "ACTIONBAR_UPDATE_STATE",
    "UPDATE_BONUS_ACTIONBAR",
    "PLAYER_CAN_GLIDE_CHANGED",
    "PLAYER_IS_GLIDING_CHANGED",
}

local function RegisterDynamicEvents()
    if eventsRegistered or not eventFrame then return end
    for _, event in ipairs(DYNAMIC_EVENTS) do
        eventFrame:RegisterEvent(event)
    end
    eventsRegistered = true
end

local function UnregisterDynamicEvents()
    if not eventsRegistered or not eventFrame then return end
    for _, event in ipairs(DYNAMIC_EVENTS) do
        eventFrame:UnregisterEvent(event)
    end
    eventsRegistered = false
end

local CDM_FRAMES = {
    BuffIconCooldownViewer = true,
    EssentialCooldownViewer = true,
    UtilityCooldownViewer = true,
    BuffBarCooldownViewer = true,
}

local function IsAnchoredToCDM()
    return CDM_FRAMES[Get("anchorFrame")] == true
end

local function StashPositionAndReanchor()
    if not mainFrame or stashedPosition then return end
    if not IsAnchoredToCDM() then return end

    local x, y = mainFrame:GetCenter()
    if not x or not y then return end
    stashedPosition = { x = x, y = y }

    local uiWidth, uiHeight = UIParent:GetSize()
    local offsetX = x - (uiWidth / 2)
    local offsetY = y - (uiHeight / 2)

    mainFrame:ClearAllPoints()
    mainFrame:SetPoint("CENTER", UIParent, "CENTER", offsetX, offsetY)
end

local function RestoreOriginalAnchor()
    if not mainFrame or not stashedPosition then return end
    stashedPosition = nil

    mainFrame:ClearAllPoints()
    local anchorParent = L.GetAnchorFrame(Get("anchorFrame")) or UIParent
    mainFrame:SetPoint(Get("point"), anchorParent, Get("anchorTo"), Get("posX"), Get("posY"))
end

local function IsSkyriding()
    if GetBonusBarIndex() == 11 and GetBonusBarOffset() == 5 then
        return true
    end
    if UnitPowerBarID("player") == 650 then return false end
    local _, canGlide = C_PlayerInfo.GetGlidingInfo()
    return canGlide and UnitPowerBarID("player") ~= 0
end

local function IsGliding()
    local gliding = C_PlayerInfo.GetGlidingInfo()
    return gliding
end

local function GetForwardSpeed()
    local _, _, spd = C_PlayerInfo.GetGlidingInfo()
    return spd or 0
end

local function GetVigorInfo()
    local data = C_Spell.GetSpellCharges(VIGOR_SPELL)
    if not data then return 0, 6, 0, 0, false, false end
    local cc = data.currentCharges
    local mc = data.maxCharges
    local cs = data.cooldownStartTime
    local cd = data.cooldownDuration
    if IsSecret(cc) or IsSecret(mc) or IsSecret(cs) or IsSecret(cd) then
        return 0, 6, 0, 0, false, false
    end
    local isThrill = cd > 0 and cd <= THRILL_THRESHOLD
    local isGroundSkim = math.abs(cd - GROUND_SKIM_DURATION) < 0.05 and not isThrill
    return cc, mc, cs, cd, isThrill, isGroundSkim
end

local function GetSecondWindCharges()
    local data = C_Spell.GetSpellCharges(SECOND_WIND_SPELL)
    if not data then return 0 end
    local cc = data.currentCharges
    if IsSecret(cc) then return 0 end
    return cc
end

local function GetWhirlingSurgeCooldown()
    local data = C_Spell.GetSpellCooldown(WHIRLING_SURGE_SPELL)
    if not data then return 0, 0 end
    local s, d = data.startTime, data.duration
    if IsSecret(s) or IsSecret(d) then return 0, 0 end
    return s, d
end

local function GetPreset()
    return COLOR_PRESETS[Get("colorPreset")] or COLOR_PRESETS.Classic
end

local function ApplyColors(isThrill, isGroundSkim)
    local state = isThrill and "thrill" or (isGroundSkim and "groundSkim" or "lowSpeed")
    if state == lastColorState then return end
    lastColorState = state

    local db = NaowhQOL.dragonriding or {}
    if isThrill then
        local r, g, b = W.GetEffectiveColor(db, "thrillColorR", "thrillColorG", "thrillColorB", "thrillColorUseClassColor")
        speedBar:SetStatusBarColor(r, g, b)
    else
        local r, g, b = W.GetEffectiveColor(db, "speedColorR", "speedColorG", "speedColorB", "speedColorUseClassColor")
        speedBar:SetStatusBarColor(r, g, b)
    end

    local cr, cg, cb = W.GetEffectiveColor(db, "chargeColorR", "chargeColorG", "chargeColorB", "chargeColorUseClassColor")
    for i = 1, NUM_CHARGES do
        chargeBars[i]:SetStatusBarColor(cr, cg, cb)
    end
end

local function UpdateLayout()
    if not mainFrame then return end
    local db = NaowhQOL.dragonriding or {}
    local speedHeight = Get("speedHeight")
    local chargeHeight = Get("chargeHeight")
    local gap = Get("gap")
    local totalHeight = speedHeight + gap + chargeHeight

    mainFrame:ClearAllPoints()
    local anchorParent = L.GetAnchorFrame(Get("anchorFrame")) or UIParent
    mainFrame:SetPoint(Get("point"), anchorParent, Get("anchorTo"), Get("posX"), Get("posY"))

    local barWidthCfg = Get("barWidth")
    local totalWidth = NUM_CHARGES * barWidthCfg + (NUM_CHARGES - 1) * gap
    local barWidth = barWidthCfg
    local borderSize = Get("borderSize")

    if Get("matchAnchorWidth") and anchorParent ~= UIParent then
        local anchorWidth = anchorParent:GetWidth()
        if anchorWidth and anchorWidth > 0 then
            totalWidth = anchorWidth - (borderSize * 2)
            barWidth = (totalWidth - (NUM_CHARGES - 1) * gap) / NUM_CHARGES
        end
    end
    mainFrame:SetSize(totalWidth + borderSize * 2, totalHeight + borderSize * 2)
    mainFrame:SetBackdrop({
        bgFile = BAR_TEXTURE,
        edgeFile = BAR_TEXTURE,
        edgeSize = borderSize,
        insets = { left = borderSize, right = borderSize, top = borderSize, bottom = borderSize }
    })
    local bgR, bgG, bgB = W.GetEffectiveColor(db, "bgColorR", "bgColorG", "bgColorB", "bgColorUseClassColor")
    local brR, brG, brB = W.GetEffectiveColor(db, "borderColorR", "borderColorG", "borderColorB", "borderColorUseClassColor")
    mainFrame:SetBackdropColor(bgR, bgG, bgB, Get("bgAlpha"))
    mainFrame:SetBackdropBorderColor(brR, brG, brB, Get("borderAlpha"))

    local swapPosition = Get("swapPosition")
    local speedY = swapPosition and -borderSize or -(chargeHeight + gap + borderSize)
    local chargeY = swapPosition and -(speedHeight + gap + borderSize) or -borderSize

    speedBar:ClearAllPoints()
    speedBar:SetSize(totalWidth, speedHeight)
    speedBar:SetPoint("TOPLEFT", mainFrame, "TOPLEFT", borderSize, speedY)

    speedText:SetFont(ns.Media.ResolveFont(Get("speedFont")), Get("speedFontSize"), "OUTLINE")
    local fontSize = Get("speedFontSize")
    speedTextFrame:SetSize(math.max(44, fontSize * 2.5), math.max(24, fontSize + 12))
    speedTextFrame:ClearAllPoints()
    speedTextFrame:SetPoint("RIGHT", mainFrame, "RIGHT", Get("speedTextOffsetX"), Get("speedTextOffsetY"))
    local moduleEnabled = IsEnabled()
    speedTextFrame:SetShown(moduleEnabled and Get("showSpeedText"))
    speedTextFrame:EnableMouse(moduleEnabled and Get("unlocked"))
    if moduleEnabled and Get("unlocked") and Get("showSpeedText") then
        speedTextFrame:SetBackdrop({
            bgFile = "Interface\\Tooltips\\UI-Tooltip-Background",
            edgeFile = "Interface\\Tooltips\\UI-Tooltip-Border",
            tile = true, tileSize = 16, edgeSize = 16,
            insets = { left = 4, right = 4, top = 4, bottom = 4 }
        })
        speedTextFrame:SetBackdropColor(0.1, 0.1, 0.1, 0.8)
        speedTextFrame:SetBackdropBorderColor(0.4, 0.4, 0.4, 1)
    else
        speedTextFrame:SetBackdrop(nil)
    end

    if thrillTick then
        thrillTick:SetShown(Get("showThrillTick"))
        thrillTick:ClearAllPoints()
        thrillTick:SetSize(1, speedHeight)
        thrillTick:SetPoint("LEFT", speedBar, "LEFT", totalWidth * THRILL_SPEED_THRESHOLD, 0)
    end

    for i = 1, NUM_CHARGES do
        local xOff = borderSize + (i - 1) * (barWidth + gap)

        secondWindBars[i]:ClearAllPoints()
        secondWindBars[i]:SetSize(barWidth, chargeHeight)
        secondWindBars[i]:SetPoint("TOPLEFT", mainFrame, "TOPLEFT", xOff, chargeY)

        chargeBars[i]:ClearAllPoints()
        chargeBars[i]:SetSize(barWidth, chargeHeight)
        chargeBars[i]:SetPoint("TOPLEFT", mainFrame, "TOPLEFT", xOff, chargeY)

        if chargeDividers[i] then
            chargeDividers[i]:ClearAllPoints()
            chargeDividers[i]:SetSize(1, chargeHeight)
            chargeDividers[i]:SetPoint("TOPLEFT", mainFrame, "TOPLEFT", xOff + barWidth, chargeY)
        end
    end

    if surgeFrame then
        surgeFrame:ClearAllPoints()
        local surgeIconSize = Get("surgeIconSize")
        local iconSize = surgeIconSize > 0 and surgeIconSize or (chargeHeight + speedHeight + gap)
        surgeFrame:SetSize(iconSize, iconSize)

        local anchor = Get("surgeAnchor")
        local ox, oy = Get("surgeOffsetX"), Get("surgeOffsetY")
        if anchor == "LEFT" then
            surgeFrame:SetPoint("RIGHT", mainFrame, "LEFT", -ox, oy)
        elseif anchor == "TOP" then
            surgeFrame:SetPoint("BOTTOM", mainFrame, "TOP", ox, oy)
        elseif anchor == "BOTTOM" then
            surgeFrame:SetPoint("TOP", mainFrame, "BOTTOM", ox, -oy)
        else
            surgeFrame:SetPoint("LEFT", mainFrame, "RIGHT", ox, oy)
        end
        surgeFrame:SetShown(Get("showWhirlingSurge"))

        if surgeBorder then
            local iconBorderSize = Get("iconBorderSize")
            surgeBorder:ClearAllPoints()
            surgeBorder:SetPoint("TOPLEFT", surgeFrame, "TOPLEFT", -iconBorderSize, iconBorderSize)
            surgeBorder:SetPoint("BOTTOMRIGHT", surgeFrame, "BOTTOMRIGHT", iconBorderSize, -iconBorderSize)
            surgeBorder:SetBackdrop({
                edgeFile = BAR_TEXTURE,
                edgeSize = iconBorderSize,
            })
            local ibR, ibG, ibB = W.GetEffectiveColor(db, "iconBorderColorR", "iconBorderColorG", "iconBorderColorB", "iconBorderColorUseClassColor")
            surgeBorder:SetBackdropBorderColor(ibR, ibG, ibB, Get("iconBorderAlpha"))
        end

        if surgeCooldownText then
            surgeCooldownText:SetFont(
                ns.Media.ResolveFont(Get("surgeIconFont")),
                Get("surgeIconFontSize") or 11,
                "OUTLINE"
            )
        end
        if surgeTextFrame and surgeCooldown then
            surgeTextFrame:SetFrameLevel(surgeCooldown:GetFrameLevel() + 2)
        end
    end

    local barTex = ns.Media.ResolveBar(Get("barStyle")) or BAR_TEXTURE
    speedBar:SetStatusBarTexture(barTex)
    for i = 1, NUM_CHARGES do
        secondWindBars[i]:SetStatusBarTexture(barTex)
        chargeBars[i]:SetStatusBarTexture(barTex)
    end

    lastColorState = nil
end

local function UpdateSpeedBar(rawSpeed)
    local scaled = math.min(rawSpeed * SPEED_RECIPROCAL, 1.0)
    prevSpeed = prevSpeed + (scaled - prevSpeed) * 0.15
    speedBar:SetValue(prevSpeed)

    if Get("showSpeedText") then
        local display = math.floor(rawSpeed * SPEED_DISPLAY_FACTOR)
        speedText:SetText(display > 0 and tostring(display) or "")
    end
end

local function UpdateCharges(charges, maxCharges, startTime, duration)
    local now = GetTime()
    for i = 1, NUM_CHARGES do
        if i > maxCharges then
            chargeBars[i]:SetValue(0)
        elseif i <= charges then
            chargeBars[i]:SetValue(1)
        elseif i == charges + 1 and duration > 0 and startTime > 0 then
            local progress = (now - startTime) / duration
            chargeBars[i]:SetValue(math.min(progress, 1))
        else
            chargeBars[i]:SetValue(0)
        end
    end
end

local function UpdateSecondWind(charges, totalFilled)
    if not Get("showSecondWind") then
        for i = 1, NUM_CHARGES do
            secondWindBars[i]:SetValue(0)
        end
        return
    end
    for i = 1, NUM_CHARGES do
        secondWindBars[i]:SetValue(i <= totalFilled and 1 or 0)
    end
end

local function FormatSurgeRemaining(remaining)
    if remaining <= 0 then return "" end
    if remaining < 3 then
        return string.format("%.1f", remaining)
    end
    return tostring(math.ceil(remaining))
end

local function UpdateSurgeCooldownText(startTime, duration)
    if not surgeCooldownText then return end
    if startTime <= 0 or duration <= 0 then
        if lastSurgeDisplayText ~= "" then
            lastSurgeDisplayText = ""
            surgeCooldownText:SetText("")
        end
        return
    end
    local remaining = duration - (GetTime() - startTime)
    local text = FormatSurgeRemaining(remaining)
    if text ~= lastSurgeDisplayText then
        lastSurgeDisplayText = text
        surgeCooldownText:SetText(text)
    end
end

local function ClearSurgeCooldown()
    surgeCdStart = 0
    surgeCdDuration = 0
    lastSurgeDisplayText = ""
    if surgeCooldown then
        surgeCooldown:Clear()
    end
    if surgeCooldownText then
        surgeCooldownText:SetText("")
    end
end

local function UpdateWhirlingSurge(startTime, duration)
    if not Get("showWhirlingSurge") or not surgeFrame then
        if surgeFrame then surgeFrame:Hide() end
        ClearSurgeCooldown()
        return
    end
    surgeFrame:Show()
    if startTime > 0 and duration > 0 then
        if startTime ~= surgeCdStart or duration ~= surgeCdDuration then
            surgeCdStart = startTime
            surgeCdDuration = duration
            surgeCooldown:SetCooldown(startTime, duration)
        end
        UpdateSurgeCooldownText(startTime, duration)
    else
        ClearSurgeCooldown()
    end
end

local CDM_VIEWER_NAMES = {
    "EssentialCooldownViewer",
    "UtilityCooldownViewer",
    "BuffIconCooldownViewer",
    "BuffBarCooldownViewer",
}

local function SetFrameTreeAlpha(frame, alpha)
    if not frame or not frame.SetAlpha then return end
    frame:SetAlpha(alpha)
    for _, child in ipairs({ frame:GetChildren() }) do
        SetFrameTreeAlpha(child, alpha)
    end
end

local function IsEllesmereCdmGlobal(name, frame)
    return type(name) == "string"
        and type(frame) == "table"
        and frame.SetAlpha
        and (name:find("^ECME_CDMBar_", 1, true) or name:find("^ECME_TBB", 1, true))
end

local function RestoreEllesmereCdmFrame(frame, name)
    if name:find("^ECME_TBBWrap", 1, true) then
        frame:SetAlpha(1)
    elseif name:find("^ECME_TBB", 1, true) then
        frame:SetAlpha(frame._opacityTarget or 1)
    end
end

local function SetEllesmereCdmAlpha(alpha)
    if not C_AddOns.IsAddOnLoaded("EllesmereUICooldownManager") then return end

    for name, frame in pairs(_G) do
        if IsEllesmereCdmGlobal(name, frame) then
            if alpha == 0 then
                SetFrameTreeAlpha(frame, 0)
            else
                RestoreEllesmereCdmFrame(frame, name)
            end
        end
    end

    if alpha == 1 and _G._ECME_ApplyVisibility then
        pcall(_G._ECME_ApplyVisibility)
    end
end

local function SetCdmAlpha(alpha)
    for _, vName in ipairs(CDM_VIEWER_NAMES) do
        local viewer = _G[vName]
        if viewer then
            viewer:SetAlpha(alpha)
            if viewer.itemFramePool then
                for frame in viewer.itemFramePool:EnumerateActive() do
                    if frame then frame:SetAlpha(alpha) end
                end
            end
        end
    end
    local ayijeCDM = _G["Ayije_CDM"]
    if ayijeCDM and ayijeCDM.anchorContainers then
        for _, container in pairs(ayijeCDM.anchorContainers) do
            if container then container:SetAlpha(alpha) end
        end
    end
    SetEllesmereCdmAlpha(alpha)
end

local cdmHidden = false
local function HideCooldownManager()
    if cdmHidden then return end
    local success = pcall(function()
        StashPositionAndReanchor()
        cdmHidden = true
        SetCdmAlpha(0)
    end)
    if not success and InCombatLockdown() then
        cdmHidden = false
        pendingCdmHide = true
        pendingCdmShow = false
    end
end

local function ShowCooldownManager()
    if not cdmHidden then return end
    local success = pcall(function()
        cdmHidden = false
        SetCdmAlpha(1)
        RestoreOriginalAnchor()
    end)
    if not success and InCombatLockdown() then
        cdmHidden = true
        pendingCdmShow = true
        pendingCdmHide = false
    end
end

local ERB_RESOURCE_BAR_NAMES = {
    "ERB_PrimaryBar",
    "ERB_SecondaryFrame",
}

local function SetEllesmereResourceBarAlpha(alpha)
    if not C_AddOns.IsAddOnLoaded("EllesmereUIResourceBars") then return end
    for _, barName in ipairs(ERB_RESOURCE_BAR_NAMES) do
        local bar = _G[barName]
        if bar then bar:SetAlpha(alpha) end
    end
end

local function SetResourceBarAlpha(alpha)
    if C_AddOns.IsAddOnLoaded("Ayije_CDM") then
        local cdm = _G["Ayije_CDM"]
        if cdm then
            if cdm.resourceBars then
                for _, bar in pairs(cdm.resourceBars) do
                    if bar then bar:SetAlpha(alpha) end
                end
            end
            if cdm.resourceUnifiedHosts then
                for _, entry in pairs(cdm.resourceUnifiedHosts) do
                    if entry and entry.host then entry.host:SetAlpha(alpha) end
                end
            end
        end
    end
    if C_AddOns.IsAddOnLoaded("BetterCooldownManager") then
        if BCDM_PowerBar then BCDM_PowerBar:SetAlpha(alpha) end
        if BCDM_SecondaryPowerBar then BCDM_SecondaryPowerBar:SetAlpha(alpha) end
    end
    SetEllesmereResourceBarAlpha(alpha)
end

local bcmHidden = false
local function HideBCM()
    if bcmHidden then return end
    local success = pcall(function()
        bcmHidden = true
        SetResourceBarAlpha(0)
    end)
    if not success and InCombatLockdown() then
        bcmHidden = false
        pendingBcmHide = true
        pendingBcmShow = false
    end
end

local function ShowBCM()
    if not bcmHidden then return end
    local success = pcall(function()
        bcmHidden = false
        SetResourceBarAlpha(1)
    end)
    if not success and InCombatLockdown() then
        bcmHidden = true
        pendingBcmShow = true
        pendingBcmHide = false
    end
end

local OnUpdate = ns.PerfMonitor:Wrap("Dragonriding", function(self, dt)
    elapsed = elapsed + dt
    if elapsed < THROTTLE then return end
    elapsed = 0

    if not mainFrame or not speedBar then return end

    if not IsEnabled() or not IsSkyriding() then
        mainFrame:SetAlpha(0)
        mainFrame:Hide()
        if speedTextFrame then speedTextFrame:Hide() end
        eventFrame:SetScript("OnUpdate", nil)
        prevSpeed = 0
        lastColorState = nil
        if Get("hideCdmWhileMounted") then
            ShowCooldownManager()
        end
        if Get("hideBcmWhileMounted") then
            ShowBCM()
        end
        return
    end

    if isCasting then
        mainFrame:SetAlpha(0)
        mainFrame:Hide()
        if speedTextFrame then speedTextFrame:Hide() end
        return
    end

    local charges, maxCharges, startTime, duration, isThrill, isGroundSkim = GetVigorInfo()

    if Get("hideWhenGroundedFull") and not IsGliding() and charges >= maxCharges then
        mainFrame:SetAlpha(0)
        mainFrame:Hide()
        if speedTextFrame then speedTextFrame:Hide() end
        eventFrame:SetScript("OnUpdate", nil)
        if Get("hideCdmWhileMounted") then
            ShowCooldownManager()
        end
        if Get("hideBcmWhileMounted") then
            ShowBCM()
        end
        return
    end

    mainFrame:Show()
    mainFrame:SetAlpha(1)
    if speedTextFrame and Get("showSpeedText") then speedTextFrame:Show() end

    if Get("hideCdmWhileMounted") then
        HideCooldownManager()
    end
    if Get("hideBcmWhileMounted") then
        HideBCM()
    end

    UpdateSpeedBar(GetForwardSpeed())
    UpdateCharges(charges, maxCharges, startTime, duration)
    ApplyColors(isThrill, isGroundSkim)

    if Get("showSecondWind") then
        local swCharges = GetSecondWindCharges()
        UpdateSecondWind(charges, charges + swCharges)
    else
        UpdateSecondWind(0, 0)
    end

    if Get("showWhirlingSurge") then
        local sStart, sDur = GetWhirlingSurgeCooldown()
        UpdateWhirlingSurge(sStart, sDur)
    else
        UpdateWhirlingSurge(0, 0)
    end
end)

local function ActivateUpdater()
    if not mainFrame then return end
    if not IsEnabled() then return end
    eventFrame:SetScript("OnUpdate", OnUpdate)
end

local function BuildUI()
    if uiBuilt then return end
    uiBuilt = true

    local db = NaowhQOL.dragonriding or {}
    local preset = GetPreset()

    mainFrame = CreateFrame("Frame", "NaowhQOL_Dragonriding", UIParent, "BackdropTemplate")
    mainFrame:SetFrameStrata("MEDIUM")
    mainFrame:SetFrameLevel(100)
    mainFrame:SetClampedToScreen(true)
    local borderSize = Get("borderSize")
    mainFrame:SetBackdrop({
        bgFile = BAR_TEXTURE,
        edgeFile = BAR_TEXTURE,
        edgeSize = borderSize,
        insets = { left = borderSize, right = borderSize, top = borderSize, bottom = borderSize }
    })
    local bgR, bgG, bgB = W.GetEffectiveColor(db, "bgColorR", "bgColorG", "bgColorB", "bgColorUseClassColor")
    local brR, brG, brB = W.GetEffectiveColor(db, "borderColorR", "borderColorG", "borderColorB", "borderColorUseClassColor")
    mainFrame:SetBackdropColor(bgR, bgG, bgB, Get("bgAlpha"))
    mainFrame:SetBackdropBorderColor(brR, brG, brB, Get("borderAlpha"))

    W.MakeDraggable(mainFrame, {
        db = NaowhQOL.dragonriding,
        unlockKey = "unlocked",
        pointKey = "point",
        anchorToKey = "anchorTo",
        xKey = "posX",
        yKey = "posY",
        userPlaced = false,
        onDragStop = function()
            if ns.SettingsIO then ns.SettingsIO:MarkDirty() end
        end,
    })

    speedBar = CreateFrame("StatusBar", nil, mainFrame)
    speedBar:SetStatusBarTexture(BAR_TEXTURE)
    speedBar:SetMinMaxValues(0, 1)
    speedBar:SetValue(0)
    local sR, sG, sB = W.GetEffectiveColor(db, "speedColorR", "speedColorG", "speedColorB", "speedColorUseClassColor")
    speedBar:SetStatusBarColor(sR, sG, sB)

    thrillTick = speedBar:CreateTexture(nil, "OVERLAY")
    thrillTick:SetColorTexture(0.01, 0.56, 0.91, 1)

    speedTextFrame = CreateFrame("Frame", "NaowhQOL_DragonridingSpeedText", UIParent, "BackdropTemplate")
    speedTextFrame:SetFrameStrata("MEDIUM")
    speedTextFrame:SetFrameLevel(mainFrame:GetFrameLevel() + 10)
    speedTextFrame:SetSize(44, 24)
    speedTextFrame:SetClampedToScreen(true)
    speedTextFrame:SetMovable(true)
    speedTextFrame:RegisterForDrag("LeftButton")

    speedTextFrame:SetScript("OnDragStart", function(self)
        if db.unlocked then
            self:StartMoving()
        end
    end)

    speedTextFrame:SetScript("OnDragStop", function(self)
        self:StopMovingOrSizing()
        local cx, cy = self:GetCenter()
        local fw = self:GetWidth()
        local mx, my = mainFrame:GetCenter()
        local mw = mainFrame:GetWidth()
        if cx and cy and mx and my then
            local frameRight = cx + fw / 2
            local mainRight = mx + mw / 2
            db.speedTextOffsetX = math.floor(frameRight - mainRight)
            db.speedTextOffsetY = math.floor(cy - my)
        end
        self:ClearAllPoints()
        self:SetPoint("RIGHT", mainFrame, "RIGHT", Get("speedTextOffsetX"), Get("speedTextOffsetY"))
        if ns.SettingsIO then ns.SettingsIO:MarkDirty() end
    end)

    speedText = speedTextFrame:CreateFontString(nil, "OVERLAY")
    speedText:SetFont(ns.Media.ResolveFont(Get("speedFont")), Get("speedFontSize"), "OUTLINE")
    speedText:SetAllPoints()
    speedText:SetJustifyH("RIGHT")
    speedText:SetJustifyV("MIDDLE")
    speedText:SetText("")

    for i = 1, NUM_CHARGES do
        local sw = CreateFrame("StatusBar", nil, mainFrame)
        sw:SetStatusBarTexture(BAR_TEXTURE)
        sw:SetMinMaxValues(0, 1)
        sw:SetValue(0)
        sw:SetStatusBarColor(preset.secondWind.r, preset.secondWind.g, preset.secondWind.b, 0.5)
        secondWindBars[i] = sw

        local cb = CreateFrame("StatusBar", nil, mainFrame)
        cb:SetStatusBarTexture(BAR_TEXTURE)
        cb:SetMinMaxValues(0, 1)
        cb:SetValue(0)
        cb:SetFrameLevel(sw:GetFrameLevel() + 1)
        cb:SetStatusBarColor(preset.charge.r, preset.charge.g, preset.charge.b)
        chargeBars[i] = cb

    end

    local dividerFrame = CreateFrame("Frame", nil, mainFrame)
    dividerFrame:SetAllPoints()
    dividerFrame:SetFrameLevel(mainFrame:GetFrameLevel() + 5)

    for i = 1, NUM_CHARGES - 1 do
        local divider = dividerFrame:CreateTexture(nil, "OVERLAY")
        divider:SetColorTexture(0, 0, 0, 1)
        chargeDividers[i] = divider
    end

    surgeFrame = CreateFrame("Frame", nil, mainFrame)
    surgeFrame:SetFrameLevel(mainFrame:GetFrameLevel() + 2)

    local icon = C_Spell.GetSpellTexture(WHIRLING_SURGE_SPELL) or 134400
    local surgeIcon = surgeFrame:CreateTexture(nil, "ARTWORK")
    surgeIcon:SetAllPoints()
    surgeIcon:SetTexCoord(0.08, 0.92, 0.08, 0.92)
    surgeIcon:SetTexture(icon)

    surgeCooldown = CreateFrame("Cooldown", nil, surgeFrame, "CooldownFrameTemplate")
    surgeCooldown:SetPoint("TOPLEFT", surgeFrame, "TOPLEFT", 0, 1)
    surgeCooldown:SetPoint("BOTTOMRIGHT", surgeFrame, "BOTTOMRIGHT", 0, 0)
    surgeCooldown:SetDrawEdge(false)
    surgeCooldown:SetDrawBling(false)
    surgeCooldown:SetSwipeColor(0, 0, 0, 0.8)
    surgeCooldown:SetHideCountdownNumbers(true)
    surgeCooldown:SetFrameLevel(surgeFrame:GetFrameLevel() + 1)

    surgeTextFrame = CreateFrame("Frame", nil, surgeFrame)
    surgeTextFrame:SetAllPoints(surgeFrame)
    surgeTextFrame:EnableMouse(false)
    surgeTextFrame:SetFrameLevel(surgeCooldown:GetFrameLevel() + 2)

    surgeCooldownText = surgeTextFrame:CreateFontString(nil, "OVERLAY")
    surgeCooldownText:SetPoint("CENTER", surgeTextFrame, "CENTER", 0, 0)
    surgeCooldownText:SetFont(ns.Media.ResolveFont(Get("surgeIconFont")), Get("surgeIconFontSize"), "OUTLINE")
    surgeCooldownText:SetTextColor(1, 1, 1, 1)

    surgeBorder = CreateFrame("Frame", nil, surgeFrame, "BackdropTemplate")
    surgeBorder:SetFrameLevel(surgeFrame:GetFrameLevel() + 3)

    mainFrame:ClearAllPoints()
    local anchorParent = L.GetAnchorFrame(Get("anchorFrame")) or UIParent
    mainFrame:SetPoint(Get("point"), anchorParent, Get("anchorTo"), Get("posX"), Get("posY"))

    UpdateLayout()

    if Get("matchAnchorWidth") and anchorParent ~= UIParent then
        C_Timer.After(0.1, function()
            if mainFrame then
                UpdateLayout()
            end
        end)
    end
    mainFrame:Hide()
    if speedTextFrame then speedTextFrame:Hide() end
end

local function ShowPreview()
    if not mainFrame then return end
    eventFrame:SetScript("OnUpdate", nil)
    mainFrame:Show()
    mainFrame:SetAlpha(1)
    if speedTextFrame then speedTextFrame:SetShown(Get("showSpeedText")) end

    local db = NaowhQOL.dragonriding or {}
    speedBar:SetValue(0.65)
    local tR, tG, tB = W.GetEffectiveColor(db, "thrillColorR", "thrillColorG", "thrillColorB", "thrillColorUseClassColor")
    speedBar:SetStatusBarColor(tR, tG, tB)
    if Get("showSpeedText") then
        speedText:SetText("456")
    end

    local cR, cG, cB = W.GetEffectiveColor(db, "chargeColorR", "chargeColorG", "chargeColorB", "chargeColorUseClassColor")
    for i = 1, NUM_CHARGES do
        if i <= 4 then
            chargeBars[i]:SetValue(1)
        elseif i == 5 then
            chargeBars[i]:SetValue(0.6)
        else
            chargeBars[i]:SetValue(0)
        end
        chargeBars[i]:SetStatusBarColor(cR, cG, cB)
        secondWindBars[i]:SetValue(i <= 5 and 1 or 0)
    end

    lastColorState = nil
end

function ns:HideDragonridingPreview()
    if not mainFrame or not uiBuilt then return end
    prevSpeed = 0
    lastColorState = nil
    if IsEnabled() then
        RegisterDynamicEvents()
        ActivateUpdater()
    else
        mainFrame:Hide()
        mainFrame:SetAlpha(0)
        if speedTextFrame then speedTextFrame:Hide() end
        eventFrame:SetScript("OnUpdate", nil)
        UnregisterDynamicEvents()
    end
end

function ns:RefreshDragonridingLayout(showPreview)
    if not uiBuilt then
        if not C_PlayerInfo or not C_PlayerInfo.GetGlidingInfo then return end
        BuildUI()
    end
    UpdateLayout()
    if IsEnabled() then
        if mainFrame then
            mainFrame:EnableMouse(Get("unlocked"))
        end
        if showPreview then
            ShowPreview()
        end
    else
        if mainFrame then
            mainFrame:Hide()
            mainFrame:SetAlpha(0)
            mainFrame:EnableMouse(false)
        end
        if speedTextFrame then speedTextFrame:Hide() end
        eventFrame:SetScript("OnUpdate", nil)
        UnregisterDynamicEvents()
    end
end

local previewHooked = false
local function HookPreviewCleanup()
    if previewHooked then return end
    if not ns.MainFrame then return end
    previewHooked = true

    if ns.MainFrame.ResetContent then
        hooksecurefunc(ns.MainFrame, "ResetContent", function()
            ns:HideDragonridingPreview()
        end)
    end

    ns.MainFrame:HookScript("OnHide", function()
        ns:HideDragonridingPreview()
    end)
end

eventFrame = CreateFrame("Frame", "NaowhQOL_DragonridingEvents")
eventFrame:RegisterEvent("PLAYER_LOGIN")
eventFrame:RegisterEvent("PLAYER_LOGOUT")
eventFrame:RegisterEvent("PLAYER_REGEN_ENABLED")
eventFrame:RegisterEvent("PLAYER_SPECIALIZATION_CHANGED")
eventFrame:RegisterEvent("PLAYER_ENTERING_WORLD")
eventFrame:RegisterUnitEvent("UNIT_SPELLCAST_START", "player")
eventFrame:RegisterUnitEvent("UNIT_SPELLCAST_STOP", "player")

eventFrame:SetScript("OnEvent", function(self, event)
    if event == "PLAYER_LOGIN" then
        HookPreviewCleanup()
        if not C_PlayerInfo or not C_PlayerInfo.GetGlidingInfo then return end
        BuildUI()
        if IsEnabled() then
            RegisterDynamicEvents()
            ActivateUpdater()
        end

        ns.SettingsIO:RegisterRefresh("dragonriding", function()
            if ns.RefreshDragonridingLayout then
                ns:RefreshDragonridingLayout()
            end
        end)
        return
    end

    if event == "PLAYER_LOGOUT" then
        ShowCooldownManager()
        ShowBCM()
        return
    end

    if event == "PLAYER_REGEN_ENABLED" then
        if pendingCdmShow then
            pendingCdmShow = false
            ShowCooldownManager()
        elseif pendingCdmHide then
            pendingCdmHide = false
            HideCooldownManager()
        end
        if pendingBcmShow then
            pendingBcmShow = false
            ShowBCM()
        elseif pendingBcmHide then
            pendingBcmHide = false
            HideBCM()
        end
        return
    end

    if event == "PLAYER_ENTERING_WORLD" then
        if uiBuilt and IsAnchoredToCDM() then
            C_Timer.After(0.5, function()
                if mainFrame then
                    UpdateLayout()
                end
            end)
        end
        return
    end

    if event == "PLAYER_SPECIALIZATION_CHANGED" then
        if Get("matchAnchorWidth") and uiBuilt then
            C_Timer.After(0.2, function()
                if mainFrame then
                    UpdateLayout()
                end
            end)
        end
        return
    end

    if event == "UNIT_SPELLCAST_START" then
        isCasting = true
        return
    end

    if event == "UNIT_SPELLCAST_STOP" then
        isCasting = false
        return
    end

    if not uiBuilt or not IsEnabled() then return end
    ActivateUpdater()
end)

ns.Dragonriding = eventFrame
