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.util.Enumeration;

import javax.swing.JOptionPane;

import jp.hishidama.zip.ZipEntry;
import jp.hishidama.zip.ZipOutputStream;
import jp.hishidama.zip.ZipFile;
import jp.hishidama.zip.ZipPasswordException;

//*****************************************************************
//*** このクラスにimportされているZIPは、jp.hishidama.zipである ***
//*****************************************************************
public class ArchivePasswordTestPassZip {
	//
	ArchivePasswordTest panel;
	File folder;
	//------------------------------------------------------------------コンストラクタ
	//
	public ArchivePasswordTestPassZip(ArchivePasswordTest parent){
		panel=parent;
	}
	//------------------------------------------------------------------ZIP情報取得
	//
	public String outputZipInfo(File file){
		StringBuffer buf=new StringBuffer();
		//
		try{
			buf.append(outputZipFileEntries(new ZipFile(file))+"\n");
		}catch(IOException e){
			return e.toString();
		}
		//
		return buf.toString();
	}
	//格納ファイルの個々の情報
	private String outputZipFileEntries(ZipFile zipFile){
		StringBuffer buf=new StringBuffer();
		//
		Enumeration enm=zipFile.entries();
		buf.append("*** ZIP File Entries (by jp.hishidama.zip) ***"+"\n");
		//
		while(enm.hasMoreElements()){
			buf.append(((ZipEntry)enm.nextElement()).getName()+"\n");
		}
		//
		return buf.toString();
	}
	//------------------------------------------------------------------ZIP圧縮（パスワード付きに対応）
	//
	public void filesToPasswordZip(File fileZip,File baseFolder,String password){
		//
		folder=baseFolder;
		//
		try{
			ZipOutputStream zos=new ZipOutputStream(new FileOutputStream(fileZip));
			if(!password.equals("")){	//パスワードがあれば設定
				zos.setPassword(password.getBytes("MS932"));
			}
			//
			filesToZip(folder,zos);
			//
			zos.close();
			//
			JOptionPane.showMessageDialog(panel,"圧縮しました(by jp.hishidama.zip)","完了",JOptionPane.INFORMATION_MESSAGE);
		} catch (IOException ioe) {
			JOptionPane.showMessageDialog(panel,ioe.toString(),"エラー",JOptionPane.ERROR_MESSAGE);
		}
	}
	//指定されたフォルダ以下にある全ファイルをZIPに書き込む（再帰呼び出し）
	private void filesToZip(File outFolder,ZipOutputStream zos) throws IOException{
		String[] outFilesList=outFolder.list();
		//
		File outFile;
		for(int f=0;f<outFilesList.length;f++){
			outFile=new File(outFolder.getPath()+File.separator+outFilesList[f]);	//絶対パス
			//
			if(outFile.isFile()){				//ファイルなら書き込む
				fileToZipEntry(outFile,zos);
			}else if(outFile.isDirectory()){	//フォルダなら再帰呼び出し
				filesToZip(outFile,zos);
			}
		}
	}
	//指定されたファイルをZIPに書き込む
	private void fileToZipEntry(File outFile,ZipOutputStream zos) throws IOException{
		BufferedInputStream bis=new BufferedInputStream(new FileInputStream(outFile));
		//エントリーを追加（絶対パスを相対パスに修正）
		ZipEntry outEntry=new ZipEntry(outFile.getPath().substring(folder.getPath().length()+1));	//+1はセパレーターの分
		zos.putNextEntry(outEntry);
		//
		int b;
		while((b=bis.read())!=-1){
			zos.write((byte)b);
		}
		//
		zos.closeEntry();
		bis.close();
	}
	//------------------------------------------------------------------ZIP解凍（パスワード付きに対応）
	//
	public void passwordZipToFolder(File fileZip,File baseFolder,String password){
		//
		folder=baseFolder;
		//
		try {
			ZipFile zf=new ZipFile(fileZip);
			if(!password.equals("")){	//パスワードがあれば設定
				zf.setPassword(password.getBytes("MS932"));
				zf.setCheckCrc(true);
			}
			//
			Enumeration enm=zf.entries();
			while(enm.hasMoreElements()){
				ZipEntry zipEntry=(ZipEntry)enm.nextElement();
				zipEntryToFile(zf,zipEntry);
			}
			//
			zf.close();
			//
			JOptionPane.showMessageDialog(panel,"解凍しました(by jp.hishidama.zip)","完了",JOptionPane.INFORMATION_MESSAGE);
		}catch(ZipPasswordException e){
			JOptionPane.showMessageDialog(panel,e.toString(),"パスワードエラー",JOptionPane.ERROR_MESSAGE);
		} catch (IOException ioe) {
			JOptionPane.showMessageDialog(panel,ioe.toString(),"エラー",JOptionPane.ERROR_MESSAGE);
		}
	}
	//
	private void zipEntryToFile(ZipFile zipFile,ZipEntry zipEntry) throws IOException{
		File outFile=new File(folder.getPath()+File.separator+zipEntry.getName());	//絶対パス
		//
		if(outFile.isDirectory()){	//ディレクトリならフォルダ作成のみ
			outFile.mkdirs();
		}else{						//ファイルならフォルダ作成（必要なら）してファイル書き出し
			//ファイルが格納されるフォルダがまだ存在しないかもしれないので作成
			File parentFolder=new File(outFile.getParent());
			parentFolder.mkdirs();
			//
			BufferedInputStream bis=new BufferedInputStream(zipFile.getInputStream(zipEntry));
			BufferedOutputStream bos=new BufferedOutputStream(new FileOutputStream(outFile));
			//
			int b;
			while((b=bis.read())!=-1){
				bos.write((byte)b);
			}
			//
			bis.close();
			bos.close();
		}
	}
}