278 lines
8.6 KiB
Lua
278 lines
8.6 KiB
Lua
local _, shared = ...
|
|
---@cast shared HeimdallShared
|
|
local ModuleName = "StinkyTracker"
|
|
|
|
---@diagnostic disable-next-line: missing-fields
|
|
shared.StinkyTracker = {}
|
|
function shared.StinkyTracker.Init()
|
|
shared.stinkyTracker = {
|
|
stinkies = 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
|
|
|
|
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))
|
|
end
|
|
for name, stinky in pairs(whoStinkies) do
|
|
if stinky.hostile then
|
|
shared.stinkyTracker.stinkies[name] = stinky
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(
|
|
string.format("[%s] Added hostile stinky from WHO: %s (%s)", ModuleName, name, stinky.class)
|
|
)
|
|
end
|
|
end
|
|
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))
|
|
end
|
|
for name, stinky in pairs(seeStinkies) do
|
|
if stinky.hostile then
|
|
shared.stinkyTracker.stinkies[name] = stinky
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(
|
|
string.format("[%s] Added hostile stinky from SEE: %s (%s)", ModuleName, name, stinky.class)
|
|
)
|
|
end
|
|
end
|
|
if not stinky.hostile then
|
|
shared.stinkyTracker.stinkies[name] = nil
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Removed non-hostile stinky from SEE: %s", ModuleName, name))
|
|
end
|
|
end
|
|
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))
|
|
end
|
|
for name, stinky in pairs(arrivedStinkies) do
|
|
shared.stinkyTracker.stinkies[name] = stinky
|
|
if Heimdall_Data.config.stinkyTracker.debug then
|
|
print(string.format("[%s] Added stinky from ARRIVED: %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
|
|
|
|
for name, stinky in pairs(shared.stinkyTracker.stinkies) do
|
|
if Heimdall_Data.config.agents[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("[Heimdall] StinkyTracker loaded")
|
|
end
|