Added formatting support for single/multiline comments in parameters / record header (#6176)

Author: Jenson3210

rewrite-java/src/main/java/org/openrewrite/java/format/SpacesVisitor.java

@@ -242,7 +242,7 @@ public J.ClassDeclaration visitClassDeclaration(J.ClassDeclaration classDecl, P
 
         if (c.getPrimaryConstructor() != null) {
             c = c.withPrimaryConstructor(ListUtils.map(c.getPrimaryConstructor(), (ix, param) -> {
-                if (param.getPrefix().getLastWhitespace().contains("\n")) {
+                if (param.getPrefix().getWhitespace().contains("\n") || (!param.getPrefix().getComments().isEmpty() && param.getPrefix().getComments().stream().noneMatch(comment -> comment.getSuffix().contains("\n")))) {
                     return param;
                 }
                 if (ix == 0 && param.getComments().isEmpty()) {

rewrite-java/src/main/java/org/openrewrite/java/format/TabsAndIndentsVisitor.java

@@ -205,8 +205,9 @@ public Space visitSpace(Space space, Space.Location loc, P p) {
         IndentType indentType = getCursor().getParent().getNearestMessage("indentType", IndentType.ALIGN);
         if (right.getElement() instanceof J) {
             J elem = (J) right.getElement();
-            if ((right.getAfter().getLastWhitespace().contains("\n") ||
-                    elem.getPrefix().getLastWhitespace().contains("\n"))) {
+            if (right.getAfter().getLastWhitespace().contains("\n") ||
+                    elem.getPrefix().getLastWhitespace().contains("\n") ||
+                    (elem.getPrefix().getWhitespace().contains("\n") && elem.getPrefix().getComments().stream().noneMatch(c -> c.getSuffix().contains("\n")))) {
                 switch (loc) {
                     case FOR_CONDITION:
                     case FOR_UPDATE: {

rewrite-java/src/test/java/org/openrewrite/java/format/AutoFormatTest.java

@@ -2134,6 +2134,64 @@ record someRecord5(@Foo @Foo String name, int age
               ) {
               }
               """
+          ),
+          java(
+            """
+              record someRecord6(
+                                   String name,
+                                   /* some comment */ int age) {
+              }
+              """,
+            """
+              record someRecord6(
+                      String name,
+                      /* some comment */ int age) {
+              }
+              """
+          ),
+          java(
+            """
+              record someRecord7(
+                                   String name,
+                                   // some comment
+                                   int age) {
+              }
+              """,
+            """
+              record someRecord7(
+                      String name,
+                      // some comment
+                      int age) {
+              }
+              """
+          ),
+          java(
+            """
+              record someRecord8(
+                                   String name, // some comment
+                                   int age) {
+              }
+              """,
+            """
+              record someRecord8(
+                      String name, // some comment
+                      int age) {
+              }
+              """
+          ),
+          java(
+            """
+              record someRecord9(
+                                     String name,       /* some comment */ int age
+              ) {
+              }
+              """,
+            """
+              record someRecord9(
+                      String name,       /* some comment */ int age
+              ) {
+              }
+              """
           )
         );
     }
@@ -2302,6 +2360,78 @@ void someMethod5(String name, int age
                   }
               }
               """
+          ),
+          java(
+            """
+              class Test6 {
+                  void someMethod6(
+                                       String name,
+                                       /* some comment */ int age) {
+                  }
+              }
+              """,
+            """
+              class Test6 {
+                  void someMethod6(
+                          String name,
+                          /* some comment */ int age) {
+                  }
+              }
+              """
+          ),
+          java(
+            """
+              class Test7 {
+                  void someMethod7(
+                                       String name,
+                                       // some comment
+                                       int age) {
+                  }
+              }
+              """,
+            """
+              class Test7 {
+                  void someMethod7(
+                          String name,
+                          // some comment
+                          int age) {
+                  }
+              }
+              """
+          ),
+          java(
+            """
+              class Test8 {
+                  void someMethod8(
+                                       String name, // some comment
+                                       int age) {
+                  }
+              }
+              """,
+            """
+              class Test8 {
+                  void someMethod8(
+                          String name, // some comment
+                          int age) {
+                  }
+              }
+              """
+          ),
+          java(
+            """
+              class Test9 {
+                  void someMethod9(
+                                       String name,       /* some comment */ int age) {
+                  }
+              }
+              """,
+            """
+              class Test9 {
+                  void someMethod9(
+                          String name,       /* some comment */ int age) {
+                  }
+              }
+              """
           )
         );
     }