package com.tax.core.util; 

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.OutputStream;

import java.util.ArrayList;

import java.util.List;

import java.util.zip.ZipEntry;

import java.util.zip.ZipOutputStream;

/**

 * ZipUtils

 * @author  ZENG.XIAO.YAN

 * @date    2017年11月19日 下午7:16:08

 * @version v1.0

 */

public class ZipUtils {

    private static final int  BUFFER_SIZE = 2 * 1024;

    /**

     * 壓縮成ZIP 方法1

     * @param srcDir 壓縮文件夾路徑 

     * @param out    壓縮文件輸出流

     * @param KeepDirStructure  是否保留原來的目錄結(jié)構(gòu),true:保留目錄結(jié)構(gòu); 

     *                          false:所有文件跑到壓縮包根目錄下(注意：不保留目錄結(jié)構(gòu)可能會(huì)出現(xiàn)同名文件,會(huì)壓縮失敗)

     * @throws RuntimeException 壓縮失敗會(huì)拋出運(yùn)行時(shí)異常

     */

    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("壓縮完成，耗時(shí)：" + (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();

                }

            }

        }

    }

    /**

     * 壓縮成ZIP 方法2

     * @param srcFiles 需要壓縮的文件列表

     * @param out           壓縮文件輸出流

     * @throws RuntimeException 壓縮失敗會(huì)拋出運(yùn)行時(shí)異常

     */

    public static void toZip(List<File> srcFiles , OutputStream out)throws RuntimeException {

        long start = System.currentTimeMillis();

        ZipOutputStream zos = null ;

        try {

            zos = new ZipOutputStream(out);

            for (File srcFile : srcFiles) {

                byte[] buf = new byte[BUFFER_SIZE];

                zos.putNextEntry(new ZipEntry(srcFile.getName()));

                int len;

                FileInputStream in = new FileInputStream(srcFile);

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

                    zos.write(buf, 0, len);

                }

                zos.closeEntry();

                in.close();

            }

            long end = System.currentTimeMillis();

            System.out.println("壓縮完成，耗時(shí)：" + (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();

                }

            }

        }

    }

    /**

     * 遞歸壓縮方法

     * @param sourceFile 源文件

     * @param zos        zip輸出流

     * @param name       壓縮后的名稱

     * @param KeepDirStructure  是否保留原來的目錄結(jié)構(gòu),true:保留目錄結(jié)構(gòu); 

     *                          false:所有文件跑到壓縮包根目錄下(注意：不保留目錄結(jié)構(gòu)可能會(huì)出現(xiàn)同名文件,會(huì)壓縮失敗)

     * @throws Exception

     */

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

            boolean KeepDirStructure) throws Exception{

        byte[] buf = new byte[BUFFER_SIZE];

        if(sourceFile.isFile()){

            // 向zip輸出流中添加一個(gè)zip實(shí)體，構(gòu)造器中name為zip實(shí)體的文件的名字

            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){

                // 需要保留原來的文件結(jié)構(gòu)時(shí),需要對空文件夾進(jìn)行處理

                if(KeepDirStructure){

                    // 空文件夾的處理

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

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

                    zos.closeEntry();

                }

            }else {

                for (File file : listFiles) {

                    // 判斷是否需要保留原來的文件結(jié)構(gòu)

                    if (KeepDirStructure) {

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

                        // 不然最后壓縮包中就不能保留原來的文件結(jié)構(gòu),即：所有文件都跑到壓縮包根目錄下了

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

                    } else {

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

                    }

                }

            }

        }

    }

    public static void main(String[] args) throws Exception {

        /** 測試壓縮方法1  */

        FileOutputStream fos1 = new FileOutputStream(new File("c:/mytest01.zip"));

        ZipUtils.toZip("D:/log", fos1,true);

        /** 測試壓縮方法2  */

        List<File> fileList = new ArrayList<>();

        fileList.add(new File("D:/Java/jdk1.7.0_45_64bit/bin/jar.exe"));

        fileList.add(new File("D:/Java/jdk1.7.0_45_64bit/bin/java.exe"));

        FileOutputStream fos2 = new FileOutputStream(new File("c:/mytest02.zip"));

        ZipUtils.toZip(fileList, fos2);

    }

}

    if(userList.size() > 0){

        /** 下面為下載zip壓縮包相關(guān)流程 */

        HttpServletRequest request = ServletActionContext.getRequest();

        FileWriter writer;

        /** 1.創(chuàng)建臨時(shí)文件夾  */

        String rootPath = request.getSession().getServletContext().getRealPath("/");

        File temDir = new File(rootPath + "/" + UUID.randomUUID().toString().replaceAll("-", ""));

        if(!temDir.exists()){

            temDir.mkdirs();

        }

        /** 2.生成需要下載的文件，存放在臨時(shí)文件夾內(nèi) */

        // 這里我們直接來10個(gè)內(nèi)容相同的文件為例，但這個(gè)10個(gè)文件名不可以相同

        for (int i = 0; i < 10; i++) {

            dataMap.put("userList", userList);

            Map<String, String> endMap = new HashMap<>();

            endMap.put("user", "老王");

            endMap.put("time", "2017-10-10 10:50:55");

            dataMap.put("endMap", endMap);

            Configuration cfg = new Configuration(Configuration.VERSION_2_3_22);

            cfg.setServletContextForTemplateLoading(ServletActionContext.getServletContext(), "/ftl");

            Template template = cfg.getTemplate("exportExcel.ftl");

            writer = new FileWriter(temDir.getPath()+"/excel"+ i +".xls");

            template.process(dataMap, writer);

            writer.flush();

            writer.close();

        }

        /** 3.設(shè)置response的header */

        HttpServletResponse response = ServletActionContext.getResponse();

        response.setContentType("application/zip");

        response.setHeader("Content-Disposition", "attachment; filename=excel.zip");  

        /** 4.調(diào)用工具類，下載zip壓縮包 */

        // 這里我們不需要保留目錄結(jié)構(gòu)

        ZipUtils.toZip(temDir.getPath(), response.getOutputStream(),false);

        /** 5.刪除臨時(shí)文件和文件夾 */

        // 這里我沒寫遞歸，直接就這樣刪除了

        File[] listFiles = temDir.listFiles();

        for (int i = 0; i < listFiles.length; i++) {

            listFiles[i].delete();

        }

        temDir.delete();

    }