LordKator_TrinityMagicMenu.lua

--[[
--
-- LordKator_TrinityMagicMenu - Lord Kator's Trinity Core MagicMenu
--
-- Author: Lord Kator <kator@karlbunch.com>
--
-- Created: Sun Nov 27 19:40:38 EST 2016
]]

LKTMM = {
    commandTips = {
        [".group summon"] = {
            tooltipTitle = "Group Summon",
            tooltipText = "Summon your group to your location."
        },
        [".appear"] = {
            tooltipTitle = "Appear At Target",
            tooltipText = "Teleports you to the target."
        },
        [".summon"] = {
            tooltipTitle = "Summon To Me",
            tooltipText = "Teleports the target to you.",
        },
        [".die"] = {
            tooltipTitle = "Die",
            tooltipText ="Kills the target instantly.",
        },
        [".recall"] = {
            tooltipTitle = "Return To Previous Location",
            tooltipText = "Returns the target to the last location they teleported from or zoned into.",
        },
        [".revive"] = {
            tooltipTitle = "Revive",
            tooltipText = "Revive the target from death or damage.",
        },
        [".repairitems"] = {
            tooltipTitle = "Repair Items",
            tooltipText = "Repair the target's items.",
        },
        [".combatstop"] = {
            tooltipTitle = "Stop Combat",
            tooltipText = "Clear target from combat.",
        },
        [".tele name $home"] = {
            tooltipTitle = "Instant Hearth",
            tooltipText = "Return target to their hearth.",
        },
        [".gobject add temp 187299 300"] = {
            tooltipTitle = "Spawn Guild Vault",
            tooltipText = "Spawn a temporary guild vault.",
        },
    },
}

function LKTMM.SetTargetDPS()
    LKTM:PromptForTargetDPS()
end

function LKTMM:SetDefaultCommand()
    LKTM:PromptForCommand()
end

function LKTMM:BuildTaxiMenu(menu)
    if LKTM_Data_TaxiMenu == nil then
        return
    end

    menu:addFlyout(1, "Instant Taxi","Choose a location you'd like to teleport to.", "TaxiMenu")

    for menuValue, menuItem in pairs(LKTM_Data_TaxiMenu) do
        if string.match(menuValue, "^%d+$") ~= nil then
            menu:addToMenu(2, menuItem, "TaxiMenu")
        else
            for _, item in pairs(menuItem) do
                if item.value == "TaxiMenuHistory" then
                    item.disabled = 1
                end
                item.func = function(entry) LKTM:GotoTaxiNode(entry) end
                menu:addToMenu(3, item, menuValue)
            end
        end
    end

    local taxiHistory = LKTM:GetTaxiHistory()

    if taxiHistory ~= nil then
        local sortByCount = function(a,b) return taxiHistory[b]['count'] < taxiHistory[a]['count'] end
        local historyItems = {}
        for _,v in LKTM:pairsByKeys(taxiHistory, sortByCount) do
            table.insert(historyItems, {
                text = v.text,
                tooltipTitle = v.text,
                tooltipText = ".go taxinode " .. v.id,
                arg1 = v.id,
                notCheckable = 1,
                func = function(entry) LKTM:GotoTaxiNode(entry) end,
            })
        end

        if #historyItems > 0 then
            table.insert(historyItems, 1, {
                ["text"] = "Recent Destinations",
                notCheckable = 1,
                isTitle = 1
            })

            menu:addItems(3, historyItems, "TaxiMenuHistory")

            for _,v in pairs(menu:getItems(2, "TaxiMenu")) do
                if v.value == "TaxiMenuHistory" then
                    v.disabled = nil
                    break
                end
            end
        end
    end
end

function LKTMM:encodeArgs(...)
    return table.concat({...}, string.char(1))
end

function LKTMM:decodeArgs(str)
    return strsplit(string.char(1), str)
end

function LKTMM:QuestTool(frame, arg1, arg2, checked) -- luacheck: no unused args
    local cmd, unit, title, questLogIndex = LKTMM:decodeArgs(arg1)
    local command = ".quest " .. cmd .. " " .. arg2

    if unit == "party" then
        local playerName = UnitName("player")
        local macroText = ""

        if cmd ~= "Add" or not IsUnitOnQuest(questLogIndex, "player") then
            macroText = macroText .. "/target player" .. "\n/whisper " .. playerName .. " " .. command .. "\n"
        end

        if GetNumPartyMembers() > 0 then
            for i=1,GetNumPartyMembers(),1 do
                if cmd ~= "Add" or not IsUnitOnQuest(questLogIndex, "party" .. i) then
                    macroText = macroText .. "/target party" .. i .. "\n/whisper " .. playerName .. " " .. command .. "\n"
                end
            end
        end

        macroText = macroText .. "/cleartarget\n"

        LordKator_TrinityMagicConfirm:confirmMacro(
            cmd .. " quest:\n\n" .. title .. "\n\nFor everyone in the party?",
            macroText,
            function() LKTM:Message(0, "Ran macro [" .. macroText .. "]") end
        )
        return
    end

    LKTM:CommandOnUnit(unit, command)
