mirror of
https://github.com/brues-code/pfUI.git
synced 2026-10-31 07:56:57 +00:00
Brues
4616c2bc5f
Each if/else that only showed one object in one branch and hid it in the other is now ClassicAPI's SetShown, which shows on any truthy value. 21 sites across 10 modules.
437 lines
17 KiB
Lua
437 lines
17 KiB
Lua
pfUI:RegisterModule("addonbuttons", function ()
|
|
if not pfUI.minimap then return end
|
|
if C.abuttons.enable == "0" then return end
|
|
|
|
local rawborder, default_border = GetBorderSize("panels")
|
|
|
|
pfUI_cache["abuttons"] = pfUI_cache["abuttons"] or {}
|
|
pfUI_cache["abuttons"]["add"] = pfUI_cache["abuttons"]["add"] or {}
|
|
pfUI_cache["abuttons"]["del"] = pfUI_cache["abuttons"]["del"] or {}
|
|
|
|
local ignored_icons = {
|
|
"Note",
|
|
"JQuest",
|
|
"Naut_",
|
|
"MinimapIcon",
|
|
"GatherMatePin",
|
|
"WestPointer",
|
|
"Chinchilla_",
|
|
"SmartMinimapZoom",
|
|
"QuestieNote",
|
|
"smm",
|
|
"pfMiniMapPin",
|
|
"MiniMapBattlefieldFrame",
|
|
"pfMinimapButton",
|
|
"GatherNote",
|
|
"MiniNotePOI",
|
|
"FWGMinimapPOI",
|
|
"RecipeRadarMinimapIcon",
|
|
"MiniMapTracking",
|
|
"CartographerNotesPOI"
|
|
}
|
|
|
|
pfUI.addonbuttons = CreateFrame("Frame", "pfMinimapButtons", UIParent)
|
|
pfUI.addonbuttons:SetFrameStrata("HIGH")
|
|
CreateBackdrop(pfUI.addonbuttons)
|
|
CreateBackdropShadow(pfUI.addonbuttons)
|
|
|
|
pfUI.addonbuttons.minimapbutton = CreateFrame("Button", "pfMinimapButton", pfUI.minimap or UIParent)
|
|
pfUI.addonbuttons.minimapbutton:SetFrameLevel(24)
|
|
pfUI.addonbuttons.minimapbutton:SetSize(12, 12)
|
|
|
|
pfUI.addonbuttons.minimapbutton:SetScript("OnClick", function()
|
|
pfUI.addonbuttons:SetShown(not pfUI.addonbuttons:IsShown())
|
|
end)
|
|
|
|
pfUI.addonbuttons.buttons = {}
|
|
pfUI.addonbuttons.overrides = {}
|
|
pfUI.addonbuttons.last_updated = 0
|
|
pfUI.addonbuttons.rows = 1
|
|
pfUI.addonbuttons.max_button_size = 40
|
|
|
|
local function GetButtonSize()
|
|
if C.abuttons.position == "bottom" or C.abuttons.position == "top" then
|
|
return (pfUI.minimap:GetWidth() - (tonumber(C.abuttons.spacing) * (tonumber(C.abuttons.rowsize) + 1))) / tonumber(C.abuttons.rowsize)
|
|
else
|
|
return (pfUI.minimap:GetHeight() - (tonumber(C.abuttons.spacing) * (tonumber(C.abuttons.rowsize) + 1))) / tonumber(C.abuttons.rowsize)
|
|
end
|
|
end
|
|
|
|
local function GetNumButtons()
|
|
local total_buttons = 0
|
|
for i, v in ipairs(pfUI.addonbuttons.buttons) do
|
|
total_buttons = total_buttons + 1
|
|
end
|
|
return total_buttons
|
|
end
|
|
|
|
local function GetStringSize()
|
|
return (GetButtonSize() + tonumber(C.abuttons.spacing))
|
|
end
|
|
|
|
local function TableMatch(table, needle)
|
|
for i,v in ipairs(table) do
|
|
if (strlower(v) == strlower(needle)) then
|
|
return i
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function TablePartialMatch(table, needle)
|
|
for i,v in ipairs(table) do
|
|
local pos_start, pos_end = strfind(strlower(needle), strlower(v))
|
|
if pos_start == 1 then
|
|
return i
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function IsButtonValid(frame)
|
|
-- ignore invalid and invisible frames
|
|
if not frame:GetName() then return false end
|
|
if not frame:IsVisible() then return false end
|
|
|
|
-- ignore frames with invalid sizes
|
|
if frame:GetHeight() > pfUI.addonbuttons.max_button_size then return false end
|
|
if frame:GetWidth() > pfUI.addonbuttons.max_button_size then return false end
|
|
|
|
-- ignore other frame types
|
|
if not frame:IsFrameType("Button") and not frame:IsFrameType("Frame") then return false end
|
|
|
|
-- ignore manually selected frames
|
|
if TablePartialMatch(ignored_icons, frame:GetName()) then return false end
|
|
|
|
-- check button/frame script handlers
|
|
if frame:IsFrameType("Button") then
|
|
if frame:GetScript("OnClick") or frame:GetScript("OnMouseDown") or frame:GetScript("OnMouseUp") then
|
|
return true
|
|
end
|
|
elseif frame:IsFrameType("Frame") and (strfind(strlower(frame:GetName()), "icon") or strfind(strlower(frame:GetName()), "button")) then
|
|
if frame:GetScript("OnMouseDown") or frame:GetScript("OnMouseUp") then
|
|
return true
|
|
end
|
|
end
|
|
|
|
-- ignore everything else
|
|
return false
|
|
end
|
|
|
|
local function FindButtons(frame)
|
|
for i, frame_child in ipairs({frame:GetChildren()}) do
|
|
-- check first level children
|
|
if IsButtonValid(frame_child) and not TableMatch(pfUI.addonbuttons.buttons, frame_child:GetName()) then
|
|
table.insert(pfUI.addonbuttons.buttons, frame_child:GetName())
|
|
else
|
|
if frame_child:GetNumChildren() > 0 then
|
|
for j, child_child in ipairs({frame_child:GetChildren()}) do
|
|
if IsButtonValid(child_child) and not TableMatch(pfUI.addonbuttons.buttons, child_child:GetName()) then
|
|
table.insert(pfUI.addonbuttons.buttons, child_child:GetName())
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local function GetScale()
|
|
local sum_size, buttons_count, calculated_scale
|
|
sum_size = 0
|
|
buttons_count = GetNumButtons()
|
|
for i, button_name in ipairs(pfUI.addonbuttons.buttons) do
|
|
if _G[button_name] ~= nil then
|
|
sum_size = sum_size + _G[button_name]:GetHeight()
|
|
end
|
|
end
|
|
calculated_scale = GetButtonSize() / (sum_size / buttons_count)
|
|
return calculated_scale > 1 and 1 or calculated_scale
|
|
end
|
|
|
|
local function ScanForButtons()
|
|
FindButtons(Minimap)
|
|
FindButtons(MinimapBackdrop)
|
|
end
|
|
|
|
local function SetupMainFrame()
|
|
pfUI.addonbuttons:ClearAllPoints()
|
|
pfUI.addonbuttons:SetScale(pfUI.minimap:GetScale())
|
|
|
|
pfUI.addonbuttons.minimapbutton:ClearAllPoints()
|
|
local mbtnWidth, mbtnHeight = pfUI.minimap:GetSize()
|
|
local dynamicSize = ceil((GetNumButtons() > 0 and GetNumButtons() or 1) / tonumber(C.abuttons.rowsize)) * GetStringSize() + tonumber(C.abuttons.spacing)
|
|
if C.abuttons.position == "bottom" then
|
|
pfUI.addonbuttons:SetSize(mbtnWidth, dynamicSize)
|
|
pfUI.addonbuttons:SetPoint("TOP", pfUI.minimap, "BOTTOM", 0 , -default_border * 3)
|
|
SkinArrowButton(pfUI.addonbuttons.minimapbutton, "down")
|
|
pfUI.addonbuttons.minimapbutton:SetPoint("BOTTOM", pfUI.minimap, "BOTTOM", 0, 4)
|
|
elseif C.abuttons.position == "left" then
|
|
pfUI.addonbuttons:SetSize(dynamicSize, mbtnHeight)
|
|
pfUI.addonbuttons:SetPoint("TOPRIGHT", pfUI.minimap, "TOPLEFT", -default_border * 3, 0)
|
|
SkinArrowButton(pfUI.addonbuttons.minimapbutton, "left")
|
|
pfUI.addonbuttons.minimapbutton:SetPoint("LEFT", pfUI.minimap, "LEFT", 4, 0)
|
|
elseif C.abuttons.position == "top" then
|
|
pfUI.addonbuttons:SetSize(mbtnWidth, dynamicSize)
|
|
pfUI.addonbuttons:SetPoint("BOTTOM", pfUI.minimap, "TOP", 0 , default_border * 3)
|
|
SkinArrowButton(pfUI.addonbuttons.minimapbutton, "up")
|
|
pfUI.addonbuttons.minimapbutton:SetPoint("TOP", pfUI.minimap, "TOP", 0, -4)
|
|
elseif C.abuttons.position == "right" then
|
|
pfUI.addonbuttons:SetSize(dynamicSize, mbtnHeight)
|
|
pfUI.addonbuttons:SetPoint("TOPLEFT", pfUI.minimap, "TOPRIGHT", default_border * 3, 0)
|
|
SkinArrowButton(pfUI.addonbuttons.minimapbutton, "right")
|
|
pfUI.addonbuttons.minimapbutton:SetPoint("RIGHT", pfUI.minimap, "RIGHT", -4, 0)
|
|
end
|
|
end
|
|
|
|
local function UpdatePanel()
|
|
ScanForButtons()
|
|
for i, button_name in ipairs(pfUI_cache["abuttons"]["add"]) do
|
|
if not TableMatch(pfUI.addonbuttons.buttons, button_name) then
|
|
if _G[button_name] ~= nil then
|
|
table.insert(pfUI.addonbuttons.buttons, button_name)
|
|
end
|
|
end
|
|
end
|
|
for i, button_name in ipairs(pfUI_cache["abuttons"]["del"]) do
|
|
if TableMatch(pfUI.addonbuttons.buttons, button_name) then
|
|
table.remove(pfUI.addonbuttons.buttons, TableMatch(pfUI.addonbuttons.buttons, button_name))
|
|
end
|
|
end
|
|
for i, button_name in ipairs(pfUI.addonbuttons.buttons) do
|
|
if _G[button_name] == nil then
|
|
table.remove(pfUI.addonbuttons.buttons, TableMatch(pfUI.addonbuttons.buttons, button_name))
|
|
end
|
|
end
|
|
SetupMainFrame()
|
|
end
|
|
|
|
local parent
|
|
local function GetTopFrame(frame)
|
|
parent = frame:GetParent()
|
|
if not parent or parent == Minimap or parent == MinimapBackdrop or parent == UIParent then
|
|
return frame
|
|
else
|
|
return GetTopFrame(parent)
|
|
end
|
|
end
|
|
|
|
local function BackupButton(frame)
|
|
if not frame then return end
|
|
if frame.backup == nil then
|
|
frame.backup = {}
|
|
frame.backup.top_frame_name = GetTopFrame(frame):GetName()
|
|
frame.backup.parent_name = GetTopFrame(frame):GetParent():GetName()
|
|
frame.backup.is_clamped_to_screen = frame:IsClampedToScreen()
|
|
frame.backup.is_movable = frame:IsMovable()
|
|
frame.backup.point = {frame:GetPoint()}
|
|
frame.backup.size = {frame:GetSize()}
|
|
frame.backup.scale = frame:GetScale()
|
|
if frame:HasScript("OnDragStart") then
|
|
frame.backup.on_drag_start = frame:GetScript("OnDragStart")
|
|
end
|
|
if frame:HasScript("OnDragStop") then
|
|
frame.backup.on_drag_stop = frame:GetScript("OnDragStop")
|
|
end
|
|
if frame:HasScript("OnUpdate") then
|
|
frame.backup.on_update = frame:GetScript("OnUpdate")
|
|
end
|
|
-- TODO: find a way to avoid such hardcoding
|
|
if frame:GetName() == "MetaMapButton" then
|
|
frame.backup.MetaMapButton_UpdatePosition = MetaMapButton_UpdatePosition
|
|
pfUI.addonbuttons.overrides.MetaMapButton_UpdatePosition = function () return end
|
|
end
|
|
end
|
|
end
|
|
|
|
local function RestoreButton(frame)
|
|
if frame.backup ~= nil then
|
|
_G[frame.backup.top_frame_name]:SetParent(frame.backup.parent_name)
|
|
frame:SetClampedToScreen(frame.backup.is_clamped_to_screen)
|
|
frame:SetMovable(frame.backup.is_movable)
|
|
frame:SetScale(frame.backup.scale)
|
|
frame:SetSize(frame.backup.size[2], frame.backup.size[1])
|
|
frame:ClearAllPoints()
|
|
frame:SetPoint(frame.backup.point[1], frame.backup.point[2], frame.backup.point[3], frame.backup.point[4], frame.backup.point[5])
|
|
if frame.backup.on_drag_start ~= nil then
|
|
frame:SetScript("OnDragStart", frame.backup.on_drag_start)
|
|
end
|
|
if frame.backup.on_drag_stop ~= nil then
|
|
frame:SetScript("OnDragStop", frame.backup.on_drag_stop)
|
|
end
|
|
if frame.backup.on_update ~= nil then
|
|
frame:SetScript("OnUpdate", frame.backup.on_update)
|
|
end
|
|
if frame.backup.MetaMapButton_UpdatePosition ~= nil then
|
|
pfUI.addonbuttons.overrides.MetaMapButton_UpdatePosition = frame.backup.MetaMapButton_UpdatePosition
|
|
end
|
|
end
|
|
end
|
|
|
|
local function MoveButton(index, frame)
|
|
if not frame then return end
|
|
local top_frame, row_index, offsetX, offsetY, final_scale
|
|
top_frame = GetTopFrame(frame)
|
|
final_scale = GetScale()
|
|
row_index = floor((index-1)/tonumber(C.abuttons.rowsize))
|
|
frame:SetFrameStrata("HIGH")
|
|
frame:SetClampedToScreen(true)
|
|
frame:SetMovable(false)
|
|
frame:SetScript("OnDragStart", nil)
|
|
frame:SetScript("OnDragStop", nil)
|
|
frame:SetScript("OnUpdate", nil)
|
|
frame:SetClampedToScreen(true)
|
|
frame:SetMovable(false)
|
|
frame:ClearAllPoints()
|
|
|
|
if top_frame ~= pfUI.addonbuttons then
|
|
top_frame:SetScale(final_scale)
|
|
top_frame:ClearAllPoints()
|
|
top_frame:SetParent(pfUI.addonbuttons)
|
|
end
|
|
|
|
if C.abuttons.position == "bottom" or C.abuttons.position == "top" then
|
|
offsetX = ((index - row_index * tonumber(C.abuttons.rowsize)) * (tonumber(C.abuttons.spacing))) + (((index - row_index * tonumber(C.abuttons.rowsize)) - 1) * GetButtonSize()) + (GetButtonSize() / 2)
|
|
offsetY = -(((row_index + 1) * tonumber(C.abuttons.spacing)) + (row_index * GetButtonSize()) + (GetButtonSize() / 2))
|
|
frame:SetPoint("CENTER", pfUI.addonbuttons, "TOPLEFT", offsetX/final_scale, offsetY/final_scale)
|
|
if top_frame ~= pfUI.addonbuttons then
|
|
top_frame:SetPoint("CENTER", pfUI.addonbuttons, "TOPLEFT", offsetX/final_scale, offsetY/final_scale)
|
|
end
|
|
else
|
|
offsetX = -(((row_index + 1) * tonumber(C.abuttons.spacing)) + (row_index * GetButtonSize()) + (GetButtonSize() / 2))
|
|
offsetY = -(((index - row_index * tonumber(C.abuttons.rowsize)) * (tonumber(C.abuttons.spacing))) + (((index - row_index * tonumber(C.abuttons.rowsize)) - 1) * GetButtonSize()) + (GetButtonSize() / 2))
|
|
frame:SetPoint("CENTER", pfUI.addonbuttons, "TOPRIGHT", offsetX/final_scale, offsetY/final_scale)
|
|
if top_frame ~= pfUI.addonbuttons then
|
|
top_frame:SetPoint("CENTER", pfUI.addonbuttons, "TOPRIGHT", offsetX/final_scale, offsetY/final_scale)
|
|
end
|
|
end
|
|
end
|
|
|
|
local function ManualAddOrRemove(msg)
|
|
local _, _, action, arg = string.find(msg, "%s?(%w+)%s?(.*)")
|
|
|
|
local button
|
|
if arg and arg ~= "" then
|
|
button = _G[arg]
|
|
else
|
|
button = GetMouseFocus()
|
|
end
|
|
|
|
if action == "" or (action ~= "reset" and action ~= "add" and action ~= "del") then
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffccpf|rUI Addon Button Panel:")
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffcc/abp add|r - " .. T["Add button to the frame"])
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffcc/abp del|r - " .. T["Remove button from the frame"])
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffcc/abp reset|r - " .. T["Reset all manually added or ignored buttons"])
|
|
return
|
|
end
|
|
|
|
if action == "reset" then
|
|
for i, button_name in ipairs(pfUI_cache["abuttons"]["add"]) do
|
|
if _G[button_name] ~= nil then
|
|
if TableMatch(pfUI.addonbuttons.buttons, button_name) then
|
|
table.remove(pfUI.addonbuttons.buttons, TableMatch(pfUI.addonbuttons.buttons, button_name))
|
|
end
|
|
RestoreButton(_G[button_name])
|
|
end
|
|
end
|
|
pfUI_cache["abuttons"]["add"] = {}
|
|
pfUI_cache["abuttons"]["del"] = {}
|
|
pfUI.addonbuttons:ProcessButtons()
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffccpf|rUI ABP|r: " .. T["Lists of added and deleted buttons are cleared"])
|
|
return
|
|
else
|
|
if IsButtonValid(button) then
|
|
if action == "add" then
|
|
if TableMatch(pfUI_cache["abuttons"]["del"], button:GetName()) then
|
|
table.remove(pfUI_cache["abuttons"]["del"], TableMatch(pfUI_cache["abuttons"]["del"], button:GetName()))
|
|
end
|
|
if not TableMatch(pfUI.addonbuttons.buttons, button:GetName()) and not TableMatch(pfUI_cache["abuttons"]["add"], button:GetName()) then
|
|
table.insert(pfUI_cache["abuttons"]["add"], button:GetName())
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffccpf|rUI ABP|r: " .. T["Added button"] .. ": " .. button:GetName())
|
|
else
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffccpf|rUI ABP|r: " .. T["Button already exists in pfMinimapButtons frame"])
|
|
return
|
|
end
|
|
elseif action == "del" then
|
|
if TableMatch(pfUI_cache["abuttons"]["add"], button:GetName()) then
|
|
table.remove(pfUI_cache["abuttons"]["add"], TableMatch(pfUI_cache["abuttons"]["add"], button:GetName()))
|
|
end
|
|
if TableMatch(pfUI.addonbuttons.buttons, button:GetName()) then
|
|
table.remove(pfUI.addonbuttons.buttons, TableMatch(pfUI.addonbuttons.buttons, button:GetName()))
|
|
else
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffccpf|rUI ABP|r: " .. T["Button not found in pfMinimapButtons frame"])
|
|
return
|
|
end
|
|
if not TableMatch(pfUI_cache["abuttons"]["del"], button:GetName()) then
|
|
table.insert(pfUI_cache["abuttons"]["del"], button:GetName())
|
|
RestoreButton(button)
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffccpf|rUI ABP|r: " .. T["Removed button"] .. ": " .. button:GetName())
|
|
end
|
|
end
|
|
pfUI.addonbuttons:ProcessButtons()
|
|
return
|
|
else
|
|
DEFAULT_CHAT_FRAME:AddMessage("|cff33ffccpf|rUI ABP|r: " .. T["Not a valid button!"])
|
|
return
|
|
end
|
|
end
|
|
end
|
|
|
|
function pfUI.addonbuttons:ProcessButtons()
|
|
UpdatePanel()
|
|
|
|
local i = 1
|
|
for _, button_name in ipairs(pfUI.addonbuttons.buttons) do
|
|
if _G[button_name] and _G[button_name]:IsVisible() then
|
|
BackupButton(_G[button_name])
|
|
MoveButton(i, _G[button_name])
|
|
i = i + 1
|
|
end
|
|
end
|
|
end
|
|
|
|
function pfUI.addonbuttons:UpdateConfig()
|
|
pfUI.addonbuttons:ProcessButtons()
|
|
end
|
|
|
|
pfUI.addonbuttons:SetScript("OnShow", function()
|
|
pfUI.addonbuttons:ProcessButtons()
|
|
end)
|
|
|
|
-- Initial setup on the next frame, once other addons' minimap buttons exist
|
|
RunNextFrame(function()
|
|
-- rescan before deciding whether the panel should be shown by default
|
|
pfUI.addonbuttons:ProcessButtons()
|
|
|
|
pfUI.addonbuttons:SetShown(C.abuttons.showdefault == "1" and GetNumButtons() > 0)
|
|
|
|
-- apply workarounds
|
|
for k, v in pairs(pfUI.addonbuttons.overrides) do
|
|
_G[k] = v
|
|
end
|
|
end)
|
|
|
|
-- Rescan minimap buttons every 5 seconds while the panel is shown
|
|
C_Timer.NewTicker(5, function()
|
|
if not pfUI.addonbuttons:IsShown() then return end
|
|
|
|
-- reload/rescan minimap buttons
|
|
pfUI.addonbuttons:ProcessButtons()
|
|
|
|
-- re-apply addon button workarounds
|
|
for k, v in pairs(pfUI.addonbuttons.overrides) do
|
|
_G[k] = v
|
|
end
|
|
end)
|
|
|
|
pfUI.addonbuttons:RegisterEvent("PLAYER_REGEN_DISABLED")
|
|
pfUI.addonbuttons:SetScript("OnEvent", function()
|
|
if C.abuttons.hideincombat == "1" and pfUI.addonbuttons:IsShown() then
|
|
pfUI.addonbuttons:Hide()
|
|
end
|
|
end)
|
|
|
|
pfUI.addonbuttons:UpdateConfig()
|
|
|
|
pfUI.api.RegisterSlashCommand("PFABP", { "/abp", "/pfabp" }, ManualAddOrRemove, true)
|
|
end)
|