/*
 * LinShare is an open source filesharing software, part of the LinPKI software
 * suite, developed by Linagora.
 * 
 * Copyright (C) 2015 LINAGORA
 * 
 * This program is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Affero General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option) any
 * later version, provided you comply with the Additional Terms applicable for
 * LinShare software by Linagora pursuant to Section 7 of the GNU Affero General
 * Public License, subsections (b), (c), and (e), pursuant to which you must
 * notably (i) retain the display of the “LinShare™” trademark/logo at the top
 * of the interface window, the display of the “You are using the Open Source
 * and free version of LinShare™, powered by Linagora © 2009–2015. Contribute to
 * Linshare R&D by subscribing to an Enterprise offer!” infobox and in the
 * e-mails sent with the Program, (ii) retain all hypertext links between
 * LinShare and linshare.org, between linagora.com and Linagora, and (iii)
 * refrain from infringing Linagora intellectual property rights over its
 * trademarks and commercial brands. Other Additional Terms apply, see
 * <http://www.linagora.com/licenses/> for more details.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Affero General Public License and
 * its applicable Additional Terms for LinShare along with this program. If not,
 * see <http://www.gnu.org/licenses/> for the GNU Affero General Public License
 * version 3 and <http://www.linagora.com/licenses/> for the Additional Terms
 * applicable to LinShare software.
 */
package org.linagora.linshare.core.utils;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ArchiveZipStream extends InputStream {
	
	private static final Logger logger = LoggerFactory.getLogger(ArchiveZipStream.class);
	
	public static final String ARCHIVE_ZIP_DOWNLOAD_NAME = "allFiles.zip";
	
	private final int BUFFERSIZE = 40 *1024;
	private Map<String,InputStream> allFilesTozip; 
	
	private BufferedOutputStream bzfout;
	private ZipOutputStream zout;
	
	private File tempFile;
	private FileInputStream fi;
	

	/**
	 * 
	 * @param out
	 * @param allFilesTozip is a map with filename as a key and inputstream associated to this file
	 * @throws IOException
	 */
	public ArchiveZipStream(Map<String,InputStream> allFilesTozip) {
			try {
				this.allFilesTozip =  allFilesTozip;
				tempFile = File.createTempFile("linshareZip", null);
				
				zout = new ZipOutputStream(new FileOutputStream(tempFile));
				bzfout= new BufferedOutputStream(zout);
				
				writeprocess();
				fi = new FileInputStream(tempFile);
			} catch (FileNotFoundException e) {
				e.printStackTrace();
			} catch (IOException e) {
				e.printStackTrace();
			}
	}


	private void releaseAll() {
		if (zout != null) {
			try {
				zout.close();
			} catch (IOException e) {
				logger.error(e.toString());
			}
		}
		if (bzfout != null) {
			try {
				bzfout.close();
			} catch (IOException e) {
				logger.error(e.toString());
			}
		}
		if (fi != null) {
			try {
				fi.close();
			} catch (IOException e) {
				logger.error(e.toString());
			}
		}
		if (tempFile != null && tempFile.exists()) {
			tempFile.delete();
		}
	}

	private void writeprocess() throws IOException {
		byte[] buffer;
		
		//*** all files in zip
		for (String filename : allFilesTozip.keySet()) {
			buffer = new byte[BUFFERSIZE];
			zout.putNextEntry(new ZipEntry(filename));
			
			InputStream is = allFilesTozip.get(filename);
			BufferedInputStream bif = new BufferedInputStream(is);
			
			int readed;
			
			while ((readed = bif.read(buffer)) >= 0) {
				bzfout.write(buffer, 0, readed);
			}
			
			bzfout.flush();
			zout.flush();
			zout.closeEntry();
			is.close();
			bif.close();
		}
		
		zout.close();
		bzfout.close();
	}
	
	public File getTempFile(){
		return tempFile;
	}
	

	@Override
	public int read() throws IOException {
		return fi.read();
	}
	@Override
	public int available() throws IOException {
		return fi.available();
	}
	@Override
	public void close() throws IOException {
		releaseAll();
		super.close();
	}
	
	
}
