process_bot 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f1bb2b0296a51e0b52b952f49f6724d95d7478181539b75fac529ff3ab3d82de
-  data.tar.gz: d7ec454928d4e72f40e9a5d5cf05e2fff930cf73de754c47fb344209173294aa
+  metadata.gz: 918adb2b2275f38d3b22eb926a5da3b021f667d4a7290a3b0b1d9ea7bb56a2c2
+  data.tar.gz: 03c557694d0a91f78f25dc4689f2b99c46b626ebeed5816f8e2e0346e29dd482
 SHA512:
-  metadata.gz: c0183b4d9a6f4023393467dc1fb794b2c1fc1c30d481a24b1cbb9728f714be5b2bd1f5e2620c71e60394e59aa79a5fef65f42cc152553007939dda0244635507
-  data.tar.gz: 8ee7928cf14abedbda792e50653fad619c9ac89196dc7effb0869fcda37141dd637fd63085075e7068cec50d1d903ff69b6216fb446939405654683b000bf4bc
+  metadata.gz: 2053d076158df45b723855cee991d47852efb18b34ace65e24005798c7d9d1f1ffff126d1fc7409ba3554e4bfa3bf9855c13dfec5bcfa14ac79aac57d9409acf
+  data.tar.gz: 5622bc3069e029485876c7329bd0658b3bf7b3c4b265a40e51421bab04e0b898367f91c864dbe7caf1de93b317e9ba1fcd4787971fd4dc9e28528bf4f14a8df8

data/.rubocop.yml

@@ -1,13 +1,189 @@
 AllCops:
+  DisplayCopNames: true
+  DisplayStyleGuide: true
+  NewCops: enable
   TargetRubyVersion: 2.6
 
-Style/StringLiterals:
+require:
+  - rubocop-performance
+  - rubocop-rake
+  - rubocop-rspec
+
+Layout/AccessModifierIndentation:
+  EnforcedStyle: outdent
+
+Layout/ArgumentAlignment:
+  EnforcedStyle: with_fixed_indentation
+
+Layout/CaseIndentation:
+  EnforcedStyle: end
+
+Layout/EmptyLines:
+  Enabled: false
+
+Layout/EmptyLinesAroundArguments:
+  Enabled: false
+
+Layout/EndAlignment:
+  EnforcedStyleAlignWith: variable
+
+Layout/LineEndStringConcatenationIndentation:
+  EnforcedStyle: indented
+
+Layout/LineLength:
+  Max: 160
+
+Layout/MultilineMethodCallIndentation:
+  EnforcedStyle: indented
+
+Layout/MultilineOperationIndentation:
+  EnforcedStyle: indented
+
+Layout/ParameterAlignment:
+  EnforcedStyle: with_fixed_indentation
+
+Layout/RescueEnsureAlignment:
+  Enabled: false
+
+Layout/SpaceAroundMethodCallOperator:
+  Enabled: true
+
+Layout/SpaceInsideHashLiteralBraces:
+  EnforcedStyle: no_space
+
+Lint/MissingSuper:
+  Enabled: false
+
+Lint/RaiseException:
+  Enabled: true
+
+Lint/StructNewOverride:
+  Enabled: true
+
+# Metrics/AbcSize:
+#   Max: 25
+
+Metrics/BlockLength:
+  Enabled: false
+
+Metrics/ClassLength:
+  Max: 250
+
+Metrics/CyclomaticComplexity:
+  Max: 12
+
+Metrics/MethodLength:
+  Max: 50
+
+Metrics/ParameterLists:
+  CountKeywordArgs: false
+
+Metrics/PerceivedComplexity:
+  Max: 12
+
+RSpec/AnyInstance:
+  Enabled: false
+
+Style/CaseLikeIf:
+  Enabled: false
+
+RSpec/ContextWording:
+  Enabled: false
+
+RSpec/DescribeClass:
+  Enabled: false
+
+RSpec/DescribedClass:
+  Enabled: false
+
+RSpec/ExampleLength:
+  Enabled: false
+
+RSpec/LetSetup:
+  Enabled: false
+
+RSpec/MessageSpies:
+  Enabled: false
+
+RSpec/MultipleExpectations:
+  Enabled: false
+
+RSpec/MultipleMemoizedHelpers:
+  Enabled: false
+
+RSpec/NamedSubject:
+  Enabled: false
+
+RSpec/NestedGroups:
+  Enabled: false
+
+RSpec/StubbedMock:
+  Enabled: false
+
+Style/ClassAndModuleChildren:
+  EnforcedStyle: compact
+
+Style/ConditionalAssignment:
+  Enabled: false
+
+Style/Documentation:
+  Enabled: false
+
+Style/ExponentialNotation:
+  Enabled: true
+
+Style/FrozenStringLiteralComment:
+  Enabled: false
+
+Style/HashAsLastArrayItem:
+  Enabled: false
+
+Style/HashEachMethods:
+  Enabled: true
+
+Style/HashTransformKeys:
+  Enabled: true
+
+Style/HashTransformValues:
   Enabled: true
