edu.pdx.cs.joy.grader.ZipFileMaker Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of grader Show documentation
There is a newer version: 1.2.0
package edu.pdx.cs.joy.grader;

import com.google.common.io.ByteStreams;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.util.Date;
import java.util.Map;
import java.util.jar.Attributes;
import java.util.jar.JarFile;
import java.util.jar.Manifest;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipFileMaker {
  protected static final Logger logger = LoggerFactory.getLogger("edu.pdx.cs.joy.grader");

  protected final Map manifestEntries;
  private OutputStream zipStream;

  public ZipFileMaker(File zipFile, Map manifestEntries) throws FileNotFoundException {
    this(new FileOutputStream(zipFile), manifestEntries);
  }

  public ZipFileMaker(OutputStream zipStream, Map manifestEntries) {
    this.manifestEntries = manifestEntries;
    this.zipStream = zipStream;
  }

  protected void makeZipFile(Map zipFileEntries) throws IOException {
    ZipOutputStream zos = new ZipOutputStream(zipStream);
    zos.setMethod(ZipOutputStream.DEFLATED);

    // Create a Manifest for the Zip file containing the name of the
    // author (userName) and a version that is based on the current
    // date/time.
    writeManifestAsEntryInZipFile(zos);

    // Add the source files to the Zip
    for (Map.Entry entry : zipFileEntries.entrySet()) {
      // Add the entry to the ZIP file
      zos.putNextEntry(entry.getKey());

      ByteStreams.copy(entry.getValue(), zos);

      zos.closeEntry();
    }

    zos.close();
  }

  private void writeManifestAsEntryInZipFile(ZipOutputStream zos) throws IOException {
    Manifest manifest = new Manifest();
    addEntriesToMainManifest(manifest);

    String entryName = JarFile.MANIFEST_NAME;

    logger.debug("Adding " + entryName + " to zip");
    ZipEntry entry = new ZipEntry(entryName);
    entry.setTime(System.currentTimeMillis());
    entry.setMethod(ZipEntry.DEFLATED);

    zos.putNextEntry(entry);
    manifest.write(new BufferedOutputStream(zos));
    zos.closeEntry();
  }

  private void addEntriesToMainManifest(Manifest manifest) {
    Attributes attrs = manifest.getMainAttributes();

    // If a manifest doesn't have a version, the other attributes won't get written out.  Lame.
    attrs.put(Attributes.Name.MANIFEST_VERSION, new Date().toString());

    for (Map.Entry entry : this.manifestEntries.entrySet()) {
      attrs.put(entry.getKey(), entry.getValue());
    }
  }
}