interrupts/Interrupts.lua at master · lua-wow/interrupts · 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
-- Blizzard
local IsInInstance = _G.IsInInstance
local IsInGroup = _G.IsInGroup
local IsInRaid = _G.IsInRaid
local GetSpellLink = C_Spell and C_Spell.GetSpellLink or _G.GetSpellLink
local CombatLogGetCurrentEventInfo = C_CombatLog.GetCurrentEventInfo
local CombatLog_Object_IsA = C_CombatLog.DoesObjectMatchFilter
local SendChatMessage = C_ChatInfo.SendChatMessage
local COMBATLOG_FILTER_ME = _G.COMBATLOG_FILTER_ME
local COMBATLOG_FILTER_MINE = _G.COMBATLOG_FILTER_MINE
local COMBATLOG_FILTER_MY_PET = _G.COMBATLOG_FILTER_MY_PET

local LE_PARTY_CATEGORY_HOME = _G.LE_PARTY_CATEGORY_HOME or 1
local LE_PARTY_CATEGORY_INSTANCE = _G.LE_PARTY_CATEGORY_INSTANCE or 2

-- Constants
local INTERRUPT = "Interrupted %s %s!"

----------------------------------------------------------------
-- Interrupt Announce
----------------------------------------------------------------

-- configurations
local cfg = {
    ["spell_link"] = true
}

-- variables used to prevent AoE spam interrupts
local lastTimestamp, lastSpellID = 0, nil

-- create a possesion form by appending ('s) to the string, unless it ends
-- with s, x or z, in which only (') is added.
local function StringPossesion(s)
    if (s:sub(-1):find("[sxzSXZ]")) then
        return s .. "\'"
    end
    return s .. "\'s"
end

local Interrupt = CreateFrame("Frame")
Interrupt:RegisterEvent("PLAYER_LOGIN")
Interrupt:RegisterEvent("PLAYER_ENTERING_WORLD")
Interrupt:RegisterEvent("GROUP_ROSTER_UPDATE")
Interrupt:SetScript("OnEvent", function(self, event, ...)
    -- call one of the event handlers
    self[event](self, ...)
end)

function Interrupt:PLAYER_LOGIN()
    self.unit = "player"
    self.guid = UnitGUID("player")
    self.chatType = "SAY"
    self:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
end

function Interrupt:UpdateChatType()
    local isInInstance = IsInInstance()
    local inInstance, inGroup, inRaid = IsInGroup(LE_PARTY_CATEGORY_INSTANCE), IsInGroup(LE_PARTY_CATEGORY_HOME), IsInRaid()

    if isInInstance then
        if inRaid then
            self.chatType = "RAID"
        else
            self.chatType = "SAY"
        end
    elseif inRaid then
        self.chatType = "RAID"
    elseif inGroup then
        self.chatType = "PARTY"
    elseif inInstance then
        self.chatType = "INSTANCE_CHAT"
    else
        -- chat type "SAY" requires a hardware event when in the outdoor world
        self.chatType = nil
    end
end

function Interrupt:PLAYER_ENTERING_WORLD()
    self:UpdateChatType()
end

function Interrupt:GROUP_ROSTER_UPDATE()
    self:UpdateChatType()
end

function Interrupt:SendChatMessage(destName, spellText)
    if not self.chatType then return end
    local text = INTERRUPT:format(StringPossesion(destName or "?"), spellText)
    SendChatMessage(text, self.chatType)
end

function Interrupt:COMBAT_LOG_EVENT_UNFILTERED()
    local timestamp, eventType, _, sourceGUID, _, sourceFlags, _, destGUID, destName, _, _,
        spellID, _, _, extraSpellID, extraSpellName = CombatLogGetCurrentEventInfo()

    -- check if event type was a spell interrupt
    if (eventType == "SPELL_INTERRUPT") then
        -- ignore self interrupts (quake from mythic affixes)
        if (sourceGUID == destGUID and destGUID == self.guid) then return end

        -- prevents spam announcements
        if (spellID == lastSpellID) and (timestamp - lastTimestamp <= 1) then return end

        -- update last timestamp e spellID
        lastTimestamp, lastSpellID = timestamp, spellID

        -- check if source is the player or belong to player
        if (CombatLog_Object_IsA(sourceFlags, COMBATLOG_FILTER_ME) or
            CombatLog_Object_IsA(sourceFlags, COMBATLOG_FILTER_MINE) or
            CombatLog_Object_IsA(sourceFlags, COMBATLOG_FILTER_MY_PET)) then
            if (cfg.spell_link) then
                local extraSpellLink = GetSpellLink(extraSpellID)
                self:SendChatMessage(destName, extraSpellLink or extraSpellName)
            else
                self:SendChatMessage(destName, extraSpellName)
            end
        end
    end
end