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
|
package com.intellij.execution.runners;
import com.intellij.execution.ExecutionManager;
import com.intellij.execution.ExecutorRegistry;
import com.intellij.execution.process.ProcessHandler;
import com.intellij.execution.ui.RunContentDescriptor;
import com.intellij.openapi.Disposable;
import com.intellij.openapi.actionSystem.*;
import com.intellij.openapi.project.DumbAwareAction;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.Disposer;
import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull;
import java.util.List;
/**
* Reruns all registered execution sessions.<p>
* The difference between this action and {@code Rerun} action (Ctrl+F5) is that this action reruns
* only explicitly registered execution sessions. For example, their tabs can be hidden by other tabs, it doesn't matter.
* <p>
* Thus it can be convenient for rerunning tests, because running non-test execution session after
* running test execution session won't hide the latter.
*
* @author Sergey Simonchik
*/
public class RerunTestsAction extends DumbAwareAction implements AnAction.TransparentUpdate {
public static final String ID = "RerunTests";
private static final List<RerunInfo> REGISTRY = ContainerUtil.createLockFreeCopyOnWriteList();
public static void register(@NotNull RunContentDescriptor descriptor,
@NotNull ExecutionEnvironment env,
@NotNull ProgramRunner runner) {
final RerunInfo rerunInfo = new RerunInfo(descriptor, env, runner);
REGISTRY.add(rerunInfo);
Disposer.register(descriptor, new Disposable() {
@Override
public void dispose() {
REGISTRY.remove(rerunInfo);
}
});
}
@Override
public void actionPerformed(AnActionEvent e) {
DataContext dataContext = e.getDataContext();
Project project = CommonDataKeys.PROJECT.getData(dataContext);
if (project == null) {
return;
}
ExecutionManager executionManager = ExecutionManager.getInstance(project);
for (RerunInfo rerunInfo : REGISTRY) {
RunContentDescriptor descriptor = rerunInfo.getDescriptor();
if (!Disposer.isDisposed(descriptor)) {
ExecutionEnvironment env = rerunInfo.getEnv();
ProgramRunner runner = rerunInfo.getRunner();
ProcessHandler processHandler = descriptor.getProcessHandler();
if (processHandler != null && processHandler.isProcessTerminated()) {
if (!ExecutorRegistry.getInstance().isStarting(project, env.getExecutor().getId(), runner.getRunnerId())) {
executionManager.restartRunProfile(runner, env, descriptor);
}
}
}
}
}
@Override
public void update(AnActionEvent e) {
Presentation presentation = e.getPresentation();
presentation.setEnabled(true);
}
private static class RerunInfo {
private final RunContentDescriptor myDescriptor;
private final ExecutionEnvironment myEnv;
private final ProgramRunner myRunner;
public RerunInfo(@NotNull RunContentDescriptor descriptor,
@NotNull ExecutionEnvironment env,
@NotNull ProgramRunner runner) {
myDescriptor = descriptor;
myEnv = env;
myRunner = runner;
}
private RunContentDescriptor getDescriptor() {
return myDescriptor;
}
private ExecutionEnvironment getEnv() {
return myEnv;
}
private ProgramRunner getRunner() {
return myRunner;
}
}
}
|