+
+# Will report offences for many places that are much more readable without using a guard clause
+Style/GuardClause:
+  Enabled: false
+
+Style/KeywordParametersOrder:
+  Enabled: false
+
+Style/Lambda:
+  Enabled: false
+
+Style/LambdaCall:
+  Enabled: false
+
+Style/MultipleComparison:
+  Enabled: false
+
+Style/RegexpLiteral:
+  Enabled: false
+
+Style/StringLiterals:
   EnforcedStyle: double_quotes
 
 Style/StringLiteralsInInterpolation:
+  Enabled: false
+
+Style/NilComparison:
+  Enabled: false
+
+Style/SignalException:
+  EnforcedStyle: only_raise
+
+Style/SymbolArray:
+  Enabled: false
+
+Style/TrivialAccessors:
+  ExactNameMatch: true
   Enabled: true
-  EnforcedStyle: double_quotes
 
-Layout/LineLength:
-  Max: 120
+Style/WordArray:
+  Enabled: false

data/Gemfile

@@ -5,8 +5,9 @@ source "https://rubygems.org"
 # Specify your gem's dependencies in process_bot.gemspec
 gemspec
 
-gem "rake", "~> 13.0"
+gem "rake"
+gem "rspec"
+gem "rubocop"
+gem "string-cases"
 
-gem "rspec", "~> 3.0"
-
-gem "rubocop", "~> 1.21"
+gem "pry"

data/Gemfile.lock

@@ -0,0 +1,75 @@
+PATH
+  remote: .
+  specs:
+    process_bot (0.1.2)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ast (2.4.2)
+    coderay (1.1.3)
+    diff-lcs (1.5.0)
+    json (2.6.2)
+    method_source (1.0.0)
+    parallel (1.22.1)
+    parser (3.1.2.1)
+      ast (~> 2.4.1)
+    pry (0.14.1)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    rainbow (3.1.1)
+    rake (13.0.6)
+    regexp_parser (2.6.0)
+    rexml (3.2.5)
+    rspec (3.11.0)
+      rspec-core (~> 3.11.0)
+      rspec-expectations (~> 3.11.0)
+      rspec-mocks (~> 3.11.0)
+    rspec-core (3.11.0)
+      rspec-support (~> 3.11.0)
+    rspec-expectations (3.11.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.11.0)
+    rspec-mocks (3.11.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.11.0)
+    rspec-support (3.11.0)
+    rubocop (1.36.0)
+      json (~> 2.3)
+      parallel (~> 1.10)
+      parser (>= 3.1.2.1)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8, < 3.0)
+      rexml (>= 3.2.5, < 4.0)
+      rubocop-ast (>= 1.20.1, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 1.4.0, < 3.0)
+    rubocop-ast (1.21.0)
+      parser (>= 3.1.1.0)
+    rubocop-performance (1.15.0)
+      rubocop (>= 1.7.0, < 2.0)
+      rubocop-ast (>= 0.4.0)
+    rubocop-rake (0.6.0)
+      rubocop (~> 1.0)
+    rubocop-rspec (2.11.1)
+      rubocop (~> 1.19)
+    ruby-progressbar (1.11.0)
+    string-cases (0.0.4)
+    unicode-display_width (2.3.0)
+
+PLATFORMS
+  x86_64-linux
+
+DEPENDENCIES
+  process_bot!
+  pry
+  rake
+  rspec
+  rubocop
+  rubocop-performance
+  rubocop-rake
+  rubocop-rspec
+  string-cases
+
+BUNDLED WITH
+   2.3.8

data/exe/process_bot

@@ -0,0 +1,36 @@
+#!/usr/bin/env ruby
+
+require "pathname"
+
+ENV["BUNDLE_GEMFILE"] ||= File.expand_path("../../Gemfile",
+  Pathname.new(__FILE__).realpath)
+
+require "bundler/setup"
+require "optparse"
+require "string-cases"
+require_relative "../lib/process_bot"
+
+options = ProcessBot::Options.new
+argv_i = 0
+
+while argv_i < ARGV.length
+  arg = ARGV.fetch(argv_i)
+
+  if (match = arg.match(/\A--(.+)\Z/))
+    key = match[1].tr("-", "_").to_sym
+    argv_i += 1
+    value = ARGV.fetch(argv_i)
+
+    options.set(key, value)
+  else
+    raise "Unknown option: #{arg}"
+  end
+
+  argv_i += 1
+end
+
+pp options.options
+
+ProcessBot::Process
+  .new(options)
+  .execute!

