Support nil argument as no argument

mruby · Dec 1, 2016 · 0f774ff · 0f774ff
1 parent 61ac564
commit 0f774ff
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/mrbgems/mruby-enumerator/mrblib/enumerator.rb b/mrbgems/mruby-enumerator/mrblib/enumerator.rb
@@ -153,9 +153,15 @@ def initialize_copy(obj)
   #
   def with_index(offset=0)
     return to_enum :with_index, offset unless block_given?
-    raise TypeError, "no implicit conversion of #{offset.class} into Integer" unless offset.respond_to?(:to_int)
+    offset = if offset.nil?
+      0
+    elsif offset.respond_to?(:to_int)
+      offset.to_int
+    else
+      raise TypeError, "no implicit conversion of #{offset.class} into Integer"
+    end
 
-    n = offset.to_int - 1
+    n = offset - 1
     enumerator_block_call do |i|
       n += 1
       yield [i,n]