Properly read username and hostname

username is read by using libc::getuid, and checking /etc/passwd for the
uid and getting the username from there

hostname is read from /proc/sys/kernel/hostname

1 files changed, 6 insertions, 3 deletions

diff --git a/src/main.rs b/src/main.rs
index 0581b13..403fd33 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,10 +7,13 @@ mod prompt;
 fn main() {
     match std::fs::read_to_string("/etc/motd") {
         Ok(motd) => println!("{}", motd),
-        Err(_) => {},
+        Err(_) => {}
     }
 
-    let prompt = prompt::Prompt::new();
+    let prompt = match prompt::Prompt::new() {
+        Ok(p) => p,
+        Err(e) => panic!(e),
+    };
     let mut cwd = match std::env::current_dir() {
         Ok(p) => p,
         Err(e) => panic!(e),
@@ -81,7 +84,7 @@ fn main() {
                         if cfg!(debug_assertions) {
                             println!("exit code [{}]", code);
                         }
-                    },
+                    }
                     None => println!("Process termed by signal"),
                 },
                 Err(e) => eprintln!("error waiting on final command: {}", e),