![JAR search and dependency download from the Maven repository](/logo.png)
com.intellij.find.findUsages.JavaFindUsagesHelper Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-analysis-impl Show documentation
Show all versions of java-analysis-impl Show documentation
A packaging of the IntelliJ Community Edition java-analysis-impl library.
This is release number 1 of trunk branch 142.
The newest version!
/*
* Copyright 2000-2014 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.find.findUsages;
import com.intellij.find.FindBundle;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.application.ReadActionProcessor;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.extensions.Extensions;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.util.Comparing;
import com.intellij.openapi.util.Computable;
import com.intellij.openapi.util.NullableComputable;
import com.intellij.openapi.util.TextRange;
import com.intellij.pom.PomTarget;
import com.intellij.pom.references.PomService;
import com.intellij.psi.*;
import com.intellij.psi.impl.search.ThrowSearchUtil;
import com.intellij.psi.meta.PsiMetaData;
import com.intellij.psi.meta.PsiMetaOwner;
import com.intellij.psi.search.*;
import com.intellij.psi.search.searches.*;
import com.intellij.psi.targets.AliasingPsiTarget;
import com.intellij.psi.targets.AliasingPsiTargetMapper;
import com.intellij.psi.util.MethodSignature;
import com.intellij.psi.util.PsiUtil;
import com.intellij.psi.xml.XmlAttributeValue;
import com.intellij.usageView.UsageInfo;
import com.intellij.util.Processor;
import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.util.*;
public class JavaFindUsagesHelper {
private static final Logger LOG = Logger.getInstance("#com.intellij.find.findUsages.JavaFindUsagesHelper");
@NotNull
static Set getElementNames(@NotNull final PsiElement element) {
if (element instanceof PsiDirectory) { // normalize a directory to a corresponding package
PsiPackage aPackage = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiPackage compute() {
return JavaDirectoryService.getInstance().getPackage((PsiDirectory)element);
}
});
return aPackage == null ? Collections.emptySet() : getElementNames(aPackage);
}
final Set result = new HashSet();
ApplicationManager.getApplication().runReadAction(new Runnable() {
@Override
public void run() {
if (element instanceof PsiPackage) {
ContainerUtil.addIfNotNull(result, ((PsiPackage)element).getQualifiedName());
}
else if (element instanceof PsiClass) {
final String qname = ((PsiClass)element).getQualifiedName();
if (qname != null) {
result.add(qname);
PsiClass topLevelClass = PsiUtil.getTopLevelClass(element);
if (topLevelClass != null) {
String topName = topLevelClass.getQualifiedName();
assert topName != null;
if (qname.length() > topName.length()) {
result.add(topName + qname.substring(topName.length()).replace('.', '$'));
}
}
}
}
else if (element instanceof PsiMethod) {
ContainerUtil.addIfNotNull(result, ((PsiMethod)element).getName());
}
else if (element instanceof PsiVariable) {
ContainerUtil.addIfNotNull(result, ((PsiVariable)element).getName());
}
else if (element instanceof PsiMetaOwner) {
final PsiMetaData metaData = ((PsiMetaOwner)element).getMetaData();
if (metaData != null) {
ContainerUtil.addIfNotNull(result, metaData.getName());
}
}
else if (element instanceof PsiNamedElement) {
ContainerUtil.addIfNotNull(result, ((PsiNamedElement)element).getName());
}
else if (element instanceof XmlAttributeValue) {
ContainerUtil.addIfNotNull(result, ((XmlAttributeValue)element).getValue());
}
else {
LOG.error("Unknown element type: " + element);
}
}
});
return result;
}
public static boolean processElementUsages(@NotNull final PsiElement element,
@NotNull final FindUsagesOptions options,
@NotNull final Processor processor) {
if (options instanceof JavaVariableFindUsagesOptions) {
final JavaVariableFindUsagesOptions varOptions = (JavaVariableFindUsagesOptions) options;
if (varOptions.isReadAccess || varOptions.isWriteAccess){
if (varOptions.isReadAccess && varOptions.isWriteAccess){
if (!addElementUsages(element, options, processor)) return false;
}
else{
if (!addElementUsages(element, varOptions, new Processor() {
@Override
public boolean process(UsageInfo info) {
final PsiElement element = info.getElement();
boolean isWrite = element instanceof PsiExpression && PsiUtil.isAccessedForWriting((PsiExpression)element);
if (isWrite == varOptions.isWriteAccess) {
if (!processor.process(info)) return false;
}
return true;
}
})) return false;
}
}
}
else if (options.isUsages) {
if (!addElementUsages(element, options, processor)) return false;
}
boolean success = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public Boolean compute() {
if (ThrowSearchUtil.isSearchable(element) && options instanceof JavaThrowFindUsagesOptions && options.isUsages) {
ThrowSearchUtil.Root root = ((JavaThrowFindUsagesOptions)options).getRoot();
if (root == null) {
final ThrowSearchUtil.Root[] roots = ThrowSearchUtil.getSearchRoots(element);
if (roots != null && roots.length > 0) {
root = roots [0];
}
}
if (root != null) {
return ThrowSearchUtil.addThrowUsages(processor, root, options);
}
}
return true;
}
});
if (!success) return false;
if (options instanceof JavaPackageFindUsagesOptions && ((JavaPackageFindUsagesOptions)options).isClassesUsages){
if (!addClassesUsages((PsiPackage)element, (JavaPackageFindUsagesOptions)options, processor)) return false;
}
if (options instanceof JavaClassFindUsagesOptions) {
final JavaClassFindUsagesOptions classOptions = (JavaClassFindUsagesOptions)options;
final PsiClass psiClass = (PsiClass)element;
PsiManager manager = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiManager compute() {
return psiClass.getManager();
}
});
if (classOptions.isMethodsUsages){
if (!addMethodsUsages(psiClass, manager, classOptions, processor)) return false;
}
if (classOptions.isFieldsUsages){
if (!addFieldsUsages(psiClass, manager, classOptions, processor)) return false;
}
if (psiClass.isInterface()) {
if (classOptions.isDerivedInterfaces){
if (classOptions.isImplementingClasses){
if (!addInheritors(psiClass, classOptions, processor)) return false;
}
else{
if (!addDerivedInterfaces(psiClass, classOptions, processor)) return false;
}
}
else if (classOptions.isImplementingClasses){
if (!addImplementingClasses(psiClass, classOptions, processor)) return false;
}
if (classOptions.isImplementingClasses) {
FunctionalExpressionSearch
.search(psiClass, classOptions.searchScope).forEach(new PsiElementProcessorAdapter(
new PsiElementProcessor() {
@Override
public boolean execute(@NotNull PsiFunctionalExpression expression) {
return addResult(expression, options, processor);
}
}));
}
}
else if (classOptions.isDerivedClasses) {
if (!addInheritors(psiClass, classOptions, processor)) return false;
}
}
if (options instanceof JavaMethodFindUsagesOptions){
final PsiMethod psiMethod = (PsiMethod)element;
boolean isAbstract = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public Boolean compute() {
return psiMethod.hasModifierProperty(PsiModifier.ABSTRACT);
}
});
final JavaMethodFindUsagesOptions methodOptions = (JavaMethodFindUsagesOptions)options;
if (isAbstract && methodOptions.isImplementingMethods || methodOptions.isOverridingMethods) {
if (!processOverridingMethods(psiMethod, processor, methodOptions)) return false;
FunctionalExpressionSearch.search(psiMethod, methodOptions.searchScope).forEach(new PsiElementProcessorAdapter(
new PsiElementProcessor() {
@Override
public boolean execute(@NotNull PsiFunctionalExpression expression) {
return addResult(expression, options, processor);
}
}));
}
}
if (element instanceof PomTarget) {
if (!addAliasingUsages((PomTarget)element, options, processor)) return false;
}
final Boolean isSearchable = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public Boolean compute() {
return ThrowSearchUtil.isSearchable(element);
}
});
if (!isSearchable && options.isSearchForTextOccurrences && options.searchScope instanceof GlobalSearchScope) {
Collection stringsToSearch = ApplicationManager.getApplication().runReadAction(new NullableComputable>() {
@Override
public Collection compute() {
return getElementNames(element);
}
});
// todo add to fastTrack
if (!FindUsagesHelper.processUsagesInText(element, stringsToSearch, (GlobalSearchScope)options.searchScope, processor)) return false;
}
return true;
}
private static boolean addAliasingUsages(@NotNull PomTarget pomTarget,
@NotNull final FindUsagesOptions options,
@NotNull final Processor processor) {
for (AliasingPsiTargetMapper aliasingPsiTargetMapper : Extensions.getExtensions(AliasingPsiTargetMapper.EP_NAME)) {
for (final AliasingPsiTarget psiTarget : aliasingPsiTargetMapper.getTargets(pomTarget)) {
boolean success = ReferencesSearch
.search(new ReferencesSearch.SearchParameters(ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiElement compute() {
return PomService.convertToPsi(psiTarget);
}
}), options.searchScope, false, options.fastTrack))
.forEach(new ReadActionProcessor() {
@Override
public boolean processInReadAction(final PsiReference reference) {
return addResult(reference, options, processor);
}
});
if (!success) return false;
}
}
return true;
}
private static boolean processOverridingMethods(@NotNull PsiMethod psiMethod,
@NotNull final Processor processor,
@NotNull final JavaMethodFindUsagesOptions options) {
return OverridingMethodsSearch.search(psiMethod, options.searchScope, options.isCheckDeepInheritance).forEach(new PsiElementProcessorAdapter(
new PsiElementProcessor() {
@Override
public boolean execute(@NotNull PsiMethod element) {
return addResult(element.getNavigationElement(), options, processor);
}
}));
}
private static boolean addClassesUsages(@NotNull PsiPackage aPackage,
@NotNull final JavaPackageFindUsagesOptions options,
@NotNull final Processor processor) {
ProgressIndicator progress = ProgressManager.getInstance().getProgressIndicator();
if (progress != null){
progress.pushState();
}
List classes = new ArrayList();
addClassesInPackage(aPackage, options.isIncludeSubpackages, classes);
for (final PsiClass aClass : classes) {
if (progress != null) {
String name = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public String compute() {
return aClass.getName();
}
});
progress.setText(FindBundle.message("find.searching.for.references.to.class.progress", name));
progress.checkCanceled();
}
boolean success = ReferencesSearch.search(new ReferencesSearch.SearchParameters(aClass, options.searchScope, false, options.fastTrack)).forEach(new ReadActionProcessor() {
@Override
public boolean processInReadAction(final PsiReference psiReference) {
return addResult(psiReference, options, processor);
}
});
if (!success) return false;
}
if (progress != null){
progress.popState();
}
return true;
}
private static void addClassesInPackage(@NotNull final PsiPackage aPackage, boolean includeSubpackages, @NotNull List array) {
PsiDirectory[] dirs = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiDirectory[] compute() {
return aPackage.getDirectories();
}
});
for (PsiDirectory dir : dirs) {
addClassesInDirectory(dir, includeSubpackages, array);
}
}
private static void addClassesInDirectory(@NotNull final PsiDirectory dir,
final boolean includeSubdirs,
@NotNull final List array) {
ApplicationManager.getApplication().runReadAction(new Runnable() {
@Override
public void run() {
PsiClass[] classes = JavaDirectoryService.getInstance().getClasses(dir);
ContainerUtil.addAll(array, classes);
if (includeSubdirs) {
PsiDirectory[] dirs = dir.getSubdirectories();
for (PsiDirectory directory : dirs) {
addClassesInDirectory(directory, includeSubdirs, array);
}
}
}
});
}
private static boolean addMethodsUsages(@NotNull final PsiClass aClass,
@NotNull final PsiManager manager,
@NotNull final JavaClassFindUsagesOptions options,
@NotNull final Processor processor) {
if (options.isIncludeInherited) {
final PsiMethod[] methods = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiMethod[] compute() {
return aClass.getAllMethods();
}
});
for(int i = 0; i < methods.length; i++){
final PsiMethod method = methods[i];
// filter overridden methods
final int finalI = i;
final PsiClass methodClass =
ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiClass compute() {
MethodSignature methodSignature = method.getSignature(PsiSubstitutor.EMPTY);
for (int j = 0; j < finalI; j++) {
if (methodSignature.equals(methods[j].getSignature(PsiSubstitutor.EMPTY))) return null;
}
return method.getContainingClass();
}
});
if (methodClass == null) continue;
boolean equivalent = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public Boolean compute() {
return manager.areElementsEquivalent(methodClass, aClass);
}
});
if (equivalent){
if (!addElementUsages(method, options, processor)) return false;
}
else {
MethodReferencesSearch.SearchParameters parameters =
new MethodReferencesSearch.SearchParameters(method, options.searchScope, true, options.fastTrack);
boolean success = MethodReferencesSearch.search(parameters)
.forEach(new PsiReferenceProcessorAdapter(new PsiReferenceProcessor() {
@Override
public boolean execute(PsiReference reference) {
addResultFromReference(reference, methodClass, manager, aClass, options, processor);
return true;
}
}));
if (!success) return false;
}
}
}
else {
PsiMethod[] methods = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiMethod[] compute() {
return aClass.getMethods();
}
});
for (PsiMethod method : methods) {
if (!addElementUsages(method, options, processor)) return false;
}
}
return true;
}
private static boolean addFieldsUsages(@NotNull final PsiClass aClass,
@NotNull final PsiManager manager,
@NotNull final JavaClassFindUsagesOptions options,
@NotNull final Processor processor) {
if (options.isIncludeInherited) {
final PsiField[] fields = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiField[] compute() {
return aClass.getAllFields();
}
});
for (int i = 0; i < fields.length; i++) {
final PsiField field = fields[i];
// filter hidden fields
final int finalI = i;
final PsiClass fieldClass =
ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiClass compute() {
for (int j = 0; j < finalI; j++) {
if (Comparing.strEqual(field.getName(), fields[j].getName())) return null;
}
return field.getContainingClass();
}
});
if (fieldClass == null) continue;
boolean equivalent = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public Boolean compute() {
return manager.areElementsEquivalent(fieldClass, aClass);
}
});
if (equivalent) {
if (!addElementUsages(fields[i], options, processor)) return false;
}
else {
boolean success = ReferencesSearch.search(new ReferencesSearch.SearchParameters(field, options.searchScope, false, options.fastTrack)).forEach(new ReadActionProcessor() {
@Override
public boolean processInReadAction(final PsiReference reference) {
return addResultFromReference(reference, fieldClass, manager, aClass, options, processor);
}
});
if (!success) return false;
}
}
}
else {
PsiField[] fields = ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public PsiField[] compute() {
return aClass.getFields();
}
});
for (PsiField field : fields) {
if (!addElementUsages(field, options, processor)) return false;
}
}
return true;
}
@Nullable
private static PsiClass getFieldOrMethodAccessedClass(@NotNull PsiReferenceExpression ref, @NotNull PsiClass fieldOrMethodClass) {
PsiElement[] children = ref.getChildren();
if (children.length > 1 && children[0] instanceof PsiExpression) {
PsiExpression expr = (PsiExpression)children[0];
PsiType type = expr.getType();
if (type != null) {
if (!(type instanceof PsiClassType)) return null;
return PsiUtil.resolveClassInType(type);
}
else {
if (expr instanceof PsiReferenceExpression) {
PsiElement refElement = ((PsiReferenceExpression)expr).resolve();
if (refElement instanceof PsiClass) return (PsiClass)refElement;
}
return null;
}
}
PsiManager manager = ref.getManager();
for(PsiElement parent = ref; parent != null; parent = parent.getParent()){
if (parent instanceof PsiClass
&& (manager.areElementsEquivalent(parent, fieldOrMethodClass) || ((PsiClass)parent).isInheritor(fieldOrMethodClass, true))){
return (PsiClass)parent;
}
}
return null;
}
private static boolean addInheritors(@NotNull PsiClass aClass,
@NotNull final JavaClassFindUsagesOptions options,
@NotNull final Processor processor) {
return ClassInheritorsSearch.search(aClass, options.searchScope, options.isCheckDeepInheritance).forEach(new PsiElementProcessorAdapter(
new PsiElementProcessor() {
@Override
public boolean execute(@NotNull PsiClass element) {
return addResult(element, options, processor);
}
}));
}
private static boolean addDerivedInterfaces(@NotNull PsiClass anInterface,
@NotNull final JavaClassFindUsagesOptions options,
@NotNull final Processor processor) {
return ClassInheritorsSearch.search(anInterface, options.searchScope, options.isCheckDeepInheritance).forEach(new PsiElementProcessorAdapter(
new PsiElementProcessor() {
@Override
public boolean execute(@NotNull PsiClass inheritor) {
return !inheritor.isInterface() || addResult(inheritor, options, processor);
}
}));
}
private static boolean addImplementingClasses(@NotNull PsiClass anInterface,
@NotNull final JavaClassFindUsagesOptions options,
@NotNull final Processor processor) {
return ClassInheritorsSearch.search(anInterface, options.searchScope, options.isCheckDeepInheritance).forEach(new PsiElementProcessorAdapter(
new PsiElementProcessor() {
@Override
public boolean execute(@NotNull PsiClass inheritor) {
return inheritor.isInterface() || addResult(inheritor, options, processor);
}
}));
}
private static boolean addResultFromReference(@NotNull PsiReference reference,
@NotNull PsiClass methodClass,
@NotNull PsiManager manager,
@NotNull PsiClass aClass,
@NotNull FindUsagesOptions options,
@NotNull Processor processor) {
PsiElement refElement = reference.getElement();
if (refElement instanceof PsiReferenceExpression) {
PsiClass usedClass = getFieldOrMethodAccessedClass((PsiReferenceExpression)refElement, methodClass);
if (usedClass != null) {
if (manager.areElementsEquivalent(usedClass, aClass) || usedClass.isInheritor(aClass, true)) {
if (!addResult(refElement, options, processor)) return false;
}
}
}
return true;
}
private static boolean addElementUsages(@NotNull final PsiElement element,
@NotNull final FindUsagesOptions options,
@NotNull final Processor processor) {
final SearchScope searchScope = options.searchScope;
final PsiClass[] parentClass = new PsiClass[1];
if (element instanceof PsiMethod && ApplicationManager.getApplication().runReadAction(new Computable() {
@Override
public Boolean compute() {
PsiMethod method = (PsiMethod)element;
parentClass[0] = method.getContainingClass();
return method.isConstructor();
}
})) {
PsiMethod method = (PsiMethod)element;
if (parentClass[0] != null) {
boolean strictSignatureSearch =
!(options instanceof JavaMethodFindUsagesOptions) || !((JavaMethodFindUsagesOptions)options).isIncludeOverloadUsages;
return MethodReferencesSearch
.search(new MethodReferencesSearch.SearchParameters(method, searchScope, strictSignatureSearch, options.fastTrack))
.forEach(new ReadActionProcessor() {
@Override
public boolean processInReadAction(final PsiReference ref) {
return addResult(ref, options, processor);
}
});
}
return true;
}
final ReadActionProcessor consumer = new ReadActionProcessor() {
@Override
public boolean processInReadAction(final PsiReference ref) {
return addResult(ref, options, processor);
}
};
if (element instanceof PsiMethod) {
final boolean strictSignatureSearch = !(options instanceof JavaMethodFindUsagesOptions) || // field with getter
!((JavaMethodFindUsagesOptions)options).isIncludeOverloadUsages;
return MethodReferencesSearch
.search(new MethodReferencesSearch.SearchParameters((PsiMethod)element, searchScope, strictSignatureSearch, options.fastTrack))
.forEach(consumer);
}
return ReferencesSearch.search(new ReferencesSearch.SearchParameters(element, searchScope, false, options.fastTrack)).forEach(consumer);
}
private static boolean addResult(@NotNull PsiElement element,
@NotNull FindUsagesOptions options,
@NotNull Processor processor) {
return !filterUsage(element, options) || processor.process(new UsageInfo(element));
}
private static boolean addResult(@NotNull PsiReference ref, @NotNull FindUsagesOptions options, @NotNull Processor processor) {
if (filterUsage(ref.getElement(), options)){
TextRange rangeInElement = ref.getRangeInElement();
return processor.process(new UsageInfo(ref.getElement(), rangeInElement.getStartOffset(), rangeInElement.getEndOffset(), false));
}
return true;
}
private static boolean filterUsage(PsiElement usage, @NotNull FindUsagesOptions options) {
if (!(usage instanceof PsiJavaCodeReferenceElement)) {
return true;
}
if (options instanceof JavaPackageFindUsagesOptions && !((JavaPackageFindUsagesOptions)options).isIncludeSubpackages &&
((PsiReference)usage).resolve() instanceof PsiPackage) {
PsiElement parent = usage.getParent();
if (parent instanceof PsiJavaCodeReferenceElement && ((PsiJavaCodeReferenceElement)parent).resolve() instanceof PsiPackage) {
return false;
}
}
if (!(usage instanceof PsiReferenceExpression)){
if (options instanceof JavaFindUsagesOptions && ((JavaFindUsagesOptions)options).isSkipImportStatements){
PsiElement parent = usage.getParent();
while(parent instanceof PsiJavaCodeReferenceElement){
parent = parent.getParent();
}
if (parent instanceof PsiImportStatement){
return false;
}
}
if (options instanceof JavaPackageFindUsagesOptions && ((JavaPackageFindUsagesOptions)options).isSkipPackageStatements){
PsiElement parent = usage.getParent();
while(parent instanceof PsiJavaCodeReferenceElement){
parent = parent.getParent();
}
if (parent instanceof PsiPackageStatement){
return false;
}
}
}
return true;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy