--- @class DialogKeyNS ## Version: v1.2.45 ## Author: Numy (previous verions by: Foxthorn, N01ch, FuriousProgrammer)
local DialogKeyNS = {}
local issecretvalue = issecretvalue or function(v) return false end

if GetLocale() == "zhCN" then
  DialogKeyLocal = "|cffe6cc80[弹窗]|r快捷对话";
elseif GetLocale() == "zhTW" then
  DialogKeyLocal = "|cffe6cc80[弹窗]|r快捷对话";
else
  DialogKeyLocal = "DialogKey";
end

DialogKeyNS.configPanelName = DialogKeyLocal
DialogKeyNS.defaultOptions = {
    keys = {
        "SPACE",
    },
    ignoreDisabledButtons = true,
    ignoreWithModifier = true,
    showGlow = false,
    dialogBlacklist = {},
    customFrames = {},
    numKeysForGossip = true,
    numKeysForQuestRewards = true,
    dontClickSummons = true,
    dontClickDuels = true,
    dontClickRevives = true,
    dontClickReleases = true,
    dontAcceptInvite = true,
    dontAcceptAbandonVote = true,
    dontAcceptVoteKick = true,
    dontAcceptInstanceLocks = false,
    useSoulstoneRez = true,
    handleCraftingOrders = true,
    handlePlayerChoice = true,
    numKeysForPlayerChoice = true,
    handleSpecFrame = true,
    postAuctions = true,
    ignoreInProgressQuests = true,
}

--- @param dialogKey DialogKey
function DialogKeyNS:InitDB(dialogKey)
    for k, v in pairs(self.defaultOptions) do
        if dialogKey.db[k] == nil then dialogKey.db[k] = v end
    end
    if select(2, next(dialogKey.db.customFrames)) == true then -- unordered list
        local i = 1
        for k, _ in pairs(dialogKey.db.customFrames) do
            dialogKey.db.customFrames[k] = i
            i = i + 1
        end
    end

    -- check for gaps
    self:GuardAgainstGaps()
end

function DialogKeyNS:GuardAgainstGaps()
    local db = self.Core.db
    local count = 0
    for _, _ in pairs(db.customFrames) do
        count = count + 1
    end
    DialogKeyNS.orderedCustomFrames = tInvert(db.customFrames)
    local orderedCount = 0
    for _, _ in ipairs(DialogKeyNS.orderedCustomFrames) do -- ipairs stops at the first gap
        orderedCount = orderedCount + 1
    end
    if count ~= orderedCount then
        for i = 1, count do
            if DialogKeyNS.orderedCustomFrames[i] == nil then
                local j = i + 1
                while j < i+100 do -- exit after 100, to avoid infinite loops in case of issues
                    if DialogKeyNS.orderedCustomFrames[j] then
                        DialogKeyNS.orderedCustomFrames[i] = DialogKeyNS.orderedCustomFrames[j]
                        DialogKeyNS.orderedCustomFrames[j] = nil
                        db.customFrames[DialogKeyNS.orderedCustomFrames[i]] = i
                        break
                    end
                    j = j + 1
                end
            end
        end
        DialogKeyNS.orderedCustomFrames = tInvert(db.customFrames)
    end
end

function DialogKeyNS:RemoveFromWatchlist(frameName)
    local db = self.Core.db
    local index = db.customFrames[frameName]
    if not index then return end
    tremove(self.orderedCustomFrames, index)
    db.customFrames[frameName] = nil

    for i = index, #self.orderedCustomFrames do
        db.customFrames[self.orderedCustomFrames[i]] = i
    end
    self:RegisterOptions()
end

function DialogKeyNS:AddToWatchlist(frameName)
    local db = self.Core.db
    if db.customFrames[frameName] then return end
    tinsert(self.orderedCustomFrames, frameName)
    db.customFrames[frameName] = #self.orderedCustomFrames
    self:RegisterOptions()
end

local arrowNext = C_Texture.GetAtlasInfo('common-dropdown-icon-next') or error('Failed to find atlas for Next icon')
local arrowBack = C_Texture.GetAtlasInfo('common-dropdown-icon-back') or error('Failed to find atlas for Back icon')
local cross = C_Texture.GetAtlasInfo('Radial_Wheel_Icon_Close') or C_Texture.GetAtlasInfo('XMarksTheSpot') or error('Failed to find atlas for cross icon')
local search = C_Texture.GetAtlasInfo('common-search-magnifyingglass') or error('Failed to find atlas for Search icon')
local width_multiplier = 170

-- only want this for toggles
local function wrapName(name)
    return "|cffffd100" .. name .. "|r"
end

------- hack to allow tooltips to work on nameless execute icons
local MAGIC_TOOLTIP_TEXTS = {
    up = 'DialogKey_Numy_MagicTooltipUp',
    down = 'DialogKey_Numy_MagicTooltipDown',
    remove = 'DialogKey_Numy_MagicTooltipRemove',
    glow = 'DialogKey_Numy_MagicTooltipGlow',
}
do
    local lastSetOwnerCall
    local ACDTooltip = LibStub("AceConfigDialog-3.0").tooltip
    hooksecurefunc(ACDTooltip, 'SetOwner', function(_, ...)
        lastSetOwnerCall = {...}
    end)
    hooksecurefunc(ACDTooltip, 'AddLine', function(tooltip, text, r, g, b, wrap)
        local title, desc
        if text == MAGIC_TOOLTIP_TEXTS.up then
            title = "上移"
            desc = "在优先级列表中将框架向上移动"
        elseif text == MAGIC_TOOLTIP_TEXTS.down then
            title = "下移"
            desc = "在优先级列表中将框架向下移动"
        elseif text == MAGIC_TOOLTIP_TEXTS.remove then
            title = "移除"
            desc = "从自定义监视列表中移除框架"
        elseif text == MAGIC_TOOLTIP_TEXTS.glow then
            title = "查找"
            desc = "在框架周围显示短暂的发光效果，使其更易找到"
        end
        if title then
            -- setting text to an empty string seems to clear the owner and effectively resets the tooltip :/
            tooltip:SetOwner(unpack(lastSetOwnerCall))
            tooltip:SetText(title, 1, .82, 0, true)
            tooltip:AddLine(desc, r, g, b, wrap)
        end
    end)
end

local initialized = false
function DialogKeyNS:RegisterOptions()
    if not initialized then
        initialized = true
        LibStub("AceConfig-3.0"):RegisterOptionsTable(self.configPanelName, function() return self:GetOptionsTable() end)
        local _, categoryID = LibStub("AceConfigDialog-3.0"):AddToBlizOptions(self.configPanelName)
        self.categoryID = categoryID
    else
        LibStub("AceConfigRegistry-3.0"):NotifyChange(self.configPanelName)
    end
