123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109 |
- /*
- * Copyright (C) 2012-present the original author or authors.
- *
- * 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.pf4j.util;
-
- import java.io.File;
- import java.io.FileInputStream;
- import java.io.FileNotFoundException;
- import java.io.FileOutputStream;
- import java.io.IOException;
- import java.util.zip.ZipEntry;
- import java.util.zip.ZipInputStream;
-
- import org.slf4j.Logger;
- import org.slf4j.LoggerFactory;
-
- /**
- * This class extracts the content of the plugin zip into a directory.
- * It's a class for only the internal use.
- *
- * @author Decebal Suiu
- */
- public class Unzip {
-
- private static final Logger log = LoggerFactory.getLogger(Unzip.class);
-
- /**
- * Holds the destination directory.
- * File will be unzipped into the destination directory.
- */
- private File destination;
-
- /**
- * Holds path to zip file.
- */
- private File source;
-
- public Unzip() {
- }
-
- public Unzip(File source, File destination) {
- this.source = source;
- this.destination = destination;
- }
-
- public void setSource(File source) {
- this.source = source;
- }
-
- public void setDestination(File destination) {
- this.destination = destination;
- }
-
- /**
- * Extract the content of zip file ({@code source}) to destination directory.
- * If destination directory already exists it will be deleted before.
- */
- public void extract() throws IOException {
- log.debug("Extract content of '{}' to '{}'", source, destination);
-
- // delete destination directory if exists
- if (destination.exists() && destination.isDirectory()) {
- FileUtils.delete(destination.toPath());
- }
-
- try (ZipInputStream zipInputStream = new ZipInputStream(new FileInputStream(source))) {
- ZipEntry zipEntry;
- while ((zipEntry = zipInputStream.getNextEntry()) != null) {
- File file = new File(destination, zipEntry.getName());
-
- // create intermediary directories - sometimes zip don't add them
- File dir = new File(file.getParent());
-
- mkdirsOrThrow(dir);
-
- if (zipEntry.isDirectory()) {
- mkdirsOrThrow(file);
- } else {
- byte[] buffer = new byte[1024];
- int length;
- try (FileOutputStream fos = new FileOutputStream(file)) {
- while ((length = zipInputStream.read(buffer)) >= 0) {
- fos.write(buffer, 0, length);
- }
- }
- }
- }
- }
- }
-
- private static void mkdirsOrThrow(File dir) throws IOException {
- if (!dir.exists() && !dir.mkdirs()) {
- throw new IOException("Failed to create directory " + dir);
- }
- }
-
- }
|