FIR IDE: add helper to resolve call for any KtElement

2021-10-28 15:24:38 -07:00
parent 523cb4b3e8
commit 1a22124c02
2 changed files with 11 additions and 6 deletions
@@ -36,4 +36,12 @@ public interface KtCallResolverMixIn : KtAnalysisSessionMixIn {

    public fun KtArrayAccessExpression.resolveCall(): KtCall? =
        analysisSession.callResolver.resolveCall(this)
-}
+
+    public fun KtElement.resolveCallIfPossible(): KtCall? = when (this) {
+        is KtCallElement -> resolveCall()
+        is KtBinaryExpression -> resolveCall()
+        is KtUnaryExpression -> resolveCall()
+        is KtArrayAccessExpression -> resolveCall()
+        else -> null
+    }
+}