adelevie/parse-ruby-client

# encoding: utf-8
module Faraday
  # A middleware to display error messages in the JSON response
  class ExtendedParseJson < FaradayMiddleware::ParseJson
    def process_response(env)
      env[:raw_body] = env[:body] if preserve_raw?(env)

      if env[:status] >= 400
        begin
          data = parse(env[:body]) || {}
        rescue StandardError
          data = {}
        end

        array_codes = [
          Parse::Protocol::ERROR_INTERNAL,
          Parse::Protocol::ERROR_TIMEOUT,
          Parse::Protocol::ERROR_EXCEEDED_BURST_LIMIT
        ]

        error_hash = {
          'error' => "HTTP Status #{env[:status]} Body #{env[:body]}",
          'http_status_code' => env[:status]
        }.merge(data)

        if data['code'] && \
          array_codes.include?(data['code']) && \
          data['code'] == Parse::Protocol::ERROR_EXCEEDED_BURST_LIMIT
          sleep 60
        end

        raise exception(env), error_hash
      else
        data = parse(env[:body]) || {}
        env[:body] = data
      end
    end

    def exception(env)
      # NOTE: decide to retry or not, the header is deleted
      #  so it won't be sent to the server
      retries = env.request_headers.delete('X-ParseRubyClient-Retries')
      (retries.to_i.zero? ? Parse::ParseProtocolError : Parse::ParseProtocolRetry)
    end
  end
end