class Sidekiq::Rescue::ServerMiddleware

def calculate_reschedule_time(delay, rescue_counter)
  # NOTE: we use the retry counter to increase the jitter
  # so that the jobs don't retry at the same time
  # inspired by sidekiq https://github.com/sidekiq/sidekiq/blob/73c150d0430a8394cadb5cd49218895b113613a0/lib/sidekiq/job_retry.rb#L188
  jitter = rand(10) * rescue_counter
  delay = delay.call(rescue_counter) if delay.is_a?(Proc)
  Time.now.to_f + delay + jitter
end

def call(job_instance, job_payload, _queue, &block)
  job_class = job_instance.class
  options = job_class.sidekiq_rescue_options if job_class.respond_to?(:sidekiq_rescue_options)
  if options
    sidekiq_rescue(job_payload, **options, &block)
  else
    yield
  end
end

def increment_rescue_counter(job_payload)
  rescue_counter = job_payload["sidekiq_rescue_counter"].to_i
  rescue_counter += 1
  rescue_counter
end

def log_reschedule_info(rescue_counter, error, reschedule_at)
  Sidekiq::Rescue.logger.info("[sidekiq_rescue] Job failed #{rescue_counter} times with error: " \
                              "#{error.message}; rescheduling at #{reschedule_at}")
end

def reschedule_job(job_payload, reschedule_at, rescue_counter)
  Sidekiq::Client.push(job_payload.merge("at" => reschedule_at, "sidekiq_rescue_counter" => rescue_counter))
end

def sidekiq_rescue(job_payload, delay:, limit:, error:, **)
  yield
rescue *error => e
  rescue_counter = increment_rescue_counter(job_payload)
  raise e if rescue_counter > limit
  reschedule_at = calculate_reschedule_time(delay, rescue_counter)
  log_reschedule_info(rescue_counter, e, reschedule_at)
  reschedule_job(job_payload, reschedule_at, rescue_counter)
end