didil/sidekiq-client-cli

require 'sidekiq'
require_relative 'sidekiq_client_cli/version'
require_relative 'sidekiq_client_cli/parser'

class SidekiqClientCLI
  COMMANDS = %w{push}
  DEFAULT_CONFIG_PATH = "config/initializers/sidekiq.rb"

  attr_accessor :settings

  def parse
    @settings = Parser.new.parse
  end

  def self.cast_retry_option(retry_option)
    return true if !!retry_option.match(/^(true|t|yes|y)$/i)
    return false if !!retry_option.match(/^(false|f|no|n|0)$/i)
    return retry_option.to_i if !!retry_option.match(/^\d+$/)
  end

  def run
    # load the config file
    load settings.config_path if File.exists?(settings.config_path)

    # set queue or retry if they are not given
    settings.queue ||= Sidekiq.default_worker_options['queue']
    settings.retry = Sidekiq.default_worker_options['retry'] if settings.retry.nil?

    self.send settings.command.to_sym
  end

  # Returns true if all args can be pushed successfully.
  # Returns false if at least one exception occured.
  def push
    settings.command_args.inject(true) do |success, arg|
      push_argument arg
    end
  end

  private

  def push_argument(arg)
    jid = Sidekiq::Client.push({ 'class' => arg,
                                 'queue' => settings.queue,
                                 'args'  => [],
                                 'retry' => settings.retry })
    p "Posted #{arg} to queue '#{settings.queue}', Job ID : #{jid}, Retry : #{settings.retry}"
    true
  rescue StandardError => ex
    p "Failed to push to queue : #{ex.message}"
    false
  end

end