LordKator_TrinityMagic/LordKator_TrinityMagicMenuBuilder.lua at master · karlbunch/LordKator_TrinityMagic · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
LKTMM_MenuBuilder = {}

function LKTMM_MenuBuilder:New(debug_level)
    return setmetatable({ menu = {}, debug_level = debug_level or 0 }, { __index = LKTMM_MenuBuilder })
end

function LKTMM_MenuBuilder:addToMenu(menuLevel, item, menuValue)
    self.lastItem = item

    if self.menu[menuLevel] == nil then
        self.menu[menuLevel] = {}
    end

    if menuValue then
        if self.menu[menuLevel][menuValue] == nil then
            self.menu[menuLevel][menuValue] = {}
        end
        table.insert(self.menu[menuLevel][menuValue], item)
    else
        table.insert(self.menu[menuLevel], item)
    end

    return self
end

function LKTMM_MenuBuilder:InitInfo(menuText, tooltipText, item)
    local i = item or {}
    i.text = menuText
    i.tooltipTitle = i.text
    i.tooltipText = tooltipText
    i.notCheckable = 1
    return i
end

function LKTMM_MenuBuilder:addItem(menuLevel, menuText, tooltipText, item, menuValue)
    local i = self:InitInfo(menuText, tooltipText, item)
    return self:addToMenu(menuLevel, i, menuValue)
end

function LKTMM_MenuBuilder:addItems(menuLevel, items, menuValue)
    if menuValue then
        if self.menu[menuLevel] == nil then
            self.menu[menuLevel] = {}
        end
        self.menu[menuLevel][menuValue] = items
    else
        self.menu[menuLevel] = items
    end

    return self
end

function LKTMM_MenuBuilder:addFlyout(menuLevel, menuText, tooltipText, menuValue, targetMenuValue, onClick)
    local i = self:InitInfo(menuText, tooltipText)
    i.hasArrow = 1
    i.value = menuValue
    i.func = onClick
    return self:addToMenu(menuLevel, i, targetMenuValue)
end

function LKTMM_MenuBuilder:addTitle(menuLevel, menuText, menuValue)
    local i = self:InitInfo(menuText)
    i.isTitle = 1
    return self:addToMenu(menuLevel, i, menuValue)
end

function LKTMM_MenuBuilder:getLastItem()
    return self.lastItem
end

function LKTMM_MenuBuilder:getItems(menuLevel, menuValue)
    if menuValue then
        if self.menu[menuLevel] then
            return self.menu[menuLevel][menuValue]
        else
            return nil
        end
    else
        if menuLevel then
            return self.menu[menuLevel]
        end
    end
    return self.menu
end

function LKTMM_MenuBuilder:build(menuLevel, menuValue)
    local info = self.menu[menuLevel]

    menuValue = menuValue or UIDROPDOWNMENU_MENU_VALUE

    if menuValue then
        info = info[menuValue]
        if not info then
            info = { { text = "Broken Menu for " .. menuValue, isTitle = 1, } }
        end
    end

    for _,v in pairs(info) do
        UIDropDownMenu_AddButton(v, menuLevel)
    end

    return info
end