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

[Analysis API] Add API for expect/actual declaration detection

Browse Source
This commit is contained in:
aleksandrina-streltsova
2023-05-02 12:05:58 +03:00
committed by teamcity
parent da5f33d762
commit 338c2433ac
3 changed files with 28 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/components/KtFirMultiplatformInfoProvider.kt
+17 -7
View File
@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.analysis.api.lifetime.KtLifetimeToken
import org.jetbrains.kotlin.analysis.api.symbols.KtDeclarationSymbol
import org.jetbrains.kotlin.analysis.low.level.api.fir.util.withFirSymbolEntry
import org.jetbrains.kotlin.analysis.utils.errors.checkWithAttachmentBuilder
import org.jetbrains.kotlin.fir.declarations.FirDeclarationStatus
import org.jetbrains.kotlin.fir.declarations.expectForActual
import org.jetbrains.kotlin.fir.symbols.impl.FirCallableSymbol
import org.jetbrains.kotlin.fir.symbols.impl.FirClassSymbol
@@ -25,13 +26,7 @@ internal class KtFirMultiplatformInfoProvider(
override fun getExpectForActual(actual: KtDeclarationSymbol): KtDeclarationSymbol? {
require(actual is KtFirSymbol<*>)
val firSymbol = actual.firSymbol
val status = when (firSymbol) {
is FirCallableSymbol -> firSymbol.rawStatus
is FirClassSymbol -> firSymbol.rawStatus
is FirTypeAliasSymbol -> firSymbol.rawStatus
else -> null
}
if (status?.isActual != true) return null
if (!isActual(actual)) return null
val expectsForActual = firSymbol.expectForActual?.get(ExpectActualCompatibility.Compatible) ?: return null
checkWithAttachmentBuilder(expectsForActual.size <= 1, message = { "expected as maximum one `expect` for the actual" }) {
@@ -43,4 +38,19 @@ internal class KtFirMultiplatformInfoProvider(
}
return expectsForActual.singleOrNull()?.let { analysisSession.firSymbolBuilder.buildSymbol(it) as? KtDeclarationSymbol }
}
override fun isActual(symbol: KtDeclarationSymbol): Boolean = getStatus(symbol)?.isActual == true
override fun isExpect(symbol: KtDeclarationSymbol): Boolean = getStatus(symbol)?.isExpect == true
private fun getStatus(symbol: KtDeclarationSymbol): FirDeclarationStatus? {
require(symbol is KtFirSymbol<*>)
val firSymbol = symbol.firSymbol
return when (firSymbol) {
is FirCallableSymbol -> firSymbol.rawStatus
is FirClassSymbol -> firSymbol.rawStatus
is FirTypeAliasSymbol -> firSymbol.rawStatus
else -> null
}
}
}