Skip to content

Instantly share code, notes, and snippets.

@stasiek
Created February 17, 2014 20:35
Show Gist options
  • Select an option

  • Save stasiek/9058584 to your computer and use it in GitHub Desktop.

Select an option

Save stasiek/9058584 to your computer and use it in GitHub Desktop.
Bash/ZSH Config
function __promptline_last_exit_code {
[[ $last_exit_code -gt 0 ]] || return 1;
printf "%s" "$last_exit_code"
}
function __promptline_ps1 {
local slice_prefix slice_empty_prefix slice_joiner slice_suffix is_prompt_empty=1
# section "a" header
slice_prefix="${a_bg}${sep}${a_fg}${a_bg}${space}" slice_suffix="$space${a_sep_fg}" slice_joiner="${a_fg}${a_bg}${alt_sep}${space}" slice_empty_prefix="${a_fg}${a_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "a" slices
__promptline_wrapper "$([[ -n ${ZSH_VERSION-} ]] && print %m || printf "%s" \\h)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# section "b" header
slice_prefix="${b_bg}${sep}${b_fg}${b_bg}${space}" slice_suffix="$space${b_sep_fg}" slice_joiner="${b_fg}${b_bg}${alt_sep}${space}" slice_empty_prefix="${b_fg}${b_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "b" slices
__promptline_wrapper "$([[ -n ${ZSH_VERSION-} ]] && print %n || printf "%s" \\u)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# modified "slice_prefix into "${slice_prefix}ⓔ "
__promptline_wrapper "${VIRTUAL_ENV##*/}" "${slice_prefix}ⓔ " "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# section "c" header
slice_prefix="${c_bg}${sep}${c_fg}${c_bg}${space}" slice_suffix="$space${c_sep_fg}" slice_joiner="${c_fg}${c_bg}${alt_sep}${space}" slice_empty_prefix="${c_fg}${c_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "c" slices
__promptline_wrapper "$(__promptline_cwd)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# section "z" header
slice_prefix="${z_bg}${sep}${z_fg}${z_bg}${space}" slice_suffix="$space${z_sep_fg}" slice_joiner="${z_fg}${z_bg}${alt_sep}${space}" slice_empty_prefix="${z_fg}${z_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "z" slices
__promptline_wrapper "$(__promptline_vcs_branch)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
__promptline_wrapper "$(__promptline_git_status)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# section "warn" header
slice_prefix="${warn_bg}${sep}${warn_fg}${warn_bg}${space}" slice_suffix="$space${warn_sep_fg}" slice_joiner="${warn_fg}${warn_bg}${alt_sep}${space}" slice_empty_prefix="${warn_fg}${warn_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "warn" slices
__promptline_wrapper "$(__promptline_last_exit_code)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# close sections
printf "%s" "${reset_bg}${sep}$reset$space"
}
function __promptline_vcs_branch {
local branch
local branch_symbol=" "
# git
if hash git 2>/dev/null; then
if branch=$( { git symbolic-ref --quiet HEAD || git rev-parse --short HEAD; } 2>/dev/null ); then
branch=${branch##*/}
printf "%s" "${branch_symbol}${branch:-unknown}"
return
fi
fi
return 1
}
function __promptline_cwd {
local dir_limit="3"
local truncation="⋯"
local first_char
local part_count=0
local formatted_cwd=""
local dir_sep="  "
local cwd="${PWD/#$HOME/~}"
# get first char of the path, i.e. tilde or slash
[[ -n ${ZSH_VERSION-} ]] && first_char=$cwd[1,1] || first_char=${cwd::1}
# remove leading tilde
cwd="${cwd#\~}"
while [[ "$cwd" == */* && "$cwd" != "/" ]]; do
# pop off last part of cwd
local part="${cwd##*/}"
cwd="${cwd%/*}"
formatted_cwd="$dir_sep$part$formatted_cwd"
part_count=$((part_count+1))
[[ $part_count -eq $dir_limit ]] && first_char="$truncation" && break
done
printf "%s" "$first_char$formatted_cwd"
}
function __promptline_left_prompt {
local slice_prefix slice_empty_prefix slice_joiner slice_suffix is_prompt_empty=1
# section "a" header
slice_prefix="${a_bg}${sep}${a_fg}${a_bg}${space}" slice_suffix="$space${a_sep_fg}" slice_joiner="${a_fg}${a_bg}${alt_sep}${space}" slice_empty_prefix="${a_fg}${a_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "a" slices
__promptline_wrapper "$([[ -n ${ZSH_VERSION-} ]] && print %m || printf "%s" \\h)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# section "b" header
slice_prefix="${b_bg}${sep}${b_fg}${b_bg}${space}" slice_suffix="$space${b_sep_fg}" slice_joiner="${b_fg}${b_bg}${alt_sep}${space}" slice_empty_prefix="${b_fg}${b_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "b" slices
__promptline_wrapper "$([[ -n ${ZSH_VERSION-} ]] && print %n || printf "%s" \\u)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
__promptline_wrapper "${VIRTUAL_ENV##*/}" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# section "c" header
slice_prefix="${c_bg}${sep}${c_fg}${c_bg}${space}" slice_suffix="$space${c_sep_fg}" slice_joiner="${c_fg}${c_bg}${alt_sep}${space}" slice_empty_prefix="${c_fg}${c_bg}${space}"
[ $is_prompt_empty -eq 1 ] && slice_prefix="$slice_empty_prefix"
# section "c" slices
__promptline_wrapper "$(__promptline_cwd)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; is_prompt_empty=0; }
# close sections
printf "%s" "${reset_bg}${sep}$reset$space"
}
function __promptline_wrapper {
# wrap the text in $1 with $2 and $3, only if $1 is not empty
# $2 and $3 typically contain non-content-text, like color escape codes and separators
[[ -n "$1" ]] || return 1
printf "%s" "${2}${1}${3}"
}
function __promptline_git_status {
[[ $(git rev-parse --is-inside-work-tree 2>/dev/null) == true ]] || return 1
local added_symbol="●"
local unmerged_symbol="✖"
local modified_symbol="✚"
local clean_symbol="✔"
local has_untracked_files_symbol="…"
local ahead_symbol="↑"
local behind_symbol="↓"
local unmerged_count=0 modified_count=0 has_untracked_files=0 added_count=0 is_clean=""
set -- $(git rev-list --left-right --count @{upstream}...HEAD 2>/dev/null)
local behind_count=$1
local ahead_count=$2
# Added (A), Copied (C), Deleted (D), Modified (M), Renamed (R), changed (T), Unmerged (U), Unknown (X), Broken (B)
while read line; do
case "$line" in
M*) modified_count=$(( $modified_count + 1 )) ;;
U*) unmerged_count=$(( $unmerged_count + 1 )) ;;
esac
done < <(git diff --name-status)
while read line; do
case "$line" in
*) added_count=$(( $added_count + 1 )) ;;
esac
done < <(git diff --name-status --cached)
if [ -n "$(git ls-files --others --exclude-standard)" ]; then
has_untracked_files=1
fi
if [ $(( unmerged_count + modified_count + has_untracked_files + added_count )) -eq 0 ]; then
is_clean=1
fi
local leading_whitespace=""
[[ $ahead_count -gt 0 ]] && { printf "%s" "$leading_whitespace$ahead_symbol$ahead_count"; leading_whitespace=" "; }
[[ $behind_count -gt 0 ]] && { printf "%s" "$leading_whitespace$behind_symbol$behind_count"; leading_whitespace=" "; }
[[ $modified_count -gt 0 ]] && { printf "%s" "$leading_whitespace$modified_symbol$modified_count"; leading_whitespace=" "; }
[[ $unmerged_count -gt 0 ]] && { printf "%s" "$leading_whitespace$unmerged_symbol$unmerged_count"; leading_whitespace=" "; }
[[ $added_count -gt 0 ]] && { printf "%s" "$leading_whitespace$added_symbol$added_count"; leading_whitespace=" "; }
[[ $has_untracked_files -gt 0 ]] && { printf "%s" "$leading_whitespace$has_untracked_files_symbol"; leading_whitespace=" "; }
[[ $is_clean -gt 0 ]] && { printf "%s" "$leading_whitespace$clean_symbol"; leading_whitespace=" "; }
}
function __promptline_right_prompt {
local slice_prefix slice_empty_prefix slice_joiner slice_suffix
# section "warn" header
slice_prefix="${warn_sep_fg}${rsep}${warn_fg}${warn_bg}${space}" slice_suffix="$space${warn_sep_fg}" slice_joiner="${warn_fg}${warn_bg}${alt_rsep}${space}" slice_empty_prefix=""
# section "warn" slices
__promptline_wrapper "$(__promptline_last_exit_code)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; }
# section "z" header
slice_prefix="${z_sep_fg}${rsep}${z_fg}${z_bg}${space}" slice_suffix="$space${z_sep_fg}" slice_joiner="${z_fg}${z_bg}${alt_rsep}${space}" slice_empty_prefix=""
# section "z" slices
__promptline_wrapper "$(__promptline_vcs_branch)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; }
__promptline_wrapper "$(__promptline_git_status)" "$slice_prefix" "$slice_suffix" && { slice_prefix="$slice_joiner"; }
# close sections
printf "%s" "$reset"
}
function __promptline {
local last_exit_code="$?"
local esc=$'[' end_esc=m
if [[ -n ${ZSH_VERSION-} ]]; then
local noprint='%{' end_noprint='%}'
else
local noprint='\[' end_noprint='\]'
fi
local wrap="$noprint$esc" end_wrap="$end_esc$end_noprint"
local space=" "
local sep=""
local rsep=""
local alt_sep=""
local alt_rsep=""
local reset="${wrap}0${end_wrap}"
local reset_bg="${wrap}49${end_wrap}"
local a_fg="${wrap}38;5;254${end_wrap}"
local a_bg="${wrap}48;5;235${end_wrap}"
local a_sep_fg="${wrap}38;5;235${end_wrap}"
local b_fg="${wrap}38;5;14${end_wrap}"
local b_bg="${wrap}48;5;242${end_wrap}"
local b_sep_fg="${wrap}38;5;242${end_wrap}"
local c_fg="${wrap}38;5;242${end_wrap}"
local c_bg="${wrap}48;5;237${end_wrap}"
local c_sep_fg="${wrap}38;5;237${end_wrap}"
local warn_fg="${wrap}38;5;232${end_wrap}"
local warn_bg="${wrap}48;5;166${end_wrap}"
local warn_sep_fg="${wrap}38;5;166${end_wrap}"
local z_fg="${wrap}38;5;254${end_wrap}"
local z_bg="${wrap}48;5;235${end_wrap}"
local z_sep_fg="${wrap}38;5;235${end_wrap}"
if [[ -n ${ZSH_VERSION-} ]]; then
PROMPT="$(__promptline_left_prompt)"
RPROMPT="$(__promptline_right_prompt)"
else
PS1="$(__promptline_ps1)"
fi
}
if [[ -n ${ZSH_VERSION-} ]]; then
if [[ ! ${precmd_functions[(r)__promptline]} == __promptline ]]; then
precmd_functions+=(__promptline)
fi
else
if [[ ! "$PROMPT_COMMAND" == *__promptline* ]]; then
PROMPT_COMMAND='__promptline;'$'\n'"$PROMPT_COMMAND"
fi
fi
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment