/*

 * Copyright (c) 2009 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.save;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.io.InputStream;

import java.lang.reflect.InvocationTargetException;

import org.eclipse.core.resources.IContainer;

import org.eclipse.core.resources.IFile;

import org.eclipse.core.resources.IResource;

import org.eclipse.core.resources.IWorkspaceRoot;

import org.eclipse.core.resources.ResourcesPlugin;

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.NullProgressMonitor;

import org.eclipse.core.runtime.Path;

import org.eclipse.core.runtime.Status;

import org.eclipse.jface.dialogs.MessageDialog;

import org.eclipse.jface.operation.IRunnableWithProgress;

import org.eclipse.jface.viewers.ISelection;

import org.eclipse.jface.viewers.IStructuredSelection;

import org.eclipse.jface.wizard.Wizard;

import org.eclipse.swt.widgets.Display;

import org.eclipse.ui.INewWizard;

import org.eclipse.ui.IWorkbench;

import org.eclipse.ui.IWorkbenchPage;

import org.eclipse.ui.IWorkbenchWizard;

import org.eclipse.ui.PartInitException;

import org.eclipse.ui.PlatformUI;

import org.eclipse.ui.ide.IDE;

import com.nokia.carbide.cpp.internal.pi.interfaces.ISaveSamples;

public class SaveSamplesWizard extends Wizard implements INewWizard {

	private SaveSamplesPage page;

	private ISelection selection;

	private ISaveSamples saveSamples;

	private String saveSamplesFileContents;

	private   boolean writeInProgress = false;

	protected boolean canceled        = false;

	private IFile file;

	/**

	 * Constructor for SampleNewWizard.

	 */

	public SaveSamplesWizard(ISaveSamples saveSamples) {

		super();

		this.saveSamples = saveSamples;

		this.setWindowTitle(Messages.getString("SaveSamplesWizard.SavingSamples")); //$NON-NLS-1$

		setNeedsProgressMonitor(true);

	}

	/**

	 * Adding the page to the wizard.

	 */

	public void addPages() {

		this.page = new SaveSamplesPage(this.selection);

		addPage(this.page);

	}

	/**

	 * This method is called when 'Finish' button is pressed in

	 * the wizard. We will create an operation and run it

	 * using the wizard as execution context.

	 */

	public boolean performFinish() {

		final IPath containerName = page.getContainerName();

		String fileName = page.getFileName();

		int dotLoc = fileName.lastIndexOf('.');

		if (dotLoc == -1)

		{

			fileName += ".csv"; //$NON-NLS-1$

		}

		final String fileNameFinal = fileName;

		IWorkspaceRoot root = ResourcesPlugin.getWorkspace().getRoot();

		final IResource resource = root.findMember(containerName);

		this.saveSamples.clear();

		this.canceled = false;

		this.file = null;

		IRunnableWithProgress op = new IRunnableWithProgress() {

			public void run(IProgressMonitor monitor)

						throws InvocationTargetException, InterruptedException {

				monitor.worked(1);

				try {

					// find the file path in the workspace

					if (!resource.exists() || !(resource instanceof IContainer)) {

						throwCoreException(Messages.getString("SaveSamplesWizard.ProjectFolder") + containerName + Messages.getString("SaveSamplesWizard.doesNotExist"));  //$NON-NLS-1$//$NON-NLS-2$

					}

					IContainer container = (IContainer) resource;

					file = container.getFile(new Path(fileNameFinal));

					// save samples, several at a time, to the output file

					writeInProgress = true;

					while (writeInProgress) {

						if (monitor.isCanceled()) {

							canceled = true;

							return;

						}

						writeOneSampleSet(monitor);

					}

					if (monitor.isCanceled()) {

						canceled = true;

						return;

					}

					// open the output file for editing

					monitor.setTaskName(Messages.getString("SaveSamplesWizard.OpeningFileForEditing")); //$NON-NLS-1$

					getShell().getDisplay().asyncExec(new Runnable() {

						public void run() {

							IWorkbenchPage page =

								PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage();

							try {

								IDE.openEditor(page, file, true);

							} catch (PartInitException e) {

							}

						}

					});

					monitor.worked(1);

				} catch (CoreException e) {

					throw new InvocationTargetException(e);

				}

				if (monitor.isCanceled()) {

					canceled = true;

					return;

				}

			}

		};

		try {

			getContainer().run(true, true, op);

		} catch (InterruptedException e) {

			canceled = true;

		} catch (InvocationTargetException e) {

			Throwable realException = e.getTargetException();

			MessageDialog.openError(getShell(), Messages.getString("SaveSamplesWizard.Error"), realException.getLocalizedMessage()); //$NON-NLS-1$

		}

		if (canceled && (file != null)) {

			try {

				file.delete(true, false, new NullProgressMonitor());

			} catch (CoreException coreEx) {

				MessageDialog.openError(getShell(), Messages.getString("SaveSamplesWizard.Error"), coreEx.getLocalizedMessage()); //$NON-NLS-1$

			}

		}

		return true;

	}

	private void writeOneSampleSet(IProgressMonitor monitor) throws CoreException {

		InputStream stream;

		try {

			if (saveSamples.getIndex() == 0) {

				monitor.setTaskName(Messages.getString("SaveSamplesWizard.WritingSamples")); //$NON-NLS-1$

				stream = openContentStream();

				if (file.exists()) {

					file.setContents(stream, true, true, monitor);

				} else {

					file.create(stream, true, monitor);

				}

				stream.close();

			} else { 

				stream = openContentStream();

				if (stream == null) {

					writeInProgress = false;

					return;

				}

				String taskName = Messages.getString("SaveSamplesWizard.WritingSamples"); //$NON-NLS-1$

				if (saveSamples.getIndex() > 0)

					taskName += Messages.getString("SaveSamplesWizard.alreadyWrtten1") + saveSamples.getIndex() + Messages.getString("SaveSamplesWizard.alreadyWrtten2"); //$NON-NLS-1$ //$NON-NLS-2$

				monitor.setTaskName(taskName);

				file.appendContents(stream, true, false, monitor);

				stream.close();

			}

		} catch (IOException e) {

		}

	}

	/**

	 * We will initialize file contents with a sample text.

	 */

	private InputStream openContentStream() {

		Display.getDefault().syncExec(new Runnable() {

			public void run() {

				saveSamplesFileContents = saveSamples.getData();

			}

		});

		if (saveSamplesFileContents == null)

			return null;

		return new ByteArrayInputStream(saveSamplesFileContents.getBytes());

	}

	private void throwCoreException(String message) throws CoreException {

		IStatus status =

			new Status(IStatus.ERROR, "junk", IStatus.OK, message, null); //$NON-NLS-1$

		throw new CoreException(status);

	}

	/**

	 * We will accept the selection in the workbench to see if

	 * we can initialize from it.

	 * @see IWorkbenchWizard#init(IWorkbench, IStructuredSelection)

	 */

	public void init(IWorkbench workbench, IStructuredSelection selection) {

		this.selection = selection;

	}

}