[formatter] Add more cases for annotations

Author: AbdullinAM

core/src/main/java/com/facebook/ktfmt/format/KotlinInputAstVisitor.kt

@@ -2240,7 +2240,14 @@ open class KotlinInputAstVisitor(
         visit(psi)
       }
 
-      if (onlyAnnotationsSoFar && forceAnnotationBreaks && psi is KtAnnotationEntry) {
+      val shouldForceBreak = forceAnnotationBreaks &&
+              // don't force break on parameter annotations
+              list.parent !is KtParameter &&
+              // don't force break on receiver type annotations
+              !(list.parent is KtTypeReference && list.parent.parent is KtFunction) &&
+              // don't force break on parameter type annotations
+              !(list.parent is KtTypeReference && list.parent.parent is KtParameter)
+      if (onlyAnnotationsSoFar && shouldForceBreak) {
         builder.forcedBreak()
       } else if (onlyAnnotationsSoFar) {
         builder.breakOp(" ")

core/src/test/resources/cases/new_codestyle/AnnotationsOnParameters.input

@@ -0,0 +1,28 @@
+private class AnnotatedConstructor(
+    @JvmField @Volatile private var state: Int,
+    @get:JvmName("ctx") val context: CoroutineContext,
+    @Suppress("UNUSED_PARAMETER") unused: String = "",
+    @property:Inject val dependency: Dependency,
+    @Deprecated("Use context instead", ReplaceWith("context")) val legacyContext: CoroutineContext,
+)
+
+private class AnnotatedParameters {
+    fun single(@NotNull a: String) = a
+
+    fun multiple(@NotNull @Size(min = 1) a: String, @Nullable b: String?) = a + b
+
+    fun wrapped(
+        @NotNull @Size(min = 1, max = 100) someVeryLongParameterName: String,
+        @Nullable anotherRatherLongParameterName: String? = null,
+    ) = someVeryLongParameterName + anotherRatherLongParameterName
+
+    fun varargs(@NotNull vararg values: String) = values.size
+
+    fun higherOrder(@NotNull block: (@Nullable Int) -> Unit) = block(1)
+
+    fun @receiver:NotNull String.extension(@NotNull other: String) = this + other
+
+    fun withLambda(@NotNull name: String, @Nullable block: () -> Unit = {}) {
+        block()
+    }
+}

core/src/test/resources/cases/new_codestyle/AnnotationsOnParameters.new.output

@@ -0,0 +1,28 @@
+private class AnnotatedConstructor(
+    @JvmField @Volatile private var state: Int,
+    @get:JvmName("ctx") val context: CoroutineContext,
+    @Suppress("UNUSED_PARAMETER") unused: String = "",
+    @property:Inject val dependency: Dependency,
+    @Deprecated("Use context instead", ReplaceWith("context")) val legacyContext: CoroutineContext,
+)
+
+private class AnnotatedParameters {
+    fun single(@NotNull a: String) = a
+
+    fun multiple(@NotNull @Size(min = 1) a: String, @Nullable b: String?) = a + b
+
+    fun wrapped(
+        @NotNull @Size(min = 1, max = 100) someVeryLongParameterName: String,
+        @Nullable anotherRatherLongParameterName: String? = null,
+    ) = someVeryLongParameterName + anotherRatherLongParameterName
+
+    fun varargs(@NotNull vararg values: String) = values.size
+
+    fun higherOrder(@NotNull block: (@Nullable Int) -> Unit) = block(1)
+
+    fun @receiver:NotNull String.extension(@NotNull other: String) = this + other
+
+    fun withLambda(@NotNull name: String, @Nullable block: () -> Unit = {}) {
+        block()
+    }
+}

core/src/test/resources/cases/new_codestyle/AnnotationsOnParameters.output

@@ -0,0 +1,28 @@
+private class AnnotatedConstructor(
+    @JvmField @Volatile private var state: Int,
+    @get:JvmName("ctx") val context: CoroutineContext,
+    @Suppress("UNUSED_PARAMETER") unused: String = "",
+    @property:Inject val dependency: Dependency,
+    @Deprecated("Use context instead", ReplaceWith("context")) val legacyContext: CoroutineContext,
+)
+
+private class AnnotatedParameters {
+  fun single(@NotNull a: String) = a
+
+  fun multiple(@NotNull @Size(min = 1) a: String, @Nullable b: String?) = a + b
+
+  fun wrapped(
+      @NotNull @Size(min = 1, max = 100) someVeryLongParameterName: String,
+      @Nullable anotherRatherLongParameterName: String? = null,
+  ) = someVeryLongParameterName + anotherRatherLongParameterName
+
+  fun varargs(@NotNull vararg values: String) = values.size
+
+  fun higherOrder(@NotNull block: (@Nullable Int) -> Unit) = block(1)
+
+  fun @receiver:NotNull String.extension(@NotNull other: String) = this + other
+
+  fun withLambda(@NotNull name: String, @Nullable block: () -> Unit = {}) {
+    block()
+  }
+}