Communications.lua

local _, addon = ...
local L = addon.L

local strformat = string.format
local BNSendGameData, SendAddonMessage, SendChatMessage = BNSendGameData, C_ChatInfo.SendAddonMessage, SendChatMessage

addon.KEYS_TEXT_COMMAND = '!keys'

-- Variables for syncing information
-- Will only accept information from other clients with same version settings
addon.UPDATE_VERSION = 'updateV8'

local versionList = {}
local highestSubVersion, highestMajorVersion = 0, 0

local PrintVersion, CheckInstanceType

-- Interval times for syncing keys between clients
-- Two different time settings for in a raid or otherwise
-- Creates a random variance between +- [.001, .100] to help prevent
-- disconnects from too many addon messages
local SEND_VARIANCE = ((-1)^math.random(1,2)) * math.random(1, 100)/ 10^3 -- random number to space out messages being sent between clients
local SEND_INTERVAL = {}
SEND_INTERVAL[1] = 0.2 + SEND_VARIANCE -- Normal operations
SEND_INTERVAL[2] = 1 + SEND_VARIANCE -- Used when in a raiding environment
SEND_INTERVAL[3] = 2 -- Used for version checks

-- Current setting to be used
-- Changes when player enters a raid instance or not
local SEND_INTERVAL_SETTING = 1 -- What intervel to use for sending key information

AstralComs = CreateFrame('FRAME', 'AstralComs')

function AstralComs:RegisterPrefix(channel, prefix, f)
	channel = channel or 'GUILD' -- Defaults to guild channel

	if self:IsPrefixRegistered(channel, prefix) then return end -- Did we register something to the same channel with the same name?

	if not self.dtbl[channel] then self.dtbl[channel] = {} end
	
	local obj = {}
	obj.method = f
	obj.prefix = prefix

	table.insert(self.dtbl[channel], obj)
end

function AstralComs:UnregisterPrefix(channel, prefix)
	local objs = self.dtbl[channel]
	if not objs then return end
	for id, obj in pairs(objs) do
		if obj.prefix == prefix then
			objs[id] = nil
			break
		end
	end
end

function AstralComs:IsPrefixRegistered(channel, prefix)
	local objs = self.dtbl[channel]
	if not objs then return false end
	for _, obj in pairs(objs) do
		if obj.prefix == prefix then
			return true
		end
	end
	return false
end

function AstralComs:OnEvent(event, prefix, msg, channel, sender)
	if not (prefix == 'AstralKeys') then return end

	if event == 'BN_CHAT_MSG_ADDON' then channel = 'BNET' end -- To handle BNET addon messages, they are actually WHISPER but I like to keep them seperate

	local objs = AstralComs.dtbl[channel]
	if not objs then return end

	local arg, content = msg:match("^(%S*)%s*(.-)$")

	for _, obj in pairs(objs) do
		if obj.prefix == arg then
			obj.method(content, sender, msg)
		end
	end
end

local msgs = setmetatable({}, {__mode='k'})

local function newMsg()
	local msg = next(msgs)
	if msg then
		msgs[msg] = nil
		return msg
	end
	return {}
end

local function delMsg(msg)
	msg[1] = nil
	msgs[msg] = true
end


