K2 Scripting: add annotation resolving for scripts

#KT-62400 fixed
2023-10-06 14:24:49 +02:00
parent 31f9e9e7a8
commit 02dc3c4ddb
2 changed files with 26 additions and 0 deletions
@@ -394,6 +394,21 @@ abstract class AbstractFirSpecificAnnotationResolveTransformer(
        }
    }

+    override fun transformScript(
+        script: FirScript,
+        data: Nothing?,
+    ): FirScript {
+        if (shouldTransformDeclaration(script)) {
+            computationSession.recordThatAnnotationsAreResolved(script)
+            transformDeclaration(script, null).also {
+                transformChildren(script) {
+                    script.transformStatements(this, data)
+                }
+            }
+        }
+        return script
+    }
+
    inline fun withRegularClass(
        regularClass: FirRegularClass,
        action: () -> Unit