org.jetbrains.kotlin.cli.jvm.compiler.findMainClass.kt Maven / Gradle / Ivy
/*
* Copyright 2010-2019 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.cli.jvm.compiler
import org.jetbrains.kotlin.config.LanguageVersionSettings
import org.jetbrains.kotlin.fileClasses.JvmFileClassUtil
import org.jetbrains.kotlin.idea.MainFunctionDetector
import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.psi.*
import org.jetbrains.kotlin.psi.psiUtil.getParentOfType
import org.jetbrains.kotlin.resolve.BindingContext
fun findMainClass(bindingContext: BindingContext, languageVersionSettings: LanguageVersionSettings, files: List): FqName? {
val mainFunctionDetector = MainFunctionDetector(bindingContext, languageVersionSettings)
return files.asSequence()
.map { file ->
mainFunctionDetector.findMainFunction(file)?.let { mainFunction ->
if (mainFunction.isTopLevel) {
JvmFileClassUtil.getFileClassInfoNoResolve(file).facadeClassFqName
} else {
val parent = mainFunction.getParentOfType(strict = true)
if (parent is KtObjectDeclaration && parent.isCompanion()) {
mainFunction.fqName?.parent()?.parent()
} else {
mainFunction.fqName?.parent()
}
}
}
}
.singleOrNull { it != null }
}
private fun MainFunctionDetector.findMainFunction(container: KtDeclarationContainer): KtNamedFunction? =
container.declarations.mapNotNull { declaration ->
when (declaration) {
is KtNamedFunction -> declaration.takeIf(::isMain)
is KtDeclarationContainer -> findMainFunction(declaration)
else -> null
}
}.singleOrNull()