import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class Zip {

    public static void main(String[] args) {
        // 將要被壓縮的文件夾路徑  
        String folderPath = "D:\Logback\yiteng\ctnNO"; 
        
        // 壓縮后生成的壓縮文件路徑及文件名
        String zipFilePath = "D:\Logback\yiteng\ctnNO.zip";
        
        try {
            zipFolder(folderPath, zipFilePath);
            System.out.println("Zip file created successfully.");
        } catch (IOException e) {
            System.out.println("Error creating zip file: " + e.getMessage());
        }
    }

    /**
     * 打包壓縮文件夾
     *
     * @param folderPath 文件夾路徑
     * @param zipFilePath 壓縮后的文件路徑
     * @throws IOException IO異常
     */
    public static void zipFolder(String folderPath, String zipFilePath) throws IOException {
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        try {
            fos = new FileOutputStream(zipFilePath);
            zos = new ZipOutputStream(fos);
            
            // 遞歸遍歷整個文件夾并添加到壓縮包
            addFolderToZip("", new File(folderPath), zos);
        } finally {
            if (zos != null) {
                zos.close();
            }
            if (fos != null) {
                fos.close();
            }
        }
    }

    /**
     * 將文件夾及其中的文件遞歸添加到壓縮流中
     *
     * @param parentPath 父級路徑
     * @param folder 文件夾
     * @param zos Zip輸出流
     * @throws FileNotFoundException 文件未找到異常
     * @throws IOException IO異常
     */
    private static void addFolderToZip(String parentPath, File folder, ZipOutputStream zos) throws FileNotFoundException, IOException {
        for (File file : folder.listFiles()) {
            if (file.isDirectory()) {
                // 遞歸添加子文件夾中的文件
                addFolderToZip(parentPath + folder.getName() + "/", file, zos);
            } else {
                FileInputStream fis = null;
                try {
                    fis = new FileInputStream(file);
                    
                    // 新建Zip條目并將輸入流加入到Zip包中
                    ZipEntry zipEntry = new ZipEntry(parentPath + folder.getName() + "/" + file.getName());
                    zos.putNextEntry(zipEntry);
                    
                    byte[] bytes = new byte[1024];
                    int length;
                    while ((length = fis.read(bytes)) >= 0) {
                        zos.write(bytes, 0, length);
                    }
                } finally {
                    if (fis != null) {
                        fis.close();
                    }
                }
            }
        }
    }
}