Files
wow-ActionBarSaverDaved/KeybindSaver.lua

279 lines
7.7 KiB
Lua

local ADDON_NAME = ...
-- Initialize saved variables
local frame = CreateFrame("Frame")
frame:RegisterEvent("ADDON_LOADED")
frame:SetScript("OnEvent", function(self, event, addon)
if addon ~= ADDON_NAME then return end
KeybindSaverDaved = KeybindSaverDaved or {}
KeybindSaverDaved.sets = KeybindSaverDaved.sets or {}
end)
---@param setName string
---@return nil
local function SaveKeybindSet(setName)
if not setName or setName == "" then
print("Set name cannot be empty")
return
end
---@type table<string, {key1: string, key2: string?}>
local set = {}
local numBindings = GetNumBindings()
for i = 1, numBindings do
local command, _, key1, key2 = GetBinding(i)
if string.find(command, "CLICK ") then
-- What a stupid fucking api, so fucking idiotic, retarded
command, key2 = strsplit(":", command)
end
if key1 or key2 then set[command] = {
key1 = key1,
key2 = key2,
} end
end
KeybindSaverDaved.sets[setName] = set
print(string.format("Saved keybind set '%s'!", setName))
end
---@param command string
---@return nil
local function UnbindCommand(command)
if not command or command == "" then
print("Cannot unbind an empty command")
return
end
local keys = { GetBindingKey(command) }
for _, key in ipairs(keys) do
-- print("Unbinding", key)
SetBinding(key)
end
end
---@param setName string
---@return nil
local function RestoreKeybindSet(setName)
if not setName or setName == "" then
print("Set name cannot be empty")
return
end
local set = KeybindSaverDaved.sets[setName]
if not set then
print(string.format("No set with the name '%s' exists", setName))
return
end
-- Clear all current bindings first
ClearOverrideBindings(frame)
-- Restore saved bindings
for command, keys in pairs(set) do
UnbindCommand(command)
-- This fucking catastrophe binds a KEY(2) to CLICK A BUTTON (command) as a MOUSE BUTTON (key1)
if string.find(command, "CLICK ") and keys.key2 then
command = string.gsub(command, "CLICK ", "")
local ok = SetBindingClick(keys.key2, command, keys.key1)
if not ok then print("Failed to bind click", keys.key1, command) end
else
if keys.key1 then
local ok = SetBinding(keys.key1, command)
if not ok then print("Failed to bind", keys.key1, command) end
end
if keys.key2 then
local ok = SetBinding(keys.key2, command)
if not ok then print("Failed to bind", keys.key2, command) end
end
end
end
-- Save the changes
SaveBindings(GetCurrentBindingSet())
SaveBindings(1)
print(string.format("Restored keybind set '%s'", setName))
end
---@param setName string
---@return nil
local function DeleteKeybindSet(setName)
if not setName or setName == "" then
print("Set name cannot be empty")
return
end
if not KeybindSaverDaved.sets[setName] then
print(string.format("No set with the name '%s' exists", setName))
return
end
KeybindSaverDaved.sets[setName] = nil
print(string.format("Deleted keybind set '%s'", setName))
end
---@return nil
local function ListKeybindSets()
local sets = {}
for setName in pairs(KeybindSaverDaved.sets) do
table.insert(sets, setName)
end
table.sort(sets)
if #sets == 0 then
print("No keybind sets found")
else
print("Saved keybind sets:")
print(table.concat(sets, ", "))
end
end
---@param setName string
---@param importStr string
---@return nil
local function ImportKeybindSet(setName, importStr)
if not setName or setName == "" then
print("Set name cannot be empty")
return
end
local set = {}
local lines = { strsplit("\n", importStr) }
for _, line in ipairs(lines) do
local command, key, key2 = strsplit(":", line)
if command and key then
if string.find(command, "CLICK ") then
-- I can't believe we have to do this bullshit...
local tmp = key
key = key2
key2 = tmp
end
if not set[command] then set[command] = {
key1 = key,
key2 = key2,
} end
end
end
KeybindSaverDaved.sets[setName] = set
print(string.format("Imported keybind set '%s'", setName))
end
-- Create the import/export frame
local importExportFrame = CreateFrame("Frame", "KBSImportExportFrame", UIParent)
importExportFrame:SetSize(512, 512)
importExportFrame:SetPoint("CENTER")
importExportFrame:SetFrameStrata("HIGH")
importExportFrame:EnableMouse(true)
importExportFrame:SetMovable(true)
importExportFrame:SetResizable(false)
importExportFrame:SetBackdrop({
bgFile = "Interface/Tooltips/UI-Tooltip-Background",
edgeFile = "Interface/Tooltips/UI-Tooltip-Border",
tile = true,
tileSize = 4,
edgeSize = 4,
insets = {
left = 4,
right = 4,
top = 4,
bottom = 4,
},
})
importExportFrame:SetBackdropColor(0, 0, 0, 0.8)
importExportFrame:SetBackdropBorderColor(0.5, 0.5, 0.5, 1)
importExportFrame:SetMovable(true)
importExportFrame:EnableMouse(true)
importExportFrame:RegisterForDrag("LeftButton")
importExportFrame:SetScript("OnDragStart", function(self) self:StartMoving() end)
importExportFrame:SetScript("OnDragStop", function(self) self:StopMovingOrSizing() end)
importExportFrame:SetScript("OnShow", function(self) self:SetScale(1) end)
importExportFrame:Hide()
local importingSet = nil
local importExportFrameTextBox = CreateFrame("EditBox", "KBSImportExportFrameTextBox", importExportFrame)
importExportFrameTextBox:SetSize(512, 512)
importExportFrameTextBox:SetPoint("TOPLEFT", importExportFrame, "TOPLEFT", 0, 0)
importExportFrameTextBox:SetFont("Fonts\\FRIZQT__.ttf", 12)
importExportFrameTextBox:SetTextColor(1, 1, 1, 1)
importExportFrameTextBox:SetTextInsets(20, 20, 20, 20)
importExportFrameTextBox:SetMultiLine(true)
importExportFrameTextBox:SetAutoFocus(true)
importExportFrameTextBox:SetMaxLetters(1000000)
importExportFrameTextBox:SetScript("OnEscapePressed", function(self)
importExportFrame:Hide()
if importingSet then
local text = self:GetText()
if text and text ~= "" then ImportKeybindSet(importingSet, text) end
importingSet = nil
end
end)
---@param setName string
---@return nil
local function ExportKeybindSet(setName)
local set = KeybindSaverDaved.sets[setName]
if not set then
print(string.format("No set with the name '%s' exists", setName))
return
end
local export = {}
for command, keys in pairs(set) do
if command and keys and (keys.key1 or keys.key2) then
local formatted = command
if keys.key1 then formatted = formatted .. ":" .. keys.key1 end
if keys.key2 then formatted = formatted .. ":" .. keys.key2 end
export[#export + 1] = formatted
end
end
table.sort(export)
local exportStr = table.concat(export, "\n")
importExportFrame:Show()
importExportFrameTextBox:SetText(exportStr)
importExportFrameTextBox:SetFocus()
end
---@param setName string
---@return nil
local function ImportKeybindSetDialogue(setName)
if not setName or setName == "" then
print("Set name cannot be empty")
return
end
importingSet = setName
importExportFrameTextBox:SetText("")
importExportFrame:Show()
importExportFrameTextBox:SetFocus()
end
---@return nil
local function PrintKeybindUsage()
print("Keybind Saver Slash Commands:")
print("/kbs save <set> - Saves your current keybinds under the given <set>")
print("/kbs restore <set> - Restores the saved <set>")
print("/kbs delete <set> - Deletes the saved <set>")
print("/kbs list - Lists all saved sets")
print("/kbs export <set> - Opens a window to export the given <set>")
print("/kbs import <set> - Opens a window to import a keybind set")
end
SLASH_KBS1 = "/kbs"
SlashCmdList["KBS"] = function(argv)
local args = { strsplit(" ", argv) }
local cmd = args[1]
if cmd == "save" then SaveKeybindSet(args[2]) end
if cmd == "restore" then RestoreKeybindSet(args[2]) end
if cmd == "delete" then DeleteKeybindSet(args[2]) end
if cmd == "list" then ListKeybindSets() end
if cmd == "export" then ExportKeybindSet(args[2]) end
if cmd == "import" then ImportKeybindSetDialogue(args[2]) end
if cmd == "" or not cmd then PrintKeybindUsage() end
end