detect nix-shell env

1 files changed, 13 insertions, 0 deletions

diff --git a/src/venv.rs b/src/venv.rs
index 7906abc..da6830c 100644
--- a/src/venv.rs
+++ b/src/venv.rs
@@ -17,3 +17,16 @@ pub fn get_name() -> colored::ColoredString {
     return "".white()
 }
 
+pub fn in_nix_shell() -> colored::ColoredString {
+    match env::var("IN_NIX_SHELL") {
+        Ok(p) => {
+            if p == "pure" {
+                "(nix) ".green()
+            } else {
+                "(nix) ".red()
+            }
+        }
+        _ => return "".white(),
+    }
+}
+