local addon = _G.HorizonSuite
if not addon then return end

addon.Presence = addon.Presence or {}

local DEFAULTS = addon.TALKING_HEAD_DEFAULTS

-- ============================================================================
-- HELPERS
-- ============================================================================

local function GetOption(key, default)
    -- Do NOT use `addon.GetDB(...) or default`: that collapses false → default,
    -- breaking any boolean option whose stored value is false.
    if not addon.GetDB then return default end
    return addon.GetDB(key, default)
end

local FONT_USE_GLOBAL = "__global__"

local function FontPath(dbKey)
    local raw = GetOption(dbKey, FONT_USE_GLOBAL)
    if raw == FONT_USE_GLOBAL or not raw or raw == "" then
        raw = addon.GetDB and addon.GetDB("fontPath", nil) or nil
    end
    if raw and raw ~= "" and raw ~= FONT_USE_GLOBAL then
        return (addon.ResolveFontPath and addon.ResolveFontPath(raw)) or raw
    end
    return (addon.GetDefaultFontPath and addon.GetDefaultFontPath()) or "Fonts\\FRIZQT__.TTF"
end

-- Font-replacement addons hook both SetFontObject and SetFont on managed FontStrings:
-- SetFontObject re-applies their object, SetFont keeps the size but substitutes their
-- font path. Font objects created with CreateFont() cannot load WoW's virtual (CASC)
-- fonts anyway. The only reliable path: call SetFont() directly and counter-hook both
-- methods so that whenever either is overridden we re-apply our desired font.
local nativeFontState = setmetatable({}, { __mode = "k" })

local function CaptureNativeFont(fontString, fontObject)
    if not fontString or not fontObject then return end
    local path, size, flags = fontString:GetFont()
    local r, g, b, a = fontString:GetTextColor()
    nativeFontState[fontString] = {
        fontObject = fontObject,
        path = path,
        size = size,
        flags = flags,
        r = r,
        g = g,
        b = b,
        a = a,
    }
end

local function RestoreNativeFont(fontString)
    if not fontString then return end
    local state = nativeFontState[fontString]
    if not state then return end
    if state.fontObject then
        fontString:SetFontObject(state.fontObject)
    elseif state.path then
        fontString:SetFont(state.path, state.size, state.flags)
    end
    if state.r and state.g and state.b then
        fontString:SetTextColor(state.r, state.g, state.b, state.a or 1)
    end
end

local function LockDirectFont(fontString, getFont)
    local busyObj  = false
    local busyFont = false

    hooksecurefunc(fontString, "SetFontObject", function(self, obj)
        if busyObj or not obj then return end
        local path, size, flags = getFont()
        if not path then return end
        CaptureNativeFont(self, obj)
        busyObj = true
        self:SetFontObject(nil)
        self:SetFont(path, size, flags or "OUTLINE")
        self:SetShadowOffset(1, -1)
        busyObj = false
    end)

    hooksecurefunc(fontString, "SetFont", function(self, path, size, flags)
        if busyFont then return end
        local targetPath, targetSize, targetFlags = getFont()
        if not targetPath or path == targetPath then return end
        busyFont = true
        self:SetFont(targetPath, targetSize or size, targetFlags or "OUTLINE")
        self:SetShadowOffset(1, -1)
        busyFont = false
    end)
end

-- ============================================================================
-- VISUAL MODES
-- ============================================================================

local function RestoreTalkingHeadContent(frame)
    if not frame then return end
    if frame.TextFrame and frame.TextFrame.Text then
        RestoreNativeFont(frame.TextFrame.Text)
    end
    if frame.NameFrame and frame.NameFrame.Name then
        RestoreNativeFont(frame.NameFrame.Name)
    end
end

local function SetTalkingHeadSuppressed(frame, suppressed)
    if not frame then return end
    frame:SetAlpha(suppressed and 0 or 1)
    if frame.EnableMouse then
        frame:EnableMouse(not suppressed)
    end
    if frame.MainFrame then
        frame.MainFrame:SetShown(not suppressed)
        frame.MainFrame:SetAlpha(suppressed and 0 or 1)
    end
    if frame.PortraitFrame then
        frame.PortraitFrame:SetShown(not suppressed)
        frame.PortraitFrame:SetAlpha(suppressed and 0 or 1)
    end
    if frame.BackgroundFrame then
        frame.BackgroundFrame:SetAlpha(suppressed and 0 or frame.BackgroundFrame:GetAlpha())
    end
