fr.gael.dhus.util.UnZip.java Source code

Introduction

Here is the source code for fr.gael.dhus.util.UnZip.java

Source

/*
 * Data Hub Service (DHuS) - For Space data distribution.
 * Copyright (C) 2013,2014,2015 GAEL Systems
 *
 * This file is part of DHuS software sources.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package fr.gael.dhus.util;

import org.apache.commons.compress.archivers.ArchiveEntry;
import org.apache.commons.compress.archivers.ArchiveException;
import org.apache.commons.compress.archivers.ArchiveInputStream;
import org.apache.commons.compress.archivers.ArchiveStreamFactory;
import org.apache.commons.compress.compressors.CompressorException;
import org.apache.commons.compress.compressors.CompressorInputStream;
import org.apache.commons.compress.compressors.CompressorStreamFactory;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.List;

public class UnZip {
    private static final Logger LOGGER = LogManager.getLogger();

    List<String> fileList;

    public static void unCompress(String zip_file, String output_folder)
            throws IOException, CompressorException, ArchiveException {
        ArchiveInputStream ais = null;
        ArchiveStreamFactory asf = new ArchiveStreamFactory();

        FileInputStream fis = new FileInputStream(new File(zip_file));

        if (zip_file.toLowerCase().endsWith(".tar")) {
            ais = asf.createArchiveInputStream(ArchiveStreamFactory.TAR, fis);
        } else if (zip_file.toLowerCase().endsWith(".zip")) {
            ais = asf.createArchiveInputStream(ArchiveStreamFactory.ZIP, fis);
        } else if (zip_file.toLowerCase().endsWith(".tgz") || zip_file.toLowerCase().endsWith(".tar.gz")) {
            CompressorInputStream cis = new CompressorStreamFactory()
                    .createCompressorInputStream(CompressorStreamFactory.GZIP, fis);
            ais = asf.createArchiveInputStream(new BufferedInputStream(cis));
        } else {
            try {
                fis.close();
            } catch (IOException e) {
                LOGGER.warn("Cannot close FileInputStream:", e);
            }
            throw new IllegalArgumentException("Format not supported: " + zip_file);
        }

        File output_file = new File(output_folder);
        if (!output_file.exists())
            output_file.mkdirs();

        // copy the existing entries
        ArchiveEntry nextEntry;
        while ((nextEntry = ais.getNextEntry()) != null) {
            File ftemp = new File(output_folder, nextEntry.getName());
            if (nextEntry.isDirectory()) {
                ftemp.mkdir();
            } else {
                FileOutputStream fos = FileUtils.openOutputStream(ftemp);
                IOUtils.copy(ais, fos);
                fos.close();
            }
        }
        ais.close();
        fis.close();
    }

    public static boolean supported(String file) {
        boolean is_supported = file.toLowerCase().endsWith(".zip")/* ||
                                                                  file.toLowerCase ().endsWith (".tar") ||
                                                                  file.toLowerCase ().endsWith (".tgz") ||
                                                                  file.toLowerCase ().endsWith (".tar.gz")*/;
        File fileObject = new File(file);
        is_supported &= fileObject.exists() && fileObject.isFile();
        return is_supported;
    }
}