From 675a42561b814004255e1198f3f82783237eb4d8 Mon Sep 17 00:00:00 2001
From: Maximilian Friedersdorff <max@friedersdorff.com>
Date: Fri, 3 Jul 2020 14:19:35 +0100
Subject: [PATCH] Print virtualenv directory if active

---
 states/zsh/files/zshrc | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/states/zsh/files/zshrc b/states/zsh/files/zshrc
index 77dec34..f02b7e3 100644
--- a/states/zsh/files/zshrc
+++ b/states/zsh/files/zshrc
@@ -32,16 +32,25 @@ bindkey -v
 NEWLINE=$'\n'
 setopt prompt_subst
 
+
 p="%(?;%F{green};%F{red})[%h]%f "
 p+="%F{red}%B%n%b%f@%F{red}%B%m%b%f"
 p+=":%F{blue}%~%f"
 p+="${NEWLINE}%(!.#.$) "
 
+virtual_env() {
+	if [ -n "${VIRTUAL_ENV+1}" ]; then
+		home_rel_path=$(echo ${VIRTUAL_ENV} | sed "s;^${HOME};~;")
+		echo -n "(${home_rel_path}) "
+	fi
+}
+
 source ~/.git-prompt.sh
 precmd () {
-	__git_ps1 "" "${p}" "${NEWLINE}[%s]${NEWLINE}"
+	__git_ps1 "" "$(virtual_env)${p}" "${NEWLINE}[%s]${NEWLINE}"
 }
 
+
 if ! pgrep -x -u "${USER}" gpg-agent >/dev/null 2>&1; then
   gpg-connect-agent /bye >/dev/null 2>&1
 fi
-- 
2.44.0