From 4b80626122466e8e3ae166d7212d5511412795bc Mon Sep 17 00:00:00 2001
From: Akshay <nerdy@peppe.rs>
Date: Tue, 18 Aug 2020 14:54:00 +0530
Subject: detect nix-shell env

---
 src/venv.rs | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'src/venv.rs')

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(),
+    }
+}
+
-- 
cgit v1.2.3