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

Extract isConstructor() check

Browse Source
This commit is contained in:
Alexey Sedunov
2013-09-16 20:49:49 +04:00
parent 07f1e61146
commit e4936549ee
2 changed files with 61 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/jet/plugin/findUsages/FindUsagesUtils.java
+59
View File
@@ -0,0 +1,59 @@
package org.jetbrains.jet.plugin.findUsages;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiConstructorCall;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.util.PsiTreeUtil;
import org.jetbrains.jet.lang.descriptors.ConstructorDescriptor;
import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
import org.jetbrains.jet.lang.psi.*;
import org.jetbrains.jet.lang.resolve.BindingContext;
import org.jetbrains.jet.lang.resolve.BindingContextUtils;
import org.jetbrains.jet.plugin.project.AnalyzerFacadeWithCache;
public class FindUsagesUtils {
private FindUsagesUtils() {
}
private static DeclarationDescriptor getCallDescriptor(PsiElement element, BindingContext bindingContext) {
JetConstructorCalleeExpression constructorCalleeExpression =
PsiTreeUtil.getParentOfType(element, JetConstructorCalleeExpression.class);
if (constructorCalleeExpression != null) {
JetReferenceExpression classReference = constructorCalleeExpression.getConstructorReferenceExpression();
return bindingContext.get(BindingContext.REFERENCE_TARGET, classReference);
}
JetCallExpression callExpression = PsiTreeUtil.getParentOfType(element, JetCallExpression.class);
if (callExpression != null) {
JetExpression callee = callExpression.getCalleeExpression();
if (callee instanceof JetReferenceExpression) {
return bindingContext.get(BindingContext.REFERENCE_TARGET, (JetReferenceExpression) callee);
}
}
return null;
}
public static boolean isConstructorUsage(PsiElement element, JetClassOrObject jetClassOrObject) {
PsiConstructorCall constructorCall = PsiTreeUtil.getParentOfType(element, PsiConstructorCall.class);
if (constructorCall != null && constructorCall == element.getParent()) {
PsiMethod constructor = constructorCall.resolveConstructor();
if (constructor == null) return false;
PsiClass constructorClass = constructor.getContainingClass();
return constructorClass != null && constructorClass.getNavigationElement() == jetClassOrObject;
}
if (!(element instanceof JetElement)) return false;
BindingContext bindingContext =
AnalyzerFacadeWithCache.analyzeFileWithCache((JetFile) element.getContainingFile()).getBindingContext();
DeclarationDescriptor descriptor = getCallDescriptor(element, bindingContext);
if (!(descriptor instanceof ConstructorDescriptor)) return false;
DeclarationDescriptor containingDescriptor = descriptor.getContainingDeclaration();
return containingDescriptor != null
&& BindingContextUtils.descriptorToDeclaration(bindingContext, containingDescriptor) == jetClassOrObject;
}
}
idea/src/org/jetbrains/jet/plugin/findUsages/handlers/KotlinFindClassUsagesHandler.java
+2 -48
View File
@@ -30,21 +30,16 @@ import com.intellij.psi.search.PsiElementProcessor;
import com.intellij.psi.search.PsiElementProcessorAdapter;
import com.intellij.psi.search.searches.ClassInheritorsSearch;
import com.intellij.psi.search.searches.ReferencesSearch;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.usageView.UsageInfo;
import com.intellij.util.Processor;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.asJava.LightClassUtil;
import org.jetbrains.jet.lang.descriptors.ConstructorDescriptor;
import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
import org.jetbrains.jet.lang.psi.*;
import org.jetbrains.jet.lang.resolve.BindingContext;
import org.jetbrains.jet.lang.resolve.BindingContextUtils;
import org.jetbrains.jet.plugin.findUsages.FindUsagesUtils;
import org.jetbrains.jet.plugin.findUsages.KotlinFindUsagesHandlerFactory;
import org.jetbrains.jet.plugin.findUsages.dialogs.KotlinFindClassUsagesDialog;
import org.jetbrains.jet.plugin.findUsages.options.KotlinClassFindUsagesOptions;
import org.jetbrains.jet.plugin.project.AnalyzerFacadeWithCache;
import java.util.Collection;
@@ -84,7 +79,7 @@ public class KotlinFindClassUsagesHandler extends KotlinFindUsagesHandler<JetCla
new ReferencesSearch.SearchParameters(jetClass, kotlinOptions.searchScope, false)
).findAll();
for (PsiReference ref : references) {
boolean constructorUsage = isConstructorUsage(ref.getElement(), jetClass);
boolean constructorUsage = FindUsagesUtils.isConstructorUsage(ref.getElement(), jetClass);
if ((constructorUsage && !kotlinOptions.searchConstructorUsages)
|| (!constructorUsage && !kotlinOptions.isUsages)) continue;
@@ -157,47 +152,6 @@ public class KotlinFindClassUsagesHandler extends KotlinFindUsagesHandler<JetCla
);
}
private static DeclarationDescriptor getCallDescriptor(PsiElement element, BindingContext bindingContext) {
JetConstructorCalleeExpression constructorCalleeExpression =
PsiTreeUtil.getParentOfType(element, JetConstructorCalleeExpression.class);
if (constructorCalleeExpression != null) {
JetReferenceExpression classReference = constructorCalleeExpression.getConstructorReferenceExpression();
return bindingContext.get(BindingContext.REFERENCE_TARGET, classReference);
}
JetCallExpression callExpression = PsiTreeUtil.getParentOfType(element, JetCallExpression.class);
if (callExpression != null) {
JetExpression callee = callExpression.getCalleeExpression();
if (callee instanceof JetReferenceExpression) {
return bindingContext.get(BindingContext.REFERENCE_TARGET, (JetReferenceExpression) callee);
}
}
return null;
}
private static boolean isConstructorUsage(PsiElement element, JetClass jetClass) {
PsiConstructorCall constructorCall = PsiTreeUtil.getParentOfType(element, PsiConstructorCall.class);
if (constructorCall != null && constructorCall == element.getParent()) {
PsiMethod constructor = constructorCall.resolveConstructor();
if (constructor == null) return false;
PsiClass constructorClass = constructor.getContainingClass();
return constructorClass != null && constructorClass.getNavigationElement() == jetClass;
}
if (!(element instanceof JetElement)) return false;
BindingContext bindingContext =
AnalyzerFacadeWithCache.analyzeFileWithCache((JetFile) element.getContainingFile()).getBindingContext();
DeclarationDescriptor descriptor = getCallDescriptor(element, bindingContext);
if (!(descriptor instanceof ConstructorDescriptor)) return false;
DeclarationDescriptor containingDescriptor = descriptor.getContainingDeclaration();
return containingDescriptor != null
&& BindingContextUtils.descriptorToDeclaration(bindingContext, containingDescriptor) == jetClass;
}
@Override
protected boolean isSearchForTextOccurencesAvailable(@NotNull PsiElement psiElement, boolean isSingleFile) {
if (isSingleFile) return false;