azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[FIR] Add util function for extracting return arguments from block

Browse Source
This commit is contained in:
Dmitriy Novozhilov
2019-08-02 12:30:38 +03:00
parent fe3824d796
commit 741ff958e8
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/inference/FirCallCompleter.kt
+1 -1
View File
@@ -122,7 +122,7 @@ class FirCallCompleter(
replacements[lambdaArgument] =
newLambdaExpression.transformSingle(transformer, FirBodyResolveTransformer.LambdaResolution(expectedReturnTypeRef))
return listOfNotNull(newLambdaExpression.body?.statements?.lastOrNull() as? FirExpression) to InferenceSession.default
return (newLambdaExpression.body?.returnExpressions() ?: emptyList()) to InferenceSession.default
}
}
}
compiler/fir/tree/src/org/jetbrains/kotlin/fir/Utils.kt
+6 -1
View File
@@ -6,6 +6,8 @@
package org.jetbrains.kotlin.fir
import org.jetbrains.kotlin.analyzer.ModuleInfo
import org.jetbrains.kotlin.fir.expressions.FirBlock
import org.jetbrains.kotlin.fir.expressions.FirExpression
import org.jetbrains.kotlin.fir.visitors.FirTransformer
fun <T : FirElement, D> MutableList<T>.transformInplace(transformer: FirTransformer<D>, data: D) {
@@ -34,4 +36,7 @@ fun <T : FirElement, D> T.transformSingle(transformer: FirTransformer<D>, data:
return (this as FirPureAbstractElement).transform<T, D>(transformer, data).single
}
fun ModuleInfo.dependenciesWithoutSelf(): Sequence<ModuleInfo> = dependencies().asSequence().filter { it != this }
fun ModuleInfo.dependenciesWithoutSelf(): Sequence<ModuleInfo> = dependencies().asSequence().filter { it != this }
// TODO: rewrite
fun FirBlock.returnExpressions(): List<FirExpression> = listOfNotNull(statements.lastOrNull() as? FirExpression)