[ruby/prism] Unescape method name for unary methods.

https://github.com/ruby/prism/commit/4386a4c0da
This commit is contained in:
Matt Valentine-House 2024-01-18 19:51:25 +00:00 committed by git
parent b4b35b1468
commit b8495048fb
2 changed files with 22 additions and 11 deletions

View File

@ -12719,20 +12719,24 @@ parse_string_part(pm_parser_t *parser) {
* automatically drop trailing `@` characters. This happens at the parser level, * automatically drop trailing `@` characters. This happens at the parser level,
* such that `~@` is parsed as `~` and `!@` is parsed as `!`. We do that here. * such that `~@` is parsed as `~` and `!@` is parsed as `!`. We do that here.
*/ */
static const uint8_t *
parse_operator_symbol_name(const pm_token_t *name) {
switch (name->type) {
case PM_TOKEN_TILDE:
case PM_TOKEN_BANG:
if (name->end[-1] == '@') return name->end - 1;
/* fallthrough */
default:
return name->end;
}
}
static pm_node_t * static pm_node_t *
parse_operator_symbol(pm_parser_t *parser, const pm_token_t *opening, pm_lex_state_t next_state) { parse_operator_symbol(pm_parser_t *parser, const pm_token_t *opening, pm_lex_state_t next_state) {
pm_token_t closing = not_provided(parser); pm_token_t closing = not_provided(parser);
pm_symbol_node_t *symbol = pm_symbol_node_create(parser, opening, &parser->current, &closing); pm_symbol_node_t *symbol = pm_symbol_node_create(parser, opening, &parser->current, &closing);
const uint8_t *end = parser->current.end; const uint8_t *end = parse_operator_symbol_name(&parser->current);
switch (parser->current.type) {
case PM_TOKEN_TILDE:
case PM_TOKEN_BANG:
if (parser->current.end[-1] == '@') end--;
break;
default:
break;
}
if (next_state != PM_LEX_STATE_NONE) lex_state_set(parser, next_state); if (next_state != PM_LEX_STATE_NONE) lex_state_set(parser, next_state);
parser_lex(parser); parser_lex(parser);
@ -15379,6 +15383,13 @@ parse_expression_prefix(pm_parser_t *parser, pm_binding_power_t binding_power, b
pm_parser_scope_pop(parser); pm_parser_scope_pop(parser);
pm_parser_current_param_name_restore(parser, saved_param_name); pm_parser_current_param_name_restore(parser, saved_param_name);
/**
* If the final character is @. As is the case when defining
* methods to override the unary operators, we should ignore
* the @ in the same way we do for symbols.
*/
name.end = parse_operator_symbol_name(&name);
return (pm_node_t *) pm_def_node_create( return (pm_node_t *) pm_def_node_create(
parser, parser,
&name, &name,

View File

@ -4,8 +4,8 @@
@ StatementsNode (location: (1,0)-(1,17)) @ StatementsNode (location: (1,0)-(1,17))
└── body: (length: 1) └── body: (length: 1)
└── @ DefNode (location: (1,0)-(1,17)) └── @ DefNode (location: (1,0)-(1,17))
├── name: :"!@" ├── name: :!
├── name_loc: (1,4)-(1,6) = "!@" ├── name_loc: (1,4)-(1,5) = "!"
├── receiver: ∅ ├── receiver: ∅
├── parameters: ∅ ├── parameters: ∅
├── body: ├── body: