[FIR] Introduce & use SourceElementPositioningStrategies.OPERATOR

This commit is contained in:
Mikhail Glukhikh
2020-11-24 16:09:50 +03:00
parent c9806c5af9
commit ea7d738ee1
6 changed files with 28 additions and 9 deletions
@@ -687,4 +687,14 @@ object PositioningStrategies {
return DEFAULT.mark(element)
}
}
val OPERATOR: PositioningStrategy<KtExpression> = object : PositioningStrategy<KtExpression>() {
override fun mark(element: KtExpression): List<TextRange> {
return when (element) {
is KtBinaryExpression -> mark(element.operationReference)
is KtUnaryExpression -> mark(element.operationReference)
else -> super.mark(element)
}
}
}
}