data/lib/process_bot/capistrano/puma/common.rb

@@ -0,0 +1,86 @@
+module ProcessBot::Capistrano::Puma::Common
+  def puma_switch_user(role)
+    user = puma_user(role)
+    if user == role.user
+      yield
+    else
+      backend.as user, &block
+    end
+  end
+
+  def puma_user(role)
+    properties = role.properties
+    properties.fetch(:puma_user) || # local property for puma only
+      fetch(:puma_user) ||
+      properties.fetch(:run_as) || # global property across multiple capistrano gems
+      role.user
+  end
+
+  def puma_bind
+    Array(fetch(:puma_bind)).collect do |bind|
+      "bind '#{bind}'"
+    end.join("\n")
+  end
+
+  def compiled_template_puma(from, role)
+    @role = role
+    file = [
+      "lib/capistrano/templates/#{from}-#{role.hostname}-#{fetch(:stage)}.rb",
+      "lib/capistrano/templates/#{from}-#{role.hostname}.rb",
+      "lib/capistrano/templates/#{from}-#{fetch(:stage)}.rb",
+      "lib/capistrano/templates/#{from}.rb.erb",
+      "lib/capistrano/templates/#{from}.rb",
+      "lib/capistrano/templates/#{from}.erb",
+      "config/deploy/templates/#{from}.rb.erb",
+      "config/deploy/templates/#{from}.rb",
+      "config/deploy/templates/#{from}.erb",
+      File.expand_path("../templates/#{from}.erb", __FILE__),
+      File.expand_path("../templates/#{from}.rb.erb", __FILE__)
+    ].detect { |path| File.file?(path) }
+    erb = File.read(file)
+    StringIO.new(ERB.new(erb, trim_mode: "-").result(binding))
+  end
+
+  def template_puma(from, to, role)
+    backend.upload! compiled_template_puma(from, role), to
+  end
+
+  PumaBind = Struct.new(:full_address, :kind, :address) do
+    def unix?
+      kind == :unix
+    end
+
+    def ssl?
+      kind == :ssl
+    end
+
+    def tcp
+      kind == :tcp || ssl?
+    end
+
+    def local
+      if unix?
+        self
+      else
+        PumaBind.new(
+          localize_address(full_address),
+          kind,
+          localize_address(address)
+        )
+      end
+    end
+
+  private
+
+    def localize_address(address)
+      address.gsub(/0\.0\.0\.0(.+)/, "127.0.0.1\\1")
+    end
+  end
+
+  def puma_binds
+    Array(fetch(:puma_bind)).map do |m|
+      etype, address  = /(tcp|unix|ssl):\/{1,2}(.+)/.match(m).captures
+      PumaBind.new(m, etype.to_sym, address)
+    end
+  end
+end

data/lib/process_bot/capistrano/puma.rake

@@ -6,7 +6,7 @@ namespace :process_bot do
     task :start do
       on roles(fetch(:puma_role)) do |role|
         git_plugin.puma_switch_user(role) do
-          if test "[ -f #{fetch(:puma_pid)} ]" and test :kill, "-0 $( cat #{fetch(:puma_pid)} )"
+          if test("[ -f #{fetch(:puma_pid)} ]") && test(:kill, "-0 $( cat #{fetch(:puma_pid)} )")
             info "Puma is already running"
           else
             within current_path do
@@ -24,7 +24,9 @@ namespace :process_bot do
                   "--control-token foobar"
                 ]
 
-                command = "/usr/bin/screen -dmS puma-#{latest_release_version} bash -c 'cd #{release_path} && #{SSHKit.config.command_map.prefix[:puma].join(" ")} puma #{puma_args.join(" ")}'"
+                command = "/usr/bin/screen -dmS puma-#{latest_release_version} " \
+                  "bash -c 'cd #{release_path} && #{SSHKit.config.command_map.prefix[:puma].join(" ")} puma #{puma_args.join(" ")}'"
+
                 execute command
               end
             end
@@ -36,7 +38,7 @@ namespace :process_bot do
     %w[halt stop status].map do |command|
       desc "#{command} puma"
       task command do
-        on roles (fetch(:puma_role)) do |role|
+        on roles(fetch(:puma_role)) do |role|
           within current_path do
             git_plugin.puma_switch_user(role) do
               with rack_env: fetch(:puma_env) do
