[FIR IDE] Refactoring, accept only KtElement sources in KtFe10AnalysisSession

This commit is contained in:
Yan Zhulanow
2021-10-27 02:58:01 +09:00
parent 0382323dad
commit 0388f1e5ca
2 changed files with 2 additions and 3 deletions
@@ -6,7 +6,6 @@
package org.jetbrains.kotlin.analysis.api.descriptors
import com.intellij.openapi.components.ServiceManager
import com.intellij.psi.PsiElement
import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
import org.jetbrains.kotlin.analysis.api.components.*
import org.jetbrains.kotlin.analysis.api.descriptors.components.*
@@ -17,7 +16,7 @@ import org.jetbrains.kotlin.psi.KtFile
@Suppress("LeakingThis")
class KtFe10AnalysisSession(
private val contextElement: PsiElement, token: ValidityToken
private val contextElement: KtElement, token: ValidityToken
) : KtAnalysisSession(token), Fe10AnalysisFacade by ServiceManager.getService(contextElement.project, Fe10AnalysisFacade::class.java) {
override val smartCastProviderImpl: KtSmartCastProvider = KtFe10SmartCastProvider(this)
override val diagnosticProviderImpl: KtDiagnosticProvider = KtFe10DiagnosticProvider(this)
@@ -26,7 +26,7 @@ class KtFe10AnalysisSessionProvider : KtAnalysisSessionProvider() {
return contextSymbol.analysisSession
} else {
val contextElement = contextSymbol.psi
if (contextElement != null) {
if (contextElement is KtElement) {
return KtFe10AnalysisSession(contextElement, contextSymbol.token)
}
}