NewRuntimeModule_Action.java

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 org.jetbrains.mps.openapi.module.SModule;
import jetbrains.mps.internal.collections.runtime.MapSequence;
import jetbrains.mps.smodel.Language;
import org.jetbrains.annotations.NotNull;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import javax.swing.tree.TreeNode;
import jetbrains.mps.project.MPSProject;
import org.jetbrains.mps.openapi.module.SRepository;
import jetbrains.mps.ide.ui.tree.MPSTree;
import jetbrains.mps.ide.ui.tree.MPSTreeNode;
import jetbrains.mps.project.ModuleInstanceCondition;
import jetbrains.mps.project.Solution;
import jetbrains.mps.workbench.choose.ModuleScopeIterable;
import jetbrains.mps.scope.ConditionalScope;
import jetbrains.mps.FilteredGlobalScope;
import jetbrains.mps.workbench.choose.ChooseByNameData;
import org.jetbrains.mps.openapi.module.SModuleReference;
import jetbrains.mps.workbench.choose.ModulesPresentation;
import com.intellij.ide.util.gotoByName.ChooseByNamePopup;
import jetbrains.mps.workbench.goTo.ui.MpsPopupFactory;
import com.intellij.ide.util.gotoByName.ChooseByNamePopupComponent;
import com.intellij.openapi.application.ModalityState;

public class NewRuntimeModule_Action extends BaseAction {
  private static final Icon ICON = null;

  public NewRuntimeModule_Action() {
    super("New Runtime Module", "", ICON);
    this.setIsAlwaysVisible(false);
    this.setExecuteOutsideCommand(false);
  }
  @Override
  public boolean isDumbAware() {
    return true;
  }
  @Override
  public boolean isApplicable(AnActionEvent event, final Map<String, Object> _params) {
    return ((SModule) MapSequence.fromMap(_params).get("contextModule")) instanceof Language;
  }
  @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;
    }
    {
      SModule p = event.getData(MPSCommonDataKeys.CONTEXT_MODULE);
      MapSequence.fromMap(_params).put("contextModule", p);
      if (p == null) {
        return false;
      }
    }
    {
      Project p = event.getData(CommonDataKeys.PROJECT);
      MapSequence.fromMap(_params).put("ideaProject", p);
      if (p == null) {
        return false;
      }
    }
    {
      TreeNode p = event.getData(MPSCommonDataKeys.TREE_NODE);
      MapSequence.fromMap(_params).put("treeNode", 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 SRepository repo = ((MPSProject) MapSequence.fromMap(_params).get("project")).getRepository();

    final Language language = (Language) ((SModule) MapSequence.fromMap(_params).get("contextModule"));
    final MPSTree mpsTree = ((MPSTreeNode) ((TreeNode) MapSequence.fromMap(_params).get("treeNode"))).getTree();

    ModuleInstanceCondition solutionOnly = new ModuleInstanceCondition(Solution.class);
    ModuleScopeIterable localScope = new ModuleScopeIterable(new ConditionalScope(((MPSProject) MapSequence.fromMap(_params).get("project")).getScope(), solutionOnly, null), repo);
    ModuleScopeIterable globalScope = new ModuleScopeIterable(new ConditionalScope(new FilteredGlobalScope(), solutionOnly, null), repo);

    ChooseByNameData<SModuleReference> gotoData = new ChooseByNameData<SModuleReference>(new ModulesPresentation(repo));
    gotoData.derivePrompts("runtime module").setScope(localScope, globalScope);

    ChooseByNamePopup popup = MpsPopupFactory.createPackagePopup(((Project) MapSequence.fromMap(_params).get("ideaProject")), gotoData, NewRuntimeModule_Action.this);
    popup.invoke(new ChooseByNamePopupComponent.Callback() {
      @Override
      public void elementChosen(final Object p0) {
        if (!(p0 instanceof SModuleReference)) {
          return;
        }
        repo.getModelAccess().runWriteInEDT(new Runnable() {
          public void run() {
            language.getModuleDescriptor().getRuntimeModules().add((SModuleReference) p0);
            language.save();
            mpsTree.rebuildLater();
          }
        });
      }
    }, ModalityState.current(), true);
  }
}