java/idea-ui/src/com/intellij/facet/impl/ui/actions/AddFacetAction.java

   1 /*
   2  * Copyright 2000-2009 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.facet.impl.ui.actions;
  18
  19 import com.intellij.facet.FacetInfo;
  20 import com.intellij.facet.FacetType;
  21 import com.intellij.facet.FacetTypeId;
  22 import com.intellij.facet.impl.ui.FacetEditorFacade;
  23 import com.intellij.openapi.actionSystem.AnAction;
  24 import com.intellij.openapi.actionSystem.AnActionEvent;
  25 import com.intellij.openapi.diagnostic.Logger;
  26 import com.intellij.openapi.module.ModuleType;
  27 import com.intellij.openapi.project.DumbAware;
  28
  29 import java.util.Collection;
  30
  31 /**
  32  * @author nik
  33 */
  34 public class AddFacetAction extends AnAction implements DumbAware {
  35   private static final Logger LOG = Logger.getInstance("#com.intellij.facet.impl.ui.actions.AddFacetAction");
  36   private final FacetEditorFacade myEditor;
  37   private final FacetType myType;
  38
  39   public AddFacetAction(final FacetEditorFacade editor, final FacetType type) {
  40     super(type.getPresentableName(), null, type.getIcon());
  41     myEditor = editor;
  42     myType = type;
  43   }
  44
  45   public void actionPerformed(AnActionEvent e) {
  46     FacetInfo parent = myEditor.getSelectedFacetInfo();
  47     final Collection<FacetInfo> facetInfos = myEditor.getFacetsByType(myType);
  48     String facetName = myType.getDefaultFacetName();
  49     int i = 2;
  50     while (facetExists(facetName, facetInfos)) {
  51       facetName = myType.getPresentableName() + i;
  52       i++;
  53     }
  54     final FacetTypeId<?> underlyingFacetType = myType.getUnderlyingFacetType();
  55     if (parent == null && underlyingFacetType == null || parent != null && parent.getFacetType().getId() == underlyingFacetType) {
  56       myEditor.createFacet(parent, myType, facetName);
  57     }
  58     else {
  59       LOG.assertTrue(parent != null);
  60       final FacetInfo grandParent = myEditor.getParent(parent);
  61       LOG.assertTrue(grandParent == null && underlyingFacetType == null ||
  62                      grandParent != null && grandParent.getFacetType().getId() == underlyingFacetType);
  63       myEditor.createFacet(grandParent, myType, facetName);
  64     }
  65   }
  66
  67   private static boolean facetExists(final String facetName, final Collection<FacetInfo> facetInfos) {
  68     for (FacetInfo facetInfo : facetInfos) {
  69       if (facetInfo.getName().equals(facetName)) {
  70         return true;
  71       }
  72     }
  73     return false;
  74   }
  75
  76   public void update(AnActionEvent e) {
  77     e.getPresentation().setVisible(isVisible(myEditor, myType));
  78   }
  79
  80   public static boolean isVisible(FacetEditorFacade editor, final FacetType<?, ?> type) {
  81     final ModuleType moduleType = editor.getSelectedModuleType();
  82     if (moduleType == null || !type.isSuitableModuleType(moduleType)) {
  83       return false;
  84     }
  85
  86     final FacetTypeId<?> underlyingTypeId = type.getUnderlyingFacetType();
  87     final FacetInfo selectedFacet = editor.getSelectedFacetInfo();
  88     if (selectedFacet == null) {
  89       return underlyingTypeId == null && canAddFacet(null, type, editor);
  90     }
  91
  92     final FacetTypeId selectedFacetType = selectedFacet.getFacetType().getId();
  93     if (selectedFacetType == underlyingTypeId) {
  94       return canAddFacet(selectedFacet, type, editor);
  95     }
  96
  97     final FacetInfo parent = editor.getParent(selectedFacet);
  98     if (!canAddFacet(parent, type, editor)) {
  99       return false;
 100     }
 101     return parent == null && underlyingTypeId == null || parent != null && parent.getFacetType().getId() == underlyingTypeId;
 102   }
 103
 104   private static boolean canAddFacet(final FacetInfo selectedFacet, final FacetType<?, ?> type, final FacetEditorFacade editor) {
 105     return !(type.isOnlyOneFacetAllowed() && editor.nodeHasFacetOfType(selectedFacet, type.getId()));
 106   }
 107 }