From 95f9610f06c2198b2a856560a93095e8033bf529 Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Fri, 28 Mar 2014 19:50:13 +0400
Subject: [PATCH] Add isAbsoluteInRootPackage and isAllUnder to
 PsiJetImportDirectiveStub

---
 .../psi/stubs/PsiJetImportDirectiveStub.java  |  2 ++
 .../JetImportDirectiveElementType.java        | 10 +++++----
 .../impl/PsiJetImportDirectiveStubImpl.java   | 21 ++++++++++++++++++-
 3 files changed, 28 insertions(+), 5 deletions(-)

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 0b1ca3c8b5b..9274b60d587 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
@@ -20,4 +20,6 @@ import com.intellij.psi.stubs.StubElement;
 import org.jetbrains.jet.lang.psi.JetImportDirective;
 
 public interface PsiJetImportDirectiveStub extends StubElement<JetImportDirective> {
+    boolean isAbsoluteInRootPackage();
+    boolean isAllUnder();
 }
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 c4c0fca3bfd..76beb0210fc 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
@@ -34,18 +34,20 @@ public class JetImportDirectiveElementType extends JetStubElementType<PsiJetImpo
 
     @Override
     public PsiJetImportDirectiveStub createStub(@NotNull JetImportDirective psi, StubElement parentStub) {
-        return new PsiJetImportDirectiveStubImpl(parentStub);
+        return new PsiJetImportDirectiveStubImpl(parentStub, psi.isAbsoluteInRootPackage(), psi.isAllUnder());
     }
 
     @Override
     public void serialize(@NotNull PsiJetImportDirectiveStub stub, @NotNull StubOutputStream dataStream) throws IOException {
-        //TODO:
+        dataStream.writeBoolean(stub.isAbsoluteInRootPackage());
+        dataStream.writeBoolean(stub.isAllUnder());
     }
 
     @NotNull
     @Override
     public PsiJetImportDirectiveStub deserialize(@NotNull StubInputStream dataStream, StubElement parentStub) throws IOException {
-        //TODO
-        return new PsiJetImportDirectiveStubImpl(parentStub);
+        boolean isAbsoluteInRootPackage = dataStream.readBoolean();
+        boolean isAllUnder = dataStream.readBoolean();
+        return new PsiJetImportDirectiveStubImpl(parentStub, isAbsoluteInRootPackage, isAllUnder);
     }
 }
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 d0c90b1027f..c92693f47e0 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
@@ -23,7 +23,26 @@ import org.jetbrains.jet.lang.psi.stubs.PsiJetImportDirectiveStub;
 import org.jetbrains.jet.lang.psi.stubs.elements.JetStubElementTypes;
 
 public class PsiJetImportDirectiveStubImpl extends StubBase<JetImportDirective> implements PsiJetImportDirectiveStub {
-    public PsiJetImportDirectiveStubImpl(StubElement parent) {
+    private final boolean isAbsoluteInRootPackage;
+    private final boolean isAllUnder;
+
+    public PsiJetImportDirectiveStubImpl(
+            StubElement parent,
+            boolean isAbsoluteInRootPackage,
+            boolean isAllUnder
+    ) {
         super(parent, JetStubElementTypes.IMPORT_DIRECTIVE);
+        this.isAbsoluteInRootPackage = isAbsoluteInRootPackage;
+        this.isAllUnder = isAllUnder;
+    }
+
+    @Override
+    public boolean isAbsoluteInRootPackage() {
+        return isAbsoluteInRootPackage;
+    }
+
+    @Override
+    public boolean isAllUnder() {
+        return isAllUnder;
     }
 }