Contents

require_relative '../provider_utils/rate_limit'

# Patch Dnsimple client method which retrieves headers for API rate limit dynamically
module Dnsimple
  class Client
    def execute(method, path, data = nil, options = {})
      response = request(method, path, data, options)
      rate_limit_sleep(response.headers['x-ratelimit-reset'].to_i, response.headers['x-ratelimit-remaining'].to_i)

      case response.code
      when 200..299
        response
      when 401
        raise AuthenticationFailed, response['message']
      when 404
        raise NotFoundError, response
      else
        raise RequestError, response
      end
    end

    private

    def rate_limit_sleep(rate_limit_reset, rate_limit_remaining)
      rate_limit_reset_in = [0, rate_limit_reset - Time.now.to_i].max
      rate_limit_periods = rate_limit_remaining + 1
      sleep_time = rate_limit_reset_in / rate_limit_periods.to_f

      rate_limit = RateLimit.new('DNSimple')
      rate_limit.sleep_for(sleep_time)
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
record_store-6.1.2	lib/record_store/provider/dnsimple/patch_api_header.rb
record_store-6.1.1	lib/record_store/provider/dnsimple/patch_api_header.rb