AddModuleToVcs_Action.java

package jetbrains.mps.vcs.plugin;

/*Generated by MPS */

import jetbrains.mps.workbench.action.BaseAction;
import javax.swing.Icon;
import jetbrains.mps.icons.MPSIcons;
import org.jetbrains.annotations.NotNull;
import com.intellij.openapi.actionSystem.AnActionEvent;
import java.util.Map;
import com.intellij.openapi.actionSystem.Presentation;
import jetbrains.mps.ide.actions.MPSCommonDataKeys;
import jetbrains.mps.internal.collections.runtime.ListSequence;
import jetbrains.mps.vcs.platform.actions.VcsActionsUtil;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import com.intellij.openapi.project.Project;
import java.util.List;
import org.jetbrains.mps.openapi.module.SModule;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.openapi.vcs.changes.ChangeListManagerImpl;

public class AddModuleToVcs_Action extends BaseAction {
  private static final Icon ICON = MPSIcons.Actions.AddToVCS;

  public AddModuleToVcs_Action() {
    super("Add Module to VCS", "", ICON);
    this.setIsAlwaysVisible(false);
    this.setExecuteOutsideCommand(true);
  }
  @Override
  public boolean isDumbAware() {
    return true;
  }
  @Override
  public void doUpdate(@NotNull AnActionEvent event, final Map<String, Object> _params) {
    Presentation presentation = event.getPresentation();
    presentation.setText(String.format("Add %s to VCS", (event.getData(MPSCommonDataKeys.MODULES).size() == 1 ? "Module" : "Modules")));
    boolean enabled = ListSequence.fromList(VcsActionsUtil.getUnversionedFilesForModules(event.getData(CommonDataKeys.PROJECT), event.getData(MPSCommonDataKeys.MODULES))).isNotEmpty();
    presentation.setEnabled(enabled);
    presentation.setVisible(enabled);
  }
  @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;
      }
    }
    {
      List<SModule> p = event.getData(MPSCommonDataKeys.MODULES);
      if (p == null) {
        return false;
      }
    }
    return true;
  }
  @Override
  public void doExecute(@NotNull final AnActionEvent event, final Map<String, Object> _params) {
    List<VirtualFile> unversionedFiles = VcsActionsUtil.getUnversionedFilesForModules(event.getData(CommonDataKeys.PROJECT), event.getData(MPSCommonDataKeys.MODULES));
    ChangeListManagerImpl changeListManager = ChangeListManagerImpl.getInstanceImpl(event.getData(CommonDataKeys.PROJECT));
    changeListManager.addUnversionedFiles(changeListManager.getDefaultChangeList(), unversionedFiles);
  }
}