[PRISM] Implement compilation for ConstantPathAndWriteNode

1 files changed, 37 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;