stream.rb

class Stream
  attr_accessor :head

  def initialize(head, &tail)
    @head, @tail = head, tail
  end

  def tail
    @tail ? @tail.call : nil
  end

  def take(i)
    results, list = [], self
    i.times { results << list.head; list = list.tail }
    results
  end
end