package jetbrains.mps.ide.actions;
/*Generated by MPS */
import jetbrains.mps.workbench.action.BaseAction;
import org.apache.log4j.Logger;
import org.apache.log4j.LogManager;
import javax.swing.Icon;
import jetbrains.mps.icons.MPSIcons;
import com.intellij.openapi.actionSystem.AnActionEvent;
import java.util.Map;
import jetbrains.mps.ide.ui.tree.smodel.SModelTreeNode;
import org.jetbrains.mps.openapi.model.SModelName;
import jetbrains.mps.smodel.SModelStereotype;
import org.jetbrains.annotations.NotNull;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import jetbrains.mps.project.MPSProject;
import org.jetbrains.mps.openapi.model.SModel;
import javax.swing.tree.TreeNode;
import jetbrains.mps.project.SModuleOperations;
import org.apache.log4j.Level;
import jetbrains.mps.smodel.ModelImports;
import jetbrains.mps.ide.projectPane.ProjectPane;
import java.util.List;
import jetbrains.mps.util.IterableUtil;
import jetbrains.mps.ide.ui.tree.SortUtil;
public class NewSubTestModel_Action extends BaseAction {
private static final Logger LOG = LogManager.getLogger(NewSubTestModel_Action.class);
private static final Icon ICON = MPSIcons.Nodes.TestModel;
public NewSubTestModel_Action() {
super("Test Model", "", ICON);
this.setIsAlwaysVisible(false);
this.setExecuteOutsideCommand(true);
}
@Override
public boolean isDumbAware() {
return true;
}
@Override
public boolean isApplicable(AnActionEvent event, final Map<String, Object> _params) {
if (!(event.getData(MPSCommonDataKeys.TREE_NODE) instanceof SModelTreeNode)) {
return false;
}
return !(event.getData(MPSCommonDataKeys.CONTEXT_MODEL).getName().hasStereotype()) && event.getData(MPSCommonDataKeys.CONTEXT_MODEL).getModelRoot().canCreateModel(new SModelName(event.getData(MPSCommonDataKeys.CONTEXT_MODEL).getName().getLongName(), SModelStereotype.TESTS).getValue());
}
@Override
public void doUpdate(@NotNull AnActionEvent event, final Map<String, Object> _params) {
this.setEnabledState(event.getPresentation(), this.isApplicable(event, _params));
}
@Override
protected boolean collectActionData(AnActionEvent event, final Map<String, Object> _params) {
if (!(super.collectActionData(event, _params))) {
return false;
}
{
Project p = event.getData(CommonDataKeys.PROJECT);
if (p == null) {
return false;
}
}
{
MPSProject p = event.getData(MPSCommonDataKeys.MPS_PROJECT);
if (p == null) {
return false;
}
}
{
SModel p = event.getData(MPSCommonDataKeys.CONTEXT_MODEL);
if (p == null) {
return false;
}
}
{
TreeNode p = event.getData(MPSCommonDataKeys.TREE_NODE);
if (p == null) {
return false;
}
}
return true;
}
@Override
public void doExecute(@NotNull final AnActionEvent event, final Map<String, Object> _params) {
event.getData(MPSCommonDataKeys.MPS_PROJECT).getModelAccess().executeCommand(new Runnable() {
public void run() {
SModelName testModelName = new SModelName(NewSubTestModel_Action.this.getTestModelName(event), SModelStereotype.TESTS);
SModel parentModel = event.getData(MPSCommonDataKeys.CONTEXT_MODEL);
SModel createdModel = SModuleOperations.createModelWithAdjustments(testModelName.getValue(), parentModel.getModelRoot());
if (createdModel == null) {
if (LOG.isEnabledFor(Level.WARN)) {
LOG.warn("Can't create submodel " + testModelName + " for model " + parentModel.getName());
}
return;
}
ModelImports imports = new ModelImports(createdModel);
imports.addModelImport(parentModel.getReference());
imports.copyImportedModelsFrom(parentModel);
imports.copyUsedLanguagesFrom(parentModel);
imports.copyEmployedDevKitsFrom(parentModel);
imports.copyLanguageEngagedOnGeneration(parentModel);
ProjectPane.getInstance(event.getData(CommonDataKeys.PROJECT)).selectModel(createdModel, false);
}
});
}
/*package*/ String getTestModelName(final AnActionEvent event) {
StringBuilder builder = new StringBuilder();
String modelBaseName = event.getData(MPSCommonDataKeys.CONTEXT_MODEL).getName().getLongName();
builder.append(modelBaseName);
int testModelCount = 0;
List<SModel> models = IterableUtil.asList(event.getData(MPSCommonDataKeys.CONTEXT_MODEL).getModule().getModels());
List<SModel> sortedModels = SortUtil.sortModels(models);
for (SModel md : sortedModels) {
if (!(SModelStereotype.isTestModel(md))) {
continue;
}
String name = (testModelCount == 0 ? modelBaseName : modelBaseName + testModelCount);
if (name.equals(md.getName().getLongName())) {
testModelCount++;
}
}
if (testModelCount != 0) {
builder.append(testModelCount);
}
return builder.toString();
}
}