From de13050d07d1670cae7adec317d5702626bf2995 Mon Sep 17 00:00:00 2001
From: Amit Beka <beka@localhost.localdomain>
Date: Sat, 8 Apr 2017 00:03:43 +0300
Subject: [PATCH] themes/steeef: support conda env in prompt

Add support for showing the conda env in the prompt and not only the
virtualenv.
---
 themes/steeef.zsh-theme | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/themes/steeef.zsh-theme b/themes/steeef.zsh-theme
index b72a41c92..b4cdcd4b8 100644
--- a/themes/steeef.zsh-theme
+++ b/themes/steeef.zsh-theme
@@ -9,8 +9,12 @@
 
 export VIRTUAL_ENV_DISABLE_PROMPT=1
 
-function virtualenv_info {
-    [ $VIRTUAL_ENV ] && echo '('%F{blue}`basename $VIRTUAL_ENV`%f') '
+function virtualenv_or_conda_info {
+    if [[ -n "$VIRTUAL_ENV" ]]; then
+        echo '('%F{blue}`basename $VIRTUAL_ENV`%f') '
+    elif [[ -n "$CONDA_DEFAULT_ENV" ]]; then
+        echo '('%F{blue}`basename $CONDA_DEFAULT_ENV`%f') '
+    fi
 }
 PR_GIT_UPDATE=1
 
@@ -96,5 +100,5 @@ function steeef_precmd {
 add-zsh-hook precmd steeef_precmd
 
 PROMPT=$'
-%{$purple%}%n${PR_RST} at %{$orange%}%m${PR_RST} in %{$limegreen%}%~${PR_RST} $vcs_info_msg_0_$(virtualenv_info)
+%{$purple%}%n${PR_RST} at %{$orange%}%m${PR_RST} in %{$limegreen%}%~${PR_RST} $vcs_info_msg_0_$(virtualenv_or_conda_info)
 $ '