X-Git-Url: http://git.iain.cx/?p=profile.git;a=blobdiff_plain;f=.profile.d%2Fps1.bashrc;h=e7d3658535bb4509319fbf4c79c6270b41964188;hp=b65e42d8447e3c3b91a00d4ddec329ed548ae59a;hb=0772e1850820e5e1bac2dbff880b0ac17386c9c9;hpb=0614eba6741fe0b5f1ef54ad7478fe003ca0e613 diff --git a/.profile.d/ps1.bashrc b/.profile.d/ps1.bashrc index b65e42d..e7d3658 100644 --- a/.profile.d/ps1.bashrc +++ b/.profile.d/ps1.bashrc @@ -1,98 +1,198 @@ #!bash Coloured prompts. -# To use, add a call to __ps1 in your .bash_profile file. +# profile-required: TERM.bashrc # -# The prompt comprises multiple parts, some of which may be hidden by unsetting +# The prompt comprises multiple parts, some of which may be hidden by unsetting # shell variables or using the ``prompt'' function. # -# The first part of the prompt is user@host where host is highlighted in +# The first part of the prompt is user@host where host is highlighted in # green if the last command exited 0 or in red otherwise. # This part will be shown only if __ps1_user is 1. By default it is 1. -# The success and failure colours can be changed by modifying -# $PROMPT_OK_COLOUR and $PROMPT_FAILURE_COLOUR respectively. +# The success and failure colours can be changed by modifying +# $PROMPT_OK_COLOUR and $PROMPT_FAILED_COLOUR respectively. # -# The second part of the prompt is taken from git-completion.bashrc. -# It is shown only if __ps1_git is 1. By default it is 0. +# Subsequent parts are taken from *.ps1 files in the .ps1.d directory. +# Each file .ps1 should contain: +# * A line __ps1_=${__ps1_:-X} where X is 0 or 1 to activate +# the plugin by default (or not). +# * A function __ps1_() which returns the text to be displayed. +# The function will be passed the return code of the last command as its +# first argument and should return the same code. +# The function should print nothing unless __ps1_ is 1. +# The function should call "__ps1_prefix $1 __ps1_" to retrieve a +# prefix string. When the string is non-empty, it should be printed before +# any output from the plugin itself. Doing so will ensure that the text is +# formatted correctly regardless of whether other parts of the prompt are +# being shown. +# * One to three lines __ps1__colour256, __ps1__colour88 and +# __ps1__colour setting the colour for the plugin. +# The plugin should not emit ANSI colour sequences itself. PS1 needs to wrap +# all escape sequences with the literal strings \[ and \] but these must be +# embedded directly into the prompt and not evaluated programmatically. +# Omitting them will cause the terminal to redraw incorrectly under certain +# circumstances. Trying to include them in a function will not work. +# Instead the plugin should define at least __ps1__colour. The +# colour will be generated before any text from the plugin is printed. +# To handle 88- and 256-colour terminals, the plugin may also declare +# __ps1__colour88 and/or __ps1__colour256. # -# The third part of the prompt is taken from p4-completion.bashrc. -# It is shown only if __ps1_p4 is 1. By default it is 0. +# Example plugin code: # -# The fourth part of the prompt is taken from svn-completion.bashrc. -# It is shown only if __ps1_svn is 1. By default it is 0. +# __ps1_example_colour='0;33' # -# The fifth part of the prompt is the exit status of the last command. -# This part will be shown only if __ps1_user is set and the exit status is +# function __ps1_example() { +# if [ -n "$__ps1_example" ]; then +# echo -n "$(__ps1_prefix $1 __ps1_example)" +# echo -n "example!" +# fi +# return $1 +# } +# +# The next part of the prompt is the exit status of the last command. +# This part will be shown only if __ps1_user is set and the exit status is # non-zero. -# +# +# The penultimate part of the prompt is the number of background jobs managed +# by the shell, in square brackets. If all background jobs are running, +# their number will be shown as [n]. If some are stopped, the number of +# running (r) and total (t) jobs will be shown as [r/t]. +# This part will be shown only if __ps1_bg is 1. By default it is 0. + # The final part of the prompt is the (full) working directory and $ string. +# If the shell is running as root the # string's colour can be changed by +# modifying $ROOT_OK_COLOUR and $ROOT_FAILED_COLOUR. # # Colouring is performed by the __ps1_col() and __ps1_ret() functions. -# We redirect stderr to /dev/null when calling these functions to prevent -# bash complaining about not knowing them when you su to another user, +# We redirect stderr to /dev/null when calling these functions to prevent +# bash complaining about not knowing them when you su to another user, # retaining PS1 but not the function definitions. # -# Note that $? is passed as an argument to - and is returned from - all -# functions. As $? is set following any shell activity it is only guaranteed +# Note that $? is passed as an argument to - and is returned from - all +# functions. As $? is set following any shell activity it is only guaranteed # to represent the return code of the last command at the beginning of __ps1(). -# By passing between subsequent functions we ensure that it is available for +# By passing between subsequent functions we ensure that it is available for # __ps1_ret(). # # Pick a colour based on the terminal capabilities. # OK: dark green. # Failed: dark red. -# Git: royal blue. -# P4: yellow. -# SVN: magenta. case $(tput colors) in 256) + __ps1_colours=256 + PROMPT_BACKGROUND_COLOUR="0;48;5;26" PROMPT_OK_COLOUR="1;38;5;34" PROMPT_FAILED_COLOUR="1;38;5;160" - GIT_COLOUR="0;38;5;33" - SVN_COLOUR="0;38;5;127" - P4_COLOUR="0;38;5;142" + ROOT_OK_COLOUR="0" + ROOT_FAILED_COLOUR="0" ;; 88) + __ps1_colours=88 + PROMPT_BACKGROUND_COLOUR="0;48;5;18" PROMPT_OK_COLOUR="1;38;5;24" PROMPT_FAILED_COLOUR="1;38;5;48" - GIT_COLOUR="0;38;5;23" - SVN_COLOUR="0;38;5;49" - P4_COLOUR="0;38;5;56" + ROOT_OK_COLOUR="0" + ROOT_FAILED_COLOUR="0" ;; *) + __ps1_colours= + PROMPT_BACKGROUND_COLOUR="0;44" PROMPT_OK_COLOUR="1;32" PROMPT_FAILED_COLOUR="1;31" - GIT_COLOUR="0;36" - SVN_COLOUR="0;35" - P4_COLOUR="0;33" + ROOT_OK_COLOUR="0" + ROOT_FAILED_COLOUR="0" ;; esac +__ps1_all='$__ps1_user' + function __ps1() { # Default __ps1_user to 1. [ -z "$__ps1_user" ] && __ps1_user=1 - PS1='$(__ps1_user $? \u@)\[\033[$(__ps1_col $? 2>/dev/null)m\]$(__ps1_user $? \h)\[\033[$(__ps1_colour_escape $? $GIT_COLOUR)m\]$(__ps1_git $? 2>/dev/null)\[\033[0m\]\[\033[$(__ps1_colour_escape $? $P4_COLOUR)m\]$(__ps1_p4 $? 2>/dev/null)\[\033[0m\]\[\033[$(__ps1_colour_escape $? $SVN_COLOUR)m\]$(__ps1_svn $? 2>/dev/null)\[\033[0m\]$(__ps1_ret $? 2>/dev/null):\w\$ ' + local pre='\[\033[$(__ps1_background $?)m\]$(__ps1_user $? \u@)\[\033[$(__ps1_col $? 2>/dev/null)m\]$(__ps1_user $? \h)\[\033[0m\]' + local post='\[\033[$(__ps1_background $?)m\]$(__ps1_ret $? 2>/dev/null)$(__ps1_bg $?)$(__ps1_colon $?)$(__ps1_short $?)\[\033[$(__ps1_root $? 2>/dev/null)m\]\$\[\033[0m\] ' + local snippets='' + local snippet= + for snippet in ${PROFILE_HOME:-~}/.ps1.d/*.ps1; do + . $snippet + local name=${snippet##*/} + name=__ps1_${name%.ps1} + __ps1_all="$__ps1_all\$$name" + snippets="$snippets"'\[$(__ps1_colour_start $? '"$name"')\]$('"$name"' $? $__ps1_all 2>/dev/null)\[$(__ps1_colour_end $? '"$name"')\]' + done + PS1=$pre$snippets$post + return 0 } +function __ps1_colour_for() { + local colour= + local ret= + for colour in "${1}_colour${__ps1_colours}" "${1}_colour"; do + ret=$(eval echo -n "\$$colour") + [ -n "$ret" ] && break + done + echo -n $ret +} + +function __ps1_prefix() { + local var=\$${2#\$} + local prefix=${__ps1_all%$var*} + local all="$(eval echo $prefix)" + [ "${all/1/}" = "$all" ] || echo -n " " + return $1 +} + # iTerm doesn't like it if you set bold and colour at the same time. function __ps1_colour_escape() { local ret=$1; shift local bold="${1%%;*}" local colour="${1#*;}" + local bgcolour="${2#*;}" echo -en "${bold}m\033[$colour" + [ "$__ps1_background" = 1 ] && echo -en "m\033[$bgcolour" + return $ret +} + +function __ps1_colour_start() { + local colour=$(__ps1_colour_for $2) + echo -en '\033[' + __ps1_colour_escape 0 "$colour" "$PROMPT_BACKGROUND_COLOUR" + echo -n m + return $1 +} + +function __ps1_colour_end() { + [ -n "$(__ps1_colour_for $2)" ] && echo -en '\033[0m' + return $1 +} + +function __ps1_background() { + local ret=$1; shift + echo -n "0" + [ "$__ps1_background" = 1 ] && echo -en "m\033[${PROMPT_BACKGROUND_COLOUR}" return $ret } function __ps1_col() { local ret=$1; shift if [ $ret -gt 0 ]; then - __ps1_colour_escape $ret "$PROMPT_FAILED_COLOUR" + __ps1_colour_escape $ret "$PROMPT_FAILED_COLOUR" "$PROMPT_BACKGROUND_COLOUR" + else + __ps1_colour_escape $ret "$PROMPT_OK_COLOUR" "$PROMPT_BACKGROUND_COLOUR" + fi + return $ret +} + +function __ps1_root() { + local ret=$1; shift + if [ $ret -gt 0 ]; then + __ps1_colour_escape $ret "$ROOT_FAILED_COLOUR" "$PROMPT_BACKGROUND_COLOUR" else - __ps1_colour_escape $ret "$PROMPT_OK_COLOUR" + __ps1_colour_escape $ret "$ROOT_OK_COLOUR" "$PROMPT_BACKGROUND_COLOUR" fi return $ret } @@ -110,33 +210,72 @@ function __ps1_user() { return $ret } -function __ps1_git() { - [ "$__ps1_git" = "1" ] || return $1 - if [ "$__ps1_user" = "1" ]; then - __git_ps1 ' %s' +function __ps1_bg() { + [ "$__ps1_bg" = "1" ] || return $1 + local job + local running=0; for job in $(builtin jobs -pr); do running=$((running+1)); done + local total=0; for job in $(builtin jobs -p); do total=$((total+1)); done + [ $total = 0 ] && return $1 + if [ -z "$2" ]; then + [ "$__ps1_user" = "1" ] && echo -n " " + echo -n "[" + [ $running = $total ] || echo -n "$running/" + echo -n "$total]" else - __git_ps1 '%s' + echo $2 fi return $1 } -function __ps1_p4() { - [ "$__ps1_p4" = "1" ] || return $1 - if [ "$__ps1_user" = "1" -o "$__ps1_git" = "1" ]; then - __p4_ps1 ' %s' - else - __p4_ps1 '%s' - fi +function __ps1_colon() { + local all="$__ps1_user$(eval echo $__ps1_all)$(__ps1_bg $1 1)" + [ "${all/1/}" = "$all" ] || echo -n ":" return $1 } -function __ps1_svn() { - [ "$__ps1_svn" = "1" ] || return $1 - if [ "$__ps1_user" = "1" -o "$__ps1_git" = "1" -o "$__ps1_p4" = "1" ]; then - __svn_ps1 ' %s' - else - __svn_ps1 '%s' +function __ps1_short() { + local home=${HOME%%/} + local pwd=${PWD/#$home/\~} + local dirtrim=${PROMPT_DIRTRIM//[^0-9]/} + + if [ "${dirtrim:0:1}" = "0" ]; then + echo "$pwd" + return $1 fi + + dirtrim=${dirtrim##0} + if [ -z "$dirtrim" ]; then + local prompt="$USER$HOSTNAME$pwd" + local width=$(((COLUMNS*2)/3)) + if [ ${#prompt} -le ${width:-53} ]; then + echo "$pwd" + return $1 + else + dirtrim=1 + fi + fi + + local dirname=${pwd##*/} + local basename=${pwd%/$dirname} + local reversed= + local component + for component in ${basename//\// }; do + reversed="$component $reversed" + done + local n=1 + local short= + for component in $reversed; do + [ $n = 1 -a "$PWD" = "$pwd" ] || short="/$short" + if [ $n -ge $dirtrim ]; then + short="${component:0:1}$short" + else + short="$component$short" + fi + n=$((n+1)) + done + + [ "${short:0:1}" = "~" ] || short="/$short" + echo "$short/$dirname" return $1 } @@ -162,3 +301,5 @@ function prompt() { what="$(echo $2 | env LANG= LC_ALL= LC_CTYPE= tr '[:upper:]' '[:lower:]')" eval __ps1_$what=$action } + +__ps1