# ------------------------------------------------------------------------
# Lori Holden oh-my-zsh theme
# Git and RVM information on the prompt
# ------------------------------------------------------------------------

function git_custom_branch() {
  local cb=$(current_branch)
  if [ -n "$cb" ]; then
    echo "$ZSH_THEME_GIT_PROMPT_BRANCH_PREFIX${cb}$ZSH_THEME_GIT_PROMPT_BRANCH_SUFFIX"
  fi
}

function rvm_custom_prompt {
  local rvmp=$(~/.rvm/bin/rvm-prompt i v g s 2> /dev/null) || return
  echo "$ZSH_THEME_RVM_PROMPT_BRANCH_PREFIX${rvmp}$ZSH_THEME_RVM_PROMPT_BRANCH_SUFFIX"
}

ZSH_THEME_RVM_PROMPT_BRANCH_PREFIX=" $reset_color$fg[magenta]["
ZSH_THEME_RVM_PROMPT_BRANCH_SUFFIX="]"
ZSH_THEME_GIT_PROMPT_BRANCH_PREFIX=" $reset_color$fg[magenta]["
ZSH_THEME_GIT_PROMPT_BRANCH_SUFFIX=""
ZSH_THEME_GIT_PROMPT_SHA_BEFORE="@"
ZSH_THEME_GIT_PROMPT_SHA_AFTER="]"
ZSH_THEME_GIT_PROMPT_DIRTY="$reset_color$fg_bold[red]*"
ZSH_THEME_GIT_PROMPT_CLEAN=""
ZSH_THEME_GIT_PROMPT_AHEAD="$reset_color$fg_bold[red]!"

# Prompt format
PROMPT='( $fg[white]%n@%m$fg[white]:$(parse_git_dirty)$(git_prompt_ahead)$reset_color$fg_bold[blue]%~%u$(git_custom_branch)$(git_prompt_short_sha)$(rvm_custom_prompt)$reset_color )
$fg_bold[blue]%%$reset_color '