dave/wow-Heimdall
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 4 Wiki Activity

Rework addon loading to properly load on ADDON_LOADED

Browse Source 
insteaad of immediately
This commit is contained in:
PhatPhuckDave
2024-12-12 16:13:36 +01:00
parent ce03160faa
commit 34a8024ce4
7 changed files with 801 additions and 736 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
DeathReporter.lua Normal file
View File

@@ -0,0 +1,8 @@
local addonname, data = ...
---@cast data HeimdallData
---@cast addonname string
data.DeathReporter = {}
function data.DeathReporter.Init()
end

3
DumpTable.lua
View File

@@ -1,5 +1,6 @@
local _, data = ... local addonname, data = ...
---@cast data HeimdallData ---@cast data HeimdallData
---@cast addonname string
if not data.dumpTable then if not data.dumpTable then
---@param table table ---@param table table

401
Heimdall.lua
View File

@@ -1,198 +1,245 @@
local _, data = ... local addonname, data = ...
---@cast data HeimdallData ---@cast data HeimdallData
---@cast addonname string
---@class Heimdall_Data local function init()
---@field who { data: table<string, Player> } ---@class Heimdall_Data
if not Heimdall_Data then Heimdall_Data = {} end ---@field who { data: table<string, Player> }
if not Heimdall_Data then Heimdall_Data = {} end
if not Heimdall_Data.config then Heimdall_Data.config = {} end
-- We don't care about these persisting -- We don't care about these persisting
-- Actually we don't want some of them to persist -- Actually we don't want some of them to persist
-- For those we DO we use (global) Heimdall_Data -- For those we DO we use (global) Heimdall_Data
---@class HeimdallData ---@class HeimdallData
---@field config HeimdallConfig ---@field config HeimdallConfig
---@field raceMap table<string, string> ---@field raceMap table<string, string>
---@field classColors table<string, string> ---@field classColors table<string, string>
---@field stinkies table<string, Player> ---@field stinkies table<string, Player>
---@field messenger HeimdallMessengerData ---@field messenger HeimdallMessengerData
---@field who HeimdallWhoData ---@field who HeimdallWhoData
---@field dumpTable fun(table: any, depth?: number): nil ---@field dumpTable fun(table: any, depth?: number): nil
---@field utf8len fun(input: string): number ---@field utf8len fun(input: string): number
---@field padString fun(input: string, targetLength: number, left?: boolean): string ---@field padString fun(input: string, targetLength: number, left?: boolean): string
---@field GetOrDefault fun(table: table<any, any>, keys: string[], default: any): any
---@field Whoer { Init: fun() }
---@field Messenger { Init: fun() }
---@field Spotter { Init: fun() }
---@field DeathReporter { Init: fun() }
--- Config --- --- Config ---
---@class HeimdallConfig ---@class HeimdallConfig
---@field spotter HeimdallSpotterConfig ---@field spotter HeimdallSpotterConfig
---@field who HeimdallWhoConfig ---@field who HeimdallWhoConfig
---@field messenger HeimdallMessengerConfig ---@field messenger HeimdallMessengerConfig
---@field whisperNotify table<string, string> ---@field whisperNotify table<string, string>
---@class HeimdallSpotterConfig ---@class HeimdallSpotterConfig
---@field enabled boolean ---@field enabled boolean
---@field everyone boolean ---@field everyone boolean
---@field hostile boolean ---@field hostile boolean
---@field alliance boolean ---@field alliance boolean
---@field stinky boolean ---@field stinky boolean
---@field notifyChannel string ---@field notifyChannel string
---@field zoneOverride string? ---@field zoneOverride string?
---@field throttleTime number ---@field throttleTime number
---@class HeimdallWhoConfig ---@class HeimdallWhoConfig
---@field enabled boolean ---@field enabled boolean
---@field ignored table<string, boolean> ---@field ignored table<string, boolean>
---@field notifyChannel string ---@field notifyChannel string
---@field ttl number ---@field ttl number
---@field doWhisper boolean ---@field doWhisper boolean
---@field zoneNotifyFor table<string, boolean> ---@field zoneNotifyFor table<string, boolean>
---@class HeimdallMessengerConfig ---@class HeimdallMessengerConfig
---@field enabled boolean ---@field enabled boolean
--- Data --- --- Data ---
---@class HeimdallMessengerData ---@class HeimdallMessengerData
---@field queue table<string, Message> ---@field queue table<string, Message>
---@field ticker number? ---@field ticker number?
---@class HeimdallWhoData ---@class HeimdallWhoData
---@field updateTicker number? ---@field updateTicker number?
---@field whoTicker number? ---@field whoTicker number?
---@field ignored table<string, boolean> ---@field ignored table<string, boolean>
data.messenger = { data.GetOrDefault = function(table, keys, default)
queue = {} local value = default
} if not table then return value end
data.who = { if not keys then return value end
ignored = {},
}
data.config = { local traverse = table
spotter = { for i = 1, #keys do
enabled = true, local key = keys[i]
everyone = false, if traverse[key] then
hostile = true, traverse = traverse[key]
alliance = false, else
stinky = false, break
notifyChannel = "Foobar", end
zoneOverride = nil,
throttleTime = 1 if i == #keys then
}, value = traverse
who = { end
enabled = true, end
ignored = {}, return value
notifyChannel = "Foobar", end
ttl = 10,
doWhisper = true, data.messenger = {
zoneNotifyFor = { queue = {}
["Orgrimmar"] = true, }
["Thunder Bluff"] = true, data.who = {
["Undercity"] = true, ignored = {},
["Durotar"] = true,
["Echo Isles"] = true,
["Valley of Trials"] = true,
}
},
messenger = {
enabled = true
},
whisperNotify = {
"Extazyk",
"Smokefire",
"Smokemantra",
"Хихихантер",
"Муркот",
"Растафаркрай",
"Frosstmorn",
"Pulsjkee",
"Paskoo",
"Totleta",
"Healleta",
"Deathleta",
"Shootleta",
"Stableta"
} }
}
data.raceMap = { data.config = {
["Orc"] = "Horde", spotter = {
["Undead"] = "Horde", enabled = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "enabled" }, true),
["Tauren"] = "Horde", everyone = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "everyone" }, false),
["Troll"] = "Horde", hostile = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "hostile" }, true),
["Blood Elf"] = "Horde", alliance = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "alliance" }, false),
["Goblin"] = "Horde", stinky = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "stinky" }, false),
["Human"] = "Alliance", notifyChannel = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "notifyChannel" }, "Foobar"),
["Dwarf"] = "Alliance", zoneOverride = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "zoneOverride" }, nil),
["Night Elf"] = "Alliance", throttleTime = data.GetOrDefault(Heimdall_Data, { "config", "spotter", "throttleTime" }, 1)
["Gnome"] = "Alliance", },
["Draenei"] = "Alliance", who = {
["Worgen"] = "Alliance", enabled = data.GetOrDefault(Heimdall_Data, { "config", "who", "enabled" }, true),
["Vulpera"] = "Horde", ignored = data.GetOrDefault(Heimdall_Data, { "config", "who", "ignored" }, {}),
["Nightborne"] = "Horde", notifyChannel = data.GetOrDefault(Heimdall_Data, { "config", "who", "notifyChannel" }, "Foobar"),
["Zandalari Troll"] = "Horde", ttl = data.GetOrDefault(Heimdall_Data, { "config", "who", "ttl" }, 10),
["Kul Tiran"] = "Alliance", doWhisper = data.GetOrDefault(Heimdall_Data, { "config", "who", "doWhisper" }, true),
["Dark Iron Dwarf"] = "Alliance", zoneNotifyFor = data.GetOrDefault(Heimdall_Data, { "config", "who", "zoneNotifyFor" }, {
["Void Elf"] = "Alliance", ["Orgrimmar"] = true,
["Lightforged Draenei"] = "Alliance", ["Thunder Bluff"] = true,
["Mechagnome"] = "Alliance", ["Undercity"] = true,
["Mag'har Orc"] = "Horde" ["Durotar"] = true,
} ["Echo Isles"] = true,
["Valley of Trials"] = true,
}),
},
messenger = {
enabled = data.GetOrDefault(Heimdall_Data, { "config", "messenger", "enabled" }, true),
},
whisperNotify = data.GetOrDefault(Heimdall_Data, { "config", "whisperNotify" }, {
-- "Extazyk",
-- "Smokefire",
-- "Smokemantra",
-- "Хихихантер",
-- "Муркот",
-- "Растафаркрай",
-- "Frosstmorn",
-- "Pulsjkee",
-- "Paskoo",
"Totleta",
"Healleta",
"Deathleta",
"Shootleta",
"Stableta"
}),
}
--/run Heimdall_Data = {config = {who = {enabled = false}}}
--/dump Heimdall_Data
print("138 " .. tostring(Heimdall_Data.config.who.enabled))
print("139 " .. tostring(data.GetOrDefault(Heimdall_Data, { "config", "who", "enabled" }, true)))
print("140 " .. tostring(data.config.who.enabled))
---@type table<string, string> data.raceMap = {
data.classColors = { ["Orc"] = "Horde",
["Warrior"] = "C69B6D", ["Undead"] = "Horde",
["Paladin"] = "F48CBA", ["Tauren"] = "Horde",
["Hunter"] = "AAD372", ["Troll"] = "Horde",
["Rogue"] = "FFF468", ["Blood Elf"] = "Horde",
["Priest"] = "FFFFFF", ["Goblin"] = "Horde",
["Death Knight"] = "C41E3A", ["Human"] = "Alliance",
["Shaman"] = "0070DD", ["Dwarf"] = "Alliance",
["Mage"] = "3FC7EB", ["Night Elf"] = "Alliance",
["Warlock"] = "8788EE", ["Gnome"] = "Alliance",
["Monk"] = "00FF98", ["Draenei"] = "Alliance",
["Druid"] = "FF7C0A", ["Worgen"] = "Alliance",
["Demon Hunter"] = "A330C9" ["Vulpera"] = "Horde",
} ["Nightborne"] = "Horde",
["Zandalari Troll"] = "Horde",
["Kul Tiran"] = "Alliance",
["Dark Iron Dwarf"] = "Alliance",
["Void Elf"] = "Alliance",
["Lightforged Draenei"] = "Alliance",
["Mechagnome"] = "Alliance",
["Mag'har Orc"] = "Horde"
}
data.stinkies = {} data.classColors = {
["Warrior"] = "C69B6D",
["Paladin"] = "F48CBA",
["Hunter"] = "AAD372",
["Rogue"] = "FFF468",
["Priest"] = "FFFFFF",
["Death Knight"] = "C41E3A",
["Shaman"] = "0070DD",
["Mage"] = "3FC7EB",
["Warlock"] = "8788EE",
["Monk"] = "00FF98",
["Druid"] = "FF7C0A",
["Demon Hunter"] = "A330C9"
}
---@param input string data.stinkies = {}
---@return number
data.utf8len = function(input) ---@param input string
if not input then ---@return number
return 0 data.utf8len = function(input)
end if not input then
local len = 0 return 0
local i = 1
local n = #input
while i <= n do
local c = input:byte(i)
if c >= 0 and c <= 127 then
i = i + 1
elseif c >= 194 and c <= 223 then
i = i + 2
elseif c >= 224 and c <= 239 then
i = i + 3
elseif c >= 240 and c <= 244 then
i = i + 4
else
i = i + 1
end end
len = len + 1 local len = 0
end local i = 1
return len local n = #input
end while i <= n do
---@param input string local c = input:byte(i)
---@param targetLength number if c >= 0 and c <= 127 then
---@param left boolean i = i + 1
---@return string elseif c >= 194 and c <= 223 then
data.padString = function(input, targetLength, left) i = i + 2
left = left or false elseif c >= 224 and c <= 239 then
local len = data.utf8len(input) i = i + 3
if len < targetLength then elseif c >= 240 and c <= 244 then
if left then i = i + 4
input = input .. string.rep(" ", targetLength - len) else
else i = i + 1
input = string.rep(" ", targetLength - len) .. input end
len = len + 1
end end
return len
end end
return input ---@param input string
---@param targetLength number
---@param left boolean
---@return string
data.padString = function(input, targetLength, left)
left = left or false
local len = data.utf8len(input)
if len < targetLength then
if left then
input = input .. string.rep(" ", targetLength - len)
else
input = string.rep(" ", targetLength - len) .. input
end
end
return input
end
data.Whoer.Init()
data.Messenger.Init()
data.Spotter.Init()
data.DeathReporter.Init()
end end
local loadedFrame = CreateFrame("Frame")
loadedFrame:RegisterEvent("ADDON_LOADED")
loadedFrame:SetScript("OnEvent", function(self, event, addonName)
if addonName == addonname then
init()
end
end)