end

function LKTMM:BuildQuestMenu(menu)
    local menuValue = "QuestMenuSelection"

    menu:addFlyout(1, "Quest Log Tools", "Tools for working with quests", menuValue)

    menu:addTitle(2, "Quests", menuValue)

    local sectionTitle = nil
    local i=0
    while 1 do
        i = i + 1
        -- luacheck: ignore 211
        local questTitle, level, questTag, suggestedGroup, isHeader, isCollapsed, isComplete, isDaily, questID, displayQuestID = GetQuestLogTitle(i);

        if questTitle == nil then
            break
        end
        if isHeader then
            sectionTitle = questTitle
        else
            if sectionTitle then
                menu:addTitle(2, sectionTitle, menuValue)
                sectionTitle = nil
            end

            local questMenuValue = "QuestMenu" .. questID

            local c = GetQuestDifficultyColor(level)
            questTitle = format("|cff%02x%02x%02x%s|r", c.r * 255, c.g * 255, c.b * 255, questTitle)

            local partyOnQuest = {}
            for j=1, MAX_PARTY_MEMBERS, 1 do
                local unit = "party" .. j

                if UnitInParty(unit) and IsUnitOnQuest(i, unit) then
                    local name = UnitName(unit)
                    table.insert(partyOnQuest, name)
                end
            end

            if #partyOnQuest > 0 then
                questTitle = "(" .. #partyOnQuest .. ") " .. questTitle
            end

            menu:addFlyout(2, questTitle .. (questTag and (" |cffffff00" .. questTag .."|r") or "") .. (isComplete and " (|cff00ff00complete|r)" or ""), "", questMenuValue, menuValue)
                :getLastItem().arg1 = questID

            if #partyOnQuest > 0 then
                menu:getLastItem().tooltipText = "Also on quest:\n\n" .. table.concat(partyOnQuest, ", ")
            end

            menu:addTitle(3, questTitle, questMenuValue)
                :addItem(3, "Complete quest", "Complete this quest for " .. UnitName("target") .. ".", {
                    arg1 = LKTMM:encodeArgs("Complete", "target", questTitle, i),
                    arg2 = questID,
                    func = function(entry, arg1, arg2, checked) LKTMM:QuestTool(entry, arg1, arg2, checked) end,
                }, questMenuValue)
                :addItem(3, "Remove quest", "Remove this quest from" .. UnitName("target") .. ".", {
                    arg1 = LKTMM:encodeArgs("Remove", "target", questTitle, i),
                    arg2 = questID,
                    func = function(entry, arg1, arg2, checked) LKTMM:QuestTool(entry, arg1, arg2, checked) end,
                }, questMenuValue)

            if GetNumPartyMembers() > 0 then
                menu:addTitle(3, "Party", questMenuValue)
                :addItem(3, "Add to party", "Add this quest to everyone in the party.", {
                    arg1 = LKTMM:encodeArgs("Add", "party", questTitle, i),
                    arg2 = questID,
                    func = function(entry, arg1, arg2, checked) LKTMM:QuestTool(entry, arg1, arg2, checked) end,
                }, questMenuValue)
                :addItem(3, "Complete for party", "Complete this quest for everyone.", {
                    arg1 = LKTMM:encodeArgs("Complete", "party" , questTitle, i),
                    arg2 = questID,
                    func = function(entry, arg1, arg2, checked) LKTMM:QuestTool(entry, arg1, arg2, checked) end,
                }, questMenuValue)
                :addItem(3, "Remove from party", "Remove this quest from everyone.", {
                    arg1 = LKTMM:encodeArgs("Remove", "party", questTitle, i),
                    arg2 = questID,
                    func = function(entry, arg1, arg2, checked) LKTMM:QuestTool(entry, arg1, arg2, checked) end,
                }, questMenuValue)
            end
        end
    end
end

