Expand descriptionWithoutComments to also remove #sourceLocation

Author: DougGregor

Sources/SwiftIfConfig/ActiveSyntaxRewriter.swift

@@ -382,12 +382,26 @@ extension IfConfigDeclSyntax {
 
 extension SyntaxProtocol {
   // Produce the source code for this syntax node with all of the comments
-  // removed. Each comment will be replaced with either a newline or a space,
-  // depending on whether the comment involved a newline.
+  // and #sourceLocations removed. Each comment will be replaced with either
+  // a newline or a space, depending on whether the comment involved a newline.
   @_spi(Compiler)
-  public var descriptionWithoutComments: String {
+  public var descriptionWithoutCommentsAndSourceLocations: String {
     var result = ""
+    var skipUntilRParen = false
     for token in tokens(viewMode: .sourceAccurate) {
+      // Skip #sourceLocation(...).
+      if token.tokenKind == .poundSourceLocation {
+        skipUntilRParen = true
+        continue
+      }
+
+      if skipUntilRParen {
+        if token.tokenKind == .rightParen {
+          skipUntilRParen = false
+        }
+        continue
+      }
+
       token.leadingTrivia.writeWithoutComments(to: &result)
       token.text.write(to: &result)
       token.trailingTrivia.writeWithoutComments(to: &result)

Tests/SwiftIfConfigTest/VisitorTests.swift

@@ -300,12 +300,13 @@ public class VisitorTests: XCTestCase {
     )
   }
 
-  func testRemoveComments() {
+  func testRemoveCommentsAndSourceLocations() {
     let original: SourceFileSyntax = """
 
       /// This is a documentation comment
       func f() { }
 
+      #sourceLocation(file: "if-configs.swift", line: 200)
       /** Another documentation comment
           that is split across
           multiple lines */
@@ -318,14 +319,13 @@ public class VisitorTests: XCTestCase {
       """
 
     assertStringsEqualWithDiff(
-      original.descriptionWithoutComments,
+      original.descriptionWithoutCommentsAndSourceLocations,
       """
 
        
       func f() { }
 
 
-
       func g() { }
 
       func h() {