From e0603314d5b2b3cbb9ca5f75a990af813cc40ae1 Mon Sep 17 00:00:00 2001
From: rmad17 <souravbasu17@gmail.com>
Date: Fri, 26 Feb 2016 15:32:22 +0530
Subject: [PATCH] Nodeenv work

---
 powerlevel9k.zsh-theme | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/powerlevel9k.zsh-theme b/powerlevel9k.zsh-theme
index f8e75b33..94059c14 100755
--- a/powerlevel9k.zsh-theme
+++ b/powerlevel9k.zsh-theme
@@ -556,6 +556,23 @@ prompt_nvm() {
 
   $1_prompt_segment "$0" "$2" "green" "011" "${node_version:1}" 'NODE_ICON'
 }
+# NodeEnv Prompt
+prompt_nodeenv(){
+  export NODE_VIRTUAL_ENV_DISABLE_PROMPT=1
+  ZSH_THEME_NODEENV_PROMPT_PREFIX="("
+  ZSH_THEME_NODEENV_PROMPT_SUFFIX=")"
+      if [ -n "$NODE_VIRTUAL_ENV" ]; then
+
+          if [ "`basename "$NODE_VIRTUAL_ENV"`" = "__" ] ; then
+              # special case for Aspen magic directories
+              # see http://www.zetadev.com/software/aspen/
+              local name="[`basename \`dirname "$NODE_VIRTUAL_ENV"\``]"
+          else
+              local name=`basename "$NODE_VIRTUAL_ENV"`
+          fi
+          echo "$ZSH_THEME_NODEENV_PROMPT_PREFIX$name$ZSH_THEME_NODEENV_PROMPT_SUFFIX"
+      fi
+}
 
 # print a little OS icon
 prompt_os_icon() {