[gedmurphy] 35264: Give Techbot the ability to check for and ghost existing bots, then rename itself

Author: gedmurphy
Date: Sun Aug 10 08:06:58 2008
New Revision: 35264

URL: http://svn.reactos.org/svn/reactos?rev=35264&view=rev
Log:
Give Techbot the ability to check for and ghost existing bots, then rename itself

Modified:
    trunk/irc/TechBot/TechBot.IRCLibrary/IRC.cs
    trunk/irc/TechBot/TechBot.IRCLibrary/IrcClient.cs
    trunk/irc/TechBot/TechBot.Library/TechBotIrcService.cs

Modified: trunk/irc/TechBot/TechBot.IRCLibrary/IRC.cs
URL:
http://svn.reactos.org/svn/reactos/trunk/irc/TechBot/TechBot.IRCLibrary/IRC…
==============================================================================
--- trunk/irc/TechBot/TechBot.IRCLibrary/IRC.cs [iso-8859-1] (original)
+++ trunk/irc/TechBot/TechBot.IRCLibrary/IRC.cs [iso-8859-1] Sun Aug 10 08:06:58 2008
@@ -17,10 +17,12 @@
 		public const string PRIVMSG = "PRIVMSG";
 		public const string USER = "USER";
 		public const string PASS = "PASS";
-		public const string GHOST = "MSG NICKSERV GHOST";
+		public const string GHOST = "NICKSERV GHOST";
+		public const string NOTICE = "NOTICE";
 
 		public const string RPL_NAMREPLY = "353";
 		public const string RPL_ENDOFNAMES = "366";
+		public const string ERR_NICKNAMEINUSE = "433";
 
 		#endregion
 

Modified: trunk/irc/TechBot/TechBot.IRCLibrary/IrcClient.cs
URL:
http://svn.reactos.org/svn/reactos/trunk/irc/TechBot/TechBot.IRCLibrary/Irc…
==============================================================================
--- trunk/irc/TechBot/TechBot.IRCLibrary/IrcClient.cs [iso-8859-1] (original)
+++ trunk/irc/TechBot/TechBot.IRCLibrary/IrcClient.cs [iso-8859-1] Sun Aug 10 08:06:58
2008
@@ -191,6 +191,7 @@
 
 		#region Private fields
 		private bool firstPingReceived = false;
+		private bool awaitingGhostDeath = false;
 		private System.Text.Encoding encoding = System.Text.Encoding.UTF8;
 		private TcpClient tcpClient;
 		private NetworkStream networkStream;
@@ -198,6 +199,9 @@
 		private LineBuffer messageStream;
 		private ArrayList ircCommandEventRegistrations = new ArrayList();
 		private ArrayList channels = new ArrayList();
+		private string reqNickname;
+		private string curNickname;
+		private string password;
 		#endregion
 
 		#region Public events
@@ -240,6 +244,16 @@
 			}
 		}
 
+		/// <summary>
+		/// Nickname for the bot.
+		/// </summary>
+		public string Nickname
+		{
+			get
+			{
+				return curNickname;
+			}
+		}
 		#endregion
 
 		#region Private methods
@@ -361,6 +375,24 @@
 			firstPingReceived = true;
 		}
 
+			/// <summary>
+		/// Send a PONG message when a PING message is received.
+		/// </summary>
+		/// <param name="message">Received IRC message.</param>
+		private void NoticeMessageReceived(IrcMessage message)
+		{
+			if (awaitingGhostDeath)
+			{
+				string str = string.Format("{0} has been ghosted", reqNickname);
+				if (message.Parameters.Contains(str))
+				{
+					ChangeNick(reqNickname);
+					SubmitPassword(password);
+					awaitingGhostDeath = false;
+				}
+			}
+		}
+
 		/// <summary>
 		/// Process RPL_NAMREPLY message.
 		/// </summary>
@@ -472,6 +504,31 @@
 			}
 		}
 
