From c33966167c4039f1db135d219c7d2bc0c8fffea0 Mon Sep 17 00:00:00 2001
From: rpkak <rpkak@users.noreply.github.com>
Date: Fri, 4 Apr 2025 11:07:15 +0200
Subject: [PATCH] zig fmt / std.zig.Ast: count addrspace tokens correctly

---
 lib/std/zig/Ast.zig         |  3 +++
 lib/std/zig/parser_test.zig | 10 ++++++++++
 2 files changed, 13 insertions(+)

diff --git a/lib/std/zig/Ast.zig b/lib/std/zig/Ast.zig
index 72ca20a6c0a6..c9294aa0af45 100644
--- a/lib/std/zig/Ast.zig
+++ b/lib/std/zig/Ast.zig
@@ -1195,6 +1195,9 @@ pub fn lastToken(tree: Ast, node: Node.Index) TokenIndex {
                 if (extra.section_node.unwrap()) |section_node| {
                     end_offset += 1; // for the rparen
                     n = section_node;
+                } else if (extra.addrspace_node.unwrap()) |addrspace_node| {
+                    end_offset += 1; // for the rparen
+                    n = addrspace_node;
                 } else if (extra.align_node.unwrap()) |align_node| {
                     end_offset += 1; // for the rparen
                     n = align_node;
diff --git a/lib/std/zig/parser_test.zig b/lib/std/zig/parser_test.zig
index d08f5b60ba9d..2c673609191b 100644
--- a/lib/std/zig/parser_test.zig
+++ b/lib/std/zig/parser_test.zig
@@ -6074,6 +6074,16 @@ test "zig fmt: do not canonicalize invalid cast builtins" {
     );
 }
 
+test "zig fmt: extern addrspace in struct" {
+    try testCanonical(
+        \\const namespace = struct {
+        \\    extern const num: u8 addrspace(.generic);
+        \\};
+        \\// comment
+        \\
+    );
+}
+
 test "recovery: top level" {
     try testError(
         \\test "" {inline}