@@ -48,7 +50,7 @@ namespace :process_bot do
                     execute :rm, fetch(:puma_pid)
                   end
                 else
-                  #pid file not found, so puma is probably not running or it using another pidfile
+                  # pid file not found, so puma is probably not running or it using another pidfile
                   warn "Puma not running"
                 end
               end
@@ -61,12 +63,12 @@ namespace :process_bot do
     %w[phased-restart restart].map do |command|
       desc "#{command} puma"
       task command do
-        on roles (fetch(:puma_role)) do |role|
+        on roles(fetch(:puma_role)) do |role|
           within current_path do
             git_plugin.puma_switch_user(role) do
               with rack_env: fetch(:puma_env) do
                 if git_plugin.puma_running?
-                  # NOTE pid exist but state file is nonsense, so ignore that case
+                  # NOTE: pid exist but state file is nonsense, so ignore that case
                   git_plugin.run_puma_command(command)
                 else
                   # Puma is not running or state file is not present : Run it
@@ -79,6 +81,7 @@ namespace :process_bot do
       end
     end
 
+    desc "Restarts Puma phased if using workers and preload and otherwise a normal restart."
     task :smart_restart do
       if !fetch(:puma_preload_app) && fetch(:puma_workers, 0).to_i > 1
         invoke "process_bot:puma:phased-restart"

data/lib/process_bot/capistrano/puma.rb

@@ -1,12 +1,10 @@
-module ProcessBot::Capistrano::Puma < Capistrano::Plugin
-  include PumaCommon
+class ProcessBot::Capistrano::Puma < Capistrano::Plugin
+  autoload :Common, "#{__dir__}/puma/common"
 
-  def register_hooks
-    after 'deploy:finished', 'process_bot:puma:smart_restart'
-  end
+  include ::ProcessBot::Capistrano::Puma::Common
 
   def define_tasks
-    eval_rakefile File.expand_path('./puma.rake', __FILE__)
+    eval_rakefile File.expand_path("./puma.rake", __dir__)
   end
 
   def puma_running?

data/lib/process_bot/capistrano/sidekiq.rake

@@ -1,6 +1,7 @@
 git_plugin = self
 
 namespace :load do
+  desc "Default variables for Sidekiq"
   task :defaults do
     set :sidekiq_default_hooks, true
 
@@ -18,17 +19,12 @@ namespace :load do
     set :chruby_map_bins, fetch(:chruby_map_bins).to_a.concat(%w[sidekiq sidekiqctl])
     # Bundler integration
     set :bundle_bins, fetch(:bundle_bins).to_a.concat(%w[sidekiq sidekiqctl])
-    # Init system integration
-    set :init_system, -> { nil }
-    # systemd integration
-    set :service_unit_name, "sidekiq-#{fetch(:stage)}.service"
-    set :upstart_service_name, "sidekiq"
   end
 end
 
 namespace :process_bot do
   namespace :sidekiq do
-    desc 'Quiet sidekiq (stop fetching new tasks from Redis)'
+    desc "Quiet sidekiq (stop fetching new tasks from Redis)"
     task :quiet do
       on roles fetch(:sidekiq_roles) do |role|
         git_plugin.switch_user(role) do
@@ -39,7 +35,7 @@ namespace :process_bot do
       end
     end
 
-    desc 'Stop sidekiq (graceful shutdown within timeout, put unfinished tasks back to Redis)'
+    desc "Stop Sidekiq (graceful shutdown within timeout, put unfinished tasks back to Redis)"
     task :stop do
       on roles fetch(:sidekiq_roles) do |role|
         git_plugin.switch_user(role) do
@@ -50,6 +46,7 @@ namespace :process_bot do
       end
     end
 
+    desc "Stops Sidekiq after a set amount of time"
     task :stop_after_time do
       on roles fetch(:sidekiq_roles) do |role|
         git_plugin.switch_user(role) do
@@ -60,7 +57,7 @@ namespace :process_bot do
       end
     end
 
-    desc 'Start sidekiq'
+    desc "Start sidekiq"
     task :start do
       on roles fetch(:sidekiq_roles) do |role|
         git_plugin.switch_user(role) do
@@ -72,7 +69,7 @@ namespace :process_bot do
       end
     end
 
-    desc 'Restart sidekiq'
+    desc "Restart sidekiq"
     task :restart do
       invoke! "process_bot:sidekiq:stop"
       invoke! "process_bot:sidekiq:start"

data/lib/process_bot/capistrano/sidekiq.rb

