local addonName, ns = ...
local L = ns.L

local COLORS = ns.COLORS

local lastCheckedZone = nil

local function ChatMsg(text)
    DEFAULT_CHAT_FRAME:AddMessage("|cff" .. COLORS.BLUE .. "Naowh QOL:|r " .. text)
end

local function GetSpecID()
    return PlayerUtil and PlayerUtil.GetCurrentSpecID() or 0
end

local function GetSpecName()
    local specIndex = GetSpecialization()
    if specIndex then
        local _, name = GetSpecializationInfo(specIndex)
        return name or "Unknown"
    end
    return "Unknown"
end

local function IsTLXAvailable()
    local success, result = pcall(function()
        return C_AddOns.IsAddOnLoaded("TalentLoadoutsEx")
            and _G.TLX ~= nil
            and _G.TalentLoadoutEx ~= nil
    end)
    return success and result
end

local function GetTLXLoadouts()
    if not IsTLXAvailable() then return nil end

    local success, result = pcall(function()
        local _, englishClass = UnitClass("player")
        local specIndex = GetSpecialization()
        if not englishClass or not specIndex then return nil end

        local tlxData = _G.TalentLoadoutEx
        if not tlxData or not tlxData[englishClass] then return nil end
        local specTable = tlxData[englishClass][specIndex]
        if not specTable then return nil end

        local loadouts = {}
        for _, entry in ipairs(specTable) do
            if entry.text and not entry.isLegacy then
                table.insert(loadouts, entry)
            end
        end
        return #loadouts > 0 and loadouts or nil
    end)

    return success and result or nil
end

local function GetTLXLoadoutByName(name)
    local loadouts = GetTLXLoadouts()
    if not loadouts then return nil end
    for _, loadout in ipairs(loadouts) do
        if loadout.name == name then return loadout end
    end
    return nil
end

local function GetCurrentTLXLoadout()
    if not IsTLXAvailable() then return nil, nil end

    local success, result = pcall(function()
        local tlx = _G.TLX
        if not tlx or not tlx.GetLoadedData then return nil end
        local loaded = { tlx.GetLoadedData() }
        if loaded[1] then
            return { name = loaded[1].name, text = loaded[1].text }
        end
        return nil
    end)

    if success and result then
        return result.name, result.text
    end
    return nil, nil
end

local function SwapToTLXLoadout(loadoutName)
    if InCombatLockdown() then
        ChatMsg("|cff" .. COLORS.ERROR .. L["TALENT_COMBAT_ERROR"] .. "|r")
        return false
    end
    local loadout = GetTLXLoadoutByName(loadoutName)
    if not loadout then
        ChatMsg("|cff" .. COLORS.ERROR .. L["TALENT_NOT_FOUND"] .. "|r")
        return false
    end
    C_Timer.After(0, function()
        if InCombatLockdown() then return end
        local success = pcall(function()
            local slashHandler = SlashCmdList["TalentLoadoutsEx_Load"]
            if slashHandler then
                slashHandler(loadoutName)
            end
        end)
        if not success then
            local editBox = ChatFrame1 and ChatFrame1.editBox
            if editBox and ChatEdit_SendText then
                ChatEdit_ActivateChat(editBox)
                editBox:SetText("/tlx " .. loadoutName)
                ChatEdit_SendText(editBox)
            end
        end
    end)
    ChatMsg(string.format(L["TALENT_SWAPPED"], "|cff" .. COLORS.ORANGE .. loadoutName .. "|r"))
    return true
end

local function GetCurrentTalentInfo()
    if IsTLXAvailable() then
        local tlxName, tlxExport = GetCurrentTLXLoadout()
        if tlxName then
            return tlxName, tlxExport, tlxName
        else
            local activeConfigID = C_ClassTalents.GetActiveConfigID()
            local exportString = activeConfigID and C_Traits.GenerateImportString(activeConfigID)
            return nil, exportString, "Unsaved Build"
        end
    end

    local specID = GetSpecID()
    local activeConfigID = C_ClassTalents.GetActiveConfigID()
    if not activeConfigID then return nil, nil, nil end

    local savedConfigID = C_ClassTalents.GetLastSelectedSavedConfigID(specID)
    local configName = "Unsaved Build"

    if savedConfigID then
        local configInfo = C_Traits.GetConfigInfo(savedConfigID)
        if configInfo and configInfo.name then
            configName = configInfo.name
        end
    end

    local exportString = C_Traits.GenerateImportString(activeConfigID)
    return savedConfigID or activeConfigID, exportString, configName
