azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Extract searchLocalFunction() method

Browse Source
This commit is contained in:
Alexey Sedunov
2013-10-24 13:05:22 +04:00
parent f91e6b8d38
commit 06ab7708ae
1 changed files with 45 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/jet/plugin/findUsages/handlers/KotlinFindFunctionUsagesHandler.java
+45 -33
View File
@@ -74,46 +74,18 @@ public class KotlinFindFunctionUsagesHandler extends KotlinFindUsagesHandler<Jet
@Override
public boolean searchReferences(
@NotNull final PsiElement element,
@NotNull PsiElement element,
@NotNull final Processor<UsageInfo> processor,
@NotNull FindUsagesOptions options
) {
JetNamedFunction function = (JetNamedFunction) element;
final KotlinMethodFindUsagesOptions kotlinOptions = (KotlinMethodFindUsagesOptions) options;
SearchScope searchScope = kotlinOptions.searchScope;
JetElement blockForLocalDeclaration = JetPsiUtil.getEnclosingBlockForLocalDeclaration((JetNamedFunction) element);
JetElement blockForLocalDeclaration = JetPsiUtil.getEnclosingBlockForLocalDeclaration(function);
if (blockForLocalDeclaration != null && kotlinOptions.isUsages) {
BindingContext bindingContext =
AnalyzerFacadeWithCache.analyzeFileWithCache((JetFile) element.getContainingFile()).getBindingContext();
final List<PsiReference> result = new ArrayList<PsiReference>();
CalleeReferenceVisitorBase visitor = new CalleeReferenceVisitorBase(bindingContext, true) {
private boolean isAcceptable(PsiElement declaration) {
if (kotlinOptions.isIncludeOverloadUsages) {
//noinspection ConstantConditions
return declaration instanceof JetNamedFunction &&
((JetNamedFunction) element).getName()
.equals(((JetNamedFunction) declaration).getName());
}
return declaration.equals(element);
}
@Override
protected void processDeclaration(JetReferenceExpression reference, PsiElement declaration) {
if (isAcceptable(declaration)) {
result.add(reference.getReference());
}
}
};
blockForLocalDeclaration.accept(visitor);
for (PsiReference ref : result) {
if (!processUsage(processor, ref, kotlinOptions)) return false;
}
return true;
return searchLocalFunction(element, processor, function, kotlinOptions, blockForLocalDeclaration);
}
final PsiMethod lightMethod = ApplicationManager.getApplication().runReadAction(new Computable<PsiMethod>() {
@@ -176,6 +148,46 @@ public class KotlinFindFunctionUsagesHandler extends KotlinFindUsagesHandler<Jet
return true;
}
private static boolean searchLocalFunction(
final PsiElement element,
Processor<UsageInfo> processor,
final JetNamedFunction function,
final KotlinMethodFindUsagesOptions kotlinOptions,
JetElement blockForLocalDeclaration
) {
BindingContext bindingContext =
AnalyzerFacadeWithCache.analyzeFileWithCache((JetFile) element.getContainingFile()).getBindingContext();
final List<PsiReference> result = new ArrayList<PsiReference>();
CalleeReferenceVisitorBase visitor = new CalleeReferenceVisitorBase(bindingContext, true) {
private boolean isAcceptable(PsiElement declaration) {
if (kotlinOptions.isIncludeOverloadUsages) {
//noinspection ConstantConditions
return declaration instanceof JetNamedFunction &&
function.getName()
.equals(((JetNamedFunction) declaration).getName());
}
return declaration.equals(element);
}
@Override
protected void processDeclaration(JetReferenceExpression reference, PsiElement declaration) {
if (isAcceptable(declaration)) {
result.add(reference.getReference());
}
}
};
blockForLocalDeclaration.accept(visitor);
for (PsiReference ref : result) {
if (!processUsage(processor, ref, kotlinOptions)) return false;
}
return true;
}
@Override
protected boolean isSearchForTextOccurencesAvailable(@NotNull PsiElement psiElement, boolean isSingleFile) {
if (isSingleFile) return false;