class Honeybadger::Plugins::Shoryuken::Middleware

def attempt_threshold
  ::Honeybadger.config[:"shoryuken.attempt_threshold"].to_i
end

def call(_worker, _queue, sqs_msg, body)
  begin
    yield
  rescue => e
    if attempt_threshold <= receive_count(sqs_msg)
      Honeybadger.notify(e, parameters: notification_params(body))
    end
    raise e
  end
ensure
  Honeybadger.clear!
end

def notification_params(body)
  body.is_a?(Array) ? {batch: body} : {body: body}
end

def receive_count(sqs_msg)
  return 0 if sqs_msg.is_a?(Array)
  sqs_msg.attributes["ApproximateReceiveCount".freeze].to_i
end