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

Convert to Kotlin: KotlinTypeHierarchyProvider (prettify)

Browse Source
This commit is contained in:
Alexey Sedunov
2016-08-18 20:38:13 +03:00
parent 23d9f923ac
commit 05abc0127b
1 changed files with 37 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/kotlin/idea/hierarchy/KotlinTypeHierarchyProvider.kt
+37 -46
View File
@@ -16,78 +16,69 @@
package org.jetbrains.kotlin.idea.hierarchy
import com.intellij.codeInsight.TargetElementUtilBase
import com.intellij.codeInsight.TargetElementUtil
import com.intellij.ide.hierarchy.type.JavaTypeHierarchyProvider
import com.intellij.openapi.actionSystem.DataContext
import com.intellij.openapi.actionSystem.LangDataKeys
import com.intellij.openapi.actionSystem.PlatformDataKeys
import com.intellij.openapi.editor.Editor
import com.intellij.openapi.project.Project
import com.intellij.psi.PsiClass
import com.intellij.psi.PsiDocumentManager
import com.intellij.psi.PsiElement
import com.intellij.psi.search.GlobalSearchScope
import com.intellij.psi.util.PsiTreeUtil
import org.jetbrains.kotlin.descriptors.SimpleFunctionDescriptor
import com.intellij.psi.PsiFile
import org.jetbrains.kotlin.idea.caches.resolve.analyze
import org.jetbrains.kotlin.idea.decompiler.navigation.SourceNavigationHelper
import org.jetbrains.kotlin.idea.search.allScope
import org.jetbrains.kotlin.idea.stubindex.KotlinClassShortNameIndex
import org.jetbrains.kotlin.idea.util.ProjectRootsUtil
import org.jetbrains.kotlin.psi.KtClassOrObject
import org.jetbrains.kotlin.psi.KtNamedFunction
import org.jetbrains.kotlin.psi.psiUtil.getNonStrictParentOfType
import org.jetbrains.kotlin.renderer.DescriptorRenderer
import org.jetbrains.kotlin.resolve.BindingContext
class KotlinTypeHierarchyProvider : JavaTypeHierarchyProvider() {
private fun getTargetByReference(project: Project, editor: Editor): PsiElement? {
val target = TargetElementUtil.findTargetElement(editor, TargetElementUtil.getInstance().allAccepted)
return when (target) {
is PsiClass -> target
is KtClassOrObject -> SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass(target)
is KtNamedFunction -> { // Factory methods
val functionName = target.name
val functionDescriptor = target.analyze()[BindingContext.FUNCTION, target] ?: return null
val type = functionDescriptor.returnType ?: return null
val returnTypeText = DescriptorRenderer.FQ_NAMES_IN_TYPES.renderType(type)
if (returnTypeText != functionName) return null
val classOrObject = KotlinClassShortNameIndex.getInstance()[functionName, project, project.allScope()].singleOrNull()
?: return null
SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass(classOrObject)
}
else -> null
}
}
private fun getTargetByContainingElement(editor: Editor, file: PsiFile): PsiElement? {
val offset = editor.caretModel.offset
val element = file.findElementAt(offset) ?: return null
val classOrObject = element.getNonStrictParentOfType<KtClassOrObject>() ?: return null
return SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass(classOrObject)
}
override fun getTarget(dataContext: DataContext): PsiElement? {
val project = PlatformDataKeys.PROJECT.getData(dataContext) ?: return null
val editor = PlatformDataKeys.EDITOR.getData(dataContext)
if (editor != null) {
val file = PsiDocumentManager.getInstance(project).getPsiFile(editor.document) ?: return null
if (!ProjectRootsUtil.isInProjectOrLibSource(file)) return null
val target = TargetElementUtilBase.findTargetElement(editor, TargetElementUtilBase.getInstance().allAccepted)
if (target is PsiClass) {
return target
}
if (target is KtClassOrObject) {
return SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass((target as KtClassOrObject?)!!)
}
else if (target is KtNamedFunction) {
val functionName = target.name
val functionDescriptor = target.analyze().get<PsiElement, SimpleFunctionDescriptor>(BindingContext.FUNCTION, target)
if (functionDescriptor != null) {
val type = functionDescriptor!!.getReturnType()
if (type != null) {
val returnTypeText = DescriptorRenderer.FQ_NAMES_IN_TYPES.renderType(type!!)
if (returnTypeText == functionName) {
val classOrObjects = KotlinClassShortNameIndex.getInstance().get(functionName, project, GlobalSearchScope.allScope(project))
if (classOrObjects.size == 1) {
val classOrObject = classOrObjects.iterator().next()
return SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass(classOrObject)
}
}
}
}
}// Factory methods
val offset = editor.caretModel.offset
val element = file.findElementAt(offset) ?: return null
val classOrObject = PsiTreeUtil.getParentOfType(element, KtClassOrObject::class.java)
if (classOrObject != null) {
return SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass(classOrObject)
}
}
else {
val element = LangDataKeys.PSI_ELEMENT.getData(dataContext)
if (element is KtClassOrObject) {
return SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass((element as KtClassOrObject?)!!)
}
return getTargetByReference(project, editor) ?: getTargetByContainingElement(editor, file)
}
val element = LangDataKeys.PSI_ELEMENT.getData(dataContext)
if (element is KtClassOrObject) return SourceNavigationHelper.getOriginalPsiClassOrCreateLightClass((element as KtClassOrObject?)!!)
return null
}
}