From 78ecb05be0c3b7b585606180ddcea72acdfa053c Mon Sep 17 00:00:00 2001
From: Daniil Ovchinnikov <daniil.ovchinnikov@jetbrains.com>
Date: Fri, 27 Dec 2019 15:29:08 +0300
Subject: [PATCH] [groovy] add GrMethodCall#getCallReference

GitOrigin-RevId: 3dc002cfc8af3f12df8f515b4bfbdde560100bae
---
 .../lang/psi/api/statements/expressions/GrMethodCall.java  | 3 +++
 .../psi/impl/statements/expressions/GrMethodCallImpl.java  | 7 +++++++
 2 files changed, 10 insertions(+)

diff --git a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/expressions/GrMethodCall.java b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/expressions/GrMethodCall.java
index 8b8802348c2e..c55e9c51bb30 100644
--- a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/expressions/GrMethodCall.java
+++ b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/expressions/GrMethodCall.java
@@ -19,6 +19,9 @@ public interface GrMethodCall extends GrCallExpression, NavigationItem {
   @Nullable
   GroovyMethodCallReference getExplicitCallReference();
 
+  @Nullable
+  GroovyMethodCallReference getCallReference();
+
   @NotNull
   GrExpression getInvokedExpression();
 
diff --git a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/expressions/GrMethodCallImpl.java b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/expressions/GrMethodCallImpl.java
index aa7d865c3535..232bd9135e1f 100644
--- a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/expressions/GrMethodCallImpl.java
+++ b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/expressions/GrMethodCallImpl.java
@@ -46,6 +46,13 @@ public abstract class GrMethodCallImpl extends GrCallExpressionImpl implements G
     return isExplicitCall(this) ? myExplicitCallReference : null;
   }
 
+  @Nullable
+  @Override
+  public GroovyMethodCallReference getCallReference() {
+    GroovyMethodCallReference explicitCallReference = getExplicitCallReference();
+    return explicitCallReference == null ? getImplicitCallReference() : explicitCallReference;
+  }
+
   @Override
   @NotNull
   public GroovyResolveResult[] getCallVariants(@Nullable GrExpression upToArgument) {