reactos/irc/TechBot/TechBot.IRCLibrary/IrcClient.cs

647 lines
16 KiB
C#
Raw Normal View History

using System;
using System.IO;
using System.Text;
using System.Collections;
using System.Net.Sockets;
namespace TechBot.IRCLibrary
{
/// <summary>
/// Delegate that delivers an IRC message.
/// </summary>
public delegate void MessageReceivedHandler(IrcMessage message);
/// <summary>
/// Delegate that notifies if the user database for a channel has changed.
/// </summary>
public delegate void ChannelUserDatabaseChangedHandler(IrcChannel channel);
/// <summary>
/// An IRC client.
/// </summary>
public class IrcClient
{
/// <summary>
/// Monitor when an IRC command is received.
/// </summary>
private class IrcCommandEventRegistration
{
/// <summary>
/// IRC command to monitor.
/// </summary>
private string command;
public string Command
{
get
{
return command;
}
}
/// <summary>
/// Handler to call when command is received.
/// </summary>
private MessageReceivedHandler handler;
public MessageReceivedHandler Handler
{
get
{
return handler;
}
}
/// <summary>
/// Constructor.
/// </summary>
/// <param name="command">IRC command to monitor.</param>
/// <param name="handler">Handler to call when command is received.</param>
public IrcCommandEventRegistration(string command,
MessageReceivedHandler handler)
{
this.command = command;
this.handler = handler;
}
}
/// <summary>
/// A buffer to store lines of text.
/// </summary>
private class LineBuffer
{
/// <summary>
/// Full lines of text in buffer.
/// </summary>
private ArrayList strings;
/// <summary>
/// Part of the last line of text in buffer.
/// </summary>
private string left = "";
/// <summary>
/// Standard constructor.
/// </summary>
public LineBuffer()
{
strings = new ArrayList();
}
/// <summary>
/// Return true if there is a complete line in the buffer or false if there is not.
/// </summary>
public bool DataAvailable
{
get
{
return (strings.Count > 0);
}
}
/// <summary>
/// Return next complete line in buffer or null if none exists.
/// </summary>
/// <returns>Next complete line in buffer or null if none exists.</returns>
public string Read()
{
if (DataAvailable)
{
string line = strings[0] as string;
strings.RemoveAt(0);
return line;
}
else
{
return null;
}
}
/// <summary>
/// Write a string to buffer splitting it into lines.
/// </summary>
/// <param name="data"></param>
public void Write(string data)
{
data = left + data;
left = "";
string[] sa = data.Split(new char[] { '\n' });
if (sa.Length <= 0)
{
left = data;
return;
}
else
{
left = "";
}
for (int i = 0; i < sa.Length; i++)
{
if (i < sa.Length - 1)
{
/* This is a complete line. Remove any \r characters at the end of the line. */
string line = sa[i].TrimEnd(new char[] { '\r', '\n'});
/* Silently ignore empty lines */
if (!line.Equals(String.Empty))
{
strings.Add(line);
}
}
else
{
/* This may be a partial line. */
left = sa[i];
}
}
}
}
/// <summary>
/// State for asynchronous reads.
/// </summary>
private class StateObject
{
/// <summary>
/// Network stream where data is read from.
/// </summary>
public NetworkStream Stream;
/// <summary>
/// Buffer where data is put.
/// </summary>
public byte[] Buffer;
/// <summary>
/// Constructor.
/// </summary>
/// <param name="stream">Network stream where data is read from.</param>
/// <param name="buffer">Buffer where data is put.</param>
public StateObject(NetworkStream stream, byte[] buffer)
{
this.Stream = stream;
this.Buffer = buffer;
}
}
#region Private fields
private bool firstPingReceived = false;
private System.Text.Encoding encoding = System.Text.Encoding.UTF8;
private TcpClient tcpClient;
private NetworkStream networkStream;
private bool connected = false;
private LineBuffer messageStream;
private ArrayList ircCommandEventRegistrations = new ArrayList();
private ArrayList channels = new ArrayList();
#endregion
#region Public events
public event MessageReceivedHandler MessageReceived;
public event ChannelUserDatabaseChangedHandler ChannelUserDatabaseChanged;
#endregion
#region Public properties
/// <summary>
/// Encoding used.
/// </summary>
public System.Text.Encoding Encoding
{
get
{
return encoding;
}
set
{
encoding = value;
}
}
/// <summary>
/// List of joined channels.
/// </summary>
public ArrayList Channels
{
get
{
return channels;
}
}
#endregion
#region Private methods
/// <summary>
/// Signal MessageReceived event.
/// </summary>
/// <param name="message">Message that was received.</param>
private void OnMessageReceived(IrcMessage message)
{
foreach (IrcCommandEventRegistration icre in ircCommandEventRegistrations)
{
if (message.Command.ToLower().Equals(icre.Command.ToLower()))
{
icre.Handler(message);
}
}
if (MessageReceived != null)
{
MessageReceived(message);
}
}
/// <summary>
/// Signal ChannelUserDatabaseChanged event.
/// </summary>
/// <param name="channel">Message that was received.</param>
private void OnChannelUserDatabaseChanged(IrcChannel channel)
{
if (ChannelUserDatabaseChanged != null)
{
ChannelUserDatabaseChanged(channel);
}
}
/// <summary>
/// Start an asynchronous read.
/// </summary>
private void Receive()
{
if ((networkStream != null) && (networkStream.CanRead))
{
byte[] buffer = new byte[1024];
networkStream.BeginRead(buffer, 0, buffer.Length,
new AsyncCallback(ReadComplete),
new StateObject(networkStream, buffer));
}
else
{
throw new Exception("Socket is closed.");
}
}
/// <summary>
/// Asynchronous read has completed.
/// </summary>
/// <param name="ar">IAsyncResult object.</param>
private void ReadComplete(IAsyncResult ar)
{
StateObject stateObject = (StateObject) ar.AsyncState;
if (stateObject.Stream.CanRead)
{
int bytesReceived = stateObject.Stream.EndRead(ar);
if (bytesReceived > 0)
{
messageStream.Write(Encoding.GetString(stateObject.Buffer, 0, bytesReceived));
while (messageStream.DataAvailable)
{
OnMessageReceived(new IrcMessage(messageStream.Read()));
}
}
}
Receive();
}
/// <summary>
/// Locate channel.
/// </summary>
/// <param name="name">Channel name.</param>
/// <returns>Channel or null if none was found.</returns>
private IrcChannel LocateChannel(string name)
{
foreach (IrcChannel channel in Channels)
{
if (name.ToLower().Equals(channel.Name.ToLower()))
{
return channel;
}
}
return null;
}
/// <summary>
/// Send a PONG message when a PING message is received.
/// </summary>
/// <param name="message">Received IRC message.</param>
private void PingMessageReceived(IrcMessage message)
{
SendMessage(new IrcMessage(IRC.PONG, message.Parameters));
firstPingReceived = true;
}
/// <summary>
/// Process RPL_NAMREPLY message.
/// </summary>
/// <param name="message">Received IRC message.</param>
private void RPL_NAMREPLYMessageReceived(IrcMessage message)
{
try
{
// :Oslo2.NO.EU.undernet.org 353 E101 = #E101 :E101 KongFu_uK @Exception
/* "( "=" / "*" / "@" ) <channel>
:[ "@" / "+" ] <nick> *( " " [ "@" / "+" ] <nick> )
- "@" is used for secret channels, "*" for private
channels, and "=" for others (public channels). */
if (message.Parameters == null)
{
System.Diagnostics.Debug.WriteLine(String.Format("Message has no parameters."));
return;
}
string[] parameters = message.Parameters.Split(new char[] { ' '});
if (parameters.Length < 5)
{
System.Diagnostics.Debug.WriteLine(String.Format("{0} is two few parameters.", parameters.Length));
return;
}
IrcChannelType type;
switch (parameters[1])
{
case "=":
type = IrcChannelType.Public;
break;
case "*":
type = IrcChannelType.Private;
break;
case "@":
type = IrcChannelType.Secret;
break;
default:
type = IrcChannelType.Public;
break;
}
IrcChannel channel = LocateChannel(parameters[2].Substring(1));
if (channel == null)
{
System.Diagnostics.Debug.WriteLine(String.Format("Channel not found '{0}'.",
parameters[2].Substring(1)));
return;
}
string nickname = parameters[3];
if (nickname[0] != ':')
{
System.Diagnostics.Debug.WriteLine(String.Format("String should start with : and not {0}.", nickname[0]));
return;
}
/* Skip : */
IrcUser user = channel.LocateUser(nickname.Substring(1));
if (user == null)
{
user = new IrcUser(this,
nickname.Substring(1));
channel.Users.Add(user);
}
for (int i = 4; i < parameters.Length; i++)
{
nickname = parameters[i];
user = channel.LocateUser(nickname);
if (user == null)
{
user = new IrcUser(this,
nickname);
channel.Users.Add(user);
}
}
}
catch (Exception ex)
{
System.Diagnostics.Debug.WriteLine(String.Format("Ex. {0}", ex));
}
}
/// <summary>
/// Process RPL_ENDOFNAMES message.
/// </summary>
/// <param name="message">Received IRC message.</param>
private void RPL_ENDOFNAMESMessageReceived(IrcMessage message)
{
try
{
/* <channel> :End of NAMES list */
if (message.Parameters == null)
{
System.Diagnostics.Debug.WriteLine(String.Format("Message has no parameters."));
return;
}
string[] parameters = message.Parameters.Split(new char[] { ' ' });
IrcChannel channel = LocateChannel(parameters[1].Substring(1));
if (channel == null)
{
System.Diagnostics.Debug.WriteLine(String.Format("Channel not found '{0}'.",
parameters[0].Substring(1)));
return;
}
OnChannelUserDatabaseChanged(channel);
}
catch (Exception ex)
{
System.Diagnostics.Debug.WriteLine(String.Format("Ex. {0}", ex));
}
}
#endregion
/// <summary>
/// Connect to the specified IRC server on the specified port.
/// </summary>
/// <param name="server">Address of IRC server.</param>
/// <param name="port">Port of IRC server.</param>
public void Connect(string server, int port)
{
if (connected)
{
throw new AlreadyConnectedException();
}
else
{
messageStream = new LineBuffer();
tcpClient = new TcpClient();
tcpClient.Connect(server, port);
tcpClient.NoDelay = true;
tcpClient.LingerState = new LingerOption(false, 0);
networkStream = tcpClient.GetStream();
connected = networkStream.CanRead && networkStream.CanWrite;
if (!connected)
{
throw new Exception("Cannot read and write from socket.");
}
/* Install PING message handler */
MonitorCommand(IRC.PING, new MessageReceivedHandler(PingMessageReceived));
/* 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));
/* Start receiving data */
Receive();
}
}
/// <summary>
/// Disconnect from IRC server.
/// </summary>
public void Diconnect()
{
if (!connected)
{
throw new NotConnectedException();
}
else
{
connected = false;
tcpClient.Close();
tcpClient = null;
}
}
/// <summary>
/// Send an IRC message.
/// </summary>
/// <param name="message">The message to be sent.</param>
public void SendMessage(IrcMessage message)
{
if (!connected)
{
throw new NotConnectedException();
}
/* Serialize sending messages */
lock (typeof(IrcClient))
{
NetworkStream networkStream = tcpClient.GetStream();
byte[] bytes = Encoding.GetBytes(message.Line);
networkStream.Write(bytes, 0, bytes.Length);
networkStream.Flush();
}
}
/// <summary>
/// Monitor when a message with an IRC command is received.
/// </summary>
/// <param name="command">IRC command to monitor.</param>
/// <param name="handler">Handler to call when command is received.</param>
public void MonitorCommand(string command, MessageReceivedHandler handler)
{
if (command == null)
{
throw new ArgumentNullException("command", "Command cannot be null.");
}
if (handler == null)
{
throw new ArgumentNullException("handler", "Handler cannot be null.");
}
ircCommandEventRegistrations.Add(new IrcCommandEventRegistration(command, handler));
}
/// <summary>
/// Talk to the channel.
/// </summary>
/// <param name="nickname">Nickname of user to talk to.</param>
/// <param name="text">Text to send to the channel.</param>
public void TalkTo(string nickname, string text)
{
}
/// <summary>
/// Change nickname.
/// </summary>
/// <param name="nickname">New nickname.</param>
public void ChangeNick(string nickname)
{
if (nickname == null)
{
throw new ArgumentNullException("nickname", "Nickname cannot be null.");
}
/* NICK <nickname> [ <hopcount> ] */
SendMessage(new IrcMessage(IRC.NICK, nickname));
}
/// <summary>
/// Register.
/// </summary>
/// <param name="nickname">New nickname.</param>
/// <param name="realname">Real name. Can be null.</param>
public void Register(string nickname, string realname)
{
if (nickname == null)
{
throw new ArgumentNullException("nickname", "Nickname cannot be null.");
}
firstPingReceived = false;
ChangeNick(nickname);
/* OLD: USER <username> <hostname> <servername> <realname> */
/* NEW: USER <user> <mode> <unused> <realname> */
SendMessage(new IrcMessage(IRC.USER, String.Format("{0} 0 * :{1}",
nickname, realname != null ? realname : "Anonymous")));
/* Wait for PING for up til 10 seconds */
int timer = 0;
while (!firstPingReceived && timer < 200)
{
System.Threading.Thread.Sleep(50);
timer++;
}
}
/// <summary>
/// Join an IRC channel.
/// </summary>
/// <param name="name">Name of channel (without leading #).</param>
/// <returns>New channel.</returns>
public IrcChannel JoinChannel(string name)
{
IrcChannel channel = new IrcChannel(this, name);
channels.Add(channel);
/* JOIN ( <channel> *( "," <channel> ) [ <key> *( "," <key> ) ] ) / "0" */
SendMessage(new IrcMessage(IRC.JOIN, String.Format("#{0}", name)));
return channel;
}
/// <summary>
/// Part an IRC channel.
/// </summary>
/// <param name="channel">IRC channel. If null, the user parts from all channels.</param>
/// <param name="message">Part message. Can be null.</param>
public void PartChannel(IrcChannel channel, string message)
{
/* PART <channel> *( "," <channel> ) [ <Part Message> ] */
if (channel != null)
{
SendMessage(new IrcMessage(IRC.PART, String.Format("#{0}{1}",
channel.Name, message != null ? String.Format(" :{0}", message) : "")));
channels.Remove(channel);
}
else
{
string channelList = null;
foreach (IrcChannel myChannel in Channels)
{
if (channelList == null)
{
channelList = "";
}
else
{
channelList += ",";
}
channelList += myChannel.Name;
}
if (channelList != null)
{
SendMessage(new IrcMessage(IRC.PART, String.Format("#{0}{1}",
channelList, message != null ? String.Format(" :{0}", message) : "")));
Channels.Clear();
}
}
}
}
}