Update trailing lambda quickfix, add tests

Case of incorrect call expression as the receiver of dot-qualified expression
was not handled properly.

Now the following order is used:
- If parent is a call expression or if this call is a receiver of parent
dot-qualified expression, parent must get last lambda as a new receiver;
- Otherwise, if parent is a dot-qualified expression, this call is in selector position.
Then grandparent is checked for being call or dot-qualified expression. If so, trailing
lambda must become grandparent's new receiver. If not, trailing nodes become standalone
epressions after parent.
- Otherwise, incorrect call expression is a standalone expression, so trailing nodes
may be lifted and put after it.

idea/src/org/jetbrains/kotlin/idea/quickfix/AddSemicolonBeforeLambdaExpressionFix.kt

@@ -36,24 +36,34 @@ class AddSemicolonBeforeLambdaExpressionFix(element: KtLambdaExpression) : Kotli
                     lastLambdaAcceptor = parent,
                     nodeBeforeSemicolon = callExpression
                 )
-                // Incorrect call is a part of dot-qualified expression before it
+                // Incorrect call is a part of the dot-qualified expression before or after it
                 is KtDotQualifiedExpression -> {
                     val grandparent = parent.parent
-                    if (grandparent.isCallOrDotExpression) {
-                        // Similar to call expression parent, but here correct lambda receiver is one level higher,
-                        // since our parent is the dot expression to the left
-                        liftTrailingNodesAndRelocateLastLambda(
-                            psiFactory, callExpression, endOfCall,
-                            lastLambdaAcceptor = grandparent,
-                            nodeBeforeSemicolon = parent
-                        )
-                    } else {
-                        // Extract trailing lambdas (and possible formatting) two levels higher,
-                        // before dot-qualified expression to the left of this call expression
-                        liftTrailingNodes(
-                            psiFactory, callExpression, endOfCall,
-                            addNodesAfter = parent
-                        )
+                    when {
+                        // Call expression is a receiver of the dot expression. Give last lambda as a new receiver to that dot expression
+                        parent.receiverExpression === callExpression -> {
+                            liftTrailingNodesAndRelocateLastLambda(
+                                psiFactory, callExpression, endOfCall,
+                                lastLambdaAcceptor = parent,
+                                nodeBeforeSemicolon = callExpression
+                            )
+                        }
+                        // Call expression is the right node of parent dot expression, possible call / dot to the right is a grandparent
+                        // If grandparent is call or dot expression, last lambda becomes its new receiver
+                        grandparent.isCallOrDotExpression -> {
+                            liftTrailingNodesAndRelocateLastLambda(
+                                psiFactory, callExpression, endOfCall,
+                                lastLambdaAcceptor = grandparent,
+                                nodeBeforeSemicolon = parent
+                            )
+                        }
+                        // Parent is dot expression, but there is no call or dot after it, so just lift everything up
+                        else -> {
+                            liftTrailingNodes(
+                                psiFactory, callExpression, endOfCall,
+                                addNodesAfter = parent
+                            )
+                        }
                     }
                 }
                 // Simple case: extract all trailing nodes right after call - it is a standalone call expression

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsEnd.kt

@@ -0,0 +1,8 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+
+fun test() {
+    foo()
+    {<caret>"test"}.invoke().toString().toString()
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsEnd.kt.after

@@ -0,0 +1,8 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+
+fun test() {
+    foo();
+    {"test"}.invoke().toString().toString()
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsStart.kt

@@ -0,0 +1,6 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun test() {
+    "test".toString().toString().toString()
+    {<caret>}
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsStart.kt.after

@@ -0,0 +1,6 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun test() {
+    "test".toString().toString().toString();
+    {}
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsStartEnd.kt

@@ -0,0 +1,6 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun test() {
+    "test".toString().toString().toString()
+    {<caret>"test"}.invoke().toString().toString()
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsStartEnd.kt.after

@@ -0,0 +1,6 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun test() {
+    "test".toString().toString().toString();
+    {"test"}.invoke().toString().toString()
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/complex.kt

@@ -0,0 +1,14 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+fun String.withLambda(lambda: (String) -> Unit) {}
+
+fun test {
+    { { "test" } }.invoke()().toString().withLambda()
+    // comment and formatting
+    {} // correct trailing lambda
+    /*
+       block comment
+    */
+    { { { foo() }<caret> } }.invoke()().invoke()
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/complex.kt.after

@@ -0,0 +1,14 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+fun String.withLambda(lambda: (String) -> Unit) {}
+
+fun test {
+    { { "test" } }.invoke()().toString().withLambda()
+    // comment and formatting
+    {}; // correct trailing lambda
+    /*
+       block comment
+    */
+    { { { foo() } } }.invoke()().invoke()
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsEnd.kt

@@ -6,4 +6,4 @@ fun test {
     foo()
     // comment and formatting
     { { { foo() }<caret> } }()()()
-}
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsEnd.kt.after

@@ -6,4 +6,4 @@ fun test {
     foo();
     // comment and formatting
     { { { foo() } } }()()()
-}
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsStart.kt

@@ -0,0 +1,9 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+
+fun test {
+    { { { foo() } } }()()()
+    // comment and formatting
+    {<caret>}
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsStart.kt.after

@@ -0,0 +1,9 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+
+fun test {
+    { { { foo() } } }()()();
+    // comment and formatting
+    {}
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsStartEnd.kt

@@ -0,0 +1,9 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+
+fun test {
+    { { { foo() } } }()()()
+    // comment and formatting
+    { { { foo() }<caret> } }()()()
+}

idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsStartEnd.kt.after

@@ -0,0 +1,9 @@
+// "Terminate preceding call with semicolon" "true"
+
+fun foo() {}
+
+fun test {
+    { { { foo() } } }()()();
+    // comment and formatting
+    { { { foo() } } }()()()
+}

idea/tests/org/jetbrains/kotlin/idea/quickfix/QuickFixTestGenerated.java

@@ -1113,14 +1113,44 @@ public class QuickFixTestGenerated extends AbstractQuickFixTest {
             runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/basic.kt");
         }
 
+        @TestMetadata("chainedDotsEnd.kt")
+        public void testChainedDotsEnd() throws Exception {
+            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsEnd.kt");
+        }
+
+        @TestMetadata("chainedDotsStart.kt")
+        public void testChainedDotsStart() throws Exception {
+            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsStart.kt");
+        }
+
+        @TestMetadata("chainedDotsStartEnd.kt")
+        public void testChainedDotsStartEnd() throws Exception {
+            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/chainedDotsStartEnd.kt");
+        }
+
+        @TestMetadata("complex.kt")
+        public void testComplex() throws Exception {
+            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/complex.kt");
+        }
+
         @TestMetadata("dotExpressionAndCallExpression.kt")
         public void testDotExpressionAndCallExpression() throws Exception {
             runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/dotExpressionAndCallExpression.kt");
         }
 
-        @TestMetadata("multipleCalls.kt")
-        public void testMultipleCalls() throws Exception {
-            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCalls.kt");
+        @TestMetadata("multipleCallsEnd.kt")
+        public void testMultipleCallsEnd() throws Exception {
+            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsEnd.kt");
+        }
+
+        @TestMetadata("multipleCallsStart.kt")
+        public void testMultipleCallsStart() throws Exception {
+            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsStart.kt");
+        }
+
+        @TestMetadata("multipleCallsStartEnd.kt")
+        public void testMultipleCallsStartEnd() throws Exception {
+            runTest("idea/testData/quickfix/addSemicolonBeforeLambdaExpression/multipleCallsStartEnd.kt");
         }
 
         @TestMetadata("multipleLambdas.kt")