local ADDON_LOADED, shared = ...

local function PickupEquipmentSet(setName)
    local setIndex = 0

    for i = 1, C_EquipmentSet.GetNumEquipmentSets() do
        local sn = C_EquipmentSet.GetEquipmentSetInfo(i)

        if sn == setName then
            setIndex = i
        end
    end

    C_EquipmentSet.PickupEquipmentSet(setIndex)
end

local pickupActionButton = {
    item = PickupItem,
    spell = PickupSpell,
    macro = PickupMacro,
    companion = PickupSpell,
    equipmentset = PickupEquipmentSet
}

local function RestoreActionButton(self, index, actionButton)
    if GetActionInfo(index) then
        PickupAction(index)
        ClearCursor()
    end

    if not actionButton then
        return true, nil
    end

    local aliases = ActionBarSaverReloaded.spellAliases[actionButton.id] or {}
    table.insert(aliases, actionButton.id)

    for _, id in ipairs(aliases) do
        pickupActionButton[actionButton.type](id)

        if GetCursorInfo() == actionButton.type then
            PlaceAction(index)
            return true, id
        end

        ClearCursor()
    end

    return false
end

local function IsMacro(actionButton)
    return actionButton and actionButton.type == "macro"
end

local function GetMacroDuplicates()
    local t = {}
    local duplicates = {}

    for i = 1, MAX_MACROS do
        local macroName = GetMacroInfo(i)

        if macroName then
            if not t[macroName] then
                t[macroName] = 1
            else
                t[macroName] = t[macroName] + 1
                duplicates[macroName] = t[macroName]
            end
        end
    end

    return duplicates
end

local function AddWarning(warnings, macroName, usages)
    table.insert(warnings, string.format("Warning: Found %d macros named '%s'. Consider renaming them to avoid issues",
        usages, macroName))
end

function SaveSet(setName)
    if not setName or setName == "" then
        print("Set name cannot be empty")
        return
    end

    local duplicates = GetMacroDuplicates()
    local set = {}
    local warnings = {}

    for i = 1, MAX_ACTION_BUTTONS do
        local type, id = GetActionInfo(i)

        if type == "macro" then
            -- use macro name as the ID
            id = GetMacroInfo(id)
            if duplicates[id] then
                AddWarning(warnings, id, duplicates[id])
            end
        end

        if type and id then
            set[i] = type and {
                type = type,
                id = id
            }
        end
    end

    ActionBarSaverReloaded.sets[setName] = set
    print(string.format("Saved set '%s'!", setName))
    for _, warning in ipairs(warnings) do
        print(warning)
    end
end

function RestoreSet(setName)
    if not setName or setName == "" then
        print("Set name cannot be empty")
        return
    end

    local set = ActionBarSaverReloaded.sets[setName]

    if not set then
        print(string.format("No set with the name '%s' exists", setName))
        return
    end
    if InCombatLockdown() then
        print("Cannot restore sets while in combat")
        return
    end

    local duplicates = GetMacroDuplicates()
    local messages = {}

    -- Start with an empty cursor
    ClearCursor()

    for i = 1, MAX_ACTION_BUTTONS do
        local actionButton = set[i]

        if IsMacro(actionButton) and duplicates[actionButton.id] then
            AddWarning(messages, actionButton.id, duplicates[actionButton.id])
        end

        local succeeded, restoredID = RestoreActionButton(self, i, actionButton)
        if not succeeded then
            table.insert(messages, string.format("Error: Unable to restore %s with id [%s] to slot %d",
                actionButton.type, actionButton.id or "", i))
        elseif actionButton and restoredID ~= actionButton.id then
            table.insert(messages,
                string.format("Info: Restored spell %d (%s) in place of spell %d", restoredID, GetSpellInfo(restoredID),
                    actionButton.id))
        end
    end

    print(string.format("Restored set '%s'", setName))
    for _, warning in ipairs(messages) do
        print(warning)
    end
end

function DeleteSet(setName)
    if not setName or setName == "" then
        print("Set name cannot be empty")
        return
    end

    if not ActionBarSaverReloaded.sets[setName] then
        print(string.format("No set with the name '%s' exists", setName))
        return
    end

    ActionBarSaverReloaded.sets[setName] = nil

    print(string.format("Deleted set '%s'", setName))
end

function ListSets()
    local sets = {}
    for setName, foo in pairs(ActionBarSaverReloaded.sets) do
        table.insert(sets, setName)
    end
    table.sort(sets)
    local setsStr = table.concat(sets, ", ")

    print(not (not setsStr or setsStr == "") and setsStr or "No sets found")
end

function AliasSpell(args)
    if not args or args == "" then
        print("Must provide args in the format 'spellID aliasID'")
        return
    end
    local spellID, aliasID = string.match(args, "(%d+)%s+(%d+)")

    spellID = tonumber(spellID)
    aliasID = tonumber(aliasID)

    if not (spellID and aliasID) then
        print(string.format("Could not parse spellID and aliasID from '%s'", args))
        return
    end

    local aliases = ActionBarSaverReloaded.spellAliases[spellID] or {}

    for _, id in ipairs(aliases) do
        if id == aliasID then
            print(string.format("Spell %d is already aliased by %d", spellID, aliasID))
            return
        end
    end

    table.insert(aliases, aliasID)

    table.insert(aliases, aliasID)
    ActionBarSaverReloaded.spellAliases[spellID] = aliases

    print(string.format("Added %d as an alias for %d", aliasID, spellID))
end

function DeleteSpellAliases(spellID)
    if not spellID or spellID == "" then
        print("Must provide a valid spellID")
        return
    end

    spellID = tonumber(spellID)

    if not ActionBarSaverReloaded.spellAliases[spellID] then
        print(string.format("No aliases to remove for spell with ID %d", spellID))
        return
    end

    ActionBarSaverReloaded.spellAliases[spellID] = nil

    print(string.format("Removed all aliases for spell with ID %d", spellID))
end

function ListAliases()
    local aliases = ActionBarSaverReloaded.spellAliases

    if Dict.isEmpty(aliases) then
        print("No aliases found")
        return
    end

    Dict.iter(ActionBarSaverReloaded.spellAliases, function(spellID, aliases)
        print(string.format("Spell %d is aliased by: %s", spellID, table.concat(aliases, ", ")))
    end)
end

function PrintUsage()
    print("ABS Slash commands")
    print("/abs save <set> - Saves your current action bar setup under the given <set>")
    print("/abs restore <set> - Restores the saved <set>")
    print("/abs delete <set> - Deletes the saved <set>")
    print("/abs list - Lists all saved sets")
    print("/abs alias <spellID> <aliasID> - Adds an alias with <aliasID> to <spellID>")
    print("/abs unalias <spellID> - Removes all aliases associated with <spellID>")
    print("/abs aliases - List all spell aliases")
end

SlashCmdList["ABS"] = function(argv)
    local args = {strsplit(" ", argv)}
    local cmd = args[1]

    if cmd == "save" then
        SaveSet(args[2])
    end
    if cmd == "restore" then
        RestoreSet(args[2])
    end
    if cmd == "delete" then
        DeleteSet(args[2])
    end
    if cmd == "list" then
        ListSets()
    end
    if cmd == "alias" then
        AliasSpell(args[2], args[3])
    end
    if cmd == "unalias" then
        DeleteSpellAliases(args[2])
    end
    if cmd == "aliases" then
        ListAliases()
    end
    if cmd == "" or not cmd then
        PrintUsage()
    end
end
SLASH_ABS1 = "/abs"