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

Get reference for invoke calls with lazy resolve

Browse Source
This commit is contained in:
Nikolay Krasko
2013-08-08 19:16:08 +04:00
parent 992ccccde0
commit 1bb036bb80
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/jet/plugin/references/JetInvokeFunctionReference.java
+8 -7
View File
@@ -23,7 +23,7 @@ import org.jetbrains.annotations.NotNull;
import org.jetbrains.jet.lang.descriptors.CallableDescriptor; import org.jetbrains.jet.lang.descriptors.CallableDescriptor;
import org.jetbrains.jet.lang.descriptors.FunctionDescriptor; import org.jetbrains.jet.lang.descriptors.FunctionDescriptor;
import org.jetbrains.jet.lang.psi.JetCallExpression; import org.jetbrains.jet.lang.psi.JetCallExpression;
import org.jetbrains.jet.lang.psi.JetFile; import org.jetbrains.jet.lang.psi.JetExpression;
import org.jetbrains.jet.lang.psi.JetValueArgumentList; import org.jetbrains.jet.lang.psi.JetValueArgumentList;
import org.jetbrains.jet.lang.resolve.BindingContext; import org.jetbrains.jet.lang.resolve.BindingContext;
import org.jetbrains.jet.lang.resolve.BindingContextUtils; import org.jetbrains.jet.lang.resolve.BindingContextUtils;
@@ -54,15 +54,16 @@ class JetInvokeFunctionReference extends JetPsiReference implements MultiRangeRe
@Override @Override
protected PsiElement doResolve() { protected PsiElement doResolve() {
BindingContext bindingContext = WholeProjectAnalyzerFacade.analyzeProjectWithCacheOnAFile( JetExpression calleeExpression = ((JetCallExpression) myExpression).getCalleeExpression();
(JetFile) getElement().getContainingFile()).getBindingContext(); BindingContext bindingContext = WholeProjectAnalyzerFacade.getContextForElement(myExpression);
ResolvedCall<? extends CallableDescriptor> invokeFunction =
bindingContext.get(RESOLVED_CALL, ((JetCallExpression) myExpression).getCalleeExpression()); ResolvedCall<? extends CallableDescriptor> invokeFunction = bindingContext.get(RESOLVED_CALL, calleeExpression);
if (invokeFunction != null && invokeFunction instanceof VariableAsFunctionResolvedCall) { if (invokeFunction != null && invokeFunction instanceof VariableAsFunctionResolvedCall) {
FunctionDescriptor resultingDescriptor = ((VariableAsFunctionResolvedCall) invokeFunction).getResultingDescriptor(); FunctionDescriptor resultingDescriptor = ((VariableAsFunctionResolvedCall) invokeFunction).getResultingDescriptor();
PsiElement invokeFunctionElement = BindingContextUtils.callableDescriptorToDeclaration(bindingContext, resultingDescriptor); return BindingContextUtils.callableDescriptorToDeclaration(bindingContext, resultingDescriptor);
return invokeFunctionElement;
} }
return null; return null;
} }