function AstralComs:NewMessage(prefix, text, channel, target)
	if channel == 'GUILD' then
		if not IsInGuild() then
			return
		end
	end

	local msg = newMsg()

	if channel == 'BNET' then
		msg.method = BNSendGameData
		msg[1] = target
		msg[2] = prefix
		msg[3] = text
	else
		msg.method = SendAddonMessage
		msg[1] = prefix
		msg[2] = text
		msg[3] = channel
		msg[4] = channel == 'WHISPER' and target or ''
	end

	--Let's add it to queue
	self.queue[#self.queue + 1] = msg

	if not self:IsShown() then
		self:Show()
	end
end

function AstralComs:SendMessage()
	local msg = table.remove(self.queue, 1)
	if msg[3] == 'BNET' then
		if select(3, BNGetGameAccountInfo(msg[4])) == 'WoW' and BNConnected() then -- Are they logged into WoW and are we connected to BNET?
			msg.method(unpack(msg, 1, #msg))
		end
	elseif msg[3] == 'WHISPER' then
		if addon.IsFriendOnline(msg[4]) then -- Are they still logged into that toon
			msg.method(unpack(msg, 1, #msg))
		end
	else-- Guild/raid message, just send it
		msg.method(unpack(msg, 1, #msg))
		delMsg(msg)
	end
end

function AstralComs:OnUpdate(elapsed)
	self.delay = self.delay + elapsed

	if self.delay < SEND_INTERVAL[SEND_INTERVAL_SETTING] + self.loadDelay then
		return
	end

	self.loadDelay = 0

	if self.versionPrint then
		CheckInstanceType()
		self.versionPrint = false
		PrintVersion()
	end

	self.delay = 0

	if #self.queue < 1 then -- Don't have any messages to send
		self:Hide()
		return
	end

	self:SendMessage()
end

function AstralComs:Init()
	self:RegisterEvent('CHAT_MSG_ADDON')
	self:RegisterEvent('BN_CHAT_MSG_ADDON')

	self:SetScript('OnEvent', self.OnEvent)
	self:SetScript('OnUpdate', self.OnUpdate)

	self.dtbl = {}
	self.queue = {}

	self:Hide()
	self.delay = 0
	self.loadDelay = 0
	self.versionPrint = false
end
AstralComs:Init()

-- Version checking

local function VersionRequest()
	SendAddonMessage('AstralKeys', 'versionPush ' .. addon.CLIENT_VERSION .. ':' .. addon.PlayerClass(), 'GUILD') -- Bypass the queue, shouldn't cause any issues, very little data is being pushed
end
AstralComs:RegisterPrefix('GUILD', 'versionRequest', VersionRequest)

local function VersionPush(msg, sender)
	local majorVersion, subVersion, class = msg:match('(%d+).(%d+):(%a+)')
	if tonumber(subVersion) > highestSubVersion then
		highestSubVersion = tonumber(subVersion)
	end
	if tonumber(majorVersion) > highestMajorVersion then
		highestMajorVersion = tonumber(majorVersion)
	end
	versionList[sender] = {subVersion = tonumber(subVersion), majorVersion = tonumber(majorVersion), class = class}
end
AstralComs:RegisterPrefix('GUILD', 'versionPush', VersionPush)

PrintVersion = function()
	local outOfDate = 'Out of date: '
	local upToDate = 'Up to date: '
	local notInstalled = 'Not installed: '

	for k,v in pairs(versionList) do
		if v.majorVersion <= highestMajorVersion and tonumber(v.subVersion) < highestSubVersion then
			outOfDate = outOfDate .. strformat('%s(%d.%d) ', WrapTextInColorCode(Ambiguate(k, 'GUILD'), select(4, GetClassColor(v.class))), v.majorVersion, v.subVersion)
		else
			upToDate = upToDate .. strformat('%s(%d.%d) ', WrapTextInColorCode(Ambiguate(k, 'GUILD'), select(4, GetClassColor(v.class))), v.majorVersion, v.subVersion)
		end
	end

	local _, numGuildMembers = GetNumGuildMembers()
	for i = 1, numGuildMembers do
		local unit = GetGuildRosterInfo(i)
		local class = select(11, GetGuildRosterInfo(i))
		if not versionList[unit] then
			notInstalled = notInstalled .. WrapTextInColorCode(Ambiguate(unit, 'GUILD'), select(4, GetClassColor(class))) .. ' '
		end
	end
	ChatFrame1:AddMessage(upToDate)
	ChatFrame1:AddMessage(outOfDate)
	ChatFrame1:AddMessage(notInstalled)
end

function addon.CheckGuildVersion()
	if not IsInGuild() then return end

	highestVersion = 0
	wipe(versionList)
	SendAddonMessage('AstralKeys', 'versionRequest', 'GUILD') -- Bypass the queue, very little data is being pushed, shouldn't cause any issues.
	AstralComs.versionPrint = true
	SEND_INTERVAL_SETTING = 3
	AstralComs.delay = 0
	AstralComs:Show()
end

-- Let's just disable sending information if we are doing a boss fight
-- but keep updating individual keys if we receive them
-- keep the addon channel overhead low
AstralEvents:Register('ENCOUNTER_START', function()
	AstralComs:UnregisterPrefix('GUILD', 'request')
	end, 'encStart')

AstralEvents:Register('ENCOUNTER_END', function()
	AstralComs:RegisterPrefix('GUILD', 'request', AstralKeys_PushKeyList)
	end, 'encStop')

-- Checks to see if we zone into a raid instance,
-- Let's increase the send interval if we are raiding, client sync can wait, dc's can't
CheckInstanceType = function()
	AstralComs.loadDelay = 3
	local inInstance, instanceType = IsInInstance()
	if inInstance and instanceType == 'raid' then
		SEND_INTERVAL_SETTING = 2
	else
		SEND_INTERVAL_SETTING = 1
	end
end
AstralEvents:Register('PLAYER_ENTERING_WORLD', CheckInstanceType, 'entering_world')

function addon.AnnounceCharacterKeys(channel, prefix)
	for i = 1, #AstralCharacters do
		local id = addon.UnitID(strformat('%s-%s', addon.CharacterName(i), addon.CharacterRealm(i)))

		if id then
			local link = addon.CreateKeyLink(addon.UnitMapID(id), addon.UnitKeyLevel(id))
			if channel == 'PARTY' and not IsInGroup() then return end
			if prefix then
				SendChatMessage(strformat('Astral Keys: %s %s', addon.CharacterName(i), link), channel)
			else
				SendChatMessage(strformat('%s %s', addon.CharacterName(i), link), channel)
			end
		end
	end
end

function addon.AnnounceKey()
	local link = addon.CreateKeyLink(addon.keystone.id, addon.keystone.level)
	if not link then return end
	local msg = strformat(L['ANNOUNCE_NEW_KEY'], link)
	if AstralKeysSettings.general.announce_party.isEnabled and IsInGroup() then
		SendChatMessage(msg, 'PARTY')
	end
	if AstralKeysSettings.general.announce_guild.isEnabled and IsInGuild() then
		SendChatMessage(msg, 'GUILD')
	end
end

local function ParseGuildChatCommands(text)
	if UnitLevel('player') ~= addon.EXPANSION_LEVEL then return end -- Don't bother checking anything if the unit is unable to acquire a key
	text = gsub(text, "^%[%a+%] ", "") -- Strip off [SomeName] from message from using Identity-2
	if text:lower() == addon.KEYS_TEXT_COMMAND then
		local guild = GetGuildInfo('player')
		if AstralKeysSettings.general.report_on_message['guild'] or (guild == 'Astral' and addon.PlayerRealm() == 'Area52') then -- Guild leader for Astral desires this setting to be foreced on for members.
			if AstralKeysSettings.general.report_on_message['all_characters'] then
				addon.AnnounceCharacterKeys('GUILD', true)
			elseif addon.keystone.id then
				local keyLink = addon.CreateKeyLink(addon.keystone.id, addon.keystone.level)
				if not keyLink then return end -- Something went wrong
				SendChatMessage(string.format('Astral Keys: %s', keyLink), 'GUILD')
				ReportTimewalkingKey('GUILD')
			else
				if AstralKeysSettings.general.report_on_message.no_key then
					SendChatMessage(strformat('%s: %s', 'Astral Keys', L['NO_KEY']), 'GUILD')
				end
			end
		end
	end
end
AstralEvents:Register('CHAT_MSG_GUILD', ParseGuildChatCommands, 'parseguildchat')

local function ParsePartyChatCommands(text)
	if UnitLevel('player') ~= addon.EXPANSION_LEVEL then return end -- Don't bother checking anything if the unit is unable to acquire a key
	text = gsub(text, "^%[%a+%] ", "") -- Strip off [SomeName] from message from using Identity-2
	if text:lower() == addon.KEYS_TEXT_COMMAND then
		if AstralKeysSettings.general.report_on_message['party'] then
			if AstralKeysSettings.general.report_on_message['all_characters'] then
				addon.AnnounceCharacterKeys('PARTY', true)
			elseif addon.keystone.id then
				local keyLink = addon.CreateKeyLink(addon.keystone.id, addon.keystone.level)
				if not keyLink then return end -- Something went wrong
				SendChatMessage(string.format('Astral Keys: %s', keyLink), 'PARTY')
				ReportTimewalkingKey('PARTY')
			else
				if AstralKeysSettings.general.report_on_message.no_key then
					SendChatMessage(strformat('%s: %s', 'Astral Keys', L['NO_KEY']), 'PARTY')
				end
			end
		end
	end
end
AstralEvents:Register('CHAT_MSG_PARTY', ParsePartyChatCommands, 'parsepartychat')
AstralEvents:Register('CHAT_MSG_PARTY_LEADER', ParsePartyChatCommands, 'parsepartychat')

function ReportTimewalkingKey(reportType)
	local timeWalkingLink
	for bag = 0, 4 do
		local numSlots = C_Container.GetContainerNumSlots(bag)
		for slot = 1, numSlots do
			if (C_Container.GetContainerItemID(bag, slot) == addon.TIMEWALKINGKEY_ITEMID) then
				timeWalkingLink = C_Container.GetContainerItemLink(bag, slot)
				break
			end
		end
		if timeWalkingLink then
			break
		end
	end
	if not timeWalkingLink then return end -- something went wrong, or key doesn't exist this week
	SendChatMessage(string.format('Astral Keys: %s', timeWalkingLink), reportType)
end

local function ParseRaidChatCommands(text)
	if UnitLevel('player') ~= addon.EXPANSION_LEVEL then return end -- Don't bother checking anything if the unit is unable to acquire a key
	text = gsub(text, "^%[%a+%] ", "") -- Strip off [SomeName] from message from using Identity-2
	if text:lower() == addon.KEYS_TEXT_COMMAND then
		if AstralKeysSettings.general.report_on_message['raid'] then
			if AstralKeysSettings.general.report_on_message['all_characters'] then
				addon.AnnounceCharacterKeys('RAID', true)
			elseif addon.keystone.id then
				local keyLink = addon.CreateKeyLink(addon.keystone.id, addon.keystone.level)
				if not keyLink then return end -- Something went wrong
				SendChatMessage(string.format('Astral Keys: %s', keyLink), 'RAID')
				ReportTimewalkingKey('RAID')
			else
				if AstralKeysSettings.general.report_on_message.no_key then
					SendChatMessage(strformat('%s: %s', 'Astral Keys', L['NO_KEY']), 'RAID')
				end
			end
		end
	end
end
AstralEvents:Register('CHAT_MSG_RAID', ParseRaidChatCommands, 'parseraidchat')
AstralEvents:Register('CHAT_MSG_RAID_LEADER', ParseRaidChatCommands, 'parseraidchat')