end

-- Custom fonts, sizes, and name colour on top of Blizzard's frame
local function ApplyTalkingHeadContent(frame)
    if not frame then return end
    if not GetOption("talkingHeadCustomise", DEFAULTS.talkingHeadCustomise) then
        RestoreTalkingHeadContent(frame)
        return
    end

    local textFont    = FontPath("talkingHeadTextFontPath")
    local textSize    = tonumber(GetOption("talkingHeadTextSize",    DEFAULTS.talkingHeadTextSize))    or DEFAULTS.talkingHeadTextSize
    local textOutline = GetOption("talkingHeadTextOutline", DEFAULTS.talkingHeadTextOutline) and "OUTLINE" or ""
    if frame.TextFrame and frame.TextFrame.Text then
        local fs = frame.TextFrame.Text
        -- Blizzard sets a FontObject on each PlayCurrent call; clear it so our
        -- direct SetFont call takes effect rather than being overridden by the object.
        fs:SetFontObject(nil)
        fs:SetFont(textFont, textSize, textOutline)
        fs:SetShadowOffset(1, -1)
    end

    local nameFont    = FontPath("talkingHeadNameFontPath")
    local nameSize    = tonumber(GetOption("talkingHeadNameSize",    DEFAULTS.talkingHeadNameSize))    or DEFAULTS.talkingHeadNameSize
    local nameOutline = GetOption("talkingHeadNameOutline", DEFAULTS.talkingHeadNameOutline) and "OUTLINE" or ""
    local nr = tonumber(GetOption("talkingHeadNameColorR", DEFAULTS.talkingHeadNameColorR)) or DEFAULTS.talkingHeadNameColorR
    local ng = tonumber(GetOption("talkingHeadNameColorG", DEFAULTS.talkingHeadNameColorG)) or DEFAULTS.talkingHeadNameColorG
    local nb = tonumber(GetOption("talkingHeadNameColorB", DEFAULTS.talkingHeadNameColorB)) or DEFAULTS.talkingHeadNameColorB
    if frame.NameFrame and frame.NameFrame.Name then
        local n = frame.NameFrame.Name
        n:SetFontObject(nil)
        n:SetFont(nameFont, nameSize, nameOutline)
        n:SetShadowOffset(1, -1)
        n:SetTextColor(nr, ng, nb)
    end
end

-- Frame-level toggles: portrait, background, close button, scale.
-- When customise is off, Blizzard defaults are restored (portrait on, background off, scale 1).
local function ApplyTalkingHeadFrame(frame)
    if not frame then return end

    if not GetOption("talkingHeadCustomise", DEFAULTS.talkingHeadCustomise) then
        if frame.MainFrame then
            frame.MainFrame:SetShown(true)
            frame.MainFrame:SetAlpha(1)
        end
        if frame.PortraitFrame then
            frame.PortraitFrame:SetShown(true)
            frame.PortraitFrame:SetAlpha(1)
        end
        if frame.BackgroundFrame then
            frame.BackgroundFrame:SetAlpha(1)
        end
        frame:SetScale(1.0)
        return
    end

    local showPortrait = GetOption("talkingHeadShowPortrait", DEFAULTS.talkingHeadShowPortrait)
    -- PortraitFrame is the decorative border ring, a sibling of MainFrame at TalkingHeadFrame
    -- level. It is independent of the model — hidden when portrait is off, or when portrait
    -- is on but the user has opted to hide the border. Both SetShown and SetAlpha are used:
    -- SetShown removes the frame from rendering; SetAlpha(0) guards against Blizzard's
    -- animation scripts calling Show() on the frame between our hooks.
    local showBorder = showPortrait and GetOption("talkingHeadShowPortraitBorder", DEFAULTS.talkingHeadShowPortraitBorder)
    if frame.MainFrame then
        frame.MainFrame:SetShown(showPortrait)
        frame.MainFrame:SetAlpha(showPortrait and 1 or 0)
        if frame.MainFrame.CloseButton then
            frame.MainFrame.CloseButton:SetShown(
                showPortrait and GetOption("talkingHeadCloseButton", DEFAULTS.talkingHeadCloseButton)
            )
        end
    end
    if frame.PortraitFrame then
        frame.PortraitFrame:SetShown(showBorder)
        frame.PortraitFrame:SetAlpha(showBorder and 1 or 0)
    end
    if frame.BackgroundFrame then
        frame.BackgroundFrame:SetAlpha(GetOption("talkingHeadBackground", DEFAULTS.talkingHeadBackground) and 1 or 0)
    end
    local scale = math.max(0.5, math.min(2.0, tonumber(GetOption("talkingHeadScale", DEFAULTS.talkingHeadScale)) or DEFAULTS.talkingHeadScale))
    frame:SetScale(scale)
