TrailingCommaInspection: should suggest fixes for call-site without warnings

#KT-39131 Fixed

idea/src/org/jetbrains/kotlin/idea/inspections/TrailingCommaInspection.kt

@@ -32,7 +32,7 @@ import kotlin.properties.Delegates
 
 class TrailingCommaInspection(
     @JvmField
-    var addCommaWarning: Boolean = false,
+    var addCommaWarning: Boolean = false
 ) : AbstractKotlinInspection() {
     override fun buildVisitor(holder: ProblemsHolder, isOnTheFly: Boolean): PsiElementVisitor = object : TrailingCommaVisitor() {
         override val recursively: Boolean = false
@@ -40,9 +40,8 @@ class TrailingCommaInspection(
 
         override fun process(trailingCommaContext: TrailingCommaContext) {
             val element = trailingCommaContext.ktElement
-            if (!element.canAddTrailingCommaWithRegistryCheck()) return
+            val kotlinCustomSettings = CodeStyle.getSettings(element.project).kotlinCustomSettings
-
+            useTrailingComma = kotlinCustomSettings.addTrailingCommaIsAllowedFor(element)
-            useTrailingComma = CodeStyle.getSettings(element.project).kotlinCustomSettings.ALLOW_TRAILING_COMMA
             when (trailingCommaContext.state) {
                 TrailingCommaState.MISSING, TrailingCommaState.EXISTS -> {
                     checkCommaPosition(element)
@@ -161,6 +160,7 @@ class TrailingCommaInspection(
                 val range = createFormatterTextRange(element)
                 val settings = CodeStyle.getSettings(project).clone()
                 settings.kotlinCustomSettings.ALLOW_TRAILING_COMMA = true
+                settings.kotlinCustomSettings.ALLOW_TRAILING_COMMA_ON_CALL_SITE = true
                 CodeStyle.doWithTemporarySettings(project, settings) {
                     CodeStyleManager.getInstance(project).reformatRange(element, range.startOffset, range.endOffset)
                 }

idea/testData/inspectionsLocal/trailingComma/destructuringDeclarations.kt

@@ -0,0 +1,8 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+fun a() {
+    val (a,
+    b,<caret>) = 1 to 2
+}

idea/testData/inspectionsLocal/trailingComma/destructuringDeclarations.kt.after

@@ -0,0 +1,10 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+fun a() {
+    val (
+        a,
+        b,
+    ) = 1 to 2
+}

idea/testData/inspectionsLocal/trailingComma/destructuringDeclarationsInLambda.kt

@@ -0,0 +1,8 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Fix comma position
+// DISABLE-ERRORS
+
+fun a() {
+    val a = { (a, b // awd
+              ,<caret>/**/), c, -> }
+}

idea/testData/inspectionsLocal/trailingComma/destructuringDeclarationsInLambda.kt.after

@@ -0,0 +1,10 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Fix comma position
+// DISABLE-ERRORS
+
+fun a() {
+    val a = {
+            (
+                a, b, // awd/**/
+            ), c, -> }
+}

idea/testData/inspectionsLocal/trailingComma/functionArguments.kt

@@ -0,0 +1,8 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add trailing comma
+// DISABLE-ERRORS
+
+fun a() {
+    b(1, 3, 2424,
+    awdawd<caret>)
+}

idea/testData/inspectionsLocal/trailingComma/functionArguments.kt.after

@@ -0,0 +1,10 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add trailing comma
+// DISABLE-ERRORS
+
+fun a() {
+    b(
+        1, 3, 2424,
+        awdawd,
+    )
+}

idea/testData/inspectionsLocal/trailingComma/indices.kt

@@ -0,0 +1,8 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+fun a() {
+    g[<caret>1, 2, 3,/**/
+    ]
+}

idea/testData/inspectionsLocal/trailingComma/indices.kt.after

@@ -0,0 +1,9 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+fun a() {
+    g[
+            1, 2, 3,/**/
+    ]
+}

idea/testData/inspectionsLocal/trailingComma/inner.kt

@@ -0,0 +1,7 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun a(
+    a: Int, b: Any = fun(a: Int,) {},<caret>) {
+
+}

idea/testData/inspectionsLocal/trailingComma/inner.kt.after

@@ -0,0 +1,8 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun a(
+    a: Int, b: Any = fun(a: Int) {},
+) {
+
+}

idea/testData/inspectionsLocal/trailingComma/lambda.kt

@@ -0,0 +1,9 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun a() {
+    val a = { a: Int,
+    b: Int, <caret>->
+
+    }
+}

idea/testData/inspectionsLocal/trailingComma/lambda.kt.after

@@ -0,0 +1,11 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun a() {
+    val a = {
+            a: Int,
+            b: Int,
+        ->
+
+    }
+}

idea/testData/inspectionsLocal/trailingComma/literal.kt

@@ -0,0 +1,8 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+@Anno(
+      [<caret>1, 2, 3,
+      ])
+class A

idea/testData/inspectionsLocal/trailingComma/literal.kt.after

@@ -0,0 +1,9 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+@Anno(
+    [
+        1, 2, 3,
+    ])
+class A

idea/testData/inspectionsLocal/trailingComma/simple.kt

@@ -0,0 +1,7 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// PROBLEM: none
+
+fun a(<caret>
+    a: Int,) {
+
+}

idea/testData/inspectionsLocal/trailingComma/typeArguments.kt

@@ -0,0 +1,8 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+fun a() {
+    b<<caret>Int,
+    >()
+}

idea/testData/inspectionsLocal/trailingComma/typeArguments.kt.after

@@ -0,0 +1,9 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+// DISABLE-ERRORS
+
+fun a() {
+    b<
+            Int,
+            >()
+}

idea/testData/inspectionsLocal/trailingComma/typeParameters.kt

@@ -0,0 +1,5 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun <<caret>T,
+        B,> a() = Unit

idea/testData/inspectionsLocal/trailingComma/typeParameters.kt.after

@@ -0,0 +1,7 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun <
+        T,
+        B,
+        > a() = Unit

idea/testData/inspectionsLocal/trailingComma/whenEntry.kt

@@ -0,0 +1,9 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun a() {
+    when (val b = 5) {
+        1, 2,
+        3, <caret>->
+    }
+}

idea/testData/inspectionsLocal/trailingComma/whenEntry.kt.after

@@ -0,0 +1,10 @@
+// COMPILER_ARGUMENTS: -XXLanguage:+TrailingCommas
+// FIX: Add line break
+
+fun a() {
+    when (val b = 5) {
+        1, 2,
+        3,
+        ->
+    }
+}

idea/tests/org/jetbrains/kotlin/idea/inspections/LocalInspectionTestGenerated.java

@@ -13115,6 +13115,41 @@ public class LocalInspectionTestGenerated extends AbstractLocalInspectionTest {
             runTest("idea/testData/inspectionsLocal/trailingComma/changeCommaPosition5.kt");
         }
 
+        @TestMetadata("destructuringDeclarations.kt")
+        public void testDestructuringDeclarations() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/destructuringDeclarations.kt");
+        }
+
+        @TestMetadata("destructuringDeclarationsInLambda.kt")
+        public void testDestructuringDeclarationsInLambda() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/destructuringDeclarationsInLambda.kt");
+        }
+
+        @TestMetadata("functionArguments.kt")
+        public void testFunctionArguments() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/functionArguments.kt");
+        }
+
+        @TestMetadata("indices.kt")
+        public void testIndices() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/indices.kt");
+        }
+
+        @TestMetadata("inner.kt")
+        public void testInner() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/inner.kt");
+        }
+
+        @TestMetadata("lambda.kt")
+        public void testLambda() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/lambda.kt");
+        }
+
+        @TestMetadata("literal.kt")
+        public void testLiteral() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/literal.kt");
+        }
+
         @TestMetadata("missingLineBreak.kt")
         public void testMissingLineBreak() throws Exception {
             runTest("idea/testData/inspectionsLocal/trailingComma/missingLineBreak.kt");
@@ -13149,6 +13184,26 @@ public class LocalInspectionTestGenerated extends AbstractLocalInspectionTest {
         public void testRemoveComma4() throws Exception {
             runTest("idea/testData/inspectionsLocal/trailingComma/removeComma4.kt");
         }
+
+        @TestMetadata("simple.kt")
+        public void testSimple() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/simple.kt");
+        }
+
+        @TestMetadata("typeArguments.kt")
+        public void testTypeArguments() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/typeArguments.kt");
+        }
+
+        @TestMetadata("typeParameters.kt")
+        public void testTypeParameters() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/typeParameters.kt");
+        }
+
+        @TestMetadata("whenEntry.kt")
+        public void testWhenEntry() throws Exception {
+            runTest("idea/testData/inspectionsLocal/trailingComma/whenEntry.kt");
+        }
     }
 
     @TestMetadata("idea/testData/inspectionsLocal/unlabeledReturnInsideLambda")