1 files changed, 165 insertions, 0 deletions
diff --git a/plugins/java-decompiler/engine/src/org/jetbrains/java/decompiler/struct/ContextUnit.java b/plugins/java-decompiler/engine/src/org/jetbrains/java/decompiler/struct/ContextUnit.java
new file mode 100644
index 000000000000..819ae6446d07
--- /dev/null
+++ b/plugins/java-decompiler/engine/src/org/jetbrains/java/decompiler/struct/ContextUnit.java
@@ -0,0 +1,165 @@
+/*
+ * 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.java.decompiler.struct;
+
+import org.jetbrains.java.decompiler.main.extern.IResultSaver;
+import org.jetbrains.java.decompiler.struct.lazy.LazyLoader;
+import org.jetbrains.java.decompiler.struct.lazy.LazyLoader.Link;
+import org.jetbrains.java.decompiler.util.DataInputFullStream;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.jar.JarFile;
+import java.util.jar.Manifest;
+
+public class ContextUnit {
+
+  public static final int TYPE_FOLDER = 0;
+  public static final int TYPE_JAR = 1;
+  public static final int TYPE_ZIP = 2;
+
+  private final int type;
+  private final boolean own;
+
+  private final String archivePath;  // relative path to jar/zip
+  private final String filename;     // folder: relative path, archive: file name
+  private final IResultSaver resultSaver;
+  private final IDecompiledData decompiledData;
+
+  private final List<String> classEntries = new ArrayList<String>();  // class file or jar/zip entry
+  private final List<String> dirEntries = new ArrayList<String>();
+  private final List<String[]> otherEntries = new ArrayList<String[]>();
+
+  private List<StructClass> classes = new ArrayList<StructClass>();
+  private Manifest manifest;
+
+  public ContextUnit(int type, String archivePath, String filename, boolean own, IResultSaver resultSaver, IDecompiledData decompiledData) {
+    this.type = type;
+    this.own = own;
+    this.archivePath = archivePath;
+    this.filename = filename;
+    this.resultSaver = resultSaver;
+    this.decompiledData = decompiledData;
+  }
+
+  public void addClass(StructClass cl, String entryName) {
+    classes.add(cl);
+    classEntries.add(entryName);
+  }
+
+  public void addDirEntry(String entry) {
+    dirEntries.add(entry);
+  }
+
+  public void addOtherEntry(String fullPath, String entry) {
+    otherEntries.add(new String[]{fullPath, entry});
+  }
+
+  public void reload(LazyLoader loader) throws IOException {
+    List<StructClass> lstClasses = new ArrayList<StructClass>();
+
+    for (StructClass cl : classes) {
+      String oldName = cl.qualifiedName;
+
+      StructClass newCl;
+      DataInputFullStream in = loader.getClassStream(oldName);
+      try {
+        newCl = new StructClass(in, cl.isOwn(), loader);
+      }
+      finally {
+        in.close();
+      }
+
+      lstClasses.add(newCl);
+
+      Link lnk = loader.getClassLink(oldName);
+      loader.removeClassLink(oldName);
+      loader.addClassLink(newCl.qualifiedName, lnk);
+    }
+
+    classes = lstClasses;
+  }
+
+  public void save() {
+    switch (type) {
+      case TYPE_FOLDER:
+        // create folder
+        resultSaver.saveFolder(filename);
+
+        // non-class files
+        for (String[] pair : otherEntries) {
+          resultSaver.copyFile(pair[0], filename, pair[1]);
+        }
+
+        // classes
+        for (int i = 0; i < classes.size(); i++) {
+          StructClass cl = classes.get(i);
+          String entryName = decompiledData.getClassEntryName(cl, classEntries.get(i));
+          if (entryName != null) {
+            String content = decompiledData.getClassContent(cl);
+            if (content != null) {
+              resultSaver.saveClassFile(filename, cl.qualifiedName, entryName, content);
+            }
+          }
+        }
+
+        break;
+
+      case TYPE_JAR:
+      case TYPE_ZIP:
+        // create archive file
+        resultSaver.saveFolder(archivePath);
+        resultSaver.createArchive(archivePath, filename, manifest);
+
+        // directory entries
+        for (String dirEntry : dirEntries) {
+          resultSaver.saveDirEntry(archivePath, filename, dirEntry);
+        }
+
+        // non-class entries
+        for (String[] pair : otherEntries) {
+          if (type != TYPE_JAR || !JarFile.MANIFEST_NAME.equalsIgnoreCase(pair[1])) {
+            resultSaver.copyEntry(pair[0], archivePath, filename, pair[1]);
+          }
+        }
+
+        // classes
+        for (int i = 0; i < classes.size(); i++) {
+          StructClass cl = classes.get(i);
+          String entryName = decompiledData.getClassEntryName(cl, classEntries.get(i));
+          if (entryName != null) {
+            String content = decompiledData.getClassContent(cl);
+            resultSaver.saveClassEntry(archivePath, filename, cl.qualifiedName, entryName, content);
+          }
+        }
+
+        resultSaver.closeArchive(archivePath, filename);
+    }
+  }
+
+  public void setManifest(Manifest manifest) {
+    this.manifest = manifest;
+  }
+
+  public boolean isOwn() {
+    return own;
+  }
+
+  public List<StructClass> getClasses() {
+    return classes;
+  }
+}