end

function DialogKeyNS:OpenConfig()
    if C_SettingsUtil and C_SettingsUtil.OpenSettingsPanel and InCombatLockdown() then
        LibStub("AceConfigDialog-3.0"):Open(self.configPanelName);
        return;
    end
    Settings.OpenToCategory(self.categoryID)
end

function DialogKeyNS:GetOptionsTable()
    local db = self.Core.db
    local function optionSetter(info, val) db[info[#info]] = val end
    local function optionGetter(info) return db[info[#info]] end
    local function swapWatchlistFrames(index1, index2)
        local frame1 = self.orderedCustomFrames[index1]
        local frame2 = self.orderedCustomFrames[index2]
        if not frame1 or not frame2 then return end

        self.orderedCustomFrames[index1] = frame2
        self.orderedCustomFrames[index2] = frame1
        db.customFrames[frame1] = index2
        db.customFrames[frame2] = index1

        self:RegisterOptions()
    end

    local increment = CreateCounter()
    local options = {
        type = "group",
        set = optionSetter,
        get = optionGetter,
        name = self.configPanelName,
        args = {
            header1 = {
                order = increment(),
                name = "主要按键绑定",
                type = "header",
            },
            key1 = {
                order = increment(),
                name = "",
                type = "keybinding",
                set = (function(_, val) db.keys[1] = val end),
                get = (function(_) return db.keys[1] end),
            },
            key2 = {
                order = increment(),
                name = "",
                type = "keybinding",
                set = (function(_, val) db.keys[2] = val end),
                get = (function(_) return db.keys[2] end),
            },
            header2 = {
                order = increment(),
                name = "选项",
                type = "header",
            },
            generalGroup = {
                order = increment(),
                name = "通用",
                desc = "设置基本的个人偏好选项",
                type = "group",
                args = {
                    showGlow = {
                        order = increment(),
                        name = wrapName("启用发光"),
                        desc = "当按钮被快捷点击时显示发光效果。",
                        descStyle = "inline", width = "full", type = "toggle",
                    },
                    ignoreWithModifier = {
                        order = increment(),
                        name = wrapName("忽略带有控制键的DialogKey"),
                        desc = "当任何控制键（Shift、Alt、Ctrl）被按下时禁用快捷点击。",
                        descStyle = "inline", width = "full", type = "toggle",
                    },
                    ignoreDisabledButtons = {
                        order = increment(),
                        name = wrapName("忽略禁用的按钮"),
                        desc = "不允许快捷点击禁用的（灰色的）按钮。",
                        descStyle = "inline", width = "full", type = "toggle",
                    },
                    ignoreInProgressQuests = {
                        order = increment(),
                        name = wrapName("忽略进行中的任务"),
                        desc = "忽略进行中任务的对话选项，只点击已完成或未接受的任务",
                        descStyle = "inline", width = "full", type = "toggle",
                    },
                    numKeysForGossip = {
                        order = increment(),
                        name = wrapName("为对话选项设置数字键"),
                        desc = "使用数字键（1到0）从NPC对话窗口中选择对话选项或任务。",
                        descStyle = "inline", width = "full", type = "toggle",
                    },
                    numKeysForQuestRewards = {
                        order = increment(),
                        name = wrapName("为任务奖励设置数字键"),
                        desc = "当有多个任务奖励可供选择时，使用数字键（1到0）来选择任务奖励。",
                        descStyle = "inline", width = "full", type = "toggle",
                    },
                    postAuctions = {
                        order = increment(),
                        name = wrapName("拍卖邮件"),
                        desc = "拍卖邮件",
                        descStyle = "inline", width = "full", type = "toggle",
                    },
                    handleCraftingOrders = {
                        order = increment(),
                        name = wrapName("制造订单"),
                        desc = "处理制造订单：启动、制作、完成。",
                        descStyle = "inline", width = "full", type = "toggle",
                        hidden = WOW_PROJECT_ID ~= WOW_PROJECT_MAINLINE,
                    },
                    handlePlayerChoice = {
                        order = increment(),
                        name = wrapName("玩家选择"),
                        desc = "使用按键绑定来选择第一个玩家选择选项。",
                        descStyle = "inline", width = "full", type = "toggle",
                        hidden = not C_AddOns.DoesAddOnExist("Blizzard_PlayerChoice"),
                    },
                    numKeysForPlayerChoice = {
                        order = increment(),
                        name = wrapName("为对话选项设置数字键"),
                        desc = "使用数字键（1到0）来选择玩家选择选项。",
                        descStyle = "inline", width = "full", type = "toggle",
                        hidden = not C_AddOns.DoesAddOnExist("Blizzard_PlayerChoice"),
                    },
                    handleSpecFrame = {
                        order = increment(),
                        name = wrapName(SPECIALIZATION),
                        desc = "当专精标签页打开时，使用数字键（1到4）选择专精",
                        descStyle = "inline", width = "full", type = "toggle",
                        hidden = WOW_PROJECT_ID ~= WOW_PROJECT_MAINLINE,
                    },
                },
            },
            watchlistGroup = {
                order = increment(),
                name = "自定义监控列表",
                desc = "快捷对话插件尝试点击的自定义按钮列表。",
                type = "group",
                args = {
                    desc = {
                        order = increment(),
                        name = [[
你可以在这里添加自定义框架，以便使用你的按键绑定来“点击”。
只需输入要处理的框架的名称，或者将鼠标悬停在框架上并写下“/dialogkey add”即可在鼠标下的框架中添加。

如果你在寻找名称时遇到困难，尝试使用“/fstack”，按下ALT键直到你找到想要突出显示的框架。如果名称中包含随机的字母和数字(例如："GameMenuFrame.2722d8f518")，那么这个框架就不能被快捷点击。
]],
                        type = "description",
                        fontSize = "medium",
                    },
                    addFrame = {
                        order = increment(),
                        type = "input",
                        name = "添加一个框架到监控",
                        width = "full",
                        set = function(_, value)
                            self:AddToWatchlist(value)
                        end,
                    },
                    priority = self:CreateCustomFramesPriorityListOptions(increment(), swapWatchlistFrames),
                },
            },
            popupBlacklistGroup = {
                order = increment(),
                name = "弹出框黑名单",
                desc = "快捷对话完全忽略的弹出对话框列表。",
                type = "group",
                args = {
                    dontAcceptInvite = {
                        order = increment(),
                        name = wrapName("不接受组队邀请"),
                        desc = "不允许快捷点击接受团队/小队邀请。",
                        width = 1.2, type = "toggle",
                    },
                    dontAcceptInstanceLocks = {
                        order = increment(),
                        name = wrapName("不接受副本锁定"),
                        desc = "不允许快捷点击接受/\"保存\"副本锁定",
                        width = 1.2, type = "toggle",
                    },
                    dontAcceptAbandonVote = {
                        order = increment(),
                        name = wrapName("不接受大秘境放弃钥匙投票"),
                        desc = "不允许快捷点击对放弃大秘境钥匙进行投票",
                        width = 1.2, type = "toggle",
                    },
                    dontAcceptVoteKick = {
                        order = increment(),
                        name = wrapName("不接受投票踢出"),
                        desc = "不允许快捷点击参与投票踢出",
                        width = 1.2, type = "toggle",
                    },
                    dontClickSummons = {
                        order = increment(),
                        name = wrapName("不接受召唤"),
                        desc = "不允许快捷点击接受召唤请求。",
                        width = 1.2, type = "toggle",
                    },
                    dontClickDuels = {
                        order = increment(),
                        name = wrapName("不接受决斗"),
                        desc = "不允许快捷点击接受决斗邀请。",
                        width = 1.2, type = "toggle",
                    },
                    dontClickRevives = {
                        order = increment(),
                        name = wrapName("不接受复活"),
                        desc = "不允许快捷点击接受复活。",
                        width = 1.2, type = "toggle",
                    },
                    useSoulstoneRez = {
                        order = increment(),
                        name = wrapName("使用职业专属复活"),
                        desc = "使用灵魂石/生命符/复生等职业专属复活。当这些复活可用时，不使用普通复活和战斗内复活。\n\n这个选项会|cffff0000忽略|r|cffffd100不接受复活|r 设置！",
                        width = 1.2, type = "toggle",
                    },
                    dontClickReleases = {
                        order = increment(),
                        name = wrapName("不要释放灵魂"),
                        desc = "不允许快捷点击释放灵魂。",
                        width = 1.2, type = "toggle",
                    },
                    desc = {
                        order = increment(),
                        name = [[
在这里，你可以创建一个自定义的弹出窗口列表，快捷对话将忽略这些窗口。
只需添加出现在弹出窗口中的（部分）文本，插件就会忽略它。
]],
                        type = "description",
                        fontSize = "medium",
                    },
                    addText = {
                        order = increment(),
                        type = "input",
                        name = "添加想忽略的文本",
                        width = "full",
                        set = function(_, value)
                            db.dialogBlacklist[value] = true
                        end,
                    },
                    removeText = {
                        order = increment(),
                        type = "select",
                        style = "dropdown",
                        name = "移除已忽略的文本",
                        width = "full",
                        values = function()
                            local tempTable = {}
                            if not next(db.dialogBlacklist) then
                                return { [''] = ' * 目前没有文本被忽略 *' }
                            end
                            for text, _ in pairs(db.dialogBlacklist) do
                                tempTable[text] = text
                            end
                            return tempTable
                        end,
                        get = function(_, _) return false end,
                        set = function(_, index, ...)
                            db.dialogBlacklist[index] = nil
                        end,
                    },
                    listOfTexts = {
                        order = increment(),
                        type = "description",
                        name = function()
                            local text = wrapName("目前被忽略的文本：") .. "\n"
                            for k, _ in pairs(db.dialogBlacklist) do
                                text = text .. " - " .. k .. "\n"
                            end
                            return text
                        end,
                    },
                },
            },
        },
    }

    return options
end

function DialogKeyNS:CreateCustomFramesPriorityListOptions(order, swapWatchlistFrames)
    local increment = CreateCounter()
    local options = {
        order = order,
        name = "管理目前监控的框架",
        type = "group",
        inline = true,
        args = {
            header = {
                order = increment(),
                name = "如果有多个框架可见，列表中位置最靠上的那个将被点击。",
                type = "description",
                width = "full",
            },
        },
    };
    if #self.orderedCustomFrames == 0 then
        options.args.noFrames = {
            order = increment(),
            name = " * 当前没有框架被监控。 * ",
            type = "description",
            fontSize = "medium",
            width = "full",
        };

        return options;
    end
    local numberOfIcons = 4;
    local iconWidth = 26;
    local groupWidth = 405; -- depends on whether there's a scrollbar
    for i, frame in ipairs(self.orderedCustomFrames) do
        options.args["glow" .. i] = {
            order = increment(),
            name = "",
            desc = MAGIC_TOOLTIP_TEXTS.glow,
            disabled = function() return not self.Core:GetFrameByName(frame) end,
            type = "execute",
            func = function()
                local frameRef = self.Core:GetFrameByName(frame)
                if not frameRef then return end
                self.Core:Glow(frameRef, 0.25, true)
            end,
            image = search.file,
            imageWidth = 16,
            imageHeight = 16,
            imageCoords = {
                search.leftTexCoord, search.rightTexCoord, search.topTexCoord, search.bottomTexCoord,
            },
            width = iconWidth / width_multiplier,
        };
        options.args["name" .. i] = {
            order = increment(),
            name = frame,
            type = "description",
            fontSize = "medium",
            width = (groupWidth - iconWidth * numberOfIcons) / width_multiplier,
        };
        options.args["up" .. i] = {
            order = increment(),
            name = "",
            desc = MAGIC_TOOLTIP_TEXTS.up,
            type = "execute",
            disabled = function() return i == 1 end,
            func = function()
                swapWatchlistFrames(i, i - 1)
            end,
            image = arrowBack.file,
            imageWidth = 16,
            imageHeight = 16,
            imageCoords = {
                arrowBack.leftTexCoord, arrowBack.bottomTexCoord, -- UL
                arrowBack.rightTexCoord, arrowBack.bottomTexCoord, -- LL
                arrowBack.leftTexCoord, arrowBack.topTexCoord, -- UR
                arrowBack.rightTexCoord, arrowBack.topTexCoord, -- LR
            },
            width = iconWidth / width_multiplier,
        };
        options.args["down" .. i] = {
            order = increment(),
            type = "execute",
            name = "",
            desc = MAGIC_TOOLTIP_TEXTS.down,
            disabled = function() return i == #self.orderedCustomFrames end,
            func = function()
                swapWatchlistFrames(i, i + 1)
            end,
            image = arrowNext.file,
            imageWidth = 16,
            imageHeight = 16,
            width = iconWidth / width_multiplier,
            imageCoords = {
                arrowNext.leftTexCoord, arrowNext.bottomTexCoord, -- UL
                arrowNext.rightTexCoord, arrowNext.bottomTexCoord, -- LL
                arrowNext.leftTexCoord, arrowNext.topTexCoord, -- UR
                arrowNext.rightTexCoord, arrowNext.topTexCoord, -- LR
            },
        };
        options.args["delete" .. i] = {
            order = increment(),
            name = "",
            desc = MAGIC_TOOLTIP_TEXTS.remove,
            type = "execute",
            func = function()
                self:RemoveFromWatchlist(frame)
            end,
            image = cross.file,
            imageWidth = 16,
            imageHeight = 16,
            imageCoords = {
                cross.leftTexCoord, cross.rightTexCoord, cross.topTexCoord, cross.bottomTexCoord,
            },
            width = iconWidth / width_multiplier,
        };
        options.args["spacer" .. i] = {
            order = increment(),
            name = "",
            type = "description",
            width = "full",
        };
    end

    return options;
end

local name = DialogKeyLocal

local GetFrameMetatable = _G.GetFrameMetatable or function() return getmetatable(CreateFrame('FRAME')) end

_G.DialogKeyNS = DialogKeyNS -- expose ourselves to the world :)

--- @class DialogKey: AceAddon, AceEvent-3.0, AceHook-3.0
local DialogKey = LibStub("AceAddon-3.0"):NewAddon(name, "AceEvent-3.0", "AceHook-3.0")
DialogKeyNS.Core = DialogKey

local defaultPopupBlacklist = { -- If a popup dialog contains one of these strings, don't click it
    AREA_SPIRIT_HEAL = true, -- Prevents cancelling the resurrection
    TOO_MANY_LUA_ERRORS = true,
    END_BOUND_TRADEABLE = true, EQUIP_BIND_TRADEABLE = true, -- Probably quite reasonable to make the user click on this one
    ADDON_ACTION_FORBIDDEN = true,                           -- Don't disable and reload UI on errors
    CONFIRM_LEAVE_RESTRICTED_CHALLENGE_MODE = true,
    WARN_LEAVE_RESTRICTED_CHALLENGE_MODE = true,
}

local function callFrameMethod(frame, method, ...)
    local functionRef = frame[method] or GetFrameMetatable().__index[method] or nop;
    local ok, result = pcall(functionRef, frame, ...);

    return ok and result or false
end
--- @return string?
local function getFrameName(frame)
    return callFrameMethod(frame, 'GetDebugName') ---@diagnostic disable-line: return-type-mismatch
        or callFrameMethod(frame, 'GetName')
end
---@return Frame?
function DialogKey:GetFrameByName(frameName)
    local frameTable = _G;

    for keyName in string.gmatch(frameName, "([^.]+)") do
        if not frameTable[keyName] then return nil; end

        frameTable = frameTable[keyName];
    end

    return frameTable;
end

--- @type Button[]
DialogKey.playerChoiceButtons = {}
--- @type Button[]
DialogKey.specButtons = {}
DialogKey.activeOverrideBindings = {}
DialogKey.activeDeferClearBindings = {}

DialogKey.dummyButton = CreateFrame("Button")

function DialogKey:OnInitialize()
    DialogKeyNumyDB = DialogKeyNumyDB or {}
    self.db = DialogKeyNumyDB
    DialogKeyNS:InitDB(self)

    self:InitGlowFrame()

    self:RegisterEvent("QUEST_GREETING")
    self:RegisterEvent("QUEST_LOG_UPDATE")
    self:RegisterEvent("QUEST_COMPLETE")
    self:RegisterEvent("PLAYER_REGEN_DISABLED")
    self:RegisterEvent("ADDON_LOADED")

    self:InitMainProxyFrame()

    self:SecureHook("QuestInfoItem_OnClick", "SelectItemReward")
    GossipFrame.GreetingPanel.ScrollBox:RegisterCallback("OnAcquiredFrame", self.OnGossipFrameAcquired, self)
    GossipFrame.GreetingPanel.ScrollBox:RegisterCallback("OnReleasedFrame", self.OnGossipFrameReleased, self)
    GossipFrame.GreetingPanel.ScrollBox:RegisterCallback(ScrollBoxListMixin.Event.OnDataProviderReassigned, self.OnGossipDataProviderChange, self)

    DialogKeyNS:RegisterOptions()

    _G.SLASH_DIALOGKEY1 = '/dialogkey'
    _G.SLASH_DIALOGKEY2 = '/dkey'
    _G.SLASH_DIALOGKEY3 = '/dk'
    SlashCmdList['DIALOGKEY'] = function(msg)
        local func, args = strsplit(" ", msg, 2)
        if func == 'add' then
            self:AddFrame(args)
        elseif func == 'remove' then
            self:RemoveFrame(args)
        else
            DialogKeyNS:OpenConfig()
        end
    end
end

function DialogKey:ADDON_LOADED(_, addon)
    if addon == 'Blizzard_PlayerChoice' then
        self:SecureHook(PlayerChoiceFrame, "TryShow", "OnPlayerChoiceShow")
        self:SecureHookScript(PlayerChoiceFrame, "OnHide", "OnPlayerChoiceHide")
    elseif addon == 'Blizzard_PlayerSpells' then
        self:SecureHookScript(PlayerSpellsFrame.SpecFrame, "OnShow", "OnSpecFrameShow")
        self:SecureHookScript(PlayerSpellsFrame.SpecFrame, "OnHide", "OnSpecFrameHide")
    end
end

function DialogKey:QUEST_COMPLETE()
    self.itemChoice = (GetNumQuestChoices() > 1 and -1 or 1)
end

function DialogKey:QUEST_GREETING()
    RunNextFrame(function() self:EnumerateGossips() end)
end

function DialogKey:QUEST_LOG_UPDATE()
    RunNextFrame(function() self:EnumerateGossips() end)
end

function DialogKey:PLAYER_REGEN_DISABLED()
    -- Disable DialogKey fully upon entering combat
    self.frame:SetPropagateKeyboardInput(true)
    self:ClearOverrideBindings()
end

function DialogKey:InitGlowFrame()
    self.glowFrame = CreateFrame("Frame", nil, UIParent)
    self.glowFrame:SetPoint("CENTER", 0, 0)
    self.glowFrame:SetFrameStrata("TOOLTIP")
    self.glowFrame:SetSize(50, 50)
    self.glowFrame:SetScript("OnUpdate", function(...) self:GlowFrameUpdate(...) end)
    self.glowFrame:Hide()
    self.glowFrame.tex = self.glowFrame:CreateTexture()
    self.glowFrame.tex:SetAllPoints()
    self.glowFrame.tex:SetColorTexture(1, 1, 0, 0.5)
end

function DialogKey:InitMainProxyFrame()
    local frame = CreateFrame("Button", "DialogKey_Numy_MainClickProxyFrame", UIParent, "InsecureActionButtonTemplate")
    frame:RegisterForClicks("AnyUp", "AnyDown")
    frame:SetAttribute("type", "click")
    frame:SetAttribute("typerelease", "click")
    frame:SetScript("PreClick", function(_, _, down)
        if InCombatLockdown() then return end

        local useOnDown = C_CVar.GetCVarBool("ActionButtonUseKeyDown")
        if down ~= useOnDown then return end

        local clickButton = frame:GetAttribute("clickbutton") --[[@as Button]]
        self:Glow(clickButton)
    end)
    frame:HookScript("OnClick", function(_, _, down)
        if InCombatLockdown() then return end

        local useOnDown = C_CVar.GetCVarBool("ActionButtonUseKeyDown")
        if down ~= useOnDown then return end

        self:ClearOverrideBindings(frame)
        frame:SetAttribute("clickbutton", nil)
        frame:SetPropagateKeyboardInput(true)
    end)
    frame:SetScript("OnKeyDown", function(_, ...) self:HandleKey(...) end)
    frame:SetFrameStrata("TOOLTIP") -- Ensure we receive keyboard events first
    frame:EnableKeyboard(true)
    frame:SetPropagateKeyboardInput(true)

    self.frame = frame
end

function DialogKey:OnPlayerChoiceShow()
    if not self.db.handlePlayerChoice and not self.db.numKeysForPlayerChoice then return end
    local frame = PlayerChoiceFrame;
    if not frame or not frame:IsVisible() then return end

    local choiceInfo = C_PlayerChoice.GetCurrentPlayerChoiceInfo()
    if not choiceInfo then return end
    local buttons = {}
    local i = 0
    for _, option in ipairs(choiceInfo.options) do
        for _, button in ipairs(option.buttons) do
            if not button.hideButtonShowText or not button.text then
                i = i + 1
                buttons[button.id] = i
            end
        end
    end

    for option in frame.optionPools:EnumerateActive() do
        if option.buttons.buttonFramePool then
            for buttonFrame in option.buttons.buttonFramePool:EnumerateActive() do
                local button = buttonFrame.Button
                local key = buttons[button.buttonID]
                if key then
                    if self.db.numKeysForPlayerChoice then
                        button.Text:SetText(key .. ' ' .. button.Text:GetText())
                    end
                    self.playerChoiceButtons[key] = button
                end
            end
        end
    end
end

function DialogKey:OnPlayerChoiceHide()
    self.playerChoiceButtons = {}
end

function DialogKey:OnSpecFrameShow()
    --- @type FramePool<Frame, ClassSpecContentFrameTemplate>
    local framePool = PlayerSpellsFrame.SpecFrame.SpecContentFramePool

    self.specButtons = {}
    for specContentFrame in framePool:EnumerateActive() do
        --- @type ClassSpecContentFrameTemplate
        local specContentFrame = specContentFrame
        self.specButtons[specContentFrame.specIndex] = specContentFrame.ActivateButton
        local text = self.db.handleSpecFrame and (specContentFrame.specIndex .. ' ' .. TALENT_SPEC_ACTIVATE) or TALENT_SPEC_ACTIVATE
        specContentFrame.ActivateButton:SetText(text)
    end
end

function DialogKey:OnSpecFrameHide()
    self.specButtons = {}
end

function DialogKey:OnGossipDataProviderChange()
    self.frames = {}
    self.orderedGossipFrames = {}
    local scrollbox = GossipFrame.GreetingPanel.ScrollBox
    local n = 1
    for _, elementData in scrollbox:GetDataProvider():Enumerate() do
        local tag
        if GOSSIP_BUTTON_TYPE_OPTION == elementData.buttonType then
            tag = "name"
        elseif GOSSIP_BUTTON_TYPE_AVAILABLE_QUEST == elementData.buttonType then
            tag = "title"
        elseif GOSSIP_BUTTON_TYPE_ACTIVE_QUEST == elementData.buttonType and (elementData.info.isComplete or not self.db.ignoreInProgressQuests) then
            tag = "title"
        end
        if tag then
            self.orderedGossipFrames[elementData] = n
            if n <= 10 then
                local info = elementData.info
                local oldText = info[tag]
                if info.flags and FlagsUtil.IsSet(info.flags, Enum.GossipOptionRecFlags.QuestLabelPrepend) then
                    oldText = GOSSIP_QUEST_OPTION_PREPEND:format(oldText);
                end
                if info.isIgnored then
                    oldText = IGNORED_QUEST_DISPLAY:format(oldText);
                elseif info.isTrivial then
                    oldText = TRIVIAL_QUEST_DISPLAY:format(oldText);
                end
                local newText = (n % 10) .. ". " .. (oldText:match("^%d. (.+)$") or oldText)

                elementData.info.DialogKeyNumy_Text = newText
            end

            n = n + 1
        end

        local calculatorFrame
        if elementData.titleOptionButton then
            calculatorFrame = elementData.titleOptionButton
        elseif elementData.availableQuestButton then
            calculatorFrame = elementData.availableQuestButton
        elseif elementData.activeQuestButton then
            calculatorFrame = elementData.activeQuestButton
        end
        if calculatorFrame and not calculatorFrame.DialogKeyNumy_Hooked then
            hooksecurefunc(calculatorFrame, 'Setup', function(_, info)
                if info.DialogKeyNumy_Text then
                    calculatorFrame:SetTextAndResize(info.DialogKeyNumy_Text)
                end
            end)
            calculatorFrame.DialogKeyNumy_Hooked = true
        end
    end
end

--- @param frame GossipTitleButtonTemplate
--- @param elementData table
function DialogKey:OnGossipFrameAcquired(frame, elementData)
    if not self.db.numKeysForGossip then return; end

    local n = self.orderedGossipFrames[elementData]
    local newText = elementData.info and elementData.info.DialogKeyNumy_Text

    if not n or n > 10 or not newText then return end

    self.frames[n] = frame
    frame:SetTextAndResize(newText)

    if not frame.DialogKeyNumy_Text then
        --- @param f GossipTitleButtonTemplate
        local function onSetText(f, text)
            local savedText = f.DialogKeyNumy_Text
            if text == savedText or not savedText or savedText == "" then return end

            f:SetTextAndResize(savedText)
        end
        hooksecurefunc(frame, 'SetText', onSetText)
        hooksecurefunc(frame, 'SetFormattedText', function(f, format, ...)
            local text = format:format(...)
            onSetText(f, text)
        end)
    end
    frame.DialogKeyNumy_Text = newText
end

function DialogKey:OnGossipFrameReleased(frame)
    if frame.DialogKeyNumy_Text then
        frame.DialogKeyNumy_Text = ""
    end
end

--- @return Button[]|nil
function DialogKey:GetValidPopupButtons()
    local buttons = {}
    local popupFrames = {}
    if StaticPopup_ForEachShownDialog then
        StaticPopup_ForEachShownDialog(function(popup) if not popup.special then table.insert(popupFrames, popup) end end)
    else
        for i = 1, 4 do
            local popup = _G["StaticPopup" .. i]
            if popup and popup:IsVisible() then
                table.insert(popupFrames, popup)
            end
        end
    end
    table.sort(popupFrames, function(a, b) return a:GetTop() > b:GetTop() end)
    for _, popupFrame in ipairs(popupFrames) do
        local button = self:GetPopupButton(popupFrame)
        if button then
            table.insert(buttons, button)
        end
    end

    return next(buttons) and buttons or nil
end

--- @param popupFrame StaticPopupTemplate # One of the StaticPopup1-4 frames
--- @return Frame|nil|false # The button to click, nil if no button should be clicked, false if the text is empty and should be checked again later
function DialogKey:GetPopupButton(popupFrame)
    local fontString = popupFrame.GetTextFontString and popupFrame:GetTextFontString() or popupFrame.text
    local text = fontString and fontString:GetText()
    local which = popupFrame.which

    local button1 = popupFrame.button1 or popupFrame:GetButton1()
    local button2 = popupFrame.button2 or popupFrame:GetButton2()

    -- Some popups have no text when they initially show, and instead get text applied OnUpdate (summons are an example)
    -- False is returned in that case, so we know to keep checking OnUpdate
    if not text or text == " " or text == "" then return false end

    if self.db.dontAcceptInvite and which == 'PARTY_INVITE' then return end
    if self.db.dontClickSummons and which == 'CONFIRM_SUMMON' then return end
    if self.db.dontClickDuels and which == 'DUEL_REQUESTED' then return end
    if self.db.dontAcceptInstanceLocks and which == 'INSTANCE_LOCK' then return end
    if self.db.dontAcceptAbandonVote and which == 'VOTE_ABANDON_INSTANCE_VOTE' or which == 'VOTE_ABANDON_INSTANCE_WAIT' then return end
    if self.db.dontAcceptVoteKick and which == 'VOTE_BOOT_PLAYER' then return end

    -- If resurrect dialog has three buttons, and the option is enabled, use the middle one instead of the first one (soulstone, etc.)
    -- Located before resurrect/release checks/returns so it happens even if you have releases/revives disabled
    -- Also, Check if Button2 is visible instead of Button3 since Recap is always 3; 2 is hidden if you can't soulstone rez

    -- the ordering here means that a revive will be taken before a battle rez before a release.
    -- if revives are disabled but soulstone battlerezzes *aren't*, nothing will happen if both are available!
    local canRelease = button1:GetText() == DEATH_RELEASE
    if self.db.useSoulstoneRez and canRelease and button2:IsVisible() then
        return button2
    end

    if self.db.dontClickRevives and (text == RECOVER_CORPSE or which == 'RESURRECT_NO_SICKNESS' or which == 'RESURRECT_NO_TIMER') then return end
    if self.db.dontClickReleases and canRelease then return end

    -- Ignore blacklisted popup dialogs
    if defaultPopupBlacklist[which] or self.db.dialogBlacklist[which] then return end
    local lowerCaseText = text:lower()
    for blacklistText, _ in pairs(self.db.dialogBlacklist) do
        -- Prepend non-alphabetical characters with '%' to escape them
        blacklistText = blacklistText:gsub("%W", "%%%0"):gsub("%%%%s", ".+")
        if lowerCaseText:find(blacklistText:lower()) then return end
    end

    return button1:IsVisible() and button1 or nil
end

--- @param frame Button
function DialogKey:GuardDisabled(frame)
    if not self.db.ignoreDisabledButtons then return true; end

    return frame:IsEnabled() and frame:IsMouseClickEnabled();
end

--- @return Button|nil
function DialogKey:GetFirstVisibleCustomFrame()
    for _, frameName in ipairs(DialogKeyNS.orderedCustomFrames) do
        local frame = self:GetFrameByName(frameName)
        if frame and frame:IsVisible() and frame:IsObjectType('Button') and self:GuardDisabled(frame) then
            return frame ---@diagnostic disable-line: return-type-mismatch
        end
    end
end

--- @return Button|nil
function DialogKey:GetFirstVisibleCraftingOrderFrame()
    if not self.db.handleCraftingOrders then return; end
    local frames = {
        "ProfessionsFrame.OrdersPage.OrderView.OrderInfo.StartOrderButton",
        "ProfessionsFrame.OrdersPage.OrderView.CreateButton",
        "ProfessionsFrame.OrdersPage.OrderView.CompleteOrderButton",
    };
    for _, frameName in ipairs(frames) do
        --- @type Button?
        local frame = self:GetFrameByName(frameName) ---@diagnostic disable-line: assign-type-mismatch
        if frame and frame:IsVisible() and self:GuardDisabled(frame) then
            return frame
        end
    end
end

function DialogKey:ShouldIgnoreInput()
    if InCombatLockdown() then return true end

    if self.db.ignoreWithModifier and (IsShiftKeyDown() or IsControlKeyDown() or IsAltKeyDown()) then return true end
    -- Ignore input while typing, unless at the Send Mail confirmation while typing into it!
    local focus = GetCurrentKeyBoardFocus()
    if focus and not (self:GetValidPopupButtons() and (focus:GetName() == "SendMailNameEditBox" or focus:GetName() == "SendMailSubjectEditBox")) then return true end

    -- Ignore input if there's nothing for DialogKey to click
    if
        not GossipFrame:IsVisible() and not QuestFrame:IsVisible() and not self:GetValidPopupButtons()
        and (not AuctionHouseFrame or not AuctionHouseFrame:IsVisible())
        and not self:GetFirstVisibleCraftingOrderFrame()
        and not self:GetFirstVisibleCustomFrame()
        and not next(self.playerChoiceButtons)
        and not next(self.specButtons)
    then
        return true
    end

    return false
end

--- subsequent calls will cancel any existing deferred call
--- @param owner Frame
--- @param delay number?
function DialogKey:DeferClearOverrideBindings(owner, delay)
    delay = delay or 0
    if self.activeDeferClearBindings[owner] then
        self.activeDeferClearBindings[owner]:Cancel()
        self.activeDeferClearBindings[owner] = nil
    end
    self.activeDeferClearBindings[owner] = C_Timer.NewTimer(delay, function()
        self:ClearOverrideBindings(owner)
        self.activeDeferClearBindings[owner] = nil
    end)
end

-- Clears all override bindings associated with an owner, clears all override bindings if no owner is passed
--- @param owner Frame?
function DialogKey:ClearOverrideBindings(owner)
    if InCombatLockdown() then return end
    if not owner then
        for owner, _ in pairs(self.activeOverrideBindings) do
            self:ClearOverrideBindings(owner)
        end
        return
    end
    if self.activeDeferClearBindings[owner] then
        self.activeDeferClearBindings[owner]:Cancel()
        self.activeDeferClearBindings[owner] = nil
    end
    if not self.activeOverrideBindings[owner] then return end
    for key in pairs(self.activeOverrideBindings[owner]) do
        SetOverrideBinding(owner, true, key, nil)
    end
    self.activeOverrideBindings[owner] = nil
end

-- Set an override click binding, these bindings can safely perform secure actions
-- Override bindings, are temporary keybinds, which can only be modified out of combat; they are tied to an owner, and need to be cleared when the target is hidden
--- @param owner Frame
--- @param targetName string
--- @param keys string[]
function DialogKey:SetOverrideBindings(owner, targetName, keys)
    if InCombatLockdown() then return end
    self.activeOverrideBindings[owner] = {}
    for _, key in pairs(keys) do
        self.activeOverrideBindings[owner][key] = owner;
        SetOverrideBindingClick(owner, true, key, targetName, 'LeftButton');
    end
end

function DialogKey:SetClickbuttonBinding(frame, key)
    if InCombatLockdown() then return end
    self.frame:SetAttribute("clickbutton", frame)
    self:SetOverrideBindings(self.frame, self.frame:GetName(), { key })

    local useOnDown = C_CVar.GetCVarBool("ActionButtonUseKeyDown")
    -- just in case something goes horribly wrong, we do NOT want to get the user stuck in a situation where the keyboard stops working
    -- 5 seconds is a rather arbitrary amount of time to wait, but meh
    self:DeferClearOverrideBindings(self.frame, useOnDown and 0 or 5)
end

function DialogKey:HandleKey(key)
    if not InCombatLockdown() then self.frame:SetPropagateKeyboardInput(true) end
    local doAction = (key == self.db.keys[1] or key == self.db.keys[2])
    local keynum = doAction and 1 or tonumber(key)
    if key:match("^NUMPAD") then
        keynum = tonumber((key:gsub("NUMPAD", "")))
    end
    if key == "0" or key == "NUMPAD0" then
        keynum = 10
    end
    if not doAction and not keynum then return end
    if self:ShouldIgnoreInput() then return end
    -- DialogKey pressed, interact with popups, accepts..
    if doAction then
        -- Popups
        local popupButtons = self:GetValidPopupButtons()
        if popupButtons then
            -- todo: set a binding for each popup button?
            self:SetClickbuttonBinding(popupButtons[1], key)
            return
        end

        -- Crafting Orders
        local craftingOrderFrame = self:GetFirstVisibleCraftingOrderFrame()
        if craftingOrderFrame then
            self:SetClickbuttonBinding(craftingOrderFrame, key)
            return
        end

        -- Custom Frames
        local customFrame = self:GetFirstVisibleCustomFrame()
        if customFrame then
            self:SetClickbuttonBinding(customFrame, key)
            return
        end

        -- Auction House
        if self.db.postAuctions and AuctionHouseFrame and AuctionHouseFrame:IsVisible() then
            if AuctionHouseFrame.displayMode == AuctionHouseFrameDisplayMode.CommoditiesSell then
                self:SetClickbuttonBinding(AuctionHouseFrame.CommoditiesSellFrame.PostButton, key)
                return
            elseif AuctionHouseFrame.displayMode == AuctionHouseFrameDisplayMode.ItemSell then
                self:SetClickbuttonBinding(AuctionHouseFrame.ItemSellFrame.PostButton, key)
                return
            end
        end

        if QuestFrameProgressPanel:IsVisible() then -- Complete Quest
            if not QuestFrameCompleteButton:IsEnabled() and self.db.ignoreDisabledButtons then
                -- click "Cencel" button when "Complete" is disabled on progress panel
                self:SetClickbuttonBinding(QuestFrameGoodbyeButton, key)
            else
                self:SetClickbuttonBinding(QuestFrameCompleteButton, key)
            end
            return
        elseif QuestFrameDetailPanel:IsVisible() then -- Accept Quest
            self:SetClickbuttonBinding(QuestFrameAcceptButton, key)
            return
        elseif QuestFrameRewardPanel:IsVisible() then -- Take Quest Reward - using manual API
            self.frame:SetPropagateKeyboardInput(false)
            if self.itemChoice == -1 and GetNumQuestChoices() > 1 then
                QuestChooseRewardError()
            else
                self:Glow(QuestFrameCompleteQuestButton)
                GetQuestReward(self.itemChoice)
            end
            return
        end
    end

    -- Player Choice
    if
        ((self.db.handlePlayerChoice and doAction) or (self.db.numKeysForPlayerChoice and not doAction))
        and next(self.playerChoiceButtons)
    then
        local button = self.playerChoiceButtons[keynum]
        if button and (not self.db.ignoreDisabledButtons or button:IsEnabled()) then
            self:SetClickbuttonBinding(button, key)
            return
        end
    end

    -- Spec Frame
    if self.db.handleSpecFrame and next(self.specButtons) then
        local button = self.specButtons[keynum]
        if button then
            -- blocks keybind for currently selected spec index
            if not button:IsVisible() then button = self.dummyButton end
            self:SetClickbuttonBinding(button, key)
            return
        end
    end

    -- GossipFrame
    if (doAction or self.db.numKeysForGossip) and GossipFrame.GreetingPanel:IsVisible() then
        while keynum and keynum > 0 and keynum <= #self.frames do
            local choice = self.frames[keynum] and self.frames[keynum].GetElementData and self.frames[keynum].GetElementData()
            -- Skip grey quest (active but not completed) when pressing DialogKey
            if doAction and choice and choice.info and choice.info.questID and choice.activeQuestButton and not choice.info.isComplete and self.db.ignoreDisabledButtons then
                keynum = keynum + 1
            else
                self:SetClickbuttonBinding(self.frames[keynum], key)
                return
            end
        end
    end

    -- QuestFrame
    if (doAction or self.db.numKeysForGossip) and QuestFrameGreetingPanel:IsVisible() and self.frame then
        while keynum and keynum > 0 and keynum <= #self.frames do
            local _, is_complete = GetActiveTitle(keynum)
            if doAction and not is_complete and self.frames[keynum].isActive == 1 and self.db.ignoreDisabledButtons then
                keynum = keynum + 1
                if keynum > #self.frames then
                    doAction = false
                    keynum = 1
                end
            else
                self:SetClickbuttonBinding(self.frames[keynum], key)
                return
            end
        end
    end

    -- QuestReward Frame (select item)
    if self.db.numKeysForQuestRewards and keynum and keynum <= GetNumQuestChoices() and QuestFrameRewardPanel:IsVisible() then
        self.itemChoice = keynum
        self:SetClickbuttonBinding(GetClickFrame("QuestInfoRewardsFrameQuestInfoItem" .. key), key)
        return
    end
end

-- QuestInfoItem_OnClick secure handler
-- allows DialogKey to update the selected quest reward when clicked as opposed to using a keybind.
function DialogKey:SelectItemReward()
    for i = 1, GetNumQuestChoices() do
        if GetClickFrame("QuestInfoRewardsFrameQuestInfoItem" .. i):IsMouseOver() then
            self.itemChoice = i
            break
        end
    end
end

-- Prefix list of QuestGreetingFrame options with 1., 2., 3. etc.
-- Also builds DialogKey.frames, used to click said options
function DialogKey:EnumerateGossips()
    if not QuestFrameGreetingPanel:IsVisible() then return end

    local checkQuestsToHandle = false
    local questsToHandle = {}

    if self.db.ignoreInProgressQuests then
        checkQuestsToHandle = true
        local numActiveQuests = GetNumActiveQuests()
        local numAvailableQuests = GetNumAvailableQuests()
        for i = 1, numActiveQuests do
            local _, isComplete = GetActiveTitle(i)
            questsToHandle[i] = isComplete
        end
        for i = (numActiveQuests + 1), (numActiveQuests + numAvailableQuests) do
            questsToHandle[i] = true
        end
    end

    local frames = {}
    self.frames = {}
    if QuestFrameGreetingPanel and QuestFrameGreetingPanel.titleButtonPool then ---@diagnostic disable-line: undefined-field
        --- @type FramePool<Button, QuestTitleButtonTemplate>
        local pool = QuestFrameGreetingPanel.titleButtonPool ---@diagnostic disable-line: undefined-field
        for tab in (pool:EnumerateActive()) do
            if tab:GetObjectType() == "Button" then
                table.insert(frames, tab)
            end
        end
    elseif QuestFrameGreetingPanel and not QuestFrameGreetingPanel.titleButtonPool then ---@diagnostic disable-line: undefined-field
        --- @type ScriptRegion[]
        local children = { QuestGreetingScrollChildFrame:GetChildren() }
        for _, child in ipairs(children) do
            if child:GetObjectType() == "Button" and child:IsVisible() then
                table.insert(frames, child)
            end
        end
    else
        return
    end

    table.sort(frames, function(a, b)
        if a.GetOrderIndex then
            return a:GetOrderIndex() < b:GetOrderIndex()
        else
            return a:GetTop() > b:GetTop()
        end
    end)

    if self.db.numKeysForGossip then
        local n = 1
        for i, frame in ipairs(frames) do
            if not checkQuestsToHandle or questsToHandle[i] then
                if n > 10 then break end
                local oldText = frame:GetText()
                local newText = (n % 10) .. ". " .. (oldText:match("^%d. (.+)$") or oldText)
                frame:SetText(newText)

                -- Make the button taller if the text inside is wrapped to multiple lines
                frame:SetHeight(frame:GetFontString():GetHeight() + 2)
                n = n + 1
            end
        end
    end
    for i, frame in ipairs(frames) do
        if not checkQuestsToHandle or questsToHandle[i] then
            table.insert(self.frames, frame)
        end
    end
end

-- Glow Functions --
--- @param frame Frame
--- @param speedModifier number? # increasing this number will speed up the fade out of the glow
--- @param forceShow boolean? # if true, the glow will be shown regardless of the showGlow setting
function DialogKey:Glow(frame, speedModifier, forceShow)
    if self.db.showGlow or forceShow then
        self.glowFrame:SetAllPoints(frame)
        self.glowFrame.tex:SetColorTexture(1, 1, 0, 0.5)
        self.glowFrame:Show()
        self.glowFrame:SetAlpha(1)
        self.glowFrame.speedModifier = speedModifier or 1
    end
end

-- Fades out the glow frame
function DialogKey:GlowFrameUpdate(frame, delta)
    local alpha = frame:GetAlpha() - (delta * 3 * frame.speedModifier)
    if alpha < 0 then
        alpha = 0
    end
    frame:SetAlpha(alpha)
    if frame:GetAlpha() <= 0 then frame:Hide() end
end

function DialogKey:print(...)
    print("|cffd2b48c[快捷对话]|r ", ...)
end

function DialogKey:AddFrame(frameName)
    local frame
    if not frameName then
        frame, frameName = self:GetFrameUnderCursor()
    else
        frame = self:GetFrameByName(frameName)
    end

    if not frame then
        self:print("在你的鼠标下方没有找到可以点击的框架。尝试使用 /fstack 命令查找框架的名称，并使用 /dialogkey add <frameName> 手动添加它。")
        return
    end

    if self.db.customFrames[frameName] then
        self:print("该框架已经在监控列表中：", frameName)
        self:Glow(frame, 0.25, true)
        return
    end
    DialogKeyNS:AddToWatchlist(frameName)
    self:Glow(frame, 0.25, true)
    self:print("已添加框架：", frameName, "。你可以使用 /dialogkey remove 命令来移除它；或者在选项界面中进行操作。")
end

function DialogKey:RemoveFrame(frameName)
    local frame
    if not frameName then
        frame, frameName = self:GetFrameUnderCursor()
    else
        frame = self:GetFrameByName(frameName)
    end

    if not frame then
        self:print("在你的鼠标下方没有找到可以点击的框架。尝试使用 /fstack 命令来查找框架的名称，并使用 /dialogkey remove <frameName> 手动移除它。")
        return
    end
    local index = self.db.customFrames[frameName]
    if not index then
        self:print("鼠标下的可点击框架不在自定义监视列表中：", frameName)
        self:Glow(frame, 0.25, true)
        return
    end

    DialogKeyNS:RemoveFromWatchlist(frameName)
    self:Glow(frame, 0.25, true)
    self:print("已移除框架：", frameName)
end

--- Returns the first clickable frame that has mouse focus
--- @return Frame?, string? # The frame under the cursor, and its name; or nil
function DialogKey:GetFrameUnderCursor()
    for _, frame in ipairs(GetMouseFoci()) do
        if
            frame ~= WorldFrame
            and frame ~= UIParent
            and not callFrameMethod(frame, 'IsForbidden')
            and callFrameMethod(frame, 'HasScript', 'OnClick')
            and getFrameName(frame)
            and self:GetFrameByName(getFrameName(frame))
        then
            return frame, getFrameName(frame);
        end
    end
end
