org.eclipse.emf.emfstore.client.ui/src/org/eclipse/emf/emfstore/client/ui/epackages/UIRegisterEPackageController.java

/*******************************************************************************
 * Copyright 2011 Chair for Applied Software Engineering,
 * Technische Universitaet Muenchen.
 * All rights reserved. This program and the accompanying materials
 * are made available under 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.epackages;

import java.util.concurrent.Callable;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.emfstore.client.model.ServerInfo;
import org.eclipse.emf.emfstore.client.model.connectionmanager.ServerCall;
import org.eclipse.emf.emfstore.client.model.util.WorkspaceUtil;
import org.eclipse.emf.emfstore.client.ui.common.RunInUI;
import org.eclipse.emf.emfstore.client.ui.handlers.AbstractEMFStoreUIController;
import org.eclipse.emf.emfstore.server.exceptions.EmfStoreException;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.swt.widgets.Shell;

/**
 * The Class UIRegisterEPackageController.
 * 
 * @author Tobias Verhoeven
 */
public class UIRegisterEPackageController extends AbstractEMFStoreUIController<Void> {

	private ServerInfo serverInfo;

	/**
	 * Instantiates a new UIRegisterPackageController.
	 * 
	 * @param shell shell
	 * @param serverInfo the server info
	 */
	public UIRegisterEPackageController(Shell shell, ServerInfo serverInfo) {
		super(shell);
		this.serverInfo = serverInfo;
	}

	/**
	 * Register a new EPackage which can be selected with a SelectionDialog.
	 * 
	 * @param serverInfo server info
	 * @throws EmfStoreException if any error in the EmfStore occurs
	 */
	public void registerEPackage(ServerInfo serverInfo) throws EmfStoreException {
		EPackageTreeSelectionDialog dialog = new EPackageTreeSelectionDialog(
			EPackageRegistryHelper.getAvailablePackages(true));
		dialog.open();
		final EPackage pkg = dialog.getSelectedEPackage();
		if (pkg != null) {
			new ServerCall<Void>(serverInfo.getLastUsersession()) {
				@Override
				protected Void run() throws EmfStoreException {
					getConnectionManager().registerEPackage(getSessionId(), pkg);
					return null;
				}
			}.execute();
		}
	}

	@Override
	public Void doRun(IProgressMonitor monitor) {
		try {
			this.registerEPackage(serverInfo);
		} catch (final EmfStoreException e) {
			RunInUI.run(new Callable<Void>() {

				public Void call() throws Exception {
					WorkspaceUtil.logException(e.getMessage(), e);
					MessageDialog.openError(getShell(), "Registration failed", e.getMessage());
					return null;
				}
			});
		}
		return null;
	}

}