[Analysis Api] Expose klibSourceFile via KtKlibSourceFileProviderMixIn

This `klibSourceFile` information is deserialized from klibs
to retain the information of the original SourceFile location
of a declaration.

^KT-66271 Fixed

analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/KtFirAnalysisSession.kt

@@ -128,6 +128,8 @@ private constructor(
 
     override val dataFlowInfoProviderImpl: KtDataFlowInfoProvider = KtFirDataFlowInfoProvider(this)
 
+    override val klibSourceFileProviderImpl: KtKlibSourceFileNameProvider = KtFirKlibSourceFileNameProvider(this)
+
     internal val useSiteSession: FirSession get() = firResolveSession.useSiteFirSession
     internal val firSymbolProvider: FirSymbolProvider get() = useSiteSession.symbolProvider
     internal val targetPlatform: TargetPlatform get() = useSiteSession.moduleData.platform

analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/components/KtFirKlibSourceFileNameProvider.kt

@@ -0,0 +1,24 @@
+/*
+ * Copyright 2010-2024 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.analysis.api.fir.components
+
+import org.jetbrains.kotlin.analysis.api.KtAnalysisNonPublicApi
+import org.jetbrains.kotlin.analysis.api.components.KtKlibSourceFileNameProvider
+import org.jetbrains.kotlin.analysis.api.fir.KtFirAnalysisSession
+import org.jetbrains.kotlin.analysis.api.fir.symbols.KtFirSymbol
+import org.jetbrains.kotlin.analysis.api.symbols.KtDeclarationSymbol
+import org.jetbrains.kotlin.fir.declarations.utils.klibSourceFile
+
+@OptIn(KtAnalysisNonPublicApi::class)
+internal class KtFirKlibSourceFileNameProvider(
+    override val analysisSession: KtFirAnalysisSession,
+) : KtKlibSourceFileNameProvider() {
+    override fun getKlibSourceFileName(declaration: KtDeclarationSymbol): String? {
+        require(declaration is KtFirSymbol<*>)
+        val sourceFile = declaration.firSymbol.klibSourceFile ?: return null
+        return sourceFile.name
+    }
+}