365 lines
11 KiB
Lua
365 lines
11 KiB
Lua
local _, shared = ...
|
|
---@cast shared HeimdallShared
|
|
local ModuleName = "StinkyTracker"
|
|
|
|
---@class Stinky
|
|
---@field name string
|
|
---@field class string
|
|
---@field seenAt number
|
|
---@field hostile boolean
|
|
|
|
---@class HeimdallStinkyTrackerConfig
|
|
---@field enabled boolean
|
|
---@field debug boolean
|
|
---@field ignoredTimeout number
|
|
---@field channels string[]
|
|
|
|
---@class StinkyTrackerData
|
|
---@field stinkies ReactiveValue<table<string, Stinky>>
|
|
---@field ignored ReactiveValue<table<string, number>>
|
|
|
|
---@class StinkyTracker
|
|
shared.StinkyTracker = {
|
|
---@param stinky Stinky
|
|
---@return boolean
|
|
Track = function(stinky)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Request to track stinky: %s (%s)", ModuleName, stinky.name, stinky.class))
|
|
end
|
|
local ignored = shared.stinkyTracker.ignored[stinky.name]
|
|
-- TODO: Add a config option for the ignored timeout
|
|
if ignored and ignored > GetTime() - 60 then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(
|
|
string.format(
|
|
"[%s] Stinky is ignored, not tracking: %s (%s)",
|
|
ModuleName,
|
|
stinky.name,
|
|
stinky.class
|
|
)
|
|
)
|
|
shared.dump(shared.stinkyTracker.ignored:get())
|
|
shared.dump(shared.stinkyTracker.stinkies:get())
|
|
end
|
|
return false
|
|
else
|
|
-- Timed out or was never ignored
|
|
shared.stinkyTracker.stinkies[stinky.name] = nil
|
|
end
|
|
|
|
shared.stinkyTracker.stinkies[stinky.name] = stinky
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Stinky is now tracked: %s (%s)", ModuleName, stinky.name, stinky.class))
|
|
shared.dump(shared.stinkyTracker.stinkies:get())
|
|
shared.dump(shared.stinkyTracker.ignored:get())
|
|
end
|
|
return true
|
|
end,
|
|
|
|
---@param name string
|
|
---@return nil
|
|
Ignore = function(name)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Request to ignore stinky: %s", ModuleName, name))
|
|
end
|
|
shared.stinkyTracker.ignored[name] = GetTime()
|
|
shared.stinkyTracker.stinkies[name] = nil
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Stinky is now ignored: %s", ModuleName, name))
|
|
shared.dump(shared.stinkyTracker.ignored:get())
|
|
shared.dump(shared.stinkyTracker.stinkies:get())
|
|
end
|
|
end,
|
|
|
|
---@param name string
|
|
---@return boolean
|
|
IsStinky = function(name)
|
|
if not shared.stinkyTracker.stinkies then return false end
|
|
if not shared.stinkyTracker.stinkies[name] then return false end
|
|
if shared.stinkyTracker.ignored[name] then return false end
|
|
return true
|
|
end,
|
|
|
|
---@param callback fun(stinkies: table<string, Stinky>)
|
|
---@return nil
|
|
OnChange = function(callback) shared.stinkyTracker.stinkies:onChange(callback) end,
|
|
|
|
---@param callback fun(name: string, stinky: Stinky)
|
|
---@return nil
|
|
ForEach = function(callback)
|
|
---@type table<string, Stinky>
|
|
local stinkies = shared.stinkyTracker.stinkies:get()
|
|
for name, stinky in pairs(stinkies) do
|
|
callback(name, stinky)
|
|
end
|
|
end,
|
|
|
|
Init = function()
|
|
shared.stinkyTracker = {
|
|
stinkies = ReactiveValue.new({}),
|
|
ignored = ReactiveValue.new({}),
|
|
}
|
|
|
|
local whoRegex = "([^ -/]+)-?%w*/(%w+)"
|
|
---@param msg string
|
|
---@return table<string, Stinky>
|
|
local function ParseWho(msg)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Parsing WHO message: '%s'", ModuleName, msg))
|
|
end
|
|
local stinkies = {}
|
|
for name, class in string.gmatch(msg, whoRegex) do
|
|
stinkies[name] = {
|
|
name = name,
|
|
class = class,
|
|
seenAt = GetTime(),
|
|
hostile = true,
|
|
}
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(
|
|
string.format(
|
|
"[%s] Found hostile player: %s (%s) at %s",
|
|
ModuleName,
|
|
name,
|
|
class,
|
|
date("%H:%M:%S", time())
|
|
)
|
|
)
|
|
shared.dump(stinkies)
|
|
end
|
|
end
|
|
return stinkies
|
|
end
|
|
|
|
local seeRegex = "I see %((%w+)%) ([^ -/]+)-?%w*/(%w+)"
|
|
---@param msg string
|
|
---@return table<string, Stinky>
|
|
local function ParseSee(msg)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Parsing SEE message: '%s'", ModuleName, msg))
|
|
end
|
|
local stinkies = {}
|
|
local aggression, name, class = string.match(msg, seeRegex)
|
|
if not name or not class then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Error: Invalid SEE message format", ModuleName))
|
|
end
|
|
return stinkies
|
|
end
|
|
local stinky = {
|
|
name = name,
|
|
class = class,
|
|
seenAt = GetTime(),
|
|
hostile = aggression == "hostile",
|
|
}
|
|
stinkies[name] = stinky
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(
|
|
string.format(
|
|
"[%s] Found stinky in SEE: %s (%s) - %s at %s",
|
|
ModuleName,
|
|
name,
|
|
class,
|
|
aggression,
|
|
date("%H:%M:%S", time())
|
|
)
|
|
)
|
|
shared.dump(stinkies)
|
|
end
|
|
return stinkies
|
|
end
|
|
|
|
local arrivedRegex = "([^ -/]+)-?%w*; c:([^;]+)"
|
|
local arrivedRegexAlt = "([^ -/]+)-?%w*%(!!!!%); c:([^;]+)"
|
|
---@param msg string
|
|
---@return table<string, Stinky>
|
|
local function ParseArrived(msg)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("%s: Parsing arrived message: %s", ModuleName, msg))
|
|
end
|
|
local stinkies = {}
|
|
local name, class = string.match(msg, arrivedRegex)
|
|
if not name or not class then
|
|
name, class = string.match(msg, arrivedRegexAlt)
|
|
end
|
|
if not name or not class then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("%s: No valid stinky found in arrived message", ModuleName))
|
|
end
|
|
return stinkies
|
|
end
|
|
local stinky = {
|
|
name = name,
|
|
class = class,
|
|
seenAt = GetTime(),
|
|
hostile = true,
|
|
}
|
|
stinkies[name] = stinky
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("%s: Found stinky in arrived: %s/%s", ModuleName, name, class))
|
|
shared.dump(stinkies)
|
|
end
|
|
return stinkies
|
|
end
|
|
|
|
local frame = CreateFrame("Frame")
|
|
frame:RegisterEvent("CHAT_MSG_CHANNEL")
|
|
frame:SetScript("OnEvent", function(self, event, msg, sender, ...)
|
|
--if Heimdall_Data.config.stinkyTracker.debug then
|
|
-- print(string.format("[%s] Event received: %s from %s", ModuleName, event, sender))
|
|
--end
|
|
if not Heimdall_Data.config.stinkyTracker.enabled then
|
|
--if Heimdall_Data.config.stinkyTracker.debug then
|
|
-- print(string.format("[%s] Module disabled, ignoring event", ModuleName))
|
|
--end
|
|
return
|
|
end
|
|
local channelId = select(6, ...)
|
|
local _, channelname = GetChannelName(channelId)
|
|
local ok = false
|
|
for _, channel in pairs(Heimdall_Data.config.stinkyTracker.channels) do
|
|
if channel == channelname then
|
|
ok = true
|
|
break
|
|
end
|
|
end
|
|
if not ok then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Ignoring message from non-master channel: %s", ModuleName, channelname))
|
|
end
|
|
return
|
|
end
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Processing message from master channel: %s", ModuleName, sender))
|
|
shared.dump(Heimdall_Data.config.stinkyTracker)
|
|
end
|
|
|
|
local stinkies = {}
|
|
if string.find(msg, "^who:") then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Processing WHO message from %s", ModuleName, sender))
|
|
end
|
|
local whoStinkies = ParseWho(msg)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Found stinkies in WHO message", ModuleName))
|
|
shared.dump(whoStinkies)
|
|
end
|
|
for name, stinky in pairs(whoStinkies) do
|
|
stinkies[name] = stinky
|
|
end
|
|
end
|
|
if string.find(msg, "^I see") then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Processing SEE message from %s", ModuleName, sender))
|
|
end
|
|
local seeStinkies = ParseSee(msg)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Found stinkies in SEE message", ModuleName))
|
|
shared.dump(seeStinkies)
|
|
end
|
|
for name, stinky in pairs(seeStinkies) do
|
|
stinkies[name] = stinky
|
|
end
|
|
end
|
|
if string.find(msg, "arrived to") or string.find(msg, "moved to") then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Processing ARRIVED message from %s", ModuleName, sender))
|
|
end
|
|
local arrivedStinkies = ParseArrived(msg)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Found stinkies in ARRIVED message", ModuleName))
|
|
shared.dump(arrivedStinkies)
|
|
end
|
|
for name, stinky in pairs(arrivedStinkies) do
|
|
stinkies[name] = stinky
|
|
end
|
|
end
|
|
|
|
for name, stinky in pairs(stinkies) do
|
|
if shared.stinkyTracker.ignored[name] then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Ignoring stinky: %s (%s)", ModuleName, name, stinky.class))
|
|
end
|
|
shared.stinkyTracker.ignored[name] = nil
|
|
else
|
|
shared.stinkyTracker.stinkies[name] = stinky
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Added stinky: %s (%s)", ModuleName, name, stinky.class))
|
|
end
|
|
end
|
|
end
|
|
|
|
-- Log total stinky count after processing
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
local count = 0
|
|
for _ in pairs(shared.stinkyTracker.stinkies:get()) do
|
|
count = count + 1
|
|
end
|
|
print(string.format("[%s] Current total stinkies tracked: %d", ModuleName, count))
|
|
end
|
|
|
|
shared.StinkyTracker.ForEach(function(name, stinky)
|
|
if shared.AgentTracker.IsAgent(name) then
|
|
shared.stinkyTracker.stinkies[name] = nil
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Removed agent from stinkies: %s", ModuleName, name))
|
|
end
|
|
end
|
|
end)
|
|
end)
|
|
|
|
local targetFrame = CreateFrame("Frame")
|
|
targetFrame:RegisterEvent("UNIT_TARGET")
|
|
targetFrame:SetScript("OnEvent", function(self, event, unit)
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Event received: %s for unit: %s", ModuleName, event, unit or "target"))
|
|
end
|
|
unit = "target"
|
|
|
|
if not Heimdall_Data.config.stinkyTracker.enabled then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Module disabled, ignoring event", ModuleName))
|
|
end
|
|
return
|
|
end
|
|
|
|
local name = UnitName(unit)
|
|
if not UnitIsPlayer(unit) then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Target %s is not a player, nothing to do", ModuleName, name))
|
|
end
|
|
return
|
|
end
|
|
|
|
local enemy = UnitCanAttack("player", unit)
|
|
if enemy then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Target %s is enemy - tracking as stinky", ModuleName, name))
|
|
end
|
|
shared.stinkyTracker.stinkies[name] = {
|
|
name = name,
|
|
class = UnitClass(unit),
|
|
seenAt = GetTime(),
|
|
hostile = true,
|
|
}
|
|
return
|
|
end
|
|
|
|
if not shared.stinkyTracker.stinkies[name] then
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Target %s is friendly and not stinky, nothing to do", ModuleName, name))
|
|
end
|
|
return
|
|
end
|
|
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Target %s is friendly and stinky - removing from stinkies", ModuleName, name))
|
|
end
|
|
shared.stinkyTracker.stinkies[name] = nil
|
|
end)
|
|
|
|
if Heimdall_Data.config.stinkyTracker.debug then print(string.format("[%s] Module initialized", ModuleName)) end
|
|
print(string.format("[%s] Module initialized", ModuleName))
|
|
end,
|
|
}
|