1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
use std::{io, str};
use constants::{CommandMessage, Nick, Pass, User};
use message::Message;
pub struct ClientBuilder {
nick: String,
pass: Option<String>,
realname: String,
username: String,
hostname: Option<String>,
servername: String,
port: u16,
}
pub struct Client {
conn: io::BufferedStream<io::TcpStream>,
}
impl ClientBuilder {
pub fn new (nick: &str, servername: &str) -> ClientBuilder {
ClientBuilder {
nick: nick.to_string(),
pass: None,
realname: nick.to_string(),
username: nick.to_string(),
hostname: None,
servername: servername.to_string(),
port: 6667,
}
}
pub fn set_pass (&mut self, pass: &str) -> &mut ClientBuilder {
self.pass = Some(pass.to_string());
self
}
pub fn set_username (&mut self, username: &str) -> &mut ClientBuilder {
self.username = username.to_string();
self
}
pub fn set_realname (&mut self, realname: &str) -> &mut ClientBuilder {
self.realname = realname.to_string();
self
}
pub fn set_hostname (&mut self, hostname: &str) -> &mut ClientBuilder {
self.hostname = Some(hostname.to_string());
self
}
pub fn set_port (&mut self, port: u16) -> &mut ClientBuilder {
self.port = port;
self
}
pub fn connect (&mut self) -> Client {
let mut client = self.connect_raw();
match self.pass {
Some(ref pass) => {
client.write(
Message::new(
None,
CommandMessage(Pass),
vec![pass.clone()],
)
);
},
None => {},
}
client.write(
Message::new(
None,
CommandMessage(Nick),
vec![self.nick.clone()],
)
);
let hostname = match &self.hostname {
&Some(ref host) => host.clone(),
&None => {
// XXX get the name of the local end of the connection
"localhost".to_string()
},
};
client.write(
Message::new(
None,
CommandMessage(User),
vec![
self.username.clone(),
hostname,
self.servername.clone(),
self.realname.clone(),
],
)
);
client
}
pub fn connect_raw (&mut self) -> Client {
let mut stream = io::TcpStream::connect(self.servername.as_slice(), self.port);
Client { conn: io::BufferedStream::new(stream.unwrap()) }
}
}
impl Client {
pub fn read (&mut self) -> Message {
// \n isn't valid inside a message, so this should be fine. if the \n
// we find isn't preceded by a \r, this will be caught by the message
// parser.
let buf = self.conn.read_until(b'\n');
// XXX handle different encodings
// XXX proper error handling
Message::parse(str::from_utf8(buf.unwrap().as_slice()).unwrap()).unwrap()
}
pub fn write (&mut self, msg: Message) {
msg.write_protocol_string(&mut self.conn);
}
}
|
