From c7f6e80ad04758e18e9c9f97850a9b2e59faeaf1 Mon Sep 17 00:00:00 2001
From: tiago <tiagobarros99@ua.pt>
Date: Fri, 22 Jan 2021 17:07:16 +0000
Subject: [PATCH] fix modern theme, added virtualenv name

---
 themes/modern/modern.theme.bash | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/themes/modern/modern.theme.bash b/themes/modern/modern.theme.bash
index 583764e2..aa01234f 100644
--- a/themes/modern/modern.theme.bash
+++ b/themes/modern/modern.theme.bash
@@ -35,18 +35,22 @@ modern_scm_prompt() {
 	fi
 }
 
-prompt() {
-	if [ $? -ne 0 ]
-	then
-		# Yes, the indenting on these is weird, but it has to be like
-		# this otherwise it won't display properly.
-
-    PS1="${TITLEBAR}${bold_red}┌─${reset_color}$(modern_scm_prompt)[${cyan}\W${normal}][$(battery_charge)]$(is_vim_shell)
-${bold_red}└─▪${normal} "
-	else
-		PS1="${TITLEBAR}┌─$(modern_scm_prompt)[${cyan}\W${normal}][$(battery_charge)]$(is_vim_shell)
-└─▪ "
+detect_venv() {
+	python_venv=""
+	# Detect python venv
+	if [[ -n "${CONDA_DEFAULT_ENV}" ]]; then
+		python_venv="($PYTHON_VENV_CHAR${CONDA_DEFAULT_ENV})"
+	elif [[ -n "${VIRTUAL_ENV}" ]]; then
+		python_venv="($PYTHON_VENV_CHAR$(basename "${VIRTUAL_ENV}"))"
 	fi
+
+}
+prompt() {
+	detect_venv
+	
+	PS1="${TITLEBAR}┌─$(modern_scm_prompt)[${cyan}\u${normal}][${cyan}\w${normal}]$(is_vim_shell)
+└─▪ ${python_venv}${dir_color} "
+
 }
 
 PS2="└─▪ "