org.egov.infra.utils.FileUtils.java Source code

Introduction

Here is the source code for org.egov.infra.utils.FileUtils.java

Source

/*
 *    eGov  SmartCity eGovernance suite aims to improve the internal efficiency,transparency,
 *    accountability and the service delivery of the government  organizations.
 *
 *     Copyright (C) 2017  eGovernments Foundation
 *
 *     The updated version of eGov suite of products as by eGovernments Foundation
 *     is available at http://www.egovernments.org
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with this program. If not, see http://www.gnu.org/licenses/ or
 *     http://www.gnu.org/licenses/gpl.html .
 *
 *     In addition to the terms of the GPL license to be adhered to in using this
 *     program, the following additional terms are to be complied with:
 *
 *         1) All versions of this program, verbatim or modified must carry this
 *            Legal Notice.
 *            Further, all user interfaces, including but not limited to citizen facing interfaces,
 *            Urban Local Bodies interfaces, dashboards, mobile applications, of the program and any
 *            derived works should carry eGovernments Foundation logo on the top right corner.
 *
 *            For the logo, please refer http://egovernments.org/html/logo/egov_logo.png.
 *            For any further queries on attribution, including queries on brand guidelines,
 *            please contact contact@egovernments.org
 *
 *         2) Any misrepresentation of the origin of the material is prohibited. It
 *            is required that all modified versions of this material be marked in
 *            reasonable ways as different from the original version.
 *
 *         3) This license does not grant any rights to any user of the program
 *            with regards to rights under trademark law for use of the trade names
 *            or trademarks of eGovernments Foundation.
 *
 *   In case of any queries, you can reach eGovernments Foundation at contact@egovernments.org.
 *
 */

package org.egov.infra.utils;

import org.apache.commons.compress.archivers.zip.ZipArchiveEntry;
import org.apache.commons.compress.archivers.zip.ZipArchiveOutputStream;
import org.apache.commons.io.IOUtils;
import org.egov.infra.exception.ApplicationRuntimeException;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.UUID;

public final class FileUtils {

    private static final String ZIP_FILE_EXTN = "zip";

    private FileUtils() {
        //only static methods
    }

    public static Path addFilesToZip(File... files) {
        try {
            Path zipFile = Files.createTempFile(UUID.randomUUID().toString(), ZIP_FILE_EXTN);
            try (FileOutputStream zipFileStream = new FileOutputStream(zipFile.toFile());
                    ZipArchiveOutputStream zipOutput = new ZipArchiveOutputStream(zipFileStream)) {
                for (File file : files) {
                    zipOutput.putArchiveEntry(new ZipArchiveEntry(file, file.getName()));
                    zipOutput.write(IOUtils.toByteArray(file.toURI()));
                    zipOutput.closeArchiveEntry();
                }
                return zipFile;
            }
        } catch (IOException e) {
            throw new ApplicationRuntimeException("Error occurred while adding files to ZIP", e);
        }
    }

    public static Path byteArrayToFile(byte[] fileBytes, String fileName, String fileExt) {
        try {
            Path file = Files.createTempFile(fileName, fileExt);
            Files.write(file, fileBytes);
            return file;
        } catch (IOException e) {
            throw new ApplicationRuntimeException("Error occurred while converting byte array to file", e);
        }

    }

    public static byte[] toByteArray(Path path) {
        try {
            return Files.readAllBytes(path);
        } catch (IOException e) {
            throw new ApplicationRuntimeException("Error occurred while converting file to byte array", e);
        }
    }
}