timcase/echonest

require 'echowrap/base'

module Echowrap
  class RateLimit
    attr_reader :attrs
    alias to_hash attrs

    # @return [Echowrap::RateLimit]
    def initialize(attrs={})
      @attrs = attrs
    end

    # @return [Integer]
    def limit
      limit = @attrs['x-rate-limit-limit']
      limit.to_i if limit
    end

    # @return [Integer]
    def remaining
      remaining = @attrs['x-rate-limit-remaining']
      remaining.to_i if remaining
    end

    # @return [Time]
    def reset_at
      reset = @attrs['x-rate-limit-reset']
      Time.at(reset.to_i) if reset
    end

    # @return [Integer]
    def reset_in
      [(reset_at - Time.now).ceil, 0].max if reset_at
    end
    alias retry_after reset_in

    # Update the attributes of a RateLimit
    #
    # @param attrs [Hash]
    # @return [Echowrap::RateLimit]
    def update(attrs)
      @attrs.update(attrs)
      self
    end

  end
end