package jetbrains.mps.ide.actions;
/*Generated by MPS */
import jetbrains.mps.workbench.action.BaseAction;
import javax.swing.Icon;
import com.intellij.openapi.actionSystem.AnActionEvent;
import java.util.Map;
import java.util.List;
import org.jetbrains.mps.openapi.module.SModule;
import jetbrains.mps.internal.collections.runtime.MapSequence;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import jetbrains.mps.project.MPSProject;
import org.jetbrains.annotations.NotNull;
import jetbrains.mps.baseLanguage.closures.runtime.Wrappers;
import org.jetbrains.mps.openapi.module.SRepository;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.progress.ProgressIndicator;
import jetbrains.mps.progress.ProgressMonitorAdapter;
import com.intellij.util.WaitForProgressToShow;
import jetbrains.mps.project.OptimizeImportsHelper;
import com.intellij.openapi.application.ApplicationManager;
import jetbrains.mps.project.Solution;
import jetbrains.mps.smodel.Language;
import com.intellij.openapi.application.ModalityState;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.ui.Messages;
public class OptimizeModuleImports_Action extends BaseAction {
private static final Icon ICON = null;
public OptimizeModuleImports_Action() {
super("Optimize Imports", "", ICON);
this.setIsAlwaysVisible(false);
this.setExecuteOutsideCommand(true);
}
@Override
public boolean isDumbAware() {
return true;
}
@Override
protected boolean collectActionData(AnActionEvent event, final Map<String, Object> _params) {
if (!(super.collectActionData(event, _params))) {
return false;
}
{
List<SModule> p = event.getData(MPSCommonDataKeys.MODULES);
MapSequence.fromMap(_params).put("modules", p);
if (p == null) {
return false;
}
}
{
Project p = event.getData(CommonDataKeys.PROJECT);
MapSequence.fromMap(_params).put("ideaProject", p);
if (p == null) {
return false;
}
}
{
MPSProject p = event.getData(MPSCommonDataKeys.MPS_PROJECT);
MapSequence.fromMap(_params).put("project", p);
if (p == null) {
return false;
}
}
return true;
}
@Override
public void doExecute(@NotNull final AnActionEvent event, final Map<String, Object> _params) {
final Wrappers._T<String> report = new Wrappers._T<String>("");
final SRepository repo = ((MPSProject) MapSequence.fromMap(_params).get("project")).getRepository();
final Task.Modal task = new Task.Modal(((Project) MapSequence.fromMap(_params).get("ideaProject")), "Optimizing module imports", true) {
public void run(@NotNull ProgressIndicator indicator) {
ProgressMonitorAdapter monitor = new ProgressMonitorAdapter(indicator);
try {
int modulesNumber = ((List<SModule>) MapSequence.fromMap(_params).get("modules")).size();
monitor.start("Optimizing the imports of the " + modulesNumber + " modules", modulesNumber + 1);
WaitForProgressToShow.runOrInvokeAndWaitAboveProgress(new Runnable() {
public void run() {
}
});
final OptimizeImportsHelper helper = new OptimizeImportsHelper(repo);
for (final SModule module : ((List<SModule>) MapSequence.fromMap(_params).get("modules"))) {
monitor.step("Optimizing imports of the " + module);
ApplicationManager.getApplication().invokeAndWait(new Runnable() {
public void run() {
repo.getModelAccess().executeCommand(new Runnable() {
public void run() {
if (module instanceof Solution) {
report.value += helper.optimizeSolutionImports(((Solution) module));
} else if (module instanceof Language) {
report.value += helper.optimizeLanguageImports(((Language) module));
}
}
});
}
}, ModalityState.defaultModalityState());
monitor.advance(1);
if (monitor.isCanceled()) {
return;
}
}
monitor.step("Saving...");
WaitForProgressToShow.runOrInvokeAndWaitAboveProgress(new Runnable() {
public void run() {
repo.getModelAccess().executeCommand(new Runnable() {
public void run() {
repo.saveAll();
}
});
}
});
monitor.advance(1);
} finally {
monitor.done();
}
}
};
ApplicationManager.getApplication().invokeLater(new Runnable() {
public void run() {
ProgressManager.getInstance().run(task);
Messages.showMessageDialog(((Project) MapSequence.fromMap(_params).get("ideaProject")), (report.value.equals("") ? "Nothing to optimize" : report.value), "Optimize Imports", Messages.getInformationIcon());
}
}, ModalityState.defaultModalityState());
}
}