org.eclipse.emf.emfstore.client.ui/src/org/eclipse/emf/emfstore/client/ui/views/emfstorebrowser/handlers/ExportProjectHistoryHandler.java

/*******************************************************************************
 * Copyright (c) 2008-2011 Chair for Applied Software Engineering,
 * Technische Universitaet Muenchen.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 ******************************************************************************/
package org.eclipse.emf.emfstore.client.ui.views.emfstorebrowser.handlers;

import java.io.File;
import java.io.IOException;

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.emf.ecp.common.util.PreferenceHelper;
import org.eclipse.emf.emfstore.client.model.ServerInfo;
import org.eclipse.emf.emfstore.client.model.Usersession;
import org.eclipse.emf.emfstore.client.model.WorkspaceManager;
import org.eclipse.emf.emfstore.client.model.util.WorkspaceUtil;
import org.eclipse.emf.emfstore.client.ui.commands.ExportProjectHandler;
import org.eclipse.emf.emfstore.server.exceptions.EmfStoreException;
import org.eclipse.emf.emfstore.server.model.ProjectHistory;
import org.eclipse.emf.emfstore.server.model.ProjectInfo;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.TreeNode;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.FileDialog;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;

/**
 * Action for to export project history from server.
 * 
 * @author hodaie
 */
public class ExportProjectHistoryHandler extends AbstractHandler {

	/**
	 * These filter names are used to filter which files are displayed.
	 */
	public static final String[] FILTER_NAMES = { "EMFStore Project Files (*.eph)", "All Files (*.*)" };

	/**
	 * These filter extensions are used to filter which files are displayed.
	 */
	public static final String[] FILTER_EXTS = { "*.eph", "*.*" };

	private static final String EXPORT_PROJECT_HISTORY_PATH = "org.eclipse.emf.emfstore.client.ui.exportProjectHistoryPath";

	private Usersession usersession;
	private ProjectInfo projectInfo;

	/**
	 * {@inheritDoc}
	 */
	public Object execute(ExecutionEvent event) throws ExecutionException {

		IWorkbenchWindow activeWorkbenchWindow = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
		ISelection selection = activeWorkbenchWindow.getSelectionService().getSelection();
		Object obj = ((IStructuredSelection) selection).getFirstElement();

		if (!(obj instanceof TreeNode)) {
			return null;
		}
		TreeNode node = (TreeNode) obj;
		projectInfo = ((ProjectInfo) node.getValue());
		ServerInfo serverInfo = ((ServerInfo) node.getParent().getValue());
		usersession = serverInfo.getLastUsersession();

		final String absoluteFileName = showSaveFileDialog();
		if (absoluteFileName == null) {
			return null;
		}

		final ProgressMonitorDialog progressDialog = new ProgressMonitorDialog(PlatformUI.getWorkbench()
			.getActiveWorkbenchWindow().getShell());

		progressDialog.open();
		progressDialog.getProgressMonitor().beginTask("Export project history...", 100);
		progressDialog.getProgressMonitor().worked(10);

		ProjectHistory projectHistory = null;
		try {
			if (usersession != null) {
				projectHistory = WorkspaceManager.getInstance().getConnectionManager()
					.exportProjectHistoryFromServer(usersession.getSessionId(), projectInfo.getProjectId());
				saveProjectHistory(projectHistory, absoluteFileName);

				MessageDialog.openInformation(null, "Import", "Exported project history to file: " + absoluteFileName);
			}

		} catch (EmfStoreException e) {
			WorkspaceUtil.logException("Error exporting project history.", e);
		} catch (IOException e) {
			WorkspaceUtil.logException("Error exporting project history.", e);
		} finally {

			progressDialog.getProgressMonitor().done();
			progressDialog.close();
		}
		return null;

	}

	private void saveProjectHistory(ProjectHistory projectHistory, String absoluteFileName) throws IOException {
		ResourceSet resourceSet = new ResourceSetImpl();
		Resource resource = resourceSet.createResource(URI.createFileURI(absoluteFileName));
		resource.getContents().add(projectHistory);
		resource.save(null);

	}

	private String showSaveFileDialog() {

		FileDialog dialog = new FileDialog(PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell(), SWT.SAVE);
		dialog.setFilterNames(ExportProjectHandler.FILTER_NAMES);
		dialog.setFilterExtensions(ExportProjectHandler.FILTER_EXTS);
		dialog.setOverwrite(true);
		String initialPath = PreferenceHelper.getPreference(EXPORT_PROJECT_HISTORY_PATH,
			System.getProperty("user.home"));
		dialog.setFilterPath(initialPath);
		String initialFileName = projectInfo.getName() + ".uph";
		dialog.setFileName(initialFileName);

		// dialog
		String fn = dialog.open();
		if (fn == null) {
			return null;
		}

		final File file = new File(fn);
		PreferenceHelper.setPreference(EXPORT_PROJECT_HISTORY_PATH, dialog.getFilterPath());

		return file.getAbsolutePath();
	}

	/**
	 * Sets the user sesssion.
	 * 
	 * @param session user session
	 */
	public void setUsersession(Usersession session) {
		this.usersession = session;
	}

	/**
	 * Sets the project info, whose history will be exported.
	 * 
	 * @param projectInfo project info
	 */
	public void setProjectInfo(ProjectInfo projectInfo) {
		this.projectInfo = projectInfo;

	}
}