Merge pull request b-ryan#81 from flyinprogramer/master

Adds support for gemsets which are a thing for RVM users
amtrivedi91 · Jul 16, 2013 · f4241ac · f4241ac
2 parents ef6ae51 + f7f0cf9
commit f4241ac
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/segments/ruby_version.py b/segments/ruby_version.py
@@ -6,6 +6,10 @@ def add_ruby_version_segment():
         p1 = subprocess.Popen(["ruby", "-v"], stdout=subprocess.PIPE)
         p2 = subprocess.Popen(["sed", "s/ (.*//"], stdin=p1.stdout, stdout=subprocess.PIPE)
         version = p2.communicate()[0].rstrip()
+        if os.environ.has_key("GEM_HOME"):
+          gem = os.environ["GEM_HOME"].split("@")
+          if len(gem) > 1:
+            version += " " + gem[1]
         powerline.append(version, 15, 1)
     except OSError:
         return