4
Heimdall.toc
View File

@@ -5,9 +5,9 @@
## SavedVariables: Heimdall_Data ## SavedVariables: Heimdall_Data
#core #core
Heimdall.lua
CLEUParser.lua CLEUParser.lua
DumpTable.lua
Spotter.lua Spotter.lua
Whoer.lua Whoer.lua
Messenger.lua Messenger.lua
DumpTable.lua Heimdall.lua

154
Messenger.lua
View File

@@ -1,91 +1,95 @@
local _, data = ... local addonname, data = ...
---@cast data HeimdallData ---@cast data HeimdallData
---@cast addonname string
if not data.config.messenger.enabled then return end data.Messenger = {}
function data.Messenger.Init()
if not data.config.messenger.enabled then return end
---@class Message ---@class Message
---@field message string ---@field message string
---@field channel string ---@field channel string
---@field data string ---@field data string
---@type table<string, number> ---@type table<string, number>
local channelIdMap = {} local channelIdMap = {}
local FindOrJoinChannel = function(channelName, password) local FindOrJoinChannel = function(channelName, password)
local function GetChannelId(channelName) local function GetChannelId(channelName)
local channels = { GetChannelList() }
for i = 1, #channels, 2 do
local id = channels[i]
local name = channels[i + 1]
if name == channelName then
return id
end
end
end
local channelId = GetChannelId(channelName)
if not channelId then
print("Channel", tostring(channelName), "not found, joining")
if password then
JoinPermanentChannel(channelName, password)
else
JoinPermanentChannel(channelName)
end
end
channelId = GetChannelId(channelName)
channelIdMap[channelName] = channelId
return channelId
end
local ScanChannels = function()
local channels = { GetChannelList() } local channels = { GetChannelList() }
for i = 1, #channels, 2 do for i = 1, #channels, 2 do
local id = channels[i] local id = channels[i]
local name = channels[i + 1] local name = channels[i + 1]
if name == channelName then channelIdMap[name] = id
return id
end
end end
end end
local channelId = GetChannelId(channelName) if not data.messenger then data.messenger = {} end
if not channelId then if not data.messenger.queue then data.messenger.queue = {} end
print("Channel", tostring(channelName), "not found, joining") if not data.messenger.ticker then
if password then data.messenger.ticker = C_Timer.NewTicker(0.2, function()
JoinPermanentChannel(channelName, password) ---@type Message
else local message = data.messenger.queue[1]
JoinPermanentChannel(channelName) if not message then return end
end if not message.message or message.message == "" then return end
end if not message.channel or message.channel == "" then return end
channelId = GetChannelId(channelName)
channelIdMap[channelName] = channelId
return channelId
end
local ScanChannels = function() -- Map channel names to ids
local channels = { GetChannelList() } if message.channel == "CHANNEL" and message.data and string.match(message.data, "%D") then
for i = 1, #channels, 2 do print("Channel presented as string:", message.data)
local id = channels[i] local channelId = channelIdMap[message.data]
local name = channels[i + 1] if not channelId then
channelIdMap[name] = id print("Channel not found, scanning")
end ScanChannels()
end channelId = channelIdMap[message.data]
end
if not data.messenger then data.messenger = {} end if not channelId then
if not data.messenger.queue then data.messenger.queue = {} end print("Channel not joined, joining")
if not data.messenger.ticker then channelId = FindOrJoinChannel(message.data)
data.messenger.ticker = C_Timer.NewTicker(0.2, function() end
---@type Message print("Channel resolved to id", channelId)
local message = data.messenger.queue[1] message.data = channelId
if not message then return end
if not message.message or message.message == "" then return end
if not message.channel or message.channel == "" then return end
-- Map channel names to ids
if message.channel == "CHANNEL" and message.data and string.match(message.data, "%D") then
print("Channel presented as string:", message.data)
local channelId = channelIdMap[message.data]
if not channelId then
print("Channel not found, scanning")
ScanChannels()
channelId = channelIdMap[message.data]
end end
if not channelId then
print("Channel not joined, joining")
channelId = FindOrJoinChannel(message.data)
end
print("Channel resolved to id", channelId)
message.data = channelId
end
table.remove(data.messenger.queue, 1) table.remove(data.messenger.queue, 1)
if not message.message or message.message == "" then return end if not message.message or message.message == "" then return end
if not message.channel or message.channel == "" then return end if not message.channel or message.channel == "" then return end
if not message.data or message.data == "" then return end if not message.data or message.data == "" then return end
SendChatMessage(message.message, message.channel, nil, message.data) SendChatMessage(message.message, message.channel, nil, message.data)
end) end)
end
--C_Timer.NewTicker(2, function()
-- print("Q")
-- table.insert(data.messenger.queue, {
-- channel = "CHANNEL",
-- data = "Foobar",
-- message = "TEST"
-- })
--end)
end end
--C_Timer.NewTicker(2, function()
-- print("Q")
-- table.insert(data.messenger.queue, {
-- channel = "CHANNEL",
-- data = "Foobar",
-- message = "TEST"
-- })
--end)