end

local function ApplyCurrent(frame)
    ApplyTalkingHeadContent(frame)
    ApplyTalkingHeadFrame(frame)
end

-- ============================================================================
-- HOOKS — OnShow + hooksecurefunc on PlayCurrent
-- ============================================================================
--
-- We hook both OnShow and PlayCurrent for different reasons:
--   OnShow  — applies frame-level settings (portrait visibility, scale, etc.)
--             the moment the frame appears, before PlayCurrent runs. This
--             prevents a one-frame flash of the portrait when it should be
--             hidden, and covers code paths where PlayCurrent is not called.
--   PlayCurrent — applies content settings (fonts, colours) AFTER Blizzard
--             has written its own values for the new dialogue line. Blizzard
--             resets FontString fonts inside PlayCurrent on every line, so
--             OnShow alone is not sufficient for font overrides.

local _hooksInstalled = false

-- Fires after Blizzard's PlayCurrent has set dialogue text/fonts for this line.
local function OnPlayCurrent(frame)
    if addon.IsModuleEnabled and not addon:IsModuleEnabled("presence") then return end

    local enabled = GetOption("talkingHeadEnabled", DEFAULTS.talkingHeadEnabled)
    if not enabled then
        SetTalkingHeadSuppressed(frame, true)
        return
    end

    SetTalkingHeadSuppressed(frame, false)
    ApplyCurrent(frame)

    if GetOption("talkingHeadMuteVoice", DEFAULTS.talkingHeadMuteVoice) and frame.voHandle then
        StopSound(frame.voHandle)
    end
end

local function InstallHooks(frame)
    if _hooksInstalled then return end
    _hooksInstalled = true
    hooksecurefunc(frame, "PlayCurrent", OnPlayCurrent)
    frame:HookScript("OnShow", function(self)
        if addon.IsModuleEnabled and not addon:IsModuleEnabled("presence") then return end
        if not GetOption("talkingHeadEnabled", DEFAULTS.talkingHeadEnabled) then
            SetTalkingHeadSuppressed(self, true)
            return
        end
        SetTalkingHeadSuppressed(self, false)
        ApplyTalkingHeadFrame(self)
    end)
    if frame.NameFrame and frame.NameFrame.Name then
        LockDirectFont(frame.NameFrame.Name, function()
            if not GetOption("talkingHeadCustomise", DEFAULTS.talkingHeadCustomise) then return end
            return FontPath("talkingHeadNameFontPath"),
                   tonumber(GetOption("talkingHeadNameSize", DEFAULTS.talkingHeadNameSize)) or DEFAULTS.talkingHeadNameSize,
                   GetOption("talkingHeadNameOutline", DEFAULTS.talkingHeadNameOutline) and "OUTLINE" or ""
        end)
    end
    if frame.TextFrame and frame.TextFrame.Text then
        LockDirectFont(frame.TextFrame.Text, function()
            if not GetOption("talkingHeadCustomise", DEFAULTS.talkingHeadCustomise) then return end
            return FontPath("talkingHeadTextFontPath"),
                   tonumber(GetOption("talkingHeadTextSize", DEFAULTS.talkingHeadTextSize)) or DEFAULTS.talkingHeadTextSize,
                   GetOption("talkingHeadTextOutline", DEFAULTS.talkingHeadTextOutline) and "OUTLINE" or ""
        end)
    end
    -- Late-install catch: first dialogue already showing when hooks were wired
    if frame:IsShown() then
        if GetOption("talkingHeadEnabled", DEFAULTS.talkingHeadEnabled) then
            SetTalkingHeadSuppressed(frame, false)
            ApplyCurrent(frame)
        else
            SetTalkingHeadSuppressed(frame, true)
        end
    end
end

