module SidekiqRunner

# File lib/sidekiq-runner.rb, line 7
def self.configure
  yield SidekiqConfiguration.default if block_given?
end

# File lib/sidekiq-runner.rb, line 11
def self.configure_god
  yield GodConfiguration.default if block_given?
end

# File lib/sidekiq-runner.rb, line 51
def self.restart(sidekiq_instances: [])
  sidekiq_config = SidekiqConfiguration.get
  god_config = GodConfiguration.get

  return unless god_alive?(god_config)

  run(:restart, sidekiq_config, god_config) do
    sidekiq_config.each_key do |name|
      God::CLI::Command.new('restart', god_config.options, ['', name]) if sidekiq_instances.empty? || sidekiq_instances.include?(name.to_sym)
    end
  end
end

# File lib/sidekiq-runner.rb, line 64
def self.running?
  god_alive? GodConfiguration.get
end

# File lib/sidekiq-runner.rb, line 15
def self.start
  sidekiq_config, god_config = SidekiqConfiguration.get, GodConfiguration.get

  abort 'God is already running.' if god_alive?(god_config)

  run(:start, sidekiq_config, god_config) do
    $0 = "SidekiqRunner/God (#{god_config.process_name})"

    puts 'Starting god.'
    God::CLI::Run.new(god_config.options)
  end

  sleep 5
  fail 'Failed to start God.' unless god_alive?(god_config)
end

# File lib/sidekiq-runner.rb, line 31
def self.stop
  sidekiq_config, god_config = SidekiqConfiguration.get, GodConfiguration.get

  run(:stop, sidekiq_config, god_config) do
    God::EventHandler.load

    if god_alive?(god_config)
      puts "Terminating god process #{god_config.process_name}..."

      sidekiq_config.each_key do |name|
        God::CLI::Command.new('stop', god_config.options, ['', name])
      end

      God::CLI::Command.new('terminate', god_config.options, [])
    else
      abort 'God is not running, so no need to stop it.'
    end
  end
end

# File lib/sidekiq-runner.rb, line 70
def self.god_alive?(god_config)
  require 'drb'
  require 'god/socket'
  DRb.start_service('druby://127.0.0.1:0')
  server = DRbObject.new(nil, God::Socket.socket(god_config.port))

  # Ping server to ensure that it is responsive.
  begin
    server.ping
  rescue DRb::DRbConnError
    return false
  end
  true
end

# File lib/sidekiq-runner.rb, line 85
def self.run(action, sidekiq_config, god_config)
  begin

    # Use this flag to actually load all of the god infrastructure.
    $load_god = true
    require 'god'
    require 'god/cli/run'

    if [:start, :stop].include? action
      cb = god_config.send("before_#{action}_cb".to_sym)
      cb.call if cb
    end

    # Peform the action.
    yield if block_given?

  rescue SystemExit => e
    sidekiq_cb = e.success? ? "#{action}_success_cb" : "#{action}_error_cb"
  ensure
    if [:start, :stop].include? action
      sidekiq_cb = "#{action}_success_cb" unless sidekiq_cb
      sidekiq_cb = sidekiq_config.send(sidekiq_cb.to_sym)
      sidekiq_cb.call if sidekiq_cb
    end
  end
end