205
Spotter.lua
View File

@@ -1,105 +1,108 @@
local _, data = ... local addonname, data = ...
---@cast data HeimdallData ---@cast data HeimdallData
---@cast addonname string
if not data.config.spotter.enabled then return end data.Spotter = {}
function data.Spotter.Init()
local function FormatHP(hp) if not data.config.spotter.enabled then return end
if hp > 1e9 then local function FormatHP(hp)
return string.format("%.1fB", hp / 1e9) if hp > 1e9 then
elseif hp > 1e6 then return string.format("%.1fB", hp / 1e9)
return string.format("%.1fM", hp / 1e6) elseif hp > 1e6 then
elseif hp > 1e3 then return string.format("%.1fM", hp / 1e6)
return string.format("%.1fK", hp / 1e3) elseif hp > 1e3 then
else return string.format("%.1fK", hp / 1e3)
return hp else
return hp
end
end end
---@type table<string, number>
local throttleTable = {}
---@param unit string
---@param name string
---@param faction string
---@param hostile boolean
---@return boolean
---@return string? error
local function ShouldNotify(unit, name, faction, hostile)
if data.config.spotter.stinky then
if data.stinkies[name] then return true end
end
if data.config.spotter.alliance then
if faction == "Alliance" then return true end
end
if data.config.spotter.hostile then
if hostile then return true end
end
return data.config.spotter.everyone
end
---@param unit string
---@return string?
local function NotifySpotted(unit)
if not unit then return string.format("Could not find unit %s", tostring(unit)) end
if not UnitIsPlayer(unit) then return nil end
local name = UnitName(unit)
if not name then return string.format("Could not find name for unit %s", tostring(unit)) end
local time = GetTime()
if throttleTable[name] and time - throttleTable[name] < data.config.spotter.throttleTime then
return string.format("Throttled %s", tostring(name))
end
throttleTable[name] = time
local race = UnitRace(unit)
if not race then return string.format("Could not find race for unit %s", tostring(unit)) end
local faction = data.raceMap[race]
if not faction then return string.format("Could not find faction for race %s", tostring(race)) end
local hostile = UnitCanAttack("player", unit)
local doNotify = ShouldNotify(unit, name, faction, hostile)
if not doNotify then return string.format("Not notifying for %s", tostring(name)) end
local hp = UnitHealth(unit)
if not hp then return string.format("Could not find hp for unit %s", tostring(unit)) end
local maxHp = UnitHealthMax(unit)
if not maxHp then return string.format("Could not find maxHp for unit %s", tostring(unit)) end
local location = data.config.spotter.zoneOverride
if not location then
local zone = GetZoneText()
if not zone then return string.format("Could not find zone for unit %s", tostring(unit)) end
local subzone = GetSubZoneText()
if not subzone then subzone = "" end
location = string.format("%s (%s)", zone, subzone)
end
local text = string.format("I see (%s) %s of race (%s) with health %s/%s at %s",
hostile and "Hostile" or "Friendly",
name,
race,
FormatHP(hp),
FormatHP(maxHp),
location)
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.spotter.notifyChannel,
message = text
}
data.dumpTable(msg)
table.insert(data.messenger.queue, msg)
end
local frame = CreateFrame("Frame")
frame:RegisterEvent("NAME_PLATE_UNIT_ADDED")
frame:RegisterEvent("TARGET_UNIT_CHANGED")
frame:SetScript("OnEvent", function(self, event, unit)
local err = NotifySpotted(unit)
if err then
print(string.format("Error notifying %s: %s", tostring(unit), tostring(err)))
end
end)
end end
---@type table<string, number>
local throttleTable = {}
---@param unit string
---@param name string
---@param faction string
---@param hostile boolean
---@return boolean
---@return string? error
local function ShouldNotify(unit, name, faction, hostile)
if data.config.spotter.stinky then
if data.stinkies[name] then return true end
end
if data.config.spotter.alliance then
if faction == "Alliance" then return true end
end
if data.config.spotter.hostile then
if hostile then return true end
end
return data.config.spotter.everyone
end
---@param unit string
---@return string?
local function NotifySpotted(unit)
if not unit then return string.format("Could not find unit %s", tostring(unit)) end
if not UnitIsPlayer(unit) then return nil end
local name = UnitName(unit)
if not name then return string.format("Could not find name for unit %s", tostring(unit)) end
local time = GetTime()
if throttleTable[name] and time - throttleTable[name] < data.config.spotter.throttleTime then
return string.format("Throttled %s", tostring(name))
end
throttleTable[name] = time
local race = UnitRace(unit)
if not race then return string.format("Could not find race for unit %s", tostring(unit)) end
local faction = data.raceMap[race]
if not faction then return string.format("Could not find faction for race %s", tostring(race)) end
local hostile = UnitCanAttack("player", unit)
local doNotify = ShouldNotify(unit, name, faction, hostile)
if not doNotify then return string.format("Not notifying for %s", tostring(name)) end
local hp = UnitHealth(unit)
if not hp then return string.format("Could not find hp for unit %s", tostring(unit)) end
local maxHp = UnitHealthMax(unit)
if not maxHp then return string.format("Could not find maxHp for unit %s", tostring(unit)) end
local location = data.config.spotter.zoneOverride
if not location then
local zone = GetZoneText()
if not zone then return string.format("Could not find zone for unit %s", tostring(unit)) end
local subzone = GetSubZoneText()
if not subzone then subzone = "" end
location = string.format("%s (%s)", zone, subzone)
end
local text = string.format("I see (%s) %s of race (%s) with health %s/%s at %s",
hostile and "Hostile" or "Friendly",
name,
race,
FormatHP(hp),
FormatHP(maxHp),
location)
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.spotter.notifyChannel,
message = text
}
data.dumpTable(msg)
table.insert(data.messenger.queue, msg)
end
local frame = CreateFrame("Frame")
frame:RegisterEvent("NAME_PLATE_UNIT_ADDED")
frame:RegisterEvent("TARGET_UNIT_CHANGED")
frame:SetScript("OnEvent", function(self, event, unit)
local err = NotifySpotted(unit)
if err then
print(string.format("Error notifying %s: %s", tostring(unit), tostring(err)))
end
end)

