io.github.nichetoolkit.rest.helper.ZipHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rest-toolkit-utils Show documentation
Rest toolkit utils project for Spring Boot
There is a newer version: 1.1.0
package io.github.nichetoolkit.rest.helper;

import io.github.nichetoolkit.rest.constant.UtilConstants;
import io.github.nichetoolkit.rest.error.often.FileCreateException;
import io.github.nichetoolkit.rest.error.often.ZipErrorException;

import java.io.*;
import java.util.List;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * ZipHelper
 * @author Cyan ([email protected])
 * @version v1.0.0
 */
public class ZipHelper {

    public static File zipFile(String zipPath, String filename, File file) throws ZipErrorException, FileCreateException {
        String zipFilePath = zipPath.concat(File.separator).concat(filename).concat(UtilConstants.SUFFIX_REGEX).concat(UtilConstants.ZIP_SUFFIX);
        if (filename.endsWith(UtilConstants.SUFFIX_REGEX.concat(UtilConstants.ZIP_SUFFIX))) {
            zipFilePath = zipPath.concat(File.separator).concat(filename);
        }
        File zipFile = FileHelper.createFile(zipFilePath);
        try (ZipOutputStream zipOutputStream = new ZipOutputStream(new FileOutputStream(zipFile))) {
            zipOutputStream.setComment(filename);
            try (InputStream inputStream = new FileInputStream(file)) {
                zipOutputStream.putNextEntry(new ZipEntry(file.getName()));
                int temp;
                while ((temp = inputStream.read()) != -1) {
                    zipOutputStream.write(temp);
                }
            }
        } catch (IOException exception) {
            throw new ZipErrorException(exception.getMessage());
        }
        return zipFile;
    }

    public static File zipFiles(String zipPath, String filename, List zipFiles) throws ZipErrorException, FileCreateException {
        if (zipFiles.size() == 1) {
            return zipFile(zipPath,filename,zipFiles.stream().findFirst().get());
        }
        String zipFilePath = zipPath.concat(File.separator).concat(filename).concat(UtilConstants.SUFFIX_REGEX).concat(UtilConstants.ZIP_SUFFIX);
        if (filename.endsWith(UtilConstants.SUFFIX_REGEX.concat(UtilConstants.ZIP_SUFFIX))) {
            zipFilePath = zipPath.concat(File.separator).concat(filename);
        }
        File zipFile = FileHelper.createFile(zipFilePath);
        try (ZipOutputStream zipOutputStream = new ZipOutputStream(new FileOutputStream(zipFile))) {
            zipOutputStream.setComment(filename);
            for (File file : zipFiles) {
                try (InputStream inputStream = new FileInputStream(file)) {
                    zipOutputStream.putNextEntry(new ZipEntry(file.getName()));
                    int temp;
                    while ((temp = inputStream.read()) != -1) {
                        zipOutputStream.write(temp);
                    }
                }
            }
        } catch (IOException exception) {
            throw new ZipErrorException(exception.getMessage());
        }
        return zipFile;
    }

    public static byte[] gzip(byte[] data) throws ZipErrorException {
        byte[] bytes;
        try (ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
             GZIPOutputStream gzipOutputStream = new GZIPOutputStream(byteArrayOutputStream)) {
            gzipOutputStream.write(data);
            gzipOutputStream.finish();
            bytes = byteArrayOutputStream.toByteArray();
        } catch (IOException exception) {
            throw new ZipErrorException(exception.getMessage());
        }
        return bytes;
    }

    public static byte[] ungzip(byte[] data) throws ZipErrorException {
        byte[] bytes;
        try (ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(data);
             GZIPInputStream gzipInputStream = new GZIPInputStream(byteArrayInputStream);
             ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream()) {
            byte[] buf = new byte[1024];
            int num;
            while ((num = gzipInputStream.read(buf, 0, buf.length)) != -1) {
                byteArrayOutputStream.write(buf, 0, num);
            }
            bytes = byteArrayOutputStream.toByteArray();
            byteArrayOutputStream.flush();
        } catch (IOException exception) {
            throw new ZipErrorException(exception.getMessage());
        }
        return bytes;
    }
}