java/java-impl/src/com/intellij/ide/navigationToolbar/JavaNavBarExtension.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 /*
  18  * User: anna
  19  * Date: 04-Feb-2008
  20  */
  21 package com.intellij.ide.navigationToolbar;
  22
  23 import com.intellij.analysis.AnalysisScopeBundle;
  24 import com.intellij.lang.StdLanguages;
  25 import com.intellij.psi.*;
  26 import com.intellij.psi.presentation.java.ClassPresentationUtil;
  27 import com.intellij.openapi.roots.ProjectFileIndex;
  28 import com.intellij.openapi.roots.ProjectRootManager;
  29 import com.intellij.openapi.vfs.VirtualFile;
  30 import org.jetbrains.annotations.Nullable;
  31
  32 public class JavaNavBarExtension implements NavBarModelExtension{
  33   public String getPresentableText(final Object object) {
  34     if (object instanceof PsiClass) {
  35       return ClassPresentationUtil.getNameForClass((PsiClass)object, false);
  36     }
  37     else if (object instanceof PsiPackage) {
  38       final String name = ((PsiPackage)object).getName();
  39       return name != null ? name : AnalysisScopeBundle.message("dependencies.tree.node.default.package.abbreviation");
  40     }
  41     return null;
  42   }
  43
  44   public PsiElement getParent(final PsiElement psiElement) {
  45     if (psiElement instanceof PsiPackage) {
  46       final PsiPackage parentPackage = ((PsiPackage)psiElement).getParentPackage();
  47       if (parentPackage != null && parentPackage.getQualifiedName().length() > 0) {
  48         return parentPackage;
  49       }
  50     }
  51     return null;
  52   }
  53
  54
  55   @Nullable
  56   public PsiElement adjustElement(final PsiElement psiElement) {
  57     final ProjectFileIndex index = ProjectRootManager.getInstance(psiElement.getProject()).getFileIndex();
  58     final PsiFile containingFile = psiElement.getContainingFile();
  59     if (containingFile != null) {
  60       final VirtualFile file = containingFile.getVirtualFile();
  61       if (file != null && (index.isInSourceContent(file) || index.isInLibraryClasses(file) || index.isInLibrarySource(file))) {
  62         if (psiElement instanceof PsiJavaFile) {
  63           final PsiJavaFile psiJavaFile = (PsiJavaFile)psiElement;
  64           if (psiJavaFile.getViewProvider().getBaseLanguage() == StdLanguages.JAVA) {
  65             final PsiClass[] psiClasses = psiJavaFile.getClasses();
  66             if (psiClasses.length == 1) {
  67               return psiClasses[0];
  68             }
  69           }
  70         }
  71         if (psiElement instanceof PsiClass) return psiElement;
  72       }
  73       return containingFile;
  74     }
  75     return psiElement.isPhysical() ? psiElement : null;
  76   }
  77 }