commit 5590992bcfd3dc5654d99dfcb4576c79b0637148
author Yan Zhulanow <yan.zhulanow@jetbrains.com> Thu Jul 13 18:20:59 2023 +0900
committer Yan Zhulanow <yan.zhulanow@jetbrains.com> Thu Jul 13 18:29:17 2023 +0900
tree bcae9eec235ef92fcbc5b8a5cf7dad2f95ca378b
parent cc5004a7a1e105b8683d35b0bcd362f92d7825e7

[FIR] Create 'FirCodeFragment' with a lazy block if requested

compiler/fir/raw-fir/psi2fir/src/org/jetbrains/kotlin/fir/builder/PsiRawFirBuilder.kt

diff --git a/compiler/fir/raw-fir/psi2fir/src/org/jetbrains/kotlin/fir/builder/PsiRawFirBuilder.kt b/compiler/fir/raw-fir/psi2fir/src/org/jetbrains/kotlin/fir/builder/PsiRawFirBuilder.kt
index fa3231c..e34f302 100644
--- a/compiler/fir/raw-fir/psi2fir/src/org/jetbrains/kotlin/fir/builder/PsiRawFirBuilder.kt
+++ b/compiler/fir/raw-fir/psi2fir/src/org/jetbrains/kotlin/fir/builder/PsiRawFirBuilder.kt
@@ -1181,10 +1181,12 @@
                 moduleData = baseModuleData
                 origin = FirDeclarationOrigin.Source
                 symbol = FirCodeFragmentSymbol()
-                block = when (file) {
-                    is KtExpressionCodeFragment -> file.getContentElement()?.toFirBlock() ?: buildEmptyExpressionBlock()
-                    is KtBlockCodeFragment -> configureBlockWithoutBuilding(file.getContentElement()).build()
-                    else -> error("Unexpected code fragment type: " + file::class.java)
+                block = buildOrLazyBlock {
+                    when (file) {
+                        is KtExpressionCodeFragment -> file.getContentElement()?.toFirBlock() ?: buildEmptyExpressionBlock()
+                        is KtBlockCodeFragment -> configureBlockWithoutBuilding(file.getContentElement()).build()
+                        else -> error("Unexpected code fragment type: " + file::class.java)
+                    }
                 }
             }
         }