ej2015 · October 10, 2018 09:12 · Mar 4, 2014 · Jan 2, 2014 · Sep 19, 2011 · Jul 30, 2011
diff --git a/deploy.rb b/deploy.rb
@@ -1,27 +1,38 @@
 # encoding: UTF-8
-
 # Place in config/deploy.rb
-
+# See: https://gist.github.com/joost/9343156
+# Adapted to work with rbenv
 namespace :rails do
-  desc "Open the rails console on each of the remote servers"
+  desc "Open the rails console on primary app server"
   task :console do
-    on roles(:app) do |host| #does it for each host, bad.
+    on roles(:app), primary: true do
       rails_env = fetch(:stage)
-      execute_interactively "ruby #{current_path}/script/rails console #{rails_env}"  
+      execute_interactively "#{bundle_cmd} #{current_path}/script/rails console #{rails_env}"
     end
   end
 
-  desc "Open the rails dbconsole on each of the remote servers"
+  desc "Open the rails dbconsole on primary db server"
   task :dbconsole do
-    on roles(:db) do |host| #does it for each host, bad.
+    on roles(:db), primary: true do
       rails_env = fetch(:stage)
-      execute_interactively "ruby #{current_path}/script/rails dbconsole #{rails_env}"  
+      execute_interactively "#{bundle_cmd} #{current_path}/script/rails dbconsole #{rails_env}"
     end
   end
 
   def execute_interactively(command)
     user = fetch(:user)
     port = fetch(:port) || 22
-    exec "ssh -l #{user} #{host} -p #{port} -t 'cd #{deploy_to}/current && #{command}'"
+    cmd = "ssh -l #{user} #{host} -p #{port} -t 'cd #{deploy_to}/current && #{command}'"
+    info "Connecting to #{host}"
+    exec cmd
+  end
+
+  def bundle_cmd
+    if fetch(:rbenv_ruby)
+      # FIXME: Is there a better way to do this? How does "execute :bundle" work?
+      "RBENV_ROOT=#{fetch(:rbenv_path)} RBENV_VERSION=#{fetch(:rbenv_ruby)} #{File.join(fetch(:rbenv_path), '/bin/rbenv')} exec bundle exec"
+    else
+      "ruby "
+    end
   end
 end
diff --git a/deploy.rb b/deploy.rb
@@ -0,0 +1,27 @@
+# encoding: UTF-8
+
+# Place in config/deploy.rb
+
+namespace :rails do
+  desc "Open the rails console on each of the remote servers"
+  task :console do
+    on roles(:app) do |host| #does it for each host, bad.
+      rails_env = fetch(:stage)
+      execute_interactively "ruby #{current_path}/script/rails console #{rails_env}"  
+    end
+  end
+
+  desc "Open the rails dbconsole on each of the remote servers"
+  task :dbconsole do
+    on roles(:db) do |host| #does it for each host, bad.
+      rails_env = fetch(:stage)
+      execute_interactively "ruby #{current_path}/script/rails dbconsole #{rails_env}"  
+    end
+  end
+
+  def execute_interactively(command)
+    user = fetch(:user)
+    port = fetch(:port) || 22
+    exec "ssh -l #{user} #{host} -p #{port} -t 'cd #{deploy_to}/current && #{command}'"
+  end
+end
diff --git a/rails.rb b/rails.rb
@@ -1,11 +0,0 @@
-# encoding: UTF-8
-
-Capistrano::Configuration.instance(:must_exist).load do
-  namespace :rails do
-    desc "Open the rails console on one of the remote servers"
-    task :console, :roles => :app do
-      hostname = find_servers_for_task(current_task).first
-      exec "ssh -l #{user} #{hostname} -t 'source ~/.profile && #{current_path}/script/rails c #{rails_env}'"
-    end
-  end
-end

diff --git a/rails.rb b/rails.rb
@@ -5,7 +5,7 @@
     desc "Open the rails console on one of the remote servers"
     task :console, :roles => :app do
       hostname = find_servers_for_task(current_task).first
-      exec "ssh -l #{user} #{hostname} 'source ~/.profile && #{current_path}/script/rails c #{rails_env}'"
+      exec "ssh -l #{user} #{hostname} -t 'source ~/.profile && #{current_path}/script/rails c #{rails_env}'"
     end
   end
 end
diff --git a/rails.rb b/rails.rb
@@ -0,0 +1,11 @@
+# encoding: UTF-8
+
+Capistrano::Configuration.instance(:must_exist).load do
+  namespace :rails do
+    desc "Open the rails console on one of the remote servers"
+    task :console, :roles => :app do
+      hostname = find_servers_for_task(current_task).first
+      exec "ssh -l #{user} #{hostname} 'source ~/.profile && #{current_path}/script/rails c #{rails_env}'"
+    end
+  end
+end