@@ -1,7 +1,9 @@
-module ProcessBot::Capistrano::Sidekiq < Capistrano::Plugin
-  include ProcessBot::Sidekiq::Helpers
+require_relative "sidekiq_helpers"
+
+class ProcessBot::Capistrano::Sidekiq < Capistrano::Plugin
+  include ProcessBot::Capistrano::SidekiqHelpers
 
   def define_tasks
-    eval_rakefile File.expand_path("./sidekiq.rake", __FILE__)
+    eval_rakefile File.expand_path("./sidekiq.rake", __dir__)
   end
 end

data/lib/process_bot/capistrano/sidekiq_helpers.rb

@@ -1,44 +1,36 @@
-module ProcessBot::Sidekiq::Helpers
+module ProcessBot::Capistrano::SidekiqHelpers
   def sidekiq_require
-    if fetch(:sidekiq_require)
-      "--require #{fetch(:sidekiq_require)}"
-    end
+    "--require #{fetch(:sidekiq_require)}" if fetch(:sidekiq_require)
   end
 
   def sidekiq_config
-    if fetch(:sidekiq_config)
-      "--config #{fetch(:sidekiq_config)}"
-    end
+    "--config #{fetch(:sidekiq_config)}" if fetch(:sidekiq_config)
   end
 
   def sidekiq_concurrency
-    if fetch(:sidekiq_concurrency)
-      "--concurrency #{fetch(:sidekiq_concurrency)}"
-    end
+    "--concurrency #{fetch(:sidekiq_concurrency)}" if fetch(:sidekiq_concurrency)
   end
 
   def sidekiq_queues
     Array(fetch(:sidekiq_queue)).map do |queue|
       "--queue #{queue}"
-    end.join(' ')
+    end.join(" ")
   end
 
   def sidekiq_logfile
     fetch(:sidekiq_log)
   end
 
-  def switch_user(role)
+  def switch_user(role, &block)
     su_user = sidekiq_user(role)
     if su_user == role.user
       yield
     else
-      as su_user do
-        yield
-      end
+      as su_user, &block
     end
   end
 
-  VALID_SIGNALS = ["TERM", "TSTP"]
+  VALID_SIGNALS = ["TERM", "TSTP"].freeze
   def stop_sidekiq(pid:, signal:)
     raise "Invalid PID: #{pid}" unless pid.to_s.match?(/\A\d+\Z/)
     raise "Invalid signal: #{signal}" unless VALID_SIGNALS.include?(signal)
@@ -53,7 +45,7 @@ module ProcessBot::Sidekiq::Helpers
     time = ENV["STOP_AFTER_TIME"] || fetch(:sidekiq_stop_after_time)
     raise "Invalid time: #{time}" unless time.to_s.match?(/\A\d+\Z/)
 
-    backend.execute "screen -dmS stopsidekiq#{pid} sleep #{time}; kill -#{signal} #{pid}"
+    backend.execute "screen -dmS stopsidekiq#{pid} bash -c \"sleep #{time} && kill -#{signal} #{pid}\""
   end
 
   def running_sidekiq_processes
@@ -94,23 +86,7 @@ module ProcessBot::Sidekiq::Helpers
     backend.capture(:echo, SSHKit.config.command_map[:bundle]).strip
   end
 
-  def start_sidekiq(idx = 0)
-    args = []
-    args.push "--environment #{fetch(:sidekiq_env)}"
-    #args.push "--logfile #{fetch(:sidekiq_log)}" if fetch(:sidekiq_log)
-    args.push "--require #{fetch(:sidekiq_require)}" if fetch(:sidekiq_require)
-    args.push "--tag #{fetch(:sidekiq_tag)}" if fetch(:sidekiq_tag)
-    Array(fetch(:sidekiq_queue)).each do |queue|
-      args.push "--queue #{queue}"
-    end
-    args.push "--config #{fetch(:sidekiq_config)}" if fetch(:sidekiq_config)
-    args.push "--concurrency #{fetch(:sidekiq_concurrency)}" if fetch(:sidekiq_concurrency)
-    if (process_options = fetch(:sidekiq_options_per_process))
-      args.push process_options[idx]
-    end
-    # use sidekiq_options for special options
-    args.push fetch(:sidekiq_options) if fetch(:sidekiq_options)
-
+  def start_sidekiq(idx = 0) # rubocop:disable Metrics/AbcSize
     releases = backend.capture(:ls, "-x", releases_path).split
     releases << release_timestamp.to_s if release_timestamp
     releases.uniq
@@ -118,13 +94,34 @@ module ProcessBot::Sidekiq::Helpers
     latest_release_version = releases.last
     raise "Invalid release timestamp: #{release_timestamp}" unless latest_release_version
 
