aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTill Hoeppner2015-01-04 23:34:37 +0100
committerTill Hoeppner2015-01-04 23:34:37 +0100
commitcd708c3ef3f8c070f4b0c566575fb043c4eb1e8e (patch)
treec600c8399d87fe18c4da05ad121f710ea11d2bac
parent326b5d9d96a5f45fa8b371f2a78394bcd87030e0 (diff)
downloadirsc-cd708c3ef3f8c070f4b0c566575fb043c4eb1e8e.tar.gz
irsc-cd708c3ef3f8c070f4b0c566575fb043c4eb1e8e.tar.xz
irsc-cd708c3ef3f8c070f4b0c566575fb043c4eb1e8e.zip
I should make smaller commits.
-rw-r--r--Cargo.toml7
-rw-r--r--examples/01.rs20
-rw-r--r--src/callback.rs14
-rw-r--r--src/ident.rs2
-rw-r--r--src/lib.rs2
-rw-r--r--src/message.rs811
-rw-r--r--src/server.rs152
7 files changed, 943 insertions, 65 deletions
diff --git a/Cargo.toml b/Cargo.toml
index f3d1253..5dcedd1 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -7,3 +7,10 @@ description = "A lightweight library for building IRC bots."
repository = "https://github.com/hoeppnertill/irsc"
keywords = ["irc", "internet", "protocol"]
license = "WTFPL"
+
+[features]
+ssl = ["openssl"]
+
+[dependencies.openssl]
+version = "*"
+optional = true
diff --git a/examples/01.rs b/examples/01.rs
index 0bf51f5..7cd7dc6 100644
--- a/examples/01.rs
+++ b/examples/01.rs
@@ -2,6 +2,10 @@
extern crate irsc;
+use std::borrow::ToOwned;
+
+use std::sync::{Once, ONCE_INIT};
+
use irsc::server::Server;
use irsc::color::bold;
use irsc::event;
@@ -10,7 +14,9 @@ use irsc::event::{ Event, ParseResult, PrivMsg };
static NAME: &'static str = "rusticbot";
static DESC: &'static str = "A bot, written in Rust.";
-fn callback(arg: (Server, Event)) {
+static START: Once = ONCE_INIT;
+
+fn callback(arg: &(Server, Event)) {
let (mut server, event) = arg;
match event.command[] {
event::PRIVMSG => {
@@ -18,20 +24,24 @@ fn callback(arg: (Server, Event)) {
let response = format!("You wrote: {}", bold(privmsg.content[]));
server.msg(privmsg.from.nickname[], response[]).unwrap();
},
+ event::MODE => {
+ START.doit(|| {
+ server.msg("Syna", "Hey, I'm poking you! *pokes you*").unwrap();
+ //server.msg("Xasin", "Hey, I'm poking you! *pokes you*").unwrap();
+ })
+ },
_ => ()
}
}
fn main() {
let mut s = Server::new();
- s.connect("irc.freenode.org".into_string(), 6667).unwrap();
+ s.connect("irc.tulpa.info".to_owned(), 6667).unwrap();
s.nick(NAME).unwrap();
s.user(NAME, "*", "*", DESC).unwrap();
s.join("#botzoo").unwrap();
- s.msg("flan3002", "Hey, I'm your example bot!").unwrap();
-
- s.events.lock().register(&(callback as fn((Server,Event))));
+ s.events.lock().register(&(callback as fn(&(Server, Event))));
// Dedicate this thread to listening and event processing
s.listen().unwrap();
diff --git a/src/callback.rs b/src/callback.rs
index c62ebf8..b2027f6 100644
--- a/src/callback.rs
+++ b/src/callback.rs
@@ -1,15 +1,15 @@
-pub struct Callback<A: Sized + Send> {
- items: Vec<fn(A)>
+pub struct Callback<A: Sized> {
+ items: Vec<fn(&mut A)>
}
-impl<A: Sized + Clone + Send> Callback<A> {
+impl<A: Sized> Callback<A> {
pub fn new() -> Callback<A> {
Callback { items: Vec::new() }
}
- pub fn register(&mut self, f: &fn(A)) {
- self.items.push(*f)
+ pub fn register(&mut self, f: fn(&mut A)) {
+ self.items.push(f)
}
- pub fn fire(&self, v: &A) {
- for _ in self.items.iter().map(|&c| c(v.clone())) {}
+ pub fn fire(&self, v: &mut A) {
+ for _ in self.items.iter().map(|&c| c(v)) {}
}
}
diff --git a/src/ident.rs b/src/ident.rs
index 6a01d12..099a129 100644
--- a/src/ident.rs
+++ b/src/ident.rs
@@ -3,7 +3,7 @@ use std::borrow::ToOwned;
static PATTERN: Regex = regex!(":(.*)!(.*)@(.*)");
-#[deriving(Show, Clone)]
+#[derive(Show, Clone)]
pub struct Ident {
pub nickname: String,
pub user: String,
diff --git a/src/lib.rs b/src/lib.rs
index 6d0b42d..1c60750 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -11,4 +11,4 @@ pub mod server;
pub mod color;
pub mod ident;
pub mod callback;
-pub mod event;
+pub mod message;
diff --git a/src/message.rs b/src/message.rs
new file mode 100644
index 0000000..edc1874
--- /dev/null
+++ b/src/message.rs
@@ -0,0 +1,811 @@
+#![allow(non_camel_case_types)]
+
+use std::str::FromStr;
+use std::borrow::ToOwned;
+
+#[derive(Clone)]
+pub struct Message {
+ pub prefix: Option<String>,
+ pub command: Command,
+ pub content: Vec<String>,
+ pub suffix: Option<String>
+}
+
+impl Message {
+ pub fn new(prefix: Option<String>, command: Command, content: Vec<String>, suffix: Option<String>) -> Message {
+ Message {
+ prefix: prefix,
+ command: command,
+ content: content,
+ suffix: suffix
+ }
+ }
+
+ pub fn parse(i: &str) -> Option<Message> {
+ let len = i.len();
+ let mut s = i;
+ let prefix = if len >= 1 && s.char_at(0) == ':' {
+ s.find(' ').map(|i| {
+ let p = s.slice_chars(1, i).to_owned();
+ s = s[i..];
+ p
+ })
+ } else { None };
+
+ let command = s.find(' ').map(|i| {
+ let p = s.slice_chars(0, i).to_owned();
+ s = s[i..];
+ p
+ }).and_then(|c| c.parse());
+
+ let mut content = Vec::with_capacity(15);
+ let mut suffix = None;
+ while s.len() > 0 {
+ if s.char_at(0) == ':' {
+ suffix = Some(s.slice_from(1).to_owned());
+ break
+ }
+ s.find(' ').map(|i| {
+ content.push(s.slice_chars(0, i).to_owned());
+ s = s[i..];
+ });
+ }
+
+ command.map(move |c| Message::new(prefix, c, content, suffix))
+ }
+
+ pub fn format(&self) -> String {
+ let mut s = String::with_capacity(512);
+ if let Some(ref p) = self.prefix {
+ s.push(':');
+ s.push_str(p[]);
+ s.push(' ');
+ }
+
+ s.push_str(format!("{} ", self.command)[]);
+
+ if let Some(ref p) = self.suffix {
+ s.push(':');
+ s.push_str(p[]);
+ }
+
+ s
+ }
+}
+
+#[derive(Copy, Clone, Show, PartialEq, Eq, Hash)]
+pub enum Command {
+ PASS,
+ NICK,
+ USER,
+ OPER,
+ MODE,
+ SERVICE,
+ QUIT,
+ SQUIT,
+ JOIN,
+ PART,
+ TOPIC,
+ NAMES,
+ LIST,
+ INVITE,
+ KICK,
+ PRIVMSG,
+ NOTICE,
+ MOTD,
+ LUSERS,
+ VERSION,
+ STATS,
+ LINKS,
+ TIME,
+ CONNECT,
+ TRACE,
+ ADMIN,
+ INFO,
+ SERVLIST,
+ SQUERY,
+ WHO,
+ WHOIS,
+ WHOWAS,
+ KILL,
+ PING,
+ PONG,
+ ERROR,
+ AWAY,
+ REHASH,
+ DIE,
+ RESTART,
+ SUMMON,
+ USERS,
+ WALLOPS,
+ USERHOST,
+ ISON
+}
+
+impl FromStr for Command {
+ fn from_str(s: &str) -> Option<Command> {
+ use self::Command::*;
+ match s {
+ "PASS" => Some(PASS),
+ "NICK" => Some(NICK),
+ "USER" => Some(USER),
+ "OPER" => Some(OPER),
+ "MODE" => Some(MODE),
+ "SERVICE" => Some(SERVICE),
+ "QUIT" => Some(QUIT),
+ "SQUIT" => Some(SQUIT),
+ "JOIN" => Some(JOIN),
+ "PART" => Some(PART),
+ "TOPIC" => Some(TOPIC),
+ "NAMES" => Some(NAMES),
+ "LIST" => Some(LIST),
+ "INVITE" => Some(INVITE),
+ "KICK" => Some(KICK),
+ "PRIVMSG" => Some(PRIVMSG),
+ "NOTICE" => Some(NOTICE),
+ "MOTD" => Some(MOTD),
+ "LUSERS" => Some(LUSERS),
+ "VERSION" => Some(VERSION),
+ "STATS" => Some(STATS),
+ "LINKS" => Some(LINKS),
+ "TIME" => Some(TIME),
+ "CONNECT" => Some(CONNECT),
+ "TRACE" => Some(TRACE),
+ "ADMIN" => Some(ADMIN),
+ "INFO" => Some(INFO),
+ "SERVLIST" => Some(SERVLIST),
+ "SQUERY" => Some(SQUERY),
+ "WHO" => Some(WHO),
+ "WHOIS" => Some(WHOIS),
+ "WHOWAS" => Some(WHOWAS),
+ "KILL" => Some(KILL),
+ "PING" => Some(PING),
+ "PONG" => Some(PONG),
+ "ERROR" => Some(ERROR),
+ "AWAY" => Some(AWAY),
+ "REHASH" => Some(REHASH),
+ "DIE" => Some(DIE),
+ "RESTART" => Some(RESTART),
+ "SUMMON" => Some(SUMMON),
+ "USERS" => Some(USERS),
+ "WALLOPS" => Some(WALLOPS),
+ "USERHOST" => Some(USERHOST),
+ "ISON" => Some(ISON),
+ _ => None
+ }
+ }
+}
+
+#[derive(Show, Copy, PartialEq, Eq)]
+pub enum Response {
+ /// "Welcome to the Internet Relay Network <nick>!<user>@<host>"
+ RPL_WELCOME = 001,
+ /// "Your host is <servername>, running version <ver>"
+ RPL_YOURHOST = 002,
+ /// "This server was created <date>"
+ RPL_CREATED = 003,
+ /// "<servername> <version> <available user modes> <available channel modes>"
+ RPL_MYINFO = 004,
+ /// "Try server <server name>, port <port number>"
+ /// Sent by the server to a user to suggest an alternative
+ /// server. This is often used when the connection is
+ /// refused because the server is already full.
+ RPL_BOUNCE = 005,
+ /// ":*1<reply> *( " " <reply> )"
+ /// - Reply format used by USERHOST to list replies to
+ /// the query list. The reply string is composed as
+ /// follows:
+ ///
+ /// reply = nickname [ "*" ] "=" ( "+" / "-" ) hostname
+ ///
+ /// The '*' indicates whether the client has registered
+ /// as an Operator. The '-' or '+' characters represent
+ /// whether the client has set an AWAY message or not
+ /// respectively.
+ RPL_USERHOST = 302,
+ /// ":*1<nick> *( " " <nick> )"
+ /// - Reply format used by ISON to list replies to the query list.
+ RPL_ISON = 303,
+ /// "<nick> :<away message>"
+ RPL_AWAY = 301,
+ /// ":You are no longer marked as being away"
+ RPL_UNAWAY = 305,
+ /// ":You have been marked as being away"
+ /// - These replies are used with the AWAY command (if
+ /// allowed). RPL_AWAY is sent to any client sending a
+ /// PRIVMSG to a client which is away. RPL_AWAY is only
+ /// sent by the server to which the client is connected.
+ /// Replies RPL_UNAWAY and RPL_NOWAWAY are sent when the
+ /// client removes and sets an AWAY message./
+ RPL_NOWAWAY = 306,
+ /// "<nick> <user> <host> * :<real name>"
+ RPL_WHOISUSER = 311,
+ /// "<nick> <server> :<server info>"
+ RPL_WHOISSERVER = 312,
+ /// "<nick> :is an IRC operator"
+ RPL_WHOISOPERATOR = 313,
+ /// "<nick> <integer> :seconds idle"
+ RPL_WHOISIDLE = 317,
+ /// "<nick> :End of WHOIS list"
+ RPL_ENDOFWHOIS = 318,
+ /// "<nick> :*( ( "@" / "+" ) <channel> " " )"
+ /// - Replies 311 - 313, 317 - 319 are all replies
+ /// generated in response to a WHOIS message. Given that
+ /// there are enough parameters present, the answering
+ /// server MUST either formulate a reply out of the above
+ /// numerics (if the query nick is found) or return an
+ /// error reply. The '*' in RPL_WHOISUSER is there as
+ /// the literal character and not as a wild card. For
+ /// each reply set, only RPL_WHOISCHANNELS may appear
+ /// more than once (for long lists of channel names).
+ /// The '@' and '+' characters next to the channel name
+ /// indicate whether a client is a channel operator or
+ /// has been granted permission to speak on a moderated
+ /// channel. The RPL_ENDOFWHOIS reply is used to mark
+ /// the end of processing a WHOIS message.
+ RPL_WHOISCHANNELS = 319,
+ /// "<nick> <user> <host> * :<real name>"
+ RPL_WHOWASUSER = 314,
+ /// "<nick> :End of WHOWAS"
+ /// - When replying to a WHOWAS message, a server MUST use
+ /// the replies RPL_WHOWASUSER, RPL_WHOISSERVER or
+ /// ERR_WASNOSUCHNICK for each nickname in the presented
+ /// list. At the end of all reply batches, there MUST
+ /// be RPL_ENDOFWHOWAS (even if there was only one reply
+ /// and it was an error).
+ RPL_ENDOFWHOWAS = 369,
+ /// Obsolete. Not used.
+ #[deprecated = "Obsolete. Not used."]
+ RPL_LISTSTART = 321,
+ /// "<channel> <# visible> :<topic>"
+ RPL_LIST = 322,
+ /// ":End of LIST"
+ /// - Replies RPL_LIST, RPL_LISTEND mark the actual replies
+ /// with data and end of the server's response to a LIST
+ /// command. If there are no channels available to return,
+ /// only the end reply MUST be sent.
+ RPL_LISTEND = 323,
+ /// "<channel> <nickname>"
+ RPL_UNIQOPIS = 325,
+ /// "<channel> <mode> <mode params>"
+ RPL_CHANNELMODEIS = 324,
+ /// "<channel> :No topic is set"
+ RPL_NOTOPIC = 331,
+ /// "<channel> :<topic>"
+ /// - When sending a TOPIC message to determine the
+ /// channel topic, one of two replies is sent. If
+ /// the topic is set, RPL_TOPIC is sent back else
+ /// RPL_NOTOPIC.
+ RPL_TOPIC = 332,
+ /// "<channel> <nick>"
+ /// - Returned by the server to indicate that the
+ /// attempted INVITE message was successful and is
+ /// being passed onto the end client.
+ RPL_INVITING = 341,
+ /// "<user> :Summoning user to IRC"
+ /// - Returned by a server answering a SUMMON message to
+ /// indicate that it is summoning that user.
+ RPL_SUMMONING = 342,
+ /// "<channel> <invitemask>"
+ RPL_INVITELIST = 346,
+ /// "<channel> :End of channel invite list"
+ /// - When listing the 'invitations masks' for a given channel,
+ /// a server is required to send the list back using the
+ /// RPL_INVITELIST and RPL_ENDOFINVITELIST messages. A
+ /// separate RPL_INVITELIST is sent for each active mask.
+ /// After the masks have been listed (or if none present) a
+ /// RPL_ENDOFINVITELIST MUST be sent.
+ RPL_ENDOFINVITELIST = 347,
+ /// "<channel> <exceptionmask>"
+ RPL_EXCEPTLIST = 348,
+ /// "<channel> :End of channel exception list"
+ /// - When listing the 'exception masks' for a given channel,
+ /// a server is required to send the list back using the
+ /// RPL_EXCEPTLIST and RPL_ENDOFEXCEPTLIST messages. A
+ /// separate RPL_EXCEPTLIST is sent for each active mask.
+ /// After the masks have been listed (or if none present)
+ /// a RPL_ENDOFEXCEPTLIST MUST be sent./
+ RPL_ENDOFEXCEPTLIST = 349,
+ /// "<version>.<debuglevel> <server> :<comments>"
+ /// - Reply by the server showing its version details.
+ /// The <version> is the version of the software being
+ /// used (including any patchlevel revisions) and the
+ /// <debuglevel> is used to indicate if the server is
+ /// running in "debug mode".
+ ///
+ /// The "comments" field may contain any comments about
+ /// the version or further version details.
+ RPL_VERSION = 351,
+ /// "<channel> <user> <host> <server> <nick>
+ /// ( "H" / "G" > ["*"] [ ( "@" / "+" ) ]
+ /// :<hopcount> <real name>"
+ RPL_WHOREPLY = 352,
+ /// "<name> :End of WHO list"
+ /// - The RPL_WHOREPLY and RPL_ENDOFWHO pair are used
+ /// to answer a WHO message. The RPL_WHOREPLY is only
+ /// sent if there is an appropriate match to the WHO
+ /// query. If there is a list of parameters supplied
+ /// with a WHO message, a RPL_ENDOFWHO MUST be sent
+ /// after processing each list item with <name> being
+ /// the item.
+ RPL_ENDOFWHO = 315,
+ /// "( "=" / "*" / "@" ) <channel>
+ /// :[ "@" / "+" ] <nick> *( " " [ "@" / "+" ] <nick> )
+ /// - "@" is used for secret channels, "*" for private
+ /// channels, and "=" for others (public channels).
+ RPL_NAMREPLY = 353,
+ /// "<channel> :End of NAMES list"
+ /// - To reply to a NAMES message, a reply pair consisting
+ /// of RPL_NAMREPLY and RPL_ENDOFNAMES is sent by the
+ /// server back to the client. If there is no channel
+ /// found as in the query, then only RPL_ENDOFNAMES is
+ /// returned. The exception to this is when a NAMES
+ /// message is sent with no parameters and all visible
+ /// channels and contents are sent back in a series of
+ /// RPL_NAMEREPLY messages with a RPL_ENDOFNAMES to mark
+ /// the end.
+ RPL_ENDOFNAMES = 366,
+ /// "<mask> <server> :<hopcount> <server info>"
+ RPL_LINKS = 364,
+ /// "<mask> :End of LINKS list"
+ /// - In replying to the LINKS message, a server MUST send
+ /// replies back using the RPL_LINKS numeric and mark the
+ /// end of the list using an RPL_ENDOFLINKS reply.
+ RPL_ENDOFLINKS = 365,
+ /// "<channel> <banmask>"
+ RPL_BANLIST = 367,
+ /// "<channel> :End of channel ban list"
+ /// - When listing the active 'bans' for a given channel,
+ /// a server is required to send the list back using the
+ /// RPL_BANLIST and RPL_ENDOFBANLIST messages. A separate
+ /// RPL_BANLIST is sent for each active banmask. After the
+ /// banmasks have been listed (or if none present) a
+ /// RPL_ENDOFBANLIST MUST be sent.
+ RPL_ENDOFBANLIST = 368,
+ /// ":<string>"
+ RPL_INFO = 371,
+ /// ":End of INFO list"
+ /// - A server responding to an INFO message is required to
+ /// send all its 'info' in a series of RPL_INFO messages
+ /// with a RPL_ENDOFINFO reply to indicate the end of the
+ /// replies.
+ RPL_ENDOFINFO = 374,
+ /// ":- <server> Message of the day - "
+ RPL_MOTDSTART = 375,
+ /// ":- <text>"
+ RPL_MOTD = 372,
+ /// ":End of MOTD command"
+ /// - When responding to the MOTD message and the MOTD file
+ /// is found, the file is displayed line by line, with
+ /// each line no longer than 80 characters, using
+ /// RPL_MOTD format replies. These MUST be surrounded
+ /// by a RPL_MOTDSTART (before the RPL_MOTDs) and an
+ /// RPL_ENDOFMOTD (after).
+ RPL_ENDOFMOTD = 376,
+ /// ":You are now an IRC operator"
+ /// - RPL_YOUREOPER is sent back to a client which has
+ /// just successfully issued an OPER message and gained
+ /// operator status.
+ RPL_YOUREOPER = 381,
+ /// "<config file> :Rehashing"
+ /// - If the REHASH option is used and an operator sends
+ /// a REHASH message, an RPL_REHASHING is sent back to
+ /// the operator.
+ RPL_REHASHING = 382,
+ /// "You are service <servicename>"
+ /// - Sent by the server to a service upon successful
+ /// registration.
+ RPL_YOURESERVICE = 383,
+ /// "<server> :<string showing server's local time>"
+ /// - When replying to the TIME message, a server MUST send
+ /// the reply using the RPL_TIME format above. The string
+ /// showing the time need only contain the correct day and
+ /// time there. There is no further requirement for the
+ /// time string.
+ RPL_TIME = 391,
+ /// ":UserID Terminal Host"
+ RPL_USERSSTART = 392,
+ /// ":<username> <ttyline> <hostname>"
+ RPL_USERS = 393,
+ /// ":End of users"
+ RPL_ENDOFUSERS = 394,
+ /// ":Nobody logged in"
+ /// - If the USERS message is handled by a server, the
+ /// replies RPL_USERSTART, RPL_USERS, RPL_ENDOFUSERS and
+ /// RPL_NOUSERS are used. RPL_USERSSTART MUST be sent
+ /// first, following by either a sequence of RPL_USERS
+ /// or a single RPL_NOUSER. Following this is
+ /// RPL_ENDOFUSERS.
+ RPL_NOUSERS = 395,
+ /// "Link <version & debug level> <destination>
+ /// <next server> V<protocol version>
+ /// <link uptime in seconds> <backstream sendq>
+ /// <upstream sendq>"
+ RPL_TRACELINK = 200,
+ /// "Try. <class> <server>"
+ RPL_TRACECONNECTING = 201,
+ /// "H.S. <class> <server>"
+ RPL_TRACEHANDSHAKE = 202,
+ /// "???? <class> [<client IP address in dot form>]"
+ RPL_TRACEUNKNOWN = 203,
+ /// "Oper <class> <nick>"
+ RPL_TRACEOPERATOR = 204,
+ /// "User <class> <nick>"
+ RPL_TRACEUSER = 205,
+ /// "Serv <class> <int>S <int>C <server>
+ /// <nick!user|*!*>@<host|server> V<protocol version>"
+ RPL_TRACESERVER = 206,
+ /// "Service <class> <name> <type> <active type>"
+ RPL_TRACESERVICE = 207,
+ /// "<newtype> 0 <client name>"
+ RPL_TRACENEWTYPE = 208,
+ /// "Class <class> <count>"
+ RPL_TRACECLASS = 209,
+ /// Unused.
+ #[deprecated = "Unused."]
+ RPL_TRACERECONNECT = 210,
+ /// "File <logfile> <debug level>"
+ RPL_TRACELOG = 261,
+ /// "<server name> <version & debug level> :End of TRACE"
+ /// - The RPL_TRACE* are all returned by the server in
+ /// response to the TRACE message. How many are
+ /// returned is dependent on the TRACE message and
+ /// whether it was sent by an operator or not. There
+ /// is no predefined order for which occurs first.
+ /// Replies RPL_TRACEUNKNOWN, RPL_TRACECONNECTING and
+ /// RPL_TRACEHANDSHAKE are all used for connections
+ /// which have not been fully established and are either
+ /// unknown, still attempting to connect or in the
+ /// process of completing the 'server handshake'.
+ /// RPL_TRACELINK is sent by any server which handles
+ /// a TRACE message and has to pass it on to another
+ /// server. The list of RPL_TRACELINKs sent in
+ /// response to a TRACE command traversing the IRC
+ /// network should reflect the actual connectivity of
+ /// the servers themselves along that path.
+ /// RPL_TRACENEWTYPE is to be used for any connection
+ /// which does not fit in the other categories but is
+ /// being displayed anyway.
+ /// RPL_TRACEEND is sent to indicate the end of the list./
+ RPL_TRACEEND = 262,
+ /// "<linkname> <sendq> <sent messages>
+ /// <sent Kbytes> <received messages>
+ /// <received Kbytes> <time open>"
+ /// - reports statistics on a connection. <linkname>
+ /// identifies the particular connection, <sendq> is
+ /// the amount of data that is queued and waiting to be
+ /// sent <sent messages> the number of messages sent,
+ /// and <sent Kbytes> the amount of data sent, in
+ /// Kbytes. <received messages> and <received Kbytes>
+ /// are the equivalent of <sent messages> and <sent
+ /// Kbytes> for received data, respectively. <time
+ /// open> indicates how long ago the connection was
+ /// opened, in seconds.
+ RPL_STATSLINKINFO = 211,
+ /// "<command> <count> <byte count> <remote count>"
+ /// - reports statistics on commands usage.
+ RPL_STATSCOMMAND = 212,
+ /// "<stats letter> :End of STATS report"
+ RPL_ENDOFSTATS = 219,
+ /// ":Server Up %d days %d:%02d:%02d"
+ /// - reports the server uptime.
+ RPL_STATSUPTIME = 242,
+ /// "O <hostmask> * <name>"
+ /// - reports the allowed hosts from where user may become IRC
+ /// operators.
+ RPL_STATSOLINE = 243,
+ /// "<user mode string>"
+ /// - To answer a query about a client's own mode,
+ /// RPL_UMODEIS is sent back.
+ RPL_UMODEIS = 221,
+ /// "<name> <server> <mask> <type> <hopcount> <info>"
+ RPL_SERVLIST = 234,
+ /// "<mask> <type> :End of service listing"
+ /// - When listing services in reply to a SERVLIST message,
+ /// a server is required to send the list back using the
+ /// RPL_SERVLIST and RPL_SERVLISTEND messages. A separate
+ /// RPL_SERVLIST is sent for each service. After the
+ /// services have been listed (or if none present) a
+ /// RPL_SERVLISTEND MUST be sent./
+ RPL_SERVLISTEND = 235,
+ /// ":There are <integer> users and <integer>
+ /// services on <integer> servers"
+ RPL_LUSERCLIENT = 251,
+ /// "<integer> :operator(s) online"
+ RPL_LUSEROP = 252,
+ /// "<integer> :unknown connection(s)"
+ RPL_LUSERUNKNOWN = 253,
+ /// "<integer> :channels formed"
+ RPL_LUSERCHANNELS = 254,
+ /// ":I have <integer> clients and <integer>
+ /// servers"
+ /// - In processing an LUSERS message, the server
+ /// sends a set of replies from RPL_LUSERCLIENT,
+ /// RPL_LUSEROP, RPL_USERUNKNOWN,
+ /// RPL_LUSERCHANNELS and RPL_LUSERME. When
+ /// replying, a server MUST send back
+ /// RPL_LUSERCLIENT and RPL_LUSERME. The other
+ /// replies are only sent back if a non-zero count
+ /// is found for them.
+ RPL_LUSERME = 255,
+ /// "<server> :Administrative info"
+ RPL_ADMINME = 256,
+ /// ":<admin info>"
+ RPL_ADMINLOC1 = 257,
+ /// ":<admin info>"
+ RPL_ADMINLOC2 = 258,
+ /// ":<admin info>"
+ /// - When replying to an ADMIN message, a server
+ /// is expected to use replies RPL_ADMINME
+ /// through to RPL_ADMINEMAIL and provide a text
+ /// message with each. For RPL_ADMINLOC1 a
+ /// description of what city, state and country
+ /// the server is in is expected, followed by
+ /// details of the institution (RPL_ADMINLOC2)
+ /// and finally the administrative contact for the
+ /// server (an email address here is REQUIRED)
+ /// in RPL_ADMINEMAIL.
+ RPL_ADMINEMAIL = 259,
+ /// "<command> :Please wait a while and try again."
+ /// - When a server drops a command without processing it,
+ /// it MUST use the reply RPL_TRYAGAIN to inform the
+ /// originating client.
+ RPL_TRYAGAIN = 263,
+
+ /// "<nickname> :No such nick/channel"
+ /// - Used to indicate the nickname parameter supplied to a
+ /// command is currently unused.
+ ERR_NOSUCHNICK = 401,
+ /// "<server name> :No such server"
+ /// - Used to indicate the server name given currently
+ /// does not exist.
+ ERR_NOSUCHSERVER = 402,
+ /// "<channel name> :No such channel"
+ /// - Used to indicate the given channel name is invalid.
+ ERR_NOSUCHCHANNEL = 403,
+ /// "<channel name> :Cannot send to channel"
+ /// - Sent to a user who is either (a) not on a channel
+ /// which is mode +n or (b) not a chanop (or mode +v) on
+ /// a channel which has mode +m set or where the user is
+ /// banned and is trying to send a PRIVMSG message to
+ /// that channel.
+ ERR_CANNOTSENDTOCHAN = 404,
+ /// "<channel name> :You have joined too many channels"
+ /// - Sent to a user when they have joined the maximum
+ /// number of allowed channels and they try to join
+ /// another channel.
+ ERR_TOOMANYCHANNELS = 405,
+ /// "<nickname> :There was no such nickname"
+ /// - Returned by WHOWAS to indicate there is no history
+ /// information for that nickname.
+ ERR_WASNOSUCHNICK = 406,
+ /// "<target> :<error code> recipients. <abort message>"
+ /// - Returned to a client which is attempting to send a
+ /// PRIVMSG/NOTICE using the user@host destination format
+ /// and for a user@host which has several occurrences.
+ /// - Returned to a client which trying to send a
+ /// PRIVMSG/NOTICE to too many recipients.
+ /// - Returned to a client which is attempting to JOIN a safe
+ /// channel using the shortname when there are more than one
+ /// such channel.
+ ERR_TOOMANYTARGETS = 407,
+ /// "<service name> :No such service"
+ /// - Returned to a client which is attempting to send a SQUERY
+ /// to a service which does not exist.
+ ERR_NOSUCHSERVICE = 408,
+ /// ":No origin specified"
+ /// - PING or PONG message missing the originator parameter.
+ ERR_NOORIGIN = 409,
+ /// ":No recipient given (<command>)"
+ ERR_NORECIPIENT = 411,
+ /// ":No text to send"
+ ERR_NOTEXTTOSEND = 412,
+ /// "<mask> :No toplevel domain specified"
+ ERR_NOTOPLEVEL = 413,
+ /// "<mask> :Wildcard in toplevel domain"
+ ERR_WILDTOPLEVEL = 414,
+ /// "<mask> :Bad Server/host mask"
+ /// - 412 - 415 are returned by PRIVMSG to indicate that
+ /// the message wasn't delivered for some reason.
+ /// ERR_NOTOPLEVEL and ERR_WILDTOPLEVEL are errors that
+ /// are returned when an invalid use of
+ /// "PRIVMSG $<server>" or "PRIVMSG #<host>" is attempted.
+ ERR_BADMASK = 415,
+ /// "<command> :Unknown command"
+ /// - Returned to a registered client to indicate that the
+ /// command sent is unknown by the server.
+ ERR_UNKNOWNCOMMAND = 421,
+ /// ":MOTD File is missing"
+ /// - Server's MOTD file could not be opened by the server.
+ ERR_NOMOTD = 422,
+ /// "<server> :No administrative info available"
+ /// - Returned by a server in response to an ADMIN message
+ /// when there is an error in finding the appropriate
+ /// information.
+ ERR_NOADMININFO = 423,
+ /// ":File error doing <file op> on <file>"
+ /// - Generic error message used to report a failed file
+ /// operation during the processing of a message.
+ ERR_FILEERROR = 424,
+ /// ":No nickname given"
+ /// - Returned when a nickname parameter expected for a
+ /// command and isn't found.
+ ERR_NONICKNAMEGIVEN = 431,
+ /// "<nick> :Erroneous nickname"
+ /// - Returned after receiving a NICK message which contains
+ /// characters which do not fall in the defined set. See
+ /// section 2.3.1 for details on valid nicknames.
+ ERR_ERRONEUSNICKNAME = 432,
+ /// "<nick> :Nickname is already in use"
+ /// - Returned when a NICK message is processed that results
+ /// in an attempt to change to a currently existing
+ /// nickname.
+ ERR_NICKNAMEINUSE = 433,
+ /// "<nick> :Nickname collision KILL from <user>@<host>"
+ /// - Returned by a server to a client when it detects a
+ /// nickname collision (registered of a NICK that
+ /// already exists by another server).
+ ERR_NICKCOLLISION = 436,
+ /// "<nick/channel> :Nick/channel is temporarily unavailable"
+ /// - Returned by a server to a user trying to join a channel
+ /// currently blocked by the channel delay mechanism.
+ /// - Returned by a server to a user trying to change nickname
+ /// when the desired nickname is blocked by the nick delay
+ /// mechanism.
+ ERR_UNAVAILRESOURCE = 437,
+ /// "<nick> <channel> :They aren't on that channel"
+ /// - Returned by the server to indicate that the target
+ /// user of the command is not on the given channel.
+ ERR_USERNOTINCHANNEL = 441,
+ /// "<channel> :You're not on that channel"
+ /// - Returned by the server whenever a client tries to
+ /// perform a channel affecting command for which the
+ /// client isn't a member.
+ ERR_NOTONCHANNEL = 442,
+ /// "<user> <channel> :is already on channel"
+ /// - Returned when a client tries to invite a user to a
+ /// channel they are already on.
+ ERR_USERONCHANNEL = 443,
+ /// "<user> :User not logged in"
+ /// - Returned by the summon after a SUMMON command for a
+ /// user was unable to be performed since they were not
+ /// logged in.
+ ERR_NOLOGIN = 444,
+ /// ":SUMMON has been disabled"
+ /// - Returned as a response to the SUMMON command. MUST be
+ /// returned by any server which doesn't implement it.
+ ERR_SUMMONDISABLED = 445,
+ /// ":USERS has been disabled"
+ /// - Returned as a response to the USERS command. MUST be
+ /// returned by any server which does not implement it.
+ ERR_USERSDISABLED = 446,
+ /// ":You have not registered"
+ /// - Returned by the server to indicate that the client
+ /// MUST be registered before the server will allow it
+ /// to be parsed in detail.
+ ERR_NOTREGISTERED = 451,
+ /// "<command> :Not enough parameters"
+ /// - Returned by the server by numerous commands to
+ /// indicate to the client that it didn't supply enough
+ /// parameters.
+ ERR_NEEDMOREPARAMS = 461,
+ /// ":Unauthorized command (already registered)"
+ /// - Returned by the server to any link which tries to
+ /// change part of the registered details (such as
+ /// password or user details from second USER message).
+ ERR_ALREADYREGISTERED = 462,
+ /// ":Your host isn't among the privileged"
+ /// - Returned to a client which attempts to register with
+ /// a server which does not been setup to allow
+ /// connections from the host the attempted connection
+ /// is tried.
+ ERR_NOPERMFORHOST = 463,
+ /// ":Password incorrect"
+ /// - Returned to indicate a failed attempt at registering
+ /// a connection for which a password was required and
+ /// was either not given or incorrect.
+ ERR_PASSWDMISMATCH = 464,
+ /// ":You are banned from this server"
+ /// - Returned after an attempt to connect and register
+ /// yourself with a server which has been setup to
+ /// explicitly deny connections to you.
+ ERR_YOUREBANNEDCREEP = 465,
+ /// - Sent by a server to a user to inform that access to the
+ /// server will soon be denied.
+ ERR_YOUWILLBEBANNED = 466,
+ /// "<channel> :Channel key already set"
+ ERR_KEYSET = 467,
+ /// "<channel> :Cannot join channel (+l)"
+ ERR_CHANNELISFULL = 471,
+ /// "<char> :is unknown mode char to me for <channel>"
+ ERR_UNKNOWNMODE = 472,
+ /// "<channel> :Cannot join channel (+i)"
+ ERR_INVITEONLYCHAN = 473,
+ /// "<channel> :Cannot join channel (+b)"
+ ERR_BANNEDFROMCHAN = 474,
+ /// "<channel> :Cannot join channel (+k)"
+ ERR_BADCHANNELKEY = 475,
+ /// "<channel> :Bad Channel Mask"
+ ERR_BADCHANMASK = 476,
+ /// "<channel> :Channel doesn't support modes"
+ ERR_NOCHANMODES = 477,
+ /// "<channel> <char> :Channel list is full"
+ ERR_BANLISTFULL = 478,
+ /// ":Permission Denied- You're not an IRC operator"
+ /// - Any command requiring operator privileges to operate
+ /// MUST return this error to indicate the attempt was
+ /// unsuccessful.
+ ERR_NOPRIVILEGES = 481,
+ /// "<channel> :You're not channel operator"
+ /// - Any command requiring 'chanop' privileges (such as
+ /// MODE messages) MUST return this error if the client
+ /// making the attempt is not a chanop on the specified
+ /// channel.
+ ERR_CHANOPRIVSNEEDED = 482,
+ /// ":You can't kill a server!"
+ /// - Any attempts to use the KILL command on a server
+ /// are to be refused and this error returned directly
+ /// to the client.
+ ERR_CANTKILLSERVER = 483,
+ /// ":Your connection is restricted!"
+ /// - Sent by the server to a user upon connection to indicate
+ /// the restricted nature of the connection (user mode "+r").
+ ERR_RESTRICTED = 484,
+ /// ":You're not the original channel operator"
+ /// - Any MODE requiring "channel creator" privileges MUST
+ /// return this error if the client making the attempt is not
+ /// a chanop on the specified channel.
+ ERR_UNIQOPPRIVSNEEDED = 485,
+ /// ":No O-lines for your host"
+ /// - If a client sends an OPER message and the server has
+ /// not been configured to allow connections from the
+ /// client's host as an operator, this error MUST be
+ /// returned.
+ ERR_NOOPERHOST = 491,
+ /// ":Unknown MODE flag"
+ /// - Returned by the server to indicate that a MODE
+ /// message was sent with a nickname parameter and that
+ /// the a mode flag sent was not recognized.
+ ERR_UMODEUNKNOWNFLAG = 501,
+ /// ":Cannot change mode for other users"
+ /// - Error sent to any user trying to view or change the
+ /// user mode for a user other than themselves.
+ ERR_USERSDONTMATCH = 502,
+}
+
+impl Response {
+ pub fn is_reply(&self) -> bool { let i = *self as uint; i >= 200 && i <= 399 }
+ pub fn is_error(&self) -> bool { let i = *self as uint; i >= 400 && i <= 599 }
+}
+
+pub fn join(v: Vec<String>, from: uint) -> String {
+ let mut msg = if v[from].chars().next().unwrap() == ':' {
+ v[from][][1..].to_owned()
+ } else { v[from].clone() };
+ for m in v.iter().skip(from + 1) {
+ msg.push_str(" ");
+ msg.push_str(m.trim_right());
+ }
+ msg
+}
+
+/*pub struct PrivMsg {
+ pub from: Ident,
+ pub to: String,
+ pub content: String
+}
+
+impl ParseResult for PrivMsg {
+ fn parse(message: Message) -> Option<PrivMsg> {
+ let from = Ident::parse(message.prefix.unwrap()[]);
+ let to = message.content[0].clone();
+ match from {
+ Some(from) => Some(PrivMsg {
+ from: from,
+ to: to,
+ content: join(message.content, 1)
+ }),
+ None => None
+ }
+ }
+}*/
diff --git a/src/server.rs b/src/server.rs
index 4c4479a..154e517 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -1,70 +1,120 @@
use std::io::{
BufferedReader,
TcpStream,
- IoError
+ IoError,
+ IoResult
};
-
-use std::sync::Arc;
-use std::sync::Mutex;
-
-use std::borrow::ToOwned;
+use std::mem;
use callback::Callback;
-use event;
-use event::Event;
+use message;
+use message::{ Command, Message };
+
+#[cfg(feature = "ssl")]
+use openssl::ssl::{ SslContext, SslMethod, SslStream };
-#[deriving(Show, PartialEq, Eq, Clone)]
+#[derive(Show, PartialEq, Eq, Clone)]
pub enum Failure {
AlreadyConnected,
NotConnected,
Io(IoError)
}
-#[deriving(Clone)]
-pub struct Server {
- stream: Arc<Mutex<Option<TcpStream>>>,
- pub events: Arc<Mutex<Callback<(Server, Event)>>>,
+/// Yes, I don't like the name either, but it's private, so...
+enum StreamKind {
+ Plain(TcpStream),
+ #[cfg(feature = "ssl")]
+ Ssl(SslStream)
+}
+
+impl Writer for StreamKind {
+ fn write(&mut self, buf: &[u8]) -> IoResult<()> {
+ match *self {
+ StreamKind::Plain(ref mut s) => s.write(buf),
+ #[cfg(feature = "ssl")]
+ StreamKind::Ssl(ref mut s) => s.write(buf)
+ }
+ }
}
-impl Server {
- pub fn new() -> Server {
+impl Reader for StreamKind {
+ fn read(&mut self, buf: &mut [u8]) -> IoResult<uint> {
+ match *self {
+ StreamKind::Plain(ref mut s) => s.read(buf),
+ #[cfg(feature = "ssl")]
+ StreamKind::Ssl(ref mut s) => s.read(buf)
+ }
+ }
+}
+
+pub struct Event<'a> {
+ pub server: &'a mut Server<'a>,
+ pub message: &'a Message
+}
+
+pub struct Server<'a> {
+ stream: Option<StreamKind>,
+ pub events: Callback<Event<'a>>,
+}
+
+impl<'a> Server<'a> {
+ pub fn new() -> Server<'a> {
Server {
- stream: Arc::new(Mutex::new(None)),
+ stream: None,
events: {
let mut c = Callback::new();
- c.register(&(Server::handle_event as fn((Server,Event))));
- Arc::new(Mutex::new(c))
+ c.register(Server::handle_event);
+ c
}
}
}
- fn handle_event(arg: (Server, Event)) {
- let (mut server, event) = arg;
- match event.command[] {
- event::PING => {
- server.sendraw(format!("PONG {}", event.content).as_slice(), true).unwrap();
+ fn handle_event(e: &mut Event) {
+ match e.message.command {
+ Command::PING => {
+ e.server.sendraw(format!("PONG :{}", message::join(e.message.content.clone(), 0)).as_slice(), true).unwrap();
}
_ => ()
}
}
pub fn connect(&mut self, host: String, port: u16) -> Result<(), Failure> {
- let mut s = self.stream.lock();
+ let s = &mut self.stream;
match *s { Some(_) => return Err(Failure::AlreadyConnected), _ => () };
*s = match TcpStream::connect((host.as_slice(), port)) {
+ Ok(tcp) => Some(StreamKind::Plain(tcp)),
+ Err(e) => return Err(Failure::Io(e))
+ };
+
+ Ok(())
+ }
+
+ #[cfg(feature = "ssl")]
+ pub fn connect_ssl(&mut self, host: String, port: u16) -> Result<(), Failure> {
+ let mut s = self.stream.lock();
+ match *s { Some(_) => return Err(Failure::AlreadyConnected), _ => () };
+ let tcp_stream = match TcpStream::connect((host.as_slice(), port)) {
Ok(tcp) => Some(tcp),
Err(e) => return Err(Failure::Io(e))
};
+ let ssl = SslContext::new(SslMethod::Tlsv1);
+ let ssl_stream = SslStream::new(&ssl, tcp_stream);
+ *s = ssl_stream;
+
Ok(())
+
}
#[inline]
fn sendraw(&mut self, s: &str, newline: bool) -> Result<(), Failure> {
- info!("OUT: {}", s);
- let mut locked_stream = self.stream.lock();
- if locked_stream.is_some() {
- locked_stream.as_mut().map(|stream| {
+ info!(">> {}", s);
+ if cfg!(not(ndebug)) && s.len() > 510 {
+ panic!("Message too long, kitties will die if this runs in release mode. Msg: {}", s)
+ }
+ let stream = &mut self.stream;
+ if stream.is_some() {
+ stream.as_mut().map(|stream| {
match stream.write_str(s) {
Ok(_) => match { if newline { stream.write_str("\r\n") } else { Ok(()) } } {
Ok(_) => match stream.flush() {
@@ -81,10 +131,18 @@ impl Server {
}
}
+ pub fn send(&mut self, msg: message::Message) -> Result<(), Failure> {
+ self.sendraw(msg.format()[], true)
+ }
+
pub fn join(&mut self, channel: &str) -> Result<(), Failure> {
self.sendraw(format!("JOIN {}", channel).as_slice(), true)
}
+ pub fn part(&mut self, channel: &str) -> Result<(), Failure> {
+ self.sendraw(format!("PART {}", channel).as_slice(), true)
+ }
+
pub fn nick(&mut self, nick: &str) -> Result<(), Failure> {
self.sendraw(format!("NICK {}", nick).as_slice(), true)
}
@@ -103,37 +161,29 @@ impl Server {
pub fn listen(&mut self) -> Result<(), Failure> {
let stream = {
- let lock = self.stream.lock();
- match *lock {
+ match self.stream {
Some(ref s) => s.clone(),
None => return Err(Failure::NotConnected)
}
};
- let mut reader = BufferedReader::new(stream);
+ let mut reader = BufferedReader::new(match *stream {
+ StreamKind::Plain(ref s) => (*s).clone(),
+ #[cfg(feature = "ssl")]
+ StreamKind::Ssl(ref s) => (*s).clone()
+ });
+
loop {
let line = reader.read_line().unwrap();
- let mut parts = line.as_slice().split(' ').collect::<Vec<&str>>();
-
- info!("IN: {}", line);
-
- if parts.len() == 0 {
- continue;
+ info!("<< {}", line);
+
+ if let Some(msg) = Message::parse(line[]) {
+ let mut e = self.events;
+ e.fire(&mut Event {
+ server: self,
+ message: &msg
+ });
}
-
- // if message has a prefix
- let prefix = if parts[0].chars().next().unwrap() == ':' {
- parts.remove(0).unwrap()
- } else { "" };
-
- let cmd = parts.remove(0).unwrap();
- let event = Event {
- prefix: prefix.to_owned(),
- command: cmd.to_owned(),
- content: parts.into_iter().map(|s| s.to_owned()).collect()
- };
-
- self.events.lock().fire(&(self.clone(), event));
}
}
}