Uast: fix for Lambda in place call identifiers (KT-35432)

plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUFunctionCallExpression.kt

@@ -83,6 +83,8 @@ class KotlinUFunctionCallExpression(
                 KotlinUIdentifier(
                     calleeExpression.constructorReferenceExpression?.getReferencedNameElement() ?: calleeExpression, this
                 )
+            is KtLambdaExpression ->
+                KotlinUIdentifier(calleeExpression.functionLiteral.lBrace, this)
             else -> KotlinUIdentifier(calleeExpression, this)
         }
     }

plugins/uast-kotlin/testData/Lambdas.identifiers.txt

@@ -0,0 +1,19 @@
+java -> USimpleNameReferenceExpression (identifier = java)
+util -> USimpleNameReferenceExpression (identifier = util)
+stream -> USimpleNameReferenceExpression (identifier = stream)
+Stream -> USimpleNameReferenceExpression (identifier = Stream)
+foo -> UMethod (name = foo)
+    Stream -> USimpleNameReferenceExpression (identifier = Stream)
+    empty -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0))
+    String -> USimpleNameReferenceExpression (identifier = String)
+    filter -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 1))
+        it -> USimpleNameReferenceExpression (identifier = it)
+        isEmpty -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0))
+doSelectItem -> UMethod (name = doSelectItem)
+selectItemFunction -> UParameter (name = selectItemFunction)
+Unit -> USimpleNameReferenceExpression (identifier = Unit)
+    selectItemFunction -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0))
+    baz -> ULocalVariable (name = baz)
+        Local -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0))
+    baz -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0))
+lambdaInPlaceCall -> UMethod (name = lambdaInPlaceCall)

plugins/uast-kotlin/testData/Lambdas.kt

@@ -11,3 +11,9 @@ fun doSelectItem(selectItemFunction: () -> Unit) {
     }
     baz()
 }
+
+fun lambdaInPlaceCall() {
+    while ({ true }()) {
+
+    }
+}

plugins/uast-kotlin/testData/Lambdas.log.txt

@@ -37,3 +37,10 @@ UFile (package = )
                 UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0))
                     UIdentifier (Identifier (baz))
                     USimpleNameReferenceExpression (identifier = invoke, resolvesTo = null)
+        UMethod (name = lambdaInPlaceCall)
+            UBlockExpression
+                UWhileExpression
+                    UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0))
+                        UIdentifier (Identifier ({))
+                        USimpleNameReferenceExpression (identifier = invoke, resolvesTo = null)
+                    UBlockExpression

plugins/uast-kotlin/testData/Lambdas.refNames.txt

@@ -0,0 +1,20 @@
+Stream -> USimpleNameReferenceExpression (identifier = Stream) from KtDotQualifiedExpression
+stream -> USimpleNameReferenceExpression (identifier = stream) from KtDotQualifiedExpression
+util -> USimpleNameReferenceExpression (identifier = util) from KtDotQualifiedExpression
+java -> USimpleNameReferenceExpression (identifier = java) from KtNameReferenceExpression
+util -> USimpleNameReferenceExpression (identifier = util) from KtNameReferenceExpression
+stream -> USimpleNameReferenceExpression (identifier = stream) from KtNameReferenceExpression
+Stream -> USimpleNameReferenceExpression (identifier = Stream) from KtNameReferenceExpression
+filter -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 1)) from KtDotQualifiedExpression
+empty -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0)) from KtDotQualifiedExpression
+Stream -> USimpleNameReferenceExpression (identifier = Stream) from KtNameReferenceExpression
+empty -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0)) from KtNameReferenceExpression
+String -> USimpleNameReferenceExpression (identifier = String) from KtNameReferenceExpression
+filter -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 1)) from KtNameReferenceExpression
+isEmpty -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0)) from KtDotQualifiedExpression
+it -> USimpleNameReferenceExpression (identifier = it) from KtNameReferenceExpression
+isEmpty -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0)) from KtNameReferenceExpression
+Unit -> USimpleNameReferenceExpression (identifier = Unit) from KtNameReferenceExpression
+selectItemFunction -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0)) from KtNameReferenceExpression
+Local -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0)) from KtNameReferenceExpression
+baz -> UCallExpression (kind = UastCallKind(name='method_call'), argCount = 0)) from KtNameReferenceExpression

plugins/uast-kotlin/testData/Lambdas.render.txt

@@ -13,4 +13,8 @@ public final class LambdasKt {
         }
         invoke()
     }
+    public static final fun lambdaInPlaceCall() : void {
+        while (invoke()) {
+        }
+    }
 }

plugins/uast-kotlin/tests/KotlinUastIdentifiersTest.kt

@@ -24,6 +24,9 @@ class KotlinUastIdentifiersTest : AbstractKotlinIdentifiersTest() {
     @Test
     fun testAnonymous() = doTest("Anonymous")
 
+    @Test
+    fun testLambdas() = doTest("Lambdas")
+
     @Test
     fun testSuperCalls() = doTest("SuperCalls")