import java.io.*; 
import org.apache.tools.zip.*; 
import java.util.Enumeration; 
/** 
*功能:zip壓縮、解壓(支援中文檔名) 
*說明:本程式通過使用Apache Ant裡提供的zip工具org.apache.tools.zip實現了zip壓縮和解壓功能. 
*   解決了由於java.util.zip包不支援漢字的問題。 
*   使用java.util.zip包時,當zip檔案中有名字為中文的檔案時, 
*   就會出現異常:"Exception  in thread "main " java.lang.IllegalArgumentException  
*               at   java.util.zip.ZipInputStream.getUTF8String(ZipInputStream.java:285) 
*注意: 
*   1、使用時把ant.jar放到classpath中,程式中使用import org.apache.tools.zip.*; 
*   2、Apache Ant 下載地址:[url]http://ant.apache.org/[/url] 
*   3、Ant ZIP API:[url]http://www.jajakarta.org/ant/ant-1.6.1/docs/mix/manual/api/org/apache/tools/zip/[/url] 
*   4、本程式使用Ant 1.7.1 中的ant.jar 

*僅供程式設計學習參考. 

*@author Winty 
*@date   2008-8-3 
*@Usage: 
*   壓縮:java AntZip -zip "directoryName" 
*   解壓:java AntZip -unzip "fileName.zip" 
*/
 

public class AntZip{ 
    private ZipFile         zipFile; 
    private ZipOutputStream zipOut;     //壓縮Zip 
    private ZipEntry        zipEntry; 
    private static int      bufSize;    //size of bytes 
    private byte[]          buf; 
    private int             readedBytes; 
     
    public AntZip(){ 
        this(512); 
    } 

    public AntZip(int bufSize){ 
        this.bufSize = bufSize; 
        this.buf = new byte[this.bufSize]; 
    } 
     
    //壓縮資料夾內的檔案 
    public void doZip(String zipDirectory){//zipDirectoryPath:需要壓縮的資料夾名 
        File file; 
        File zipDir; 

        zipDir = new File(zipDirectory); 
        String zipFileName = zipDir.getName() + ".zip";//壓縮後生成的zip檔名 

        try
            this.zipOut = new ZipOutputStream(new BufferedOutputStream(new FileOutputStream(zipFileName))); 
            handleDir(zipDir , this.zipOut); 
            this.zipOut.close(); 
        }catch(IOException ioe){ 
            ioe.printStackTrace(); 
        } 
    } 

    //由doZip呼叫,遞迴完成目錄檔案讀取 
    private void handleDir(File dir , ZipOutputStream zipOut)throws IOException{ 
        FileInputStream fileIn; 
        File[] files; 

        files = dir.listFiles(); 
     
        if(files.length == 0){//如果目錄為空,則單獨建立之. 
            //ZipEntry的isDirectory()方法中,目錄以"/"結尾. 
            this.zipOut.putNextEntry(new ZipEntry(dir.toString() + "/")); 
            this.zipOut.closeEntry(); 
        } 
        else{//如果目錄不為空,則分別處理目錄和檔案. 
            for(File fileName : files){ 
                //System.out.println(fileName); 

                if(fileName.isDirectory()){ 
                    handleDir(fileName , this.zipOut); 
                } 
                else
                    fileIn = new FileInputStream(fileName); 
                    this.zipOut.putNextEntry(new ZipEntry(fileName.toString())); 

                    while((this.readedBytes = fileIn.read(this.buf))>0){ 
                        this.zipOut.write(this.buf , 0 , this.readedBytes); 
                    } 

                    this.zipOut.closeEntry(); 
                } 
            } 
        } 
    } 

    //解壓指定zip檔案 
    public void unZip(String unZipfileName){//unZipfileName需要解壓的zip檔名 
        FileOutputStream fileOut; 
        File file; 
        InputStream inputStream; 

        try
            this.zipFile = new ZipFile(unZipfileName); 

            for(Enumeration entries = this.zipFile.getEntries(); entries.hasMoreElements();){ 
                ZipEntry entry = (ZipEntry)entries.nextElement(); 
                file = new File(entry.getName()); 

                if(entry.isDirectory()){ 
                    file.mkdirs(); 
                } 
                else
                    //如果指定檔案的目錄不存在,則建立之. 
                    File parent = file.getParentFile(); 
                    if(!parent.exists()){ 
                        parent.mkdirs(); 
                    } 

                    inputStream = zipFile.getInputStream(entry); 

                    fileOut = new FileOutputStream(file); 
                    while(( this.readedBytes = inputStream.read(this.buf) ) > 0){ 
                        fileOut.write(this.buf , 0 , this.readedBytes ); 
                    } 
                    fileOut.close(); 

                    inputStream.close(); 
                }    
            } 
            this.zipFile.close(); 
        }catch(IOException ioe){ 
            ioe.printStackTrace(); 
        } 
    } 

    //設定緩衝區大小 
    public void setBufSize(int bufSize){ 
        this.bufSize = bufSize; 
    } 

    //測試AntZip類 
    public static void main(String[] args)throws Exception{ 
        if(args.length==2){ 
            String name = args[1]; 
            AntZip zip = new AntZip(); 

            if(args[0].equals("-zip")) 
                zip.doZip(name); 
            else if(args[0].equals("-unzip")) 
                zip.unZip(name); 
        } 
        else
            System.out.println("Usage:"); 
            System.out.println("壓縮:java AntZip -zip directoryName"); 
            System.out.println("解壓:java AntZip -unzip fileName.zip"); 
            throw new Exception("Arguments error!"); 
        } 
    } 
}