+    args = [
+      "--id", "sidekiq-#{latest_release_version}-#{idx}",
+      "--handler", "sidekiq",
+      "--bundle-prefix", SSHKit.config.command_map.prefix[:bundle].join(" "),
+      "--sidekiq-environment", fetch(:sidekiq_env),
+      "--port", 7050 + idx
+    ]
+    args += ["--log-file-path", fetch(:sidekiq_log)] if fetch(:sidekiq_log)
+    args += ["--sidekiq-require", fetch(:sidekiq_require)] if fetch(:sidekiq_require)
+    args += ["--sidekiq-tag", fetch(:sidekiq_tag)] if fetch(:sidekiq_tag)
+    args += ["--sidekiq-queues", Array(fetch(:sidekiq_queue)).join(",")] if fetch(:sidekiq_queue)
+    args += ["--sidekiq-config", fetch(:sidekiq_config)] if fetch(:sidekiq_config)
+    args += ["--sidekiq-concurrency", fetch(:sidekiq_concurrency)] if fetch(:sidekiq_concurrency)
+    if (process_options = fetch(:sidekiq_options_per_process))
+      args += process_options[idx]
+    end
+    args += fetch(:sidekiq_options) if fetch(:sidekiq_options)
+
     screen_args = ["-dmS sidekiq-#{idx}-#{latest_release_version}"]
     screen_args << "-L -Logfile #{fetch(:sidekiq_log)}" if fetch(:sidekiq_log)
 
-    # command = "/usr/bin/tmux new -d -s sidekiq#{idx} '#{SSHKit.config.command_map.prefix[:sidekiq].join(" ")} sidekiq #{args.compact.join(' ')}'"
-    command = "/usr/bin/screen #{screen_args.join(" ")} bash -c 'cd #{release_path} && #{SSHKit.config.command_map.prefix[:sidekiq].join(" ")} sidekiq #{args.compact.join(' ')}'"
+    process_bot_args = args.compact.map { |arg| "\"#{arg}\"" }
+
+    command = "/usr/bin/screen #{screen_args.join(" ")} " \
+      "bash -c 'cd #{release_path} && #{SSHKit.config.command_map.prefix[:bundle].join(" ")} bundle exec process_bot #{process_bot_args.join(" ")}'"
 
     puts "WARNING: A known bug prevents Sidekiq from starting when pty is set (which it is)" if fetch(:pty)
+    puts "ProcessBot Sidekiq command: #{command}"
 
     backend.execute command
   end

data/lib/process_bot/capistrano.rb

@@ -0,0 +1,4 @@
+class ProcessBot::Capistrano
+  autoload :Puma, "#{__dir__}/capistrano/puma"
+  autoload :Sidekiq, "#{__dir__}/capistrano/sidekiq"
+end

data/lib/process_bot/control_socket.rb

@@ -0,0 +1,39 @@
+class ProcessBot::ControlSocket
+  attr_reader :options, :process, :server
+
+  def initialize(options:, process:)
+    @options = options
+    @process = process
+  end
+
+  def port
+    options.fetch(:port).to_i
+  end
+
+  def start
+    require "socket"
+
+    @server = TCPServer.new(port)
+  end
+
+  def run_client_loop
+    Thread.new do
+      client = server.accept
+
+      Thread.new do
+        handle_client(client)
+      end
+    end
+  end
+
+  def handle_client(client)
+    command = JSON.parse(client.gets)
+    type = command.fetch("type")
+
+    if type == "stop"
+      process.stop
+    else
+      client.puts(JSON.generate(type: "error", message: "Unknown type: #{type}"))
+    end
+  end
+end

data/lib/process_bot/logger.rb

@@ -0,0 +1,22 @@
+class ProcessBot::Logger
+  attr_reader :fp_log, :options
+
+  def initialize(options:)
+    @options = options
+
+    open_file
+  end
+
+  def log(output)
+    fp_log&.write(output)
+    fp_log&.flush
+  end
+
+  def log_file_path
+    options.fetch(:log_file_path)
+  end
+
+  def open_file
+    @fp_log = File.open(log_file_path, "a")
+  end
+end

data/lib/process_bot/options.rb

@@ -0,0 +1,21 @@
+class ProcessBot::Options
+  attr_reader :options
+
+  def initialize(options = {})
+    @options = options
+  end
+
+  def fetch(*args, **opts, &blk)
+    options.fetch(*args, **opts, &blk)
+  end
+
+  def present?(key)
+    return true if options.key?(key) && options[key]
+
+    false
+  end
+
+  def set(key, value)
+    options[key] = value
+  end
+end

