blob: a81cc0b9241503b3f69bc808b546202fcc05f061 [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
27# Use http://geoff.greer.fm/lscolors/
28
29# Displays the current virtualenv information
30function curr_virtualenv_info() {
31 [ ! -z "$VIRTUAL_ENV" ] && echo "`basename $VIRTUAL_ENV`"
32}
33
34# Displays the current rvm information w/gemset
35function curr_rvm_info() {
36 local ruby_version=$(echo $MY_RUBY_HOME | awk -F'-' '{print $2}')
37 local ruby_gemset=$(echo $GEM_HOME | awk -F'@' '{print $2}')
38
39 if [ "$ruby_version" != "" ]; then
40 [ "$ruby_gemset" != "" ] && ruby_gemset="@$ruby_gemset"
41 echo "$ruby_version$ruby_gemset"
42 fi
43}
44
45# Displays using ...
46function virtual_info() {
47 local virtual_env_info=$(curr_virtualenv_info)
48 local rvm_info=$(curr_rvm_info)
49 local prompt="using"
50
51 # If no virtual info, just return
52 [ "$virtual_env_info" == "" -a "$rvm_info" == "" ] && return
53
54 # If virtual_env info present, append to prompt
55 [ "$virtual_env_info" != "" ] && prompt="$prompt virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}"
56
57 if [ "$rvm_info" != "" ]
58 then
59 [ "$virtual_env_info" != "" ] && prompt="$prompt,"
60 prompt="$prompt rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}"
61 fi
62 echo $prompt
63}
64
65
66# SCM information
67function scm_info() {
68 SCM_CHAR=$(scm_char)
69 [ "$SCM_CHAR" == "$SCM_NONE_CHAR" ] && return
70 local prompt="on"
71 [ "$SCM_CHAR" == "$SCM_GIT_CHAR" ] && echo "$prompt$(parse_git_info)" && return
72}
73
74# Parse git info
75function parse_git_info() {
76 if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then
77 state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
78 else
79 state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
80 fi
81 prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
82 suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
83 ref=$(git symbolic-ref HEAD 2> /dev/null) || return
84 rawhex=$(git rev-parse HEAD 2>/dev/null) || return
85
86 echo -e "$prefix${ref#refs/heads/} (${rawhex:0:$MAX_GIT_HEX_LENGTH})$state$suffix"
87}
88
89# Displays last X characters of pwd
90function limited_pwd() {
91
92 # Replace $HOME with ~ if possible
93 RELATIVE_PWD=${PWD/#$HOME/\~}
94
95 local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH))
96
97 if [ $offset -gt "0" ]
98 then
99 local truncated_symbol="..."
100 TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH}
101 echo "${truncated_symbol}/${TRUNCATED_PWD#*/}"
102 else
103 echo "${RELATIVE_PWD}"
104 fi
105}
106
107# Displays the current prompt
108function prompt() {
109
110 local UC=$USER_COLOR
111 [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR
112
113 PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h${DEFAULT_COLOR} in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR} $(virtual_info) $(scm_info) \$ "
114 PS2='> '
115 PS4='+ '
116}
117
118PROMPT_COMMAND=prompt