/*
    GNU GENERAL LICENSE
    Copyright (C) 2006 The Lobo Project. Copyright (C) 2014 - 2017 Lobo Evolution

    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public
    License as published by the Free Software Foundation; either
    verion 3 of the License, or (at your option) any later version.

    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
    General License for more details.

    You should have received a copy of the GNU General Public
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
    

    Contact info: lobochief@users.sourceforge.net; ivan.difrancesco@yahoo.it
 */
package org.lobobrowser.primary.clientlets.download;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;

import org.lobobrowser.clientlet.CancelClientletException;
import org.lobobrowser.clientlet.Clientlet;
import org.lobobrowser.clientlet.ClientletContext;
import org.lobobrowser.clientlet.ClientletException;
import org.lobobrowser.clientlet.ClientletResponse;
import org.lobobrowser.primary.gui.download.DownloadDialog;
import org.lobobrowser.util.Strings;
import org.lobobrowser.util.Urls;

/**
 * The Class DownloadClientlet.
 */
public class DownloadClientlet implements Clientlet {

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * org.lobobrowser.clientlet.Clientlet#process(org.lobobrowser.clientlet.
	 * ClientletContext)
	 */
	@Override
	public void process(ClientletContext context) throws ClientletException {
		ClientletResponse response = context.getResponse();
		URL url = response.getResponseURL();
		if (url.getProtocol().equals("file") && "".equals(url.getHost())) {
			String shorterPath = Strings.truncate(Urls.getNoRefForm(url), 64);
			context.getNavigatorFrame().alert("There are no extensions that can render\r\n" + shorterPath + ".");
			throw new CancelClientletException("cancel");
		}
		if (!"GET".equals(response.getLastRequestMethod())) {
			String shorterPath = Strings.truncate(Urls.getNoRefForm(url), 64);
			context.getNavigatorFrame()
					.alert("Cannot download document that is not accessed with method GET:\r\n" + shorterPath + ".");
			throw new CancelClientletException("cancel");
		}
		// Load a bit of content to determine transfer speed
		int transferSpeed = -1;
		int contentLength = response.getContentLength();
		if (contentLength > 0) {
			try {
				InputStream in = response.getInputStream();
				try {
					long baseTime = System.currentTimeMillis();
					long maxElapsed = 1000;
					byte[] buffer = new byte[4096];
					int numRead;
					int totalRead = 0;
					while (((System.currentTimeMillis() - baseTime) < maxElapsed)
							&& ((numRead = in.read(buffer)) != -1)) {
						totalRead += numRead;
					}
					// Note: This calcuation depends on
					// content not being stored in cache.
					// It works just because downloads
					// are not stored in the cache.
					long elapsed = System.currentTimeMillis() - baseTime;
					if (elapsed > 0) {
						transferSpeed = (int) Math.round((double) totalRead / elapsed);
					}
				} finally {
					in.close();
				}
			} catch (IOException ioe) {
				throw new ClientletException(ioe);
			}
		}
		DownloadDialog dialog = new DownloadDialog(response, url, transferSpeed);
		dialog.setTitle("Download " + Urls.getNoRefForm(url));
		dialog.pack();
		dialog.setLocationByPlatform(true);
		dialog.setVisible(true);

		// Cancel current transfer
		throw new CancelClientletException("download");
	}
}