platform/lang-impl/src/com/intellij/openapi/roots/ui/configuration/actions/ContentEntryEditingAction.java

   1 package com.intellij.openapi.roots.ui.configuration.actions;
   2
   3 import com.intellij.openapi.actionSystem.*;
   4 import com.intellij.openapi.actionSystem.ex.CustomComponentAction;
   5 import com.intellij.openapi.actionSystem.impl.ActionButtonWithText;
   6 import com.intellij.openapi.fileChooser.ex.FileNodeDescriptor;
   7 import com.intellij.openapi.project.DumbAware;
   8 import com.intellij.openapi.vfs.VirtualFile;
   9 import org.jetbrains.annotations.Nullable;
  10
  11 import javax.swing.*;
  12 import javax.swing.tree.DefaultMutableTreeNode;
  13 import javax.swing.tree.TreePath;
  14
  15 /**
  16  * @author Eugene Zhuravlev
  17  *         Date: Oct 14
  18  * @author 2003
  19  *         Time: 3:07:14 PM
  20  */
  21 public abstract class ContentEntryEditingAction extends ToggleAction implements CustomComponentAction, DumbAware {
  22   protected final JTree myTree;
  23
  24   protected ContentEntryEditingAction(JTree tree) {
  25     myTree = tree;
  26     getTemplatePresentation().setEnabled(true);
  27   }
  28
  29   public void update(AnActionEvent e) {
  30     super.update(e);
  31     final Presentation presentation = e.getPresentation();
  32     presentation.setEnabled(true);
  33     final VirtualFile[] files = getSelectedFiles();
  34     if (files == null || files.length == 0) {
  35       presentation.setEnabled(false);
  36       return;
  37     }
  38     for (VirtualFile file : files) {
  39       if (file == null || !file.isDirectory()) {
  40         presentation.setEnabled(false);
  41         break;
  42       }
  43     }
  44   }
  45
  46   @Nullable
  47   protected final VirtualFile[] getSelectedFiles() {
  48     final TreePath[] selectionPaths = myTree.getSelectionPaths();
  49     if (selectionPaths == null) {
  50       return null;
  51     }
  52     final VirtualFile[] selected = new VirtualFile[selectionPaths.length];
  53     for (int i = 0; i < selectionPaths.length; i++) {
  54       TreePath treePath = selectionPaths[i];
  55       final DefaultMutableTreeNode node = (DefaultMutableTreeNode)treePath.getLastPathComponent();
  56       final Object nodeDescriptor = node.getUserObject();
  57       if (!(nodeDescriptor instanceof FileNodeDescriptor)) {
  58         return null;
  59       }
  60       selected[i] = ((FileNodeDescriptor)nodeDescriptor).getElement().getFile();
  61     }
  62     return selected;
  63   }
  64
  65   public JComponent createCustomComponent(Presentation presentation) {
  66     return new ActionButtonWithText(this, presentation, ActionPlaces.UNKNOWN, ActionToolbar.DEFAULT_MINIMUM_BUTTON_SIZE);
  67   }
  68 }