[PRISM] Implemented ConstantPathOperatorWriteNode

2 files changed, 30 insertions, 0 deletions

diff --git a/prism_compile.c b/prism_compile.c
index f881bcc672..fd1ba3c421 100644
--- a/prism_compile.c
+++ b/prism_compile.c
@@ -1512,6 +1512,32 @@ 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_OPERATOR_WRITE_NODE: {
+        pm_constant_path_operator_write_node_t *constant_path_operator_write_node = (pm_constant_path_operator_write_node_t*) node;
+
+        pm_constant_path_node_t *target = constant_path_operator_write_node->target;
+        PM_COMPILE(target->parent);
+
+        ADD_INSN(ret, &dummy_line_node, dup);
+        ADD_INSN1(ret, &dummy_line_node, putobject, Qtrue);
+
+        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_INSN1(ret, &dummy_line_node, getconstant, child_name);
+
+        PM_COMPILE(constant_path_operator_write_node->value);
+        ID method_id = pm_constant_id_lookup(scope_node, constant_path_operator_write_node->operator);
+        ADD_CALL(ret, &dummy_line_node, method_id, INT2FIX(1));
+        ADD_INSN(ret, &dummy_line_node, swap);
+
+        if (!popped) {
+            ADD_INSN1(ret, &dummy_line_node, topn, INT2FIX(1));
+            ADD_INSN(ret, &dummy_line_node, swap);
+        }
+
+        ADD_INSN1(ret, &dummy_line_node, setconstant, child_name);
+        return ;
+      }
       case PM_CONSTANT_PATH_TARGET_NODE: {
         pm_constant_path_target_node_t *cast = (pm_constant_path_target_node_t *)node;
 
diff --git a/test/ruby/test_compile_prism.rb b/test/ruby/test_compile_prism.rb
index 8d1ea20399..b33ed5319c 100644
--- a/test/ruby/test_compile_prism.rb
+++ b/test/ruby/test_compile_prism.rb
@@ -160,6 +160,10 @@ module Prism
       assert_prism_eval("::CPWN = 1")
     end
 
+    def test_ConstantPathOperatorWriteNode
+      assert_prism_eval("Prism::CPOWN = 0; Prism::CPOWN += 1")
+    end
+
     def test_GlobalVariableAndWriteNode
       assert_prism_eval("$pit = 0; $pit &&= 1")
     end