package javazip.util;

import java.util.ArrayList;
import java.io.*;
import java.util.zip.*;
import javazip.base.*;
import javazip.base.FileInfo;

/**
 * 實現gz的壓縮與解壓縮 目前只找到了壓縮一個文件
 * 
 * @author dragon
 * 
 */
public class GzipUtil extends CompressUtil {

	/**
	 * Gzip壓縮文件的方法 注意:只能壓縮一個文件 壓縮文件的文件名為:原文件名+.gz
	 * 
	 * @see javazip.util.CompressUtil#compressFile(java.util.ArrayList,
	 *      java.lang.String)
	 */
	public void compressFile(ArrayList fileList, String target) {
		try {
			File f = (File) fileList.get(0);
			/* 打開需壓縮文件作為文件輸入流 */
			FileInputStream fin = new FileInputStream(f);
			/* 建立壓縮文件輸出流 */
			target = generatePath(target, f);
			FileOutputStream fout = new FileOutputStream(target);
			/* 建立gzip壓縮輸出流 */
			GZIPOutputStream gzout = new GZIPOutputStream(fout);
			byte[] buf = new byte[1024];/* 設定讀入緩沖區尺寸 */
			int num;
			while ((num = fin.read(buf)) != -1) {
				gzout.write(buf, 0, num);
			}
			gzout.close();/* !!!關閉流,必須關閉所有輸入輸出流.保證輸入輸出完整和釋放系統資源. */
			fout.close();
			fin.close();
		} catch (IOException e) {
			System.out.println(e);
		}
	}
	/**
	 * 
	 */
	public void compressFile(File fileList[], String target)
	{
		try {
			File f = fileList[0];
			/* 打開需壓縮文件作為文件輸入流 */
			FileInputStream fin = new FileInputStream(f);
			/* 建立壓縮文件輸出流 */
			target = generatePath(target, f);
			FileOutputStream fout = new FileOutputStream(target);
			/* 建立gzip壓縮輸出流 */
			GZIPOutputStream gzout = new GZIPOutputStream(fout);
			byte[] buf = new byte[1024];/* 設定讀入緩沖區尺寸 */
			int num;
			while ((num = fin.read(buf)) != -1) {
				gzout.write(buf, 0, num);
			}
			gzout.close();/* !!!關閉流,必須關閉所有輸入輸出流.保證輸入輸出完整和釋放系統資源. */
			fout.close();
			fin.close();
		} catch (IOException e) {
			System.out.println(e);
		}
	}
	/**
	 * Gzip解壓縮文件的方法
	 * 
	 * @see javazip.util.CompressUtil#extractFile(java.lang.String,
	 *      java.lang.String)
	 */
	public void extractFile(String source, String target) {
		try {
			/* 建立gzip壓縮文件輸入流 */
			FileInputStream fin = new FileInputStream(source);
			/* 建立gzip解壓工作流 */
			GZIPInputStream gzin = new GZIPInputStream(fin);
			target = getFileName(target, new File(source));
			/* 建立解壓文件輸出流 */
			FileOutputStream fout = new FileOutputStream(target);
			byte[] buf = new byte[1024];
			int num;
			while ((num = gzin.read(buf, 0, buf.length)) != -1) {
				fout.write(buf, 0, num);
			}
			gzin.close();
			fout.close();
			fin.close();
		} catch (IOException e) {
			System.out.println(e);
		}
	}

	/**
	 * 生成保存文件的路徑
	 * 
	 * @param target
	 * @param source
	 * @return
	 */
	private String generatePath(String target, File source) {
		File temp = new File(target);
		target = temp.getParent();
		target = target + source.getName() + ".gz";
		System.out.println(target);
		return target;
	}

	/**
	 * 取得要解壓縮的文件名
	 * 
	 * @return
	 */
	private String getFileName(String target, File source) {
		File t = new File(target);
		if (!t.isDirectory()) {
			target = t.getParent();
		}
		target = target
				+ source.getName().substring(0, source.getName().length() - 3);
		System.out.println(target);
		return target;
	}
	/**
	 * 
	 */
	public Long getOriginalLength(File f)
	{
		return null;
	}
	
	/**
	 * 
	 */
	public boolean testFile(File f)
	{
		return true;
	}
	/**
	 * @param args
	 */
	public static void main(String[] args) {
		GzipUtil gzip = new GzipUtil();
		ArrayList a = new ArrayList();
		a.add(new File("F:\\sevenzip.exe"));
		gzip.compressFile(a, "D:\\sevenzip.exe.gz");
		gzip.extractFile("D:\\sevenzip.exe.gz", "D:\\");
	}

}