bmHotbarClass.lua

local mq                            = require('mq')
local Set                           = require('mq.Set')
local btnUtils                      = require('lib.buttonUtils')
local BMButtonHandlers              = require('bmButtonHandlers')
local themes                        = require('extras.themes')
local BMTab                         = require('bmTab')

local WINDOW_SETTINGS_ICON_SIZE     = 22

local editTabPopup                  = "edit_tab_popup"

---@class BMHotbarClass
local BMHotbarClass                 = {}
BMHotbarClass.__index               = BMHotbarClass
BMHotbarClass.id                    = 1
BMHotbarClass.openGUI               = true
BMHotbarClass.shouldDrawGUI         = true
BMHotbarClass.setupComplete         = false
BMHotbarClass.lastWindowX           = 0
BMHotbarClass.lastWindowY           = 0
BMHotbarClass.lastButtonPageHeight  = 0
BMHotbarClass.lastButtonPageWidth   = 0
BMHotbarClass.lastWindowHeight      = 0
BMHotbarClass.lastWindowWidth       = 0
BMHotbarClass.buttonSizeDirty       = false
BMHotbarClass.visibleButtonCount    = 0
BMHotbarClass.cachedCols            = 0
BMHotbarClass.cachedRows            = 0
BMHotbarClass.highestRenderTime     = 0

BMHotbarClass.importObjectPopupOpen = false

BMHotbarClass.validDecode           = false
BMHotbarClass.importText            = ""
BMHotbarClass.decodedObject         = {}

BMHotbarClass.newSetName            = ""
BMHotbarClass.editingLabel          = ""
BMHotbarClass.windowTitleEdit       = ""
BMHotbarClass.currentSelectedSet    = 0

BMHotbarClass.lastFrameTime         = 0

BMHotbarClass.importTextChanged     = false

BMHotbarClass.updateWindowPosSize   = false
BMHotbarClass.newWidth              = 0
BMHotbarClass.newHeight             = 0
BMHotbarClass.newX                  = 0
BMHotbarClass.newY                  = 0

BMHotbarClass.searchText            = ""

BMHotbarClass.currentDnDData        = nil
BMHotbarClass.tabBar                = nil

BMHotbarClass.alphaMenu             = {
    { name = "A-D",   filter = function(i) return i >= "A" and i <= "D" end, items = {}, },
    { name = "E-H",   filter = function(i) return i >= "E" and i <= "H" end, items = {}, },
    { name = "I-L",   filter = function(i) return i >= "I" and i <= "L" end, items = {}, },
    { name = "M-P",   filter = function(i) return i >= "M" and i <= "P" end, items = {}, },
    { name = "Q-T",   filter = function(i) return i >= "Q" and i <= "T" end, items = {}, },
    { name = "U-X",   filter = function(i) return i >= "U" and i <= "X" end, items = {}, },
    { name = "Y-Z",   filter = function(i) return i >= "Y" and i <= "Z" end, items = {}, },
    { name = "0-9",   filter = function(i) return i >= "0" and i <= "9" end, items = {}, },
    { name = "Other", filter = function(i) return true end,                  items = {}, }, }


function BMHotbarClass.new(id, createFresh)
    local newBMHotbar = setmetatable({ id = id, }, BMHotbarClass)

    if createFresh then
        BMSettings:GetCharConfig().Windows[id] = {
            Visible = true,
            Sets = {},
            Locked = false,
            HideTitleBar = false,
            CompactMode = false,
            AdvTooltips = true,
            ShowSearch = false,
            PerCharacterPositioning = false,
        }

        -- if this character doesn't have the sections in the config, create them
        newBMHotbar.updateWindowPosSize = true
        newBMHotbar.newWidth = 1000
        newBMHotbar.newHeight = 150
        newBMHotbar.newX = 500
        newBMHotbar.newY = 500

        BMSettings:SaveSettings(true)
    end

    BMSettings:GetCharConfig().Windows[id].Sets = BMSettings:GetCharConfig().Windows[id].Sets or {}

    newBMHotbar.tabBar = BMTab.new(
        tostring(id),
        BMSettings:GetCharacterWindowSets(id),
        function(newLabels)
            BMSettings:GetCharConfig().Windows[id].Sets = newLabels
            BMSettings:SaveSettings(true)
        end,
        function(label, idx)
            -- Rename popup injected into each tab's context menu.
            if newBMHotbar.editingLabel ~= label then
                newBMHotbar.editingLabel = label
                newBMHotbar.newSetName   = label
            end
            ImGui.Text("Edit Name:")
            local tmp, changed = ImGui.InputText("##bmtab_edit_" .. id, newBMHotbar.newSetName, 0)
            if changed then newBMHotbar.newSetName = tmp end
            if ImGui.Button("Save##bmtab_save_" .. id) then
                local newName = newBMHotbar.newSetName
                if newName and newName:len() > 0 and newName ~= label then
                    -- Rename in global Sets table.
                    BMSettings:GetSettings().Sets[newName] = BMSettings:GetSettings().Sets[label]
                    BMSettings:GetSettings().Sets[label]   = nil
                    -- Update all characters that reference this set name.
                    for curCharKey, curCharData in pairs(BMSettings:GetSettings().Characters) do
                        for windowIdx, windowData in ipairs(curCharData.Windows or {}) do
                            for setIdx, oldSetName in ipairs(windowData.Sets or {}) do
                                if oldSetName == label then
                                    BMSettings:GetSettings().Characters[curCharKey].Windows[windowIdx].Sets[setIdx] = newName
                                end
                            end
                        end
                    end
                    newBMHotbar.tabBar:RenameTab(label, newName)
                    BMSettings:SaveSettings(true)
                end
                newBMHotbar.editingLabel = ""
                newBMHotbar.newSetName   = ""
                ImGui.CloseCurrentPopup()
            end
        end
    )

    return newBMHotbar
end

function BMHotbarClass:SetVisible(bVisible)
    BMSettings:GetCharacterWindow(self.id).Visible = bVisible
    self.openGUI = bVisible
    BMSettings:SaveSettings(true)