end

local function GetConfigName(configID)
    local info = C_Traits.GetConfigInfo(configID)
    return info and info.name or "Unknown"
end

local function BuildDungeonKey(specID, instanceID, difficulty)
    return specID .. ":" .. instanceID .. ":" .. difficulty
end

local function SwapToSaved(saved)
    if InCombatLockdown() then
        ChatMsg("|cff" .. COLORS.ERROR .. L["TALENT_COMBAT_ERROR"] .. "|r")
        return false
    end

    if saved.tlxMode and saved.tlxName then
        return SwapToTLXLoadout(saved.tlxName)
    end

    local specID = GetSpecID()
    local configs = C_ClassTalents.GetConfigIDsBySpecID(specID) or {}
    for index, id in ipairs(configs) do
        if id == saved.configID then
            if ClassTalentHelper and ClassTalentHelper.SwitchToLoadoutByIndex then
                ClassTalentHelper.SwitchToLoadoutByIndex(index)
                ChatMsg(string.format(
                    L["TALENT_SWAPPED"],
                    "|cff" .. COLORS.ORANGE .. (saved.configName or "saved build") .. "|r"
                ))
                return true
            end
        end
    end
    ChatMsg("|cff" .. COLORS.ERROR .. L["TALENT_NOT_FOUND"] .. "|r")
    return false
end

StaticPopupDialogs["NAOWHQOL_TALENT_SAVE"] = {
    text = "%s",
    button1 = L["TALENT_SAVE_BTN"],
    button2 = L["COMBATLOGGER_SKIP_BTN"],
    OnAccept = function(self)
        local data = self.data
        if not data then return end

        local db = NaowhQOL.talentReminder
        if not db then return end
        db.loadouts = db.loadouts or {}

        if IsTLXAvailable() then
            local tlxName, tlxExport = GetCurrentTLXLoadout()
            if tlxName then
                db.loadouts[data.key] = {
                    tlxMode = true,
                    tlxName = tlxName,
                    exportString = tlxExport,
                    name = data.name,
                    diffName = data.diffName,
                }
                ChatMsg(string.format(
                    L["TALENT_SAVED"],
                    "|cff" .. COLORS.ORANGE .. data.name .. "|r"
                ))
                return
            else
                ChatMsg("|cff" .. COLORS.ERROR .. "No TLX loadout active. Select a loadout first.|r")
                return
            end
        end

        local configID, exportString, configName = GetCurrentTalentInfo()
        db.loadouts[data.key] = {
            configID = configID,
            exportString = exportString,
            configName = configName,
            name = data.name,
            diffName = data.diffName,
        }

        ChatMsg(string.format(
            L["TALENT_SAVED"],
            "|cff" .. COLORS.ORANGE .. data.name .. "|r"
        ))
    end,
    timeout = 0,
    whileDead = false,
    hideOnEscape = true,
    preferredIndex = 3,
}

StaticPopupDialogs["NAOWHQOL_TALENT_MISMATCH"] = {
    text = "%s",
    button1 = L["TALENT_SWAP_BTN"],
    button2 = L["TALENT_OVERWRITE_BTN"],
    button3 = L["TALENT_IGNORE_BTN"],
    OnAccept = function(self)
        local data = self.data
        if not data or not data.saved then return end
        SwapToSaved(data.saved)
    end,
    OnCancel = function(self)
        local data = self.data
        if not data then return end

        local db = NaowhQOL.talentReminder
        if not db then return end
        db.loadouts = db.loadouts or {}

        if IsTLXAvailable() then
            local tlxName, tlxExport = GetCurrentTLXLoadout()
            if tlxName then
                db.loadouts[data.key] = {
                    tlxMode = true,
                    tlxName = tlxName,
                    exportString = tlxExport,
                    name = data.name,
                    diffName = data.diffName,
                }
                ChatMsg(string.format(
                    L["TALENT_OVERWRITTEN"],
                    "|cff" .. COLORS.ORANGE .. data.name .. "|r"
                ))
                return
            else
                ChatMsg("|cff" .. COLORS.ERROR .. "No TLX loadout active. Select a loadout first.|r")
                return
            end
        end

        local configID, exportString, configName = GetCurrentTalentInfo()
        db.loadouts[data.key] = {
            configID = configID,
            exportString = exportString,
            configName = configName,
            name = data.name,
            diffName = data.diffName,
        }

        ChatMsg(string.format(
            L["TALENT_OVERWRITTEN"],
            "|cff" .. COLORS.ORANGE .. data.name .. "|r"
        ))
    end,
    OnAlt = function() end,
    timeout = 0,
    whileDead = false,
    hideOnEscape = true,
    preferredIndex = 3,
}

