std.zig.Ast: store lbrace and rbrace token in data of .error_set_decl

This makes the `.data` field the better choice over the `.main_token` for this tag.
This commit is contained in:
Techatrix 2025-02-25 15:29:51 +01:00
parent d84055f9c6
commit 4129f7ff5a
No known key found for this signature in database
5 changed files with 15 additions and 13 deletions

View File

@ -444,9 +444,7 @@ fn walkExpression(w: *Walk, node: Ast.Node.Index) Error!void {
}, },
.error_set_decl => { .error_set_decl => {
const error_token = ast.nodeMainToken(node); const lbrace, const rbrace = ast.nodeData(node).token_and_token;
const lbrace = error_token + 1;
const rbrace = ast.nodeData(node).token;
var i = lbrace + 1; var i = lbrace + 1;
while (i < rbrace) : (i += 1) { while (i < rbrace) : (i += 1) {

View File

@ -996,9 +996,8 @@ pub fn lastToken(tree: Ast, node: Node.Index) TokenIndex {
.unwrap_optional, .unwrap_optional,
.asm_simple, .asm_simple,
=> return tree.nodeData(n).node_and_token[1] + end_offset, => return tree.nodeData(n).node_and_token[1] + end_offset,
.error_set_decl => return tree.nodeData(n).token + end_offset,
.grouped_expression, .asm_input => return tree.nodeData(n).node_and_token[1] + end_offset, .grouped_expression, .asm_input => return tree.nodeData(n).node_and_token[1] + end_offset,
.multiline_string_literal => return tree.nodeData(n).token_and_token[1] + end_offset, .multiline_string_literal, .error_set_decl => return tree.nodeData(n).token_and_token[1] + end_offset,
.asm_output => return tree.nodeData(n).opt_node_and_token[1] + end_offset, .asm_output => return tree.nodeData(n).opt_node_and_token[1] + end_offset,
.error_value => return tree.nodeMainToken(n) + 2 + end_offset, .error_value => return tree.nodeMainToken(n) + 2 + end_offset,
@ -3758,7 +3757,9 @@ pub const Node = struct {
builtin_call_comma, builtin_call_comma,
/// `error{a, b}`. /// `error{a, b}`.
/// ///
/// The `data` field is a `.token` to the '}'. /// The `data` field is a `.token_and_token`:
/// 1. a `TokenIndex` to the `{` token.
/// 2. a `TokenIndex` to the `}` token.
/// ///
/// The `main_token` field is the `error`. /// The `main_token` field is the `error`.
error_set_decl, error_set_decl,

View File

@ -5974,9 +5974,9 @@ fn errorSetDecl(gz: *GenZir, ri: ResultInfo, node: Ast.Node.Index) InnerError!Zi
var idents: std.AutoHashMapUnmanaged(Zir.NullTerminatedString, Ast.TokenIndex) = .empty; var idents: std.AutoHashMapUnmanaged(Zir.NullTerminatedString, Ast.TokenIndex) = .empty;
defer idents.deinit(gpa); defer idents.deinit(gpa);
const error_token = tree.nodeMainToken(node); const lbrace, const rbrace = tree.nodeData(node).token_and_token;
var tok_i = error_token + 2; for (lbrace + 1..rbrace) |i| {
while (true) : (tok_i += 1) { const tok_i: Ast.TokenIndex = @intCast(i);
switch (tree.tokenTag(tok_i)) { switch (tree.tokenTag(tok_i)) {
.doc_comment, .comma => {}, .doc_comment, .comma => {},
.identifier => { .identifier => {
@ -6003,7 +6003,6 @@ fn errorSetDecl(gz: *GenZir, ri: ResultInfo, node: Ast.Node.Index) InnerError!Zi
try astgen.extra.append(gpa, @intFromEnum(str_index)); try astgen.extra.append(gpa, @intFromEnum(str_index));
fields_len += 1; fields_len += 1;
}, },
.r_brace => break,
else => unreachable, else => unreachable,
} }
} }

View File

@ -2763,7 +2763,12 @@ fn parsePrimaryTypeExpr(p: *Parse) !?Node.Index {
return try p.addNode(.{ return try p.addNode(.{
.tag = .error_set_decl, .tag = .error_set_decl,
.main_token = error_token, .main_token = error_token,
.data = .{ .token = p.tok_i - 1 }, // rbrace .data = .{
.token_and_token = .{
error_token + 1, // lbrace
p.tok_i - 1, // rbrace
},
},
}); });
}, },
else => { else => {

View File

@ -739,8 +739,7 @@ fn renderExpression(r: *Render, node: Ast.Node.Index, space: Space) Error!void {
.error_set_decl => { .error_set_decl => {
const error_token = tree.nodeMainToken(node); const error_token = tree.nodeMainToken(node);
const lbrace = error_token + 1; const lbrace, const rbrace = tree.nodeData(node).token_and_token;
const rbrace = tree.nodeData(node).token;
try renderToken(r, error_token, .none); try renderToken(r, error_token, .none);