ChatSaver.lua

@@ -2,6 +2,8 @@ ChatSaver = LibStub('AceAddon-3.0'):NewAddon('ChatSaver','AceConsole-3.0','AceHo
 local core = ChatSaver;
 
 function core:OnInitialize()	
+	self:RegisterChatCommand('cs','SlashCommand');
+	
 	if(ChatSaverDB == nil) then 
 		core.firstrun = true;
 		ChatSaverDB = {}; 
@@ -11,6 +13,8 @@ function core:OnInitialize()
 end
 
 function core:OnEnable()
+	self:RawHook(SlashCmdList,'JOIN','JoinChannel',true);
+	self:RawHook(SlashCmdList,'LEAVE','LeaveChannel',true);
 	self:Hook('ToggleChatChannel','ToggleChatChannel',true);
 	
 	self:RegisterEvent('CHANNEL_UI_UPDATE','RejoinChannels');
@@ -19,68 +23,8 @@ function core:OnEnable()
 	end
 end
 
---[[ 
---	CHANNEL FUNCTIONS 
---	Because GetChannelName() and GetChannelDisplayInfo() are less
---  than good.	
-]] --
-
-function core:GetChannelInfo(id)
-	local channelNumber,channelName = GetChannelName(id);
-	local channelTable = core:GetChannelTable();
-
-	if(channelName == nil) then
-		id = channelTable[id];
-		
-		if(id ~= nil) then
-			channelNumber,channelName = GetChannelName(id);
-		end
-	end
-	
-	return channelNumber,channelName,core:GetChannelCategory(channelNumber);
-end
-
-function core:GetChannelTable()
-	local channelList = { GetChannelList() };
-	local channelTable = {};
-	for i = 1,#channelList,2 do
-		channelTable[channelList[i]] = channelList[i + 1];
-		channelTable[channelList[i + 1]] = channelList[i];
-		if(type(channelList[i + 1]) == 'string') then
-			channelTable[channelList[i + 1]:lower()] = channelList[i];
-		end
-	end
-
-	return channelTable;
-end
-
-function core:GetChannelCategory(number)
-	for i = 1,GetNumDisplayChannels(),1 do
-		local _,_,_,channelNumber,_,_,category = GetChannelDisplayInfo(i);
-		
-		if(channelNumber == number) then
-			return category;
-		end
-	end
-end
-
---[[ EVENT FUNCTIONS ]] --
-
-function core:ProcessChannelChange(_,message,_,_,_,_,_,_,index,name)
-	if message == 'YOU_JOINED' then
-		local number,_,category = core:GetChannelInfo(name)
-                
-		if category == 'CHANNEL_CATEGORY_CUSTOM' then
-			if ChatSaverDB[name] == nil then
-				ChatSaverDB[name] = {}
-				ChatSaverDB[name]['frames'] = {}
-				ChatSaverDB[name]['index'] = number
-				ChatSaverDB[name]['frames'][DEFAULT_CHAT_FRAME:GetID()] = true
-			end
-		end
-	elseif message == 'YOU_LEFT' then
-		ChatSaverDB[name] = nil
-	end
+function core:SlashCommand()
+	core:RejoinChannels();
 end
 
 function core:RejoinChannels(...)
@@ -89,13 +33,7 @@ function core:RejoinChannels(...)
 		currentChannels[select(i,GetChannelList())] = true
 	end
 	
-	local sortedChannels = {};
 	for channel,_ in pairs(ChatSaverDB) do
-		table.insert(sortedChannels,channel);
-	end
-	table.sort(sortedChannels,function(a,b) return ChatSaverDB[a].index < ChatSaverDB[b].index end);
-	
-	for _,channel in pairs(sortedChannels) do
 		if(currentChannels[channel] == nil) then
 			JoinPermanentChannel(channel); -- does not place in chat frame properly
 			for index,_ in pairs(ChatSaverDB[channel].frames) do
@@ -105,20 +43,20 @@ function core:RejoinChannels(...)
 	end
 	
 	self:UnregisterEvent('CHANNEL_UI_UPDATE');
-	self:RegisterEvent('CHAT_MSG_CHANNEL_NOTICE','ProcessChannelChange')
 end
 
 function core:SetupChatSaver(...)
-	for frame = 1,NUM_CHAT_WINDOWS do 
-		local chatWindowChannels = { GetChatWindowChannels(frame) };
-		for i = 1,#chatWindowChannels,2 do
-			local number,name,category = core:GetChannelInfo(chatWindowChannels[i]);
+	local chatWindowChannels = { GetChatWindowChannels(frame) };
 
-			if(category == 'CHANNEL_CATEGORY_CUSTOM') then
+	for frame = 1,NUM_CHAT_WINDOWS do 
+		for i = 1,#chatWindowChannels,2 do
+			local name,zone = chatWindowChannels[i], chatWindowChannels[i + 1];
+			
+			if(zone == 0) then
 				if(ChatSaverDB[name] == nil) then
 					ChatSaverDB[name] = {};
 					ChatSaverDB[name]['frames'] = {};
-					ChatSaverDB[name]['index'] = number;
+					ChatSaverDB[name]['index'] = GetChannelName(name);
 				end
 			
 				ChatSaverDB[name]['frames'][frame] = true;
@@ -127,10 +65,29 @@ function core:SetupChatSaver(...)
 	end
 	
 	self:UnregisterEvent('CHAT_MSG_CHANNEL_NOTICE');
-	self:RegisterEvent('CHAT_MSG_CHANNEL_NOTICE','ProcessChannelChange')
 end
 
---[[ HOOKED FUNCTIONS ]] --
+function core:JoinChannel(msg)
+	self.hooks[SlashCmdList].JOIN(msg);
+	
+	local name = gsub(msg, "%s*([^%s]+).*", "%1");
+
+	if(strlen(name) > 0 and string.match(name,"%a+")) then
+		ChatSaverDB[name] = {};
+		ChatSaverDB[name]['frames'] = {};
+		ChatSaverDB[name]['index'] = GetChannelName(name);
+		ChatSaverDB[name]['frames'][DEFAULT_CHAT_FRAME:GetID()] = true;
+	end
+end
+
+function core:LeaveChannel(msg)
+	self.hooks[SlashCmdList].LEAVE(msg);
+	
+	local number = gsub(msg, "%s*([^%s]+).*", "%1");
+	local _,name = GetChannelName(number);
+	
+	ChatSaverDB[name] = nil;
+end
 
 function core:ToggleChatChannel(checked,channel)
 	if(ChatSaverDB[channel] == nil) then 

ChatSaver.toc

@@ -1,4 +1,4 @@
-## Interface: 42000
+## Interface: 40100
 ## Title: ChatSaver
 ## Version: @project-version@
 ## Author: Gaffer