239 lines
9.7 KiB
C#
239 lines
9.7 KiB
C#
using System;
|
|
using System.Text;
|
|
using System.Collections.Generic;
|
|
using System.Net.Sockets;
|
|
using ChatSharp.Events;
|
|
using System.Timers;
|
|
using ChatSharp.Handlers;
|
|
|
|
namespace ChatSharp
|
|
{
|
|
public partial class IrcClient
|
|
{
|
|
public delegate void MessageHandler(IrcClient client, IrcMessage message);
|
|
private Dictionary<string, MessageHandler> Handlers { get; set; }
|
|
public void SetHandler(string message, MessageHandler handler)
|
|
{
|
|
#if DEBUG
|
|
// This is the default behavior if 3rd parties want to handle certain messages themselves
|
|
// However, if it happens from our own code, we probably did something wrong
|
|
if (Handlers.ContainsKey(message.ToUpper()))
|
|
Console.WriteLine("Warning: {0} handler has been overwritten", message);
|
|
#endif
|
|
message = message.ToUpper();
|
|
Handlers[message] = handler;
|
|
}
|
|
|
|
internal static DateTime DateTimeFromIrcTime(int time)
|
|
{
|
|
return new DateTime(1970, 1, 1).AddSeconds(time);
|
|
}
|
|
|
|
private const int ReadBufferLength = 1024;
|
|
|
|
private byte[] ReadBuffer { get; set; }
|
|
private int ReadBufferIndex { get; set; }
|
|
private string ServerHostname { get; set; }
|
|
private int ServerPort { get; set; }
|
|
private Timer PingTimer { get; set; }
|
|
|
|
public string ServerAddress
|
|
{
|
|
get
|
|
{
|
|
return ServerHostname + ":" + ServerPort;
|
|
}
|
|
internal set
|
|
{
|
|
string[] parts = value.Split(':');
|
|
if (parts.Length > 2 || parts.Length == 0)
|
|
throw new FormatException("Server address is not in correct format ('hostname:port')");
|
|
ServerHostname = parts[0];
|
|
if (parts.Length > 1)
|
|
ServerPort = int.Parse(parts[1]);
|
|
else
|
|
ServerPort = 6667;
|
|
}
|
|
}
|
|
|
|
public Socket Socket { get; set; }
|
|
public Encoding Encoding { get; set; }
|
|
public IrcUser User { get; set; }
|
|
public ChannelCollection Channels { get; private set; }
|
|
public ClientSettings Settings { get; set; }
|
|
|
|
public IrcClient(string serverAddress, IrcUser user)
|
|
{
|
|
if (serverAddress == null) throw new ArgumentNullException("serverAddress");
|
|
if (user == null) throw new ArgumentNullException("user");
|
|
|
|
User = user;
|
|
ServerAddress = serverAddress;
|
|
Encoding = Encoding.UTF8;
|
|
Channels = new ChannelCollection(this);
|
|
Settings = new ClientSettings();
|
|
Handlers = new Dictionary<string, MessageHandler>();
|
|
MessageHandlers.RegisterDefaultHandlers(this);
|
|
}
|
|
|
|
public void ConnectAsync()
|
|
{
|
|
if (Socket != null && Socket.Connected) throw new InvalidOperationException("Socket is already connected to server.");
|
|
Socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
|
|
ReadBuffer = new byte[ReadBufferLength];
|
|
ReadBufferIndex = 0;
|
|
Socket.BeginConnect(ServerHostname, ServerPort, ConnectComplete, null);
|
|
PingTimer = new Timer(30000);
|
|
}
|
|
|
|
private void ConnectComplete(IAsyncResult result)
|
|
{
|
|
Socket.EndConnect(result);
|
|
Socket.BeginReceive(ReadBuffer, ReadBufferIndex, ReadBuffer.Length, SocketFlags.None, DataRecieved, null);
|
|
// Write login info
|
|
if (!string.IsNullOrEmpty(User.Password))
|
|
SendRawMessage("PASS {0}", User.Password);
|
|
SendRawMessage("NICK {0}", User.Nick);
|
|
// hostname, servername are ignored by most IRC servers
|
|
SendRawMessage("USER {0} hostname servername :{1}", User.User, User.RealName);
|
|
}
|
|
|
|
private void DataRecieved(IAsyncResult result)
|
|
{
|
|
SocketError error;
|
|
int length = Socket.EndReceive(result, out error) + ReadBufferIndex;
|
|
if (error != SocketError.Success)
|
|
{
|
|
OnNetworkError(new SocketErrorEventArgs(error));
|
|
return;
|
|
}
|
|
ReadBufferIndex = 0;
|
|
while (length > 0)
|
|
{
|
|
int messageLength = Array.IndexOf(ReadBuffer, (byte)'\n', 0, length);
|
|
if (messageLength == -1) // Incomplete message
|
|
{
|
|
ReadBufferIndex = length;
|
|
break;
|
|
}
|
|
messageLength++;
|
|
var message = Encoding.GetString(ReadBuffer, 0, messageLength - 2); // -2 to remove \r\n
|
|
HandleMessage(message);
|
|
Array.Copy(ReadBuffer, messageLength, ReadBuffer, 0, length - messageLength);
|
|
length -= messageLength;
|
|
}
|
|
Socket.BeginReceive(ReadBuffer, ReadBufferIndex, ReadBuffer.Length - ReadBufferIndex, SocketFlags.None, DataRecieved, null);
|
|
}
|
|
|
|
private void HandleMessage(string rawMessage)
|
|
{
|
|
OnRawMessageRecieved(new RawMessageEventArgs(rawMessage, false));
|
|
var message = new IrcMessage(rawMessage);
|
|
if (Handlers.ContainsKey(message.Command.ToUpper()))
|
|
Handlers[message.Command.ToUpper()](this, message);
|
|
else
|
|
{
|
|
// TODO: Fire an event or something
|
|
}
|
|
}
|
|
|
|
public void SendRawMessage(string message, params object[] format)
|
|
{
|
|
message = string.Format(message, format);
|
|
var data = Encoding.GetBytes(message + "\r\n");
|
|
Socket.BeginSend(data, 0, data.Length, SocketFlags.None, MessageSent, message);
|
|
}
|
|
|
|
public void SendIrcMessage(IrcMessage message)
|
|
{
|
|
SendRawMessage(message.RawMessage);
|
|
}
|
|
|
|
private void MessageSent(IAsyncResult result)
|
|
{
|
|
SocketError error;
|
|
Socket.EndSend(result, out error);
|
|
if (error != SocketError.Success)
|
|
OnNetworkError(new SocketErrorEventArgs(error));
|
|
else
|
|
OnRawMessageSent(new RawMessageEventArgs((string)result.AsyncState, true));
|
|
}
|
|
|
|
public event EventHandler<SocketErrorEventArgs> NetworkError;
|
|
protected internal virtual void OnNetworkError(SocketErrorEventArgs e)
|
|
{
|
|
if (NetworkError != null) NetworkError(this, e);
|
|
}
|
|
public event EventHandler<RawMessageEventArgs> RawMessageSent;
|
|
protected internal virtual void OnRawMessageSent(RawMessageEventArgs e)
|
|
{
|
|
if (RawMessageSent != null) RawMessageSent(this, e);
|
|
}
|
|
public event EventHandler<RawMessageEventArgs> RawMessageRecieved;
|
|
protected internal virtual void OnRawMessageRecieved(RawMessageEventArgs e)
|
|
{
|
|
if (RawMessageRecieved != null) RawMessageRecieved(this, e);
|
|
}
|
|
public event EventHandler<IrcNoticeEventArgs> NoticeRecieved;
|
|
protected internal virtual void OnNoticeRecieved(IrcNoticeEventArgs e)
|
|
{
|
|
if (NoticeRecieved != null) NoticeRecieved(this, e);
|
|
}
|
|
public event EventHandler<ServerMOTDEventArgs> MOTDPartRecieved;
|
|
protected internal virtual void OnMOTDPartRecieved(ServerMOTDEventArgs e)
|
|
{
|
|
if (MOTDPartRecieved != null) MOTDPartRecieved(this, e);
|
|
}
|
|
public event EventHandler<ServerMOTDEventArgs> MOTDRecieved;
|
|
protected internal virtual void OnMOTDRecieved(ServerMOTDEventArgs e)
|
|
{
|
|
if (MOTDRecieved != null) MOTDRecieved(this, e);
|
|
}
|
|
public event EventHandler<PrivateMessageEventArgs> PrivateMessageRecieved;
|
|
protected internal virtual void OnPrivateMessageRecieved(PrivateMessageEventArgs e)
|
|
{
|
|
if (PrivateMessageRecieved != null) PrivateMessageRecieved(this, e);
|
|
}
|
|
public event EventHandler<PrivateMessageEventArgs> ChannelMessageRecieved;
|
|
protected internal virtual void OnChannelMessageRecieved(PrivateMessageEventArgs e)
|
|
{
|
|
if (ChannelMessageRecieved != null) ChannelMessageRecieved(this, e);
|
|
}
|
|
public event EventHandler<PrivateMessageEventArgs> UserMessageRecieved;
|
|
protected internal virtual void OnUserMessageRecieved(PrivateMessageEventArgs e)
|
|
{
|
|
if (UserMessageRecieved != null) UserMessageRecieved(this, e);
|
|
}
|
|
public event EventHandler<ErronousNickEventArgs> NickInUse;
|
|
protected internal virtual void OnNickInUse(ErronousNickEventArgs e)
|
|
{
|
|
if (NickInUse != null) NickInUse(this, e);
|
|
}
|
|
public event EventHandler<ModeChangeEventArgs> ModeChanged;
|
|
protected internal virtual void OnModeChanged(ModeChangeEventArgs e)
|
|
{
|
|
if (ModeChanged != null) ModeChanged(this, e);
|
|
}
|
|
public event EventHandler<ChannelUserEventArgs> UserJoinedChannel;
|
|
protected internal virtual void OnUserJoinedChannel(ChannelUserEventArgs e)
|
|
{
|
|
if (UserJoinedChannel != null) UserJoinedChannel(this, e);
|
|
}
|
|
public event EventHandler<ChannelUserEventArgs> UserPartedChannel;
|
|
protected internal virtual void OnUserPartedChannel(ChannelUserEventArgs e)
|
|
{
|
|
if (UserPartedChannel != null) UserPartedChannel(this, e);
|
|
}
|
|
public event EventHandler<ChannelEventArgs> ChannelListRecieved;
|
|
protected internal virtual void OnChannelListRecieved(ChannelEventArgs e)
|
|
{
|
|
if (ChannelListRecieved != null) ChannelListRecieved(this, e);
|
|
}
|
|
public event EventHandler<EventArgs> ConnectionComplete;
|
|
protected internal virtual void OnConnectionComplete(EventArgs e)
|
|
{
|
|
if (ConnectionComplete != null) ConnectionComplete(this, e);
|
|
}
|
|
}
|
|
}
|