DGNum/rust-irc
6
0
Fork 0
Code Issues 1 Pull requests 3 Projects Releases Packages Wiki Activity Actions

Added load_utf8 to Config and load takes a Path.

Browse source
This commit is contained in:
Aaron Weiss 2014-11-02 13:47:22 -05:00
parent f8a6987fcf
commit 0ad276fdae
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/bot.rs
View file

@ -14,7 +14,7 @@ pub struct IrcBot<'a, T, U> where T: IrcWriter, U: IrcReader {
impl<'a> IrcBot<'a, BufferedWriter<TcpStream>, BufferedReader<TcpStream>> {
pub fn new(process: |&IrcBot<BufferedWriter<TcpStream>, BufferedReader<TcpStream>>, &str, &str, &[&str]|:'a -> IoResult<()>) -> IoResult<IrcBot<'a, BufferedWriter<TcpStream>, BufferedReader<TcpStream>>> {
let config = try!(Config::load("config.json"));
let config = try!(Config::load_utf8("config.json"));
let conn = try!(Connection::connect(config.server[], config.port));
Ok(IrcBot {
conn: conn,
@ -123,7 +123,7 @@ impl<'a, T, U> IrcBot<'a, T, U> where T: IrcWriter, U: IrcReader {
pub fn from_connection(conn: Connection<T, U>, process: |&IrcBot<T, U>, &str, &str, &[&str]|:'a -> IoResult<()>) -> IoResult<IrcBot<'a, T, U>> {
Ok(IrcBot {
conn: conn,
config: try!(Config::load("config.json")),
config: try!(Config::load_utf8("config.json")),
process: RefCell::new(process),
chanlists: RefCell::new(HashMap::new()),
})

12
src/data.rs
View file

@ -86,8 +86,8 @@ pub struct Config {
}
impl Config {
pub fn load(path: &str) -> IoResult<Config> {
let mut file = try!(File::open(&Path::new(path)));
pub fn load(path: Path) -> IoResult<Config> {
let mut file = try!(File::open(&path));
let data = try!(file.read_to_string());
decode(data[]).map_err(|e| IoError {
kind: InvalidInput,
@ -96,6 +96,10 @@ impl Config {
})
}
pub fn load_utf8(path: &str) -> IoResult<Config> {
Config::load(Path::new(path))
}
pub fn is_owner(&self, nickname: &str) -> bool {
self.owners[].contains(&String::from_str(nickname))
}
@ -119,12 +123,12 @@ mod test {
#[test]
fn load_config() {
assert!(Config::load("config.json").is_ok());
assert!(Config::load_utf8("config.json").is_ok());
}
#[test]
fn is_owner() {
let cfg = Config::load("config.json").unwrap();
let cfg = Config::load_utf8("config.json").unwrap();
assert!(cfg.is_owner("test"));
assert!(!cfg.is_owner("test2"));
}