org.molgenis.util.file.ZipFileUtil Maven / Gradle / Ivy

Go to download
package org.molgenis.util.file;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

public class ZipFileUtil
{
	private static final Logger LOG = LoggerFactory.getLogger(ZipFileUtil.class);

	private static void copyInputStream(InputStream in, OutputStream out) throws IOException
	{
		byte[] buffer = new byte[1024];
		int len;
		try
		{
			while ((len = in.read(buffer)) >= 0) out.write(buffer, 0, len);
		}
		finally
		{
			if (in != null) in.close();
			if (out != null) out.close();
		}
	}

	public static List unzip(File file) throws IOException
	{

		List unzippedFiles = new ArrayList<>();
		Enumeration entries;
		ZipFile zipFile = null;
		try
		{
			zipFile = new ZipFile(file);
			entries = zipFile.entries();
			while (entries.hasMoreElements())
			{
				ZipEntry entry = entries.nextElement();
				if (entry.getName().startsWith(".") || entry.getName().startsWith("_"))
				{
					continue;
				}

				if (entry.isDirectory())
				{
					LOG.info("Extracting directory: " + entry.getName());
					if (!(new File(file.getParentFile(), entry.getName())).mkdir())
					{
						throw new RuntimeException("Failed to create directory");
					}
					continue;
				}
				LOG.info("Extracting directory: " + entry.getName());
				File newFile = new File(file.getParent(), entry.getName());
				copyInputStream(zipFile.getInputStream(entry), new BufferedOutputStream(new FileOutputStream(newFile)));

				unzippedFiles.add(newFile);
			}
		}
		finally
		{
			if (zipFile != null) zipFile.close();
		}
		return unzippedFiles;
	}
}