detect nix-shell env

1 files changed, 3 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index b839325..1a77759 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -50,12 +50,13 @@ fn pista(zsh: bool) -> String {
         )
     } else {
         format!(
-            "{cwd} {branch} {status}\n{venv}{pchar} ",
+            "{cwd} {branch} {status}\n{venv}{nix}{pchar} ",
             cwd = cwd,
             branch = branch,
             status = status,
             venv = venv,
-            pchar = prompt_char
+            pchar = prompt_char,
+            nix = venv::in_nix_shell()
         )
     }
 }