platform/lang-impl/src/com/intellij/ide/impl/SelectInTargetPsiWrapper.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 package com.intellij.ide.impl;
  18
  19 import com.intellij.ide.SelectInContext;
  20 import com.intellij.ide.SelectInTarget;
  21 import com.intellij.openapi.editor.Document;
  22 import com.intellij.openapi.fileEditor.FileDocumentManager;
  23 import com.intellij.openapi.project.Project;
  24 import com.intellij.openapi.vfs.VirtualFile;
  25 import com.intellij.psi.*;
  26 import org.jetbrains.annotations.NotNull;
  27
  28
  29 public abstract class SelectInTargetPsiWrapper implements SelectInTarget {
  30   protected final Project myProject;
  31
  32   protected SelectInTargetPsiWrapper(@NotNull final Project project) {
  33     myProject = project;
  34   }
  35
  36   public abstract String toString();
  37
  38   protected abstract boolean canSelect(PsiFileSystemItem file);
  39
  40   public final boolean canSelect(SelectInContext context) {
  41     if (myProject.isDisposed()) return false;
  42
  43     VirtualFile virtualFile = context.getVirtualFile();
  44     final Document document = FileDocumentManager.getInstance().getDocument(virtualFile);
  45     final PsiFileSystemItem psiFile;
  46     if (document != null) {
  47       psiFile = PsiDocumentManager.getInstance(myProject).getPsiFile(document);
  48     }
  49     else if (context.getSelectorInFile() instanceof PsiFile) {
  50       psiFile = (PsiFile)context.getSelectorInFile();
  51     }
  52     else if (virtualFile.isDirectory()) {
  53       psiFile = PsiManager.getInstance(myProject).findDirectory(virtualFile);
  54     }
  55     else {
  56       psiFile = PsiManager.getInstance(myProject).findFile(virtualFile);
  57     }
  58     return psiFile != null && canSelect(psiFile) || canWorkWithCustomObjects();
  59   }
  60
  61   public final void selectIn(SelectInContext context, final boolean requestFocus) {
  62     VirtualFile file = context.getVirtualFile();
  63     Object selector = context.getSelectorInFile();
  64     if (selector == null) {
  65       PsiManager psiManager = PsiManager.getInstance(myProject);
  66       selector = file.isDirectory() ? psiManager.findDirectory(file) : psiManager.findFile(file);
  67     }
  68
  69     if (selector instanceof PsiElement) {
  70       select(((PsiElement)selector).getOriginalElement(), requestFocus);
  71     } else {
  72       select(selector, file, requestFocus);
  73     }
  74   }
  75
  76   protected abstract void select(final Object selector, VirtualFile virtualFile, final boolean requestFocus);
  77
  78   protected abstract boolean canWorkWithCustomObjects();
  79
  80   protected abstract void select(PsiElement element, boolean requestFocus);
  81 }