blob: 1b0a5998cca7f097c9ecb80d5d00822de955dce6 [file] [log] [blame]
Ryan0df98ca2011-06-01 11:22:22 -10001#!/bin/bash
Ryanf3787fd2011-06-01 11:33:41 -10002#
Ryan0df98ca2011-06-01 11:22:22 -10003# This theme was obviously inspired a lot by
Ryanf3787fd2011-06-01 11:33:41 -10004#
Ryan0df98ca2011-06-01 11:22:22 -10005# - Demula theme
6#
7# which in itself was inspired by :
8#
9# - Ronacher's dotfiles (mitsuhikos) - http://github.com/mitsuhiko/dotfiles/tree/master/bash/
10# - Glenbot - http://theglenbot.com/custom-bash-shell-for-development/
11# - My extravagant zsh - http://stevelosh.com/blog/2010/02/my-extravagant-zsh-prompt/
12# - Monokai colors - http://monokai.nl/blog/2006/07/15/textmate-color-theme/
13# - Bash_it modern theme
14#
Ryanf3787fd2011-06-01 11:33:41 -100015# Hawaii50 theme supports :
16#
17# - configurable directory length
18# - hg, svn, git detection (I work in all of them)
19# - virtualenv, rvm + gemsets
Ryan0df98ca2011-06-01 11:22:22 -100020#
Ryan Kannodcb0dde2011-06-07 08:25:45 -100021# Screenshot: http://i.imgur.com/4IAMJ.png
Ryan0df98ca2011-06-01 11:22:22 -100022#
23# by Ryan Kanno <ryankanno@localkinegrinds.com>
24#
25# And yes, we code out in Hawaii. :D
26#
27# Note: I also am really new to this bash scripting game, so if you see things
28# that are flat out wrong, or if you think of something neat, just send a pull
Ryanf3787fd2011-06-01 11:33:41 -100029# request. This probably only works on a Mac - as some functions are OS
30# specific like getting ip, etc.
31#
Ryan0df98ca2011-06-01 11:22:22 -100032
33# COLORS ======================================================================
Ryan Kanno8e110772011-06-17 10:31:49 -100034ORANGE='\[\e[0;33m\]'
Ryan1a83a4d2011-05-31 13:19:20 -100035
JFSIII8b380c12011-06-18 10:57:59 -040036DEFAULT_COLOR="${white}"
Ryan1a83a4d2011-05-31 13:19:20 -100037
JFSIII8b380c12011-06-18 10:57:59 -040038USER_COLOR="${purple}"
39SUPERUSER_COLOR="${red}"
Ryan1a83a4d2011-05-31 13:19:20 -100040MACHINE_COLOR=$ORANGE
Ryan0df98ca2011-06-01 11:22:22 -100041IP_COLOR=$ORANGE
JFSIII8b380c12011-06-18 10:57:59 -040042DIRECTORY_COLOR="${green}"
Ryan1a83a4d2011-05-31 13:19:20 -100043
JFSIII8b380c12011-06-18 10:57:59 -040044VE_COLOR="${cyan}"
45RVM_COLOR="${cyan}"
Ryan1a83a4d2011-05-31 13:19:20 -100046
JFSIII8b380c12011-06-18 10:57:59 -040047REF_COLOR="${purple}"
Ryan Kanno9bd5b542011-05-31 21:23:33 -100048
Ryan1a83a4d2011-05-31 13:19:20 -100049# SCM prompts
JFSIII8b380c12011-06-18 10:57:59 -040050SCM_THEME_PROMPT_DIRTY=" ${bold_red}✗${normal}"
51SCM_THEME_PROMPT_CLEAN=" ${bold_green}✓${normal}"
Ryan09bd0ef2011-06-08 13:34:27 -100052SCM_THEME_PROMPT_PREFIX=' on '
Ryan0713fdc2011-06-08 11:32:50 -100053SCM_THEME_PROMPT_SUFFIX=''
54
Ryan Kannob96345e2011-06-09 12:18:11 -100055# rvm prompts
Ryan0713fdc2011-06-08 11:32:50 -100056RVM_THEME_PROMPT_PREFIX=''
57RVM_THEME_PROMPT_SUFFIX=''
58
Ryan Kannob96345e2011-06-09 12:18:11 -100059# virtualenv prompts
Ryan0713fdc2011-06-08 11:32:50 -100060VIRTUALENV_THEME_PROMPT_PREFIX=''
61VIRTUALENV_THEME_PROMPT_SUFFIX=''
Ryan1a83a4d2011-05-31 13:19:20 -100062
Ryan Kannob96345e2011-06-09 12:18:11 -100063# virtual prompts
64VIRTUAL_PROMPT_ENABLED=1
65
66VIRTUAL_THEME_PROMPT_PREFIX=' using '
67VIRTUAL_THEME_PROMPT_SUFFIX=''
68
Ryan1a83a4d2011-05-31 13:19:20 -100069# Max length of PWD to display
70MAX_PWD_LENGTH=20
71
72# Max length of Git Hex to display
73MAX_GIT_HEX_LENGTH=5
74
Ryan Kanno49f897a2011-06-09 13:20:41 -100075# IP address
76IP_SEPARATOR=', '
77
Ryan Kanno86aa9ff2011-07-07 22:13:29 -100078# Show IP
79IP_ENABLED=1
80
Ryan0df98ca2011-06-01 11:22:22 -100081# FUNCS =======================================================================
Ryan Kannodcb0dde2011-06-07 08:25:45 -100082
Ryan09dfe752011-05-31 19:17:10 -100083function ip {
Ryan Kanno49f897a2011-06-09 13:20:41 -100084 myip=$(curl -s checkip.dyndns.org | grep -Eo '[0-9\.]+')
85 echo -e "$(ips | sed -e :a -e '$!N;s/\n/${IP_SEPARATOR}/;ta' | sed -e 's/127\.0\.0\.1\${IP_SEPARATOR}//g'), ${myip}"
Ryan09dfe752011-05-31 19:17:10 -100086}
87
Ryan Kanno86aa9ff2011-07-07 22:13:29 -100088# Displays ip prompt
89function ip_prompt_info() {
90 if [[ $IP_ENABLED == 1 ]]; then
91 echo -e " ${DEFAULT_COLOR}(${IP_COLOR}$(ip)${DEFAULT_COLOR})"
92 fi
93}
94
Ryan Kannob96345e2011-06-09 12:18:11 -100095# Displays virtual info prompt (virtualenv/rvm)
96function virtual_prompt_info() {
Ryan0713fdc2011-06-08 11:32:50 -100097 local virtual_env_info=$(virtualenv_prompt)
98 local rvm_info=$(rvm_version_prompt)
Ryan Kannob96345e2011-06-09 12:18:11 -100099 local virtual_prompt=""
100
101 local prefix=${VIRTUAL_THEME_PROMPT_PREFIX}
102 local suffix=${VIRTUAL_THEME_PROMPT_SUFFIX}
Ryan1a83a4d2011-05-31 13:19:20 -1000103
104 # If no virtual info, just return
Ryan Kannob96345e2011-06-09 12:18:11 -1000105 [[ -z "$virtual_env_info" && -z "$rvm_info" ]] && return
Ryan Kanno0611f862011-05-31 21:38:41 -1000106
Ryan1a83a4d2011-05-31 13:19:20 -1000107 # If virtual_env info present, append to prompt
Ryan Kannob96345e2011-06-09 12:18:11 -1000108 [[ -n "$virtual_env_info" ]] && virtual_prompt="virtualenv: ${VE_COLOR}$virtual_env_info${DEFAULT_COLOR}"
Ryan1a83a4d2011-05-31 13:19:20 -1000109
Ryan Kannob96345e2011-06-09 12:18:11 -1000110 if [[ -n "$rvm_info" ]]
Ryan1a83a4d2011-05-31 13:19:20 -1000111 then
Ryan Kannob96345e2011-06-09 12:18:11 -1000112 [[ -n "$virtual_env_info" ]] && virtual_prompt="$virtual_prompt, "
113 virtual_prompt="${virtual_prompt}rvm: ${RVM_COLOR}$rvm_info${DEFAULT_COLOR}"
Ryan1a83a4d2011-05-31 13:19:20 -1000114 fi
Ryan Kannob96345e2011-06-09 12:18:11 -1000115 echo -e "$prefix$virtual_prompt$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000116}
117
Ryan1a83a4d2011-05-31 13:19:20 -1000118# Parse git info
Ryan28e2aed2011-06-08 11:57:16 -1000119function git_prompt_info() {
Ryan1a83a4d2011-05-31 13:19:20 -1000120 if [[ -n $(git status -s 2> /dev/null |grep -v ^# |grep -v "working directory clean") ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000121 state=${GIT_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan1a83a4d2011-05-31 13:19:20 -1000122 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000123 state=${GIT_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan1a83a4d2011-05-31 13:19:20 -1000124 fi
125 prefix=${GIT_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
126 suffix=${GIT_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
127 ref=$(git symbolic-ref HEAD 2> /dev/null) || return
Ryan Kannob96345e2011-06-09 12:18:11 -1000128 commit_id=$(git rev-parse HEAD 2>/dev/null) || return
Ryan1a83a4d2011-05-31 13:19:20 -1000129
Ryan Kannob96345e2011-06-09 12:18:11 -1000130 echo -e "$prefix${REF_COLOR}${ref#refs/heads/}${DEFAULT_COLOR}:${commit_id:0:$MAX_GIT_HEX_LENGTH}$state$suffix"
Ryan1a83a4d2011-05-31 13:19:20 -1000131}
132
Ryan06d71ab2011-05-31 16:10:30 -1000133# Parse hg info
Ryan28e2aed2011-06-08 11:57:16 -1000134function hg_prompt_info() {
Ryanaa2f1cb2011-05-31 18:41:07 -1000135 if [[ -n $(hg status 2> /dev/null) ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000136 state=${HG_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan06d71ab2011-05-31 16:10:30 -1000137 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000138 state=${HG_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan06d71ab2011-05-31 16:10:30 -1000139 fi
140 prefix=${HG_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
141 suffix=${HG_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
Ryanaa2f1cb2011-05-31 18:41:07 -1000142 branch=$(hg summary 2> /dev/null | grep branch | awk '{print $2}')
143 changeset=$(hg summary 2> /dev/null | grep parent | awk '{print $2}')
Ryan06d71ab2011-05-31 16:10:30 -1000144
Ryan Kannob96345e2011-06-09 12:18:11 -1000145 echo -e "$prefix${REF_COLOR}${branch}${DEFAULT_COLOR}:${changeset#*:}$state$suffix"
Ryan06d71ab2011-05-31 16:10:30 -1000146}
147
Ryan2a6d1da2011-05-31 19:03:28 -1000148# Parse svn info
Ryan28e2aed2011-06-08 11:57:16 -1000149function svn_prompt_info() {
Ryan2a6d1da2011-05-31 19:03:28 -1000150 if [[ -n $(svn status --ignore-externals -q 2> /dev/null) ]]; then
Ryan Kannob96345e2011-06-09 12:18:11 -1000151 state=${SVN_THEME_PROMPT_DIRTY:-$SCM_THEME_PROMPT_DIRTY}
Ryan2a6d1da2011-05-31 19:03:28 -1000152 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000153 state=${SVN_THEME_PROMPT_CLEAN:-$SCM_THEME_PROMPT_CLEAN}
Ryan2a6d1da2011-05-31 19:03:28 -1000154 fi
155 prefix=${SVN_THEME_PROMPT_PREFIX:-$SCM_THEME_PROMPT_PREFIX}
156 suffix=${SVN_THEME_PROMPT_SUFFIX:-$SCM_THEME_PROMPT_SUFFIX}
157 ref=$(svn info 2> /dev/null | awk -F/ '/^URL:/ { for (i=0; i<=NF; i++) { if ($i == "branches" || $i == "tags" ) { print $(i+1); break }; if ($i == "trunk") { print $i; break } } }') || return
Ryan2a6d1da2011-05-31 19:03:28 -1000158 [[ -z $ref ]] && return
Ryan Kannob96345e2011-06-09 12:18:11 -1000159
160 revision=$(svn info 2> /dev/null | sed -ne 's#^Revision: ##p' )
161
Ryan Kanno9bd5b542011-05-31 21:23:33 -1000162 echo -e "$prefix${REF_COLOR}$ref${DEFAULT_COLOR}:$revision$state$suffix"
Ryan2a6d1da2011-05-31 19:03:28 -1000163}
Ryan06d71ab2011-05-31 16:10:30 -1000164
Ryan1a83a4d2011-05-31 13:19:20 -1000165# Displays last X characters of pwd
166function limited_pwd() {
167
168 # Replace $HOME with ~ if possible
169 RELATIVE_PWD=${PWD/#$HOME/\~}
170
171 local offset=$((${#RELATIVE_PWD}-$MAX_PWD_LENGTH))
172
173 if [ $offset -gt "0" ]
174 then
175 local truncated_symbol="..."
176 TRUNCATED_PWD=${RELATIVE_PWD:$offset:$MAX_PWD_LENGTH}
Ryan Kannob96345e2011-06-09 12:18:11 -1000177 echo -e "${truncated_symbol}/${TRUNCATED_PWD#*/}"
Ryan1a83a4d2011-05-31 13:19:20 -1000178 else
Ryan Kannob96345e2011-06-09 12:18:11 -1000179 echo -e "${RELATIVE_PWD}"
Ryan1a83a4d2011-05-31 13:19:20 -1000180 fi
181}
182
183# Displays the current prompt
184function prompt() {
Ryan Kannob96345e2011-06-09 12:18:11 -1000185 local UC=$USER_COLOR
186 [ $UID -eq "0" ] && UC=$SUPERUSER_COLOR
Ryan1a83a4d2011-05-31 13:19:20 -1000187
Ryan Kannob96345e2011-06-09 12:18:11 -1000188 if [[ $VIRTUAL_PROMPT_ENABLED == 1 ]]; then
Ryan Kanno86aa9ff2011-07-07 22:13:29 -1000189 PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h$(ip_prompt_info) ${DEFAULT_COLOR}in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR}$(virtual_prompt_info)$(scm_prompt_info)${reset_color} \$ "
Ryan Kannob96345e2011-06-09 12:18:11 -1000190 else
Ryan Kanno86aa9ff2011-07-07 22:13:29 -1000191 PS1="$(scm_char) ${UC}\u ${DEFAULT_COLOR}at ${MACHINE_COLOR}\h$(ip_prompt_info) ${DEFAULT_COLOR}in ${DIRECTORY_COLOR}$(limited_pwd)${DEFAULT_COLOR}$(scm_prompt_info)${reset_color} \$ "
Ryan Kannob96345e2011-06-09 12:18:11 -1000192 fi
193 PS2='> '
194 PS4='+ '
Ryan1a83a4d2011-05-31 13:19:20 -1000195}
196
197PROMPT_COMMAND=prompt