1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
|
/*******************************************************************************
* Copyright (c) 2006 IBM Corporation.
* 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 - Jeff Briggs, Henry Hughes, Ryan Morse
*******************************************************************************/
package org.eclipse.linuxtools.internal.systemtap.ui.ide.actions;
import java.text.MessageFormat;
import java.util.LinkedList;
import java.util.List;
import org.eclipse.jface.action.Action;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.ArrayContentProvider;
import org.eclipse.jface.viewers.DoubleClickEvent;
import org.eclipse.jface.viewers.IDoubleClickListener;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.jface.window.Window;
import org.eclipse.linuxtools.internal.systemtap.ui.ide.Localization;
import org.eclipse.linuxtools.internal.systemtap.ui.ide.editors.stp.STPEditor;
import org.eclipse.linuxtools.internal.systemtap.ui.ide.views.ProbeAliasBrowserView;
import org.eclipse.linuxtools.systemtap.structures.TreeNode;
import org.eclipse.linuxtools.systemtap.ui.editor.actions.file.NewFileAction;
import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IEditorReference;
import org.eclipse.ui.ISelectionListener;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.dialogs.ListDialog;
/**
* This <code>Action</code> is fired when the user selects an item in the <code>ProbeAliasBrowserView</code>.
* The action taken is to insert a template probe in the current <code>STPEditor</code>, if available, or to
* insert the probe into a new <code>STPEditor</code> if one does not exist.
* @author Henry Hughes
* @author Ryan Morse
* @see org.eclipse.linuxtools.systemtap.ui.editor.SimpleEditor#insertText(String)
* @see org.eclipse.linuxtools.internal.systemtap.ui.ide.views.ProbeAliasBrowserView
* @see org.eclipse.jface.action.Action
*/
public class ProbeAliasAction extends Action implements ISelectionListener, IDoubleClickListener {
private final IWorkbenchWindow window;
private final ProbeAliasBrowserView viewer;
private static final String ID = "org.eclipse.linuxtools.systemtap.ui.ide.ProbeAliasAction"; //$NON-NLS-1$
private IStructuredSelection selection;
/**
* The Default Constructor. Takes the <code>IWorkbenchWindow</code> that it effects
* as well as the <code>ProbeAliasBrowserView</code> that will fire this action.
* @param window window effected by this event
* @param view browser that fires this action
*/
public ProbeAliasAction(IWorkbenchWindow window, ProbeAliasBrowserView view) {
this.window = window;
setId(ID);
setActionDefinitionId(ID);
setText(Localization.getString("ProbeAliasAction.Insert")); //$NON-NLS-1$
setToolTipText(Localization
.getString("ProbeAliasAction.InsertSelectedProbe")); //$NON-NLS-1$
window.getSelectionService().addSelectionListener(this);
viewer = view;
}
/**
* Updates <code>selection</code> with the current selection whenever the user changes
* the current selection.
*/
@Override
public void selectionChanged(IWorkbenchPart part, ISelection incoming) {
if (incoming instanceof IStructuredSelection) {
selection = (IStructuredSelection) incoming;
setEnabled(selection.size() == 1);
} else {
// Other selections, for example containing text or of other kinds.
setEnabled(false);
}
}
public void dispose() {
window.getSelectionService().removeSelectionListener(this);
}
/**
* The main body of the action. This method checks for the current editor, creating one
* if there is no active <code>STPEditor</code>, and then inserts a template probe for the
* item that the user clicked on.
*/
@Override
public void run() {
IWorkbenchPage page = window.getActivePage();
IEditorPart editor = page.getActiveEditor();
if (!(editor instanceof STPEditor)) {
editor = findEditor();
if (null == editor) {
return;
}
}
page.activate(editor);
ISelection incoming = viewer.getViewer().getSelection();
IStructuredSelection selection = (IStructuredSelection)incoming;
Object o = selection.getFirstElement();
if (o instanceof TreeNode && editor instanceof STPEditor) {
buildString((STPEditor) editor, (TreeNode) o);
}
}
private void buildString(STPEditor stpeditor, TreeNode t) {
//build the string
StringBuilder s = new StringBuilder("\nprobe " + t.toString()); //$NON-NLS-1$
if(!t.isClickable()) {
if(0 <t.getChildCount()) {
s.append(".*"); //$NON-NLS-1$
} else {
return;
}
}
s.append("\n{\n"); //$NON-NLS-1$
if(t.isClickable() && t.getChildCount() > 0) {
s.append("\t/*\n\t * " + //$NON-NLS-1$
Localization
.getString("ProbeAliasAction.AvailableVariables") + //$NON-NLS-1$
"\n\t * "); //$NON-NLS-1$
boolean first = true;
for(int i = 0; i < t.getChildCount(); i++) {
if(first) {
first = false;
} else {
s.append(", "); //$NON-NLS-1$
}
s.append(t.getChildAt(i).toString());
}
s.append("\n\t */\n"); //$NON-NLS-1$
}
s.append("\n}\n"); //$NON-NLS-1$
stpeditor.insertText(s.toString());
}
private IEditorPart getRestoredEditor(final IEditorReference ref) {
if (ref.getEditor(false) == null) {
Display.getDefault().syncExec(new Runnable() {
@Override
public void run() {
ref.getEditor(true);
}
});
}
return ref.getEditor(false);
}
private IEditorPart findEditor() {
final List<IEditorPart> allEditors = new LinkedList<>();
for (IEditorReference ref : window.getActivePage().getEditorReferences()) {
IEditorPart editor = getRestoredEditor(ref);
if (editor instanceof STPEditor) {
allEditors.add(editor);
}
}
switch (allEditors.size()) {
// If only one file is found, open it. Give user the option to open another file.
case 1:
MessageDialog messageDialog = new MessageDialog(window.getShell(),
Messages.ProbeAliasAction_DialogTitle, null,
MessageFormat.format(Messages.ProbeAliasAction_AskBeforeAddMessage,
allEditors.get(0).getEditorInput().getName() ),
MessageDialog.QUESTION,
new String[]{Messages.ProbeAliasAction_AskBeforeAddCancel,
Messages.ProbeAliasAction_AskBeforeAddAnother,
Messages.ProbeAliasAction_AskBeforeAddYes}, 2);
switch (messageDialog.open()) {
case 2:
return allEditors.get(0);
case 1:
return openNewFile();
default:
return null;
}
// If no files found, prompt user to open a new file
case 0:
return openNewFile();
// If multiple files found, prompt user to select one of them
default:
return openNewFileFromMultiple(allEditors);
}
}
private IEditorPart openNewFileFromMultiple(final List<IEditorPart> allEditors) {
ListDialog listDialog = new ListDialog(window.getShell());
listDialog.setTitle(Messages.ProbeAliasAction_DialogTitle);
listDialog.setContentProvider(new ArrayContentProvider());
listDialog.setLabelProvider(new LabelProvider() {
@Override
public String getText(Object element) {
int i = (Integer) element;
return i != -1 ? allEditors.get(i).getEditorInput().getName()
: Messages.ProbeAliasAction_OtherFile;
}
});
Integer[] editorIndexes = new Integer[allEditors.size() + 1];
for (int i = 0; i < editorIndexes.length - 1; i++) {
editorIndexes[i] = i;
}
editorIndexes[editorIndexes.length - 1] = -1;
listDialog.setInput(editorIndexes);
listDialog.setMessage(Messages.ProbeAliasAction_SelectEditor);
if (listDialog.open() == Window.OK) {
int result = (Integer) listDialog.getResult()[0];
return result != -1 ? allEditors.get(result) : openNewFile();
}
// Abort if user cancels
return null;
}
private IEditorPart openNewFile() {
NewFileAction action = new NewFileAction();
action.run();
if (action.isSuccessful()) {
return window.getActivePage().getActiveEditor();
}
return null;
}
@Override
public void doubleClick(DoubleClickEvent event) {
run();
}
}
|
