Reorganize methods

lib/capistrano/tasks/sneakers.rake

 namespace :load do
 namespace :load do
   task :defaults do
   task :defaults do
-    set :sneakers_default_hooks, -> { true }
+    set :sneakers_default_hooks, true
 
 
     set :sneakers_pid, -> { File.join(shared_path, 'tmp', 'pids', 'sneakers.pid') }
     set :sneakers_pid, -> { File.join(shared_path, 'tmp', 'pids', 'sneakers.pid') }
     set :sneakers_env, -> { fetch(:rack_env, fetch(:rails_env, fetch(:stage))) }
     set :sneakers_env, -> { fetch(:rack_env, fetch(:rails_env, fetch(:stage))) }
@@ -29,96 +29,6 @@ namespace :deploy do
 end
 end
 
 
 namespace :sneakers do
 namespace :sneakers do
-  def for_each_sneakers_process(reverse = false, &block)
-    pids = processes_sneakers_pids
-    pids.reverse! if reverse
-    pids.each_with_index do |pid_file, idx|
-      within current_path do
-        yield(pid_file, idx)
-      end
-    end
-  end
-
-  def processes_sneakers_pids
-    pids = []
-    raise "sneaker_processes is nil class, cannot continue, please [set :sneaker_processes]" if fetch(:sneakers_processes).nil?
-    fetch(:sneakers_processes).times do |idx|
-      pids.push (idx.zero? && fetch(:sneakers_processes) <= 1) ?
-                    fetch(:sneakers_pid) :
-                    fetch(:sneakers_pid).gsub(/\.pid$/, "-#{idx}.pid")
-
-    end
-    pids
-  end
-
-  def sneakers_pid_process_exists?(pid_file)
-    sneakers_pid_file_exists?(pid_file) and test(:kill, "-0 $( cat #{pid_file} )")
-  end
-
-  def sneakers_pid_file_exists?(pid_file)
-    test(*("[ -f #{pid_file} ]").split(' '))
-  end
-
-  def stop_sneakers(pid_file)
-    if fetch(:sneakers_run_config) == true
-      execute :kill, "-SIGTERM `cat #{pid_file}`"
-    else
-      if fetch(:stop_sneakers_in_background, fetch(:sneakers_run_in_background))
-        if fetch(:sneakers_use_signals)
-          background :kill, "-TERM `cat #{pid_file}`"
-        else
-          background :bundle, :exec, :sneakersctl, 'stop', "#{pid_file}", fetch(:sneakers_timeout)
-        end
-      else
-        execute :bundle, :exec, :sneakersctl, 'stop', "#{pid_file}", fetch(:sneakers_timeout)
-      end
-    end
-  end
-
-  def quiet_sneakers(pid_file)
-    if fetch(:sneakers_use_signals) || fetch(:sneakers_run_config)
-      execute :kill, "-USR1 `cat #{pid_file}`"
-    else
-      begin
-        execute :bundle, :exec, :sneakersctl, 'quiet', "#{pid_file}"
-      rescue SSHKit::Command::Failed
-        # If gems are not installed eq(first deploy) and sneakers_default_hooks as active
-        warn 'sneakersctl not found (ignore if this is the first deploy)'
-      end
-    end
-  end
-
-  def start_sneakers(pid_file, idx = 0)
-    if fetch(:sneakers_run_config) == true
-      # Use sneakers configuration prebuilt in
-      raise "[ set :workers, ['worker1', 'workerN'] ] not configured properly, please configure the workers you wish to use" if fetch(:sneakers_workers).nil? or fetch(:sneakers_workers) == false or !fetch(:sneakers_workers).kind_of? Array
-
-      workers = fetch(:sneakers_workers).compact.join(',')
-
-      info "Starting the sneakers processes"
-
-      with rails_env: fetch(:sneakers_env), workers: workers do
-        rake 'sneakers:run'
-      end
-    end
-  end
-
-  def as_sneakers_user(role, &block)
-    user = sneakers_user(role)
-    if user == role.user
-      block.call
-    else
-      as user do
-        block.call
-      end
-    end
-  end
-
-  def sneakers_user(role)
-    properties = role.properties
-    properties.fetch(:sneakers_user) || fetch(:sneakers_user) || properties.fetch(:run_as) || role.user
-  end
-
   task :add_default_hooks do
   task :add_default_hooks do
     after 'deploy:starting',  'sneakers:quiet'
     after 'deploy:starting',  'sneakers:quiet'
     after 'deploy:updated',   'sneakers:stop'
     after 'deploy:updated',   'sneakers:stop'