StaticPopupDialogs["NAOWHQOL_TALENT_TLX_UNAVAILABLE"] = {
    text = "%s",
    button1 = L["TALENT_OVERWRITE_BTN"],
    button2 = L["TALENT_IGNORE_BTN"],
    OnAccept = function(self)
        local data = self.data
        if not data then return end

        local db = NaowhQOL.talentReminder
        if not db then return end
        db.loadouts = db.loadouts or {}

        if IsTLXAvailable() then
            local tlxName, tlxExport = GetCurrentTLXLoadout()
            if tlxName then
                db.loadouts[data.key] = {
                    tlxMode = true,
                    tlxName = tlxName,
                    exportString = tlxExport,
                    name = data.name,
                    diffName = data.diffName,
                }
                ChatMsg(string.format(
                    L["TALENT_OVERWRITTEN"],
                    "|cff" .. COLORS.ORANGE .. data.name .. "|r"
                ))
                return
            end
        end

        local configID, exportString, configName = GetCurrentTalentInfo()
        db.loadouts[data.key] = {
            configID = configID,
            exportString = exportString,
            configName = configName,
            name = data.name,
            diffName = data.diffName,
        }
        ChatMsg(string.format(
            L["TALENT_OVERWRITTEN"],
            "|cff" .. COLORS.ORANGE .. data.name .. "|r"
        ))
    end,
    OnCancel = function() end,
    timeout = 0,
    whileDead = false,
    hideOnEscape = true,
    preferredIndex = 3,
}

local function CheckTalents(key, displayName, diffName)
    if InCombatLockdown() then return end

    local db = NaowhQOL.talentReminder
    if not db or not db.enabled then return end
    db.loadouts = db.loadouts or {}

    local saved = db.loadouts[key]
    local configID, exportString, configName = GetCurrentTalentInfo()

    if not saved then
        if IsTLXAvailable() then
            local tlxName = GetCurrentTLXLoadout()
            if not tlxName then
                ChatMsg(string.format("No TLX loadout active for %s. Select a loadout first.",
                    "|cff" .. COLORS.ORANGE .. displayName .. "|r"))
                return
            end
        end

        local promptText = "|cff" .. COLORS.BLUE .. "Naowh QOL|r\n\n"
            .. string.format(
                L["TALENT_SAVE_POPUP"],
                "|cff" .. COLORS.ORANGE .. displayName .. "|r",
                diffName and ("(" .. diffName .. ")") or "",
                "(" .. GetSpecName() .. ")",
                "|cff" .. COLORS.SUCCESS .. configName .. "|r"
            )

        local dialog = StaticPopup_Show("NAOWHQOL_TALENT_SAVE", promptText)
        if dialog then
            dialog.data = {
                key = key,
                name = displayName,
                diffName = diffName,
            }
        end
    else
        local isMismatch = false
        local savedDisplayName = saved.configName or saved.tlxName or "Saved Build"

        if saved.tlxMode then
            local tlxUnavailableReason = nil

            if not IsTLXAvailable() then
                tlxUnavailableReason = "TalentLoadoutsEx not loaded"
            else
                local tlxLoadout = GetTLXLoadoutByName(saved.tlxName)
                if not tlxLoadout then
                    tlxUnavailableReason = "Loadout '" .. saved.tlxName .. "' not found"
                end
            end

            if tlxUnavailableReason then
                local promptText = "|cff" .. COLORS.BLUE .. "Naowh QOL|r\n\n"
                    .. "|cff" .. COLORS.ERROR .. tlxUnavailableReason .. "|r\n\n"
                    .. "Saved TLX loadout for:\n"
                    .. "|cff" .. COLORS.ORANGE .. displayName .. "|r\n\n"
                    .. "Current: |cff" .. COLORS.SUCCESS .. configName .. "|r"

                local dialog = StaticPopup_Show("NAOWHQOL_TALENT_TLX_UNAVAILABLE", promptText)
                if dialog then
                    dialog.data = {
                        key = key,
                        name = displayName,
                        diffName = diffName,
                    }
                end
                return
            end

            local currentTLX = GetCurrentTLXLoadout()
            isMismatch = (currentTLX ~= saved.tlxName)
            savedDisplayName = saved.tlxName
        else
            isMismatch = (saved.configID ~= configID)
        end

        if isMismatch then
            local promptText = "|cff" .. COLORS.BLUE .. "Naowh QOL|r\n\n"
                .. string.format(
                    L["TALENT_MISMATCH_POPUP"],
                    "|cff" .. COLORS.ORANGE .. displayName .. "|r",
                    "|cff" .. COLORS.ERROR .. configName .. "|r",
                    "|cff" .. COLORS.SUCCESS .. savedDisplayName .. "|r"
                )

            local dialog = StaticPopup_Show("NAOWHQOL_TALENT_MISMATCH", promptText)
            if dialog then
                dialog.data = {
                    key = key,
                    name = displayName,
                    diffName = diffName,
                    saved = saved,
                }
            end
        end
    end
