FIR: introduce & use SELECTOR_BY_QUALIFIED positioning strategy

2021-02-17 16:26:17 +03:00
parent 5892646a27
commit 5ca3ce9e37
6 changed files with 60 additions and 6 deletions
@@ -725,4 +725,15 @@ object PositioningStrategies {
            return super.mark(element)
        }
    }
+
+    val SELECTOR_BY_QUALIFIED: PositioningStrategy<PsiElement> = object : PositioningStrategy<PsiElement>() {
+        override fun mark(element: PsiElement): List<TextRange> {
+            if (element is KtQualifiedExpression) {
+                when (val selectorExpression = element.selectorExpression) {
+                    is KtCallExpression, is KtReferenceExpression -> return mark(selectorExpression)
+                }
+            }
+            return super.mark(element)
+        }
+    }
 }