Added `NODE_SPECIAL_EXCESSIVE_COMMA` info to `ARGS` of `RubyVM::AbstractSyntaxTree`.

This commit is contained in:
manga_osyo 2020-07-07 20:18:18 +09:00 коммит произвёл Nobuyoshi Nakada
Родитель 57cde28c20
Коммит ff5e660340
2 изменённых файлов: 18 добавлений и 1 удалений

4
ast.c
Просмотреть файл

@ -547,7 +547,9 @@ node_children(rb_ast_t *ast, NODE *node)
var_name(ainfo->first_post_arg),
INT2NUM(ainfo->post_args_num),
NEW_CHILD(ast, ainfo->post_init),
var_name(ainfo->rest_arg),
(ainfo->rest_arg == NODE_SPECIAL_EXCESSIVE_COMMA
? ID2SYM(rb_intern("NODE_SPECIAL_EXCESSIVE_COMMA"))
: var_name(ainfo->rest_arg)),
(ainfo->no_kwarg ? Qfalse : NEW_CHILD(ast, ainfo->kw_args)),
(ainfo->no_kwarg ? Qfalse : NEW_CHILD(ast, ainfo->kw_rest_arg)),
var_name(ainfo->block_arg));

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

@ -330,4 +330,19 @@ class TestAst < Test::Unit::TestCase
assert_equal(:+, op)
assert_equal(:VCALL, value.type)
end
def test_args
rest = 6
node = RubyVM::AbstractSyntaxTree.parse("proc { |a| }")
_, args = *node.children.last.children[1].children
assert_equal(nil, args.children[rest])
node = RubyVM::AbstractSyntaxTree.parse("proc { |a,| }")
_, args = *node.children.last.children[1].children
assert_equal(:NODE_SPECIAL_EXCESSIVE_COMMA, args.children[rest])
node = RubyVM::AbstractSyntaxTree.parse("proc { |*a| }")
_, args = *node.children.last.children[1].children
assert_equal(:a, args.children[rest])
end
end