| /* |
| * Copyright 2000-2014 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.intellij.ide.structureView.impl.java; |
| |
| import com.intellij.ide.structureView.impl.common.PsiTreeElementBase; |
| import com.intellij.ide.util.FileStructureNodeProvider; |
| import com.intellij.ide.util.treeView.smartTree.ActionPresentation; |
| import com.intellij.ide.util.treeView.smartTree.ActionPresentationData; |
| import com.intellij.ide.util.treeView.smartTree.TreeElement; |
| import com.intellij.navigation.AnonymousElementProvider; |
| import com.intellij.openapi.actionSystem.KeyboardShortcut; |
| import com.intellij.openapi.actionSystem.Shortcut; |
| import com.intellij.openapi.extensions.Extensions; |
| import com.intellij.openapi.util.PropertyOwner; |
| import com.intellij.openapi.util.SystemInfo; |
| import com.intellij.psi.PsiAnonymousClass; |
| import com.intellij.psi.PsiClass; |
| import com.intellij.psi.PsiElement; |
| import com.intellij.util.PlatformIcons; |
| import com.intellij.util.containers.hash.HashSet; |
| import org.jetbrains.annotations.NotNull; |
| |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.List; |
| |
| /** |
| * @author Konstantin Bulenkov |
| */ |
| public class JavaAnonymousClassesNodeProvider implements FileStructureNodeProvider<JavaAnonymousClassTreeElement>, PropertyOwner { |
| public static final String ID = "SHOW_ANONYMOUS"; |
| public static final String JAVA_ANONYMOUS_PROPERTY_NAME = "java.anonymous.provider"; |
| |
| @NotNull |
| @Override |
| public Collection<JavaAnonymousClassTreeElement> provideNodes(@NotNull TreeElement node) { |
| if (node instanceof PsiMethodTreeElement || node instanceof PsiFieldTreeElement || node instanceof ClassInitializerTreeElement) { |
| final PsiElement el = ((PsiTreeElementBase)node).getElement(); |
| if (el != null) { |
| for (AnonymousElementProvider provider : Extensions.getExtensions(AnonymousElementProvider.EP_NAME)) { |
| final PsiElement[] elements = provider.getAnonymousElements(el); |
| if (elements.length > 0) { |
| List<JavaAnonymousClassTreeElement> result = new ArrayList<JavaAnonymousClassTreeElement>(elements.length); |
| for (PsiElement element : elements) { |
| result.add(new JavaAnonymousClassTreeElement((PsiAnonymousClass)element, new HashSet<PsiClass>())); |
| } |
| return result; |
| } |
| } |
| } |
| } |
| return Collections.emptyList(); |
| } |
| |
| @NotNull |
| @Override |
| public String getCheckBoxText() { |
| return "Show Anonymous Classes"; |
| } |
| |
| @NotNull |
| @Override |
| public Shortcut[] getShortcut() { |
| return new Shortcut[]{KeyboardShortcut.fromString(SystemInfo.isMac ? "meta I" : "control I")}; |
| } |
| |
| @NotNull |
| @Override |
| public ActionPresentation getPresentation() { |
| return new ActionPresentationData(getCheckBoxText(), null, PlatformIcons.ANONYMOUS_CLASS_ICON); |
| } |
| |
| @NotNull |
| @Override |
| public String getName() { |
| return ID; |
| } |
| |
| @NotNull |
| @Override |
| public String getPropertyName() { |
| return JAVA_ANONYMOUS_PROPERTY_NAME; |
| } |
| } |