package com.lingtu.poixls;
import java.io.BufferedInputStream;   
import java.io.BufferedOutputStream;   
import java.io.File;   
import java.io.FileInputStream;   
import java.io.FileOutputStream;   
import java.io.IOException;   
import java.io.InputStream;   
import java.io.OutputStream;   
import java.util.ArrayList;   
import java.util.Enumeration;   
import java.util.List;   
import java.util.zip.ZipEntry;   
import java.util.zip.ZipFile;   
import java.util.zip.ZipOutputStream;   

public class ZipToFile {
	
	public static final String ZIP_FILENAME	= "d:\\a.zip";//需要解壓縮的文件名
	public static final String ZIP_DIR = "d:\\a\\";//需要壓縮的文件夾
	public static final String UN_ZIP_DIR = "d:\\";//要解壓的文件目錄
	public static final int BUFFER = 1024 ;//緩存大小
	public static void main(String[] args) {
		ZipToFile ztf=new ZipToFile();
		try{
		ztf.zipFile("d://a","681.png");
		}catch(Exception ex){
			ex.printStackTrace();
		}
	}
	
	/**
	* zip壓縮功能.
	* 壓縮baseDir(文件夾目錄)下所有文件，包括子目錄
	* @throws Exception
	*/
	public static void zipFile(String baseDir,String fileName) throws Exception{
		List fileList=getSubFiles(new File(baseDir));
		ZipOutputStream zos=new ZipOutputStream(new FileOutputStream(fileName));
		ZipEntry ze=null;
		byte[] buf=new byte[BUFFER];
		int readLen=0;
		for(int i = 0; i <fileList.size(); i++) {
			File f=(File)fileList.get(i);
			ze=new ZipEntry(getAbsFileName(baseDir, f));
			ze.setSize(f.length());
			ze.setTime(f.lastModified());	
			zos.putNextEntry(ze);
			InputStream is=new BufferedInputStream(new FileInputStream(f));
			while ((readLen=is.read(buf, 0, BUFFER))!=-1) {
				zos.write(buf, 0, readLen);
			}
			is.close();
		}
		zos.close();
	}
	
	/**
	* 給定根目錄，返回另一個文件名的相對路徑，用于zip文件中的路徑.
	* @param baseDir java.lang.String 根目錄
	* @param realFileName java.io.File 實際的文件名
	* @return 相對文件名
	*/
	private static String getAbsFileName(String baseDir, File realFileName){
		File real=realFileName;
		File base=new File(baseDir);
		String ret=real.getName();
		while (true) {
			real=real.getParentFile();
			if(real==null) 
				break;
			if(real.equals(base)) 
				break;
			else
				ret=real.getName()+"/"+ret;
		}
		return ret;
	}
	
	/**
	* 取得指定目錄下的所有文件列表，包括子目錄.
	* @param baseDir File 指定的目錄
	* @return 包含java.io.File的List
	*/
	private static List getSubFiles(File baseDir){
		List ret=new ArrayList();
		File[] tmp=baseDir.listFiles();
		for (int i = 0; i <tmp.length; i++) {
			if(tmp[i].isFile())
				ret.add(tmp[i]);
			if(tmp[i].isDirectory())
				ret.addAll(getSubFiles(tmp[i]));
		}
		return ret;
	}
	
	/**
	* 解壓縮功能.
	* 將ZIP_FILENAME文件解壓到ZIP_DIR目錄下.
	* @throws Exception
	*/
	public static void upZipFile() throws Exception{
		ZipFile zfile=new ZipFile(ZIP_FILENAME);
		Enumeration zList=zfile.entries();
		ZipEntry ze=null;
		byte[] buf=new byte[1024];
		while(zList.hasMoreElements()){
			ze=(ZipEntry)zList.nextElement();		
			if(ze.isDirectory()){
	            File f=new File(ZIP_DIR+ze.getName());
	            f.mkdir();
				continue;
			}
			OutputStream os=new BufferedOutputStream(new FileOutputStream(getRealFileName(ZIP_DIR, ze.getName())));
			InputStream is=new BufferedInputStream(zfile.getInputStream(ze));
			int readLen=0;
			while ((readLen=is.read(buf, 0, 1024))!=-1) {
				os.write(buf, 0, readLen);
			}
			is.close();
			os.close();	
		}
		zfile.close();
	}

	/**
	* 給定根目錄，返回一個相對路徑所對應的實際文件名.
	* @param baseDir 指定根目錄
	* @param absFileName 相對路徑名，來自于ZipEntry中的name
	* @return java.io.File 實際的文件
	*/
	public static File getRealFileName(String baseDir, String absFileName){
		String[] dirs=absFileName.split("/");
		File ret=new File(baseDir);
		if(dirs.length>1){
			for (int i = 0; i < dirs.length-1;i++) {
				ret=new File(ret, dirs[i]);
			}
			if(!ret.exists())
				ret.mkdirs();
			ret=new File(ret, dirs[dirs.length-1]);
			return ret;
		}
		return ret;
	}

}
/*

不過上面存在中文問題 
是因為java.util.zip下的格式轉換有問題 
將 
import java.util.zip.ZipEntry; 
import java.util.zip.ZipFile; 
import java.util.zip.ZipOutputStream; 
改成 
import org.apache.tools.zip.*; 
即可！ 
apache里修正了這一現象。 
*/