[Analysis API] Allow obtaining original for fake declaration/file

It is required to remove dependency on low-level API from completion
This commit is contained in:
aleksandrina-streltsova
2023-03-09 16:30:50 +02:00
committed by teamcity
parent 6d3b911dfc
commit 3c8531aad8
6 changed files with 109 additions and 0 deletions
@@ -108,6 +108,8 @@ private constructor(
override val multiplatformInfoProviderImpl: KtMultiplatformInfoProvider = KtFirMultiplatformInfoProvider(this, token)
override val originalPsiProviderImpl: KtOriginalPsiProvider = KtFirOriginalPsiProvider(this, token)
override val symbolInfoProviderImpl: KtSymbolInfoProvider = KtFirSymbolInfoProvider(this, token)
override val typesCreatorImpl: KtTypeCreator = KtFirTypeCreator(this, token)
@@ -0,0 +1,31 @@
/*
* Copyright 2010-2023 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.components.KtOriginalPsiProvider
import org.jetbrains.kotlin.analysis.api.fir.KtFirAnalysisSession
import org.jetbrains.kotlin.analysis.api.lifetime.KtLifetimeToken
import org.jetbrains.kotlin.analysis.low.level.api.fir.util.originalDeclaration
import org.jetbrains.kotlin.analysis.low.level.api.fir.util.originalKtFile
import org.jetbrains.kotlin.psi.KtDeclaration
import org.jetbrains.kotlin.psi.KtFile
internal class KtFirOriginalPsiProvider(
override val analysisSession: KtFirAnalysisSession,
override val token: KtLifetimeToken,
) : KtOriginalPsiProvider(), KtFirAnalysisSessionComponent {
override fun getOriginalDeclaration(declaration: KtDeclaration): KtDeclaration? = declaration.originalDeclaration
override fun getOriginalKtFile(file: KtFile): KtFile? = file.originalKtFile
override fun recordOriginalDeclaration(fakeDeclaration: KtDeclaration, originalDeclaration: KtDeclaration) {
fakeDeclaration.originalDeclaration = originalDeclaration
}
override fun recordOriginalKtFile(fakeFile: KtFile, originalFile: KtFile) {
fakeFile.originalKtFile = originalFile
}
}