diff options
-rw-r--r-- | prism/prism.c | 13 | ||||
-rw-r--r-- | test/prism/snapshots/seattlerb/defn_unary_not.txt | 2 |
2 files changed, 9 insertions, 6 deletions
diff --git a/prism/prism.c b/prism/prism.c index 9df29a2de5..533280c0b9 100644 --- a/prism/prism.c +++ b/prism/prism.c @@ -2890,7 +2890,8 @@ pm_def_node_receiver_check(pm_parser_t *parser, const pm_node_t *node) { static pm_def_node_t * pm_def_node_create( pm_parser_t *parser, - const pm_token_t *name, + pm_constant_id_t name, + const pm_location_t *name_loc, pm_node_t *receiver, pm_parameters_node_t *parameters, pm_node_t *body, @@ -2920,8 +2921,8 @@ pm_def_node_create( .type = PM_DEF_NODE, .location = { .start = def_keyword->start, .end = end }, }, - .name = pm_parser_constant_id_token(parser, name), - .name_loc = PM_LOCATION_TOKEN_VALUE(name), + .name = name, + .name_loc = *name_loc, .receiver = receiver, .parameters = parameters, .body = body, @@ -15626,11 +15627,13 @@ parse_expression_prefix(pm_parser_t *parser, pm_binding_power_t binding_power, b * 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); + pm_constant_id_t name_id = pm_parser_constant_id_token(parser, &name); + pm_location_t name_loc = { .start = name.start, .end = parse_operator_symbol_name(&name) }; return (pm_node_t *) pm_def_node_create( parser, - &name, + name_id, + &name_loc, receiver, params, statements, diff --git a/test/prism/snapshots/seattlerb/defn_unary_not.txt b/test/prism/snapshots/seattlerb/defn_unary_not.txt index df2e3a87bb..a659aeb9c2 100644 --- a/test/prism/snapshots/seattlerb/defn_unary_not.txt +++ b/test/prism/snapshots/seattlerb/defn_unary_not.txt @@ -4,7 +4,7 @@ @ StatementsNode (location: (1,0)-(1,17)) └── body: (length: 1) └── @ DefNode (location: (1,0)-(1,17)) - ├── name: :! + ├── name: :"!@" ├── name_loc: (1,4)-(1,5) = "!" ├── receiver: ∅ ├── parameters: ∅ |