81 lines
2.5 KiB
Rust
81 lines
2.5 KiB
Rust
use std::cell::{RefCell, RefMut};
|
|
use std::io::{BufferedReader, BufferedWriter, IoResult, TcpStream, Writer};
|
|
use data::{IrcReader, IrcWriter, Message};
|
|
|
|
pub struct Connection<T, U> where T: IrcWriter, U: IrcReader {
|
|
writer: RefCell<T>,
|
|
reader: RefCell<U>,
|
|
}
|
|
|
|
impl Connection<BufferedWriter<TcpStream>, BufferedReader<TcpStream>> {
|
|
pub fn connect(host: &str, port: u16) -> IoResult<Connection<BufferedWriter<TcpStream>, BufferedReader<TcpStream>>> {
|
|
let socket = try!(TcpStream::connect(host, port));
|
|
Connection::new(BufferedWriter::new(socket.clone()), BufferedReader::new(socket.clone()))
|
|
}
|
|
}
|
|
|
|
impl<T, U> Connection<T, U> where T: IrcWriter, U: IrcReader {
|
|
pub fn new(writer: T, reader: U) -> IoResult<Connection<T, U>> {
|
|
Ok(Connection {
|
|
writer: RefCell::new(writer),
|
|
reader: RefCell::new(reader),
|
|
})
|
|
}
|
|
|
|
fn send_internal(&self, msg: &str) -> IoResult<()> {
|
|
let mut send = self.writer.borrow_mut();
|
|
try!(send.write_str(msg));
|
|
send.flush()
|
|
}
|
|
|
|
pub fn send(&self, msg: Message) -> IoResult<()> {
|
|
let mut send = msg.command.to_string();
|
|
if msg.args.init().len() > 0 {
|
|
send.push_str(" ");
|
|
send.push_str(msg.args.init().connect(" ").as_slice());
|
|
}
|
|
send.push_str(" :");
|
|
send.push_str(*msg.args.last().unwrap());
|
|
send.push_str("\r\n");
|
|
self.send_internal(send.as_slice())
|
|
}
|
|
|
|
pub fn writer<'a>(&'a self) -> RefMut<'a, T> {
|
|
self.writer.borrow_mut()
|
|
}
|
|
|
|
pub fn reader<'a>(&'a self) -> RefMut<'a, U> {
|
|
self.reader.borrow_mut()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::Connection;
|
|
use std::io::MemWriter;
|
|
use std::io::util::NullReader;
|
|
use data::Message;
|
|
|
|
#[test]
|
|
fn new_connection() {
|
|
let w = MemWriter::new();
|
|
assert!(Connection::new(w, NullReader).is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn send_internal() {
|
|
let w = MemWriter::new();
|
|
let c = Connection::new(w, NullReader).unwrap();
|
|
c.send_internal("string of text").unwrap();
|
|
assert_eq!(c.writer().deref_mut().get_ref(), "string of text".as_bytes());
|
|
}
|
|
|
|
#[test]
|
|
fn send() {
|
|
let w = MemWriter::new();
|
|
let c = Connection::new(w, NullReader).unwrap();
|
|
let args = ["flare.to.ca.fyrechat.net"];
|
|
c.send(Message::new(None, "PING", args)).unwrap();
|
|
assert_eq!(c.writer().deref_mut().get_ref(), "PING :flare.to.ca.fyrechat.net\r\n".as_bytes());
|
|
}
|
|
}
|