third-party/mruby/mrbgems/mruby-enum-chain/mrblib/chain.rb

   1 ##
   2 # chain.rb Enumerator::Chain class
   3 # See Copyright Notice in mruby.h
   4
   5 module Enumerable
   6   def chain(*args)
   7     Enumerator::Chain.new(self, *args)
   8   end
   9 end
  10
  11 class Enumerator
  12   def +(other)
  13     Chain.new(self, other)
  14   end
  15
  16   class Chain
  17     include Enumerable
  18
  19     def initialize(*args)
  20       @enums = args.freeze
  21       @pos = -1
  22     end
  23
  24     def each(&block)
  25       return to_enum unless block
  26
  27       i = 0
  28       while i < @enums.size
  29         @pos = i
  30         @enums[i].each(&block)
  31         i += 1
  32       end
  33
  34       self
  35     end
  36
  37     def size
  38       @enums.reduce(0) do |a, e|
  39         return nil unless e.respond_to?(:size)
  40         a + e.size
  41       end
  42     end
  43
  44     def rewind
  45       while 0 <= @pos && @pos < @enums.size
  46         e = @enums[@pos]
  47         e.rewind if e.respond_to?(:rewind)
  48         @pos -= 1
  49       end
  50
  51       self
  52     end
  53
  54     def +(other)
  55       self.class.new(self, other)
  56     end
  57
  58     def inspect
  59       "#<#{self.class}: #{@enums.inspect}>"
  60     end
  61   end
  62 end