end

function BMHotbarClass:ToggleVisible()
    BMSettings:GetCharacterWindow(self.id).Visible = not BMSettings:GetCharacterWindow(self.id).Visible
    self.openGUI = BMSettings:GetCharacterWindow(self.id).Visible
    BMSettings:SaveSettings(true)
end

function BMHotbarClass:IsVisible()
    return BMSettings:GetCharacterWindow(self.id).Visible
end

function BMHotbarClass:PerCharacterPositioning()
    return BMSettings:GetCharacterWindow(self.id).PerCharacterPositioning
end

---@return integer, integer
function BMHotbarClass:StartTheme()
    local theme = BMSettings:GetSettings().Themes and BMSettings:GetSettings().Themes[self.id] or nil

    if not theme then
        theme = BMSettings.Globals.CustomThemes and
            BMSettings.Globals.CustomThemes[BMSettings:GetCharacterWindow(self.id).Theme] or nil
    end

    if not theme then
        theme = themes[BMSettings:GetCharacterWindow(self.id).Theme or ""] or nil
    end

    local themeColorPop = 0
    local themeStylePop = 0

    if theme ~= nil then
        for n, t in pairs(theme) do
            if t.color then
                ImGui.PushStyleColor(ImGuiCol[t.element], t.color.r, t.color.g, t.color.b, t.color.a)
                themeColorPop = themeColorPop + 1
            elseif t.stylevar then
                ImGui.PushStyleVar(ImGuiStyleVar[t.stylevar], t.value)
                themeStylePop = themeStylePop + 1
            else
                if type(t) == 'table' then
                    if t['Dynamic_Color'] then
                        local ret, colors = btnUtils.EvaluateLua(t['Dynamic_Color'])
                        if ret then
                            ---@diagnostic disable-next-line: param-type-mismatch
                            ImGui.PushStyleColor(ImGuiCol[n], colors)
                            themeColorPop = themeColorPop + 1
                        end
                    elseif t['Dynamic_Var'] then
                        local ret, var = btnUtils.EvaluateLua(t['Dynamic_Var'])
                        if ret then
                            if type(var) == 'table' then
                                ---@diagnostic disable-next-line: param-type-mismatch, deprecated
                                ImGui.PushStyleVar(ImGuiStyleVar[n], unpack(var))
                            else
                                ---@diagnostic disable-next-line: param-type-mismatch
                                ImGui.PushStyleVar(ImGuiStyleVar[n], var)
                            end
                            themeStylePop = themeStylePop + 1
                        end
                    elseif #t == 4 then
                        local colors = btnUtils.shallowcopy(t)
                        for i = 1, 4 do
                            if type(colors[i]) == 'string' then
                                local ret, color = btnUtils.EvaluateLua(colors[i])
                                if ret then
                                    colors[i] = color
                                end
                            end
                        end
                        ---@diagnostic disable-next-line: param-type-mismatch, deprecated
                        ImGui.PushStyleColor(ImGuiCol[n], unpack(colors))
                        themeColorPop = themeColorPop + 1
                    else
                        ---@diagnostic disable-next-line: param-type-mismatch, deprecated
                        ImGui.PushStyleVar(ImGuiStyleVar[n], unpack(t))
                        themeStylePop = themeStylePop + 1
                    end
                end
            end
        end
    end

    return themeColorPop, themeStylePop
end

---@param themeColorPop integer
---@param themeStylePop integer
function BMHotbarClass:EndTheme(themeColorPop, themeStylePop)
    if themeColorPop > 0 then
        ImGui.PopStyleColor(themeColorPop)
    end
    if themeStylePop > 0 then
        ImGui.PopStyleVar(themeStylePop)
    end
end

function BMHotbarClass:RenderHotbar(flags)
    if not self:IsVisible() then return end

    if self.updateWindowPosSize then
        btnUtils.Debug("Setting new(%d: %s) pos: %d, %d and size: %d, %d", self.id, tostring(self), self.newX, self.newY,
            self.newWidth, self.newHeight)
        self.updateWindowPosSize = false
        ImGui.SetNextWindowSize(self.newWidth, self.newHeight)

        ImGui.SetNextWindowPos(self.newX, self.newY)
        self.lastButtonPageHeight = self.newHeight
        self.lastButtonPageWidth  = self.newWidth
        self.lastWindowX          = self.newX
        self.lastWindowY          = self.newY
    end

    local colorPop, stylePop = self:StartTheme()
    local windowTitle = BMSettings:GetCharacterWindow(self.id).Title or string.format('Button Master - %d', self.id)
    local renderName  = windowTitle

    if self:PerCharacterPositioning() then
        renderName = windowTitle .. "##" .. mq.TLO.EverQuest.Server() .. "_" .. mq.TLO.Me.Name()
    end

    ImGui.PushID("##MainWindow_" .. tostring(self.id))
    self.openGUI, self.shouldDrawGUI = ImGui.Begin(renderName, self.openGUI,
        bit32.bor(flags))

    if not ImGui.IsMouseDown(ImGuiMouseButton.Left) then
        self.lastWindowX, self.lastWindowY = ImGui.GetWindowPos()
        self.lastWindowHeight = ImGui.GetWindowHeight()
        self.lastWindowWidth = ImGui.GetWindowWidth()
    end

    if self.openGUI and self.shouldDrawGUI then
        local startTimeMS = os.clock() * 1000
        local cursorScreenPos = ImGui.GetCursorScreenPosVec()

        self:RenderTabs()

        self:RenderImportButtonPopup()

        local endTimeMS = os.clock() * 1000

        local renderTimeMS = math.ceil(endTimeMS - startTimeMS)

        if btnUtils.enableDebug then
            if renderTimeMS > self.highestRenderTime then self.highestRenderTime = renderTimeMS end
            ImGui.SetWindowFontScale(0.8)
            self:RenderDebugText(cursorScreenPos, tostring(self.highestRenderTime))
            ImGui.SetWindowFontScale(1)
        end
    end

    ImGui.End()
    ImGui.PopID()

    self:EndTheme(colorPop, stylePop)

    if self.openGUI ~= self:IsVisible() then
        self:SetVisible(self.openGUI)
        self.openGUI = true
        if not self:IsVisible() then
            btnUtils.Output("Hotbar %d hidden! Use `/btn %d` to bring it back.", self.id, self.id)
        end
    end

    self.setupComplete = true
