/*

 * Hedgewars for Android. An Android port of Hedgewars, a free turn based strategy game

 * Copyright (c) 2011-2012 Richard Deurwaarder <xeli@xelification.com>

 *

 * 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 version 2

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

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

 */

package org.hedgewars.hedgeroid.Downloader;

import java.util.LinkedList;

import java.util.List;

import org.hedgewars.hedgeroid.R;

import android.app.Notification;

import android.app.NotificationManager;

import android.app.PendingIntent;

import android.app.Service;

import android.content.Intent;

import android.os.Handler;

import android.os.IBinder;

import android.os.Message;

import android.os.Messenger;

import android.os.RemoteException;

import android.widget.RemoteViews;

public class DownloadService extends Service {

	public final static String INTENT_TASKID = "taskId";

	public final static String INTENT_TASK = "task";

	public static final String PREF_DOWNLOADED = "downloaded";

	public static final int MSG_CANCEL = 0;

	public static final int MSG_UNREGISTER_CLIENT = 2;

	public final static int MSG_ADDTASK = 4;

	public static final int NOTIFICATION_PROCESSING = 0;

	public static final int NOTIFICATION_DONE = 1;

	private DownloadAsyncTask asyncExecutor;

	private DownloadHandler handler = new DownloadHandler();

	private final Messenger messenger = new Messenger(handler);

	private NotificationManager nM;

	private RemoteViews contentView;

	private LinkedList<DownloadTask> downloadTasks = new LinkedList<DownloadTask>();

	private DownloadTask currentTask = null;

	public class DownloadHandler extends Handler{

		public void handleMessage(Message msg){

			if(msg.obj != null){

				DownloadPackage pack = (DownloadPackage) msg.obj;

				DownloadTask task = null;

				Messenger replyToMessenger = msg.replyTo;

				for(DownloadTask _task : downloadTasks){

					if(_task.getPackage().equals(pack)){

						task = _task;

						break;

					}

				}

				switch(msg.what){

				case MSG_ADDTASK:

					if(task == null){

						task = new DownloadTask(pack);

						downloadTasks.add(task);

					}

					task.addClient(replyToMessenger);

					runNextTask();

					return;

				case MSG_CANCEL:

					if(task != null && task.getPackage().equals(pack) && task.getStatus() == TASK_STATE.PENDING){

						downloadTasks.remove(task);

					}

					if(currentTask != null && currentTask.getPackage().equals(pack)){//TODO synchronization problem?

						asyncExecutor.cancel(false);

					}

					return;

				case MSG_UNREGISTER_CLIENT:

					if(task != null){

						task.removeClient(replyToMessenger);

					}

					return;

				}

			}

		}

	}

	public void onCreate(){

		super.onCreate();

		nM = (NotificationManager) getSystemService(NOTIFICATION_SERVICE);

	}

	public IBinder onBind(Intent intent) {

		return messenger.getBinder();

	}

	private void runNextTask(){

		if(asyncExecutor == null){//if (task isnt running right now) ...

			currentTask = downloadTasks.poll();

			if(currentTask != null){

				asyncExecutor = new DownloadAsyncTask(currentTask);

				asyncExecutor.execute(currentTask.getPackage());

			}

		}

	}

	public void onDestroy(){

		super.onDestroy();

		asyncExecutor.cancel(false);	

	}

	class DownloadTask {

		private final DownloadPackage pack;

		private TASK_STATE status = TASK_STATE.PENDING;

		private Notification progressNotification, doneNotification;

		//I expect little to no removeClient calls that's why we go for a list rather than a map

		private final List<Messenger> clients;

		public DownloadTask(DownloadPackage _pack){

			pack = _pack;

			clients = new LinkedList<Messenger>();

		}

		public void addClient(Messenger messenger){

			clients.add(messenger);

		}

		public void removeClient(Messenger messenger){

			clients.remove(messenger);

		}

		public DownloadPackage getPackage(){

			return pack;

		}

		public TASK_STATE getStatus(){

			return status;

		}

