diff options
| author | Jemma Issroff <jemmaissroff@gmail.com> | 2023-10-26 07:02:55 -0300 |
|---|---|---|
| committer | Jemma Issroff <jemmaissroff@gmail.com> | 2023-10-26 11:42:15 -0300 |
| commit | 9570a88bb3c86c79a25562a341452002be457c12 (patch) | |
| tree | 242d59c55c659816bf69751e40c5d0971f795be7 | |
| parent | ec2878c11bb0b5549c49ef2ae842a886c3905aa9 (diff) | |
[PRISM] Implement compilation for ConstantPathAndWriteNode
| -rw-r--r-- | prism_compile.c | 37 | ||||
| -rw-r--r-- | test/ruby/test_compile_prism.rb | 5 |
2 files changed, 42 insertions, 0 deletions
diff --git a/prism_compile.c b/prism_compile.c index 2059369bd4..829ff1695d 100644 --- a/prism_compile.c +++ b/prism_compile.c @@ -1512,6 +1512,43 @@ pm_compile_node(rb_iseq_t *iseq, const pm_node_t *node, LINK_ANCHOR *const ret, PM_POP_IF_POPPED; return; } + case PM_CONSTANT_PATH_AND_WRITE_NODE: { + pm_constant_path_and_write_node_t *constant_path_and_write_node = (pm_constant_path_and_write_node_t*) node; + + LABEL *lfin = NEW_LABEL(lineno); + + pm_constant_path_node_t *target = constant_path_and_write_node->target; + PM_COMPILE(target->parent); + + pm_constant_read_node_t *child = (pm_constant_read_node_t *)target->child; + VALUE child_name = ID2SYM(pm_constant_id_lookup(scope_node, child->name)); + + ADD_INSN(ret, &dummy_line_node, dup); + ADD_INSN1(ret, &dummy_line_node, putobject, Qtrue); + ADD_INSN1(ret, &dummy_line_node, getconstant, child_name); + + PM_DUP_UNLESS_POPPED; + ADD_INSNL(ret, &dummy_line_node, branchunless, lfin); + + PM_POP_UNLESS_POPPED; + PM_COMPILE(constant_path_and_write_node->value); + + if (popped) { + ADD_INSN1(ret, &dummy_line_node, topn, INT2FIX(1)); + } + else { + ADD_INSN1(ret, &dummy_line_node, dupn, INT2FIX(2)); + ADD_INSN(ret, &dummy_line_node, swap); + } + + ADD_INSN1(ret, &dummy_line_node, setconstant, child_name); + ADD_LABEL(ret, lfin); + + if (!popped) ADD_INSN(ret, &dummy_line_node, swap); + PM_POP; + + return; + } case PM_CONSTANT_PATH_OR_WRITE_NODE: { pm_constant_path_or_write_node_t *constant_path_or_write_node = (pm_constant_path_or_write_node_t*) node; diff --git a/test/ruby/test_compile_prism.rb b/test/ruby/test_compile_prism.rb index 6ab143d93a..9062d5ffc2 100644 --- a/test/ruby/test_compile_prism.rb +++ b/test/ruby/test_compile_prism.rb @@ -160,6 +160,11 @@ module Prism assert_prism_eval("::CPWN = 1") end + def test_ConstantPathAndWriteNode + assert_prism_eval("Prism::CPAWN = 1; Prism::CPAWN &&= 2") + assert_prism_eval("Prism::CPAWN &&= 1") + end + def test_ConstantPathOrWriteNode assert_prism_eval("Prism::CPOrWN = nil; Prism::CPOrWN ||= 1") assert_prism_eval("Prism::CPOrWN ||= 1") |