[java] Resolve lombok extension methods in JavaTargetElementEvaluator

GitOrigin-RevId: 47fb6ac1ba4c143bb4c423120c86079b55b135af
This commit is contained in:
Tagir Valeev
2020-11-20 10:02:20 +07:00
committed by intellij-monorepo-bot
parent f4cd548a9f
commit 56a375c9c1
2 changed files with 48 additions and 0 deletions

View File

@@ -5,6 +5,7 @@ import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.util.Computable;
import com.intellij.psi.*;
import com.intellij.psi.augment.PsiAugmentProvider;
import com.intellij.psi.impl.light.LightRecordCanonicalConstructor;
import com.intellij.psi.impl.light.LightRecordMember;
import com.intellij.psi.javadoc.PsiDocTag;
@@ -151,6 +152,13 @@ public class JavaTargetElementEvaluator extends TargetElementEvaluatorEx2 implem
}
}
if (refElement instanceof PsiMethod) {
PsiMethod extensionMethod = PsiAugmentProvider.resolveExtensionMethod((PsiMethod)refElement);
if (extensionMethod != null) {
return extensionMethod;
}
}
if (refElement instanceof LightRecordMember) {
return ((LightRecordMember)refElement).getRecordComponent();
}