From 64bdcc11311ac42b0df3fd987789174da2a0f2da Mon Sep 17 00:00:00 2001
From: Dmitrii Gridin <dmitry.gridin@jetbrains.com>
Date: Sat, 29 Jul 2023 01:06:11 +0200
Subject: [PATCH] [LL FIR] support scripts in COMPILER_REQUIRED_ANNOTATIONS
 transformer

^KT-60728
---
 .../fir/transformers/LLFirCompilerAnnotationsLazyResolver.kt  | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/transformers/LLFirCompilerAnnotationsLazyResolver.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/transformers/LLFirCompilerAnnotationsLazyResolver.kt
index 3d76905f4fe..51f9cdeb134 100644
--- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/transformers/LLFirCompilerAnnotationsLazyResolver.kt
+++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/transformers/LLFirCompilerAnnotationsLazyResolver.kt
@@ -105,8 +105,7 @@ private class LLFirCompilerRequiredAnnotationsTargetResolver(
 
     override fun doResolveWithoutLock(target: FirElementWithResolveState): Boolean {
         when {
-            target is FirScript -> return false
-            target is FirRegularClass || target.isRegularDeclarationWithAnnotation -> {}
+            target is FirRegularClass || target is FirScript || target.isRegularDeclarationWithAnnotation -> {}
             else -> throwUnexpectedFirElementError(target)
         }
 
@@ -297,7 +296,6 @@ private class LLFirCompilerRequiredAnnotationsTargetResolver(
     }
 
     override fun doLazyResolveUnderLock(target: FirElementWithResolveState) {
-        if (target is FirScript) return
         throwUnexpectedFirElementError(target)
     }
 }