end

function BMHotbarClass:RenderTabs()
    local lockedIcon = BMSettings:GetCharacterWindow(self.id).Locked and Icons.FA_LOCK .. '##lockTabButton' or
        Icons.FA_UNLOCK .. '##lockTablButton'

    if BMSettings:GetCharacterWindow(self.id).CompactMode then
        local start_x, start_y = ImGui.GetCursorPos()

        local iconPadding = 2
        local settingsIconSize = math.ceil(((BMSettings:GetCharacterWindow(self.id).ButtonSize or 6) * 10) / 2) -
            iconPadding
        ImGui.PushStyleVar(ImGuiStyleVar.ItemSpacing, 0, iconPadding)
        ImGui.PushStyleVar(ImGuiStyleVar.FramePadding, 0, 0)

        if ImGui.Button(lockedIcon, settingsIconSize, settingsIconSize) then
            --ImGuiWindowFlags.NoMove
            BMSettings:GetCharacterWindow(self.id).Locked = not BMSettings:GetCharacterWindow(self.id).Locked
            BMSettings:SaveSettings(true)
        end

        ImGui.SetCursorPosY(ImGui.GetCursorPosY() + (iconPadding))
        ImGui.Button(Icons.MD_SETTINGS, settingsIconSize, settingsIconSize)
        ImGui.PopStyleVar(2)

        ImGui.SameLine()

        self:RenderTabContextMenu()
        self:RenderCreateTab()
        self.currentSelectedSet = 1

        local style = ImGui.GetStyle()
        ImGui.SetCursorPos(ImVec2(start_x + settingsIconSize + (style.ItemSpacing.x), start_y))

        ImGui.PushStyleVar(ImGuiStyleVar.WindowPadding, 0, 0)
        ImGui.PushStyleVar(ImGuiStyleVar.FramePadding, 0, 0)
        ImGui.BeginChild("##buttons_child", nil, nil, bit32.bor(ImGuiChildFlags.AlwaysAutoResize, ImGuiChildFlags.AutoResizeY))

        if BMSettings:GetCharacterWindowSets(self.id)[1] ~= nil then
            self:RenderButtons(BMSettings:GetCharacterWindowSets(self.id)[1], "")
        end

        ImGui.EndChild()
        ImGui.PopStyleVar(2)
    else
        if ImGui.Button(lockedIcon, WINDOW_SETTINGS_ICON_SIZE, WINDOW_SETTINGS_ICON_SIZE) then
            --ImGuiWindowFlags.NoMove
            BMSettings:GetCharacterWindow(self.id).Locked = not BMSettings:GetCharacterWindow(self.id).Locked
            BMSettings:SaveSettings(true)
        end

        ImGui.SameLine()
        ImGui.Button(Icons.MD_SETTINGS, WINDOW_SETTINGS_ICON_SIZE, WINDOW_SETTINGS_ICON_SIZE)
        ImGui.SameLine()
        self:RenderTabContextMenu()
        self:RenderCreateTab()

        -- Sync tab bar labels in case sets were added/removed via context menu.
        self.tabBar:SetLabels(BMSettings:GetCharacterWindowSets(self.id))

        local selectedLabel, selectedIdx = self.tabBar:Render()

        if selectedLabel and selectedIdx then
            self.currentSelectedSet = selectedIdx
            if BMSettings:GetCharacterWindow(self.id).ShowSearch then
                ImGui.Text("Search")
                ImGui.SameLine()
                self.searchText = ImGui.InputText("##SearchText", self.searchText, ImGuiInputTextFlags.None)
            else
                self.searchText = ""
            end
            self:RenderButtons(selectedLabel, self.searchText)
        else
            ImGui.Text(string.format("No Sets Added! Add one by right-clicking on %s", Icons.MD_SETTINGS))
        end
    end
end

---@param cursorScreenPos ImVec2 # cursor position on screen
---@param text string
function BMHotbarClass:RenderDebugText(cursorScreenPos, text)
    local buttonLabelCol = IM_COL32(255, 0, 0, 255)
    local draw_list = ImGui.GetWindowDrawList()

    draw_list:AddText(ImVec2(cursorScreenPos.x, cursorScreenPos.y), buttonLabelCol, text)
end