function LKTMM:BuildNPCToolsMenu(menu)
    local npcList = LKTM:GetSavedNPClist()

    local npcCount = 0
    for _,_ in pairs(npcList or {}) do
        npcCount = npcCount + 1
    end

    if npcCount <= 0 then
        return
    end

    local menuValue = "NPCToolsMenu"

    menu:addFlyout(1, "NPC Tools", "Tools for working with NPC's", menuValue)

    menu:addTitle(2, "Saved NPC's", menuValue)

    local sortNPC = function(a,b) return npcList[a]['unitName'] < npcList[b]['unitName'] end

    for _,v in LKTM:pairsByKeys(npcList, sortNPC) do
        local npcMenuValue = "NPCToolsMenuNPC" .. v.entryID

        menu:addFlyout(2, v.unitName, "", npcMenuValue, menuValue, function()
            LKTM:CommandOnUnit("player", ".npc add temp loot " .. v.entryID)
            CloseDropDownMenus()
        end)
        :addTitle(3, v.unitName, npcMenuValue)
        :addItem(3, "Spawn NPC", "Spawn this NPC at your position.", {
            arg1 = v.entryID,
            func = function(_, arg1)
                LKTM:CommandOnUnit("player", ".npc add temp loot " .. arg1)
                CloseDropDownMenus()
            end
        }, npcMenuValue)
        :addItem(3, "Forget NPC", "Forget this NPC.", {
            arg1 = v.entryID,
            func = function(_, arg1)
                local list = LKTM:GetSavedNPClist()
                list[arg1] = nil
                CloseDropDownMenus()
            end
        }, npcMenuValue)
    end
end

function LKTMM:BuildWaypointMenu(menu)
    local menuValue = "WaypointToolsMenu"

    menu:addFlyout(1, "Waypoint Tools", "Tools for working with Waypoint's", menuValue)

    menu:addTitle(2, "Saved Waypoint's", menuValue)
        :addItem(2, "Save waypoint", "Save current location as a waypoint", {
            func = function()
                CloseDropDownMenus()
                StaticPopup_Show("LKTMM_PromptWaypointName", "", "", LKTM:UserWaypointNew())
            end
        }, menuValue)

    local wpList = LKTM:UserWaypointGetList()

    if wpList == nil then
        return
    end

    local sortWaypoint = function(a,b)
        local wpa, wpb = wpList[a], wpList[b]

        if wpa.map == nil or wpb.map == nil then
            return wpa.name < wpb.name
        end

        if wpa.map.name == wpb.map.name then
            if wpa.map.zoneName == wpb.map.zoneName then
                if wpa.map.areaName == wpb.map.areaName then
                    return wpa.name < wpb.name
                else
                    return wpa.map.areaName < wpa.map.areaName
                end
            else
                return wpa.map.zoneName < wpb.map.zoneName
            end
        end

        return wpa.map.name < wpb.map.name
    end

    for _,v in LKTM:pairsByKeys(wpList, sortWaypoint) do
        local wpMenuValue = menuValue .. v.key

        menu:addFlyout(2, v.name, "", wpMenuValue, menuValue, function() LKTM:UserWaypointGoto(v) end)
        :addTitle(3, v.name, wpMenuValue)
        :addItem(3, "Teleport to waypoint", "Teleport to this waypoint.", {
            arg1 = v,
            func = function(_, arg1)
                LKTM:UserWaypointGoto(arg1)
                CloseDropDownMenus()
            end
        }, wpMenuValue)
        :addItem(3, "Forget Waypoint", "Forget this Waypoint.", {
            arg1 = v,
            func = function(_, arg1)
                LKTM:UserWaypointDelete(arg1)
                CloseDropDownMenus()
            end
        }, wpMenuValue)
    end
end

