From 48068bc1a4bbb272def06139fab392cf964ddc8a Mon Sep 17 00:00:00 2001 From: Jeremy Kemper Date: Sun, 29 Mar 2009 02:43:13 -0700 Subject: Feature detection for Range#step extension --- .../lib/active_support/core_ext/range/blockless_step.rb | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) (limited to 'activesupport/lib/active_support') diff --git a/activesupport/lib/active_support/core_ext/range/blockless_step.rb b/activesupport/lib/active_support/core_ext/range/blockless_step.rb index 98f333786d..f4792d03b7 100644 --- a/activesupport/lib/active_support/core_ext/range/blockless_step.rb +++ b/activesupport/lib/active_support/core_ext/range/blockless_step.rb @@ -1,21 +1,24 @@ class Range - if RUBY_VERSION < '1.9' + begin + (1..2).step + # Range#step doesn't return an Enumerator + rescue LocalJumpError # Return an array when step is called without a block. - def step_with_blockless(value = 1, &block) + def step_with_blockless(*args, &block) if block_given? - step_without_blockless(value, &block) + step_without_blockless(*args, &block) else array = [] - step_without_blockless(value) { |step| array << step } + step_without_blockless(*args) { |step| array << step } array end end else - def step_with_blockless(value = 1, &block) #:nodoc: + def step_with_blockless(*args, &block) #:nodoc: if block_given? - step_without_blockless + step_without_blockless(*args, &block) else - step_without_blockless(value).to_a + step_without_blockless(*args).to_a end end end -- cgit v1.2.3