Fix parsing keyword args starting with plus and minus

ruby · Sep 28, 2023 · 95913a4 · 95913a4
1 parent 62469a5
commit 95913a4
Show file tree

Hide file tree

Showing 3 changed files with 158 additions and 17 deletions.
diff --git a/src/prism.c b/src/prism.c
@@ -6912,7 +6912,7 @@ parser_lex(pm_parser_t *parser) {
                         );
                     }
 
-                    if (lex_state_beg_p(parser) || spcarg) {
+                    if (lex_state_beg_p(parser) || spcarg || parser->in_keyword_arg) {
                         lex_state_set(parser, PM_LEX_STATE_BEG);
 
                         if (pm_char_is_decimal_digit(peek(parser))) {
@@ -6961,7 +6961,7 @@ parser_lex(pm_parser_t *parser) {
                         );
                     }
 
-                    if (lex_state_beg_p(parser) || spcarg) {
+                    if (lex_state_beg_p(parser) || spcarg || parser->in_keyword_arg) {
                         lex_state_set(parser, PM_LEX_STATE_BEG);
                         LEX(pm_char_is_decimal_digit(peek(parser)) ? PM_TOKEN_UMINUS_NUM : PM_TOKEN_UMINUS);
                     }

diff --git a/test/prism/fixtures/methods.txt b/test/prism/fixtures/methods.txt
@@ -166,3 +166,9 @@ end
 
 foo = 1
 def foo.bar; end
+
+def f x:-a; end
+
+def f x:+a; end
+
+def f x:!a; end
diff --git a/test/prism/snapshots/methods.txt b/test/prism/snapshots/methods.txt