mirror of
https://gitflic.ru/project/openide/openide.git
synced 2026-05-04 20:30:42 +07:00
IDEA-158821 - added an edge between refFrom and lambda/method ref parameters where it's needed
GitOrigin-RevId: 946c0073473e6a5f68da422518cdc137f9224601
This commit is contained in:
committed by
intellij-monorepo-bot
parent
064b127d6b
commit
1b4c297f74
@@ -157,18 +157,20 @@ public class RefJavaUtilImpl extends RefJavaUtil {
|
||||
@Override
|
||||
public boolean visitCallableReferenceExpression(@NotNull UCallableReferenceExpression node) {
|
||||
visitReferenceExpression(node);
|
||||
processFunctionalExpression(node, getFunctionalInterfaceType(node));
|
||||
markParametersReferenced(node);
|
||||
// todo doesn't work for kotlin
|
||||
PsiType interfaceType = getFunctionalInterfaceType(node);
|
||||
processFunctionalExpression(node, interfaceType);
|
||||
markParametersReferenced(node, interfaceType);
|
||||
return false;
|
||||
}
|
||||
|
||||
private void markParametersReferenced(@NotNull UCallableReferenceExpression node) {
|
||||
PsiElement resolved = node.resolve();
|
||||
if (resolved == null) return;
|
||||
RefElement refElement = refFrom.getRefManager().getReference(resolved);
|
||||
if (refElement instanceof RefMethod) {
|
||||
for (RefParameter parameter : ((RefMethod)refElement).getParameters()) {
|
||||
refFrom.addReference(parameter, parameter.getPsiElement(), decl, false, true, node);
|
||||
private void markParametersReferenced(@NotNull UCallableReferenceExpression node, @Nullable PsiType type) {
|
||||
PsiMethod method = LambdaUtil.getFunctionalInterfaceMethod(type);
|
||||
if (method == null) return;
|
||||
for (PsiParameter param : method.getParameterList().getParameters()) {
|
||||
RefElement paramRef = refFrom.getRefManager().getReference(param);
|
||||
if (paramRef != null) {
|
||||
refFrom.addReference(paramRef, param, decl, false, true, node);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -245,6 +247,13 @@ public class RefJavaUtilImpl extends RefJavaUtil {
|
||||
//TODO support kotlin
|
||||
addClassReferenceForStaticImport(node, (PsiMember)psiResolved, refFrom, decl);
|
||||
}
|
||||
else {
|
||||
// todo currently if psiResolved is KtParameter, it doesn't convert to UParameter, that seems wrong
|
||||
UParameter uParam = UastContextKt.toUElement(psiResolved, UParameter.class);
|
||||
if (uParam != null) {
|
||||
addReferenceToLambdaParameter(uParam, psiResolved, decl, refFrom);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
@@ -367,6 +376,26 @@ public class RefJavaUtilImpl extends RefJavaUtil {
|
||||
}
|
||||
}
|
||||
|
||||
private static void addReferenceToLambdaParameter(@NotNull UParameter uParam, @NotNull PsiElement param, @NotNull UDeclaration decl,
|
||||
@NotNull RefJavaElementImpl refFrom) {
|
||||
ULambdaExpression lambda = UastUtils.getParentOfType(uParam, ULambdaExpression.class);
|
||||
if (lambda == null) return;
|
||||
int paramIndex = -1;
|
||||
List<UParameter> lambdaParams = lambda.getParameters();
|
||||
for (int i = 0; i < lambdaParams.size(); i++) {
|
||||
if (lambdaParams.get(i).equals(uParam)) {
|
||||
paramIndex = i;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (paramIndex == -1) return;
|
||||
RefElement method = refFrom.getRefManager().getReference(LambdaUtil.getFunctionalInterfaceMethod(lambda.getFunctionalInterfaceType()));
|
||||
if (method instanceof RefMethod) {
|
||||
RefParameter[] methodParams = ((RefMethod)method).getParameters();
|
||||
refFrom.addReference(methodParams[paramIndex], param, decl, false, true, null);
|
||||
}
|
||||
}
|
||||
|
||||
private static void addClassReferenceForStaticImport(UExpression node,
|
||||
PsiMember psiResolved,
|
||||
RefJavaElementImpl refFrom, UDeclaration decl) {
|
||||
|
||||
Reference in New Issue
Block a user