allow to set capistrano role (closes #33)

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Change Log
 
+## Unreleased
+### Feature
+- Allow to set capistrano role to select server for remote console #33
+
 ## 2.2.1 (2017-09-04)
 ### Feature
 - [add rails:db alias to rails:dbconsole](https://github.com/ydkn/capistrano-rails-console/pull/32)

README.md

@@ -66,6 +66,12 @@ set :console_user, :appuser # run rails console as appuser through sudo
 set :console_user, nil # explicitly disable switching user through sudo
 ```
 
+### User
+
+```ruby
+set :console_role, :app # start remote console on primary server for this role
+```
+
 
 ## Contributing
 

lib/capistrano/rails/console/tasks.cap

@@ -9,6 +9,7 @@ namespace :load do
     # Default values
     set :console_env,  -> { fetch(:rails_env, fetch(:stage, 'production')) }
     set :console_user, -> { fetch(:app_user, nil) }
+    set :console_role, :app
   end
 end
 
@@ -18,7 +19,7 @@ namespace :rails do
     args  = []
     args << '--sandbox' if ENV.key?('sandbox') || ENV.key?('s')
 
-    run_interactively primary(:app) do
+    run_interactively primary(fetch(:console_role)) do
       within current_path do
         as user: fetch(:console_user) do
           execute(:rails, :console, fetch(:console_env), *args)
@@ -29,7 +30,7 @@ namespace :rails do
 
   desc 'Interact with a remote rails dbconsole'
   task :dbconsole do
-    run_interactively primary(:app) do
+    run_interactively primary(fetch(:console_role)) do
       within current_path do
         as user: fetch(:console_user) do
           execute(:rails, :dbconsole, '-p', '-e', fetch(:console_env))