From 74d3b10a01321b34bd5e26ef456d6b4d201b062e Mon Sep 17 00:00:00 2001
From: NerdyPepper <akshayoppiliappan@gmail.com>
Date: Fri, 12 Jul 2019 14:54:44 +0530
Subject: improve output in non existent directories

---
 src/main.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 0603cba..3096635 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -4,6 +4,7 @@ mod vcs;
 mod venv;
 
 use clap::{Arg, App};
+use colored::*;
 
 fn main() {
     let matches = App::new("Pista")
@@ -29,7 +30,10 @@ fn main() {
 }
 
 fn pista(zsh: bool) -> String {
-    let cwd = cwd::cwd();
+    let cwd = match cwd::cwd() {
+        Some(c) => c,
+        None => "[directory does not exist]".color("red")
+    };
     let (branch, status) = vcs::vcs_status().unwrap_or(("".into(), "".into()));
     let venv = venv::get_name();
     let prompt_char = prompt_char::get_char();
@@ -53,7 +57,10 @@ fn pista(zsh: bool) -> String {
 }
 
 fn pista_minimal(zsh: bool) -> String {
-    let cwd = cwd::cwd();
+    let cwd = match cwd::cwd() {
+        Some(c) => c,
+        None => "[directory does not exist]".color("red")
+    };
     let vcs_tuple = vcs::vcs_status();
     let mut vcs_component = String::new();
     if let Some((branch, status)) = vcs_tuple {
-- 
cgit v1.2.3