/*
 * Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies). 
 * All rights reserved.
 * This component and the accompanying materials are made available
 * under the terms of the License "Eclipse Public License v1.0"
 * which accompanies this distribution, and is available
 * at the URL "http://www.eclipse.org/legal/epl-v10.html".
 *
 * Initial Contributors:
 * Nokia Corporation - initial contribution.
 *
 * Contributors:
 *
 * Description: 
 *
 */

package com.nokia.carbide.cpp.internal.pi.wizards.ui;

import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.List;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.jface.wizard.IWizardContainer;
import org.eclipse.swt.widgets.Display;

import com.nokia.carbide.cpp.internal.pi.plugin.model.ITrace;
import com.nokia.carbide.cpp.pi.PiPlugin;
import com.nokia.carbide.cpp.pi.export.ITraceClientNotificationsIf;
import com.nokia.carbide.cpp.pi.wizards.WizardsPlugin;

/**
 * Helper class to handle traces
 */
public class TraceHandler implements ITraceClientNotificationsIf {

	private NewPIWizardPage wizardPage;
	private IWizardContainer wizardContainer;
	private ProfilerActivatorGroup profilerActivatorGroup;

	/**
	 * Constructor
	 * 
	 * @param profilerActivatorGroup
	 *            instance of the ProfilerActivatorGroup
	 */
	public TraceHandler(ProfilerActivatorGroup profilerActivatorGroup) {
		this.profilerActivatorGroup = profilerActivatorGroup;
		this.wizardPage = profilerActivatorGroup.getWizardPage();
		this.wizardContainer = profilerActivatorGroup.getWizardContainer();

	}

	/**
	 * Update current connection
	 * 
	 */
	public void updateCurrenConnection() {
		try {
			IRunnableWithProgress runnableWithProgress = new IRunnableWithProgress() {
				public void run(IProgressMonitor monitor)
						throws InvocationTargetException, InterruptedException {
					monitor.beginTask(Messages.getString("TraceHandler.resolvingCurrentConnection"), //$NON-NLS-1$
							IProgressMonitor.UNKNOWN);
					try {
						String currenConnection = PiPlugin.getTraceProvider()
								.getDisplayNameForCurrentConnection(monitor);
						profilerActivatorGroup
								.setCurrentConnection(currenConnection);
					} catch (CoreException e) {
						throw new InvocationTargetException(e);
					} finally {
						monitor.done();
					}
				}
			};
			wizardContainer.run(true, true, runnableWithProgress);
		} catch (InvocationTargetException e) {
			if(e.getCause() instanceof CoreException){
				updateStatus(((CoreException)e.getCause()).getStatus());
			}else{
				notifyError(e.getMessage());
			}
		} catch (InterruptedException e) {
			notifyError(e.getMessage());
		}
	}

	/**
	 * Fetch available plug-ins list
	 * 
	 */
	public List<ITrace> fetchAvailablePlugins() {
		final List<ITrace> plugins = new ArrayList<ITrace>();
		try {		
			IRunnableWithProgress runnableWithProgress = new IRunnableWithProgress() {
				public void run(IProgressMonitor monitor)
						throws InvocationTargetException, InterruptedException{
					monitor.beginTask(Messages.getString("TraceHandler.fetchingPluginsList"), //$NON-NLS-1$
							IProgressMonitor.UNKNOWN);
					try {				
						plugins.addAll(PiPlugin.getTraceProvider()
								.getAvailableSamplers(TraceHandler.this, monitor));				
					} catch (CoreException e) {
						throw new InvocationTargetException(e);					
					} finally {
						monitor.done();
					}

				}
			};
			wizardContainer.run(true, false, runnableWithProgress);		
			if(wizardPage instanceof INewPIWizardSettings){
				((INewPIWizardSettings)wizardPage).validatePage();
			}
			
			return plugins;
		}catch (InvocationTargetException e) {
			if(e.getCause() instanceof CoreException){
				updateStatus(((CoreException)e.getCause()).getStatus());
			}else{
				notifyError(e.getMessage());	
			}			
		} catch (InterruptedException e) {
			notifyError(e.getMessage());
		}
		return null;
	}

