package com.chinaweal.aiccs.common.util;


import java.io.*;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
import javax.servlet.http.HttpServletResponse;

import com.chinaweal.aiccs.common.util.filestorage.FileStorageService;
import com.chinaweal.youfool.framework.springboot.util.SpringUtil;
import org.springframework.stereotype.Component;

/**
 * @author Lee
 * @since 2021/6/7 18:02
 */
@Component
public class ZipUtil {

    /**
     * 把文件打成压缩包并输出到客户端浏览器中
     */
    public static void downloadZipFiles(HttpServletResponse response, Map<String, String> fileMap, String zipFileName) {
        try {
            response.reset(); // 重点突出
            response.setCharacterEncoding("UTF-8"); // 重点突出
            response.setContentType("application/x-msdownload"); // 不同类型的文件对应不同的MIME类型 // 重点突出
            // 对文件名进行编码处理中文问题
            zipFileName = new String(zipFileName.getBytes(), StandardCharsets.UTF_8);
            // inline在浏览器中直接显示，不提示用户下载
            // attachment弹出对话框，提示用户进行下载保存本地
            // 默认为inline方式
            response.setHeader("Content-Disposition", "attachment;filename=" + zipFileName);

            // --设置成这样可以不用保存在本地，再输出， 通过response流输出,直接输出到客户端浏览器中。
            ZipOutputStream zos = new ZipOutputStream(response.getOutputStream());
            zipFile(fileMap, zos);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 压缩文件
     *
     * @param fileMap 需要压缩的文件路径集合
     * @throws IOException
     */
    private static void zipFile(Map<String, String> fileMap, ZipOutputStream zos) {
        //设置读取数据缓存大小
        byte[] buffer = new byte[4096];
        try {
            //循环读取文件路径集合，获取每一个文件的路径
            Set<String> filePaths = fileMap.keySet();
            for (String filePath : filePaths) {

                File inputFile = new File(filePath);

                FileStorageService fileStorageService = SpringUtil.getBean(FileStorageService.class);
                fileStorageService.writeToFile(filePath, inputFile);
                //判断文件是否存在
                if (inputFile.exists()) {
                    //判断是否属于文件，还是文件夹
                    if (inputFile.isFile()) {
                        //创建输入流读取文件

                        try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(inputFile))) {
                            //将文件写入zip内，即将文件进行打包
                            zos.putNextEntry(new ZipEntry(fileMap.get(filePath)));
                            //写入文件的方法，同上
                            int size = 0;
                            //设置读取数据缓存大小
                            while ((size = bis.read(buffer)) > 0) {
                                zos.write(buffer, 0, size);
                            }
                            //关闭输入输出流
                            zos.closeEntry();
                            inputFile.delete();
                        }
                    } else {  //如果是文件夹，则使用穷举的方法获取文件，写入zip
                        File[] files = inputFile.listFiles();
                        List<String> filePathsTem = new ArrayList<String>();
                        HashMap<String, String> map = new HashMap<>();
                        for (File fileTem : files) {
                            filePathsTem.add(fileTem.toString());
                            map.put(fileTem.toString(), fileTem.getName());
                        }
                        zipFile(map, zos);
                    }
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (null != zos) {
                try {
                    zos.close();

                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}