end

local function OnZoneChanged(zoneData)
    if InCombatLockdown() then return end

    local db = NaowhQOL.talentReminder
    if not db or not db.enabled then return end

    if zoneData.instanceType ~= "party" or zoneData.difficulty ~= 23 then
        lastCheckedZone = nil
        return
    end

    local specID = GetSpecID()
    if specID == 0 then return end

    local key = BuildDungeonKey(specID, zoneData.instanceID, zoneData.difficulty)

    if lastCheckedZone == key then return end
    lastCheckedZone = key

    C_Timer.After(1, function()
        if InCombatLockdown() then return end
        CheckTalents(key, zoneData.zoneName, zoneData.difficultyName)
    end)
end

local loader = CreateFrame("Frame", "NaowhQOL_TalentReminder")
loader:RegisterEvent("PLAYER_LOGIN")
loader:RegisterEvent("PLAYER_SPECIALIZATION_CHANGED")
loader:RegisterEvent("TRAIT_CONFIG_UPDATED")

loader:SetScript("OnEvent", function(self, event, ...)
    if event == "PLAYER_LOGIN" then
        if not NaowhQOL.talentReminder then
            NaowhQOL.talentReminder = { enabled = true, loadouts = {} }
        end
        local db = NaowhQOL.talentReminder
        if db.enabled == nil then db.enabled = true end
        db.loadouts = db.loadouts or {}

        if ns.ZoneUtil and ns.ZoneUtil.RegisterCallback then
            ns.ZoneUtil.RegisterCallback("TalentReminder", OnZoneChanged)
            C_Timer.After(1, function()
                if not InCombatLockdown() then
                    OnZoneChanged(ns.ZoneUtil.GetCurrentZone())
                end
            end)
        end

        self:UnregisterEvent("PLAYER_LOGIN")

    elseif event == "PLAYER_SPECIALIZATION_CHANGED" then
        lastCheckedZone = nil

    elseif event == "TRAIT_CONFIG_UPDATED" then
        lastCheckedZone = nil
        C_Timer.After(0.5, function()
            if InCombatLockdown() then return end
            if ns.ZoneUtil then
                OnZoneChanged(ns.ZoneUtil.GetCurrentZone())
            end
        end)
    end
end)

function loader:TriggerZoneCheck()
    lastCheckedZone = nil

    C_Timer.After(0.5, function()
        if InCombatLockdown() then return end
        if ns.ZoneUtil then
            OnZoneChanged(ns.ZoneUtil.GetCurrentZone())
        end
    end)
end

ns.TalentReminder = loader
