lang-impl/src/com/intellij/ide/util/DirectoryChooserUtil.java

   1 package com.intellij.ide.util;
   2
   3 import com.intellij.ide.IdeBundle;
   4 import com.intellij.ide.IdeView;
   5 import com.intellij.openapi.application.ApplicationManager;
   6 import com.intellij.openapi.project.Project;
   7 import com.intellij.openapi.roots.ProjectFileIndex;
   8 import com.intellij.openapi.roots.ProjectRootManager;
   9 import com.intellij.psi.PsiDirectory;
  10 import com.intellij.refactoring.RefactoringBundle;
  11 import org.jetbrains.annotations.Nullable;
  12
  13 import java.util.ArrayList;
  14 import java.util.Map;
  15
  16 public class DirectoryChooserUtil {
  17   private DirectoryChooserUtil() {
  18   }
  19
  20   public static PsiDirectory getOrChooseDirectory(IdeView view) {
  21     PsiDirectory[] dirs = view.getDirectories();
  22     if (dirs.length == 0) return null;
  23     if (dirs.length == 1) {
  24       return dirs[0];
  25     }
  26     else {
  27       Project project = dirs[0].getProject();
  28       return selectDirectory(project, dirs, null, "");
  29     }
  30   }
  31
  32   public static PsiDirectory selectDirectory(Project project,
  33                                              PsiDirectory[] packageDirectories,
  34                                              PsiDirectory defaultDirectory,
  35                                              String postfixToShow) {
  36     ProjectFileIndex projectFileIndex = ProjectRootManager.getInstance(project).getFileIndex();
  37
  38     ArrayList<PsiDirectory> possibleDirs = new ArrayList<PsiDirectory>();
  39     for (PsiDirectory dir : packageDirectories) {
  40       if (!dir.isValid()) continue;
  41       if (!dir.isWritable()) continue;
  42       if (possibleDirs.contains(dir)) continue;
  43       if (!projectFileIndex.isInContent(dir.getVirtualFile())) continue;
  44       possibleDirs.add(dir);
  45     }
  46
  47     if (possibleDirs.isEmpty()) return null;
  48     if (possibleDirs.size() == 1) return possibleDirs.get(0);
  49
  50     if (ApplicationManager.getApplication().isUnitTestMode()) return possibleDirs.get(0);
  51
  52     DirectoryChooser chooser = new DirectoryChooser(project);
  53     chooser.setTitle(IdeBundle.message("title.choose.destination.directory"));
  54     chooser.fillList(possibleDirs.toArray(new PsiDirectory[possibleDirs.size()]), defaultDirectory, project, postfixToShow);
  55     chooser.show();
  56     return chooser.isOK() ? chooser.getSelectedDirectory() : null;
  57   }
  58
  59   @Nullable
  60   public static
  61   PsiDirectory chooseDirectory(PsiDirectory[] targetDirectories,
  62                                                        @Nullable PsiDirectory initialDirectory, Project project,
  63                                                        Map<PsiDirectory, String> relativePathsToCreate) {
  64     final DirectoryChooser chooser = new DirectoryChooser(project, new DirectoryChooserModuleTreeView(project));
  65     chooser.setTitle(RefactoringBundle.message("choose.destination.directory"));
  66     chooser.fillList(
  67       targetDirectories,
  68       initialDirectory,
  69       project,
  70       relativePathsToCreate
  71     );
  72     chooser.show();
  73     if (!chooser.isOK()) return null;
  74     return chooser.getSelectedDirectory();
  75   }
  76 }