Merge branch 'check-for-alias-in-file'

3 files changed, 171 insertions, 21 deletions

diff --git a/src/main.rs b/src/main.rs
index 49c30d3..77f230a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,6 @@
-use std::io::{self, BufRead};
+use std::io::{self, BufRead, BufReader, Write};
+use std::fs::{File, OpenOptions};
+use std::path::Path;
 
 #[cfg(test)]
 mod tests;
@@ -47,37 +49,81 @@ impl Alias {
             format!("alias {} {} {}", self.alias, self.name, self.email)
         }
     }
+
+    fn write_to_file<P: AsRef<Path>>(&self, file: P) -> Result<(), io::Error> {
+        let mut f = try!(OpenOptions::new().append(true).open(file));
+        try!(f.write_all(format!("{}\n", self.to_string()).as_bytes()));
+        Ok(())
+    }
+
+    fn update_alias_id(&mut self, similar_aliases: Vec<String>) {
+        if !similar_aliases.is_empty() {
+            self.alias = format!("{}-{}", self.alias, similar_aliases.len() + 1);
+        }
+    }
 }
 
 fn handle_alias(s: &str) {
-    let alias = build_alias(s);
 }
 
-fn build_alias(s: &str) -> String {
-    let mut split: Vec<&str> = s.split_whitespace().collect();
+#[derive(Debug)]
+enum AliasSearchError {
+    NotFound,
+    EmailExists,
+    Io(io::Error),
+}
 
-    // Remove "From: "
-    split.remove(0);
+// impl fmt::Display for AliasSearchError {}
+// impl error::Error for AliasSearchError {}
 
-    let mut alias_line = String::from("alias ");
-    let mut alias = String::new();
+impl From<io::Error> for AliasSearchError {
+    fn from(err: io::Error) -> AliasSearchError {
+        AliasSearchError::Io(err)
+    }
+}
 
-    if split.len() == 1 {
-        alias = format!("{} ", split[0].to_lowercase());
-    } else if split.len() == 2 {
-        alias = format!("{} ", split[0].to_lowercase());
-    } else if split.len() > 2 {
-        alias = format!("{}-{} ", split[split.len() - 2], split[0]).to_lowercase();
+#[cfg(test)]
+impl PartialEq<AliasSearchError> for AliasSearchError {
+    fn eq(&self, other: &AliasSearchError) -> bool {
+        match *self {
+            AliasSearchError::NotFound => match *other {
+                AliasSearchError::NotFound => true,
+                _ => false,
+            },
+            AliasSearchError::EmailExists => match *other {
+                AliasSearchError::EmailExists => true,
+                _ => false,
+            },
+            AliasSearchError::Io(_) => match *other {
+                AliasSearchError::Io(_) => true,
+                _ => false,
+            },
+        }
     }
+}
+
+fn find_alias_in_file(alias: &Alias, file: &str) -> Result<Vec<String>, AliasSearchError> {
+    let mut matches = Vec::new();
+    let f = try!(File::open(file));
+    let file = BufReader::new(&f);
+    for line in file.lines() {
+        let line = try!(line);
+        let split: Vec<&str> = line.split_whitespace().collect();
 
-    alias = alias.replace(',', "");
-    alias = alias.replace('\'', "");
-    alias = alias.replace('"', "");
+        if line.contains(&alias.email) {
+            return Err(AliasSearchError::EmailExists)
+        }
 
-    alias_line.push_str(&alias);
-    alias_line.push_str(&split.join(" "));
+        if split[1].starts_with(&alias.alias) {
+            matches.push(split[1].to_owned());
+        }
+    }
 
-    alias_line
+    if matches.is_empty() {
+        Err(AliasSearchError::NotFound)
+    } else {
+        Ok(matches)
+    }
 }
 
 fn main() {
diff --git a/src/tests.rs b/src/tests.rs
index c12bab1..2a6e264 100644
--- a/src/tests.rs
+++ b/src/tests.rs
@@ -1,4 +1,7 @@
-use super::Alias;
+use std::fs::{self, File};
+use std::io::Read;
+
+use super::{Alias, AliasSearchError, find_alias_in_file};
 
 #[test]
 fn new_alias_with_only_email() {
@@ -39,3 +42,102 @@ fn new_alias_with_special_characters() {
         Alias::new("From: \"O'Strulson, Celty\" <celty@dollars.co>").to_string()
     );
 }
+
+
+#[test]
+fn find_alias_in_file_email_already_exists() {
+    assert_eq!(
+        Err(AliasSearchError::EmailExists),
+        find_alias_in_file(
+            &Alias {
+                alias: "farnsworth-hubert".to_owned(),
+                name: "Hubert Farnsworth".to_owned(),
+                email: "<professor@planetexpress.com>".to_owned()
+            },
+            "./testdata/aliases"
+        )
+    );
+}
+
+#[test]
+fn find_alias_in_file_alias_is_new() {
+    assert_eq!(
+        Err(AliasSearchError::NotFound),
+        find_alias_in_file(
+            &Alias {
+                alias: "fry-philip".to_owned(),
+                name: "Philip Fry".to_owned(),
+                email: "<fry@planetexpress.com>".to_owned()
+            },
+            "./testdata/aliases"
+        )
+    );
+}
+
+#[test]
+fn find_alias_in_file_finds_a_match() {
+    assert_eq!(
+        Ok(vec![
+            "farnsworth-hubert".to_owned(),
+            "farnsworth-hubert-2".to_owned()
+        ]),
+        find_alias_in_file(
+            &Alias {
+                alias: "farnsworth-hubert".to_owned(),
+                name: "Hubert Farnsworth".to_owned(),
+                email: "<goodnewseveryone@planetexpress.com>".to_owned()
+            },
+            "./testdata/aliases"
+        )
+    );
+}
+
+
+const update_alias_id_alias_identifier: &'static str = "hooves-derpy";
+
+fn update_alias_id_sample_alias() -> Alias {
+    Alias {
+        alias: update_alias_id_alias_identifier.to_owned(),
+        name: "Derpy Hooves".to_owned(),
+        email: "derpyhooves@postmaster.pv".to_owned()
+    }
+}
+
+#[test]
+fn update_alias_id_does_nothing_given_an_empty_vector() {
+    let mut alias = update_alias_id_sample_alias();
+    alias.update_alias_id(vec![]);
+
+    assert_eq!(update_alias_id_alias_identifier, &alias.alias);
+}
+
+#[test]
+fn update_alias_id_increments_alias() {
+    let mut alias = update_alias_id_sample_alias();
+    alias.update_alias_id(vec![
+        "hooves-derpy".to_owned(),
+        "hooves-derpy-2".to_owned(),
+        "hooves-derpy-3".to_owned(),
+        "hooves-derpy-4".to_owned()
+    ]);
+
+    assert_eq!("hooves-derpy-5", &alias.alias);
+}
+
+
+#[test]
+fn alias_write_to_file_must_write_given_alias_to_file() {
+    let alias = update_alias_id_sample_alias();
+
+    let test_file = "./testdata/write_to_file";
+    fs::copy("./testdata/aliases", test_file).unwrap();
+    alias.write_to_file(test_file).unwrap();
+
+    let mut f = File::open(test_file).unwrap();
+    let mut file_contents = String::new();
+    f.read_to_string(&mut file_contents).unwrap();
+    let file_contents: Vec<&str> = file_contents.split('\n').collect();
+    fs::remove_file(test_file).unwrap();
+
+    assert_eq!(alias.to_string(), file_contents[file_contents.len() - 2]);
+}
diff --git a/testdata/aliases b/testdata/aliases
new file mode 100644
index 0000000..d9c167a
--- /dev/null
+++ b/testdata/aliases
@@ -0,0 +1,2 @@
+alias farnsworth-hubert Hubert Farnsworth <professor@planetexpress.com>
+alias farnsworth-hubert-2 Hubert Farnsworth <davincifan@planetexpress.com>