package com.zhl.push.Utils;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.OutputStream;

import java.util.zip.ZipEntry;

import java.util.zip.ZipOutputStream;

/**

 * @Author

 * @ClassName ToZIPUtil 压缩文件

 * @Description TODO

 * @Date 2019/3/25 17:39

 * @Version 1.0

 */

public class ToZIPUtil {

    private static final int  BUFFER_SIZE = 2 * 1024;

    /**

     23

     * 压缩成ZIP

     24

     * @param srcDir 压缩文件夹路径

    25

     * @param out    压缩文件输出流

    26

     * @param KeepDirStructure  是否保留原来的目录结构,true:保留目录结构;

    27

     *                          false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)

    28

     * @throws RuntimeException 压缩失败会抛出运行时异常

    29

     */

    public static void toZip(String srcDir, OutputStream out, boolean KeepDirStructure)

            throws RuntimeException{

        long start = System.currentTimeMillis();

        ZipOutputStream zos = null ;

        try {

            zos = new ZipOutputStream(out);

            File sourceFile = new File(srcDir);

            compress(sourceFile,zos,sourceFile.getName(),KeepDirStructure);

            long end = System.currentTimeMillis();

            System.out.println("压缩完成，耗时：" + (end - start) +" ms");

        } catch (Exception e) {

            throw new RuntimeException("zip error from ZipUtils",e);

        }finally{

            if(zos != null) {

                try {

                    zos.close();

                } catch (IOException e) {

                    e.printStackTrace();

                }

            }

        }

    }

    /**

     94

     * 递归压缩方法

     95

     * @param sourceFile 源文件

    96

     * @param zos        zip输出流

    97

     * @param name       压缩后的名称

    98

     * @param KeepDirStructure  是否保留原来的目录结构,true:保留目录结构;

    99

     *                          false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)

    100

     * @throws Exception

    101

     */

    private static void compress(File sourceFile, ZipOutputStream zos, String name,

                                 boolean KeepDirStructure) throws Exception{

        byte[] buf = new byte[BUFFER_SIZE];

        if(sourceFile.isFile()){

            // 向zip输出流中添加一个zip实体，构造器中name为zip实体的文件的名字

            zos.putNextEntry(new ZipEntry(name));

            // copy文件到zip输出流中

            int len;

            FileInputStream in = new FileInputStream(sourceFile);

            while ((len = in.read(buf)) != -1){

                zos.write(buf, 0, len);

            }

            // Complete the entry

            zos.closeEntry();

            in.close();

        } else {

            File[] listFiles = sourceFile.listFiles();

            if(listFiles == null || listFiles.length == 0){

                // 需要保留原来的文件结构时,需要对空文件夹进行处理

                if(KeepDirStructure){

                    // 空文件夹的处理

                    zos.putNextEntry(new ZipEntry(name + "/"));

                    // 没有文件，不需要文件的copy

                    zos.closeEntry();

                }

            }else {

                for (File file : listFiles) {

                    // 判断是否需要保留原来的文件结构

                    if (KeepDirStructure) {

                        // 注意：file.getName()前面需要带上父文件夹的名字加一斜杠,

                        // 不然最后压缩包中就不能保留原来的文件结构,即：所有文件都跑到压缩包根目录下了

                        compress(file, zos, name + "/" + file.getName(),KeepDirStructure);

                    } else {

                        compress(file, zos, file.getName(),KeepDirStructure);

                    }

                }

            }

        }

    }

}

 /*

     **author:weijiakun

     *获取目录工具类

     */

    public static String  getPath(String subdirectory){

        //获取跟目录---与jar包同级目录的upload目录下指定的子目录subdirectory

        File upload = null;

        try {

            //本地测试时获取到的是"工程目录/target/upload/subdirectory

            File path = new File(ResourceUtils.getURL("classpath:").getPath());

            if(!path.exists())

                path = new File("");

            upload = new File(path.getAbsolutePath(),subdirectory);

            if(!upload.exists()) upload.mkdirs();//如果不存在则创建目录

            String realPath = upload + "/";

            return realPath;

        } catch (FileNotFoundException e) {

            throw new RuntimeException("获取服务器路径发生错误！");

        }

    }

    //创建临时文件夹

    private String createFile(String filename){

        File file =new File(filename);

        //如果文件夹不存在则创建

        if (!file .exists() && !file .isDirectory())

        {

            file .mkdir();

        }

        return file.getAbsolutePath();

    }

//删除文件

    private void deleteFile(File file) {

        if (file.exists()) {//判断文件是否存在

            if (file.isFile()) {//判断是否是文件

                file.delete();//删除文件

            } else if (file.isDirectory()) {//否则如果它是一个目录

                File[] files = file.listFiles();//声明目录下所有的文件 files[];

                for (int i = 0; i < files.length; i++) {//遍历目录下所有的文件

                    this.deleteFile(files[i]);//把每个文件用这个方法进行迭代

                }

                file.delete();//删除文件夹

            }

        } else {

            System.out.println("所删除的文件不存在");

        }

    }

    response.setContentType("multipart/form-data");

    response.setCharacterEncoding("utf-8");

    response.addHeader("Content-Disposition", "filename=" + fileName);