Mirrors/rink-rs
2
0
Fork 0
mirror of https://github.com/tiffany352/rink-rs synced 2024-11-10 13:44:15 +00:00
Code Issues Projects Releases Packages Wiki Activity

Multi-server support in IRC bot

Browse source
This commit is contained in:
Tiffany Bennett 2016-08-08 22:21:07 -04:00
parent e692e623a5
commit 4a142003af
2 changed files with 54 additions and 35 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Cargo.toml
View file

@ -11,11 +11,13 @@ keywords = ["unit", "math", "conversion", "cli", "tool"]
[features]
default = ["rustyline"]
ircbot = ["irc", "glob"]
[dependencies]
rust-gmp = "0.3.2"
rustyline = { version = "0.2.3", optional = true }
irc = { version = "0.11.3", optional = true }
glob = { version = "0.2.11", optional = true }
[[bin]]
name = "rink"

87
src/bin/rink-irc.rs
View file

@ -1,52 +1,69 @@
#[cfg(feature = "irc")]
#[cfg(feature = "ircbot")]
extern crate irc;
#[cfg(feature = "irc")]
#[cfg(feature = "ircbot")]
extern crate glob;
#[cfg(feature = "ircbot")]
extern crate rink;
#[cfg(feature = "irc")]
#[cfg(feature = "ircbot")]
fn main() {
use irc::client::prelude::*;
use rink::*;
use glob::glob;
use std::thread;
let mut ctx = load().unwrap();
ctx.short_output = true;
let server = IrcServer::new("config.json").unwrap();
server.identify().unwrap();
let nick = server.config().nickname.clone().unwrap();
let mut prefix = nick.clone();
prefix.push(':');
for message in server.iter() {
if let Ok(Message { command: Command::PRIVMSG(ref chan, ref message_str), ..}) = message {
if message_str.starts_with(&*prefix) {
let reply_to = if &*chan == &*nick {
message.as_ref().unwrap().source_nickname().unwrap()
} else {
&*chan
};
let line = message_str[prefix.len()..].trim();
let reply = match one_line(&mut ctx, line) {
Ok(v) => v,
Err(e) => e
};
let mut i = 0;
for line in reply.lines() {
if line.trim().len() > 0 {
server.send(Command::NOTICE(reply_to.to_owned(), line.to_owned())).unwrap();
i += 1;
}
// cut off early
if i > 4 {
break;
fn run(config: &str) {
let mut ctx = load().unwrap();
ctx.short_output = true;
let server = IrcServer::new(config).unwrap();
server.identify().unwrap();
let nick = server.config().nickname.clone().unwrap();
let mut prefix = nick.clone();
prefix.push(':');
for message in server.iter() {
if let Ok(Message { command: Command::PRIVMSG(ref chan, ref message_str), ..}) = message {
if message_str.starts_with(&*prefix) {
let reply_to = if &*chan == &*nick {
message.as_ref().unwrap().source_nickname().unwrap()
} else {
&*chan
};
let line = message_str[prefix.len()..].trim();
let reply = match one_line(&mut ctx, line) {
Ok(v) => v,
Err(e) => e
};
let mut i = 0;
for line in reply.lines() {
if line.trim().len() > 0 {
server.send(Command::NOTICE(reply_to.to_owned(), line.to_owned())).unwrap();
i += 1;
}
// cut off early
if i > 4 {
break;
}
}
}
} else if let Err(e) = message {
println!("{}", e);
}
} else if let Err(e) = message {
println!("{}", e);
}
}
let mut threads = vec![];
for config in glob("servers/*.json").expect("Glob failed") {
match config {
Ok(config) => threads.push(thread::spawn(move || run(config.to_str().unwrap()))),
Err(e) => println!("{:?}", e)
}
}
for thread in threads {
thread.join().unwrap()
}
}
#[cfg(not(feature = "irc"))]
#[cfg(not(feature = "ircbot"))]
fn main() {
println!("Rink was not compiled with IRC support.");
}