From 69abd4b626471c7be1ccebd950265d64451a09da Mon Sep 17 00:00:00 2001
From: Matthieu Vachon <mvachon@samsao.co>
Date: Tue, 23 Jan 2018 16:08:12 -0500
Subject: [PATCH] Added possibility to show shorten project dir on Git

Added a new option `BULLETTRAIN_DIR_SHORTEN_GIT_REPO` available through
`prompt_dir` that will displays a shorten prompt dir if the current
working directory is within a git repository.

For a `prompt_dir` of `/Users/john/work/project/a` where `a` has a
`.git` folder, the prompt dir when using
`BULLETTRAIN_DIR_SHORTEN_GIT_REPO` would be simply `a`. For a
`prompt_dir` of `/Users/john/work/project/b/nested/deep` where `b` has a
`.git` folder, the prompt dir when using
`BULLETTRAIN_DIR_SHORTEN_GIT_REPO` would be `b/nested/deep`.

If there is no `.git` directory within the working directory or
its parent, then `prompt_dir` behaves as before.
---
 bullet-train.zsh-theme | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/bullet-train.zsh-theme b/bullet-train.zsh-theme
index c31cc96..0843fb8 100644
--- a/bullet-train.zsh-theme
+++ b/bullet-train.zsh-theme
@@ -474,6 +474,23 @@ prompt_hg() {
   fi
 }
 
+prompt_pwd_maybe_short_dir() {
+  git_root=$PWD
+  while [[ $git_root != / && ! -e $git_root/.git ]]; do
+    git_root=$git_root:h
+  done
+
+  if [[ $git_root = / || $git_root = ~ || $git_root = "$HOME" ]]; then
+    unset git_root
+    prompt_short_dir='%~'
+  else
+    parent=${git_root%\/*}
+    prompt_short_dir=${PWD#$parent/}
+  fi
+
+  echo $prompt_short_dir
+}
+
 # Dir: current working directory
 prompt_dir() {
   local dir=''
@@ -486,6 +503,8 @@ prompt_dir() {
   elif [[ $BULLETTRAIN_DIR_EXTENDED == 2 ]]; then
     #long directories
     dir="${dir}%0~"
+  elif [[ $BULLETTRAIN_DIR_SHORTEN_GIT_REPO == true ]] then
+    prompt_segment $BULLETTRAIN_DIR_BG $BULLETTRAIN_DIR_FG `prompt_pwd_maybe_short_dir`
   else
     #medium directories (default case)
     dir="${dir}%4(c:...:)%3c"
@@ -558,7 +577,7 @@ prompt_kctx() {
   if command -v kubectl > /dev/null 2>&1; then
     if [[ -f $BULLETTRAIN_KCTX_KCONFIG ]]; then
       prompt_segment $BULLETTRAIN_KCTX_BG $BULLETTRAIN_KCTX_FG $BULLETTRAIN_KCTX_PREFIX" $(cat $BULLETTRAIN_KCTX_KCONFIG|grep current-context| awk '{print $2}')"
-    fi  
+    fi
   fi
 }