+		/// <summary>
+		/// Process ERR_NICKNAMEINUSE message.
+		/// </summary>
+		/// <param name="message">Received IRC message.</param>
+		private void ERR_NICKNAMEINUSEMessageReceived(IrcMessage message)
+		{
+			try
+			{
+				if (message.Parameters == null)
+				{
+					System.Diagnostics.Debug.WriteLine(String.Format("Message has no
parameters."));
+					return;
+				}
+				
+				/* Connect with a different name */
+				string[] parameters = message.Parameters.Split(new char[] { ' ' });
+				string nickname = parameters[1];
+				ChangeNick(nickname + "__");
+			}
+			catch (Exception ex)
+			{
+				System.Diagnostics.Debug.WriteLine(String.Format("Ex. {0}", ex));
+			}
+		}
+
 		#endregion
 
 		/// <summary>
@@ -500,10 +557,14 @@
 				}
 				/* Install PING message handler */
 				MonitorCommand(IRC.PING, new MessageReceivedHandler(PingMessageReceived));
+				/* Install NOTICE message handler */
+				MonitorCommand(IRC.NOTICE, new MessageReceivedHandler(NoticeMessageReceived));
 				/* Install RPL_NAMREPLY message handler */
 				MonitorCommand(IRC.RPL_NAMREPLY, new
MessageReceivedHandler(RPL_NAMREPLYMessageReceived));
 				/* Install RPL_ENDOFNAMES message handler */
 				MonitorCommand(IRC.RPL_ENDOFNAMES, new
MessageReceivedHandler(RPL_ENDOFNAMESMessageReceived));
+				/* Install ERR_NICKNAMEINUSE message handler */
+				MonitorCommand(IRC.ERR_NICKNAMEINUSE, new
MessageReceivedHandler(ERR_NICKNAMEINUSEMessageReceived));
 				/* Start receiving data */
 				Receive();
 			}
@@ -520,8 +581,6 @@
 			}
 			else
 			{
-				
-
 				connected = false;
 				tcpClient.Close();
 				tcpClient = null;
@@ -606,10 +665,32 @@
 			if (nickname == null)
 				throw new ArgumentNullException("nickname", "Nickname cannot be
null.");
 
+			Console.WriteLine("Changing nick to {0}\n", nickname);
+			curNickname = nickname;
+
 			/* NICK <nickname> [ <hopcount> ] */
 			SendMessage(new IrcMessage(IRC.NICK, nickname));
 		}
 
+        /// <summary>
+        /// Ghost nickname.
+        /// </summary>
+        /// <param name="nickname">Nickname.</param>
+        public void GhostNick(string nickname,
+                              string password)
+        {
+            if (nickname == null)
+                throw new ArgumentNullException("nickname", "Nickname
cannot be null.");
+
+            if (password == null)
+                throw new ArgumentNullException("password", "Password
cannot be null.");
+
+            awaitingGhostDeath = true;
+
+            /* GHOST <nickname> <password> */
+            SendMessage(new IrcMessage(IRC.GHOST, nickname + " " + password));
+        }
+
 		/// <summary>
 		/// Submit password to identify user.
 		/// </summary>
@@ -618,6 +699,8 @@
 		{
 			if (password == null)
 				throw new ArgumentNullException("password", "Password cannot be
null.");
+
+			this.password = password;
 
 			/* PASS <password> */
 			SendMessage(new IrcMessage(IRC.PASS, password));
@@ -635,12 +718,10 @@
 		{
 			if (nickname == null)
 				throw new ArgumentNullException("nickname", "Nickname cannot be
null.");
+			reqNickname = nickname;
 			firstPingReceived = false;
 			if (password != null)
 			{
-				/* First ghost ourself and then register */
-				if (nickname != null)
-					SendMessage(new IrcMessage(IRC.GHOST, nickname + " " + password));
 				SubmitPassword(password);
 			}
 			ChangeNick(nickname);

Modified: trunk/irc/TechBot/TechBot.Library/TechBotIrcService.cs
URL:
http://svn.reactos.org/svn/reactos/trunk/irc/TechBot/TechBot.Library/TechBo…
==============================================================================
--- trunk/irc/TechBot/TechBot.Library/TechBotIrcService.cs [iso-8859-1] (original)
+++ trunk/irc/TechBot/TechBot.Library/TechBotIrcService.cs [iso-8859-1] Sun Aug 10
08:06:58 2008
@@ -88,7 +88,15 @@
             m_IrcClient.Connect(hostname, port);
             
             m_IrcClient.Register(botname, password, null);
-            Console.WriteLine("Registered as {0}...", botname);
+            Console.WriteLine("Registered as {0}...", m_IrcClient.Nickname);
+
+            /* Did we get the nick we wanted? */
+            if (m_IrcClient.Nickname != botname)
+            {
+                /* there must have been an existing one, kill it */
+                m_IrcClient.GhostNick(botname, password);;
+            }
+
             JoinChannels();
 
             while (!isStopped)
@@ -279,7 +287,7 @@
 						                                injectMessage,
 						                                GetMessageSource(context)));
 						InjectMessage(context,
-						                      injectMessage);
+						              injectMessage);
 					}
 					else
 					{