extract compact logic into separate function

Author: tje

src/main.rs

@@ -7,12 +7,42 @@ extern crate dirs;
 
 fn main() -> std::io::Result<()> {
     let path = get_cwd()?;
+
+    print!("{}", compact_path(path));
+    Ok(())
+}
+
+/// Gets the current working directory, or parsed from first CLI argument if
+/// provided.
+fn get_cwd() -> std::io::Result<PathBuf> {
+    let argv: Vec<String> = env::args().collect();
+    let path = match argv.get(1) {
+        Some(p) => fs::canonicalize(PathBuf::from(p))
+            .or(Ok(PathBuf::from(p))),
+        _ => env::current_dir(),
+    };
+    path
+}
+
+/// Gets the home directory.
+fn get_home() -> PathBuf {
+    let home = match dirs::home_dir() {
+        Some(p) => p,
+        None => Path::new("").to_path_buf(),
+    };
+    home
+}
+
+/// Compacts a provided PathBuf and returns it as a string
+fn compact_path(path: PathBuf) -> String {
     let home = get_home();
 
+    let mut output = String::new();
+
     // If the path includes the home directory, strip it out and print "~"
     let np = match path.strip_prefix(format!("{}", home.display())) {
         Ok(p) => {
-            print!("~/");
+            output.push_str(&"~/");
             p.to_path_buf()
         },
         Err(_) => path,
@@ -30,13 +60,13 @@ fn main() -> std::io::Result<()> {
     let mut first: bool = false;
     for frag in components.iter() {
         if first {
-            print!("/");
+            output.push_str(&"/");
         }
         first = true;
 
         // Just print the full fragment if it's already short
         if frag.len() <= 4 {
-            print!("{}", frag);
+            output.push_str(frag);
             continue;
         }
 
@@ -50,35 +80,14 @@ fn main() -> std::io::Result<()> {
 
         // If there's only one word, just print the first 3 characters
         if parts.len() == 1 {
-            print!("{}", &frag[..3]);
+            output.push_str(&frag[..3]);
             continue;
         }
 
         // Print the first character of each part
         for part in parts {
-            print!("{}", &part[..1]);
+            output.push_str(&part[..1]);
         }
     }
-    Ok(())
-}
-
-/// Gets the current working directory, or parsed from first CLI argument if
-/// provided.
-fn get_cwd() -> std::io::Result<PathBuf> {
-    let argv: Vec<String> = env::args().collect();
-    let path = match argv.get(1) {
-        Some(p) => fs::canonicalize(PathBuf::from(p))
-            .or(Ok(PathBuf::from(p))),
-        _ => env::current_dir(),
-    };
-    path
-}
-
-/// Gets the home directory.
-fn get_home() -> PathBuf {
-    let home = match dirs::home_dir() {
-        Some(p) => p,
-        None => Path::new("").to_path_buf(),
-    };
-    home
+    output
 }