Attempt to use CHANNEL_UI_UPDATE for initial setup.

This commit is contained in:
Andrew Tomaka 2011-08-09 05:43:46 -04:00
parent 7536a5e14d
commit e34a237c84

View file

@ -1,21 +1,21 @@
ChatSaver = LibStub('AceAddon-3.0'):NewAddon('ChatSaver','AceConsole-3.0','AceHook-3.0','AceEvent-3.0'); ChatSaver = LibStub('AceAddon-3.0'):NewAddon('ChatSaver','AceConsole-3.0','AceHook-3.0','AceEvent-3.0')
local core = ChatSaver; local core = ChatSaver
function core:OnInitialize() function core:OnInitialize()
if(ChatSaverDB == nil) then if ChatSaverDB == nil then
core.firstrun = true; core.firstrun = true
ChatSaverDB = {}; ChatSaverDB = {}
else else
core.firstrun = false; core.firstrun = false
end end
end end
function core:OnEnable() function core:OnEnable()
self:Hook('ToggleChatChannel','ToggleChatChannel',true); self:Hook('ToggleChatChannel','ToggleChatChannel',true)
self:RegisterEvent('CHAT_MSG_CHANNEL_NOTICE','ProcessChannelChange'); self:RegisterEvent('CHAT_MSG_CHANNEL_NOTICE','ProcessChannelChange')
if(core.firstrun) then if core.firstrun then
--self:RegisterEvent('CHAT_MSG_CHANNEL_NOTICE','SetupChatSaver'); self:RegisterEvent('CHANNEL_UI_UPDATE','SetupChatSaver')
end end
end end
@ -26,55 +26,55 @@ end
]] -- ]] --
function core:GetChannelInfo(id) function core:GetChannelInfo(id)
local channelNumber,channelName = GetChannelName(id); local channelNumber,channelName = GetChannelName(id)
local channelTable = core:GetChannelTable(); local channelTable = core:GetChannelTable()
if(channelName == nil) then if channelName == nil then
id = channelTable[id]; id = channelTable[id]
if(id ~= nil) then if id ~= nil then
channelNumber,channelName = GetChannelName(id); channelNumber,channelName = GetChannelName(id)
end end
end end
return channelNumber,channelName,core:GetChannelCategory(channelNumber); return channelNumber,channelName,core:GetChannelCategory(channelNumber)
end end
function core:GetChannelTable() function core:GetChannelTable()
local channelList = { GetChannelList() }; local channelList = { GetChannelList() }
local channelTable = {}; local channelTable = {}
for i = 1,#channelList,2 do for i = 1,#channelList,2 do
channelTable[channelList[i]] = channelList[i + 1]; channelTable[channelList[i]] = channelList[i + 1]
channelTable[channelList[i + 1]] = channelList[i]; channelTable[channelList[i + 1]] = channelList[i]
if type(channelList[i + 1]) == 'string' then if type(channelList[i + 1]) == 'string' then
channelTable[channelList[i + 1]:lower()] = channelList[i]; channelTable[channelList[i + 1]:lower()] = channelList[i]
end end
end end
return channelTable; return channelTable
end end
function core:GetChannelCategory(number) function core:GetChannelCategory(number)
for i = 1,GetNumDisplayChannels(),1 do for i = 1,GetNumDisplayChannels(),1 do
local _,_,_,channelNumber,_,_,category = GetChannelDisplayInfo(i); local _,_,_,channelNumber,_,_,category = GetChannelDisplayInfo(i)
if(channelNumber == number) then if channelNumber == number then
return category; return category
end end
end end
end end
--[[ EVENT FUNCTIONS ]] -- --[[ EVENT FUNCTIONS ]] --
function ProcessChannelChange(_,message,_,_,_,_,_,_,index,name) function core:ProcessChannelChange(_,message,_,_,_,_,_,_,index,name)
if message == 'YOU_JOINED' then if message == 'YOU_JOINED' then
local number,_,category = core:GetChannelInfo(name) local number,_,category = core:GetChannelInfo(name)
if category == 'CHANNEL_CATEGORY_CUSTOM' then if category == 'CHANNEL_CATEGORY_CUSTOM' then
ChatSaverDB[name] = {}; ChatSaverDB[name] = {}
ChatSaverDB[name]['frames'] = {}; ChatSaverDB[name]['frames'] = {}
ChatSaverDB[name]['index'] = number; ChatSaverDB[name]['index'] = number
ChatSaverDB[name]['frames'][DEFAULT_CHAT_FRAME:GetID()] = true; ChatSaverDB[name]['frames'][DEFAULT_CHAT_FRAME:GetID()] = true
end end
elseif message == 'YOU_LEFT' then elseif message == 'YOU_LEFT' then
ChatSaverDB[name] = nil ChatSaverDB[name] = nil
@ -82,73 +82,60 @@ function ProcessChannelChange(_,message,_,_,_,_,_,_,index,name)
end end
function core:RejoinChannels(...) function core:RejoinChannels(...)
local currentChannels = {}; local currentChannels = {}
for i = 1,select('#',GetChannelList()) do for i = 1,select('#',GetChannelList()) do
currentChannels[select(i,GetChannelList())] = true currentChannels[select(i,GetChannelList())] = true
end end
local sortedChannels = {}; local sortedChannels = {}
for channel,_ in pairs(ChatSaverDB) do for channel,_ in pairs(ChatSaverDB) do
table.insert(sortedChannels,channel); table.insert(sortedChannels,channel)
end end
table.sort(sortedChannels,function(a,b) return ChatSaverDB[a].index < ChatSaverDB[b].index end); table.sort(sortedChannels,function(a,b) return ChatSaverDB[a].index < ChatSaverDB[b].index end)
for _,channel in pairs(sortedChannels) do for _,channel in pairs(sortedChannels) do
if currentChannels[channel] == nil then if currentChannels[channel] == nil then
JoinPermanentChannel(channel); -- does not place in chat frame properly JoinPermanentChannel(channel) -- does not place in chat frame properly
for index,_ in pairs(ChatSaverDB[channel].frames) do for index,_ in pairs(ChatSaverDB[channel].frames) do
ChatFrame_AddChannel(_G['ChatFrame'..index],channel); ChatFrame_AddChannel(_G['ChatFrame'..index],channel)
end end
end end
end end
self:UnregisterEvent('CHANNEL_UI_UPDATE'); self:UnregisterEvent('CHANNEL_UI_UPDATE')
end end
function core:SetupChatSaver(...) function core:SetupChatSaver(...)
for frame = 1,NUM_CHAT_WINDOWS do for frame = 1,NUM_CHAT_WINDOWS do
local chatWindowChannels = { GetChatWindowChannels(frame) }; local chatWindowChannels = { GetChatWindowChannels(frame) }
for i = 1,#chatWindowChannels,2 do for i = 1,#chatWindowChannels,2 do
local number,name,category = core:GetChannelInfo(chatWindowChannels[i]); local number,name,category = core:GetChannelInfo(chatWindowChannels[i])
if category == 'CHANNEL_CATEGORY_CUSTOM' then if category == 'CHANNEL_CATEGORY_CUSTOM' then
if ChatSaverDB[name] == nil then if ChatSaverDB[name] == nil then
ChatSaverDB[name] = {}; ChatSaverDB[name] = {}
ChatSaverDB[name]['frames'] = {}; ChatSaverDB[name]['frames'] = {}
ChatSaverDB[name]['index'] = number; ChatSaverDB[name]['index'] = number
end end
ChatSaverDB[name]['frames'][frame] = true; ChatSaverDB[name]['frames'][frame] = true
end end
end end
end end
self:UnregisterEvent('CHAT_MSG_CHANNEL_NOTICE'); self:UnregisterEvent('CHANNEL_UI_UPDATE')
end
function core:StoreChannel(_,_,_,_,_,_,_,_,_,name)
local number,channelName,category = core:GetChannelInfo(name);
if category == 'CHANNEL_CATEGORY_CUSTOM' then
ChatSaverDB[name] = {};
ChatSaverDB[name]['frames'] = {};
ChatSaverDB[name]['index'] = number;
ChatSaverDB[name]['frames'][DEFAULT_CHAT_FRAME:GetID()] = true;
end
self:UnregisterEvent('CHAT_MSG_CHANNEL_NOTICE');
end end
--[[ HOOKED FUNCTIONS ]] -- --[[ HOOKED FUNCTIONS ]] --
function core:ToggleChatChannel(checked,channel) function core:ToggleChatChannel(checked,channel)
if ChatSaverDB[channel] == nil then if ChatSaverDB[channel] == nil then
return; return
end end
if checked then if checked then
ChatSaverDB[channel]['frames'][FCF_GetCurrentChatFrameID()] = true; ChatSaverDB[channel]['frames'][FCF_GetCurrentChatFrameID()] = true
else else
ChatSaverDB[channel]['frames'][FCF_GetCurrentChatFrameID()] = nil; ChatSaverDB[channel]['frames'][FCF_GetCurrentChatFrameID()] = nil
end end
end end