package com.atlassian.core.util.zip;
   
   import org.apache.commons.io.IOUtils;
   import org.apache.commons.lang.StringUtils;
   
   import java.io.BufferedInputStream;
   import java.io.File;
   import java.io.FileInputStream;
   import java.io.IOException;
  import java.io.InputStream;
  import java.util.zip.ZipEntry;
  import java.util.zip.ZipOutputStream;
  
  /**
   * Handles adding files to zip archives. Supports appending files to existing zip output stream.
   *
   * @since v4.6.13
   */
  public class FileArchiver
  {
      private final ZipOutputStream zipOutputStream;
  
      public FileArchiver(final ZipOutputStream zipOutputStream)
      {
          if (zipOutputStream == null)
          {
              throw new NullPointerException("output stream can't be null");
          }
          this.zipOutputStream = zipOutputStream;
      }
  
      /**
       * Adds file to the current archive output stream
       *
       * @param file file to add to archive
       * @param archiveFolderToCreate name of topmost folder to create inside of the archive, leave blank to not create.
       * @throws java.io.IOException if can't read from file or can't write to output
       */
      public void addToArchive(final File file, final String path, final String archiveFolderToCreate) throws IOException
      {
          if ((file == null) || !file.exists())
          {
              return;
          }
          final String escapedPath = getEntryPath(FilePathUtils.stripSlashes(path), archiveFolderToCreate);
          final ZipEntry entry = new ZipEntry(escapedPath);
  
          entry.setTime(file.lastModified());
          zipOutputStream.putNextEntry(entry);
  
          final InputStream in = new BufferedInputStream(new FileInputStream(file));
          try
          {
              IOUtils.copyLarge(in, zipOutputStream);
          }
          finally
          {
              IOUtils.closeQuietly(in);
          }
  
      }
  
      public void addDirectoryToArchive(final File file, final String path, final String archiveFolderName)
              throws IOException
      {
          if ((file == null) || !file.exists() || !file.isDirectory())
          {
              return;
          }
          final String escapedPath = getEntryPath(FilePathUtils.stripSlashes(path), archiveFolderName);
          final String directoryEscapedPath = escapedPath.endsWith("/") ? escapedPath : escapedPath + "/";
  
          final ZipEntry entry = new ZipEntry(directoryEscapedPath);
  
          entry.setTime(file.lastModified());
          zipOutputStream.putNextEntry(entry);
      }
  
      /**
       * Builds a file path for an entry inside of the archive Replaces forward slashes, removes extra slashes. Also
       * handles creation of a topmost directory in archive, if necessary.
       *
       * @param path related path inside of archive
       * @param archiveFolderToCreate name of topmost folder to create inside of the archive, leave blank to not create.
       * @return path to use inside of archive
       */
      private static String getEntryPath(final String path, final String archiveFolderToCreate)
      {
          // if creating zip on a windows system, convert all backslashes to forward slashes so that the resulting file is platform-independent (work on unix and windows)
          String sanitisedPath = path;
          sanitisedPath = sanitisedPath.replaceAll("\\\\", "/");
  
          // also drop the leading slash
          if (sanitisedPath.length() > 0 && sanitisedPath.charAt(0) == '/')
          {
              sanitisedPath = sanitisedPath.substring(1);
          }
          if (StringUtils.isNotBlank(archiveFolderToCreate))
          {
             sanitisedPath = archiveFolderToCreate + "/" + path;
         }
         return sanitisedPath;
     }
 }