From c92396069ab5429c88b149ea5527880fae9ce070 Mon Sep 17 00:00:00 2001
From: Marco Pennekamp <marco.pennekamp@jetbrains.com>
Date: Mon, 20 Nov 2023 18:25:30 +0100
Subject: [PATCH] [LL] Implement
 `LLFirResolveExtensionToolDeclarationProvider.computePackageNames`

^KT-62553
---
 .../fir/resolve/extensions/LLFirResolveExtensionTool.kt  | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/resolve/extensions/LLFirResolveExtensionTool.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/resolve/extensions/LLFirResolveExtensionTool.kt
index bfdb3642596..2a4c3439e06 100644
--- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/resolve/extensions/LLFirResolveExtensionTool.kt
+++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/resolve/extensions/LLFirResolveExtensionTool.kt
@@ -249,7 +249,14 @@ class LLFirResolveExtensionToolDeclarationProvider internal constructor(
             .mapNotNullTo(mutableListOf()) { it.kotlinFile.script }
     }
 
-    override fun computePackageNames(): Set<String>? = null
+    override fun computePackageNames(): Set<String>? =
+        buildSet {
+            extensionProvider.extensions.forEach { extension ->
+                extension.getContainedPackages().forEach { fqName ->
+                    add(fqName.asString())
+                }
+            }
+        }
 
     override val hasSpecificClassifierPackageNamesComputation: Boolean get() = false
     override val hasSpecificCallablePackageNamesComputation: Boolean get() = false