function LKTMM:InitializeDropDown(frame, level) -- luacheck: no unused args
    local menu = LKTMM_MenuBuilder:New()

    menu.addCommand = function (thisMenu, menuLevel, command, tipsHint)
        local tips = LKTMM.commandTips[tipsHint] or LKTMM.commandTips[command] or tipsHint

        if tips then
            return thisMenu:addItem(menuLevel, tips.tooltipTitle, tips.tooltipText, {
                func = function() LKTM:CommandOnUnit("target", command) end,
            })
        else
            return thisMenu:addItem(menuLevel, command, "", {
                func = function() LKTM:CommandOnUnit("target", command) end,
            })
        end
    end

    -- Build Menu
    menu:addTitle(1, "Trinity Core Magic Menu")
    menu:addFlyout(1, "Set Control-Click Command","Set the command to run when you control-click a target", "SetControlClickCommand")

    -- Context Sensitive Items
    if UnitIsUnit("player", "target") then
        LKTMM:BuildWaypointMenu(menu)
        LKTMM:BuildTaxiMenu(menu)
        LKTMM:BuildNPCToolsMenu(menu)
        LKTMM:BuildQuestMenu(menu)
        if GetNumPartyMembers() > 0 then
            menu:addCommand(1, ".group summon")
        end
        menu:addCommand(1, ".gobject add temp 187299 300")
        menu:addCommand(1, ".tele name $home")
    elseif UnitIsPlayer("target") then
        menu:addCommand(1, ".appear"):addCommand(1, ".summon")
            :addCommand(1, ".tele name " .. UnitName("target") .. " $home", ".tele name $home")
    else
        local unitName = UnitName("target")

        if GameTooltip:NumLines() >= 3 then
            unitName = GameTooltipTextLeft1:GetText() .. " <" .. GameTooltipTextLeft2:GetText() .. ">"
        end

        menu:addItem(1, "Save: " .. unitName, "Save NPC so you can summon it later.", {
                arg1 = unitName,
                func = function(entry) LKTM:CopyNPC(entry, unitName) end
            })
        :addItem(1, "Waypoint: " .. unitName, "Save waypoint at this NPC so you can teleport to it later.", {
                arg1 = unitName,
                func = function() LKTM:UserWaypointNew(unitName, unitName) end
            })
        :addCommand(1, ".die")
    end

    if UnitIsPlayer("target") then
        local dpsPrompt = "Set Target DPS"

        if LKTM:GetTargetDPS() > 0 then
            dpsPrompt = "Change Target DPS (" .. LKTM:GetTargetDPS() .. ")"
        end

        menu:addCommand(1, ".recall"):addCommand(1, ".revive"):addCommand(1, ".repairitems")
        :addItem(1, dpsPrompt, "Prompt for target DPS.", {
            func = LKTMM.SetTargetDPS
        })
        :addCommand(1, ".combatstop")
    end

    menu:addTitle(2, "Set Control-Click Command", "SetControlClickCommand")

    menu:addItem(2, "Prompt For Command","Prompt for command to use as control-click default.", {
        func = LKTMM.SetDefaultCommand,
    }, "SetControlClickCommand")

    for cmd, tips in pairs(LKTMM.commandTips) do
        menu:addItem(2, tips.tooltipTitle, tips.tooltipText, {
            func = function()
                LKTM:SetDefaultCommand(cmd)
                CloseDropDownMenus()
            end
        }, "SetControlClickCommand")
    end

    local history, history_count = LKTM:GetCommandHistory()

    if history_count > 0 then
        menu:addItem(2, "<Clear History>", "", {
            func = function()
                LKTM:ClearCommandHistory()
                CloseDropDownMenus()
            end
        }, "SetControlClickCommand")

        for cmd, _ in LKTM:pairsByKeys(history, function(a,b) return a < b end) do
            if LKTMM.commandTips[cmd] == nil then
                menu:addItem(2, cmd, "", {
                    func = function()
                        LKTM:SetDefaultCommand(cmd)
                        CloseDropDownMenus()
                    end
                }, "SetControlClickCommand")
            end
        end
    end

    menu:addItem(1, CLOSE, "Close Menu", { func = function() CloseDropDownMenus() end })

    if LKTM.debugLevel >= 9 then
        LKTM:SetGlobalPreference("lastMenu", menu:getItems())
    end

    menu:build(level or 1)
end

function LKTMM:Init()
end

function LKTMM:Show(parent, unit) -- luacheck: no unused args
    local anchorName, xOfs, yOfs = "TargetFrame", 120, 10

    if UnitIsUnit("player", "target") then
        anchorName, xOfs, yOfs = "PlayerFrame", 106, 27
    end

    if string.sub(unit, 1, 5) == "party" then
        anchorName, xOfs, yOfs = "PartyMemberFrame" .. string.sub(unit, 6, -1), 47, 15
    end

    local LordKator_TrinityMagicMenuDropDown = CreateFrame("Frame", "LordKator_TrinityMagicMenuDropDown", _G[anchorName], "UIDropDownMenuTemplate")

    UIDropDownMenu_Initialize(LordKator_TrinityMagicMenuDropDown, function(frame, level, menuList) LKTMM:InitializeDropDown(frame, level, menuList) end, "MENU")
    ToggleDropDownMenu(1, nil, LordKator_TrinityMagicMenuDropDown, anchorName, xOfs, yOfs)
end

StaticPopupDialogs["LKTMM_PromptWaypointName"] = {
    text = "Waypoint Note:",
    button1 = "Save Note",
    button2 = "None",
    hasEditBox = 1,
    whileDead = 1,
    hideOnEscape = 1,
    timeout = 20,
    --[[
    OnShow = function(self, data)
        getglobal(this:GetName().."EditBox"):SetText(data.name)
    end,
    ]]
    OnAccept = function(self, data)
        local newName = self.editBox:GetText();
        LKTM:UserWaypointSetNote(data, newName)
    end,
    EditBoxOnEnterPressed = function(self, data)
        local newName = self:GetParent().editBox:GetText();
        LKTM:UserWaypointSetNote(data, newName)
        self:GetParent():Hide();
    end,
}