1 files changed, 84 insertions, 0 deletions
diff --git a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/expressions/GrSuperReferenceResolver.java b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/expressions/GrSuperReferenceResolver.java
new file mode 100644
index 000000000000..0b45314b34a0
--- /dev/null
+++ b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/expressions/GrSuperReferenceResolver.java
@@ -0,0 +1,84 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.jetbrains.plugins.groovy.lang.psi.impl.statements.expressions;
+
+import com.intellij.psi.PsiClass;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiSubstitutor;
+import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.psi.util.TypeConversionUtil;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.jetbrains.plugins.groovy.lang.psi.api.GroovyResolveResult;
+import org.jetbrains.plugins.groovy.lang.psi.api.statements.GrConstructorInvocation;
+import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrExpression;
+import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrReferenceExpression;
+import org.jetbrains.plugins.groovy.lang.psi.api.statements.typedef.GrTypeDefinition;
+import org.jetbrains.plugins.groovy.lang.psi.impl.GroovyResolveResultImpl;
+import org.jetbrains.plugins.groovy.lang.psi.util.GrTraitUtil;
+import org.jetbrains.plugins.groovy.lang.psi.util.PsiUtil;
+
+/**
+ * Created by Max Medvedev on 15/06/14
+ */
+public class GrSuperReferenceResolver {
+  @Nullable("null if ref is not 'super' reference")
+  public static GroovyResolveResult[] resolveSuperExpression(@NotNull GrReferenceExpression ref) {
+    GrExpression qualifier = ref.getQualifier();
+
+    if (qualifier == null) {
+      final PsiElement parent = ref.getParent();
+      if (parent instanceof GrConstructorInvocation) {
+        return ((GrConstructorInvocation)parent).multiResolve(false);
+      }
+      PsiClass aClass = PsiUtil.getContextClass(ref);
+      if (aClass != null) {
+        return getSuperClass(aClass);
+      }
+    }
+    else if (qualifier instanceof GrReferenceExpression) {
+      GroovyResolveResult result = ((GrReferenceExpression)qualifier).advancedResolve();
+      PsiElement resolved = result.getElement();
+      if (resolved instanceof PsiClass) {
+        PsiClass superClass = (PsiClass)resolved;
+
+        GrTypeDefinition scopeClass = PsiTreeUtil.getParentOfType(ref, GrTypeDefinition.class, true);
+        if (scopeClass != null && GrTraitUtil.isTrait(superClass) && scopeClass.isInheritor(superClass, false)) {
+          PsiSubstitutor superClassSubstitutor = TypeConversionUtil.getSuperClassSubstitutor(superClass, scopeClass, PsiSubstitutor.EMPTY);
+          return new GroovyResolveResultImpl[]{new GroovyResolveResultImpl(superClass, null, null, superClassSubstitutor, true, true)};
+        }
+
+        if (PsiUtil.hasEnclosingInstanceInScope(superClass, ref, false)) {
+          return getSuperClass(superClass);
+        }
+      }
+    }
+
+    return null;
+  }
+
+  @NotNull
+  private static GroovyResolveResult[] getSuperClass(@NotNull PsiClass aClass) {
+    PsiClass superClass = aClass.getSuperClass();
+    if (superClass != null) {
+      PsiSubstitutor superClassSubstitutor = TypeConversionUtil.getSuperClassSubstitutor(superClass, aClass, PsiSubstitutor.EMPTY);
+      return new GroovyResolveResultImpl[]{new GroovyResolveResultImpl(superClass, null, null, superClassSubstitutor, true, true)};
+    }
+    else {
+      return GroovyResolveResult.EMPTY_ARRAY; //no super class, but the reference is definitely super-reference
+    }
+  }
+}