omg/prompt.sh

180 lines
5.6 KiB
Bash
Raw Normal View History

2013-11-06 10:32:06 -05:00
# Symbols
2013-11-07 07:21:19 -05:00
: ${is_a_git_repo_symbol:='❤'}
: ${has_untracked_files_symbol:='∿'}
: ${has_adds_symbol:='+'}
: ${has_deletions_symbol:='-'}
: ${has_deletions_cached_symbol:='✖'}
: ${has_modifications_symbol:='✎'}
: ${has_modifications_cached_symbol:='☲'}
: ${ready_to_commit_symbol:='→'}
: ${is_on_a_tag_symbol:='⌫'}
: ${needs_to_merge_symbol:='ᄉ'}
: ${has_upstream_symbol:='⇅'}
: ${detached_symbol:='⚯ '}
: ${can_fast_forward_symbol:='»'}
: ${has_diverged_symbol:='Ⴤ'}
: ${rebase_tracking_branch_symbol:='↶'}
: ${merge_tracking_branch_symbol:='ᄉ'}
: ${should_push_symbol:='↑'}
: ${has_stashes_symbol:='★'}
2013-11-06 10:32:06 -05:00
# Flags
: ${display_has_upstream:=false}
: ${display_tag:=false}
: ${display_tag_name:=true}
: ${two_lines:=true}
2013-11-07 07:21:19 -05:00
: ${finally:='\w ∙ '}
2013-11-06 10:32:06 -05:00
: ${use_color_off:=false}
# Colors
2013-11-07 07:21:19 -05:00
: ${on='\[\033[0;37m\]'}
: ${off='\[\033[1;30m\]'}
: ${red='\[\033[0;31m\]'}
: ${green='\[\033[0;32m\]'}
: ${yellow='\[\033[0;33m\]'}
: ${violet='\[\033[0;35m\]'}
: ${branch_color='\[\033[0;34m\]'}
#: ${blinking='\[\033[1;5;17m\]'}
: ${reset='\[\033[0m\]'}
2013-11-06 10:32:06 -05:00
2013-09-02 02:31:05 -04:00
function enrich {
2013-09-23 10:35:34 -04:00
flag=$1
symbol=$2
if [[ -n $3 ]]; then
coloron=$3
else
coloron=$on
2013-09-23 10:35:34 -04:00
fi
2013-11-07 18:36:12 -05:00
if [[ $use_color_off == false && $flag == false ]]; then symbol=' '; fi
if [[ $flag == true ]]; then color=$coloron; else color=$off; fi
2013-09-23 10:36:43 -04:00
PS1="${PS1}${color}${symbol}${reset} "
2013-09-02 02:31:05 -04:00
}
function build_prompt {
2013-09-23 10:35:34 -04:00
PS1=""
# Git info
2013-11-07 18:16:02 -05:00
current_commit_hash=$(git rev-parse HEAD 2> /dev/null)
2013-09-23 10:35:34 -04:00
if [[ -n $current_commit_hash ]]; then is_a_git_repo=true; else is_a_git_repo=false; fi
2013-11-07 18:59:55 -05:00
if [[ $is_a_git_repo == true ]]; then
current_branch=$(git rev-parse --abbrev-ref HEAD 2> /dev/null)
if [[ $current_branch == 'HEAD' ]]; then detached=true; else detached=false; fi
2013-11-07 18:59:55 -05:00
number_of_logs=$(git log --pretty=oneline -n1 2> /dev/null | wc -l)
if [[ $number_of_logs -eq 0 ]]; then
just_init=true
else
upstream=$(git rev-parse --symbolic-full-name --abbrev-ref @{upstream} 2> /dev/null)
if [[ $upstream != '@{upstream}' ]]; then has_upstream=true; else has_upstream=false; upstream=''; fi
2013-09-23 10:35:34 -04:00
2013-11-07 18:59:55 -05:00
git_status=$(git status --porcelain 2> /dev/null)
2013-11-07 18:59:55 -05:00
if [[ $git_status =~ ^.M ]]; then has_modifications=true; else has_modifications=false; fi
2013-09-23 10:35:34 -04:00
2013-11-07 18:59:55 -05:00
if [[ $git_status =~ ^M ]]; then has_modifications_cached=true; else has_modifications_cached=false; fi
2013-11-06 08:48:23 -05:00
2013-11-07 18:59:55 -05:00
if [[ $git_status =~ ^A ]]; then has_adds=true; else has_adds=false; fi
2013-09-03 17:35:13 -04:00
2013-11-07 18:59:55 -05:00
if [[ $git_status =~ ^.D ]]; then has_deletions=true; else has_deletions=false; fi
2013-09-02 16:59:09 -04:00
2013-11-07 18:59:55 -05:00
if [[ $git_status =~ ^D ]]; then has_deletions_cached=true; else has_deletions_cached=false; fi
2013-09-02 16:59:09 -04:00
2013-11-07 18:59:55 -05:00
if [[ $git_status =~ ^[MAD] && ! $git_status =~ ^.[MAD\?] ]]; then ready_to_commit=true; else ready_to_commit=false; fi
2013-09-02 13:12:27 -04:00
2013-11-07 18:59:55 -05:00
if [[ $git_status =~ ^\?\? ]]; then has_untracked_files=true; else has_untracked_files=false; fi
2013-11-06 08:48:23 -05:00
2013-11-07 18:59:55 -05:00
tag_at_current_commit=$(git describe --exact-match --tags $current_commit_hash 2> /dev/null)
if [[ -n $tag_at_current_commit ]]; then is_on_a_tag=true; else is_on_a_tag=false; fi
2013-11-07 18:59:55 -05:00
has_diverged=false
can_fast_forward=false
2013-11-07 18:59:55 -05:00
commits_diff=$(git log --pretty=oneline --topo-order --left-right ${current_commit_hash}...${upstream} 2> /dev/null)
2013-11-07 19:06:11 -05:00
commits_ahead=$(grep -c '^<' <<< "$commits_diff")
commits_behind=$(grep -c '^>' <<< "$commits_diff")
2013-11-07 07:21:19 -05:00
2013-11-07 18:59:55 -05:00
if [[ $commits_ahead -gt 0 && $commits_behind -gt 0 ]]; then
has_diverged=true
fi
if [[ $commits_ahead -eq 0 && $commits_behind -gt 0 ]]; then
can_fast_forward=true
fi
2013-11-07 18:59:55 -05:00
will_rebase=$(git config --get branch.${current_branch}.rebase 2> /dev/null)
2013-11-07 18:59:55 -05:00
number_of_stashes=$(wc -l 2> /dev/null < .git/refs/stash)
if [[ $number_of_stashes -gt 0 ]]; then has_stashes=true; else has_stashes=false; fi
fi
2013-09-23 10:35:34 -04:00
fi
2013-09-03 02:35:50 -04:00
if [[ $is_a_git_repo == true ]]; then
enrich $is_a_git_repo $is_a_git_repo_symbol $violet
enrich $has_stashes $has_stashes_symbol $yellow
enrich $has_untracked_files $has_untracked_files_symbol $red
enrich $has_adds $has_adds_symbol $yellow
2013-09-03 01:47:52 -04:00
enrich $has_deletions $has_deletions_symbol $red
enrich $has_deletions_cached $has_deletions_cached_symbol $yellow
2013-09-03 01:47:52 -04:00
enrich $has_modifications $has_modifications_symbol $red
enrich $has_modifications_cached $has_modifications_cached_symbol $yellow
enrich $ready_to_commit $ready_to_commit_symbol $green
2013-09-03 01:47:52 -04:00
enrich $detached $detached_symbol $red
2013-09-02 16:59:09 -04:00
if [[ $display_has_upstream == true ]]; then
enrich $has_upstream $has_upstream_symbol
fi
if [[ $detached == true ]]; then
if [[ $just_init == true ]]; then
2013-11-07 18:36:12 -05:00
PS1="${PS1} ${red}detached"
2013-09-23 10:35:34 -04:00
else
2013-11-07 18:36:12 -05:00
PS1="${PS1} ${on}(${current_commit_hash:0:7})"
2013-09-23 10:35:34 -04:00
fi
else
2013-09-23 10:35:34 -04:00
if [[ $has_upstream == true ]]; then
if [[ $will_rebase == true ]]; then
type_of_upstream=$rebase_tracking_branch_symbol
2013-09-23 10:35:34 -04:00
else
type_of_upstream=$merge_tracking_branch_symbol
2013-09-23 10:35:34 -04:00
fi
if [[ $has_diverged == true ]]; then
PS1="${PS1}-${commits_behind} ${has_diverged_symbol} +${commits_ahead} "
2013-09-23 10:35:34 -04:00
else
2013-11-07 07:33:29 -05:00
if [[ $commits_behind -gt 0 ]]; then
PS1="${PS1}${on} -${commits_behind} ${can_fast_forward_symbol} "
2013-09-23 10:35:34 -04:00
fi
2013-11-07 07:33:29 -05:00
if [[ $commits_ahead -gt 0 ]]; then
PS1="${PS1}${on} ${should_push_symbol} +${commits_ahead} "
2013-09-23 10:35:34 -04:00
fi
fi
2013-09-23 10:36:43 -04:00
PS1="${PS1}(${green}${current_branch}${reset} ${type_of_upstream} ${upstream//\/$current_branch/})"
2013-09-23 10:35:34 -04:00
else
2013-09-23 10:36:43 -04:00
PS1="${PS1}${on}(${green}${current_branch}${reset})"
2013-09-23 10:35:34 -04:00
fi
fi
2013-11-07 18:59:55 -05:00
if [[ $display_tag == true && $is_on_a_tag == true ]]; then
2013-11-07 18:36:12 -05:00
PS1="${PS1} ${yellow}${is_on_a_tag_symbol}${reset}"
2013-09-23 10:35:34 -04:00
fi
2013-11-06 15:55:38 -05:00
if [[ $display_tag_name == true && $is_on_a_tag == true ]]; then
2013-11-07 18:36:12 -05:00
PS1="${PS1} ${yellow}[${tag_at_current_commit}]${reset}"
2013-09-23 10:35:34 -04:00
fi
2013-09-02 13:00:26 -04:00
fi
2013-11-06 15:55:38 -05:00
if [[ $two_lines == true && $is_a_git_repo == true ]]; then
2013-11-07 07:21:19 -05:00
break='\n'
2013-11-06 08:48:23 -05:00
else
2013-11-07 07:21:19 -05:00
break=''
fi
2013-09-23 10:35:34 -04:00
PS1="${PS1}${reset}${break}${finally}"
}
2013-09-23 10:36:43 -04:00
PS2="${yellow}${reset} "
PROMPT_COMMAND=build_prompt