	/**
	 * Start trace
	 * 
	 * @param fileName
	 *            for new profiler data file
	 * @param traceIDs
	 *            selected plug-ins to be profiling
	 */
	public void startTrace(final String fileName, final int[] traceIDs) {

		try {
			IRunnableWithProgress runnableWithProgress = new IRunnableWithProgress() {
				public void run(IProgressMonitor monitor)
						throws InvocationTargetException, InterruptedException {
					monitor.beginTask(Messages.getString("TraceHandler.preparingTracing"), //$NON-NLS-1$
							IProgressMonitor.UNKNOWN);			
					try {	
						
						PiPlugin.getTraceProvider().startTrace(
								fileName,
								traceIDs,
								TraceHandler.this,
								new SubProgressMonitor(monitor,
										IProgressMonitor.UNKNOWN));						
					} catch (CoreException e) {
						throw new InvocationTargetException(e);
					} finally {
						monitor.done();
					}

				}
			};
			wizardContainer.run(true, false, runnableWithProgress);

		} catch (InvocationTargetException e) {
			if(e.getCause() instanceof CoreException){
				updateStatus(((CoreException)e.getCause()).getStatus());
			}else{
				notifyError(e.getMessage());
			}
		} catch (InterruptedException e) {
			notifyError(e.getMessage());
		}
	}

	/**
	 * Stop trace
	 * 
	 * @return
	 */
	public IPath stopTrace() {
		final IPath[] path = new Path[1];
		try {
			IRunnableWithProgress runnableWithProgress = new IRunnableWithProgress() {
				public void run(IProgressMonitor monitor)
						throws InvocationTargetException, InterruptedException {
					monitor.beginTask(Messages.getString("TraceHandler.creatingProfilerDataFile"), //$NON-NLS-1$
							IProgressMonitor.UNKNOWN);
					try {					
						path[0] = PiPlugin.getTraceProvider().stopTrace(false);	
					} catch (CoreException e) {
						throw new InvocationTargetException(e);
					} finally {
						monitor.done();
					}
				}
			};
			wizardContainer.run(true, false, runnableWithProgress);
			return path[0];
		} catch (InvocationTargetException e) {
			if(e.getCause() instanceof CoreException){
				updateStatus(((CoreException)e.getCause()).getStatus());
			}else{
				notifyError(e.getMessage());
			}
			
		} catch (InterruptedException e) {
			notifyError(e.getMessage());
		}
		return null;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * com.nokia.carbide.cpp.pi.export.ITraceClientNotificationsIf#notifyError
	 * (java.lang.String)
	 */
	public void notifyError(String message) {
		updateStatus(createStatus(IStatus.ERROR, message));

	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * com.nokia.carbide.cpp.pi.export.ITraceClientNotificationsIf#notifyInformation
	 * (java.lang.String)
	 */
	public void notifyInformation(String message) {
		updateStatus(createStatus(IStatus.INFO, message));

	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * com.nokia.carbide.cpp.pi.export.ITraceClientNotificationsIf#notifyWarning
	 * (java.lang.String)
	 */
	public void notifyWarning(String message) {
		updateStatus(createStatus(IStatus.WARNING, message));

	}

	/**
	 * Update status for the wizard
	 * 
	 * @param status
	 */
	private void updateStatus(final IStatus status) {	
		Display.getDefault().asyncExec(new Runnable() {
			public void run() {				
				wizardPage.setErrorMessage(null);		
				profilerActivatorGroup.updateButtons();
				if (status.getSeverity() == IStatus.OK) {
					wizardPage.setMessage(status.getMessage());
					wizardPage.setPageComplete(true);
				} else if (status.getSeverity() == IStatus.ERROR) {					
					wizardPage.setErrorMessage(status.getMessage());
					wizardPage.setPageComplete(false);
				} else if (status.getSeverity() == IStatus.INFO) {
					wizardPage.setMessage(status.getMessage());
					wizardPage.setPageComplete(false);
				} else if (status.getSeverity() == IStatus.WARNING) {
					wizardPage.setMessage(status.getMessage());
					wizardPage.setPageComplete(true);
				}else if (status.getSeverity() == IStatus.CANCEL) {
					wizardPage.setMessage(status.getMessage());
					wizardPage.setPageComplete(false);
				}

			}
		});
	}

	/**
	 * Create a status
	 * 
	 * @param severity
	 *            of the status
	 * @param message
	 *            of the status
	 * @return created status
	 */
	private IStatus createStatus(int severity, String message) {
		return new Status(severity, WizardsPlugin.PLUGIN_ID, message);
	}

	/*
	 * (non-Javadoc)
	 * @see com.nokia.carbide.cpp.pi.export.ITraceClientNotificationsIf#updateTraceDataFile(org.eclipse.core.runtime.IPath, long, long)
	 */
	public void updateTraceDataFile(IPath path, long time, long size) {
		profilerActivatorGroup.updateTraceDataFile(path, time, size);
		
	}
}