[FIR][LightTree] Keep source in FOR loop elements

This commit is contained in:
Andrey Zinovyev
2021-04-16 11:01:26 +03:00
committed by TeamCityServer
parent 27766c2575
commit 70f9b77cce
@@ -58,7 +58,7 @@ class ExpressionsConverter(
return expression?.let { return expression?.let {
convertExpression(it, errorReason) convertExpression(it, errorReason)
} as? R ?: buildErrorExpression( } as? R ?: buildErrorExpression(
null, ConeSimpleDiagnostic(errorReason, DiagnosticKind.ExpressionExpected) expression?.toFirSourceElement(), ConeSimpleDiagnostic(errorReason, DiagnosticKind.ExpressionExpected)
) as R ) as R
} }
@@ -962,7 +962,7 @@ class ExpressionsConverter(
) )
statements += iteratorVal statements += iteratorVal
statements += FirWhileLoopBuilder().apply { statements += FirWhileLoopBuilder().apply {
source = forLoop.toFirSourceElement() source = fakeSource
condition = buildFunctionCall { condition = buildFunctionCall {
source = fakeSource source = fakeSource
calleeReference = buildSimpleNamedReference { calleeReference = buildSimpleNamedReference {