array_enumerator 0.0.2 → 0.0.3

data/VERSION

@@ -1 +1 @@
-0.0.2
+0.0.3

data/array_enumerator.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{array_enumerator}
-  s.version = "0.0.2"
+  s.version = "0.0.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Kasper Johansen"]

data/lib/array_enumerator.rb

@@ -47,6 +47,49 @@ class Array_enumerator
     end
   end
   
+  #This method should only be used with 'each_index'.
+  def [](key)
+    if @each_index and @each_index.key?(key)
+      ret = @each_index[key]
+      @each_index.delete(key)
+      return ret
+    end
+    
+    raise "This only works when also using 'each_index'. Invalid key: '#{key}'."
+  end
+  
+  #Yields each count-key and caches element for returning it by using the []-method.
+  def each_index(&block)
+    enum = Enumerator.new do |yielder|
+      begin
+        @each_index = {}
+        
+        count = 0
+        self.each do |ele|
+          #Remove previous element to not take up memory.
+          count_before = count - 1
+          @each_index.delete(count_before) if @each_index.key?(count_before)
+          
+          #Add current element to cache.
+          @each_index[count] = ele
+          yield(count)
+          
+          #Increase count for next run.
+          count += 1
+        end
+      ensure
+        @each_index = nil
+      end
+    end
+    
+    if block
+      enum.each(&block)
+      return nil
+    else
+      return enum
+    end
+  end
+  
   #Returns a enumerator that can yield the all the lements (both cached and future un-cached ones).
   def to_enum
     check_corrupted

data/spec/array_enumerator_spec.rb

@@ -128,4 +128,17 @@ describe "ArrayEnumerator" do
       end
     end
   end
+  
+  it "each_index" do
+    arr = %w[a b c d e f g]
+    ae = Array_enumerator.new(arr.to_enum)
+    
+    expect = 0
+    ae.each_index do |num|
+      raise "Expected #{expect} but got: #{num}" if num != expect
+      ele = ae[num]
+      raise "Expected #{arr[num]} but got: #{ele}" if ele != arr[num]
+      expect += 1
+    end
+  end
 end

metadata

@@ -2,7 +2,7 @@
 name: array_enumerator
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors: 
 - Kasper Johansen
@@ -104,7 +104,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 813322090081366183
+      hash: -947760670988037179
       segments: 
       - 0
       version: "0"