core/plugins/org.eclipse.dltk.testing/src/org/eclipse/dltk/internal/testing/ui/GotoReferencedTestAction.java

/*******************************************************************************
 * Copyright (c) 2000, 2017 IBM Corporation and others.
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.dltk.internal.testing.ui;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.dltk.core.IField;
import org.eclipse.dltk.core.IMember;
import org.eclipse.dltk.core.IMethod;
import org.eclipse.dltk.core.IModelElement;
import org.eclipse.dltk.core.ISourceModule;
import org.eclipse.dltk.core.IType;
import org.eclipse.dltk.core.ModelException;
import org.eclipse.dltk.internal.testing.Messages;
import org.eclipse.dltk.internal.ui.actions.SelectionConverter;
import org.eclipse.dltk.internal.ui.editor.EditorUtility;
import org.eclipse.dltk.internal.ui.editor.ScriptEditor;
import org.eclipse.dltk.testing.DLTKTestingMessages;
import org.eclipse.dltk.testing.DLTKTestingPlugin;
import org.eclipse.dltk.ui.DLTKUIPlugin;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.text.ITextSelection;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.window.Window;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.IWorkbenchWindowActionDelegate;
import org.eclipse.ui.PartInitException;
import org.eclipse.ui.dialogs.SelectionStatusDialog;

/**
 * Shows a dialog with test methods that refer to the selection.
 */
public class GotoReferencedTestAction
		implements IWorkbenchWindowActionDelegate {
	ISelection fSelection;
	IWorkbenchWindow fWorkbench;

	private void run(IStructuredSelection selection) {
		IModelElement[] elements = getSelectedElements(selection);
		if (elements.length == 0) {
			MessageDialog.openInformation(getShell(),
					DLTKTestingMessages.GotoReferencedTestAction_dialog_title,
					DLTKTestingMessages.GotoReferencedTestAction_dialog_message);
			return;
		}
		try {
			run(elements);
		} catch (CoreException e) {
			ErrorDialog.openError(getShell(),
					DLTKTestingMessages.GotoReferencedTestAction_dialog_title,
					DLTKTestingMessages.GotoReferencedTestAction_dialog_error,
					e.getStatus());
		}
	}

	private void run(ITextSelection ITextSelection) {
		try {
			ScriptEditor editor = getActiveEditor();
			if (editor == null)
				return;
			IModelElement element = SelectionConverter
					.getElementAtOffset(editor);
			int type = element != null ? element.getElementType() : -1;
			if (type != IModelElement.METHOD && type != IModelElement.TYPE) {
				element = SelectionConverter.getElementAtOffset(editor);
				if (element == null) {
					MessageDialog.openInformation(getShell(),
							DLTKTestingMessages.GotoReferencedTestAction_dialog_title,
							DLTKTestingMessages.GotoReferencedTestAction_dialog_error_nomethod);
					return;
				}
			}
			run(new IMember[] { (IMember) element });
		} catch (CoreException e) {
			ErrorDialog.openError(getShell(),
					DLTKTestingMessages.GotoReferencedTestAction_dialog_title,
					DLTKTestingMessages.GotoReferencedTestAction_dialog_error,
					e.getStatus());
		}
	}

	private void run(IModelElement[] elements)
			throws PartInitException, ModelException {
		IModelElement element = elements[0];

		SelectionStatusDialog dialog = new TestMethodSelectionDialog(getShell(),
				element);
		dialog.setTitle(
				DLTKTestingMessages.GotoReferencedTestAction_selectdialog_title);
		String msg = Messages.format(
				DLTKTestingMessages.GotoReferencedTestAction_dialog_select_message,
				element.getElementName());
		dialog.setMessage(msg);

		if (dialog.open() == Window.CANCEL)
			return;

		Object result = dialog.getFirstResult();
		if (result == null)
			return;

		openElement((IModelElement) result);
	}

	private void openElement(IModelElement result)
			throws ModelException, PartInitException {
		IEditorPart part = DLTKUIPlugin.openInEditor(result);
		EditorUtility.revealInEditor(part, result);
	}

	private IModelElement[] getSelectedElements(
			IStructuredSelection selection) {
		List elements = selection.toList();
		int size = elements.size();
		if (size == 0)
			return new IModelElement[0];

		ArrayList<IModelElement> result = new ArrayList<>(size);

		for (int i = 0; i < size; i++) {
			Object e = elements.get(i);
			if (e instanceof ISourceModule) {
				ISourceModule unit = (ISourceModule) e;
				IType[] types = new IType[0];
				try {
					types = unit.getTypes();
				} catch (ModelException ex) {
				}
				for (int j = 0; j < types.length; j++) {
					result.add(types[j]);
				}
			} else if (e instanceof IMethod || e instanceof IType
					|| e instanceof IField) {
				result.add((IModelElement) e);
			} else {
				return new IModelElement[0];
			}
		}
		return result.toArray(new IModelElement[result.size()]);
	}

	@Override
	public void run(IAction action) {
		if (fSelection instanceof IStructuredSelection)
			run((IStructuredSelection) fSelection);
		else if (fSelection instanceof ITextSelection)
			run((ITextSelection) fSelection);
	}

	@Override
	public void selectionChanged(IAction action, ISelection selection) {
		fSelection = selection;
		action.setEnabled(getActiveEditor() != null);
	}

	private Shell getShell() {
		if (fWorkbench != null)
			return fWorkbench.getShell();
		return DLTKTestingPlugin.getActiveWorkbenchShell();
	}

	@Override
	public void dispose() {
	}

	@Override
	public void init(IWorkbenchWindow window) {
		fWorkbench = window;
	}

	private ScriptEditor getActiveEditor() {
		IEditorPart editor = fWorkbench.getActivePage().getActiveEditor();
		if (editor instanceof ScriptEditor)
			return (ScriptEditor) editor;
		return null;
	}
}