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 129
extern crate exec; pub mod error; use std::io::BufRead; pub fn load<P: AsRef<std::path::Path>>(path: P) -> Result<(), error::Error> { for (k, v) in try!(build_env(path)) { std::env::set_var(k, v); } return Ok(()); } pub fn build_env<P: AsRef<std::path::Path>>(path: P) -> Result<Vec<(String, String)>, error::Error> { match std::fs::File::open(path) { Ok(f) => parse(f), Err(e) => Err(error::Error::from_io_error(e)), } } pub fn parse<R: std::io::Read>(file: R) -> Result<Vec<(String, String)>, error::Error> { let mut env = vec![]; let reader = std::io::BufReader::new(file); for line_result in reader.lines() { match line_result { Ok(line) => { if let Some((k, v)) = try!(parse_line(&line)) { env.push((k.to_owned(), v.to_owned())); } } Err(e) => { return Err(error::Error::from_io_error(e)); } } } return Ok(env); } fn parse_line(line: &str) -> Result<Option<(&str, &str)>, error::Error> { let line = line.trim_left(); if line.is_empty() || line.starts_with("#") { return Ok(None); } else { let xs: Vec<&str> = line.splitn(2, "=").collect(); if xs.len() == 2 { let key = xs[0]; let val = xs[1]; if key.find(char::is_whitespace).is_some() { return Err(error::Error::new("key cannot contain whitespaces")); } else { return Ok(Some((key, val))); } } else { return Err(error::Error::new("key and value must be separated by `=`")); } } } #[cfg(test)] mod tests { use std::error::Error; use super::parse; #[test] fn parse_denv_file() { let result = parse("x=123\ny=345\n".as_bytes()); assert!(result.is_ok()); assert_eq!(vec![("x".to_owned(), "123".to_owned()), ("y".to_owned(), "345".to_owned())], result.unwrap()); } #[test] fn ignore_comment() { let result = parse("x=123\n # comment\ny=345\n".as_bytes()); assert!(result.is_ok()); assert_eq!(vec![("x".to_owned(), "123".to_owned()), ("y".to_owned(), "345".to_owned())], result.unwrap()); } #[test] fn ignore_empty() { let result = parse("x=123\n\n \n\ny=345\n".as_bytes()); assert!(result.is_ok()); assert_eq!(vec![("x".to_owned(), "123".to_owned()), ("y".to_owned(), "345".to_owned())], result.unwrap()); } #[test] fn hash_in_the_middle() { let result = parse("x=123\ny=345# comment\n".as_bytes()); assert!(result.is_ok()); assert_eq!(vec![("x".to_owned(), "123".to_owned()), ("y".to_owned(), "345# comment".to_owned())], result.unwrap()); } #[test] fn equal_in_the_middle() { let result = parse("x=12=3\ny=345\n".as_bytes()); assert!(result.is_ok()); assert_eq!(vec![("x".to_owned(), "12=3".to_owned()), ("y".to_owned(), "345".to_owned())], result.unwrap()); } #[test] fn include_whitespaces() { let result = parse("x=a text with whitespaces\ny=345\n".as_bytes()); assert!(result.is_ok()); assert_eq!(vec![("x".to_owned(), "a text with whitespaces".to_owned()), ("y".to_owned(), "345".to_owned())], result.unwrap()); } #[test] fn malformed_line() { let result = parse("x=123\ny".as_bytes()); assert!(result.is_err()); assert!(result.unwrap_err().cause().is_none()); } #[test] fn malformed_key() { let result = parse("xx x=123\ny".as_bytes()); assert!(result.is_err()); assert!(result.unwrap_err().cause().is_none()); } }