@@ -128,11 +38,11 @@ namespace :sneakers do
 
 
   desc 'Quiet sneakers (stop processing new tasks)'
   desc 'Quiet sneakers (stop processing new tasks)'
   task :quiet do
   task :quiet do
-    on roles fetch(:sneakers_role) do |role|
-      as_sneakers_user(role) do
-        if test("[ -d #{current_path} ]") # fixes #11
-          for_each_sneakers_process(true) do |pid_file, idx|
-            if sneakers_pid_process_exists?(pid_file)
+    on roles fetch(:sneakers_roles) do |role|
+      switch_user(role) do
+        if test("[ -d #{current_path} ]")
+          each_process_with_index(true) do |pid_file, idx|
+            if pid_file_exists?(pid_file) && process_exists?(pid_file)
               quiet_sneakers(pid_file)
               quiet_sneakers(pid_file)
             end
             end
           end
           end
@@ -143,11 +53,11 @@ namespace :sneakers do
 
 
   desc 'Stop sneakers'
   desc 'Stop sneakers'
   task :stop do
   task :stop do
-    on roles fetch(:sneakers_role) do |role|
-      as_sneakers_user(role) do
+    on roles fetch(:sneakers_roles) do |role|
+      switch_user(role) do
         if test("[ -d #{current_path} ]")
         if test("[ -d #{current_path} ]")
-          for_each_sneakers_process(true) do |pid_file, idx|
-            if sneakers_pid_process_exists?(pid_file)
+          each_process_with_index(true) do |pid_file, idx|
+            if pid_file_exists?(pid_file) && process_exists?(pid_file)
               stop_sneakers(pid_file)
               stop_sneakers(pid_file)
             end
             end
           end
           end
@@ -158,10 +68,10 @@ namespace :sneakers do
 
 
   desc 'Start sneakers'
   desc 'Start sneakers'
   task :start do
   task :start do
-    on roles fetch(:sneakers_role) do |role|
-      as_sneakers_user(role) do
-        for_each_sneakers_process do |pid_file, idx|
-          unless sneakers_pid_process_exists?(pid_file)
+    on roles fetch(:sneakers_roles) do |role|
+      switch_user(role) do
+        each_process_with_index do |pid_file, idx|
+          unless pid_file_exists?(pid_file) && process_exists?(pid_file)
             start_sneakers(pid_file, idx)
             start_sneakers(pid_file, idx)
           end
           end
         end
         end
@@ -180,10 +90,10 @@ namespace :sneakers do
 
 
   desc 'Rolling-restart sneakers'
   desc 'Rolling-restart sneakers'
   task :rolling_restart do
   task :rolling_restart do
-    on roles fetch(:sneakers_role) do |role|
-      as_sneakers_user(role) do
-        for_each_sneakers_process(true) do |pid_file, idx|
-          if sneakers_pid_process_exists?(pid_file)
+    on roles fetch(:sneakers_roles) do |role|
+      switch_user(role) do
+        each_process_with_index(true) do |pid_file, idx|
+          if pid_file_exists?(pid_file) && process_exists?(pid_file)
             stop_sneakers(pid_file)
             stop_sneakers(pid_file)
           end
           end
           start_sneakers(pid_file, idx)
           start_sneakers(pid_file, idx)
@@ -194,11 +104,13 @@ namespace :sneakers do
 
 
   # Delete any pid file not in use
   # Delete any pid file not in use
   task :cleanup do
   task :cleanup do
-    on roles fetch(:sneakers_role) do |role|
-      as_sneakers_user(role) do
-        for_each_sneakers_process do |pid_file, idx|
-          if sneakers_pid_file_exists?(pid_file)
-            execute "rm #{pid_file}" unless sneakers_pid_process_exists?(pid_file)
+    on roles fetch(:sneakers_roles) do |role|
+      switch_user(role) do
+        each_process_with_index do |pid_file, idx|
+          unless process_exists?(pid_file)
+            if pid_file_exists?(pid_file)
+              execute "rm #{pid_file}"
+            end
           end
           end
         end
         end
       end
       end
