Handle parsing local variable singleton method definition

ruby · Sep 13, 2023 · 943204d · 943204d
1 parent 41430f4
commit 943204d
Show file tree

Hide file tree

Showing 3 changed files with 60 additions and 32 deletions.
diff --git a/src/yarp.c b/src/yarp.c
@@ -11806,12 +11806,12 @@ parse_expression_prefix(yp_parser_t *parser, yp_binding_power_t binding_power) {
                     name = parser->previous;
                     break;
                 case YP_TOKEN_IDENTIFIER: {
-                    yp_parser_scope_push(parser, true);
                     parser_lex(parser);
 
                     if (match_any_type_p(parser, 2, YP_TOKEN_DOT, YP_TOKEN_COLON_COLON)) {
                         receiver = parse_variable_call(parser);
 
+                        yp_parser_scope_push(parser, true);
                         lex_state_set(parser, YP_LEX_STATE_FNAME);
                         parser_lex(parser);
 
@@ -11822,6 +11822,7 @@ parse_expression_prefix(yp_parser_t *parser, yp_binding_power_t binding_power) {
                             yp_diagnostic_list_append(&parser->error_list, parser->previous.start, parser->previous.end, YP_ERR_DEF_NAME_AFTER_RECEIVER);
                         }
                     } else {
+                        yp_parser_scope_push(parser, true);
                         name = parser->previous;
                     }
 

diff --git a/test/yarp/fixtures/methods.txt b/test/yarp/fixtures/methods.txt
@@ -163,3 +163,6 @@ end
 
 def foo(_a, _a, b, c)
 end
+
+foo = 1
+def foo.bar; end
diff --git a/test/yarp/snapshots/methods.txt b/test/yarp/snapshots/methods.txt