add DumbService.filterByDumbAwareness(T[])
[idea/community.git] / platform / lang-impl / src / com / intellij / ide / impl / ProjectViewSelectInTarget.java
1 /*
2  * Copyright 2000-2014 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.CompositeSelectInTarget;
20 import com.intellij.ide.SelectInContext;
21 import com.intellij.ide.SelectInTarget;
22 import com.intellij.ide.projectView.ProjectView;
23 import com.intellij.ide.projectView.SelectableTreeStructureProvider;
24 import com.intellij.ide.projectView.TreeStructureProvider;
25 import com.intellij.ide.projectView.impl.AbstractProjectViewPane;
26 import com.intellij.ide.projectView.impl.ProjectViewPane;
27 import com.intellij.openapi.application.ApplicationManager;
28 import com.intellij.openapi.extensions.Extensions;
29 import com.intellij.openapi.project.DumbService;
30 import com.intellij.openapi.project.Project;
31 import com.intellij.openapi.util.ActionCallback;
32 import com.intellij.openapi.vfs.VirtualFile;
33 import com.intellij.openapi.wm.ToolWindow;
34 import com.intellij.openapi.wm.ToolWindowId;
35 import com.intellij.openapi.wm.ToolWindowManager;
36 import com.intellij.psi.*;
37 import com.intellij.psi.util.PsiUtilBase;
38 import com.intellij.util.ObjectUtils;
39 import org.jetbrains.annotations.NotNull;
40 import org.jetbrains.annotations.Nullable;
41
42 import java.util.ArrayList;
43 import java.util.Collection;
44 import java.util.List;
45
46 public abstract class ProjectViewSelectInTarget extends SelectInTargetPsiWrapper implements CompositeSelectInTarget {
47   private String mySubId;
48
49   protected ProjectViewSelectInTarget(Project project) {
50     super(project);
51   }
52
53   @Override
54   protected final void select(final Object selector, final VirtualFile virtualFile, final boolean requestFocus) {
55     select(myProject, selector, getMinorViewId(), mySubId, virtualFile, requestFocus);
56   }
57
58   @NotNull
59   public static ActionCallback select(@NotNull Project project,
60                                       final Object toSelect,
61                                       @Nullable final String viewId,
62                                       @Nullable final String subviewId,
63                                       final VirtualFile virtualFile,
64                                       final boolean requestFocus) {
65     final ActionCallback result = new ActionCallback();
66
67
68     final ProjectView projectView = ProjectView.getInstance(project);
69     if (ApplicationManager.getApplication().isUnitTestMode()) {
70       AbstractProjectViewPane pane = projectView.getProjectViewPaneById(ProjectViewPane.ID);
71       pane.select(toSelect, virtualFile, requestFocus);
72       return result;
73     }
74
75     ToolWindowManager windowManager = ToolWindowManager.getInstance(project);
76     final ToolWindow projectViewToolWindow = windowManager.getToolWindow(ToolWindowId.PROJECT_VIEW);
77     final Runnable runnable = new Runnable() {
78       @Override
79       public void run() {
80         Runnable r = new Runnable() {
81           @Override
82           public void run() {
83             projectView.selectCB(toSelect, virtualFile, requestFocus).notify(result);
84           }
85         };
86         if (requestFocus) {
87           projectView.changeViewCB(ObjectUtils.chooseNotNull(viewId, ProjectViewPane.ID), subviewId).doWhenProcessed(r);
88         }
89         else {
90           r.run();
91         }
92       }
93     };
94
95     if (requestFocus) {
96       projectViewToolWindow.activate(runnable, false);
97     }
98     else {
99       projectViewToolWindow.show(runnable);
100     }
101
102     return result;
103   }
104
105
106   @Override
107   @NotNull
108   public Collection<SelectInTarget> getSubTargets(@NotNull SelectInContext context) {
109     List<SelectInTarget> result = new ArrayList<SelectInTarget>();
110     AbstractProjectViewPane pane = ProjectView.getInstance(myProject).getProjectViewPaneById(getMinorViewId());
111     int index = 0;
112     for (String subId : pane.getSubIds()) {
113       result.add(new ProjectSubViewSelectInTarget(this, subId, index++));
114     }
115     return result;
116   }
117
118   public boolean isSubIdSelectable(String subId, SelectInContext context) {
119     return false;
120   }
121
122   @Override
123   protected boolean canSelect(PsiFileSystemItem file) {
124     return true;
125   }
126
127   public String getSubIdPresentableName(String subId) {
128     AbstractProjectViewPane pane = ProjectView.getInstance(myProject).getProjectViewPaneById(getMinorViewId());
129     return pane.getPresentableSubIdName(subId);
130   }
131
132   @Override
133   public void select(PsiElement element, final boolean requestFocus) {
134     PsiElement toSelect = null;
135     for (TreeStructureProvider provider : getProvidersDumbAware()) {
136       if (provider instanceof SelectableTreeStructureProvider) {
137         toSelect = ((SelectableTreeStructureProvider) provider).getTopLevelElement(element);
138       }
139       if (toSelect != null) break;
140     }
141     if (toSelect == null) {
142       if (element instanceof PsiFile || element instanceof PsiDirectory) {
143         toSelect = element;
144       }
145       else {
146         final PsiFile containingFile = element.getContainingFile();
147         if (containingFile == null) return;
148         final FileViewProvider viewProvider = containingFile.getViewProvider();
149         toSelect = viewProvider.getPsi(viewProvider.getBaseLanguage());
150       }
151     }
152     if (toSelect == null) return;
153     PsiElement originalElement = toSelect.getOriginalElement();
154     final VirtualFile virtualFile = PsiUtilBase.getVirtualFile(originalElement);
155     select(originalElement, virtualFile, requestFocus);
156   }
157
158   private TreeStructureProvider[] getProvidersDumbAware() {
159     TreeStructureProvider[] allProviders = Extensions.getExtensions(TreeStructureProvider.EP_NAME, myProject);
160     List<TreeStructureProvider> dumbAware = DumbService.getInstance(myProject).filterByDumbAwareness(allProviders);
161     return dumbAware.toArray(new TreeStructureProvider[dumbAware.size()]);
162   }
163
164   @Override
165   public final String getToolWindowId() {
166     return ToolWindowId.PROJECT_VIEW;
167   }
168
169   @Override
170   protected boolean canWorkWithCustomObjects() {
171     return true;
172   }
173
174   public final void setSubId(String subId) {
175     mySubId = subId;
176   }
177
178   public final String getSubId() {
179     return mySubId;
180   }
181 }