[VISUALIZER] Add handler for dot qualified expression in fir

For some reason source element of such expression in fir
was changed, so to get FirFunctionCall we should seek it
in KtDotQualifiedExpression
This commit is contained in:
Ivan Kylchik
2021-02-25 13:49:06 +03:00
committed by TeamCityServer
parent 0160174c7a
commit 65fcd72ce1
@@ -249,6 +249,11 @@ class FirVisualizer(private val firFile: FirFile) : BaseRenderer() {
super.visitWhenExpression(expression)
}
override fun visitDotQualifiedExpression(expression: KtDotQualifiedExpression) {
expression.firstOfTypeWithLocalReplace<FirFunctionCall>(expression.selectorExpression) { this.calleeReference.name.asString() }
super.visitDotQualifiedExpression(expression)
}
override fun visitCallExpression(expression: KtCallExpression) {
expression.firstOfTypeWithLocalReplace<FirFunctionCall> { this.calleeReference.name.asString() }
expression.children.filter { it.node.elementType != KtNodeTypes.REFERENCE_EXPRESSION }.forEach { psi ->