		public void sendMessageToClients(Message msg){

			for(Messenger messenger : clients){

				try {

					messenger.send(msg);		

				} catch (RemoteException e) {

					e.printStackTrace();

				}

			}

		}

		/*

		 * Callbacks called from the async tasks

		 */

		//Thread safe method to let clients know the processing is starting and will process int max kbytes

		public void start(int max){

			progressNotification = new Notification(R.drawable.statusbar, getString(R.string.notification_title), System.currentTimeMillis());

			progressNotification.flags |= Notification.FLAG_ONGOING_EVENT;

			contentView = new RemoteViews(getPackageName(), R.layout.notification);

			contentView.setProgressBar(R.id.notification_progress, 100, 34, false);

			progressNotification.contentView = contentView;

			PendingIntent contentIntent = PendingIntent.getActivity(DownloadService.this, 0, new Intent(DownloadService.this, DownloadListActivity.class), Intent.FLAG_ACTIVITY_NEW_TASK);

			progressNotification.contentIntent = contentIntent;

			startForeground(NOTIFICATION_PROCESSING, progressNotification);

			Message msg = Message.obtain(null, DownloadFragment.MSG_START, max, 0);

			sendMessageToClients(msg);

		}

		//periodically gets called by the ASyncTask, we can't tell for sure when it's called

		public void update(int progress, int max, String fileName){

			progress = (progress/1024);

			contentView.setProgressBar(R.id.notification_progress, max, progress, false);

			contentView.setTextViewText(R.id.progressbar_sub, String.format("%dkb/%dkb (Compressed sizes)", progress, max));

			nM.notify(NOTIFICATION_PROCESSING, progressNotification);

			sendMessageToClients(Message.obtain(handler, DownloadFragment.MSG_UPDATE, progress, max, fileName));

		}

		//Call back from the ASync task when the task has either run into an error or finished otherwise

		public void done(int result){

			switch(result){

			case DownloadAsyncTask.EXIT_SUCCESS: 	sendMessageToClients(Message.obtain(handler, DownloadFragment.MSG_DONE)); break;

			case DownloadAsyncTask.EXIT_CONNERROR:  sendMessageToClients(Message.obtain(handler, DownloadFragment.MSG_FAILED, DownloadAsyncTask.EXIT_CONNERROR, 0)); break;

			case DownloadAsyncTask.EXIT_FNF:		sendMessageToClients(Message.obtain(handler, DownloadFragment.MSG_FAILED, DownloadAsyncTask.EXIT_FNF, 0)); break;

			case DownloadAsyncTask.EXIT_MD5:		sendMessageToClients(Message.obtain(handler, DownloadFragment.MSG_FAILED, DownloadAsyncTask.EXIT_MD5, 0)); break;

			case DownloadAsyncTask.EXIT_URLFAIL:	sendMessageToClients(Message.obtain(handler, DownloadFragment.MSG_FAILED, DownloadAsyncTask.EXIT_URLFAIL, 0)); break;

			case DownloadAsyncTask.EXIT_CANCELLED:	sendMessageToClients(Message.obtain(handler, DownloadFragment.MSG_DONE)); break;

			}

			stopForeground(true);

			nM.cancel(NOTIFICATION_PROCESSING);

			String title = getString(R.string.notification_title);

			doneNotification = new Notification(R.drawable.icon, title, System.currentTimeMillis());

			doneNotification.flags |= Notification.FLAG_AUTO_CANCEL;

			PendingIntent contentIntent = PendingIntent.getActivity(DownloadService.this, 0, new Intent(DownloadService.this, DownloadListActivity.class), Intent.FLAG_ACTIVITY_NEW_TASK);

			doneNotification.setLatestEventInfo(DownloadService.this, title, getString(R.string.notification_done) + pack, contentIntent);

			nM.notify(pack.getId(), doneNotification);

			asyncExecutor = null;

			runNextTask();//see if there are more tasks

		}

	}

	enum TASK_STATE{

		RUNNING, FINISHED, PENDING;

	}

}