From be6c0320252d2c750884a90664a26b58eb824920 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christian=20H=C3=B6ltje?= <docwhat@gerf.org>
Date: Wed, 26 Jul 2017 23:39:49 -0400
Subject: [PATCH] rvm: handle -head, etc. more gracefully.

Closes #575
---
 powerlevel9k.zsh-theme | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/powerlevel9k.zsh-theme b/powerlevel9k.zsh-theme
index 16904af7..d6ce38f1 100755
--- a/powerlevel9k.zsh-theme
+++ b/powerlevel9k.zsh-theme
@@ -1070,13 +1070,10 @@ prompt_rspec_stats() {
 
 # Ruby Version Manager information
 prompt_rvm() {
-  local gemset=$(echo $GEM_HOME | awk -F'@' '{print $2}')
-  [ "$gemset" != "" ] && gemset="@$gemset"
+  local version_and_gemset=${rvm_env_string/ruby-}
 
-  local version=$(echo $MY_RUBY_HOME | awk -F'-' '{print $NF}')
-
-  if [[ -n "$version$gemset" ]]; then
-    "$1_prompt_segment" "$0" "$2" "240" "$DEFAULT_COLOR" "$version$gemset" 'RUBY_ICON'
+  if [[ -n "$version_and_gemset" ]]; then
+    "$1_prompt_segment" "$0" "$2" "240" "$DEFAULT_COLOR" "$version_and_gemset" 'RUBY_ICON'
   fi
 }