/*******************************************************************************
  * Copyright (c) 2009 - 2012 Red Hat, Inc.
  * Distributed under license by Red Hat, Inc. All rights reserved.
  * This program is made available under the terms of the
  * Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  *
  * Contributor:
  *     Red Hat, Inc. - initial API and implementation
  ******************************************************************************/
package org.jboss.tools.common.zip;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

public class UnzipOperation {
	
	File zipFile;
	
	public static final int BUFFER_SIZE = 1024*4;
	
	public UnzipOperation(String zipFile) {
		this(new File(zipFile));
	}
	
	public UnzipOperation(File zipFile) {
		this.zipFile = zipFile;
	}
	
	public void execute(File destination) throws IOException {
		ZipArchive archive = new ZipArchive(zipFile);
		archive.acceptVisitor(new UnzipEntryVisitor(destination));
	}
	
	public void execute(String destination) throws IOException {
		execute(new File(destination));
	}

	public void execute(File destination,String filter) throws IOException {
		ZipArchive archive = new ZipArchive(zipFile);
		archive.acceptVisitor(new FilteredZipEntryVisitor(filter,new UnzipEntryVisitor(destination)));
	}
	
	public void execute(String destination,String filter) throws IOException {
		execute(new File(destination), filter);
	}
	
	public static class FilteredZipEntryVisitor implements IZipEntryVisitor{

		private String filter;
		private IZipEntryVisitor visitor;

		public FilteredZipEntryVisitor(String filter, IZipEntryVisitor visitor) {
			this.filter = filter;
			this.visitor = visitor;
		}
		
		public void visiteDirectoryEntry(ZipFile zipFile, ZipEntry dir) throws IOException {
			if(dir.getName().matches(filter)) {
				visitor.visiteDirectoryEntry(zipFile, dir);
			}
		}

		public void visiteFileEntry(ZipFile zipFile, ZipEntry file) throws IOException {
			if(file.getName().matches(filter)) {
				visitor.visiteFileEntry(zipFile, file);
			}
		}
	}

	public static class UnzipEntryVisitor implements IZipEntryVisitor {

		private File destination;

		public UnzipEntryVisitor(File destination) {
			this.destination = destination;
		}
	
		public void visiteDirectoryEntry(ZipFile zipFIle, ZipEntry dir) throws IOException {
			File entryDir = new File(destination,dir.getName());
			entryDir.mkdirs();
		}

		public void visiteFileEntry(ZipFile zipFile, ZipEntry file) throws IOException {
			File outputFile = new File(destination,file.getName());
			outputFile.getParentFile().mkdirs();
			outputFile.createNewFile();
			BufferedOutputStream out = null;
			BufferedInputStream in =null; 
			try {
				out = new BufferedOutputStream(new FileOutputStream(outputFile),BUFFER_SIZE);
				in= new BufferedInputStream(zipFile.getInputStream(file));
				byte[] buff = new byte[BUFFER_SIZE];
				int n = -1;
				while((n = in.read(buff,0,buff.length))>-1) {
					out.write(buff, 0, n);
				}
				out.flush();
			} finally {
				if (in != null) {
					in.close();
				}
				if (out != null) {
					out.close();
				}
			}
		}
	}
}
