platform/platform-impl/src/com/intellij/ide/actions/OpenProjectFileChooserDescriptor.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 package com.intellij.ide.actions;
  17
  18 import com.intellij.ide.highlighter.ProjectFileType;
  19 import com.intellij.ide.impl.ProjectUtil;
  20 import com.intellij.openapi.application.impl.ApplicationInfoImpl;
  21 import com.intellij.openapi.fileChooser.FileChooserDescriptor;
  22 import com.intellij.openapi.project.Project;
  23 import com.intellij.openapi.util.IconLoader;
  24 import com.intellij.openapi.vfs.VirtualFile;
  25 import com.intellij.projectImport.ProjectOpenProcessor;
  26 import org.jetbrains.annotations.Nullable;
  27
  28 import javax.swing.*;
  29
  30 public class OpenProjectFileChooserDescriptor extends FileChooserDescriptor {
  31   private final Icon myProjectIcon = IconLoader.getIcon(ApplicationInfoImpl.getInstanceEx().getSmallIconUrl());
  32
  33   public OpenProjectFileChooserDescriptor(final boolean chooseFiles) {
  34     super(chooseFiles, true, false, false, false, false);
  35   }
  36
  37   public boolean isFileSelectable(final VirtualFile file) {
  38     return isProjectDirectory(file) || isProjectFile(file);
  39   }
  40
  41   public Icon getOpenIcon(final VirtualFile virtualFile) {
  42     if (isProjectDirectory(virtualFile)) return myProjectIcon;
  43     final Icon icon = getImporterIcon(virtualFile, true);
  44     if(icon!=null){
  45       return icon;
  46     }
  47     return super.getOpenIcon(virtualFile);
  48   }
  49
  50   public Icon getClosedIcon(final VirtualFile virtualFile) {
  51     if (isProjectDirectory(virtualFile)) return myProjectIcon;
  52     final Icon icon = getImporterIcon(virtualFile, false);
  53     if(icon!=null){
  54       return icon;
  55     }
  56     return super.getClosedIcon(virtualFile);
  57   }
  58
  59   @Nullable
  60   public static Icon getImporterIcon(final VirtualFile virtualFile, final boolean open) {
  61     final ProjectOpenProcessor provider = ProjectUtil.getImportProvider(virtualFile);
  62     if(provider!=null) {
  63       return provider.getIcon();
  64     }
  65     return null;
  66   }
  67
  68   public boolean isFileVisible(final VirtualFile file, final boolean showHiddenFiles) {
  69     return isProjectFile(file) || super.isFileVisible(file, showHiddenFiles) && file.isDirectory();
  70   }
  71
  72   private static boolean isProjectFile(final VirtualFile file) {
  73     return (!file.isDirectory() && file.getName().toLowerCase().endsWith(ProjectFileType.DOT_DEFAULT_EXTENSION)) ||
  74            (ProjectUtil.getImportProvider(file) != null);
  75   }
  76
  77   private static boolean isProjectDirectory(final VirtualFile virtualFile) {
  78     // the root directory of any drive is never an IDEA project
  79     if (virtualFile.getParent() == null) return false;
  80     if (virtualFile.isDirectory() && virtualFile.isValid() && virtualFile.findChild(Project.DIRECTORY_STORE_FOLDER) != null) return true;
  81     return false;
  82   }
  83 }