org.abysm.onionzip.ZipFileExtractHelper.java Source code

Introduction

Here is the source code for org.abysm.onionzip.ZipFileExtractHelper.java

Source

/*
 * Copyright 2016 Kuan-Yi Li
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.abysm.onionzip;

import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.FileTime;
import java.util.Enumeration;

import org.apache.commons.compress.archivers.zip.ZipArchiveEntry;
import org.apache.commons.compress.archivers.zip.ZipFile;
import org.apache.commons.compress.utils.IOUtils;
import org.mozilla.universalchardet.UniversalDetector;

class ZipFileExtractHelper {
    private String zipFilename;
    private String encoding;

    ZipFileExtractHelper(String zipFilename) {
        this.zipFilename = zipFilename;
    }

    String getEncoding() {
        return encoding;
    }

    void setEncoding(String encoding) {
        this.encoding = encoding;
    }

    void setEncodingByDetection() throws IOException {
        UniversalDetector filenameCharsetDetector = new UniversalDetector(null);
        ZipFile zipFile = new ZipFile(zipFilename);
        try {
            for (Enumeration<ZipArchiveEntry> zipArchiveEntryEnumeration = zipFile
                    .getEntries(); zipArchiveEntryEnumeration.hasMoreElements();) {
                ZipArchiveEntry entry = zipArchiveEntryEnumeration.nextElement();
                if (!entry.getGeneralPurposeBit().usesUTF8ForNames()) {
                    byte[] filename = entry.getRawName();
                    filenameCharsetDetector.handleData(filename, 0, filename.length);
                }
            }
        } finally {
            ZipFile.closeQuietly(zipFile);
        }
        filenameCharsetDetector.dataEnd();
        this.encoding = filenameCharsetDetector.getDetectedCharset();
    }

    void listZipArchiveEntries() throws IOException {
        ZipFile zipFile = new ZipFile(zipFilename, encoding);
        try {
            System.out.println("Length\tDatetime\tName\tEFS\tUnix Mode");
            for (Enumeration<ZipArchiveEntry> zipArchiveEntryEnumeration = zipFile
                    .getEntries(); zipArchiveEntryEnumeration.hasMoreElements();) {
                ZipArchiveEntry entry = zipArchiveEntryEnumeration.nextElement();
                System.out.format("%d\t%s\t%s\t%b\t%o\n", entry.getSize(), entry.getLastModifiedDate().toString(),
                        entry.getName(), entry.getGeneralPurposeBit().usesUTF8ForNames(), entry.getUnixMode());
            }
        } finally {
            ZipFile.closeQuietly(zipFile);
        }
    }

    void extractZipArchiveEntries() throws IOException {
        ZipFile zipFile = new ZipFile(zipFilename, encoding);
        try {
            for (Enumeration<ZipArchiveEntry> zipArchiveEntryEnumeration = zipFile
                    .getEntries(); zipArchiveEntryEnumeration.hasMoreElements();) {
                ZipArchiveEntry entry = zipArchiveEntryEnumeration.nextElement();
                System.out.println(entry.getName());
                if (entry.isDirectory()) {
                    Path directory = Paths.get(entry.getName());
                    Files.createDirectories(directory);
                } else if (entry.isUnixSymlink()) {
                    Path symlink = Paths.get(entry.getName());
                    Path parentDirectory = symlink.getParent();
                    Path target = Paths.get(zipFile.getUnixSymlink(entry));
                    if (parentDirectory != null) {
                        Files.createDirectories(parentDirectory);
                    }
                    Files.createSymbolicLink(symlink, target);
                } else {
                    Path file = Paths.get(entry.getName());
                    Path parentDirectory = file.getParent();
                    if (parentDirectory != null) {
                        Files.createDirectories(parentDirectory);
                    }
                    InputStream contentInputStream = zipFile.getInputStream(entry);
                    FileOutputStream extractedFileOutputStream = new FileOutputStream(entry.getName());
                    try {
                        IOUtils.copy(contentInputStream, extractedFileOutputStream);
                    } finally {
                        IOUtils.closeQuietly(contentInputStream);
                        IOUtils.closeQuietly(extractedFileOutputStream);
                    }
                    FileTime fileTime = FileTime.fromMillis(entry.getLastModifiedDate().getTime());
                    Files.setLastModifiedTime(file, fileTime);
                }
            }
        } finally {
            ZipFile.closeQuietly(zipFile);
        }
    }
}