@@ -209,14 +121,105 @@ namespace :sneakers do
   desc 'Respawn missing sneakers proccesses'
   desc 'Respawn missing sneakers proccesses'
   task :respawn do
   task :respawn do
     invoke 'sneakers:cleanup'
     invoke 'sneakers:cleanup'
-    on roles fetch(:sneakers_role) do |role|
-      as_sneakers_user(role) do
-        for_each_sneakers_process do |pid_file, idx|
-          unless sneakers_pid_file_exists?(pid_file)
+    on roles fetch(:sneakers_roles) do |role|
+      switch_user(role) do
+        each_process_with_index do |pid_file, idx|
+          unless pid_file_exists?(pid_file)
             start_sneakers(pid_file, idx)
             start_sneakers(pid_file, idx)
           end
           end
         end
         end
       end
       end
     end
     end
   end
   end
+
+  def each_process_with_index(reverse = false, &block)
+    _pid_files = pid_files
+    _pid_files.reverse! if reverse
+    _pid_files.each_with_index do |pid_file, idx|
+      within release_path do
+        yield(pid_file, idx)
+      end
+    end
+  end
+
+  def pid_files
+    sneakers_roles = Array(fetch(:sneakers_roles))
+    sneakers_roles.select! { |role| host.roles.include?(role) }
+    sneakers_roles.flat_map do |role|
+      processes = fetch(:sneakers_processes)
+      if processes == 1
+        fetch(:sneakers_pid)
+      else
+        Array.new(processes) { |idx| fetch(:sneakers_pid).gsub(/\.pid$/, "-#{idx}.pid") }
+      end
+    end
+  end
+
+  def pid_file_exists?(pid_file)
+    test(*("[ -f #{pid_file} ]").split(' '))
+  end
+
+  def process_exists?(pid_file)
+    test(*("kill -0 $( cat #{pid_file} )").split(' '))
+  end
+
+  def quiet_sneakers(pid_file)
+    if fetch(:sneakers_use_signals) || fetch(:sneakers_run_config)
+      execute :kill, "-USR1 `cat #{pid_file}`"
+    else
+      begin
+        execute :bundle, :exec, :sneakersctl, 'quiet', "#{pid_file}"
+      rescue SSHKit::Command::Failed
+        # If gems are not installed eq(first deploy) and sneakers_default_hooks as active
+        warn 'sneakersctl not found (ignore if this is the first deploy)'
+      end
+    end
+  end
+
+  def stop_sneakers(pid_file)
+    if fetch(:sneakers_run_config) == true
+      execute :kill, "-SIGTERM `cat #{pid_file}`"
+    else
+      if fetch(:stop_sneakers_in_background, fetch(:sneakers_run_in_background))
+        if fetch(:sneakers_use_signals)
+          background :kill, "-TERM `cat #{pid_file}`"
+        else
+          background :bundle, :exec, :sneakersctl, 'stop', "#{pid_file}", fetch(:sneakers_timeout)
+        end
+      else
+        execute :bundle, :exec, :sneakersctl, 'stop', "#{pid_file}", fetch(:sneakers_timeout)
+      end
+    end
+  end
+
+  def start_sneakers(pid_file, idx = 0)
+    if fetch(:sneakers_run_config) == true
+      # Use sneakers configuration prebuilt in
+      raise "[ set :workers, ['worker1', 'workerN'] ] not configured properly, please configure the workers you wish to use" if fetch(:sneakers_workers).nil? or fetch(:sneakers_workers) == false or !fetch(:sneakers_workers).kind_of? Array
+
+      workers = fetch(:sneakers_workers).compact.join(',')
+
+      info "Starting the sneakers processes"
+
+      with rails_env: fetch(:sneakers_env), workers: workers do
+        rake 'sneakers:run'
+      end
+    end
+  end
+
+  def switch_user(role, &block)
+    user = sneakers_user(role)
+    if user == role.user
+      block.call
+    else
+      as user do
+        block.call
+      end
+    end
+  end
+
+  def sneakers_user(role)
+    properties = role.properties
+    properties.fetch(:sneakers_user) || fetch(:sneakers_user) || properties.fetch(:run_as) || role.user
+  end
 end
 end