From 9a9b16ed4147782eb0fbfe054bd6a8b2952a6284 Mon Sep 17 00:00:00 2001
From: Yan Zhulanow <yan.zhulanow@jetbrains.com>
Date: Tue, 1 Aug 2023 03:51:18 +0900
Subject: [PATCH] [LL API] Do not wrap body or lambdas in 'forFunctionBody()'

The function is only intended to be called for 'FirSimpleFunction's.
---
 .../low/level/api/fir/util/ContextCollector.kt   | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
index 8cc7b822a28..2848af30e74 100644
--- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
+++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
@@ -331,17 +331,15 @@ private class ContextCollectorVisitor(
         processSignatureAnnotations(anonymousFunction)
 
         context.withAnonymousFunction(anonymousFunction, holder, ResolutionMode.ContextIndependent) {
-            context.forFunctionBody(anonymousFunction, holder) {
-                for (parameter in anonymousFunction.valueParameters) {
-                    process(parameter)
-                    context.storeVariable(parameter, holder.session)
-                }
+            for (parameter in anonymousFunction.valueParameters) {
+                process(parameter)
+                context.storeVariable(parameter, holder.session)
+            }
 
-                dumpContext(anonymousFunction.psi, ContextKind.BODY)
+            dumpContext(anonymousFunction.psi, ContextKind.BODY)
 
-                onActive {
-                    process(anonymousFunction.body)
-                }
+            onActive {
+                process(anonymousFunction.body)
             }
 
             onActive {