This allows passing keywords through a normal argument splat in a
Proc.  While needing ruby2_keywords support for methods is more
common, there is code that delegates keywords through normal
argument splats in procs, including code in Rails.  For that
reason, it makes sense to expose this for procs as well.

Internally, ruby2_keywords is not tied to methods, but iseqs,
so this just allows for setting the ruby2_keywords for the iseq
related to the proc.
This commit is contained in:
Jeremy Evans 2019-12-05 11:01:20 -08:00
Родитель c2dc27d643
Коммит f45c0dc239
2 изменённых файлов: 104 добавлений и 0 удалений

65
proc.c
Просмотреть файл

@ -3464,6 +3464,70 @@ rb_method_compose_to_right(VALUE self, VALUE g)
return proc_compose_to_right(self, g); return proc_compose_to_right(self, g);
} }
/*
* call-seq:
* proc.ruby2_keywords -> proc
*
* Marks the proc as passing keywords through a normal argument splat.
* This should only be called on procs that accept an argument splat
* (<tt>*args</tt>) but not explicit keywords or a keyword splat. It
* marks the proc such that if the proc is called with keyword arguments,
* the final hash argument is marked with a special flag such that if it
* is the final element of a normal argument splat to another method call,
* and that method call does not include explicit keywords or a keyword
* splat, the final element is interpreted as keywords. In other words,
* keywords will be passed through the proc to other methods.
*
* This should only be used for procs that delegate keywords to another
* method, and only for backwards compatibility with Ruby versions before
* 2.7.
*
* This method will probably be removed at some point, as it exists only
* for backwards compatibility. As it does not exist in Ruby versions
* before 2.7, check that the proc responds to this method before calling
* it. Also, be aware that if this method is removed, the behavior of the
* proc will change so that it does not pass through keywords.
*
* module Mod
* foo = ->(meth, *args, &block) do
* send(:"do_#{meth}", *args, &block)
* end
* foo.ruby2_keywords if foo.respond_to?(:ruby2_keywords)
* end
*/
static VALUE
proc_ruby2_keywords(VALUE procval)
{
rb_proc_t *proc;
GetProcPtr(procval, proc);
rb_check_frozen(procval);
if (proc->is_from_method) {
rb_warn("Skipping set of ruby2_keywords flag for proc (proc created from method)");
return procval;
}
switch (proc->block.type) {
case block_type_iseq:
if (proc->block.as.captured.code.iseq->body->param.flags.has_rest &&
!proc->block.as.captured.code.iseq->body->param.flags.has_kw &&
!proc->block.as.captured.code.iseq->body->param.flags.has_kwrest) {
proc->block.as.captured.code.iseq->body->param.flags.ruby2_keywords = 1;
}
else {
rb_warn("Skipping set of ruby2_keywords flag for proc (proc accepts keywords or proc does not accept argument splat)");
}
break;
default:
rb_warn("Skipping set of ruby2_keywords flag for proc (proc not defined in Ruby)");
break;
}
return procval;
}
/* /*
* Document-class: LocalJumpError * Document-class: LocalJumpError
* *
@ -3789,6 +3853,7 @@ Init_Proc(void)
rb_define_method(rb_cProc, ">>", proc_compose_to_right, 1); rb_define_method(rb_cProc, ">>", proc_compose_to_right, 1);
rb_define_method(rb_cProc, "source_location", rb_proc_location, 0); rb_define_method(rb_cProc, "source_location", rb_proc_location, 0);
rb_define_method(rb_cProc, "parameters", rb_proc_parameters, 0); rb_define_method(rb_cProc, "parameters", rb_proc_parameters, 0);
rb_define_method(rb_cProc, "ruby2_keywords", proc_ruby2_keywords, 0);
/* Exceptions */ /* Exceptions */
rb_eLocalJumpError = rb_define_class("LocalJumpError", rb_eStandardError); rb_eLocalJumpError = rb_define_class("LocalJumpError", rb_eStandardError);

Просмотреть файл

@ -2684,6 +2684,45 @@ class TestKeywordArguments < Test::Unit::TestCase
assert_raise(ArgumentError) { m.call(42, a: 1, **h2) } assert_raise(ArgumentError) { m.call(42, a: 1, **h2) }
end end
def test_proc_ruby2_keywords
h1 = {:a=>1}
foo = ->(*args, &block){block.call(*args)}
assert_same(foo, foo.ruby2_keywords)
assert_equal([[1], h1], foo.call(1, :a=>1, &->(*args, **kw){[args, kw]}))
assert_equal([1, h1], foo.call(1, :a=>1, &->(*args){args}))
assert_warn(/The last argument is used as the keyword parameter/) do
assert_equal([[1], h1], foo.call(1, {:a=>1}, &->(*args, **kw){[args, kw]}))
end
assert_equal([1, h1], foo.call(1, {:a=>1}, &->(*args){args}))
assert_warn(/The keyword argument is passed as the last hash parameter/) do
assert_equal([h1, {}], foo.call(:a=>1, &->(arg, **kw){[arg, kw]}))
end
assert_equal(h1, foo.call(:a=>1, &->(arg){arg}))
[->(){}, ->(arg){}, ->(*args, **kw){}, ->(*args, k: 1){}, ->(*args, k: ){}].each do |pr|
assert_warn(/Skipping set of ruby2_keywords flag for proc \(proc accepts keywords or proc does not accept argument splat\)/) do
pr.ruby2_keywords
end
end
o = Object.new
def o.foo(*args)
yield *args
end
foo = o.method(:foo).to_proc
assert_warn(/Skipping set of ruby2_keywords flag for proc \(proc created from method\)/) do
foo.ruby2_keywords
end
foo = :foo.to_proc
assert_warn(/Skipping set of ruby2_keywords flag for proc \(proc not defined in Ruby\)/) do
foo.ruby2_keywords
end
assert_raise(FrozenError) { ->(*args){}.freeze.ruby2_keywords }
end
def test_ruby2_keywords def test_ruby2_keywords
c = Class.new do c = Class.new do
ruby2_keywords def foo(meth, *args) ruby2_keywords def foo(meth, *args)