function BMHotbarClass:RenderTabContextMenu()
    local openPopup = false

    local unassigned = {}
    local charLoadedSets = {}
    for _, v in ipairs(BMSettings:GetCharacterWindowSets(self.id) or {}) do
        charLoadedSets[v] = true
    end
    for k, _ in pairs(BMSettings:GetSettings().Sets) do
        if charLoadedSets[k] == nil then
            unassigned[k] = true
        end
    end

    if ImGui.BeginPopupContextItem() then
        if btnUtils.getTableSize(unassigned) > 0 then
            if ImGui.BeginMenu("Add Set") then
                for k, _ in pairs(unassigned) do
                    if ImGui.MenuItem(k) then
                        table.insert(BMSettings:GetCharacterWindowSets(self.id), k)
                        BMSettings:SaveSettings(true)
                        break
                    end
                end
                ImGui.EndMenu()
            end
        end

        if ImGui.BeginMenu("Remove Set") then
            for i, v in ipairs(BMSettings:GetCharacterWindowSets(self.id)) do
                if ImGui.MenuItem(v) then
                    table.remove(BMSettings:GetCharConfig().Windows[self.id].Sets, i)
                    BMSettings:SaveSettings(true)
                    break
                end
            end
            ImGui.EndMenu()
        end

        if ImGui.BeginMenu("Delete Set") then
            for k, _ in pairs(BMSettings:GetSettings().Sets) do
                if ImGui.MenuItem(k) then
                    -- clean up any references to this set.
                    for charConfigKey, charConfigValue in pairs(BMSettings:GetSettings().Characters or {}) do
                        for windowKey, windowData in ipairs(charConfigValue.Windows or {}) do
                            for setKey, setName in pairs(windowData.Sets or {}) do
                                if setName == k then
                                    BMSettings:GetSettings().Characters[charConfigKey].Windows[windowKey].Sets[setKey] = nil
                                end
                            end
                        end
                    end
                    BMSettings:GetSettings().Sets[k] = nil
                    BMSettings:SaveSettings(true)
                    break
                end
            end
            ImGui.EndMenu()
        end

        if ImGui.BeginMenu("Delete Hotkey") then
            local sortedKeys = {}
            for k, v in pairs(BMSettings:GetSettings().Buttons) do
                table.insert(sortedKeys,
                    { Label = BMButtonHandlers.ResolveButtonLabel(v, true), id = k, })
            end
            table.sort(sortedKeys, function(a, b) return a.Label < b.Label end)

            local sortedAlphaMenu = {}

            for idx, buttonData in ipairs(sortedKeys) do
                for _, menuGroup in ipairs(self.alphaMenu) do
                    sortedAlphaMenu[menuGroup.name] = sortedAlphaMenu[menuGroup.name] or {}
                    if menuGroup.filter(buttonData.Label:sub(1, 1):upper()) then
                        table.insert(sortedAlphaMenu[menuGroup.name], { idx = idx, buttonData = buttonData, })
                        break
                    end
                end
            end

            for _, menuGroup in pairs(self.alphaMenu) do
                local items = sortedAlphaMenu[menuGroup.name] or {}
                if #items > 0 then
                    if ImGui.BeginMenu(menuGroup.name) then
                        for _, item in ipairs(items) do
                            if ImGui.MenuItem(item.buttonData.Label .. "##delete_menu_" .. tostring(item.idx)) then
                                -- clean up any references to this Button.
                                for setNameKey, setButtons in pairs(BMSettings:GetSettings().Sets) do
                                    for buttonKey, buttonName in pairs(setButtons) do
                                        if buttonName == item.buttonData.id then
                                            BMSettings:GetSettings().Sets[setNameKey][buttonKey] = nil
                                        end
                                    end
                                end
                                BMSettings:GetSettings().Buttons[item.buttonData.id] = nil
                                BMSettings:SaveSettings(true)
                                break
                            end
                        end
                        ImGui.EndMenu()
                    end
                end
            end
            ImGui.EndMenu()
        end

        if ImGui.MenuItem("Create New Set") then
            openPopup = true
        end

        ImGui.Separator()

        if ImGui.BeginMenu("Button Size") then
            for i = 3, 10 do
                local checked = BMSettings:GetCharacterWindow(self.id).ButtonSize == i
                if ImGui.MenuItem(tostring(i), nil, checked) then
                    BMSettings:GetCharacterWindow(self.id).ButtonSize = i
                    self.buttonSizeDirty = true
                    BMSettings:SaveSettings(true)
                    break
                end
            end
            ImGui.EndMenu()
        end

        local font_scale = {
            {
                label = "Tiny",
                size = 8,
            },
            {
                label = "Small",
                size = 9,
            },
            {
                label = "Normal",
                size = 10,
            },
            {
                label = "Large",
                size  = 11,
            },
        }

        if ImGui.BeginMenu("Font Scale") then
            for i, v in ipairs(font_scale) do
                local checked = BMSettings:GetCharacterWindow(self.id).Font == v.size
                if ImGui.MenuItem(v.label, nil, checked) then
                    BMSettings:GetCharacterWindow(self.id).Font = v.size
                    BMSettings:SaveSettings(true)
                    break
                end
            end
            ImGui.EndMenu()
        end

        if ImGui.BeginMenu("Set Theme") then
            local checked = BMSettings:GetCharacterWindow(self.id).Theme == nil
            if ImGui.MenuItem("Default", nil, checked) then
                BMSettings:GetCharacterWindow(self.id).Theme = nil
                BMSettings:SaveSettings(true)
            end
            for n, _ in pairs(themes) do
                checked = (BMSettings:GetCharacterWindow(self.id).Theme or "") == n
                if ImGui.MenuItem(n, nil, checked) then
                    BMSettings:GetCharacterWindow(self.id).Theme = n
                    BMSettings:SaveSettings(true)
                    break
                end
            end
            for n, _ in pairs(BMSettings.Globals.CustomThemes or {}) do
                checked = (BMSettings:GetCharacterWindow(self.id).Theme or "") == n
                if ImGui.MenuItem(n, nil, checked) then
                    BMSettings:GetCharacterWindow(self.id).Theme = n
                    BMSettings:SaveSettings(true)
                    break
                end
            end
            ImGui.EndMenu()
        end

        ImGui.Separator()

        if ImGui.BeginMenu("Share Set") then
            for k, _ in pairs(BMSettings:GetSettings().Sets) do
                if ImGui.MenuItem(k) then
                    BMButtonHandlers:ExportSetToClipBoard(k)
                    btnUtils.Output("Set: '%s' has been copied to your clipboard!", k)
                end
            end
            ImGui.EndMenu()
        end

        if ImGui.MenuItem("Import Button or Set") then
            self.importObjectPopupOpen = true
            self.importText = ImGui.GetClipboardText() or ""
            self.importTextChanged = true
        end

        if ImGui.BeginMenu("Copy Local Set") then
            local charList = {}
            for k, _ in pairs(BMSettings:GetSettings().Characters) do
                local menuItem = k:sub(1, 1):upper() .. k:sub(2)
                menuItem = menuItem:gsub("_", ": ")
                table.insert(charList, { displayName = menuItem, key = k, })
            end
            table.sort(charList, function(a, b) return a.key < b.key end)
            for _, value in ipairs(charList) do
                if ImGui.MenuItem(value.displayName) then
                    CopyLocalSet(value.key)
                end
            end
            ImGui.EndMenu()
        end

        ImGui.Separator()

        if ImGui.BeginMenu("Display Settings") then
            if ImGui.MenuItem((BMSettings:GetCharacterWindow(self.id).HideTitleBar and "Show" or "Hide") .. " Title Bar") then
                BMSettings:GetCharacterWindow(self.id).HideTitleBar = not BMSettings:GetCharacterWindow(self.id)
                    .HideTitleBar
                BMSettings:SaveSettings(true)
            end
            if ImGui.MenuItem((BMSettings:GetCharacterWindow(self.id).CompactMode and "Normal" or "Compact") .. " Mode") then
                BMSettings:GetCharacterWindow(self.id).CompactMode = not BMSettings:GetCharacterWindow(self.id)
                    .CompactMode
                BMSettings:SaveSettings(true)
            end
            if ImGui.MenuItem((BMSettings:GetCharacterWindow(self.id).PerCharacterPositioning and "Global Window Positioning" or "Per Char Window Positioning")) then
                BMSettings:GetCharacterWindow(self.id).PerCharacterPositioning = not BMSettings:GetCharacterWindow(self.id)
                    .PerCharacterPositioning
                BMSettings:SaveSettings(true)
            end
            if ImGui.MenuItem((BMSettings:GetCharacterWindow(self.id).AdvTooltips and "Disable" or "Enable") .. " Advanced Tooltips") then
                BMSettings:GetCharacterWindow(self.id).AdvTooltips = not BMSettings:GetCharacterWindow(self.id)
                    .AdvTooltips
                BMSettings:SaveSettings(true)
            end
            if ImGui.MenuItem((BMSettings:GetCharacterWindow(self.id).HideScrollbar and "Show" or "Hide") .. " Scrollbar") then
                BMSettings:GetCharacterWindow(self.id).HideScrollbar = not BMSettings:GetCharacterWindow(self.id)
                    .HideScrollbar
                BMSettings:SaveSettings(true)
            end
            if ImGui.MenuItem((BMSettings:GetCharacterWindow(self.id).ShowSearch and "Disable" or "Enable") .. " Search") then
                BMSettings:GetCharacterWindow(self.id).ShowSearch = not BMSettings:GetCharacterWindow(self.id)
                    .ShowSearch
                BMSettings:SaveSettings(true)
            end
            local fps_scale = {
                {
                    label = "Instant",
                    fps = 0,
                },
                {
                    label = "10 FPS",
                    fps = 1,
                },
                {
                    label = "4 FPS",
                    fps = 2.5,
                },
                {
                    label = "1 FPS",
                    fps   = 10,
                },
            }

            ImGui.Separator()
            ImGui.Text("Window Name:")
            local titleTmp, titleChanged = ImGui.InputText("##winTitle_" .. self.id, self.windowTitleEdit, 0)
            if titleChanged then self.windowTitleEdit = titleTmp end
            if ImGui.Button("Rename##winRename_" .. self.id) then
                local newTitle = self.windowTitleEdit:len() > 0 and self.windowTitleEdit or nil
                BMSettings:GetCharacterWindow(self.id).Title = newTitle
                BMSettings:SaveSettings(true)
                self.windowTitleEdit = ""
            end

            if ImGui.BeginMenu("Update FPS") then
                for _, v in ipairs(fps_scale) do
                    local checked = BMSettings:GetCharacterWindow(self.id).FPS == v.fps
                    if ImGui.MenuItem(v.label, nil, checked) then
                        BMSettings:GetCharacterWindow(self.id).FPS = v.fps
                        BMSettings:SaveSettings(true)
                        break
                    end
                end
                ImGui.EndMenu()
            end
            -- TODO: Make this a reference to a character since it can dynamically change.
            --if ImGui.MenuItem("Save Layout as Default") then
            --    BMSettings:GetSettings().Defaults = {
            --        width = self.lastButtonPageWidth,
            --        height = self.lastButtonPageHeight,
            --        x = self.lastWindowX,
            --        y = self.lastWindowY,
            --        CharSettings = BMSettings:GetCharConfig(),
            --    }
            --    BMSettings:SaveSettings(true)
            --end
            ImGui.EndMenu()
        end

        if ImGui.MenuItem("Create New Hotbar") then
            table.insert(BMHotbars, BMHotbarClass.new(BMSettings:GetNextWindowId(), true))
        end

        if ImGui.BeginMenu("Show/Hide Hotbar") then
            for hbIdx, hotbarClass in ipairs(BMHotbars) do
                local hbTitle = BMSettings:GetCharacterWindow(hbIdx).Title or string.format("Button Master - %d", hbIdx)
                if ImGui.MenuItem(hbTitle, nil, hotbarClass:IsVisible()) then
                    hotbarClass:ToggleVisible()
                end
            end
            ImGui.EndMenu()
        end

        --[[if ImGui.MenuItem("Replicate Size/Pos") then
            local x, y = ImGui.GetWindowPos()
            ButtonActors.send({
                from = mq.TLO.Me.DisplayName(),
                script = "ButtonMaster",
                event = "CopyLoc",
                width = self.lastButtonPageWidth,
                height = self.lastButtonPageHeight,
                x = self.lastWindowX,
                y = self.lastWindowY,
                windowId = self.id,
                hideTitleBar = BMSettings:GetCharacterWindow(self.id).HideTitleBar,
                compactMode = BMSettings:GetCharacterWindow(self.id).CompactMode,
            })
        end]]

        ImGui.Separator()

        if ImGui.BeginMenu("Dev") then
            if ImGui.MenuItem((btnUtils.enableDebug and "Disable" or "Enable") .. " Debug") then
                btnUtils.enableDebug = not btnUtils.enableDebug
            end
            if ImGui.MenuItem("Remove All Duped Buttons") then
                local duplicatekeys = Set.new({})
                for buttonKey, buttonData in pairs(BMSettings:GetSettings().Buttons or {}) do
                    btnUtils.Output("\awTesting Button: \am%s", buttonKey)
                    for curBtnKey, curBtn in pairs(BMSettings:GetSettings().Buttons or {}) do
                        if buttonKey ~= curBtnKey and curBtn.Cmd == buttonData.Cmd then
                            btnUtils.Output("\awButton: \am%s \awis a duplicate!", buttonKey)
                            duplicatekeys:add(curBtnKey)
                            duplicatekeys:add(buttonKey)
                            break
                        end
                    end
                end

                for _, key in ipairs(duplicatekeys:toList()) do
                    btnUtils.Output("\awDuplicate: \am%s \aw(\at%s\aw)", key, BMSettings:GetSettings().Buttons[key].Label)
                    local isUsed = false
                    for _, setButtons in pairs(BMSettings:GetSettings().Sets) do
                        for _, buttonName in pairs(setButtons) do
                            if buttonName == key then
                                isUsed = true
                            end
                        end
                    end

                    if isUsed then
                        btnUtils.Output("   \ag-> Used")
                    else
                        if BMSettings:GetSettings().Buttons[key] then
                            btnUtils.Output("   \ay-> Unused - Removing!")
                            BMSettings:GetSettings().Buttons[key] = nil
                        else
                            btnUtils.Output("   \ay-> Unused - Previosuly Removed!")
                        end
                    end
                end
                BMSettings:SaveSettings(true)
            end
            ImGui.EndMenu()
        end

        ImGui.EndPopup()
    end

    if openPopup and ImGui.IsPopupOpen(editTabPopup) == false then
        ImGui.OpenPopup(editTabPopup)
        openPopup = false
    end
