core.lua

local addonName, ns = ...

-- Global table for saving positions
WoWUI_DB = WoWUI_DB or {}

local WoWUI = CreateFrame("Frame")
ns.WoWUI = WoWUI

ns.WoWUI.actionBars = {}
ns.WoWUI.unitFrames = {}

local movableFrames = {}

WoWUI:RegisterEvent("ADDON_LOADED")
WoWUI:SetScript("OnEvent", function(self, event, loadedAddon)
    if loadedAddon == addonName then
        WoWUI_DB.positions = WoWUI_DB.positions or {}
        self:UnregisterEvent("ADDON_LOADED")

        -- Restore saved positions now that the DB is loaded
        for _, frame in ipairs(movableFrames) do
            local name = frame.dbKey
            if WoWUI_DB.positions[name] then
                local pos = WoWUI_DB.positions[name]
                frame:ClearAllPoints()
                frame:SetPoint(pos[1], UIParent, pos[2], pos[3], pos[4])
            end
        end
    end
end)

-- Movability system
local unlocked = false

function WoWUI:RegisterMovable(frame, name)
    frame.dbKey = name
    table.insert(movableFrames, frame)

    local mover = CreateFrame("Frame", nil, frame, "BackdropTemplate")
    mover:SetAllPoints(frame)
    mover:SetFrameStrata("TOOLTIP")
    mover:EnableMouse(true)
    mover:RegisterForDrag("LeftButton")

    mover:SetScript("OnDragStart", function(self)
        frame:StartMoving()
    end)

    mover:SetScript("OnDragStop", function(self)
        frame:StopMovingOrSizing()
        local point, _, relativePoint, x, y = frame:GetPoint()
        WoWUI_DB.positions[frame.dbKey] = { point, relativePoint, x, y }
    end)

    -- Simple background to show it's unlocked
    local tex = mover:CreateTexture(nil, "BACKGROUND")
    tex:SetAllPoints()
    tex:SetColorTexture(0, 1, 0, 0.4)

    local text = mover:CreateFontString(nil, "OVERLAY", "GameFontNormal")
    text:SetPoint("CENTER")
    text:SetText(name)

    mover:Hide()
    frame.mover = mover

    -- If the frame is registered after ADDON_LOADED has fired, try to load it now
    if WoWUI_DB and WoWUI_DB.positions and WoWUI_DB.positions[name] then
        local pos = WoWUI_DB.positions[name]
        frame:ClearAllPoints()
        frame:SetPoint(pos[1], UIParent, pos[2], pos[3], pos[4])
    end
end

SLASH_WOWUI1 = "/wowui"
SlashCmdList["WOWUI"] = function(msg)
    if msg == "unlock" then
        unlocked = true
        for _, frame in ipairs(movableFrames) do
            frame:SetMovable(true)
            frame:SetUserPlaced(true)
            frame.mover:Show()
        end
        print("WoWUI: Frames unlocked.")
    elseif msg == "lock" then
        unlocked = false
        for _, frame in ipairs(movableFrames) do
            frame:SetMovable(false)
            frame.mover:Hide()
        end
        print("WoWUI: Frames locked.")
    else
        print("WoWUI Commands:")
        print("  /wowui unlock - Unlock frames to move them")
        print("  /wowui lock   - Lock frames")
    end
end


-- Keybinding System
local keybindMode = false

local function UpdateKeybindOverlay(btn)
    if not btn.keybindOverlay then
        local overlay = CreateFrame("Button", nil, btn, "BackdropTemplate")
        overlay:SetAllPoints(btn)
        overlay:SetFrameStrata("DIALOG")
        overlay:EnableMouse(true)
        overlay:EnableKeyboard(true)
        overlay:RegisterForClicks("AnyDown")

        local tex = overlay:CreateTexture(nil, "OVERLAY")
        tex:SetAllPoints()
        tex:SetColorTexture(0, 1, 0, 0.4)
        tex:Hide()

        overlay:SetScript("OnEnter", function(self)
            tex:Show()
            GameTooltip:SetOwner(self, "ANCHOR_TOP")
            GameTooltip:SetText("Hover to bind\nPress ESC to close\nPress Backspace/Delete to clear")
            GameTooltip:Show()
        end)

        overlay:SetScript("OnLeave", function(self)
            tex:Hide()
            GameTooltip:Hide()
        end)

        local function assignBind(key)
            if key == "ESCAPE" then
                ns.WoWUI:ToggleKeybindMode()
                return
            end

            if key == "BACKSPACE" or key == "DELETE" then
                btn:ClearBindings()
                print("WoWUI: Cleared bindings for " .. (btn:GetName() or "button"))
                return
            end

            if key == "LSHIFT" or key == "RSHIFT" or key == "LCTRL" or key == "RCTRL" or key == "LALT" or key == "RALT" or key == "LMETA" or key == "RMETA" then
                return -- Wait for actual key
            end

            local modifier = ""
            if IsShiftKeyDown() then modifier = modifier .. "SHIFT-" end
            if IsControlKeyDown() then modifier = modifier .. "CTRL-" end
            if IsAltKeyDown() then modifier = modifier .. "ALT-" end

            local bindKey = modifier .. key
            btn:SetKey(bindKey)
            SaveBindings(GetCurrentBindingSet())
            print("WoWUI: Bound " .. bindKey .. " to " .. (btn:GetName() or "button"))
        end

        overlay:SetScript("OnKeyDown", function(self, key)
            assignBind(key)
        end)

        overlay:SetScript("OnMouseDown", function(self, button)
            if button == "LeftButton" or button == "RightButton" then
                return
            end
            local bindKey = button
            if button == "MiddleButton" then bindKey = "BUTTON3" end

            local btnNum = button:match("Button(%d+)")
            if btnNum then
                bindKey = "BUTTON" .. btnNum
            end

            assignBind(bindKey)
        end)

        overlay:SetScript("OnMouseWheel", function(self, delta)
            local bindKey = delta > 0 and "MOUSEWHEELUP" or "MOUSEWHEELDOWN"
            assignBind(bindKey)
        end)

        btn.keybindOverlay = overlay
    end

    if keybindMode then
        btn.keybindOverlay:Show()
    else
        btn.keybindOverlay:Hide()
    end
end

function ns.WoWUI:ToggleKeybindMode()
    keybindMode = not keybindMode

    if keybindMode then
        print("WoWUI: Keybind mode ENABLED. Hover over buttons to bind. Press ESC to exit.")
    else
        print("WoWUI: Keybind mode DISABLED.")
    end

    for _, bar in ipairs(ns.WoWUI.actionBars) do
        if bar.buttons then
            for _, btn in ipairs(bar.buttons) do
                UpdateKeybindOverlay(btn)
            end
        end
    end
end

SLASH_WOWUIKBIND1 = "/kbind"
SlashCmdList["WOWUIKBIND"] = function(msg)
    ns.WoWUI:ToggleKeybindMode()
end