tylerhunt/brewery_db

module BreweryDB
  class Collection
    include Enumerable

    attr_reader :size
    alias length size

    def initialize(response)
      self.response = response
    end

    def each
      return to_enum unless block_given?
      @collection.each { |element| yield(element) }
    end

  private

    def response=(response)
      @response = response
      @collection = Array(response.data) || []
      @size = @collection.length
    end
  end
end