br.com.tecsinapse.dataio.util.ZipUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tecsinapse-data-io Show documentation

A simple way to work with CSV, XLSX? and TXT files

The newest version!

/*
 * Tecsinapse Data Input and Output
 *
 * License: GNU Lesser General Public License (LGPL), version 3 or later
 * See the LICENSE file in the root directory or .
 */
package br.com.tecsinapse.dataio.util;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import com.google.common.io.ByteStreams;

import lombok.experimental.UtilityClass;

@UtilityClass
public final class ZipUtil {

    public static void zip(File newZip, File... filesToAdd) throws IOException {
        try (ZipOutputStream zos = new ZipOutputStream(
                new BufferedOutputStream(new FileOutputStream(newZip)))) {
            for (File file : filesToAdd) {
                addFile(zos, file);
            }
        }
    }

    public static void addFile(ZipOutputStream zos, File file) throws IOException {
        if (file.isDirectory()) {
            ZipEntry entry = new ZipEntry(file.getName());
            zos.putNextEntry(entry);
            for (File subFile : file.listFiles()) {
                addFile(zos, subFile);
            }
            zos.closeEntry();
        }

        try (FileInputStream fis = new FileInputStream(file)) {
            ZipEntry entry = new ZipEntry(file.getName());
            zos.putNextEntry(entry);
            ByteStreams.copy(fis, zos);
            zos.closeEntry();
        }
    }

}