prism 0.13.0 → 0.15.0

data/ext/prism/api_node.c

@@ -82,6 +82,9 @@ static VALUE rb_cPrismIfNode;
 static VALUE rb_cPrismImaginaryNode;
 static VALUE rb_cPrismImplicitNode;
 static VALUE rb_cPrismInNode;
+static VALUE rb_cPrismIndexAndWriteNode;
+static VALUE rb_cPrismIndexOperatorWriteNode;
+static VALUE rb_cPrismIndexOrWriteNode;
 static VALUE rb_cPrismInstanceVariableAndWriteNode;
 static VALUE rb_cPrismInstanceVariableOperatorWriteNode;
 static VALUE rb_cPrismInstanceVariableOrWriteNode;
@@ -358,7 +361,6 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
                 case PM_CALL_AND_WRITE_NODE: {
                     pm_call_and_write_node_t *cast = (pm_call_and_write_node_t *) node;
                     pm_node_stack_push(&node_stack, (pm_node_t *) cast->receiver);
-                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->arguments);
                     pm_node_stack_push(&node_stack, (pm_node_t *) cast->value);
                     break;
                 }
@@ -374,7 +376,6 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
                 case PM_CALL_OPERATOR_WRITE_NODE: {
                     pm_call_operator_write_node_t *cast = (pm_call_operator_write_node_t *) node;
                     pm_node_stack_push(&node_stack, (pm_node_t *) cast->receiver);
-                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->arguments);
                     pm_node_stack_push(&node_stack, (pm_node_t *) cast->value);
                     break;
                 }
@@ -382,7 +383,6 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
                 case PM_CALL_OR_WRITE_NODE: {
                     pm_call_or_write_node_t *cast = (pm_call_or_write_node_t *) node;
                     pm_node_stack_push(&node_stack, (pm_node_t *) cast->receiver);
-                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->arguments);
                     pm_node_stack_push(&node_stack, (pm_node_t *) cast->value);
                     break;
                 }
@@ -640,6 +640,33 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
                     pm_node_stack_push(&node_stack, (pm_node_t *) cast->statements);
                     break;
                 }
+#line 108 "api_node.c.erb"
+                case PM_INDEX_AND_WRITE_NODE: {
+                    pm_index_and_write_node_t *cast = (pm_index_and_write_node_t *) node;
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->receiver);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->arguments);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->block);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->value);
+                    break;
+                }
+#line 108 "api_node.c.erb"
+                case PM_INDEX_OPERATOR_WRITE_NODE: {
+                    pm_index_operator_write_node_t *cast = (pm_index_operator_write_node_t *) node;
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->receiver);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->arguments);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->block);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->value);
+                    break;
+                }
+#line 108 "api_node.c.erb"
+                case PM_INDEX_OR_WRITE_NODE: {
+                    pm_index_or_write_node_t *cast = (pm_index_or_write_node_t *) node;
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->receiver);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->arguments);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->block);
+                    pm_node_stack_push(&node_stack, (pm_node_t *) cast->value);
+                    break;
+                }
 #line 108 "api_node.c.erb"
                 case PM_INSTANCE_VARIABLE_AND_WRITE_NODE: {
                     pm_instance_variable_and_write_node_t *cast = (pm_instance_variable_and_write_node_t *) node;
@@ -1222,12 +1249,18 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
                 }
 #line 134 "api_node.c.erb"
                 case PM_BACK_REFERENCE_READ_NODE: {
-                    VALUE argv[1];
+                    pm_back_reference_read_node_t *cast = (pm_back_reference_read_node_t *) node;
+                    VALUE argv[2];
+
+                    // name
+#line 157 "api_node.c.erb"
+                    assert(cast->name != 0);
+                    argv[0] = rb_id2sym(constants[cast->name - 1]);
 
                     // location
-                    argv[0] = pm_location_new(parser, node->location.start, node->location.end, source);
+                    argv[1] = pm_location_new(parser, node->location.start, node->location.end, source);
 
-                    rb_ary_push(value_stack, rb_class_new_instance(1, argv, rb_cPrismBackReferenceReadNode));
+                    rb_ary_push(value_stack, rb_class_new_instance(2, argv, rb_cPrismBackReferenceReadNode));
                     break;
                 }
 #line 134 "api_node.c.erb"
