Don't enforce LF before annotations on primary constructor

openrewrite · Oct 2, 2023 · a66d25d · a66d25d
1 parent fa88e44
commit a66d25d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/format/WrappingAndBracesVisitor.java b/src/main/java/org/openrewrite/kotlin/format/WrappingAndBracesVisitor.java
@@ -22,6 +22,7 @@
 import org.openrewrite.java.tree.*;
 import org.openrewrite.kotlin.KotlinIsoVisitor;
 import org.openrewrite.kotlin.marker.OmitBraces;
+import org.openrewrite.kotlin.marker.PrimaryConstructor;
 import org.openrewrite.kotlin.style.WrappingAndBracesStyle;
 
 import java.util.List;
@@ -97,6 +98,10 @@ public J.VariableDeclarations visitVariableDeclarations(J.VariableDeclarations m
     @Override
     public J.MethodDeclaration visitMethodDeclaration(J.MethodDeclaration method, P p) {
         J.MethodDeclaration m = super.visitMethodDeclaration(method, p);
+        if (m.getMarkers().findFirst(PrimaryConstructor.class).isPresent()) {
+            return m;
+        }
+
         m = m.withLeadingAnnotations(withNewlines(m.getLeadingAnnotations()));
 
         List<J.Modifier> modifiers = method.getModifiers();

diff --git a/src/test/java/org/openrewrite/kotlin/format/WrappingAndBracesTest.java b/src/test/java/org/openrewrite/kotlin/format/WrappingAndBracesTest.java
@@ -562,4 +562,15 @@ class Test {}
           )
         );
     }
+
+    @Test
+    void primaryConstructorAnnotations() {
+        rewriteRun(
+          kotlin(
+            """
+              class T @Suppress constructor()
+              """
+          )
+        );
+    }
 }