java/java-impl/src/com/intellij/ide/structureView/impl/java/JavaAnonymousClassesNodeProvider.java

   1 /*
   2  * Copyright 2000-2011 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 package com.intellij.ide.structureView.impl.java;
  17
  18 import com.intellij.ide.structureView.impl.common.PsiTreeElementBase;
  19 import com.intellij.ide.util.FileStructureNodeProvider;
  20 import com.intellij.ide.util.treeView.smartTree.ActionPresentation;
  21 import com.intellij.ide.util.treeView.smartTree.ActionPresentationData;
  22 import com.intellij.ide.util.treeView.smartTree.TreeElement;
  23 import com.intellij.navigation.AnonymousElementProvider;
  24 import com.intellij.openapi.actionSystem.KeyboardShortcut;
  25 import com.intellij.openapi.actionSystem.Shortcut;
  26 import com.intellij.openapi.extensions.Extensions;
  27 import com.intellij.openapi.util.PropertyOwner;
  28 import com.intellij.openapi.util.SystemInfo;
  29 import com.intellij.psi.PsiAnonymousClass;
  30 import com.intellij.psi.PsiClass;
  31 import com.intellij.psi.PsiElement;
  32 import com.intellij.util.PlatformIcons;
  33 import com.intellij.util.containers.hash.HashSet;
  34 import org.jetbrains.annotations.NotNull;
  35
  36 import java.util.ArrayList;
  37 import java.util.Collection;
  38 import java.util.Collections;
  39 import java.util.List;
  40
  41 /**
  42  * @author Konstantin Bulenkov
  43  */
  44 public class JavaAnonymousClassesNodeProvider implements FileStructureNodeProvider<JavaAnonymousClassTreeElement>, PropertyOwner {
  45   public static final String ID = "SHOW_ANONYMOUS";
  46
  47   @Override
  48   public Collection<JavaAnonymousClassTreeElement> provideNodes(TreeElement node) {
  49     if (node instanceof PsiMethodTreeElement || node instanceof PsiFieldTreeElement || node instanceof ClassInitializerTreeElement) {
  50       final PsiElement el = ((PsiTreeElementBase)node).getElement();
  51       for (AnonymousElementProvider provider : Extensions.getExtensions(AnonymousElementProvider.EP_NAME)) {
  52         final PsiElement[] elements = provider.getAnonymousElements(el);
  53         if (elements != null && elements.length > 0) {
  54           List<JavaAnonymousClassTreeElement> result = new ArrayList<JavaAnonymousClassTreeElement>(elements.length);
  55           for (PsiElement element : elements) {
  56             result.add(new JavaAnonymousClassTreeElement((PsiAnonymousClass)element, new HashSet<PsiClass>()));
  57           }
  58           return result;
  59         }
  60       }
  61     }
  62     return Collections.emptyList();
  63   }
  64
  65   @Override
  66   public String getCheckBoxText() {
  67     return "Show Anonymous Classes";
  68   }
  69
  70   @Override
  71   public Shortcut[] getShortcut() {
  72     return new Shortcut[]{KeyboardShortcut.fromString(SystemInfo.isMac ? "meta I" : "control I")};
  73   }
  74
  75   @NotNull
  76   @Override
  77   public ActionPresentation getPresentation() {
  78     return new ActionPresentationData(getCheckBoxText(), null, PlatformIcons.ANONYMOUS_CLASS_ICON);
  79   }
  80
  81   @NotNull
  82   @Override
  83   public String getName() {
  84     return ID;
  85   }
  86
  87   @NotNull
  88   @Override
  89   public String getPropertyName() {
  90     return "java.anonymous.provider";
  91   }
  92 }