diff options
Diffstat (limited to 'lib/irb')
-rw-r--r-- | lib/irb/extend-command.rb | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/lib/irb/extend-command.rb b/lib/irb/extend-command.rb index 84c4c05e15..862857466a 100644 --- a/lib/irb/extend-command.rb +++ b/lib/irb/extend-command.rb @@ -126,8 +126,12 @@ module IRB def #{cmd_name}(*opts, &b) require "#{load_file}" arity = ExtendCommand::#{cmd_class}.instance_method(:execute).arity - args = (1..arity).map {|i| "arg" + i.to_s } - args << "*opts" if arity < 0 + args = [] + if arity < 0 + args << "*opts" + arity = -arity - 1 + end + args.unshift *(1..arity).map {|i| "arg" + i.to_s } args << "&block" args = args.join(", ") eval <<-"EOS2", binding, __FILE__, __LINE__+1 |