Faster completion of non-imported extensions

2014-11-07 22:35:15 +03:00
parent ff4d91d47b
commit 2512962134
4 changed files with 73 additions and 33 deletions
@@ -17,6 +17,7 @@
 package org.jetbrains.jet.lang.psi;

 import com.intellij.lang.ASTNode;
+import com.intellij.psi.PsiElement;
 import com.intellij.psi.stubs.IStubElementType;
 import com.intellij.psi.stubs.StubElement;
 import org.jetbrains.annotations.NotNull;
@@ -38,4 +39,13 @@ abstract class JetDeclarationStub<T extends StubElement> extends JetModifierList
    public KDoc getDocComment() {
        return FindDocCommentPackage.findDocComment(this);
    }
+
+    @Override
+    public PsiElement getParent() {
+        T stub = getStub();
+        if (stub != null) {
+            return stub.getParentStub().getPsi();
+        }
+        return super.getParent();
+    }
 }