end

function BMHotbarClass:RenderContextMenu(Set, Index, buttonID)
    local button = BMSettings:GetButtonBySetIndex(Set, Index)

    if ImGui.BeginPopupContextItem(buttonID) then
        local unassigned = {}
        local keys = {}
        for _, v in pairs(BMSettings:GetSettings().Sets[Set] or {}) do keys[v] = true end
        for k, v in pairs(BMSettings:GetSettings().Buttons) do
            if keys[k] == nil then
                unassigned[k] = v
            end
        end
        --editPopupName = "edit_button_popup|" .. Index
        -- only list hotkeys that aren't already assigned to the button set
        if btnUtils.getTableSize(unassigned) > 0 then
            if ImGui.BeginMenu("Assign Hotkey") then
                -- hytiek: BEGIN ADD
                -- Create an array to store the sorted keys
                local sortedKeys = {}

                -- Populate the array with non-nil keys from the original table
                for key, value in pairs(unassigned) do
                    if value ~= nil then
                        table.insert(sortedKeys, key)
                    end
                end

                -- Sort the keys based on the Label field
                table.sort(sortedKeys, function(a, b)
                    local labelA = unassigned[a] and BMButtonHandlers.ResolveButtonLabel(unassigned[a], true)
                    local labelB = unassigned[b] and BMButtonHandlers.ResolveButtonLabel(unassigned[b], true)
                    return labelA < labelB
                end)

                local sortedAlphaMenu = {}

                for idx, key in ipairs(sortedKeys) do
                    local value = unassigned[key]
                    if value ~= nil then
                        for _, menuGroup in ipairs(self.alphaMenu) do
                            sortedAlphaMenu[menuGroup.name] = sortedAlphaMenu[menuGroup.name] or {}
                            local label = BMButtonHandlers.ResolveButtonLabel(value, true)
                            if menuGroup.filter(label:sub(1, 1):upper()) then
                                table.insert(sortedAlphaMenu[menuGroup.name], { idx = idx, key = key, value = value, label = label, })
                                break
                            end
                        end
                    end
                end


                for _, menuGroup in pairs(self.alphaMenu) do
                    local items = sortedAlphaMenu[menuGroup.name] or {}
                    if #items > 0 then
                        if ImGui.BeginMenu(menuGroup.name) then
                            for _, item in ipairs(items) do
                                if ImGui.MenuItem(item.label .. "##assign_menu_" .. tostring(item.idx)) then
                                    BMSettings:GetSettings().Sets[Set][Index] = item.key
                                    BMSettings:SaveSettings(true)
                                    break
                                end
                            end
                            ImGui.EndMenu()
                        end
                    end
                end

                ImGui.EndMenu()
            end
        end

        -- only show create new for unassigned buttons
        if button.Unassigned == true then
            if ImGui.MenuItem("Create New") then
                BMEditPopup:OpenEditPopup(Set, Index)
            end
        else
            if ImGui.MenuItem("Edit") then
                BMEditPopup:OpenEditPopup(Set, Index)
            end
            if ImGui.MenuItem("Unassign") then
                BMSettings:GetSettings().Sets[Set][Index] = nil
                BMSettings:SaveSettings(true)
            end
            if ImGui.MenuItem("Delete") then
                local buttonID = BMSettings:GetSettings().Sets[Set][Index]
                for setNameKey, setButtons in pairs(BMSettings:GetSettings().Sets) do
                    for buttonKey, buttonName in pairs(setButtons) do
                        if buttonName == buttonID then
                            BMSettings:GetSettings().Sets[setNameKey][buttonKey] = nil
                        end
                    end
                end
                BMSettings:GetSettings().Buttons[buttonID] = nil
                BMSettings:SaveSettings(true)
            end

            if ImGui.MenuItem(Icons.MD_SHARE) then
                BMButtonHandlers.ExportButtonToClipBoard(button)
            end
            btnUtils.Tooltip("Copy contents of this button to share with friends.")
        end

        ImGui.EndPopup()
    end
