platform/usageView/src/com/intellij/usages/UsageTargetUtil.java

   1 /*
   2  * Copyright 2000-2013 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.usages;
  17
  18 import com.intellij.openapi.actionSystem.CommonDataKeys;
  19 import com.intellij.openapi.actionSystem.DataProvider;
  20 import com.intellij.openapi.editor.Editor;
  21 import com.intellij.openapi.extensions.ExtensionPointName;
  22 import com.intellij.openapi.extensions.Extensions;
  23 import com.intellij.openapi.project.DumbService;
  24 import com.intellij.openapi.project.Project;
  25 import com.intellij.psi.PsiElement;
  26 import com.intellij.psi.PsiFile;
  27 import org.jetbrains.annotations.NotNull;
  28
  29 import java.util.ArrayList;
  30 import java.util.Arrays;
  31 import java.util.Collections;
  32 import java.util.List;
  33
  34 public class UsageTargetUtil {
  35   private static final ExtensionPointName<UsageTargetProvider> EP_NAME = ExtensionPointName.create("com.intellij.usageTargetProvider");
  36
  37   public static UsageTarget[] findUsageTargets(DataProvider dataProvider) {
  38     Editor editor = CommonDataKeys.EDITOR.getData(dataProvider);
  39     PsiFile file = CommonDataKeys.PSI_FILE.getData(dataProvider);
  40
  41     List<UsageTarget> result = new ArrayList<UsageTarget>();
  42     if (file != null && editor != null) {
  43       UsageTarget[] targets = findUsageTargets(editor, file);
  44       if (targets != null) Collections.addAll(result, targets);
  45     }
  46     PsiElement psiElement = CommonDataKeys.PSI_ELEMENT.getData(dataProvider);
  47     if (psiElement != null) {
  48       UsageTarget[] targets = findUsageTargets(psiElement);
  49       if (targets != null)Collections.addAll(result, targets);
  50     }
  51
  52     return result.isEmpty() ? null : result.toArray(new UsageTarget[result.size()]);
  53   }
  54
  55   public static UsageTarget[] findUsageTargets(Editor editor, PsiFile file) {
  56     List<UsageTarget> result = new ArrayList<UsageTarget>();
  57     for (UsageTargetProvider provider : getProviders(file.getProject())) {
  58       UsageTarget[] targets = provider.getTargets(editor, file);
  59       if (targets != null) Collections.addAll(result, targets);
  60     }
  61     return result.isEmpty() ? null : result.toArray(new UsageTarget[result.size()]);
  62   }
  63
  64   public static UsageTarget[] findUsageTargets(PsiElement psiElement) {
  65     List<UsageTarget> result = new ArrayList<UsageTarget>();
  66     for (UsageTargetProvider provider : getProviders(psiElement.getProject())) {
  67       UsageTarget[] targets = provider.getTargets(psiElement);
  68       if (targets != null) Collections.addAll(result, targets);
  69     }
  70     return result.isEmpty() ? null : result.toArray(new UsageTarget[result.size()]);
  71   }
  72
  73   @NotNull
  74   private static List<UsageTargetProvider> getProviders(Project project) {
  75     return DumbService.getInstance(project).filterByDumbAwareness(Arrays.asList(Extensions.getExtensions(EP_NAME)));
  76   }
  77 }