summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--prism/prism.c13
-rw-r--r--test/prism/snapshots/seattlerb/defn_unary_not.txt2
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: ∅