From b6f34ad8c09bd126ebf8adb5bc1d5fca74e0ad66 Mon Sep 17 00:00:00 2001
From: Dmitrii Gridin <dmitry.gridin@jetbrains.com>
Date: Wed, 23 Aug 2023 20:57:18 +0200
Subject: [PATCH] [AA] add API to get KtScriptSymbol for KtScript

^KT-61431
---
 .../jetbrains/kotlin/analysis/api/symbols/KtSymbolProvider.kt  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/symbols/KtSymbolProvider.kt b/analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/symbols/KtSymbolProvider.kt
index ab97b2f22d9..b452eb18a1d 100644
--- a/analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/symbols/KtSymbolProvider.kt
+++ b/analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/symbols/KtSymbolProvider.kt
@@ -135,6 +135,9 @@ public interface KtSymbolProviderMixIn : KtAnalysisSessionMixIn {
     public fun KtFile.getFileSymbol(): KtFileSymbol =
         withValidityAssertion { analysisSession.symbolProvider.getFileSymbol(this) }
 
+    public fun KtScript.getScriptSymbol(): KtScriptSymbol =
+        withValidityAssertion { analysisSession.symbolProvider.getScriptSymbol(this) }
+
     /**
      * Returns [KtPackageSymbol] corresponding to [packageFqName] if corresponding package is exists and visible from current uses-site scope,
      * `null` otherwise