add flag to disable vcsv0.1.5

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 1a77759..6ccca7b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,6 +3,8 @@ mod prompt_char;
 mod vcs;
 mod venv;
 
+use std::env;
+
 use clap::{App, Arg};
 use colored::*;
 
@@ -36,7 +38,10 @@ fn pista(zsh: bool) -> String {
         Some(c) => c,
         None => "[directory does not exist]".color("red"),
     };
-    let (branch, status) = vcs::vcs_status().unwrap_or(("".into(), "".into()));
+    let (branch, status) = match env::var("DISABLE_VCS").unwrap_or("0".into()).as_ref() {
+        "0" => vcs::vcs_status().unwrap_or(("".into(), "".into())),
+        _ => ("".into(), "".into())
+    };
     let venv = venv::get_name();
     let prompt_char = prompt_char::get_char();
     if zsh {