@@ -1409,7 +1442,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 #line 134 "api_node.c.erb"
                 case PM_CALL_AND_WRITE_NODE: {
                     pm_call_and_write_node_t *cast = (pm_call_and_write_node_t *) node;
-                    VALUE argv[12];
+                    VALUE argv[9];
 
                     // receiver
 #line 145 "api_node.c.erb"
@@ -1423,42 +1456,32 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 #line 173 "api_node.c.erb"
                     argv[2] = cast->message_loc.start == NULL ? Qnil : pm_location_new(parser, cast->message_loc.start, cast->message_loc.end, source);
 
-                    // opening_loc
-#line 173 "api_node.c.erb"
-                    argv[3] = cast->opening_loc.start == NULL ? Qnil : pm_location_new(parser, cast->opening_loc.start, cast->opening_loc.end, source);
-
-                    // arguments
-#line 145 "api_node.c.erb"
-                    argv[4] = rb_ary_pop(value_stack);
-
-                    // closing_loc
-#line 173 "api_node.c.erb"
-                    argv[5] = cast->closing_loc.start == NULL ? Qnil : pm_location_new(parser, cast->closing_loc.start, cast->closing_loc.end, source);
-
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[6] = ULONG2NUM(node->flags >> 2);
+                    argv[3] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // read_name
-#line 154 "api_node.c.erb"
-                    argv[7] = pm_string_new(&cast->read_name, encoding);
+#line 157 "api_node.c.erb"
+                    assert(cast->read_name != 0);
+                    argv[4] = rb_id2sym(constants[cast->read_name - 1]);
 
                     // write_name
-#line 154 "api_node.c.erb"
-                    argv[8] = pm_string_new(&cast->write_name, encoding);
+#line 157 "api_node.c.erb"
+                    assert(cast->write_name != 0);
+                    argv[5] = rb_id2sym(constants[cast->write_name - 1]);
 
                     // operator_loc
 #line 170 "api_node.c.erb"
-                    argv[9] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
+                    argv[6] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
 
                     // value
 #line 145 "api_node.c.erb"
-                    argv[10] = rb_ary_pop(value_stack);
+                    argv[7] = rb_ary_pop(value_stack);
 
                     // location
-                    argv[11] = pm_location_new(parser, node->location.start, node->location.end, source);
+                    argv[8] = pm_location_new(parser, node->location.start, node->location.end, source);
 
-                    rb_ary_push(value_stack, rb_class_new_instance(12, argv, rb_cPrismCallAndWriteNode));
+                    rb_ary_push(value_stack, rb_class_new_instance(9, argv, rb_cPrismCallAndWriteNode));
                     break;
                 }
 #line 134 "api_node.c.erb"
