From cdbeffc30b1146272cca79115289842aac3e01b4 Mon Sep 17 00:00:00 2001
From: Roman Inflianskas <infroma@gmail.com>
Date: Thu, 20 Nov 2014 23:58:23 +0300
Subject: [PATCH] bobthefish: added promt support for virtualfish

---
 fish_prompt.fish | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/fish_prompt.fish b/fish_prompt.fish
index b6a5413..965cc6e 100644
--- a/fish_prompt.fish
+++ b/fish_prompt.fish
@@ -32,6 +32,11 @@ set __bobthefish_nonzero_exit_glyph      '! '
 set __bobthefish_superuser_glyph         '$ '
 set __bobthefish_bg_job_glyph            '% '
 
+# Python glyphs
+set __bobthefish_superscript             \u00B1 \u00B2 \u00B3
+set __bobthefish_virtualenv_glyph        \u25F0
+set __bobthefish_pypy                    \u1D56
+
 # Colors
 set __bobthefish_lt_green   addc10
 set __bobthefish_med_green  189303
@@ -50,6 +55,10 @@ set __bobthefish_dk_grey    333
 set __bobthefish_med_grey   999
 set __bobthefish_lt_grey    ccc
 
+set __bobthefish_dk_brown   4d2600
+set __bobthefish_med_brown  803F00
+set __bobthefish_lt_brown   BF5E00
+
 # ===========================
 # Helper methods
 # ===========================
@@ -249,6 +258,34 @@ function __bobthefish_prompt_dir -d 'Display a shortened form of the current dir
   __bobthefish_path_segment "$PWD"
 end
 
+function __bobthefish_in_virtualfish_virtualenv
+  set -q VIRTUAL_ENV
+end
+
+function __bobthefish_virtualenv_python_version -d 'Get current python version'
+  switch (readlink (which python))
+    case python2
+      echo $__bobthefish_superscript[2]
+    case python3
+      echo $__bobthefish_superscript[3]
+    case pypy
+      echo $__bobthefish_pypy
+    end
+end
+
+function __bobthefish_virtualenv -d 'Get the current virtualenv'
+  echo $__bobthefish_virtualenv_glyph(__bobthefish_virtualenv_python_version) (basename "$VIRTUAL_ENV")
+end
+
+function __bobthefish_promt_virtualfish -d "Display activated virtual environment (only for virtualfish, virtualenv's activate.fish changes promt by itself)"
+  set flag_bg $__bobthefish_lt_brown
+  set flag_fg $__bobthefish_dk_brown
+  __bobthefish_start_segment $flag_bg $flag_fg
+  set_color $flag_fg --bold
+  echo -n -s (__bobthefish_virtualenv) $flags ' '
+  set_color normal
+end
+
 
 # ===========================
 # Apply theme
@@ -263,5 +300,8 @@ function fish_prompt -d 'bobthefish, a fish theme optimized for awesome'
   else
     __bobthefish_prompt_dir
   end
+  if __bobthefish_in_virtualfish_virtualenv
+    __bobthefish_promt_virtualfish
+  end
   __bobthefish_finish_segments
 end