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