QuitHandler.java

/*=============================================================================#
 # Copyright (c) 2007-2016 Stephan Wahlbrink (WalWare.de) and others.
 # 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:
 #     Stephan Wahlbrink - initial API and implementation
 #=============================================================================*/

package de.walware.statet.nico.ui.util;

import java.lang.reflect.InvocationTargetException;
import java.util.Map;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.osgi.util.NLS;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.statushandlers.StatusManager;

import de.walware.ecommons.ts.ITool;
import de.walware.ecommons.ts.IToolRunnable;
import de.walware.ecommons.ts.IToolService;
import de.walware.ecommons.ui.util.UIAccess;

import de.walware.statet.nico.core.runtime.IConsoleService;
import de.walware.statet.nico.core.runtime.IRemoteEngineController;
import de.walware.statet.nico.core.runtime.IToolEventHandler;
import de.walware.statet.nico.core.runtime.ToolController;
import de.walware.statet.nico.core.runtime.ToolProcess;
import de.walware.statet.nico.internal.ui.Messages;
import de.walware.statet.nico.ui.NicoUI;


/**
 * @see IToolEventHandler#SCHEDULE_QUIT_EVENT_ID
 */
public class QuitHandler implements IToolEventHandler {
	
	
	private static class UIRunnable implements Runnable {
		
		private ToolController fController;
		private String fDialogTitle;
		private String fDialogMessage;
		private String[] fDialogOptions;
		private volatile int fResult;
		
		@Override
		public void run() {
			final IWorkbenchWindow window = UIAccess.getActiveWorkbenchWindow(true);
			final MessageDialog dialog = new MessageDialog(window.getShell(), fDialogTitle, null,
					fDialogMessage, MessageDialog.QUESTION, fDialogOptions, 0 );
			fResult = dialog.open();
			
			if (fResult == 1) {
				try {
					window.run(true, true, new IRunnableWithProgress() {
						@Override
						public void run(final IProgressMonitor monitor) throws InvocationTargetException {
							try {
								fController.kill(monitor);
							} catch (final CoreException e) {
								throw new InvocationTargetException(e);
							}
						}
					});
				}
				catch (final InvocationTargetException e) {
					StatusManager.getManager().handle(new Status(IStatus.ERROR, NicoUI.PLUGIN_ID, -1,
							Messages.TerminatingMonitor_Force_error_message, e.getTargetException()),
							StatusManager.LOG | StatusManager.SHOW);
				}
				catch (final InterruptedException e) {
				}
			}
		}
		
	}
	
	@Override
	public IStatus execute(final String id, final IToolService service, final Map<String, Object> data,
			final IProgressMonitor monitor) {
		final IConsoleService console = (IConsoleService) service;
		if (PlatformUI.getWorkbench().isClosing()) {
			final ToolController controller = console.getController();
			if (controller != null) {
				if (console.getTool().isProvidingFeatureSet(IRemoteEngineController.FEATURE_SET_ID)) {
					try {
						((IRemoteEngineController) controller).disconnect(monitor);
						return Status.CANCEL_STATUS;
					}
					catch (final CoreException e) {}
				}
				try {
					controller.kill(monitor);
					return Status.CANCEL_STATUS;
				}
				catch (final CoreException e) {}
			}
			return Status.CANCEL_STATUS;
		}
		
		final IToolRunnable[] quitRunnables = (IToolRunnable[]) data.get("scheduledQuitTasks");
		if (quitRunnables.length == 0) {
			return Status.OK_STATUS; // run default = schedule quit
		}
		
		final UIRunnable runner = new UIRunnable();
		runner.fController = console.getController();
		final ToolProcess process = runner.fController.getTool();
		runner.fDialogTitle = NLS.bind(Messages.TerminatingMonitor_title, process.getLabel(ITool.DEFAULT_LABEL));
		runner.fDialogMessage = NLS.bind(Messages.TerminatingMonitor_message, process.getLabel(ITool.LONG_LABEL));
		runner.fDialogOptions = new String[] {
				Messages.TerminatingMonitor_WaitButton_label,
				Messages.TerminatingMonitor_ForceButton_label,
				Messages.TerminatingMonitor_CancelButton_label
		};
		
		UIAccess.getDisplay().syncExec(runner);
		if (runner.fResult == 2) {
			runner.fController.cancelQuit();
		}
		return Status.CANCEL_STATUS; // do nothing
	}
	
}