public static void visitAnnotatedElements( @Nullable PsiElement element, AnnotatedElementVisitor visitor) { if (element == null) return; for (PsiElement cur = ContextComputationProcessor.getTopLevelInjectionTarget(element); cur != null; cur = cur.getParent()) { if (!visitAnnotatedElementInner(cur, visitor)) return; } }
/** * Determines the PsiModifierListOwner for the passed element depending of the specified * LookupType. The LookupType decides whether to prefer the element a reference expressions * resolves to, or the element that is implied by the usage context ("expected type"). */ @Nullable public static PsiModifierListOwner getAnnotatedElementFor( @Nullable PsiElement element, LookupType type) { while (element != null) { if (type == LookupType.PREFER_DECLARATION || type == LookupType.DECLARATION_ONLY) { if (element instanceof PsiReferenceExpression) { final PsiElement e = ((PsiReferenceExpression) element).resolve(); if (e instanceof PsiModifierListOwner) { return (PsiModifierListOwner) e; } if (type == LookupType.DECLARATION_ONLY) { return null; } } } element = ContextComputationProcessor.getTopLevelInjectionTarget(element); final PsiElement parent = element.getParent(); if (element instanceof PsiAssignmentExpression && ((PsiAssignmentExpression) element).getOperationTokenType() == JavaTokenType.PLUSEQ) { element = ((PsiAssignmentExpression) element).getLExpression(); continue; } else if (parent instanceof PsiAssignmentExpression) { final PsiAssignmentExpression p = (PsiAssignmentExpression) parent; if (p.getRExpression() == element) { element = p.getLExpression(); continue; } } else if (parent instanceof PsiReturnStatement) { final PsiMethod m = PsiTreeUtil.getParentOfType(parent, PsiMethod.class); if (m != null) { return m; } } else if (parent instanceof PsiModifierListOwner) { return (PsiModifierListOwner) parent; } else if (parent instanceof PsiArrayInitializerMemberValue) { final PsiArrayInitializerMemberValue value = (PsiArrayInitializerMemberValue) parent; final PsiElement pair = value.getParent(); if (pair instanceof PsiNameValuePair) { return AnnotationUtil.getAnnotationMethod((PsiNameValuePair) pair); } } else if (parent instanceof PsiNameValuePair) { return AnnotationUtil.getAnnotationMethod((PsiNameValuePair) parent); } else { return PsiUtilEx.getParameterForArgument(element); } // If no annotation has been found through the usage context, check if the element // (i.e. the element the reference refers to) is annotated itself if (type != LookupType.DECLARATION_ONLY) { if (element instanceof PsiReferenceExpression) { final PsiElement e = ((PsiReferenceExpression) element).resolve(); if (e instanceof PsiModifierListOwner) { return (PsiModifierListOwner) e; } } } return null; } return null; }