From 12310b5bd94ff3805f9b74e538d5802a54bcfdff Mon Sep 17 00:00:00 2001
From: Alexey Sedunov <alexey.sedunov@jetbrains.com>
Date: Mon, 3 Mar 2014 19:11:36 +0400
Subject: [PATCH] Reuse getQualifiedElementSelector() function

---
 .../src/org/jetbrains/jet/lang/psi/JetPsiUtil.java     | 10 ++--------
 .../org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt   |  2 +-
 2 files changed, 3 insertions(+), 9 deletions(-)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java
index 6f26c31a31c..aec5d03b885 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java
@@ -399,14 +399,8 @@ public class JetPsiUtil {
 
     @Nullable
     public static JetSimpleNameExpression getLastReference(@NotNull JetExpression importedReference) {
-        if (importedReference instanceof JetDotQualifiedExpression) {
-            JetExpression selectorExpression = ((JetDotQualifiedExpression) importedReference).getSelectorExpression();
-            return (selectorExpression instanceof JetSimpleNameExpression) ? (JetSimpleNameExpression) selectorExpression : null;
-        }
-        if (importedReference instanceof JetSimpleNameExpression) {
-            return (JetSimpleNameExpression) importedReference;
-        }
-        return null;
+        JetElement selector = PsiUtilPackage.getQualifiedElementSelector(importedReference);
+        return selector instanceof JetSimpleNameExpression ? (JetSimpleNameExpression) selector : null;
     }
 
     public static boolean isSelectorInQualified(@NotNull JetSimpleNameExpression nameExpression) {
diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt b/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt
index 3e5fd21f5d1..9b6a81a082e 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt
@@ -242,7 +242,7 @@ fun JetElement.getQualifiedElementSelector(): JetElement? {
             if (selector is JetCallExpression) selector.getCalleeExpression() else selector
         }
         is JetUserType -> getReferenceExpression()
-        else -> this
+        else -> null
     }
 }