From b0f8cea364469281134788ff617d9403cff1347d Mon Sep 17 00:00:00 2001
From: Justin Hileman <justin@justinhileman.info>
Date: Wed, 30 Mar 2016 16:57:34 -0700
Subject: [PATCH] Make bobthefish git-workdir aware

Fixes #29
---
 fish_prompt.fish | 91 ++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 84 insertions(+), 7 deletions(-)

diff --git a/fish_prompt.fish b/fish_prompt.fish
index 4e7e2ee..3689e1c 100644
--- a/fish_prompt.fish
+++ b/fish_prompt.fish
@@ -67,7 +67,41 @@ end
 
 function __bobthefish_git_project_dir -S -d 'Print the current git project base directory'
   [ "$theme_display_git" = 'no' ]; and return
-  command git rev-parse --show-toplevel ^/dev/null
+  set -l git_dir (command git rev-parse --git-dir ^/dev/null); or return
+
+  pushd $git_dir
+  set git_dir $PWD
+  popd
+
+  switch $PWD/
+    case $git_dir/\*
+      # Nothing works quite right if we're inside the git dir
+      # TODO: fix the underlying issues then re-enable the stuff below
+
+      # # if we're inside the git dir, sweet. just return that.
+      # set -l toplevel (command git rev-parse --show-toplevel ^/dev/null)
+      # if [ "$toplevel" ]
+      #   switch $git_dir/
+      #     case $toplevel/\*
+      #       echo $git_dir
+      #   end
+      # end
+      return
+  end
+
+  set -l project_dir (dirname $git_dir)
+
+  switch $PWD/
+    case $project_dir/\*
+      echo $project_dir
+      return
+  end
+
+  set project_dir (command git rev-parse --show-toplevel ^/dev/null)
+  switch $PWD/
+    case $project_dir/\*
+      echo $project_dir
+  end
 end
 
 function __bobthefish_hg_project_dir -S -d 'Print the current hg project base directory'
@@ -378,15 +412,58 @@ function __bobthefish_prompt_git -S -a current_dir -d 'Display the actual git st
   echo -ns (__bobthefish_git_branch) $flags ' '
   set_color normal
 
-  set -l project_pwd (__bobthefish_project_pwd $current_dir)
-  if [ "$project_pwd" ]
-    if [ -w "$PWD" ]
-      __bobthefish_start_segment $__bobthefish_dk_grey $__bobthefish_med_grey
-    else
-      __bobthefish_start_segment $__bobthefish_med_red $__bobthefish_lt_red
+  set -l project_pwd (command git rev-parse --show-prefix ^/dev/null | sed -e 's#/$##')
+  set -l work_dir (command git rev-parse --show-toplevel ^/dev/null)
+
+  # only show work dir if it's a parent…
+  if [ "$work_dir" ]
+    switch $PWD/
+      case $work_dir/\*
+        set work_dir (echo $work_dir | sed -e "s#^$current_dir##")
+      case \*
+        set -e work_dir
+    end
+  end
+
+  if [ "$project_pwd" -o "$work_dir" ]
+    set -l bg_color $__bobthefish_dk_grey
+    set -l fg_color $__bobthefish_med_grey
+    if not [ -w "$PWD" ]
+      set bg_color $__bobthefish_med_red
+      set fg_color $__bobthefish_lt_red
+    end
+
+    __bobthefish_start_segment $bg_color $fg_color
+
+    # handle work_dir != project dir
+    if [ "$work_dir" ]
+      set -l work_parent (dirname $work_dir | sed -e 's#^/##')
+      if [ "$work_parent" ]
+        set_color --background $bg_color $fg_color
+        echo -n "$work_parent/"
+      end
+      set_color fff --bold
+      echo -n (basename $work_dir)
+      set_color --background $bg_color $fg_color
+      [ "$project_pwd" ]
+        and echo -n '/'
     end
 
     echo -ns $project_pwd ' '
+  else
+    set project_pwd (echo $PWD | sed -e "s#^$current_dir##" -e 's#^/##')
+    if [ "$project_pwd" ]
+      set -l bg_color $__bobthefish_dk_grey
+      set -l fg_color $__bobthefish_med_grey
+      if not [ -w "$PWD" ]
+        set bg_color $__bobthefish_med_red
+        set fg_color $__bobthefish_lt_red
+      end
+
+      __bobthefish_start_segment $bg_color $fg_color
+
+      echo -ns $project_pwd ' '
+    end
   end
 end