Merge pull request #41 from cheald/master

Allow for overriding of RVM roles

README.md

@@ -64,12 +64,16 @@ or
 
     set :rvm_ruby_version, '2.0.0@mygemset'
 
-
 ### Custom RVM path: `:rvm_custom_path`
 
 If you have a custom RVM setup with a different path then expected, you have
 to define a custom RVM path to tell capistrano where it is.
 
+### Custom Roles: `:rvm_roles`
+
+If you want to restrict RVM usage to a subset of roles, you may set `:rvm_roles`:
+
+    set :rvm_roles, [:app, :web]
 
 ## Restrictions
 

lib/capistrano/tasks/rvm.rake

@@ -4,15 +4,15 @@ RVM_USER_PATH = "~/.rvm"
 namespace :rvm do
   desc "Prints the RVM and Ruby version on the target host"
   task :check do
-    on roles(:all) do
+    on roles(fetch(:rvm_roles, :all)) do
       puts capture(:rvm, "version")
       puts capture(:rvm, "current")
       puts capture(:ruby, "--version")
     end
   end
 
   task :hook do
-    on roles(:all) do
+    on roles(fetch(:rvm_roles, :all)) do
       rvm_path = fetch(:rvm_custom_path)
       rvm_path ||= case fetch(:rvm_type)
       when :auto