azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files
analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/KtAnalysisSession.kt
+4
View File
@@ -55,6 +55,7 @@ public abstract class KtAnalysisSession(final override val token: KtLifetimeToke
KtVisibilityCheckerMixIn,
KtMemberSymbolProviderMixin,
KtMultiplatformInfoProviderMixin,
KtOriginalPsiProviderMixIn,
KtInheritorsProviderMixIn,
KtTypeCreatorMixIn,
KtAnalysisScopeProviderMixIn,
@@ -141,6 +142,9 @@ public abstract class KtAnalysisSession(final override val token: KtLifetimeToke
internal val multiplatformInfoProvider: KtMultiplatformInfoProvider get() = multiplatformInfoProviderImpl
protected abstract val multiplatformInfoProviderImpl: KtMultiplatformInfoProvider
internal val originalPsiProvider: KtOriginalPsiProvider get() = originalPsiProviderImpl
protected abstract val originalPsiProviderImpl: KtOriginalPsiProvider
internal val symbolInfoProvider: KtSymbolInfoProvider get() = symbolInfoProviderImpl
protected abstract val symbolInfoProviderImpl: KtSymbolInfoProvider
analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/components/KtOriginalPsiProvider.kt
+47
View File
@@ -0,0 +1,47 @@
/*
* 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.components
import org.jetbrains.kotlin.analysis.api.lifetime.withValidityAssertion
import org.jetbrains.kotlin.psi.KtDeclaration
import org.jetbrains.kotlin.psi.KtFile
public abstract class KtOriginalPsiProvider : KtAnalysisSessionComponent() {
public abstract fun getOriginalDeclaration(declaration: KtDeclaration): KtDeclaration?
public abstract fun getOriginalKtFile(file: KtFile): KtFile?
public abstract fun recordOriginalDeclaration(fakeDeclaration: KtDeclaration, originalDeclaration: KtDeclaration)
public abstract fun recordOriginalKtFile(fakeFile: KtFile, originalFile: KtFile)
}
public interface KtOriginalPsiProviderMixIn : KtAnalysisSessionMixIn {
/**
* If [KtDeclaration] is a non-local declaration in a fake file analyzed in dependent session, returns the original declaration
* for [this]. Otherwise, returns `null`.
*/
public fun KtDeclaration.getOriginalDeclaration(): KtDeclaration? =
withValidityAssertion { analysisSession.originalPsiProvider.getOriginalDeclaration(this) }
/**
* If [this] is a fake file analyzed in dependent session, returns the original file for [this]. Otherwise, returns `null`.
*/
public fun KtFile.getOriginalKtFile(): KtFile? =
withValidityAssertion { analysisSession.originalPsiProvider.getOriginalKtFile(this) }
/**
* Records [declaration] as an original declaration for [this].
*/
public fun KtDeclaration.recordOriginalDeclaration(declaration: KtDeclaration) {
withValidityAssertion { analysisSession.originalPsiProvider.recordOriginalDeclaration(this, declaration) }
}
/**
* Records [file] as an original file for [this].
*/
public fun KtFile.recordOriginalKtFile(file: KtFile) {
withValidityAssertion { analysisSession.originalPsiProvider.recordOriginalKtFile(this, file) }
}
}