data/lib/process_bot/process/handlers/sidekiq.rb

@@ -0,0 +1,55 @@
+class ProcessBot::Process::Handlers::Sidekiq
+  attr_reader :options
+
+  def initialize(options)
+    @options = options
+
+    set_defaults
+  end
+
+  def fetch(*args, **opts)
+    options.fetch(*args, **opts)
+  end
+
+  def set_option(key, value)
+    raise "Unknown option for Sidekiq handler: #{key}" unless options.key?(key)
+
+    set(key, value)
+  end
+
+  def set(*args, **opts)
+    options.set(*args, **opts)
+  end
+
+  def set_defaults
+    set :sidekiq_default_hooks, true
+    set :sidekiq_pid, -> { File.join(shared_path, "tmp", "pids", "sidekiq.pid") }
+    set :sidekiq_timeout, 10
+    set :sidekiq_roles, fetch(:sidekiq_role, :app)
+    set :sidekiq_processes, 1
+    set :sidekiq_options_per_process, nil
+  end
+
+  def command # rubocop:disable Metrics/AbcSize
+    args = []
+
+    options.options.each do |key, value|
+      if (match = key.to_s.match(/\Asidekiq-(.+)\Z/))
+        sidekiq_key = match[1]
+
+        if sidekiq_key == "queue"
+          value.split(",").each do |queue|
+            args.push "--queue #{value}"
+          end
+        else
+          args.push "--#{sidekiq_key} #{value}"
+        end
+      end
+    end
+
+    command = ""
+    command << "#{options.fetch(:bundle_prefix)} " if options.present?(:bundle_prefix)
+    command << "bundle exec sidekiq #{args.compact.join(' ')}"
+    command
+  end
+end

data/lib/process_bot/process/handlers.rb

@@ -0,0 +1,3 @@
+class ProcessBot::Process::Handlers
+  autoload :Sidekiq, "#{__dir__}/handlers/sidekiq"
+end

data/lib/process_bot/process/runner.rb

@@ -0,0 +1,55 @@
+class ProcessBot::Process::Runner
+  attr_reader :command, :exit_status, :logger, :monitor, :options, :stop_time
+
+  def initialize(command:, logger:, options:)
+    @command = command
+    @logger = logger
+    @monitor = Monitor.new
+    @options = options
+    @output = []
+  end
+
+  def output(output:, type:) # rubocop:disable Lint/UnusedMethodArgument
+    logger.log(output)
+  end
+
+  def run # rubocop:disable Metrics/AbcSize
+    @start_time = Time.new
+    stderr_reader, stderr_writer = IO.pipe
+
+    require "pty"
+
+    PTY.spawn(command, err: stderr_writer.fileno) do |stdout, _stdin, pid|
+      @pid = pid
+      logger.log "Command running with PID #{pid}: #{command}"
+
+      stdout_reader_thread = Thread.new do
+        stdout.each_char do |chunk|
+          monitor.synchronize do
+            output(type: :stdout, output: chunk)
+          end
+        end
+      rescue Errno::EIO
+        # Process done
+      ensure
+        status = Process::Status.wait(@pid, 0)
+
+        @exit_status = status.exitstatus
+        stderr_writer.close
+      end
+
+      stderr_reader_thread = Thread.new do
+        stderr_reader.each_char do |chunk|
+          monitor.synchronize do
+            output(type: :stderr, output: chunk)
+          end
+        end
+      end
+
+      stdout_reader_thread.join
+      stderr_reader_thread.join
+
+      @stop_time = Time.new
+    end
+  end
+end

data/lib/process_bot/process.rb

@@ -1,5 +1,52 @@
 class ProcessBot::Process
-  def initialize(command)
-    @command = command
+  autoload :Handlers, "#{__dir__}/process/handlers"
+  autoload :Runner, "#{__dir__}/process/runner"
+
+  attr_reader :options, :stopped
+
+  def initialize(options)
+    @options = options
+    @stopped = false
+  end
+
+  def logger
+    @logger ||= ProcessBot::Logger.new(options: options)
+  end
+
+  def start_control_socket
+    @control_socket = ProcessBot::ControlSocket.new(options: options, process: self)
+    @control_socket.start
+  end
+
+  def stop
+    @stopped = true
+  end
+
+  def handler_class
+    @handler_class ||= begin
+      require_relative "process/handlers/#{options.fetch(:handler)}"
+      ProcessBot::Process::Handlers.const_get(StringCases.snake_to_camel(options.fetch(:handler)))
+    end
+  end
+
+  def execute!
+    start_control_socket
+
+    loop do
+      run
+
+      if stopped
+        break
+      else
+        puts "Process stopped - starting again after 1 sec"
+        sleep 1
+      end
+    end
+  end
+
+  def run
+    handler_instance = handler_class.new(options)
+    runner = ProcessBot::Process::Runner.new(command: handler_instance.command, logger: logger, options: options)
+    runner.run
   end
 end

