blob: 987153b3cc177d74ad3792be1c3da700f98f731d [file] [log] [blame]
Ryan1a83a4d2011-05-31 13:19:20 -10001#!/bin/basORANGEh
2
3# Colors
4ORANGE='\e[0;33m'
5GREY='\e[1:37m'
6
7DEFAULT_COLOR='\[${white}\]'
8
9USER_COLOR='\[${purple}\]'
10SUPERUSER_COLOR='\[${red}\]'
11MACHINE_COLOR=$ORANGE
12DIRECTORY_COLOR='\[${bold_green}\]'
13
14VE_COLOR='\[${red}\]'
15RVM_COLOR='\[${purple}\]'
16
17# SCM prompts
18SCM_THEME_PROMPT_DIRTY=' ${bold_red}✗${normal}'
19SCM_THEME_PROMPT_CLEAN=' ${bold_green}✓${normal}'
20
21# Max length of PWD to display
22MAX_PWD_LENGTH=20
23
24# Max length of Git Hex to display
25MAX_GIT_HEX_LENGTH=5
26
Ryan06d71ab2011-05-31 16:10:30 -100027GIT_THEME_PROMPT_PREFIX=' |git:'
28GIT_THEME_PROMPT_SUFFIX='|'
29
30HG_THEME_PROMPT_PREFIX=' |hg:'
31HG_THEME_PROMPT_SUFFIX='|'
32
Ryan1a83a4d2011-05-31 13:19:20 -100033# Use http://geoff.greer.fm/lscolors/
34
35# Displays the current virtualenv information
36function curr_virtualenv_info() {
37 [ ! -z "$VIRTUAL_ENV" ] && echo "`basename $VIRTUAL_ENV`"
38}
39
40# Displays the current rvm information w/gemset
41function 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 ...
52function 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
Ryan1a83a4d2011-05-31 13:19:20 -100071# SCM information
72function 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
Ryan06d71ab2011-05-31 16:10:30 -100077 [ "$SCM_CHAR" == "$SCM_SVN_CHAR" ] && echo "$prompt$(parse_svn_info)" && return
78 [ "$SCM_CHAR" == "$SCM_HG_CHAR" ] && echo "$prompt$(parse_hg_info)" && return
Ryan1a83a4d2011-05-31 13:19:20 -100079}
80
81# Parse git info
82function 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
Ryan06d71ab2011-05-31 16:10:30 -100093 echo "$prefix${ref#refs/heads/}=#${rawhex:0:$MAX_GIT_HEX_LENGTH}$state$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -100094}
95
Ryan06d71ab2011-05-31 16:10:30 -100096# Parse hg info
97function 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
Ryan1a83a4d2011-05-31 13:19:20 -1000111# Displays last X characters of pwd
112function 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
130function 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
140PROMPT_COMMAND=prompt