dom/openapi/src/com/intellij/util/xml/ui/actions/AddDomElementAction.java

   1 /*
   2  * Copyright 2000-2007 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package com.intellij.util.xml.ui.actions;
  18
  19 import com.intellij.openapi.actionSystem.*;
  20 import com.intellij.openapi.application.ApplicationBundle;
  21 import com.intellij.openapi.project.Project;
  22 import com.intellij.openapi.ui.popup.JBPopupFactory;
  23 import com.intellij.openapi.ui.popup.ListPopup;
  24 import com.intellij.util.ReflectionUtil;
  25 import com.intellij.util.xml.*;
  26 import com.intellij.util.xml.reflect.DomCollectionChildDescription;
  27 import com.intellij.util.xml.ui.DomCollectionControl;
  28 import org.jetbrains.annotations.NotNull;
  29 import org.jetbrains.annotations.Nullable;
  30
  31 import javax.swing.*;
  32 import java.awt.*;
  33 import java.awt.event.KeyEvent;
  34 import java.lang.reflect.Type;
  35 import java.util.ArrayList;
  36 import java.util.List;
  37
  38 /**
  39  * User: Sergey.Vasiliev
  40  */
  41 public abstract class AddDomElementAction extends AnAction {
  42
  43   private final static ShortcutSet shortcutSet = new CustomShortcutSet(KeyStroke.getKeyStroke(KeyEvent.VK_INSERT, 0));
  44
  45   public AddDomElementAction() {
  46     super(ApplicationBundle.message("action.add"), null, DomCollectionControl.ADD_ICON);
  47   }
  48
  49   public void update(AnActionEvent e) {
  50     if (!isEnabled(e)) {
  51       e.getPresentation().setEnabled(false);
  52       return;
  53     }
  54
  55     final AnAction[] actions = getChildren(e);
  56     for (final AnAction action : actions) {
  57       e.getPresentation().setEnabled(true);
  58       action.update(e);
  59       if (e.getPresentation().isEnabled()) {
  60         break;
  61       }
  62     }
  63     if (actions.length == 1) {
  64       e.getPresentation().setText(actions[0].getTemplatePresentation().getText());
  65     } else {
  66       final String actionText = getActionText(e);
  67       if (!actionText.endsWith("...")) {
  68         e.getPresentation().setText(actionText + (actions.length > 1 ? "..." : ""));
  69       }
  70     }
  71     e.getPresentation().setIcon(DomCollectionControl.ADD_ICON);
  72
  73     super.update(e);
  74   }
  75
  76   public void actionPerformed(AnActionEvent e) {
  77     final AnAction[] actions = getChildren(e);
  78     if (actions.length > 1) {
  79       final DefaultActionGroup group = new DefaultActionGroup();
  80       for (final AnAction action : actions) {
  81         group.add(action);
  82       }
  83
  84       final DataContext dataContext = e.getDataContext();
  85       final ListPopup groupPopup =
  86         JBPopupFactory.getInstance().createActionGroupPopup(null,//J2EEBundle.message("label.menu.title.add.activation.config.property"),
  87                                                             group, dataContext, JBPopupFactory.ActionSelectionAid.NUMBERING, true);
  88
  89       showPopup(groupPopup, e);
  90     }
  91     else if (actions.length == 1) {
  92       actions[0].actionPerformed(e);
  93     }
  94   }
  95
  96   protected String getActionText(final AnActionEvent e) {
  97     return e.getPresentation().getText();
  98   }
  99
 100   protected boolean isEnabled(final AnActionEvent e) {
 101     return true;
 102   }
 103
 104   protected void showPopup(final ListPopup groupPopup, final AnActionEvent e) {
 105     final Component component = e.getInputEvent().getComponent();
 106
 107     if (component instanceof ActionButtonComponent) {
 108       groupPopup.showUnderneathOf(component);
 109     } else {
 110       groupPopup.showInBestPositionFor(e.getDataContext());
 111     }
 112   }
 113
 114   @NotNull
 115   public AnAction[] getChildren(final AnActionEvent e) {
 116     Project project = PlatformDataKeys.PROJECT.getData(e.getDataContext());
 117     if (project == null) return AnAction.EMPTY_ARRAY;
 118
 119     DomCollectionChildDescription[] descriptions = getDomCollectionChildDescriptions(e);
 120     final List<AnAction> actions = new ArrayList<AnAction>();
 121     for (DomCollectionChildDescription description : descriptions) {
 122       final TypeChooser chooser = DomManager.getDomManager(project).getTypeChooserManager().getTypeChooser(description.getType());
 123       for (Type type : chooser.getChooserTypes()) {
 124
 125         final Class<?> rawType = ReflectionUtil.getRawType(type);
 126         String name = TypeNameManager.getTypeName(rawType);
 127         Icon icon = null;
 128         if (!showAsPopup() || descriptions.length == 1) {
 129 //          if (descriptions.length > 1) {
 130             icon = ElementPresentationManager.getIconForClass(rawType);
 131 //          }
 132         }
 133         actions.add(createAddingAction(e, ApplicationBundle.message("action.add") + " " + name, icon, type, description));
 134       }
 135     }
 136     if (actions.size() > 1 && showAsPopup()) {
 137       ActionGroup group = new ActionGroup() {
 138         public AnAction[] getChildren(@Nullable AnActionEvent e) {
 139           return actions.toArray(AnAction.EMPTY_ARRAY);
 140         }
 141       };
 142       return new AnAction[]{new ShowPopupAction(group)};
 143     }
 144     else {
 145       if (actions.size() > 1) {
 146         actions.add(Separator.getInstance());
 147       } else if (actions.size() == 1) {
 148
 149       }
 150     }
 151     return actions.toArray(AnAction.EMPTY_ARRAY);
 152   }
 153
 154   protected abstract AnAction createAddingAction(final AnActionEvent e,
 155                                                  final String name,
 156                                                  final Icon icon,
 157                                                  final Type type,
 158                                                  final DomCollectionChildDescription description);
 159
 160   @NotNull
 161   protected abstract DomCollectionChildDescription[] getDomCollectionChildDescriptions(final AnActionEvent e);
 162
 163   @Nullable
 164   protected abstract DomElement getParentDomElement(final AnActionEvent e);
 165
 166   protected abstract JComponent getComponent(AnActionEvent e);
 167
 168   protected boolean showAsPopup() {
 169     return true;
 170   }
 171
 172   protected class ShowPopupAction extends AnAction {
 173
 174     protected final ActionGroup myGroup;
 175
 176     protected ShowPopupAction(ActionGroup group) {
 177       super(ApplicationBundle.message("action.add"), null, DomCollectionControl.ADD_ICON);
 178       myGroup = group;
 179       setShortcutSet(shortcutSet);
 180     }
 181
 182     public void actionPerformed(AnActionEvent e) {
 183       final ListPopup groupPopup =
 184         JBPopupFactory.getInstance().createActionGroupPopup(null,//J2EEBundle.message("label.menu.title.add.activation.config.property"),
 185                                                             myGroup, e.getDataContext(), JBPopupFactory.ActionSelectionAid.NUMBERING, true);
 186
 187       showPopup(groupPopup, e);
 188     }
 189   }
 190 }