From 194f058b8a9e6ee1abacc4f4ee9565f0d3ac038d Mon Sep 17 00:00:00 2001
From: bretello <bretello@distruzione.org>
Date: Fri, 4 Dec 2020 01:56:14 +0100
Subject: [PATCH] zsh-theme: change color virtualenv segment

---
 brethil.zsh-theme | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/brethil.zsh-theme b/brethil.zsh-theme
index 895544a..897360a 100644
--- a/brethil.zsh-theme
+++ b/brethil.zsh-theme
@@ -41,7 +41,19 @@ ZSH_THEME_GIT_PROMPT_BEHIND_REMOTE="%F{255}<%f"
 ZSH_THEME_GIT_PROMPT_DIVERGED_REMOTE="%F{255}<>%f"
 
 function virtualenv_info {
-    [ $VIRTUAL_ENV ] && echo ' %F{27}<'$(basename $VIRTUAL_ENV)'> %f'
+    [ -z "$VIRTUAL_ENV" ] && return
+
+    local BOLD_PURPLE='%B%F{5}'
+    local CLEAR='%f%b'
+    local venv
+    venv="$(basename "$VIRTUAL_ENV")"
+    if [[ "$venv" == ".venv" ]]; then
+      venv="$(basename $(dirname "$VIRTUAL_ENV") )"
+    fi
+
+    echo -n "$BOLD_PURPLE"
+    echo -n " <$venv> "
+    echo -n "$CLEAR"
 }
 
 function git_prompt() {