class Sidekiq::Rescue::ServerMiddleware

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 sidekiq_rescue(job_payload, delay:, limit:, error:, **)
  yield
rescue *error => e
  rescue_counter = job_payload["sidekiq_rescue_counter"].to_i
  rescue_counter += 1
  raise e if rescue_counter > limit
  # 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
  reschedule_at = Time.now.to_f + delay + jitter
  Sidekiq::Rescue.logger.info("[sidekiq_rescue] Job failed #{rescue_counter} times with error:" \
                              "#{e.message}; rescheduling at #{reschedule_at}")
  Sidekiq::Client.push(job_payload.merge("at" => reschedule_at, "sidekiq_rescue_counter" => rescue_counter))
end