selection tests
[idea/community.git] / 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   public static final String JAVA_ANONYMOUS_PROPERTY_NAME = "java.anonymous.provider";
47
48   @Override
49   public Collection<JavaAnonymousClassTreeElement> provideNodes(TreeElement node) {
50     if (node instanceof PsiMethodTreeElement || node instanceof PsiFieldTreeElement || node instanceof ClassInitializerTreeElement) {
51       final PsiElement el = ((PsiTreeElementBase)node).getElement();
52       for (AnonymousElementProvider provider : Extensions.getExtensions(AnonymousElementProvider.EP_NAME)) {
53         final PsiElement[] elements = provider.getAnonymousElements(el);
54         if (elements != null && elements.length > 0) {
55           List<JavaAnonymousClassTreeElement> result = new ArrayList<JavaAnonymousClassTreeElement>(elements.length);
56           for (PsiElement element : elements) {
57             result.add(new JavaAnonymousClassTreeElement((PsiAnonymousClass)element, new HashSet<PsiClass>()));
58           }
59           return result;
60         }
61       }
62     }
63     return Collections.emptyList();
64   }
65
66   @Override
67   public String getCheckBoxText() {
68     return "Show Anonymous Classes";
69   }
70
71   @Override
72   public Shortcut[] getShortcut() {
73     return new Shortcut[]{KeyboardShortcut.fromString(SystemInfo.isMac ? "meta I" : "control I")};
74   }
75
76   @NotNull
77   @Override
78   public ActionPresentation getPresentation() {
79     return new ActionPresentationData(getCheckBoxText(), null, PlatformIcons.ANONYMOUS_CLASS_ICON);
80   }
81
82   @NotNull
83   @Override
84   public String getName() {
85     return ID;
86   }
87
88   @NotNull
89   @Override
90   public String getPropertyName() {
91     return JAVA_ANONYMOUS_PROPERTY_NAME;
92   }
93 }