build_alias: Remove [,'"] characters

Make the function a bit more DRY by taking the `push_str` calls out of
the `if` block.

Add a new test and some code to remove commas and quotes from aliases.

1 files changed, 11 insertions, 6 deletions

diff --git a/src/main.rs b/src/main.rs
index 2a0a471..1f3ff29 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -16,18 +16,23 @@ fn build_alias(s: &str) -> String {
     split.remove(0);
 
     let mut alias_line = String::from("alias ");
+    let mut alias = String::new();
 
     if split.len() == 1 {
-        alias_line.push_str(&format!("{} ", split[0].to_lowercase()));
-        alias_line.push_str(&split.join(" "))
+        alias = format!("{} ", split[0].to_lowercase());
     } else if split.len() == 2 {
-        alias_line.push_str(&format!("{} ", split[0].to_lowercase()));
-        alias_line.push_str(&split.join(" "));
+        alias = format!("{} ", split[0].to_lowercase());
     } else if split.len() > 2 {
-        alias_line.push_str(&format!("{}-{} ", split[split.len() - 2], split[0]).to_lowercase());
-        alias_line.push_str(&split.join(" "));
+        alias = format!("{}-{} ", split[split.len() - 2], split[0]).to_lowercase();
     }
 
+    alias = alias.replace(',', "");
+    alias = alias.replace('\'', "");
+    alias = alias.replace('"', "");
+
+    alias_line.push_str(&alias);
+    alias_line.push_str(&split.join(" "));
+
     alias_line
 }