end

---@param Set string
---@param searchText string
function BMHotbarClass:RenderButtons(Set, searchText)
    ImGui.PushStyleVar(ImGuiStyleVar.ItemSpacing, ImVec2(4, 4))
    if ImGui.GetWindowWidth() ~= self.lastButtonPageWidth or ImGui.GetWindowHeight() ~= self.lastButtonPageHeight or self.buttonSizeDirty then
        self:RecalculateVisibleButtons(Set)
    end

    local btnSize = (BMSettings:GetCharacterWindow(self.id).ButtonSize or 6) * 10

    local renderButtonCount = self.visibleButtonCount

    for ButtonIndex = 1, renderButtonCount do
        local button = BMSettings:GetButtonBySetIndex(Set, ButtonIndex)
        local searchMatch = true

        if searchText:len() > 0 then
            searchMatch =
                (button.CachedLabel or ""):lower():find(searchText:lower()) ~= nil
                or
                (button.Cmd or ""):lower():find(searchText:lower()) ~= nil
        end

        if searchMatch then
            local clicked = false

            local buttonID = string.format("##Button_%s_%d", Set, ButtonIndex)
            local showLabel = true
            local btnKey = BMSettings:GetButtonSectionKeyBySetIndex(Set, ButtonIndex)
            if BMSettings.settings.Buttons[btnKey] ~= nil and BMSettings.settings.Buttons[btnKey].ShowLabel ~= nil then
                showLabel = BMSettings.settings.Buttons[btnKey].ShowLabel
            end
            ImGui.PushID(buttonID)
            clicked = BMButtonHandlers.Render(button, btnSize, showLabel, (BMSettings:GetCharacterWindow(self.id).Font or 10) / 10,
                BMSettings:GetCharacterWindow(self.id).AdvTooltips)
            ImGui.PopID()
            -- TODO Move this to button config class and out of the UI thread.
            if clicked then
                if button.Unassigned then
                    BMEditPopup:CreateButtonFromCursor(Set, ButtonIndex)
                else
                    BMButtonHandlers.Exec(button)
                end
            else
                -- setup drag and drop
                if ImGui.BeginDragDropSource() then
                    self.currentDnDData = { Set = Set, Index = ButtonIndex, }
                    ImGui.SetDragDropPayload("BTN", self.id)
                    ImGui.Button(button.Label, btnSize, btnSize)
                    ImGui.EndDragDropSource()
                end
                if ImGui.BeginDragDropTarget() then
                    local payload = ImGui.AcceptDragDropPayload("BTN")

                    if payload ~= nil then
                        ---@diagnostic disable-next-line: undefined-field
                        local dndData = BMHotbars[payload.Data].currentDnDData
                        local success = dndData ~= nil
                        if success then
                            local to_set = dndData.Set
                            local to_num = dndData.Index
                            btnUtils.Output("Dropping button from set '" ..
                                tostring(to_set) .. "' index " .. tostring(to_num) .. " to set '" .. tostring(Set) .. "' index " .. tostring(ButtonIndex))

                            -- swap the keys in the button set
                            BMSettings:GetSettings().Sets[to_set][to_num], BMSettings:GetSettings().Sets[Set][ButtonIndex] =
                                BMSettings:GetSettings().Sets[Set][ButtonIndex], BMSettings:GetSettings().Sets[to_set][to_num]
                            BMSettings:SaveSettings(true)
                        else
                            btnUtils.Output("\arError: Failed to decode dropped button payload :: %s!\ax", payload.Data or "nil")
                        end
                    end

                    local rgmercsPayload = ImGui.AcceptDragDropPayload("RGMERCS_BUTTON")
                    if rgmercsPayload ~= nil then
                        local success, sharableButton = btnUtils.decodeTable(rgmercsPayload.Data)
                        if success and type(sharableButton) == 'table' and sharableButton.Type == "Button" then
                            BMSettings:GetSettings().Sets[Set][ButtonIndex] = BMSettings:ImportButtonAndSave(sharableButton.Button, false)
                            BMSettings:SaveSettings(true)
                        else
                            btnUtils.Output("\arError: Failed to decode dropped RGMercs button!\ax")
                        end
                    end
                    ImGui.EndDragDropTarget()
                end

                self:RenderContextMenu(Set, ButtonIndex, buttonID)
            end

            -- button grid
            if ButtonIndex % self.cachedCols ~= 0 then ImGui.SameLine() end
        end
    end
    ImGui.PopStyleVar(1)
