From 2541a149a583c964ea60f967891e8272373d89e1 Mon Sep 17 00:00:00 2001
From: Yuta Saito <kateinoigakukun@gmail.com>
Date: Fri, 14 Jun 2024 03:02:00 +0000
Subject: [PATCH] Fix `@_expose` attribute argument spacing

Pretty-print the `@_expose` with the correct spacing. It was formatted
as `@_expose(wasm,"foo")` instead of `@_expose(wasm, "foo")`.
---
 .../PrettyPrint/TokenStreamCreator.swift      |  5 ++++
 .../PrettyPrint/AttributeTests.swift          | 24 +++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/Sources/SwiftFormat/PrettyPrint/TokenStreamCreator.swift b/Sources/SwiftFormat/PrettyPrint/TokenStreamCreator.swift
index 08f1aa4f0..95525d058 100644
--- a/Sources/SwiftFormat/PrettyPrint/TokenStreamCreator.swift
+++ b/Sources/SwiftFormat/PrettyPrint/TokenStreamCreator.swift
@@ -1802,6 +1802,11 @@ fileprivate final class TokenStreamCreator: SyntaxVisitor {
     return .visitChildren
   }
 
+  override func visit(_ node: ExposeAttributeArgumentsSyntax) -> SyntaxVisitorContinueKind {
+    after(node.comma, tokens: .break(.same, size: 1))
+    return .visitChildren
+  }
+
   override func visit(_ node: AvailabilityLabeledArgumentSyntax) -> SyntaxVisitorContinueKind {
     before(node.label, tokens: .open)
 
diff --git a/Tests/SwiftFormatTests/PrettyPrint/AttributeTests.swift b/Tests/SwiftFormatTests/PrettyPrint/AttributeTests.swift
index 3031fc31b..3ff5db02d 100644
--- a/Tests/SwiftFormatTests/PrettyPrint/AttributeTests.swift
+++ b/Tests/SwiftFormatTests/PrettyPrint/AttributeTests.swift
@@ -444,4 +444,28 @@ final class AttributeTests: PrettyPrintTestCase {
 
     assertPrettyPrintEqual(input: input, expected: expected, linelength: 100)
   }
+
+  func testAttributeParamSpacingInExpose() {
+    let input =
+      """
+      @_expose( wasm  , "foo"  )
+      func f() {}
+
+      @_expose( Cxx  ,   "bar")
+      func b() {}
+
+      """
+
+    let expected =
+      """
+      @_expose(wasm, "foo")
+      func f() {}
+
+      @_expose(Cxx, "bar")
+      func b() {}
+
+      """
+
+    assertPrettyPrintEqual(input: input, expected: expected, linelength: 100)
+  }
 }