data/lib/process_bot/version.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module ProcessBot
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

data/lib/process_bot.rb

@@ -1,8 +1,11 @@
-# frozen_string_literal: true
-
 require_relative "process_bot/version"
 
 module ProcessBot
   class Error < StandardError; end
-  # Your code goes here...
+
+  autoload :Capistrano, "#{__dir__}/process_bot/capistrano"
+  autoload :ControlSocket, "#{__dir__}/process_bot/control_socket"
+  autoload :Logger, "#{__dir__}/process_bot/logger"
+  autoload :Options, "#{__dir__}/process_bot/options"
+  autoload :Process, "#{__dir__}/process_bot/process"
 end

data/peak_flow.yml

@@ -0,0 +1,4 @@
+rvm: true
+script:
+  - bundle exec rspec
+  - bundle exec rubocop

data/process_bot.gemspec

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require_relative "lib/process_bot/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "process_bot"
+  spec.version = ProcessBot::VERSION
+  spec.authors = ["kaspernj"]
+  spec.email = ["k@spernj.org"]
+
+  spec.summary = "Run and control processes."
+  spec.description = "Run and control processes."
+  spec.homepage = "https://github.com/kaspernj/process_bot"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 2.6.0"
+
+  spec.metadata["allowed_push_host"] = "https://rubygems.org"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/kaspernj/process_bot"
+  spec.metadata["changelog_uri"] = "https://github.com/kaspernj/process_bot/blob/master/CHANGELOG.md"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (f == __FILE__) || f.match(%r{\A(?:(?:bin|test|spec|features)/|\.(?:git|travis|circleci)|appveyor)})
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  # Uncomment to register a new dependency of your gem
+  # spec.add_dependency "example-gem", "~> 1.0"
+
+  spec.add_development_dependency "rubocop"
+  spec.add_development_dependency "rubocop-performance"
+  spec.add_development_dependency "rubocop-rake"
+  spec.add_development_dependency "rubocop-rspec"
+
+  # For more information and examples about making a new gem, check out our
+  # guide at: https://bundler.io/guides/creating_gem.html
+  spec.metadata["rubygems_mfa_required"] = "true"
+end

metadata

@@ -1,19 +1,76 @@
 --- !ruby/object:Gem::Specification
 name: process_bot
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - kaspernj
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-04-03 00:00:00.000000000 Z
-dependencies: []
+date: 2022-10-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop-performance
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Run and control processes.
 email:
 - k@spernj.org
-executables: []
+executables:
+- process_bot
 extensions: []
 extra_rdoc_files: []
 files:
@@ -21,17 +78,29 @@ files:
 - ".rubocop.yml"
 - CHANGELOG.md
 - Gemfile
+- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
+- exe/process_bot
 - lib/process_bot.rb
+- lib/process_bot/capistrano.rb
 - lib/process_bot/capistrano/puma.rake
 - lib/process_bot/capistrano/puma.rb
+- lib/process_bot/capistrano/puma/common.rb
 - lib/process_bot/capistrano/sidekiq.rake
 - lib/process_bot/capistrano/sidekiq.rb
 - lib/process_bot/capistrano/sidekiq_helpers.rb
+- lib/process_bot/control_socket.rb
+- lib/process_bot/logger.rb
+- lib/process_bot/options.rb
 - lib/process_bot/process.rb
+- lib/process_bot/process/handlers.rb
+- lib/process_bot/process/handlers/sidekiq.rb
+- lib/process_bot/process/runner.rb
 - lib/process_bot/version.rb
+- peak_flow.yml
+- process_bot.gemspec
 - sig/process_bot.rbs
 homepage: https://github.com/kaspernj/process_bot
 licenses:
@@ -41,6 +110,7 @@ metadata:
   homepage_uri: https://github.com/kaspernj/process_bot
   source_code_uri: https://github.com/kaspernj/process_bot
   changelog_uri: https://github.com/kaspernj/process_bot/blob/master/CHANGELOG.md
+  rubygems_mfa_required: 'true'
 post_install_message: 
 rdoc_options: []
 require_paths:
@@ -56,7 +126,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.32
+rubygems_version: 3.3.7
 signing_key: 
 specification_version: 4
 summary: Run and control processes.