end

function BMHotbarClass:RecalculateVisibleButtons(Set)
    self.buttonSizeDirty = false

    btnUtils.Debug("\arHave old lW=%d lH=%d", self.lastButtonPageWidth, self.lastButtonPageHeight)

    self.lastButtonPageWidth = ImGui.GetWindowWidth()
    self.lastButtonPageHeight = ImGui.GetWindowHeight()

    btnUtils.Debug("\arSetting new lW=%d lH=%d", self.lastButtonPageWidth, self.lastButtonPageHeight)

    local cursorX, cursorY = ImGui.GetCursorPos() -- this will get us the x pos we start at which tells us of the offset from the main window border
    local style = ImGui.GetStyle()                -- this will get us ItemSpacing.x which is the amount of space between buttons

    -- global button configs
    local btnSize = (BMSettings:GetCharacterWindow(self.id).ButtonSize or 6) * 10
    self.cachedCols = math.floor((self.lastButtonPageWidth - cursorX) / (btnSize + style.ItemSpacing.x))
    self.cachedRows = math.floor((self.lastButtonPageHeight - cursorY) / (btnSize + style.ItemSpacing.y))

    local count = 100
    if self.cachedRows * self.cachedCols < 100 then count = self.cachedRows * self.cachedCols end

    -- get the last assigned button and make sure it is visible.
    local lastAssignedButton = 1
    for i = 1, 100 do if not BMSettings:GetButtonBySetIndex(Set, i).Unassigned then lastAssignedButton = i end end

    -- if the last forced visible buttons isn't the last in a row then render to the end of that row.
    -- stay with me here. The last button needs to look at the number of buttons per row (cols) and
    -- the position of this button in that row (button%cols) and add enough to get to the end of the row.
    if lastAssignedButton % self.cachedCols ~= 0 then
        lastAssignedButton = lastAssignedButton + (self.cachedCols - (lastAssignedButton % self.cachedCols))
    end

    self.visibleButtonCount = math.min(math.max(count, lastAssignedButton), 100)
