Ejemplo n.º 1
1
  /**
   * 创建zip文件 使用文件列表
   *
   * @param zipfileName
   * @param filesToAdd
   */
  public static void CreateZipWithFilelist(String zipfileName, ArrayList<File> filesToAdd) {

    // Input and OutputStreams are defined outside of the try/catch block
    // to use them in the finally block
    ZipOutputStream outputStream = null;
    InputStream inputStream = null;

    try {
      // Prepare the files to be added
      // ArrayList filesToAdd = new ArrayList();
      // filesToAdd.add(new File("c:\\ZipTest\\sample.txt"));
      // filesToAdd.add(new File("c:\\ZipTest\\myvideo.avi"));
      // filesToAdd.add(new File("c:\\ZipTest\\mysong.mp3"));

      // Initiate output stream with the path/file of the zip file
      // Please note that ZipOutputStream will overwrite zip file if it
      // already exists
      outputStream = new ZipOutputStream(new FileOutputStream(new File(zipfileName)));

      // Initiate Zip Parameters which define various properties such
      // as compression method, etc. More parameters are explained in
      // other
      // examples
      ZipParameters parameters = new ZipParameters();

      // Deflate compression or store(no compression) can be set below
      parameters.setCompressionMethod(Zip4jConstants.COMP_DEFLATE);

      // Set the compression level. This value has to be in between 0 to 9
      // Several predefined compression levels are available
      // DEFLATE_LEVEL_FASTEST - Lowest compression level but higher speed
      // of compression
      // DEFLATE_LEVEL_FAST - Low compression level but higher speed of
      // compression
      // DEFLATE_LEVEL_NORMAL - Optimal balance between compression
      // level/speed
      // DEFLATE_LEVEL_MAXIMUM - High compression level with a compromise
      // of speed
      // DEFLATE_LEVEL_ULTRA - Highest compression level but low speed
      parameters.setCompressionLevel(Zip4jConstants.DEFLATE_LEVEL_NORMAL);

      // This flag defines if the files have to be encrypted.
      // If this flag is set to false, setEncryptionMethod, as described
      // below,
      // will be ignored and the files won't be encrypted
      parameters.setEncryptFiles(true);

      // Zip4j supports AES or Standard Zip Encryption (also called
      // ZipCrypto)
      // If you choose to use Standard Zip Encryption, please have a look
      // at example
      // as some additional steps need to be done while using
      // ZipOutputStreams with
      // Standard Zip Encryption
      parameters.setEncryptionMethod(Zip4jConstants.ENC_METHOD_AES);

      // If AES encryption is used, this defines the key strength
      parameters.setAesKeyStrength(Zip4jConstants.AES_STRENGTH_256);

      // self descriptive
      parameters.setPassword("YourPassword");

      // Now we loop through each file and read this file with an
      // inputstream
      // and write it to the ZipOutputStream.
      for (int i = 0; i < filesToAdd.size(); i++) {
        File file = (File) filesToAdd.get(i);

        // This will initiate ZipOutputStream to include the file
        // with the input parameters
        outputStream.putNextEntry(file, parameters);

        // If this file is a directory, then no further processing is
        // required
        // and we close the entry (Please note that we do not close the
        // outputstream yet)
        if (file.isDirectory()) {
          outputStream.closeEntry();
          continue;
        }

        // Initialize inputstream
        inputStream = new FileInputStream(file);
        byte[] readBuff = new byte[4096];
        int readLen = -1;

        // Read the file content and write it to the OutputStream
        while ((readLen = inputStream.read(readBuff)) != -1) {
          outputStream.write(readBuff, 0, readLen);
        }

        // Once the content of the file is copied, this entry to the zip
        // file
        // needs to be closed. ZipOutputStream updates necessary header
        // information
        // for this file in this step
        outputStream.closeEntry();

        inputStream.close();
      }

      // ZipOutputStream now writes zip header information to the zip file
      outputStream.finish();

    } catch (Exception e) {
      e.printStackTrace();
    } finally {
      if (outputStream != null) {
        try {
          outputStream.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }

      if (inputStream != null) {
        try {
          inputStream.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
    }
  }
Ejemplo n.º 2
0
  public static ByteArrayOutputStream requestsToZip(
      ServletContext context,
      String globalTitle,
      Date generationDate,
      ScrambleRequest[] scrambleRequests,
      String password,
      String generationUrl)
      throws IOException, DocumentException, ZipException {
    ByteArrayOutputStream baosZip = new ByteArrayOutputStream();

    ZipParameters parameters = new ZipParameters();
    parameters.setCompressionMethod(Zip4jConstants.COMP_DEFLATE);
    parameters.setCompressionLevel(Zip4jConstants.DEFLATE_LEVEL_NORMAL);
    if (password != null) {
      parameters.setEncryptFiles(true);
      parameters.setEncryptionMethod(Zip4jConstants.ENC_METHOD_STANDARD);
      parameters.setPassword(password);
    }
    parameters.setSourceExternalStream(true);

    ZipOutputStream zipOut = new ZipOutputStream(baosZip);
    HashMap<String, Boolean> seenTitles = new HashMap<String, Boolean>();
    for (ScrambleRequest scrambleRequest : scrambleRequests) {
      String safeTitle = toFileSafeString(scrambleRequest.title);
      int salt = 0;
      String tempNewSafeTitle = safeTitle;
      while (seenTitles.get(tempNewSafeTitle) != null) {
        tempNewSafeTitle = safeTitle + " (" + (++salt) + ")";
      }
      safeTitle = tempNewSafeTitle;
      seenTitles.put(safeTitle, true);

      String pdfFileName = "pdf/" + safeTitle + ".pdf";
      parameters.setFileNameInZip(pdfFileName);
      zipOut.putNextEntry(null, parameters);

      PdfReader pdfReader = createPdf(globalTitle, generationDate, scrambleRequest);
      byte[] b = new byte[(int) pdfReader.getFileLength()];
      pdfReader.getSafeFile().readFully(b);
      zipOut.write(b);

      zipOut.closeEntry();

      String txtFileName = "txt/" + safeTitle + ".txt";
      parameters.setFileNameInZip(txtFileName);
      zipOut.putNextEntry(null, parameters);
      zipOut.write(join(stripNewlines(scrambleRequest.getAllScrambles()), "\r\n").getBytes());
      zipOut.closeEntry();
    }

    String safeGlobalTitle = toFileSafeString(globalTitle);
    String jsonFileName = safeGlobalTitle + ".json";
    parameters.setFileNameInZip(jsonFileName);
    zipOut.putNextEntry(null, parameters);
    HashMap<String, Object> jsonObj = new HashMap<String, Object>();
    jsonObj.put("sheets", scrambleRequests);
    jsonObj.put("competitionName", globalTitle);
    jsonObj.put("version", Utils.getProjectName() + "-" + Utils.getVersion());
    jsonObj.put("generationDate", generationDate);
    jsonObj.put("generationUrl", generationUrl);
    String json = GSON.toJson(jsonObj);
    zipOut.write(json.getBytes());
    zipOut.closeEntry();

    String jsonpFileName = safeGlobalTitle + ".jsonp";
    parameters.setFileNameInZip(jsonpFileName);
    zipOut.putNextEntry(null, parameters);
    String jsonp = "var SCRAMBLES_JSON = " + json + ";";
    zipOut.write(jsonp.getBytes());
    zipOut.closeEntry();

    parameters.setFileNameInZip(safeGlobalTitle + ".html");
    zipOut.putNextEntry(null, parameters);

    InputStream is = context.getResourceAsStream(HTML_SCRAMBLE_VIEWER);
    BufferedReader in = new BufferedReader(new InputStreamReader(is));
    StringBuilder sb = new StringBuilder();
    String line;
    while ((line = in.readLine()) != null) {
      line = line.replaceAll("%SCRAMBLES_JSONP_FILENAME%", jsonpFileName);
      sb.append(line).append("\n");
    }
    zipOut.write(sb.toString().getBytes());
    zipOut.closeEntry();

    parameters.setFileNameInZip(safeGlobalTitle + ".pdf");
    zipOut.putNextEntry(null, parameters);
    // Note that we're not passing the password into this function. It seems pretty silly
    // to put a password protected pdf inside of a password protected zip file.
    ByteArrayOutputStream baos = requestsToPdf(globalTitle, generationDate, scrambleRequests, null);
    zipOut.write(baos.toByteArray());
    zipOut.closeEntry();

    zipOut.finish();
    zipOut.close();

    return baosZip;
  }