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

[lc] KotlinAsJavaSupportBase: add more lightweight checks for facades

Browse Source 
^KT-50241
This commit is contained in:
Dmitry Gridin
2022-09-06 14:19:16 +02:00
committed by Space
parent ec1b6f4f3e
commit b618855e2c
1 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/light-classes-base/src/org/jetbrains/kotlin/asJava/KotlinAsJavaSupportBase.kt
+22 -5
View File
@@ -23,12 +23,11 @@ import org.jetbrains.kotlin.psi.*
abstract class KotlinAsJavaSupportBase<TModule>(protected val project: Project) : KotlinAsJavaSupport() {
fun createLightFacade(file: KtFile): Pair<KtLightClassForFacade?, ModificationTracker>? {
if (file.isScript()) return null
if (file.isCompiled && !file.name.endsWith(".class")) return null
if (!file.facadeIsPossible()) return null
val module = file.findModule().takeIf { facadeIsApplicable(it, file) } ?: return null
val facadeFqName = file.javaFileFacadeFqName
val facadeFiles = if (file.isJvmMultifileClassFile && !file.isCompiled) {
val facadeFiles = if (file.canHaveAdditionalFilesInFacade()) {
findFilesForFacade(facadeFqName, module.contentSearchScope).filter(KtFile::isJvmMultifileClassFile)
} else {
listOf(file)
@@ -48,6 +47,18 @@ abstract class KotlinAsJavaSupportBase<TModule>(protected val project: Project)
}
}
/**
* lightweight applicability check
*/
private fun KtFile.facadeIsPossible(): Boolean = when {
isCompiled && !name.endsWith(".class") -> false
isScript() -> false
canHaveAdditionalFilesInFacade() -> true
else -> hasTopLevelCallables()
}
private fun KtFile.canHaveAdditionalFilesInFacade(): Boolean = !isCompiled && isJvmMultifileClassFile
protected abstract fun KtFile.findModule(): TModule
protected abstract fun facadeIsApplicable(module: TModule, file: KtFile): Boolean
protected abstract val TModule.contentSearchScope: GlobalSearchScope
@@ -88,11 +99,17 @@ abstract class KotlinAsJavaSupportBase<TModule>(protected val project: Project)
override fun getFacadeNames(packageFqName: FqName, scope: GlobalSearchScope): Collection<String> {
return findFilesForFacadeByPackage(packageFqName, scope).mapNotNullTo(mutableSetOf()) { file ->
file.takeIf { facadeIsApplicable(it.findModule(), file) }?.javaFileFacadeFqName?.shortName()?.asString()
file.takeIf { it.facadeIsPossible() }
?.takeIf { facadeIsApplicable(it.findModule(), file) }
?.javaFileFacadeFqName
?.shortName()
?.asString()
}.toSet()
}
private fun Collection<KtFile>.toFacadeClasses(): List<KtLightClassForFacade> = groupBy {
private fun Collection<KtFile>.toFacadeClasses(): List<KtLightClassForFacade> = filter {
it.facadeIsPossible()
}.groupBy {
FacadeKey(it.javaFileFacadeFqName, it.isJvmMultifileClassFile, it.findModule())
}.mapNotNull { (key, files) ->
files.firstOrNull { facadeIsApplicable(key.module, it) }?.let(::getLightFacade)