MountJournalEnhanced/Events.lua at master · exochron/MountJournalEnhanced · 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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
local _, ADDON = ...

local activeMount

function ADDON:IsPlayerMounted()
    return activeMount ~= nil
end

local function FetchActiveMountFromJournal()
    local numMounts = C_MountJournal.GetNumDisplayedMounts()
    for index = 1, numMounts do
        local _, _, _, isActive, _, _, _, _, _, _, isCollected, mountID = C_MountJournal.GetDisplayedMountInfo(index)
        if isActive then
            return mountID
        end
        if not isCollected then
            return
        end
    end
end

local function checkAuras(auras)
    for _, aura in ipairs(auras) do
        if aura then
            local mountId = C_MountJournal.GetMountFromSpell(aura.spellId)
            if mountId then
                return mountId
            end
        end
    end
end

function ADDON:ScanAuras(unit)
    if InCombatLockdown() or C_Secrets.ShouldAurasBeSecret() then
        return
    end

    local continuationToken, aura1, aura2, aura3, aura4, aura5
    repeat
        continuationToken, aura1, aura2, aura3, aura4, aura5 = C_UnitAuras.GetAuraSlots(unit, "HELPFUL|CANCELABLE", 5, continuationToken)
        local mountId = checkAuras({
            aura1 and C_UnitAuras.GetAuraDataBySlot(unit, aura1),
            aura2 and C_UnitAuras.GetAuraDataBySlot(unit, aura2),
            aura3 and C_UnitAuras.GetAuraDataBySlot(unit, aura3),
            aura4 and C_UnitAuras.GetAuraDataBySlot(unit, aura4),
            aura5 and C_UnitAuras.GetAuraDataBySlot(unit, aura5),
        })
        if mountId then
            return mountId
        end
    until continuationToken == nil
end

ADDON.Events:RegisterCallback("OnLogin", function()

    activeMount = FetchActiveMountFromJournal()
    if activeMount then
        ADDON.Events:TriggerEvent("OnMountUp", activeMount, true)
    end

    -- can be triggered by anyone surrounding you for mounting up and down
    ADDON.Events:RegisterFrameEventAndCallback("COMPANION_UPDATE", function(_, type)
        if type == "MOUNT" and activeMount then
            local _,_,_, isActive = C_MountJournal.GetMountInfoByID(activeMount)
            if not isActive then
                local mountId = activeMount
                activeMount = nil
                ADDON.Events:TriggerEvent("OnMountDown", mountId)
            end
        end
    end)

    ADDON.Events:RegisterFrameEventAndCallback("UNIT_SPELLCAST_START", function(_, unit, _, spellId)
        if InCombatLockdown() -- no use case for now in combat
            or (unit ~= "player" and unit ~= "target")
        then
            return
        end

        local mountId = C_MountJournal.GetMountFromSpell(spellId)
        if mountId then
            if unit == "player" then
                ADDON.Events:TriggerEvent("CastMount", mountId)
            elseif unit == "target" then
                ADDON.Events:TriggerEvent("CastMountTarget", mountId)
            end
        end
    end, 'external events')

    ADDON.Events:RegisterFrameEventAndCallback("UNIT_SPELLCAST_SUCCEEDED", function(_, target, _, spellId)
        if target == "player" then
            local mountId = C_MountJournal.GetMountFromSpell(spellId)
            if mountId then
                local _,_,_, isActive = C_MountJournal.GetMountInfoByID(mountId)
                if isActive then
                    activeMount = mountId
                    ADDON.Events:TriggerEvent("OnMountUp", activeMount, false)
                end
            end
        elseif target == "target" then
            local mountId = C_MountJournal.GetMountFromSpell(spellId)
            if mountId then
                ADDON.Events:TriggerEvent("OnMountUpTarget", mountId)
            end
        end
    end, 'external events')

    ADDON.Events:RegisterFrameEventAndCallback("PLAYER_TARGET_CHANGED", function()
        C_Timer.After(0, function()
            local mountId = ADDON:ScanAuras("target")
            if mountId then
                ADDON.Events:TriggerEvent("OnMountUpTarget", mountId)
            end
        end)
    end, 'external events')

    ADDON.Events:RegisterFrameEventAndCallback("NEW_MOUNT_ADDED", function(_, ...)
        local param1, param2, param3 = ...
        C_Timer.After(1, function() -- mount infos are not properly updated in current frame
            ADDON:FilterMounts()
            ADDON.Events:TriggerEvent("OnNewMount", param1, param2, param3)
        end)
    end, 'external events')

    -- on unpacking mount
    hooksecurefunc(C_MountJournal, "ClearFanfare", function(mountId)
        ADDON:FilterMounts()
        ADDON.Events:TriggerEvent("OnNewMount", mountId)
    end)
end, 'external events')