| /******************************************************************************* |
| * Copyright (c) 2000, 2016 IBM Corporation 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: |
| * 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.swt.widgets.Shell; |
| |
| import org.eclipse.jface.action.IAction; |
| import org.eclipse.jface.dialogs.ErrorDialog; |
| import org.eclipse.jface.dialogs.MessageDialog; |
| import org.eclipse.jface.viewers.ISelection; |
| import org.eclipse.jface.viewers.IStructuredSelection; |
| import org.eclipse.jface.window.Window; |
| |
| import org.eclipse.jface.text.ITextSelection; |
| |
| 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; |
| |
| 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; |
| |
| /** |
| * 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<IModelElement>(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; |
| } |
| } |