tekkub / greedbeacon

local colorneed, colorgreed = "|cffff0000", "|cffffff00"
local coloredwords = {Greed = colorgreed.."Greed", Need = colorneed.."Need"}
local rolls = {}
 
local function Print(...) ChatFrame1:AddMessage("|cFF33FF99GreedBeacon|r: "..string.join(", ", ...)) end
 
 
local chatframes = {[ChatFrame1] = false, [ChatFrame2] = false, [ChatFrame3] = false, [ChatFrame4] = false, [ChatFrame5] = false, [ChatFrame6] = false, [ChatFrame7] = false}
for frame in pairs(chatframes) do
	for i,v in pairs(frame.messageTypeList) do
		if v == "LOOT" then
			chatframes[frame] = true
			Print("Initializing - "..frame:GetName().. " has loot enabled")
		end
	end
end
 
local origadd, origrem = ChatFrame_AddMessageGroup, ChatFrame_RemoveMessageGroup
ChatFrame_AddMessageGroup = function(frame, channel, ...)
	if channel == "LOOT" then
		chatframes[frame] = true
		Print("ChatFrame_AddMessageGroup", frame:GetName())
	end
	return origadd(frame, channel, ...)
end
ChatFrame_RemoveMessageGroup = function(frame, channel, ...)
	if channel == "LOOT" then
		chatframes[frame] = false
		Print("ChatFrame_RemoveMessageGroup", frame:GetName())
	end
	return origrem(frame, channel, ...)
end
 
 
local function FindRoll(link, player, hasselected)
	for i,roll in ipairs(rolls) do
		if roll._link == link and not roll._winner and (not roll[player] or hasselected) then return roll end
	end
	local newroll = {_link = link}
	table.insert(rolls, newroll)
	return newroll
end
 
 
local f = CreateFrame("Frame")
f:RegisterEvent("CHAT_MSG_LOOT")
f:SetScript("OnEvent", function(self, event, msg)
	local rolltype, rollval, link, player = msg:match("(.+) Roll . (%d+) for (.+) by (.+)")
	if player then
		local roll = FindRoll(link, player, true)
		roll[player] = (rolltype == "Need" and colorneed or colorgreed)..rollval
		roll._type = rolltype
		return
	end
 
	local player, selection, link = msg:match("(.*) has?v?e? selected (.+) for: (.+)")
	if player then
		player = player == "You" and UnitName("player") or player
		FindRoll(link, player)[player] = coloredwords[selection]
		return
	end
 
	local player, link = msg:match("(.*) won: (.+)")
	if player then
		player = player == "You" and UnitName("player") or player
		for i,roll in ipairs(rolls) do
			if roll._link == link and roll[player] and not roll._printed then
				roll._printed = true
				roll._winner = player
				local msg = string.format("%s|Hgreedbeacon:%d|h[%s roll]|h|r %s won %s ", roll._type == "Need" and colorneed or colorgreed, i, roll._type or "???", player, link)
				for frame,val in pairs(chatframes) do if val then frame:AddMessage(msg) end end
				return
			end
		end
		ChatFrame6:AddMessage("GB NO MATCH: "..msg)
		return
	end
end)
 
 
ChatFrame_AddMessageEventFilter("CHAT_MSG_LOOT", function(msg)
	if msg:match("(.*) won: (.+)")
		or msg:match(" has?v?e? selected .+ for: ")
		or msg:match(" Roll . %d+ for .+ by ")
		or (msg:match(" passed on: ") and not msg:match("Everyone passed on: ")) then return true end
end)
 
 
local orig2 = SetItemRef
function SetItemRef(link, text, button)
	local id = link:match("greedbeacon:(%d+)")
	if id then
		ShowUIPanel(ItemRefTooltip)
		if not ItemRefTooltip:IsShown() then ItemRefTooltip:SetOwner(UIParent, "ANCHOR_PRESERVE") end
 
		local i = tonumber(id)
		local val = rolls[i]
		ItemRefTooltip:ClearLines()
		ItemRefTooltip:AddLine(coloredwords[val._type].." roll|r - "..val._link)
		ItemRefTooltip:AddDoubleLine("Winner:", "|cffffffff"..val._winner)
		for i,v in pairs(val) do if string.sub(i, 1, 1) ~= "_" then ItemRefTooltip:AddDoubleLine(i, v) end end
		ItemRefTooltip:Show()
	else return orig2(link, text, button) end
end