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

Uast-pattens: registering Uast-patterns only on PsiLanguageInjectionHosts

Browse Source 
as long as now we can rely on `UInjectionHost` existence for both Java and Kotlin

GitOrigin-RevId: 121e36a242e4155cf8ee46f970ed67e0f3dd6d89
This commit is contained in:
Nicolay Mitropolsky
2019-04-28 18:21:31 +03:00
committed by intellij-monorepo-bot
parent 08274007f1
commit 4e2ac7f238
1 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
uast/uast-common/src/com/intellij/psi/UastReferenceRegistrar.kt
+19 -2
View File
@@ -23,15 +23,17 @@ import com.intellij.openapi.util.RecursionManager
import com.intellij.patterns.ElementPattern
import com.intellij.patterns.ElementPatternCondition
import com.intellij.patterns.InitialPatternCondition
import com.intellij.patterns.StandardPatterns
import com.intellij.util.ProcessingContext
import org.jetbrains.uast.UElement
import org.jetbrains.uast.UExpression
import org.jetbrains.uast.expressions.UInjectionHost
import org.jetbrains.uast.toUElement
fun PsiReferenceRegistrar.registerUastReferenceProvider(pattern: (UElement, ProcessingContext) -> Boolean,
provider: UastReferenceProvider,
priority: Double = PsiReferenceRegistrar.DEFAULT_PRIORITY) {
this.registerReferenceProvider(UastPatternAdapter(pattern, provider.supportedUElementTypes),
this.registerReferenceProvider(adaptPattern(pattern, provider.supportedUElementTypes),
UastReferenceProviderAdapter(provider),
priority)
}
@@ -39,7 +41,7 @@ fun PsiReferenceRegistrar.registerUastReferenceProvider(pattern: (UElement, Proc
fun PsiReferenceRegistrar.registerUastReferenceProvider(pattern: ElementPattern<out UElement>,
provider: UastReferenceProvider,
priority: Double = PsiReferenceRegistrar.DEFAULT_PRIORITY) {
this.registerReferenceProvider(UastPatternAdapter(pattern::accepts, provider.supportedUElementTypes),
this.registerReferenceProvider(adaptPattern(pattern::accepts, provider.supportedUElementTypes),
UastReferenceProviderAdapter(provider), priority)
}
@@ -80,6 +82,21 @@ private fun getOrCreateCachedElement(element: PsiElement,
element.toUElement(it)
}.firstOrNull()?.also { context?.put(cachedUElement, it) }
private fun adaptPattern(
predicate: (UElement, ProcessingContext) -> Boolean,
supportedUElementTypes: List<Class<out UElement>>
): ElementPattern<out PsiElement> {
val uastPatternAdapter = UastPatternAdapter(predicate, supportedUElementTypes)
// optimisation until IDEA-211738 is implemented
if (supportedUElementTypes == listOf(UInjectionHost::class.java)) {
return StandardPatterns.instanceOf(PsiLanguageInjectionHost::class.java).and(uastPatternAdapter)
}
return uastPatternAdapter
}
private class UastPatternAdapter(
val predicate: (UElement, ProcessingContext) -> Boolean,
val supportedUElementTypes: List<Class<out UElement>>