end

function BMHotbarClass:RenderImportButtonPopup()
    if not self.importObjectPopupOpen then return end

    local shouldDrawImportPopup = false

    self.importObjectPopupOpen, shouldDrawImportPopup = ImGui.Begin("Import Button or Set", self.importObjectPopupOpen,
        ImGuiWindowFlags.None)
    if ImGui.GetWindowWidth() < 500 or ImGui.GetWindowHeight() < 100 then
        ImGui.SetWindowSize(math.max(500, ImGui.GetWindowWidth()), math.max(100, ImGui.GetWindowHeight()))
    end
    if self.importObjectPopupOpen and shouldDrawImportPopup then
        if ImGui.SmallButton(Icons.MD_CONTENT_PASTE) then
            self.importText = ImGui.GetClipboardText()
            self.importTextChanged = true
        end
        btnUtils.Tooltip("Paste from Clipboard")
        ImGui.SameLine()

        if self.importTextChanged then
            self.validDecode, self.decodedObject = btnUtils.decodeTable(self.importText)
            self.validDecode = type(self.decodedObject) == 'table' and self.validDecode or false
        end

        if self.validDecode then
            ImGui.PushStyleColor(ImGuiCol.Text, 0.02, 0.8, 0.02, 1.0)
        else
            ImGui.PushStyleColor(ImGuiCol.Text, 0.8, 0.02, 0.02, 1.0)
        end
        self.importText, self.importTextChanged = ImGui.InputText(
            (self.validDecode and Icons.MD_CHECK or Icons.MD_NOT_INTERESTED) .. " Import Code", self.importText,
            ImGuiInputTextFlags.None)
        ImGui.PopStyleColor()

        -- save button
        if self.validDecode and self.decodedObject then
            if ImGui.Button("Import " .. (self.decodedObject.Type or "Failed")) then
                if self.decodedObject.Type == "Button" then
                    BMSettings:ImportButtonAndSave(self.decodedObject.Button, true)
                elseif self.decodedObject.Type == "Set" then
                    BMSettings:ImportSetAndSave(self.decodedObject, self.id)
                else
                    btnUtils.Output("\arError: imported object was not a button or a set!")
                end
                -- reset everything
                self.decodedObject = {}
                self.importText = ""
                self.importObjectPopupOpen = false
            end
        end
    end
    ImGui.End()
end

function BMHotbarClass:RenderCreateTab()
    if ImGui.BeginPopup(editTabPopup) then
        ImGui.Text("New Button Set:")
        local tmp, selected = ImGui.InputText("##edit", '', 0)
        if selected then self.newSetName = tmp end
        if ImGui.Button("Save") then
            if self.newSetName ~= nil and self.newSetName:len() > 0 then
                if BMSettings:GetSettings().Sets[self.newSetName] == nil then
                    table.insert(BMSettings:GetCharConfig().Windows[self.id].Sets, self.newSetName)
                    BMSettings:GetSettings().Sets[self.newSetName] = {}
                    BMSettings:SaveSettings(true)
                else
                    btnUtils.Output("\arError Saving Set: A set with this name already exists!\ax")
                end
            else
                btnUtils.Output("\arError Saving Set: Name cannot be empty.\ax")
            end
            ImGui.CloseCurrentPopup()
        end
        ImGui.EndPopup()
    end
end

function BMHotbarClass:ReloadConfig()
    local config = BMSettings:GetCharacterWindow(self.id)
    btnUtils.Debug("\ayWindow(%d: %s) config: \n%s", self.id, tostring(self), btnUtils.dumpTable(config))
    self.updateWindowPosSize           = true
    self.newWidth                      = config.Width or 100
    self.newHeight                     = config.Height or 40
    self.newX                          = config.Pos and (config.Pos.x or 10)
    self.newY                          = config.Pos and (config.Pos.y or 10)
    self.buttonSizeDirty               = true

    self.lastWindowX, self.lastWindowY = self.newX, self.newY
    self.lastWindowWidth               = self.newWidth
    self.lastWindowHeight              = self.newHeight
    btnUtils.Debug("\agWindow(%d: %s) config set!", self.id, tostring(self))
end

function BMHotbarClass:GiveTime()
    local now = os.clock()

    -- update every visible button to save on our FPS.
    if not BMSettings:GetCharacterWindow(self.id).FPS then
        BMSettings:GetCharacterWindow(self.id).FPS = 0
        BMSettings:SaveSettings(true)
    end

    local fps = BMSettings:GetCharacterWindow(self.id).FPS / 10

    if now - self.lastFrameTime < fps then return end
    self.lastFrameTime = now

    for i, set in ipairs(BMSettings:GetCharacterWindowSets(self.id)) do
        if self.currentSelectedSet == i then
            --btnUtils.Debug("Caching Visibile Buttons for Set: %s / %d", set, i)
            local renderButtonCount = self.visibleButtonCount

            for ButtonIndex = 1, renderButtonCount do
                local button = BMSettings:GetButtonBySetIndex(set, ButtonIndex)

                BMButtonHandlers.EvaluateAndCache(button)
            end
        end
    end

    local config = BMSettings:GetCharacterWindow(self.id)

    if config then
        if self.setupComplete and not BMUpdateSettings then -- wont have valid positions until the render loop has run once.
            if not config.Pos or (config.Pos.x ~= self.lastWindowX or config.Pos.y ~= self.lastWindowY) or config.Height ~= self.lastWindowHeight or config.Width ~= self.lastWindowWidth then
                config.Pos    = config.Pos or {}
                config.Pos.x  = self.lastWindowX
                config.Pos.y  = self.lastWindowY
                config.Height = self.lastWindowHeight
                config.Width  = self.lastWindowWidth
                BMSettings:SaveSettings(true)
            end
        end
    else
        btnUtils.Output("\ayError: No config found for bar: %d", self.id)
    end
end

return BMHotbarClass