762
Whoer.lua
View File

@@ -1,402 +1,404 @@
local _, data = ... local addonname, data = ...
---@cast data HeimdallData ---@cast data HeimdallData
---@cast addonname string
if not data.config.who.enabled then return end data.Whoer = {}
if not Heimdall_Data.who then Heimdall_Data.who = {} end function data.Whoer.Init()
if not Heimdall_Data.who.data then Heimdall_Data.who.data = {} end if not data.config.who.enabled then return end
print(Heimdall_Data) if not Heimdall_Data.who then Heimdall_Data.who = {} end
print(Heimdall_Data.who) if not Heimdall_Data.who.data then Heimdall_Data.who.data = {} end
print(Heimdall_Data.who.data)
---@type table<string, Player> ---@type table<string, Player>
local players = {} local players = {}
---@class Player ---@class Player
---@field name string ---@field name string
---@field guild string ---@field guild string
---@field race string ---@field race string
---@field class string ---@field class string
---@field zone string ---@field zone string
---@field lastSeenInternal number ---@field lastSeenInternal number
---@field lastSeen string ---@field lastSeen string
---@field firstSeen string ---@field firstSeen string
---@field seenCount number ---@field seenCount number
Player = { Player = {
---@param name string ---@param name string
---@param guild string ---@param guild string
---@param race string ---@param race string
---@param class string ---@param class string
---@param zone string ---@param zone string
---@return Player ---@return Player
new = function(name, guild, race, class, zone) new = function(name, guild, race, class, zone)
local self = setmetatable({}, { local self = setmetatable({}, {
__index = Player __index = Player
}) })
self.name = name self.name = name
self.guild = guild self.guild = guild
self.race = race self.race = race
self.class = class self.class = class
self.zone = zone self.zone = zone
self.lastSeenInternal = GetTime() self.lastSeenInternal = GetTime()
self.lastSeen = "never" self.lastSeen = "never"
self.firstSeen = "never" self.firstSeen = "never"
self.seenCount = 0 self.seenCount = 0
return self return self
end, end,
---@return string ---@return string
ToString = function(self) ToString = function(self)
local out = string.format("%s %s %s\nFirst: %s Last: %s Seen: %3d", local out = string.format("%s %s %s\nFirst: %s Last: %s Seen: %3d",
data.padString(self.name, 16, true), data.padString(self.name, 16, true),
data.padString(self.guild, 26, false), data.padString(self.guild, 26, false),
data.padString(self.zone, 26, false), data.padString(self.zone, 26, false),
data.padString(self.firstSeen, 10, true), data.padString(self.firstSeen, 10, true),
data.padString(self.lastSeen, 10, true), data.padString(self.lastSeen, 10, true),
self.seenCount) self.seenCount)
return string.format("|cFF%s%s|r", data.classColors[self.class], out) return string.format("|cFF%s%s|r", data.classColors[self.class], out)
end, end,
---@return string ---@return string
NotifyMessage = function(self) NotifyMessage = function(self)
local text = string.format("%s of class %s and guild %s in %s, first seen: %s, last seen: %s, times seen: %d", local text = string.format(
self.name, "%s of class %s and guild %s in %s, first seen: %s, last seen: %s, times seen: %d",
self.class, self.name,
self.guild, self.class,
self.zone, self.guild,
self.firstSeen, self.zone,
self.lastSeen, self.firstSeen,
self.seenCount) self.lastSeen,
return text self.seenCount)
end return text
}
---@class WHOQuery
---@field query string
---@field filters WHOFilter[]
WHOQuery = {
---@param query string
---@param filters WHOFilter[]
---@return WHOQuery
new = function(query, filters)
local self = setmetatable({}, {
__index = WHOQuery
})
self.query = query
self.filters = filters
return self
end
}
---@alias WHOFilter fun(name: string, guild: string, level: number, race: string, class: string, zone: string): boolean
---@type WHOFilter
local NotSiegeOfOrgrimmarFilter = function(name, guild, level, race, class, zone)
if not zone then
return false
end
return zone ~= "Siege of Orgrimmar"
end
---@type WHOFilter
local AllianceFilter = function(name, guild, level, race, class, zone)
if not race then
return false
end
if not data.raceMap[race] then
return false
end
return data.raceMap[race] == "Alliance"
end
local whoQueryIdx = 1
---@type table<number, WHOQuery>
local whoQueries = {
WHOQuery.new("g-\"БеспредеЛ\"", {}),
WHOQuery.new(
"z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Human\" r-\"Dwarf\" r-\"Night Elf\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new(
"z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Gnome\" r-\"Draenei\" r-\"Worgen\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new(
"z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Kul Tiran\" r-\"Dark Iron Dwarf\" r-\"Void Elf\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new(
"z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Lightforged Draenei\" r-\"Mechagnome\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new("Kekv Demonboo Dotmada Firobot Verminal", {})
}
local queryPending = false
local ttl = #whoQueries * 2
---@type WHOQuery?
local lastQuery = nil
---@param player Player
---@return string?
local function Notify(player)
if not player then return string.format("Cannot notify for nil player %s", tostring(player)) end
if not data.config.who.zoneNotifyFor[player.zone] then
return string.format("Not notifying for zone %s",
tostring(player.zone))
end
local text = player:NotifyMessage()
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.who.notifyChannel,
message = text
}
table.insert(data.messenger.queue, msg)
if data.config.who.doWhisper then
for _, name in pairs(data.config.whisperNotify) do
---@type Message
local msg = {
channel = "WHISPER",
data = name,
message = text
}
table.insert(data.messenger.queue, msg)
end end
end
return nil
end
---@param player Player
---@param zone string
---@return string?
local function NotifyZoneChanged(player, zone)
if not player then return string.format("Cannot notify for nil player %s", tostring(player)) end
if not data.config.who.zoneNotifyFor[zone]
and not data.config.who.zoneNotifyFor[player.zone] then
return string.format("Not notifying for zones %s and %s", tostring(zone), tostring(player.zone))
end
local text = string.format("%s of class %s and guild %s moved to %s",
player.name,
player.class,
player.guild,
zone)
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.who.notifyChannel,
message = text
} }
table.insert(data.messenger.queue, msg)
if data.config.who.doWhisper then ---@class WHOQuery
for _, name in pairs(data.config.whisperNotify) do ---@field query string
---@type Message ---@field filters WHOFilter[]
local msg = { WHOQuery = {
channel = "WHISPER", ---@param query string
data = name, ---@param filters WHOFilter[]
message = text ---@return WHOQuery
} new = function(query, filters)
table.insert(data.messenger.queue, msg) local self = setmetatable({}, {
__index = WHOQuery
})
self.query = query
self.filters = filters
return self
end end
end
return nil
end
---@param player Player
---@return string?
local function NotifyGone(player)
if not player then return string.format("Cannot notify for nil player %s", tostring(player)) end
if not data.config.who.zoneNotifyFor[player.zone] then
return string.format("Not notifying for zone %s",
tostring(player.zone))
end
local text = string.format("%s of class %s and guild %s left %s",
player.name,
player.class,
player.guild,
player.zone)
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.who.notifyChannel,
message = text
} }
table.insert(data.messenger.queue, msg)
if data.config.who.doWhisper then ---@alias WHOFilter fun(name: string, guild: string, level: number, race: string, class: string, zone: string): boolean
for _, name in pairs(data.config.whisperNotify) do ---@type WHOFilter
---@type Message local NotSiegeOfOrgrimmarFilter = function(name, guild, level, race, class, zone)
local msg = { if not zone then
channel = "WHISPER", return false
data = name,
message = text
}
table.insert(data.messenger.queue, msg)
end end
return zone ~= "Siege of Orgrimmar"
end
---@type WHOFilter
local AllianceFilter = function(name, guild, level, race, class, zone)
if not race then
return false
end
if not data.raceMap[race] then
return false
end
return data.raceMap[race] == "Alliance"
end end
return nil local whoQueryIdx = 1
end ---@type table<number, WHOQuery>
local whoQueries = {
local frame = CreateFrame("Frame") WHOQuery.new("g-\"БеспредеЛ\"", {}),
frame:RegisterEvent("WHO_LIST_UPDATE") WHOQuery.new(
frame:SetScript("OnEvent", function(self, event, ...) "z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Human\" r-\"Dwarf\" r-\"Night Elf\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new(
"z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Gnome\" r-\"Draenei\" r-\"Worgen\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new(
"z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Kul Tiran\" r-\"Dark Iron Dwarf\" r-\"Void Elf\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new(
"z-\"Orgrimmar\" z-\"Durotar\" z-\"Valley of Trials\" z-\"Echo Isles\" r-\"Lightforged Draenei\" r-\"Mechagnome\"",
{ NotSiegeOfOrgrimmarFilter, AllianceFilter }),
WHOQuery.new("Kekv Demonboo Dotmada Firobot Verminal", {})
}
local queryPending = false
local ttl = #whoQueries * 2
---@type WHOQuery? ---@type WHOQuery?
local query = lastQuery local lastQuery = nil
if not query then
print("No query wtf?") ---@param player Player
return ---@return string?
local function Notify(player)
if not player then return string.format("Cannot notify for nil player %s", tostring(player)) end
if not data.config.who.zoneNotifyFor[player.zone] then
return string.format("Not notifying for zone %s",
tostring(player.zone))
end
local text = player:NotifyMessage()
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.who.notifyChannel,
message = text
}
table.insert(data.messenger.queue, msg)
if data.config.who.doWhisper then
for _, name in pairs(data.config.whisperNotify) do
---@type Message
local msg = {
channel = "WHISPER",
data = name,
message = text
}
table.insert(data.messenger.queue, msg)
end
end
return nil
end
---@param player Player
---@param zone string
---@return string?
local function NotifyZoneChanged(player, zone)
if not player then return string.format("Cannot notify for nil player %s", tostring(player)) end
if not data.config.who.zoneNotifyFor[zone]
and not data.config.who.zoneNotifyFor[player.zone] then
return string.format("Not notifying for zones %s and %s", tostring(zone), tostring(player.zone))
end
local text = string.format("%s of class %s and guild %s moved to %s",
player.name,
player.class,
player.guild,
zone)
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.who.notifyChannel,
message = text
}
table.insert(data.messenger.queue, msg)
if data.config.who.doWhisper then
for _, name in pairs(data.config.whisperNotify) do
---@type Message
local msg = {
channel = "WHISPER",
data = name,
message = text
}
table.insert(data.messenger.queue, msg)
end
end
return nil
end
---@param player Player
---@return string?
local function NotifyGone(player)
if not player then return string.format("Cannot notify for nil player %s", tostring(player)) end
if not data.config.who.zoneNotifyFor[player.zone] then
return string.format("Not notifying for zone %s",
tostring(player.zone))
end
local text = string.format("%s of class %s and guild %s left %s",
player.name,
player.class,
player.guild,
player.zone)
---@type Message
local msg = {
channel = "CHANNEL",
data = data.config.who.notifyChannel,
message = text
}
table.insert(data.messenger.queue, msg)
if data.config.who.doWhisper then
for _, name in pairs(data.config.whisperNotify) do
---@type Message
local msg = {
channel = "WHISPER",
data = name,
message = text
}
table.insert(data.messenger.queue, msg)
end
end
return nil
end end
for i = 1, GetNumWhoResults() do local frame = CreateFrame("Frame")
local name, guild, level, race, class, zone = GetWhoInfo(i) frame:RegisterEvent("WHO_LIST_UPDATE")
if data.who.ignored[name] then return end frame:SetScript("OnEvent", function(self, event, ...)
local continue = false ---@type WHOQuery?
local query = lastQuery
---@type WHOFilter[] if not query then
local filters = query.filters print("No query wtf?")
for _, filter in pairs(filters) do
if not filter(name, guild, level, race, class, zone) then
-- Mega scuffed, yes...
-- But wow does not have gotos
continue = true
end
end
if not continue then
local timestamp = date("%Y-%m-%dT%H:%M:%S")
local player = players[name]
if not player then
player = Player.new(name, guild, race, class, zone)
if not Heimdall_Data.who then Heimdall_Data.who = {} end
if not Heimdall_Data.who.data then Heimdall_Data.who.data = {} end
local existing = Heimdall_Data.who.data[name]
if existing then
player.lastSeen = existing.lastSeen or "never"
player.firstSeen = existing.firstSeen or "never"
player.seenCount = existing.seenCount or 0
end
if player.firstSeen == "never" then
player.firstSeen = timestamp
end
local stinky = data.stinkies[name]
if stinky then
PlaySoundFile("Interface\\Sounds\\Domination.ogg", "Master")
else
PlaySoundFile("Interface\\Sounds\\Cloak.ogg", "Master")
end
local err = Notify(player)
if err then
print(string.format("Error notifying for %s: %s", tostring(name), tostring(err)))
end
player.lastSeen = timestamp
player.seenCount = player.seenCount + 1
players[name] = player
end
player.lastSeenInternal = GetTime()
if player.zone ~= zone then
local err = NotifyZoneChanged(player, zone)
if err then
print(string.format("Error notifying for %s: %s", tostring(name), tostring(err)))
end
end
player.zone = zone
player.lastSeen = timestamp
players[name] = player
if not Heimdall_Data.who then Heimdall_Data.who = {} end
if not Heimdall_Data.who.data then Heimdall_Data.who.data = {} end
Heimdall_Data.who.data[name] = player
end
end
-- Turns out WA cannot do this (
-- aura_env.UpdateMacro()
_G["FriendsFrameCloseButton"]:Click()
queryPending = false
print(queryPending)
end)
if not data.who.updateTicker then
data.who.updateTicker = C_Timer.NewTicker(0.5, function()
for name, player in pairs(players) do
if player.lastSeenInternal + data.config.who.ttl < GetTime() then
NotifyGone(player)
PlaySoundFile("Interface\\Sounds\\Uncloak.ogg", "Master")
players[name] = nil
end
end
end)
end
if not data.who.whoTicker then
data.who.whoTicker = C_Timer.NewTicker(1, function()
if queryPending then
print("Tried running a who query while one is already pending, previous query:")
data.dumpTable(lastQuery)
return return
end end
queryPending = true
local query = whoQueries[whoQueryIdx] for i = 1, GetNumWhoResults() do
whoQueryIdx = whoQueryIdx + 1 local name, guild, level, race, class, zone = GetWhoInfo(i)
if whoQueryIdx > #whoQueries then if data.who.ignored[name] then return end
whoQueryIdx = 1 local continue = false
---@type WHOFilter[]
local filters = query.filters
for _, filter in pairs(filters) do
if not filter(name, guild, level, race, class, zone) then
-- Mega scuffed, yes...
-- But wow does not have gotos
continue = true
end
end
if not continue then
local timestamp = date("%Y-%m-%dT%H:%M:%S")
local player = players[name]
if not player then
player = Player.new(name, guild, race, class, zone)
if not Heimdall_Data.who then Heimdall_Data.who = {} end
if not Heimdall_Data.who.data then Heimdall_Data.who.data = {} end
local existing = Heimdall_Data.who.data[name]
if existing then
player.lastSeen = existing.lastSeen or "never"
player.firstSeen = existing.firstSeen or "never"
player.seenCount = existing.seenCount or 0
end
if player.firstSeen == "never" then
player.firstSeen = timestamp
end
local stinky = data.stinkies[name]
if stinky then
PlaySoundFile("Interface\\Sounds\\Domination.ogg", "Master")
else
PlaySoundFile("Interface\\Sounds\\Cloak.ogg", "Master")
end
local err = Notify(player)
if err then
print(string.format("Error notifying for %s: %s", tostring(name), tostring(err)))
end
player.lastSeen = timestamp
player.seenCount = player.seenCount + 1
players[name] = player
end
player.lastSeenInternal = GetTime()
if player.zone ~= zone then
local err = NotifyZoneChanged(player, zone)
if err then
print(string.format("Error notifying for %s: %s", tostring(name), tostring(err)))
end
end
player.zone = zone
player.lastSeen = timestamp
players[name] = player
if not Heimdall_Data.who then Heimdall_Data.who = {} end
if not Heimdall_Data.who.data then Heimdall_Data.who.data = {} end
Heimdall_Data.who.data[name] = player
end
end
-- Turns out WA cannot do this (
-- aura_env.UpdateMacro()
_G["FriendsFrameCloseButton"]:Click()
queryPending = false
print(queryPending)
end)
if not data.who.updateTicker then
data.who.updateTicker = C_Timer.NewTicker(0.5, function()
for name, player in pairs(players) do
if player.lastSeenInternal + data.config.who.ttl < GetTime() then
NotifyGone(player)
PlaySoundFile("Interface\\Sounds\\Uncloak.ogg", "Master")
players[name] = nil
end
end
end)
end
if not data.who.whoTicker then
data.who.whoTicker = C_Timer.NewTicker(1, function()
if queryPending then
print("Tried running a who query while one is already pending, previous query:")
data.dumpTable(lastQuery)
return
end
queryPending = true
local query = whoQueries[whoQueryIdx]
whoQueryIdx = whoQueryIdx + 1
if whoQueryIdx > #whoQueries then
whoQueryIdx = 1
end
lastQuery = query
print(string.format("Running who query: %s", tostring(query.query)))
SetWhoToUI(1)
SendWho(query.query)
end)
end
local whoQueryWhisperFrame = CreateFrame("Frame")
whoQueryWhisperFrame:RegisterEvent("CHAT_MSG_WHISPER")
whoQueryWhisperFrame:SetScript("OnEvent", function(self, event, msg, sender)
if msg == "who" then
for _, player in pairs(players) do
local text = player:NotifyMessage()
---@type Message
local msg = {
channel = "WHISPER",
data = sender,
message = text
}
table.insert(data.messenger.queue, msg)
end
end
end)
local whoQueryChannelFrame = CreateFrame("Frame")
whoQueryChannelFrame:RegisterEvent("CHAT_MSG_CHANNEL")
whoQueryChannelFrame:SetScript("OnEvent", function(self, event, msg, sender, ...)
local channelId = select(6, ...)
local channelname = ""
---@type any[]
local channels = { GetChannelList() }
for i = 1, #channels, 2 do
---@type number
local id = channels[i]
---@type string
local name = channels[i + 1]
if id == channelId then
channelname = name
end
end
if channelname ~= data.config.who.notifyChannel then
return
end
if msg == "who" then
for _, player in pairs(players) do
local text = player:NotifyMessage()
---@type Message
local msg = {
channel = "CHANNEL",
data = sender,
message = text
}
table.insert(data.messenger.queue, msg)
end
end end
lastQuery = query
print(string.format("Running who query: %s", tostring(query.query)))
SetWhoToUI(1)
SendWho(query.query)
end) end)
end end
local whoQueryWhisperFrame = CreateFrame("Frame")
whoQueryWhisperFrame:RegisterEvent("CHAT_MSG_WHISPER")
whoQueryWhisperFrame:SetScript("OnEvent", function(self, event, msg, sender)
if msg == "who" then
for _, player in pairs(players) do
local text = player:NotifyMessage()
---@type Message
local msg = {
channel = "WHISPER",
data = sender,
message = text
}
table.insert(data.messenger.queue, msg)
end
end
end)
local whoQueryChannelFrame = CreateFrame("Frame")
whoQueryChannelFrame:RegisterEvent("CHAT_MSG_CHANNEL")
whoQueryChannelFrame:SetScript("OnEvent", function(self, event, msg, sender, ...)
local channelId = select(6, ...)
local channelname = ""
---@type any[]
local channels = { GetChannelList() }
for i = 1, #channels, 2 do
---@type number
local id = channels[i]
---@type string
local name = channels[i + 1]
if id == channelId then
channelname = name
end
end
if channelname ~= data.config.who.notifyChannel then
return
end
if msg == "who" then
for _, player in pairs(players) do
local text = player:NotifyMessage()
---@type Message
local msg = {
channel = "CHANNEL",
data = sender,
message = text
}
table.insert(data.messenger.queue, msg)
end
end
end)