From 2858552ba299fdc2d376336be62246962c99fe9d Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Tue, 13 May 2014 20:51:52 +0400
Subject: [PATCH] Add isValid() to PsiJetImportDirectiveStub

---
 .../org/jetbrains/jet/lang/psi/JetImportDirective.java |  2 +-
 .../jet/lang/psi/stubs/PsiJetImportDirectiveStub.java  |  1 +
 .../stubs/elements/JetImportDirectiveElementType.java  |  7 +++++--
 .../psi/stubs/impl/PsiJetImportDirectiveStubImpl.java  | 10 +++++++++-
 4 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetImportDirective.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetImportDirective.java
index 3644b8e4ecc..64e20eb6f81 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetImportDirective.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetImportDirective.java
@@ -120,7 +120,7 @@ public class JetImportDirective extends JetElementImplStub<PsiJetImportDirective
         return new ImportPath(importFqn, isAllUnder(), alias);
     }
 
-    private boolean isValidImport() {
+    public boolean isValidImport() {
         return !PsiTreeUtil.hasErrorElements(this);
     }
 
diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetImportDirectiveStub.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetImportDirectiveStub.java
index 476b881a339..be616eba3f0 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetImportDirectiveStub.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetImportDirectiveStub.java
@@ -26,4 +26,5 @@ public interface PsiJetImportDirectiveStub extends StubElement<JetImportDirectiv
     boolean isAllUnder();
     @Nullable
     String getAliasName();
+    boolean isValid();
 }
diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetImportDirectiveElementType.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetImportDirectiveElementType.java
index dbb98ac8e47..b5bc184b941 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetImportDirectiveElementType.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetImportDirectiveElementType.java
@@ -36,7 +36,8 @@ public class JetImportDirectiveElementType extends JetStubElementType<PsiJetImpo
     @Override
     public PsiJetImportDirectiveStub createStub(@NotNull JetImportDirective psi, StubElement parentStub) {
         StringRef aliasName = StringRef.fromString(psi.getAliasName());
-        return new PsiJetImportDirectiveStubImpl(parentStub, psi.isAbsoluteInRootPackage(), psi.isAllUnder(), aliasName);
+        return new PsiJetImportDirectiveStubImpl(parentStub, psi.isAbsoluteInRootPackage(), psi.isAllUnder(),
+                                                 aliasName, psi.isValidImport());
     }
 
     @Override
@@ -44,6 +45,7 @@ public class JetImportDirectiveElementType extends JetStubElementType<PsiJetImpo
         dataStream.writeBoolean(stub.isAbsoluteInRootPackage());
         dataStream.writeBoolean(stub.isAllUnder());
         dataStream.writeName(stub.getAliasName());
+        dataStream.writeBoolean(stub.isValid());
     }
 
     @NotNull
@@ -52,6 +54,7 @@ public class JetImportDirectiveElementType extends JetStubElementType<PsiJetImpo
         boolean isAbsoluteInRootPackage = dataStream.readBoolean();
         boolean isAllUnder = dataStream.readBoolean();
         StringRef aliasName = dataStream.readName();
-        return new PsiJetImportDirectiveStubImpl(parentStub, isAbsoluteInRootPackage, isAllUnder, aliasName);
+        boolean isValid = dataStream.readBoolean();
+        return new PsiJetImportDirectiveStubImpl(parentStub, isAbsoluteInRootPackage, isAllUnder, aliasName, isValid);
     }
 }
diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetImportDirectiveStubImpl.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetImportDirectiveStubImpl.java
index 19422024175..e46611b5ecd 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetImportDirectiveStubImpl.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetImportDirectiveStubImpl.java
@@ -29,17 +29,20 @@ public class PsiJetImportDirectiveStubImpl extends StubBase<JetImportDirective>
     private final boolean isAllUnder;
     @Nullable
     private final StringRef aliasName;
+    private final boolean isValid;
 
     public PsiJetImportDirectiveStubImpl(
             StubElement parent,
             boolean isAbsoluteInRootPackage,
             boolean isAllUnder,
-            @Nullable StringRef aliasName
+            @Nullable StringRef aliasName,
+            boolean isValid
     ) {
         super(parent, JetStubElementTypes.IMPORT_DIRECTIVE);
         this.isAbsoluteInRootPackage = isAbsoluteInRootPackage;
         this.isAllUnder = isAllUnder;
         this.aliasName = aliasName;
+        this.isValid = isValid;
     }
 
     @Override
@@ -57,4 +60,9 @@ public class PsiJetImportDirectiveStubImpl extends StubBase<JetImportDirective>
     public String getAliasName() {
         return StringRef.toString(aliasName);
     }
+
+    @Override
+    public boolean isValid() {
+        return isValid;
+    }
 }