MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-09-27 10:03:11 +07:00
Code Issues Packages Projects Releases Wiki Activity

cleanup: rename method and clarify javadoc

Browse Source
This commit is contained in:
Konstantin Aleev
2018-07-09 19:36:58 +03:00
parent 9538284b49
commit d585ccb554
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
uast/uast-common/src/org/jetbrains/uast/qualifiedUtils.kt
+14 -8
View File
@@ -210,14 +210,20 @@ fun UElement.asRecursiveLogString(render: (UElement) -> String = { it.asLogStrin
* or companion object's containing class if the given method is main method annotated with [kotlin.jvm.JvmStatic] in companion object,
* otherwise *null*.
*/
fun findMainClass(mainMethod: UMethod): PsiClass? {
val psiMethod = mainMethod.javaPsi
if ("main" != psiMethod.name) return null
var mainClassCandidate = psiMethod.containingClass ?: return null
if (PsiMethodUtil.isMainMethod(psiMethod)) return mainClassCandidate
fun getMainMethodClass(uMainMethod: UMethod): PsiClass? {
if ("main" != uMainMethod.name) return null
val containingClass = uMainMethod.uastParent as? UClass ?: return null
val mainMethod = uMainMethod.javaPsi
if (PsiMethodUtil.isMainMethod(mainMethod)) return containingClass.javaPsi
// Check for @JvmStatic main method in companion object
mainClassCandidate = mainClassCandidate.containingClass ?: return null
val mainInClass = PsiMethodUtil.findMainInClass(mainClassCandidate)
return if (psiMethod.manager.areElementsEquivalent(psiMethod, mainInClass)) mainClassCandidate else null
val parentClassForCompanionObject = (containingClass.uastParent as? UClass)?.javaPsi ?: return null
val mainInClass = PsiMethodUtil.findMainInClass(parentClassForCompanionObject)
if (mainMethod.manager.areElementsEquivalent(mainMethod, mainInClass)) {
return parentClassForCompanionObject
}
return null
}