-- ============================================================================
-- SETUP — Three timing windows for when TalkingHeadFrame becomes available:
--   1. File load:     TalkingHeadFrame already exists (e.g. after /reload).
--   2. PLAYER_LOGIN:  Blizzard addons fully initialised; covers most fresh logins.
--   3. TALKINGHEAD_REQUESTED: frame created on first TH if it wasn't ready earlier.
-- ============================================================================

local _setupFrame = CreateFrame("Frame")
_setupFrame:RegisterEvent("TALKINGHEAD_REQUESTED")
_setupFrame:RegisterEvent("PLAYER_LOGIN")
_setupFrame:SetScript("OnEvent", function(self, event)
    if event == "PLAYER_LOGIN" then
        -- Always drop PLAYER_LOGIN on fire. If frame is still absent, case 3 covers it.
        self:UnregisterEvent("PLAYER_LOGIN")
    end
    local frame = _G.TalkingHeadFrame
    if frame then
        InstallHooks(frame)
        self:UnregisterEvent("TALKINGHEAD_REQUESTED")
        self:UnregisterEvent("PLAYER_LOGIN")
    end
end)

-- Case 1: frame exists right now (e.g. /reload).
if _G.TalkingHeadFrame then
    InstallHooks(_G.TalkingHeadFrame)
    _setupFrame:UnregisterEvent("TALKINGHEAD_REQUESTED")
    _setupFrame:UnregisterEvent("PLAYER_LOGIN")
end

-- ============================================================================
-- PREVIEW WIDGET (options dashboard)
-- ============================================================================

local PREVIEW_HEIGHT     = 110
local PREVIEW_PORTRAIT_W = 110

