r/org.eclipse.statet.r.ui/srcDebug/org/eclipse/statet/r/launching/SubmitFileViaCommandLaunchShortcut.java

/*=============================================================================#
 # Copyright (c) 2008, 2020 Stephan Wahlbrink and others.
 # 
 # This program and the accompanying materials are made available under the
 # terms of the Eclipse Public License 2.0 which is available at
 # https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
 # which is available at https://www.apache.org/licenses/LICENSE-2.0.
 # 
 # SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
 # 
 # Contributors:
 #     Stephan Wahlbrink <sw@wahlbrink.eu> - initial API and implementation
 #=============================================================================*/

package org.eclipse.statet.r.launching;

import java.util.HashMap;
import java.util.Map;

import org.eclipse.core.commands.Command;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.NotEnabledException;
import org.eclipse.core.commands.NotHandledException;
import org.eclipse.core.commands.common.NotDefinedException;
import org.eclipse.debug.ui.ILaunchShortcut;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.IWorkbenchPartSite;
import org.eclipse.ui.commands.ICommandService;
import org.eclipse.ui.handlers.HandlerUtil;
import org.eclipse.ui.handlers.IHandlerService;

import org.eclipse.statet.ecommons.ui.util.UIAccess;

import org.eclipse.statet.internal.r.debug.ui.RLaunchingMessages;
import org.eclipse.statet.internal.r.debug.ui.launcher.LaunchShortcutUtil;
import org.eclipse.statet.internal.r.debug.ui.launcher.SubmitFileViaCommandHandler;


/**
 * Launch Shortcut for {@link SubmitFileViaCommandHandler}
 */
public class SubmitFileViaCommandLaunchShortcut implements ILaunchShortcut {
	
	
	private final boolean fGotoConsole;
	private final String fFileCommandId;
	
	
	public SubmitFileViaCommandLaunchShortcut(final String fileCommandId, final boolean gotoConsole) {
		fFileCommandId = fileCommandId;
		fGotoConsole = gotoConsole;
	}
	
	
	@Override
	public void launch(final ISelection selection, final String mode) {
		assert mode.equals("run"); //$NON-NLS-1$
		
		final IWorkbenchPart workbenchPart = UIAccess.getActiveWorkbenchPart(false);
		final IWorkbenchPartSite site = workbenchPart.getSite();
		if (site != null) {
			final IHandlerService handlerService = site.getService(IHandlerService.class);
			final ICommandService commandService = site.getService(ICommandService.class);
			
			final Map<String, String> parameters= new HashMap<>();
			parameters.put(RCodeLaunching.FILE_COMMAND_ID_PARAMTER_ID, fFileCommandId);
			final Command command = commandService.getCommand(!fGotoConsole ?
					RCodeLaunching.SUBMIT_FILEVIACOMMAND_COMMAND_ID : RCodeLaunching.SUBMIT_FILEVIACOMMAND_GOTOCONSOLE_COMMAND_ID);
			final ExecutionEvent executionEvent = new ExecutionEvent(command, parameters, null, handlerService.getCurrentState());
			if (!selection.equals(HandlerUtil.getCurrentSelection(executionEvent))) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
				return;
			}
			try {
				command.executeWithChecks(executionEvent);
			}
			catch (final ExecutionException e) {
				LaunchShortcutUtil.handleRLaunchException(e, RLaunchingMessages.RScriptLaunch_error_message, executionEvent); 
			}
			catch (final NotDefinedException e) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
			}
			catch (final NotEnabledException e) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
			}
			catch (final NotHandledException e) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
			}
		}
	}
	
	@Override
	public void launch(final IEditorPart editor, final String mode) {
		assert mode.equals("run"); //$NON-NLS-1$
		
		final IWorkbenchPart workbenchPart = UIAccess.getActiveWorkbenchPart(false);
		final IWorkbenchPartSite site = workbenchPart.getSite();
		if (site != null) {
			final IHandlerService handlerService = site.getService(IHandlerService.class);
			final ICommandService commandService = site.getService(ICommandService.class);
			
			final Map<String, String> parameters= new HashMap<>();
			parameters.put(RCodeLaunching.FILE_COMMAND_ID_PARAMTER_ID, fFileCommandId);
			final Command command = commandService.getCommand(!fGotoConsole ?
					RCodeLaunching.SUBMIT_FILEVIACOMMAND_COMMAND_ID : RCodeLaunching.SUBMIT_FILEVIACOMMAND_GOTOCONSOLE_COMMAND_ID);
			final ExecutionEvent executionEvent = new ExecutionEvent(command, parameters, null, handlerService.getCurrentState());
			if (!editor.equals(HandlerUtil.getActivePart(executionEvent))) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
				return;
			}
			try {
				command.executeWithChecks(executionEvent);
			}
			catch (final ExecutionException e) {
				LaunchShortcutUtil.handleRLaunchException(e, RLaunchingMessages.RScriptLaunch_error_message, executionEvent); 
			}
			catch (final NotDefinedException e) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
			}
			catch (final NotEnabledException e) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
			}
			catch (final NotHandledException e) {
				LaunchShortcutUtil.handleUnsupportedExecution(executionEvent);
			}
		}
	}
	
}