import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.Closeable;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;


/**
 *  Zip壓縮解壓縮
 *  @author LuoZq
 */
public final class ZipUtils {
	
	public static final String SUFFIX_NAME = "zip";
	
	/**
	 *  壓縮檔案
	 *  @param zipFile 壓縮檔名路徑
	 *  @param fileList 待壓縮的檔案集合
	 *  @param isDeleteRes 如果出現同名壓縮包, 是否刪除原本的壓縮包, 如果false,且出現同名則拋異常
	 *  @return 壓縮後的壓縮包File
	 */
	public static File compress(String zipFile, List<File> fileList, boolean isDelete){
		File f = isCheckFileNameByZip(zipFile);
		return compress( f.getName(), f.getParent(), getFilePathArray(fileList), false, isDelete );
	}
	/**
	 *  壓縮檔案
	 *  @param zipFile 壓縮檔名路徑
	 *  @param fileList 待壓縮的檔案集合
	 *  @return 壓縮後的壓縮包File
	 */
	public static File compress(String zipFile, List<File> fileList){
		return compress( zipFile, fileList, true );
	}
	/**
	 *  校驗目標檔名稱是否是zip格式
	 *  @param zipFile
	 *  @return
	 */
	private static File isCheckFileNameByZip(String zipFile){
		return new File(zipFile.substring(0, isCheckFileNameInZip(zipFile)));
	}
	/**
	 *  校驗目標檔名稱是否是zip格式
	 *  @param zipFile
	 *  @return
	 */
	private static int isCheckFileNameInZip(String zipFile){
		int las = zipFile.lastIndexOf(".");
		if(las == -1){
			throw new RuntimeException( zipFile + " is not zip format! this format = ??? ");
		}
		String format = zipFile.substring(las + 1);
		if(!SUFFIX_NAME.equalsIgnoreCase(format)){
			throw new RuntimeException( zipFile + " is not zip format! this format = " + format);
		}
		return las;
	}
	private static String[] getFilePathArray (List<File> list){
		String[] strs = new String[list.size()];
		for(int index = 0; index < strs.length; index++){
			strs[index] = list.get(index).getPath();
		}
		return strs;
	}
	/**
	 *  壓縮檔案
	 *  @param zipName 壓縮檔名, 無需加字尾
	 *  @param zipFilePath 壓縮路徑
	 *  @param filePaths 待壓縮的檔案路徑
	 *  @param isNewFolder 是否在壓縮包新建同名資料夾
	 *  @param isDeleteRes 如果出現同名壓縮包, 是否刪除原本的壓縮包, 如果false,且出現同名則拋異常
	 *  @return 壓縮後的壓縮包File
	 */
    private static File compress(String zipName,String zipFilePath, String[] filePaths, boolean isNewFolder, boolean isDeleteRes) {
        File target = null;
        File source = new File(zipFilePath);
        if (source.exists()) {
        	String base = isNewFolder? zipName + File.separator: "";
            zipName = zipName + "." + SUFFIX_NAME;// 壓縮檔名=原始檔名.zip
            target = new File(source.getPath(), zipName);
            if (target.exists()) {
            	if(!isDeleteRes){
            		throw new RuntimeException("Compression package name repetition !");
            	}
            	target.delete(); // 刪除舊的檔案
            }
            FileOutputStream fos = null;
            ZipOutputStream zos = null;
            try {
                fos = new FileOutputStream(target);
                zos = new ZipOutputStream(new BufferedOutputStream(fos));
                // 新增對應的檔案Entry
                for(String fip:filePaths){
                	addEntry(base, new File(fip), zos);
                }
            } catch (IOException e) {
                throw new RuntimeException(e);
            } finally {
                closeQuietly(zos, fos);
            }
        }
        return target;
    }
	
    /**
     *  解壓檔案
     *  filePath所代表檔案系統不能與targetStr一致
     *  @param filePath 壓縮檔案路徑
     *  @param targetStr 解壓至所在檔案目錄
     */
    public static void decompression(String filePath, String targetStr) {
    	isCheckFileNameInZip(filePath);
        File source = new File(filePath);
        if (source.exists()) {
            ZipInputStream zis = null;
            BufferedOutputStream bos = null;
            try {
                zis = new ZipInputStream(new FileInputStream(source));
                ZipEntry entry = null;
                while ((entry = zis.getNextEntry()) != null && !entry.isDirectory()) {
                    File target = new File(targetStr, entry.getName());
                    if (!target.getParentFile().exists()) {
                    	target.getParentFile().mkdirs();// 建立檔案父目錄
                    }
                    // 寫入檔案
                    bos = new BufferedOutputStream(new FileOutputStream(target));
                    int read = 0;
                    byte[] buffer = new byte[1024 * 10];
                    while ((read = zis.read(buffer, 0, buffer.length)) != -1) {
                        bos.write(buffer, 0, read);
                    }
                    bos.flush();
                }
                zis.closeEntry();
            } catch (Exception e) {
                throw new RuntimeException(e);
            } finally {
                closeQuietly(zis, bos);
            }
        }
    }
    
	/**
     *  掃描新增檔案Entry
     *  @param base 基路徑
     *  @param source 原始檔
     *  @param zos Zip檔案輸出流
     *  @throws IOException
     */
    private static void addEntry(String base, File source, ZipOutputStream zos) throws IOException {
        // 按目錄分級，形如：/aaa/bbb.txt
        String entry = base.concat(source.getName());
        if (source.isDirectory()) {
            for (File file : source.listFiles()) {
                // 遞迴列出目錄下的所有檔案，新增檔案Entry
                addEntry(entry + File.separator, file, zos);
            }
        } else {
            FileInputStream fis = null;
            BufferedInputStream bis = null;
            try {
                byte[] buffer = new byte[1024 * 10];
                fis = new FileInputStream(source);
                bis = new BufferedInputStream(fis, buffer.length);
                int read = 0;
                zos.putNextEntry(new ZipEntry(entry));
                while ((read = bis.read(buffer, 0, buffer.length)) != -1) {
                    zos.write(buffer, 0, read);
                }
                zos.closeEntry();
            } finally {
                closeQuietly(bis, fis);
            }
        }
    }
	
	private ZipUtils(){
		
	}
	
	/**
     * 關閉一個或多個流物件
     * 
     * @param closeables
     *            可關閉的流物件列表
     */
    public static void closeQuietly(Closeable... closeables) {
        try {
        	if (closeables != null && closeables.length > 0) {
                for (Closeable closeable : closeables) {
                    if (closeable != null) {
                        closeable.close();
                    }
                }
            }
        } catch (IOException e) {
            // do nothing
        }
    }
    
    public static void main(String[] args) {
    	String res = "G://test";
    	List<File> list = Arrays.asList( new File(res).listFiles() );
    	compress("G://testOne.zip", list);
    	decompression("G://testOne.zip", "G://testOne1.zip");
	}
}