io.proximax.upload.FilesAsZipParameterData Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-chain-xipfs-sdk Show documentation
Official ProximaX P2P Storage SDK Library in Java.
The newest version!
package io.proximax.upload;

import io.proximax.exceptions.ParamDataCreationException;
import org.apache.commons.io.FileUtils;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import static io.proximax.utils.ParameterValidationUtils.checkParameter;

/**
 * This model class is one type of the upload parameter data that defines a zip upload
 */
public class FilesAsZipParameterData extends AbstractByteStreamParameterData {

    private final List files;
    private final byte[] zipData;

    private FilesAsZipParameterData(List files, String description, String name, Map metadata) {
        super(description, name, "application/zip", metadata);

        checkParameter(files != null && !files.isEmpty(), "files cannot be null or empty");
        checkParameter(files.stream().allMatch(File::isFile), "not all files are file");

        this.files = files;
        zipData = zipFiles(files);
    }

    /**
     * Get the byte stream
     * @return the byte stream
     */
    @Override
    public InputStream getByteStream() {
        return new ByteArrayInputStream(zipData);
    }

    /**
     * Get the list of files to upload as zip
     * @return the list of files to zip
     */
    public List getFiles() {
        return files;
    }

    private static byte[] zipFiles(List files) {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        try (ZipOutputStream zos = new ZipOutputStream(baos)) {

            for (File file : files) {
                ZipEntry entry = new ZipEntry(file.getName());

                zos.putNextEntry(entry);
                zos.write(FileUtils.readFileToByteArray(file));
                zos.closeEntry();
            }

            return baos.toByteArray();
        } catch (Exception e) {
            throw new ParamDataCreationException("Failed to create zip file", e);
        }
    }

    /**
     * Create instance by providing the list of files
     * @param files the list of files to upload as zip
     * @return the instance of this class
     */
    public static FilesAsZipParameterData create(List files) {
        return create(files, null, null, null);
    }

    /**
     * Create instance by providing the list of files
     * @param files the list of files to upload as zip
     * @param description a searchable description attach on the upload
     * @param name a searchable name attach on the upload
     * @param metadata a searchable key-pair metadata attach on the upload
     * @return the instance of this class
     */
    public static FilesAsZipParameterData create(List files, String description, String name, Map metadata) {
        return new FilesAsZipParameterData(files, description, name, metadata);
    }
}