Ryan | 1a83a4d | 2011-05-31 13:19:20 -1000 | [diff] [blame] | 1 | #!/bin/basORANGEh |
| 2 | |
| 3 | # Colors |
| 4 | ORANGE='\e[0;33m' |
| 5 | GREY='\e[1:37m' |
| 6 | |
| 7 | DEFAULT_COLOR='\[${white}\]' |
| 8 | |
| 9 | USER_COLOR='\[${purple}\]' |
| 10 | SUPERUSER_COLOR='\[${red}\]' |
| 11 | MACHINE_COLOR=$ORANGE |
| 12 | DIRECTORY_COLOR='\[${bold_green}\]' |
| 13 | |
| 14 | VE_COLOR='\[${red}\]' |
| 15 | RVM_COLOR='\[${purple}\]' |
| 16 | |
| 17 | # SCM prompts |
| 18 | SCM_THEME_PROMPT_DIRTY=' ${bold_red}✗${normal}' |
| 19 | SCM_THEME_PROMPT_CLEAN=' ${bold_green}✓${normal}' |
| 20 | |
| 21 | # Max length of PWD to display |
| 22 | MAX_PWD_LENGTH=20 |
| 23 | |
| 24 | # Max length of Git Hex to display |
| 25 | MAX_GIT_HEX_LENGTH=5 |
| 26 | |
Ryan | 06d71ab | 2011-05-31 16:10:30 -1000 | [diff] [blame^] | 27 | GIT_THEME_PROMPT_PREFIX=' |git:' |
| 28 | GIT_THEME_PROMPT_SUFFIX='|' |
| 29 | |
| 30 | HG_THEME_PROMPT_PREFIX=' |hg:' |
| 31 | HG_THEME_PROMPT_SUFFIX='|' |
| 32 | |
Ryan | 1a83a4d | 2011-05-31 13:19:20 -1000 | [diff] [blame] | 33 | # Use http://geoff.greer.fm/lscolors/ |
| 34 | |
| 35 | # Displays the current virtualenv information |
| 36 | function curr_virtualenv_info() { |
| 37 | [ ! -z "$VIRTUAL_ENV" ] && echo "`basename $VIRTUAL_ENV`" |
| 38 | } |
| 39 | |
| 40 | # Displays the current rvm information w/gemset |
| 41 | function curr_rvm_info() { |
| 42 | local ruby_version=$(echo $MY_RUBY_HOME | awk -F'-' '{print $2}') |
| 43 | local ruby_gemset=$(echo $GEM_HOME | awk -F'@' '{print $2}') |
| 44 | |
| 45 | if [ "$ruby_version" != "" ]; then |
| 46 | [ "$ruby_gemset" != "" ] && ruby_gemset="@$ruby_gemset" |
| 47 | echo "$ruby_version$ruby_gemset" |
| 48 | fi |
| 49 | } |
| 50 | |
| 51 | # Displays using ... |
| 52 | function virtual_info() { |
| 53 | local virtual_env_info=$(curr_virtualenv_info) |
| 54 | local rvm_info=$(curr_rvm_info) |
| 55 | local prompt="using" |
| 56 | |
| 57 | # If no virtual info, just return |
| 58 | [ "$virtual_env_info" == "" -a "$rvm_info" == "" ] && return |
| 59 | |
| 60 | # If virtual_env info present, append to prompt |
| 61 | [ "$virtual_env_info" != "" ] && prompt="$prompt virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}" |
| 62 | |
| 63 | if [ "$rvm_info" != "" ] |
| 64 | then |
| 65 | [ "$virtual_env_info" != "" ] && prompt="$prompt," |
| 66 | prompt="$prompt rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}" |
| 67 | fi |
| 68 | echo $prompt |
| 69 | } |
| 70 | |
Ryan | 1a83a4d | 2011-05-31 13:19:20 -1000 | [diff] [blame] | 71 | # SCM information |
| 72 | function scm_info() { |
| 73 | SCM_CHAR=$(scm_char) |
| 74 | [ "$SCM_CHAR" == "$SCM_NONE_CHAR" ] && return |
| 75 | local prompt="on" |
| 76 | [ "$SCM_CHAR" == "$SCM_GIT_CHAR" ] && echo "$prompt$(parse_git_info)" && return |
Ryan | 06d71ab | 2011-05-31 16:10:30 -1000 | [diff] [blame^] | 77 | [ "$SCM_CHAR" == "$SCM_SVN_CHAR" ] && echo "$prompt$(parse_svn_info)" && return |
| 78 | [ "$SCM_CHAR" == "$SCM_HG_CHAR" ] && echo "$prompt$(parse_hg_info)" && return |
Ryan | 1a83a4d | 2011-05-31 13:19:20 -1000 | [diff] [blame] | 79 | } |
| 80 | |
| 81 | # Parse git info |
| 82 | function parse_git_info() { |
| 83 | if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then |
| 84 | state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY} |
| 85 | else |
| 86 | state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN} |
| 87 | fi |
| 88 | prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX} |
| 89 | suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX} |
| 90 | ref=$(git symbolic-ref HEAD 2> /dev/null) || return |
| 91 | rawhex=$(git rev-parse HEAD 2>/dev/null) || return |
| 92 | |
Ryan | 06d71ab | 2011-05-31 16:10:30 -1000 | [diff] [blame^] | 93 | echo "$prefix${ref#refs/heads/}=#${rawhex:0:$MAX_GIT_HEX_LENGTH}$state$suffix" |
Ryan | 1a83a4d | 2011-05-31 13:19:20 -1000 | [diff] [blame] | 94 | } |
| 95 | |
Ryan | 06d71ab | 2011-05-31 16:10:30 -1000 | [diff] [blame^] | 96 | # Parse hg info |
| 97 | function parse_hg_info() { |
| 98 | if [[ -n $(hg status --no-color 2> /dev/null| awk '$1 == "?" { print "?" } $1 = "?" { print "!" }' | sort | uniq | head -c1) ]]; then |
| 99 | state=${HG_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY} |
| 100 | else |
| 101 | state=${HG_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN} |
| 102 | fi |
| 103 | prefix=${HG_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX} |
| 104 | suffix=${HG_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX} |
| 105 | branch=$(hg branch 2> /dev/null | awk '{print $1}') |
| 106 | |
| 107 | echo "$prefix$branch$state$suffix" |
| 108 | } |
| 109 | |
| 110 | |
Ryan | 1a83a4d | 2011-05-31 13:19:20 -1000 | [diff] [blame] | 111 | # Displays last X characters of pwd |
| 112 | function limited_pwd() { |
| 113 | |
| 114 | # Replace $HOME with ~ if possible |
| 115 | RELATIVE_PWD=${PWD/#$HOME/\~} |
| 116 | |
| 117 | local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH)) |
| 118 | |
| 119 | if [ $offset -gt "0" ] |
| 120 | then |
| 121 | local truncated_symbol="..." |
| 122 | TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH} |
| 123 | echo "${truncated_symbol}/${TRUNCATED_PWD#*/}" |
| 124 | else |
| 125 | echo "${RELATIVE_PWD}" |
| 126 | fi |
| 127 | } |
| 128 | |
| 129 | # Displays the current prompt |
| 130 | function prompt() { |
| 131 | |
| 132 | local UC=$USER_COLOR |
| 133 | [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR |
| 134 | |
| 135 | PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h${DEFAULT_COLOR} in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR} $(virtual_info) $(scm_info) \$ " |
| 136 | PS2='> ' |
| 137 | PS4='+ ' |
| 138 | } |
| 139 | |
| 140 | PROMPT_COMMAND=prompt |