package com.moon.server.helper;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import java.io.*;
import java.nio.charset.Charset;
import java.util.Enumeration;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;

@SuppressWarnings("ALL")
public class ZipHelper {
    private final static int BUFFER_SIZE = 1024;

    private final static byte[] BUFFER = new byte[1024];

    private final static Log log = LogFactory.getLog(ZipHelper.class);

    public static boolean unzip(String filePath, String zipDir) {
        ZipFile zipFile = null;
        try {
            File dir = new File(zipDir);
            if (!dir.exists() || !dir.isDirectory()) {
                dir.mkdirs();
            }

            zipFile = new ZipFile(filePath, Charset.forName("GBK"));
            createDirs(zipFile, zipDir);
            writeFiles(zipFile, zipDir);

            return true;
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return false;
        } finally {
            try {
                if (null != zipFile) {
                    zipFile.close();
                }
            } catch (Exception e) {
                log.error(e.getMessage(), e);
            }
        }
    }

    private static void createDirs(ZipFile zipFile, String zipDir) {
        Enumeration<?> e = zipFile.entries();
        while (e.hasMoreElements()) {
            ZipEntry entry = (ZipEntry) e.nextElement();
            if (entry.isDirectory()) {
                File f = new File(zipDir + File.separator + entry.getName());
                if (!f.exists()) {
                    f.mkdirs();
                }
            }
        }
    }

    private static void writeFiles(ZipFile zipFile, String zipDir) throws IOException {
        Enumeration<?> e = zipFile.entries();
        while (e.hasMoreElements()) {
            ZipEntry entry = (ZipEntry) e.nextElement();
            if (!entry.isDirectory()) {
                writeFile(zipFile, entry, zipDir);
            }
        }
    }

    private static void writeFile(ZipFile zipFile, ZipEntry entry, String zipDir) throws IOException {
        int count;

        BufferedInputStream is = new BufferedInputStream(zipFile.getInputStream(entry));
        FileOutputStream fos = new FileOutputStream(zipDir + File.separator + entry.getName());
        BufferedOutputStream dest = new BufferedOutputStream(fos, BUFFER_SIZE);
        while ((count = is.read(BUFFER, 0, BUFFER_SIZE)) != -1) {
            dest.write(BUFFER, 0, count);
        }

        dest.flush();
        dest.close();
        fos.close();
        is.close();
    }

    public static boolean zip(String zipFile, String sourceDir) {
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        try {
            fos = new FileOutputStream(zipFile);
            zos = new ZipOutputStream(fos);

            File sourceFile = new File(sourceDir);
            compress(sourceFile, zos, sourceFile.getName());

            return true;
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return false;
        } finally {
            try {
                if (null != zos) {
                    zos.close();
                }
                if (null != fos) {
                    fos.close();
                }
            } catch (Exception e) {
                log.error(e.getMessage(), e);
            }
        }
    }

    public static void compress(File sourceFile, ZipOutputStream zos, String name) throws Exception {
        if (sourceFile.isFile()) {
            zos.putNextEntry(new ZipEntry(name));

            int len;
            FileInputStream in = new FileInputStream(sourceFile);
            while ((len = in.read(BUFFER)) != -1) {
                zos.write(BUFFER, 0, len);
            }
            zos.closeEntry();
            in.close();
        } else {
            File[] listFiles = sourceFile.listFiles();
            if (listFiles == null || listFiles.length == 0) {
                zos.putNextEntry(new ZipEntry(name + "/"));
                zos.closeEntry();
            } else {
                for (File file : listFiles) {
                    compress(file, zos, name + "/" + file.getName());
                }
            }
        }
    }
}