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

Uast: add "strict" parameter to getParentOfType()

Browse Source
This commit is contained in:
Yan Zhulanow
2016-04-07 19:45:51 +03:00
parent c13c24becc
commit 29412693db
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/uast-common/src/org/jetbrains/uast/UastUtils.kt
+7 -7
View File
@@ -162,18 +162,18 @@ fun UCallExpression.getQualifiedCallElement(): UExpression {
inline fun <reified T: UElement> UElement.getParentOfType(): T? = getParentOfType(T::class.java) inline fun <reified T: UElement> UElement.getParentOfType(): T? = getParentOfType(T::class.java)
fun <T: UElement> UElement.getParentOfType(clazz: Class<T>): T? { @JvmOverloads
tailrec fun findParent(element: UElement): UElement? { fun <T: UElement> UElement.getParentOfType(clazz: Class<T>, strict: Boolean = true): T? {
val parent = element.parent tailrec fun findParent(element: UElement?): UElement? {
return when { return when {
parent == null -> null element == null -> null
clazz.isInstance(parent) -> parent clazz.isInstance(element) -> element
else -> findParent(parent) else -> findParent(element.parent)
} }
} }
@Suppress("UNCHECKED_CAST") @Suppress("UNCHECKED_CAST")
return findParent(this) as T return findParent(if (!strict) this else parent) as? T
} }
fun <T> UClass.findStaticMemberOfType(name: String, type: Class<out T>): T? { fun <T> UClass.findStaticMemberOfType(name: String, type: Class<out T>): T? {