From d7a8b87e4fc032ec9eb266d732439699524f1024 Mon Sep 17 00:00:00 2001
From: FriedUmami <54095148+FriedUmami@users.noreply.github.com>
Date: Sat, 28 Mar 2020 03:18:45 -0700
Subject: [PATCH] Add Conda support to virtual env

Code from pull request https://github.com/caiogondim/bullet-train.zsh/pull/214
---
 bullet-train.zsh-theme | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/bullet-train.zsh-theme b/bullet-train.zsh-theme
index d72c4f5..7d125e4 100644
--- a/bullet-train.zsh-theme
+++ b/bullet-train.zsh-theme
@@ -575,10 +575,13 @@ prompt_kctx() {
 # Virtualenv: current working virtualenv
 prompt_virtualenv() {
   local virtualenv_path="$VIRTUAL_ENV"
-  if [[ -n $virtualenv_path && -n $VIRTUAL_ENV_DISABLE_PROMPT ]]; then
-    prompt_segment $BULLETTRAIN_VIRTUALENV_BG $BULLETTRAIN_VIRTUALENV_FG $BULLETTRAIN_VIRTUALENV_PREFIX" $(basename $virtualenv_path)"
-  elif which pyenv &> /dev/null; then
-    if [[ "$(pyenv version | sed -e 's/ (set.*$//' | tr '\n' ' ' | sed 's/.$//')" != "system" ]]; then
+  if [[ -n $VIRTUAL_ENV_DISABLE_PROMPT ]]; then
+    local virtualenv_path="$VIRTUAL_ENV"
+    if [[ -n $CONDA_DEFAULT_ENV ]]; then
+      prompt_segment $BULLETTRAIN_VIRTUALENV_BG $BULLETTRAIN_VIRTUALENV_FG $BULLETTRAIN_VIRTUALENV_PREFIX" $CONDA_DEFAULT_ENV"
+    elif [[ -n $virtualenv_path ]]; then
+      prompt_segment $BULLETTRAIN_VIRTUALENV_BG $BULLETTRAIN_VIRTUALENV_FG $BULLETTRAIN_VIRTUALENV_PREFIX" $(basename $virtualenv_path)"
+    elif which pyenv &> /dev/null; then
       prompt_segment $BULLETTRAIN_VIRTUALENV_BG $BULLETTRAIN_VIRTUALENV_FG $BULLETTRAIN_VIRTUALENV_PREFIX" $(pyenv version | sed -e 's/ (set.*$//' | tr '\n' ' ' | sed 's/.$//')"
     fi
   fi