lang-impl/src/com/intellij/ide/projectView/impl/nodes/PsiFileNode.java

   1 package com.intellij.ide.projectView.impl.nodes;
   2
   3 import com.intellij.ide.projectView.PresentationData;
   4 import com.intellij.ide.projectView.ViewSettings;
   5 import com.intellij.ide.util.treeView.AbstractTreeNode;
   6 import com.intellij.openapi.fileTypes.StdFileTypes;
   7 import com.intellij.openapi.project.Project;
   8 import com.intellij.openapi.util.Iconable;
   9 import com.intellij.openapi.vfs.VirtualFile;
  10 import com.intellij.psi.PsiFile;
  11
  12 import java.util.ArrayList;
  13 import java.util.Collection;
  14
  15 public class PsiFileNode extends BasePsiNode<PsiFile>{
  16
  17   public PsiFileNode(Project project, PsiFile value, ViewSettings viewSettings) {
  18     super(project, value, viewSettings);
  19   }
  20
  21   public Collection<AbstractTreeNode> getChildrenImpl() {
  22     return new ArrayList<AbstractTreeNode>();
  23   }
  24
  25   protected void updateImpl(PresentationData data) {
  26     final PsiFile value = getValue();
  27     data.setPresentableText(value.getName());
  28     data.setIcons(value.getIcon(Iconable.ICON_FLAG_READ_STATUS));
  29   }
  30
  31   public VirtualFile getVirtualFile() {
  32     return getValue().getVirtualFile();
  33   }
  34
  35   public int getWeight() {
  36     return 20;
  37   }
  38
  39   @Override
  40   public String getTitle() {
  41     final PsiFile file = getValue();
  42     if (file != null) {
  43       return file.getVirtualFile().getPresentableUrl();
  44     }
  45     return super.getTitle();
  46   }
  47
  48   @Override
  49   protected boolean isMarkReadOnly() {
  50     return true;
  51   }
  52
  53   public Comparable getTypeSortKey() {
  54     String extension = extension(getValue());
  55     return extension == null ? null : new ExtensionSortKey(extension);
  56   }
  57
  58   public static String extension(final PsiFile file) {
  59     return file == null || file.getVirtualFile() == null ? null : file.getVirtualFile().getFileType().getDefaultExtension();
  60   }
  61
  62   public static class ExtensionSortKey implements Comparable {
  63     private String myExtension;
  64
  65     public ExtensionSortKey(final String extension) {
  66       myExtension = extension;
  67     }
  68
  69     public int compareTo(final Object o) {
  70       if (!(o instanceof ExtensionSortKey)) return 0;
  71       ExtensionSortKey rhs = (ExtensionSortKey) o;
  72       return myExtension.compareTo(rhs.myExtension);
  73     }
  74   }
  75
  76   @Override
  77   public boolean shouldDrillDownOnEmptyElement() {
  78     final PsiFile file = getValue();
  79     return file != null && file.getFileType() == StdFileTypes.JAVA;
  80   }
  81
  82   public boolean canRepresent(final Object element) {
  83     return super.canRepresent(element) || getValue().getVirtualFile() == element;
  84   }
  85 }