class Restforce::Collection

def count(*args)
  # By default, `Enumerable`'s `#count` uses `#each`, which means going through all
  # of the pages of results, one by one. Instead, we can use `#size` which we have
  # already overridden to work in a smarter, more efficient way. This only works for
  # the simple version of `#count` with no arguments. When called with an argument or
  # a block, you need to know what the items in the collection actually are, so we
  # call `super` and end up iterating through each item in the collection.
  return size unless block_given? || !args.empty?
  super
end

def current_page
  first(@raw_page['records'].size)
end

def each(&block)
  @raw_page['records'].each { |record| yield Restforce::Mash.build(record, @client) }
  np = next_page
  while np
    np.current_page.each(&block)
    np = np.next_page
  end
end

def empty?
  size.zero?
end

def has_next_page?
  !@raw_page['nextRecordsUrl'].nil?
end

def initialize(hash, client)
  @client = client
  @raw_page = hash
end

def next_page
  @client.get(@raw_page['nextRecordsUrl']).body if has_next_page?
end

def page_size
  @raw_page['records'].size
end

def pages
  [self] + (has_next_page? ? next_page.pages : [])
end

def size
  @raw_page['totalSize'] || @raw_page['size']
end