@@ -1496,11 +1519,12 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[7] = ULONG2NUM(node->flags >> 2);
+                    argv[7] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // name
-#line 154 "api_node.c.erb"
-                    argv[8] = pm_string_new(&cast->name, encoding);
+#line 157 "api_node.c.erb"
+                    assert(cast->name != 0);
+                    argv[8] = rb_id2sym(constants[cast->name - 1]);
 
                     // location
                     argv[9] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -1511,7 +1535,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 #line 134 "api_node.c.erb"
                 case PM_CALL_OPERATOR_WRITE_NODE: {
                     pm_call_operator_write_node_t *cast = (pm_call_operator_write_node_t *) node;
-                    VALUE argv[13];
+                    VALUE argv[10];
 
                     // receiver
 #line 145 "api_node.c.erb"
@@ -1525,53 +1549,43 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 #line 173 "api_node.c.erb"
                     argv[2] = cast->message_loc.start == NULL ? Qnil : pm_location_new(parser, cast->message_loc.start, cast->message_loc.end, source);
 
-                    // opening_loc
-#line 173 "api_node.c.erb"
-                    argv[3] = cast->opening_loc.start == NULL ? Qnil : pm_location_new(parser, cast->opening_loc.start, cast->opening_loc.end, source);
-
-                    // arguments
-#line 145 "api_node.c.erb"
-                    argv[4] = rb_ary_pop(value_stack);
-
-                    // closing_loc
-#line 173 "api_node.c.erb"
-                    argv[5] = cast->closing_loc.start == NULL ? Qnil : pm_location_new(parser, cast->closing_loc.start, cast->closing_loc.end, source);
-
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[6] = ULONG2NUM(node->flags >> 2);
+                    argv[3] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // read_name
-#line 154 "api_node.c.erb"
-                    argv[7] = pm_string_new(&cast->read_name, encoding);
+#line 157 "api_node.c.erb"
+                    assert(cast->read_name != 0);
+                    argv[4] = rb_id2sym(constants[cast->read_name - 1]);
 
                     // write_name
-#line 154 "api_node.c.erb"
-                    argv[8] = pm_string_new(&cast->write_name, encoding);
+#line 157 "api_node.c.erb"
+                    assert(cast->write_name != 0);
+                    argv[5] = rb_id2sym(constants[cast->write_name - 1]);
 
                     // operator
 #line 157 "api_node.c.erb"
                     assert(cast->operator != 0);
-                    argv[9] = rb_id2sym(constants[cast->operator - 1]);
+                    argv[6] = rb_id2sym(constants[cast->operator - 1]);
 
                     // operator_loc
 #line 170 "api_node.c.erb"
-                    argv[10] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
+                    argv[7] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
 
                     // value
 #line 145 "api_node.c.erb"
-                    argv[11] = rb_ary_pop(value_stack);
+                    argv[8] = rb_ary_pop(value_stack);
 
                     // location
-                    argv[12] = pm_location_new(parser, node->location.start, node->location.end, source);
+                    argv[9] = pm_location_new(parser, node->location.start, node->location.end, source);
 
-                    rb_ary_push(value_stack, rb_class_new_instance(13, argv, rb_cPrismCallOperatorWriteNode));
+                    rb_ary_push(value_stack, rb_class_new_instance(10, argv, rb_cPrismCallOperatorWriteNode));
                     break;
                 }
 #line 134 "api_node.c.erb"
                 case PM_CALL_OR_WRITE_NODE: {
                     pm_call_or_write_node_t *cast = (pm_call_or_write_node_t *) node;
-                    VALUE argv[12];
+                    VALUE argv[9];
 
                     // receiver
 #line 145 "api_node.c.erb"
@@ -1585,42 +1599,32 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 #line 173 "api_node.c.erb"
                     argv[2] = cast->message_loc.start == NULL ? Qnil : pm_location_new(parser, cast->message_loc.start, cast->message_loc.end, source);
 
-                    // opening_loc
-#line 173 "api_node.c.erb"
-                    argv[3] = cast->opening_loc.start == NULL ? Qnil : pm_location_new(parser, cast->opening_loc.start, cast->opening_loc.end, source);
-
-                    // arguments
-#line 145 "api_node.c.erb"
-                    argv[4] = rb_ary_pop(value_stack);
-
-                    // closing_loc
-#line 173 "api_node.c.erb"
-                    argv[5] = cast->closing_loc.start == NULL ? Qnil : pm_location_new(parser, cast->closing_loc.start, cast->closing_loc.end, source);
-
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[6] = ULONG2NUM(node->flags >> 2);
+                    argv[3] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // read_name
-#line 154 "api_node.c.erb"
-                    argv[7] = pm_string_new(&cast->read_name, encoding);
+#line 157 "api_node.c.erb"
+                    assert(cast->read_name != 0);
+                    argv[4] = rb_id2sym(constants[cast->read_name - 1]);
 
                     // write_name
-#line 154 "api_node.c.erb"
-                    argv[8] = pm_string_new(&cast->write_name, encoding);
+#line 157 "api_node.c.erb"
+                    assert(cast->write_name != 0);
+                    argv[5] = rb_id2sym(constants[cast->write_name - 1]);
 
                     // operator_loc
 #line 170 "api_node.c.erb"
-                    argv[9] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
+                    argv[6] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
 
                     // value
 #line 145 "api_node.c.erb"
-                    argv[10] = rb_ary_pop(value_stack);
+                    argv[7] = rb_ary_pop(value_stack);
 
                     // location
-                    argv[11] = pm_location_new(parser, node->location.start, node->location.end, source);
+                    argv[8] = pm_location_new(parser, node->location.start, node->location.end, source);
 
-                    rb_ary_push(value_stack, rb_class_new_instance(12, argv, rb_cPrismCallOrWriteNode));
+                    rb_ary_push(value_stack, rb_class_new_instance(9, argv, rb_cPrismCallOrWriteNode));
                     break;
                 }
 #line 134 "api_node.c.erb"
@@ -2415,7 +2419,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[3] = ULONG2NUM(node->flags >> 2);
+                    argv[3] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[4] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -2801,6 +2805,152 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
                     rb_ary_push(value_stack, rb_class_new_instance(5, argv, rb_cPrismInNode));
                     break;
                 }
