Ensure newline after trailing line comments to prevent formatting issues

Author: TTOzzi

Sources/SwiftFormat/PrettyPrint/PrettyPrint.swift

@@ -434,7 +434,14 @@ public class PrettyPrinter {
         mustBreak = true
       }
 
-      let suppressBreaking = isBreakingSuppressed && !overrideBreakingSuppressed
+      let previousToken = idx > 0 ? tokens[idx - 1] : nil
+      let isAfterLineComment: Bool
+      if case .comment(let comment, _)? = previousToken {
+        isAfterLineComment = comment.kind == .line || comment.kind == .docLine
+      } else {
+        isAfterLineComment = false
+      }
+      let suppressBreaking = !isAfterLineComment && isBreakingSuppressed && !overrideBreakingSuppressed
       if !suppressBreaking && (!canFit(length) || mustBreak) {
         currentLineIsContinuation = isContinuationIfBreakFires
         if case .escaped = newline {

Tests/SwiftFormatTests/PrettyPrint/AttributeTests.swift

@@ -623,4 +623,42 @@ final class AttributeTests: PrettyPrintTestCase {
 
     assertPrettyPrintEqual(input: input, expected: expected, linelength: 45)
   }
+
+  func testAttributesWithComment() {
+    let input =
+      """
+      @foo // comment
+      @bar
+      import Baz
+
+      """
+    let expected =
+      """
+      @foo  // comment
+      @bar import Baz
+
+      """
+
+    assertPrettyPrintEqual(input: input, expected: expected, linelength: 45)
+  }
+
+  func testAttributesWithLineAndBlockComments() {
+    let input =
+      """
+      @foo // comment
+      @bar /* comment */
+      @zoo // comment
+      import Baz
+
+      """
+    let expected =
+      """
+      @foo  // comment
+      @bar /* comment */ @zoo  // comment
+      import Baz
+
+      """
+
+    assertPrettyPrintEqual(input: input, expected: expected, linelength: 45)
+  }
 }