org.jetbrains.kotlin.utils.psiUtils.kt Maven / Gradle / Ivy
/*
* Copyright 2010-2022 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.utils
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiElementVisitor
import com.intellij.psi.PsiFile
import com.intellij.psi.impl.source.tree.LeafPsiElement
fun getElementTextWithContext(psiElement: PsiElement): String {
if (!psiElement.isValid) return ""
@Suppress("LocalVariableName") val ELEMENT_TAG = "ELEMENT"
val containingFile = psiElement.containingFile
val context = psiElement.parentOfType("KtImportDirective")
?: psiElement.parentOfType("KtPackageDirective")
?: psiElement.parentOfType("KtDeclarationWithBody", "KtClassOrObject", "KtScript")
?: psiElement.parentOfType("KtProperty")
?: containingFile
val elementTextInContext = buildString {
context.accept(object : PsiElementVisitor() {
override fun visitElement(element: PsiElement) {
if (element === psiElement) append("<$ELEMENT_TAG>")
if (element is LeafPsiElement) {
append(element.text)
} else {
element.acceptChildren(this)
}
if (element === psiElement) append("")
}
})
}.trimIndent().trim()
return buildString {
appendLine("")
append(elementTextInContext)
}
}
private fun PsiElement.parentOfType(vararg psiClassNames: String): PsiElement? {
fun acceptsClass(javaClass: Class<*>): Boolean {
if (javaClass.simpleName in psiClassNames) return true
javaClass.superclass?.let { if (acceptsClass(it)) return true }
for (superInterface in javaClass.interfaces) {
if (acceptsClass(superInterface)) return true
}
return false
}
return generateSequence(this) { it.parent }
.filter { it !is PsiFile }
.firstOrNull { acceptsClass(it::class.java) }
}