1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
// Copyright 2000-2021 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package org.jetbrains.uast.kotlin
import com.intellij.psi.PsiType
import org.jetbrains.kotlin.psi.KtBlockExpression
import org.jetbrains.kotlin.psi.KtLambdaExpression
import org.jetbrains.uast.*
import org.jetbrains.uast.kotlin.psi.UastKotlinPsiParameter
class KotlinULambdaExpression(
override val sourcePsi: KtLambdaExpression,
givenParent: UElement?
) : KotlinAbstractUExpression(givenParent), ULambdaExpression, KotlinUElementWithType {
override val functionalInterfaceType: PsiType? by lz {
baseResolveProviderService.getFunctionalInterfaceType(this)
}
override val body by lz {
sourcePsi.bodyExpression?.let { Body(it, this) } ?: UastEmptyExpression(this)
}
class Body(bodyExpression: KtBlockExpression, parent: KotlinULambdaExpression) : KotlinUBlockExpression(bodyExpression, parent) {
override val expressions: List<UExpression> by lz {
val statements = sourcePsi.statements
if (statements.isEmpty()) return@lz emptyList<UExpression>()
ArrayList<UExpression>(statements.size).also { result ->
statements.subList(0, statements.size - 1).mapTo(result) {
baseResolveProviderService.baseKotlinConverter.convertOrEmpty(it, this)
}
result.add(implicitReturn ?: baseResolveProviderService.baseKotlinConverter.convertOrEmpty(statements.last(), this))
}
}
val implicitReturn: KotlinUImplicitReturnExpression? by lz {
baseResolveProviderService.getImplicitReturn(parent.sourcePsi, this)
}
}
override val valueParameters by lz {
getParameters(includeExplicitParameters = false)
}
override val parameters: List<UParameter> by lz {
getParameters(includeExplicitParameters = true)
}
private fun getParameters(includeExplicitParameters: Boolean): List<UParameter> {
val explicitParameters = sourcePsi.valueParameters.mapIndexed { i, p ->
KotlinUParameter(UastKotlinPsiParameter.create(p, sourcePsi, this, i), p, this)
}
if (explicitParameters.isNotEmpty()) return explicitParameters
return baseResolveProviderService.getImplicitParameters(sourcePsi, this, includeExplicitParameters)
}
override fun asRenderString(): String {
val renderedValueParameters = if (valueParameters.isEmpty())
""
else
valueParameters.joinToString { it.asRenderString() } + " ->\n"
val expressions =
(body as? UBlockExpression)?.expressions?.joinToString("\n") { it.asRenderString().withMargin } ?: body.asRenderString()
return "{ $renderedValueParameters\n$expressions\n}"
}
}
|