function addon.Presence.CreateTalkingHeadPreviewWidget(parent)
    if not parent then return nil end

    local L = addon.L

    local frame = CreateFrame("Frame", nil, parent)
    frame:SetHeight(PREVIEW_HEIGHT)

    local bg = frame:CreateTexture(nil, "BACKGROUND")
    bg:SetAllPoints(frame)
    bg:SetColorTexture(0.05, 0.05, 0.07, 0.92)

    if addon.CreateBorder then
        addon.CreateBorder(frame, { 0.18, 0.18, 0.24, 0.95 })
    end

    -- Plain Frame container for the portrait area. We show/hide this rather than
    -- the PlayerModel directly: PlayerModel:SetShown(false) can be overridden when
    -- the model finishes loading asynchronously, whereas hiding a plain parent
    -- Frame makes all its children effectively invisible regardless of their own state.
    local portraitArea = CreateFrame("Frame", nil, frame)
    portraitArea:SetWidth(PREVIEW_PORTRAIT_W)
    portraitArea:SetPoint("TOPLEFT", frame, "TOPLEFT", 0, 0)
    portraitArea:SetPoint("BOTTOMLEFT", frame, "BOTTOMLEFT", 0, 0)

    local portraitBg = portraitArea:CreateTexture(nil, "BACKGROUND", nil, -1)
    portraitBg:SetAllPoints(portraitArea)
    portraitBg:SetColorTexture(0.03, 0.03, 0.05, 0.95)

    local portrait = CreateFrame("PlayerModel", nil, portraitArea)
    portrait:SetPoint("TOPLEFT", portraitArea, "TOPLEFT", 2, -2)
    portrait:SetPoint("BOTTOMRIGHT", portraitArea, "BOTTOMRIGHT", -2, 2)
    portrait:SetUnit("player")
    portrait:SetCamDistanceScale(0.85)
    portrait:SetPortraitZoom(1)

    local sep = frame:CreateTexture(nil, "ARTWORK")
    sep:SetWidth(1)
    sep:SetPoint("TOP", frame, "TOP", 0, -6)
    sep:SetPoint("BOTTOM", frame, "BOTTOM", 0, 6)
    sep:SetPoint("LEFT", portraitArea, "RIGHT", 0, 0)
    sep:SetColorTexture(0.25, 0.28, 0.35, 0.8)

    local nameText = frame:CreateFontString(nil, "OVERLAY")
    nameText:SetPoint("TOPLEFT", frame, "TOPLEFT", PREVIEW_PORTRAIT_W + 12, -10)
    nameText:SetPoint("RIGHT", frame, "RIGHT", -8, 0)
    nameText:SetJustifyH("LEFT")

    local dialogueText = frame:CreateFontString(nil, "OVERLAY")
    dialogueText:SetPoint("TOPLEFT", nameText, "BOTTOMLEFT", 0, -8)
    dialogueText:SetPoint("RIGHT", frame, "RIGHT", -8, 0)
    dialogueText:SetJustifyH("LEFT")
    dialogueText:SetNonSpaceWrap(true)

    LockDirectFont(nameText, function()
        return FontPath("talkingHeadNameFontPath"),
               tonumber(GetOption("talkingHeadNameSize", DEFAULTS.talkingHeadNameSize)) or DEFAULTS.talkingHeadNameSize,
               GetOption("talkingHeadNameOutline", DEFAULTS.talkingHeadNameOutline) and "OUTLINE" or ""
    end)
    LockDirectFont(dialogueText, function()
        return FontPath("talkingHeadTextFontPath"),
               tonumber(GetOption("talkingHeadTextSize", DEFAULTS.talkingHeadTextSize)) or DEFAULTS.talkingHeadTextSize,
               GetOption("talkingHeadTextOutline", DEFAULTS.talkingHeadTextOutline) and "OUTLINE" or ""
    end)

    local function Refresh()
        local nameFont = FontPath("talkingHeadNameFontPath")
        local nameSize = tonumber(GetOption("talkingHeadNameSize", DEFAULTS.talkingHeadNameSize)) or DEFAULTS.talkingHeadNameSize
        local nameOutline = GetOption("talkingHeadNameOutline", DEFAULTS.talkingHeadNameOutline) and "OUTLINE" or ""
        nameText:SetFontObject(nil)  -- clear any inherited FontObject so SetFont applies directly
        nameText:SetFont(nameFont, nameSize, nameOutline)
        nameText:SetShadowOffset(1, -1)
        nameText:SetText(UnitName("player") or L["TALKING_HEAD_PREVIEW_NPC_NAME"])
        local nr = tonumber(GetOption("talkingHeadNameColorR", DEFAULTS.talkingHeadNameColorR)) or DEFAULTS.talkingHeadNameColorR
        local ng = tonumber(GetOption("talkingHeadNameColorG", DEFAULTS.talkingHeadNameColorG)) or DEFAULTS.talkingHeadNameColorG
        local nb = tonumber(GetOption("talkingHeadNameColorB", DEFAULTS.talkingHeadNameColorB)) or DEFAULTS.talkingHeadNameColorB
        nameText:SetTextColor(nr, ng, nb)

        local textFont = FontPath("talkingHeadTextFontPath")
        local textSize = tonumber(GetOption("talkingHeadTextSize", DEFAULTS.talkingHeadTextSize)) or DEFAULTS.talkingHeadTextSize
        local textOutline = GetOption("talkingHeadTextOutline", DEFAULTS.talkingHeadTextOutline) and "OUTLINE" or ""
        dialogueText:SetFontObject(nil)
        dialogueText:SetFont(textFont, textSize, textOutline)
        dialogueText:SetShadowOffset(1, -1)
        dialogueText:SetText(L["TALKING_HEAD_PREVIEW_DIALOGUE"])

        local showPortrait = GetOption("talkingHeadShowPortrait", DEFAULTS.talkingHeadShowPortrait)
        portraitArea:SetShown(showPortrait)
        sep:SetShown(showPortrait)
    end

    frame.Refresh = Refresh
    frame:SetScript("OnShow", Refresh)
    Refresh()

    return { frame = frame, Refresh = Refresh }
end

-- ============================================================================
-- INIT / UPDATE (public API)
-- ============================================================================

function addon.Presence.InitTalkingHead()
    local frame = _G.TalkingHeadFrame
    if frame then InstallHooks(frame) end
end

function addon.Presence.UpdateTalkingHead()
    local frame = _G.TalkingHeadFrame
    if not frame then return end
    InstallHooks(frame)

    local enabled = GetOption("talkingHeadEnabled", DEFAULTS.talkingHeadEnabled)
    if not enabled then
        if frame:IsShown() then SetTalkingHeadSuppressed(frame, true) end
        return
    end

    -- Only update a visible frame; never force-show (Blizzard owns visibility)
    if frame:IsShown() then
        SetTalkingHeadSuppressed(frame, false)
        ApplyCurrent(frame)
    end
end