+#line 134 "api_node.c.erb"
+                case PM_INDEX_AND_WRITE_NODE: {
+                    pm_index_and_write_node_t *cast = (pm_index_and_write_node_t *) node;
+                    VALUE argv[10];
+
+                    // receiver
+#line 145 "api_node.c.erb"
+                    argv[0] = rb_ary_pop(value_stack);
+
+                    // call_operator_loc
+#line 173 "api_node.c.erb"
+                    argv[1] = cast->call_operator_loc.start == NULL ? Qnil : pm_location_new(parser, cast->call_operator_loc.start, cast->call_operator_loc.end, source);
+
+                    // opening_loc
+#line 170 "api_node.c.erb"
+                    argv[2] = pm_location_new(parser, cast->opening_loc.start, cast->opening_loc.end, source);
+
+                    // arguments
+#line 145 "api_node.c.erb"
+                    argv[3] = rb_ary_pop(value_stack);
+
+                    // closing_loc
+#line 170 "api_node.c.erb"
+                    argv[4] = pm_location_new(parser, cast->closing_loc.start, cast->closing_loc.end, source);
+
+                    // block
+#line 145 "api_node.c.erb"
+                    argv[5] = rb_ary_pop(value_stack);
+
+                    // flags
+#line 179 "api_node.c.erb"
+                    argv[6] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
+
+                    // operator_loc
+#line 170 "api_node.c.erb"
+                    argv[7] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
+
+                    // value
+#line 145 "api_node.c.erb"
+                    argv[8] = rb_ary_pop(value_stack);
+
+                    // location
+                    argv[9] = pm_location_new(parser, node->location.start, node->location.end, source);
+
+                    rb_ary_push(value_stack, rb_class_new_instance(10, argv, rb_cPrismIndexAndWriteNode));
+                    break;
+                }
+#line 134 "api_node.c.erb"
+                case PM_INDEX_OPERATOR_WRITE_NODE: {
+                    pm_index_operator_write_node_t *cast = (pm_index_operator_write_node_t *) node;
+                    VALUE argv[11];
+
+                    // receiver
+#line 145 "api_node.c.erb"
+                    argv[0] = rb_ary_pop(value_stack);
+
+                    // call_operator_loc
+#line 173 "api_node.c.erb"
+                    argv[1] = cast->call_operator_loc.start == NULL ? Qnil : pm_location_new(parser, cast->call_operator_loc.start, cast->call_operator_loc.end, source);
+
+                    // opening_loc
+#line 170 "api_node.c.erb"
+                    argv[2] = pm_location_new(parser, cast->opening_loc.start, cast->opening_loc.end, source);
+
+                    // arguments
+#line 145 "api_node.c.erb"
+                    argv[3] = rb_ary_pop(value_stack);
+
+                    // closing_loc
+#line 170 "api_node.c.erb"
+                    argv[4] = pm_location_new(parser, cast->closing_loc.start, cast->closing_loc.end, source);
+
+                    // block
+#line 145 "api_node.c.erb"
+                    argv[5] = rb_ary_pop(value_stack);
+
+                    // flags
+#line 179 "api_node.c.erb"
+                    argv[6] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
+
+                    // operator
+#line 157 "api_node.c.erb"
+                    assert(cast->operator != 0);
+                    argv[7] = rb_id2sym(constants[cast->operator - 1]);
+
+                    // operator_loc
+#line 170 "api_node.c.erb"
+                    argv[8] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
+
+                    // value
+#line 145 "api_node.c.erb"
+                    argv[9] = rb_ary_pop(value_stack);
+
+                    // location
+                    argv[10] = pm_location_new(parser, node->location.start, node->location.end, source);
+
+                    rb_ary_push(value_stack, rb_class_new_instance(11, argv, rb_cPrismIndexOperatorWriteNode));
+                    break;
+                }
+#line 134 "api_node.c.erb"
+                case PM_INDEX_OR_WRITE_NODE: {
+                    pm_index_or_write_node_t *cast = (pm_index_or_write_node_t *) node;
+                    VALUE argv[10];
+
+                    // receiver
+#line 145 "api_node.c.erb"
+                    argv[0] = rb_ary_pop(value_stack);
+
+                    // call_operator_loc
+#line 173 "api_node.c.erb"
+                    argv[1] = cast->call_operator_loc.start == NULL ? Qnil : pm_location_new(parser, cast->call_operator_loc.start, cast->call_operator_loc.end, source);
+
+                    // opening_loc
+#line 170 "api_node.c.erb"
+                    argv[2] = pm_location_new(parser, cast->opening_loc.start, cast->opening_loc.end, source);
+
+                    // arguments
+#line 145 "api_node.c.erb"
+                    argv[3] = rb_ary_pop(value_stack);
+
+                    // closing_loc
+#line 170 "api_node.c.erb"
+                    argv[4] = pm_location_new(parser, cast->closing_loc.start, cast->closing_loc.end, source);
+
+                    // block
+#line 145 "api_node.c.erb"
+                    argv[5] = rb_ary_pop(value_stack);
+
+                    // flags
+#line 179 "api_node.c.erb"
+                    argv[6] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
+
+                    // operator_loc
+#line 170 "api_node.c.erb"
+                    argv[7] = pm_location_new(parser, cast->operator_loc.start, cast->operator_loc.end, source);
+
+                    // value
+#line 145 "api_node.c.erb"
+                    argv[8] = rb_ary_pop(value_stack);
+
+                    // location
+                    argv[9] = pm_location_new(parser, node->location.start, node->location.end, source);
+
+                    rb_ary_push(value_stack, rb_class_new_instance(10, argv, rb_cPrismIndexOrWriteNode));
+                    break;
+                }
 #line 134 "api_node.c.erb"
                 case PM_INSTANCE_VARIABLE_AND_WRITE_NODE: {
                     pm_instance_variable_and_write_node_t *cast = (pm_instance_variable_and_write_node_t *) node;
@@ -2956,7 +3106,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[0] = ULONG2NUM(node->flags >> 2);
+                    argv[0] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[1] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -2986,7 +3136,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[3] = ULONG2NUM(node->flags >> 2);
+                    argv[3] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[4] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -3016,7 +3166,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[3] = ULONG2NUM(node->flags >> 2);
+                    argv[3] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[4] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -3401,7 +3551,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[4] = ULONG2NUM(node->flags >> 2);
+                    argv[4] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[5] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -3918,7 +4068,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[3] = ULONG2NUM(node->flags >> 2);
+                    argv[3] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[4] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -3973,7 +4123,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[4] = ULONG2NUM(node->flags >> 2);
+                    argv[4] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[5] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -4281,7 +4431,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[0] = ULONG2NUM(node->flags >> 2);
+                    argv[0] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // opening_loc
 #line 173 "api_node.c.erb"
@@ -4449,7 +4599,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[4] = ULONG2NUM(node->flags >> 2);
+                    argv[4] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[5] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -4506,7 +4656,7 @@ pm_ast_new(pm_parser_t *parser, pm_node_t *node, rb_encoding *encoding) {
 
                     // flags
 #line 179 "api_node.c.erb"
-                    argv[4] = ULONG2NUM(node->flags >> 2);
+                    argv[4] = ULONG2NUM(node->flags & ~PM_NODE_FLAG_COMMON_MASK);
 
                     // location
                     argv[5] = pm_location_new(parser, node->location.start, node->location.end, source);
@@ -4649,6 +4799,9 @@ Init_prism_api_node(void) {
     rb_cPrismImaginaryNode = rb_define_class_under(rb_cPrism, "ImaginaryNode", rb_cPrismNode);
     rb_cPrismImplicitNode = rb_define_class_under(rb_cPrism, "ImplicitNode", rb_cPrismNode);
     rb_cPrismInNode = rb_define_class_under(rb_cPrism, "InNode", rb_cPrismNode);
+    rb_cPrismIndexAndWriteNode = rb_define_class_under(rb_cPrism, "IndexAndWriteNode", rb_cPrismNode);
+    rb_cPrismIndexOperatorWriteNode = rb_define_class_under(rb_cPrism, "IndexOperatorWriteNode", rb_cPrismNode);
+    rb_cPrismIndexOrWriteNode = rb_define_class_under(rb_cPrism, "IndexOrWriteNode", rb_cPrismNode);
     rb_cPrismInstanceVariableAndWriteNode = rb_define_class_under(rb_cPrism, "InstanceVariableAndWriteNode", rb_cPrismNode);
     rb_cPrismInstanceVariableOperatorWriteNode = rb_define_class_under(rb_cPrism, "InstanceVariableOperatorWriteNode", rb_cPrismNode);
     rb_cPrismInstanceVariableOrWriteNode = rb_define_class_under(rb_cPrism, "InstanceVariableOrWriteNode", rb_cPrismNode);