Examples with PsiRecursiveElementVisitor com.intellij.psi.PsiRecursiveElementVisitor used on opensource projects

Example 6 with PsiRecursiveElementVisitor

use of com.intellij.psi.PsiRecursiveElementVisitor in project intellij-elixir by KronicDeth.

the class ModuleAttribute method annotate.

/*
     * Public Instance Methods
     */
/**
     * Annotates the specified PSI element.
     * It is guaranteed to be executed in non-reentrant fashion.
     * I.e there will be no call of this method for this instance before previous call get completed.
     * Multiple instances of the annotator might exist simultaneously, though.
     *
     * @param element to annotate.
     * @param holder  the container which receives annotations created by the plugin.
     */
@Override
public void annotate(@NotNull final PsiElement element, @NotNull final AnnotationHolder holder) {
    element.accept(new PsiRecursiveElementVisitor() {

        /*
                     *
                     * Instance Methods
                     *
                     */
        /*
                     * Public Instance Methods
                     */
        @Override
        public void visitElement(@NotNull final PsiElement element) {
            if (element instanceof AtNonNumericOperation) {
                visitMaybeUsage((AtNonNumericOperation) element);
            } else if (element instanceof AtUnqualifiedNoParenthesesCall) {
                visitDeclaration((AtUnqualifiedNoParenthesesCall) element);
            }
        }

        /*
                     * Private Instance Methods
                     */
        private void visitDeclaration(@NotNull final AtUnqualifiedNoParenthesesCall atUnqualifiedNoParenthesesCall) {
            ElixirAtIdentifier atIdentifier = atUnqualifiedNoParenthesesCall.getAtIdentifier();
            TextRange textRange = atIdentifier.getTextRange();
            String identifier = identifierName(atIdentifier);
            if (isCallbackName(identifier)) {
                highlight(textRange, holder, ElixirSyntaxHighlighter.MODULE_ATTRIBUTE);
                highlightCallback(atUnqualifiedNoParenthesesCall, holder);
            } else if (isDocumentationName(identifier)) {
                highlight(textRange, holder, ElixirSyntaxHighlighter.DOCUMENTATION_MODULE_ATTRIBUTE);
                highlightDocumentationText(atUnqualifiedNoParenthesesCall, holder);
            } else if (isTypeName(identifier)) {
                highlight(textRange, holder, ElixirSyntaxHighlighter.MODULE_ATTRIBUTE);
                highlightType(atUnqualifiedNoParenthesesCall, holder);
            } else if (isSpecificationName(identifier)) {
                highlight(textRange, holder, ElixirSyntaxHighlighter.MODULE_ATTRIBUTE);
                highlightSpecification(atUnqualifiedNoParenthesesCall, holder);
            } else {
                highlight(textRange, holder, ElixirSyntaxHighlighter.MODULE_ATTRIBUTE);
            }
        }

        private void visitMaybeUsage(@NotNull final AtNonNumericOperation element) {
            PsiElement operand = element.operand();
            if (operand != null && !(operand instanceof ElixirAccessExpression)) {
                visitUsage(element);
            }
        }

        private void visitUsage(@NotNull final AtNonNumericOperation atNonNumericOperation) {
            highlight(atNonNumericOperation.getTextRange(), holder, ElixirSyntaxHighlighter.MODULE_ATTRIBUTE);
            if (!isNonReferencing(atNonNumericOperation) && atNonNumericOperation.getReference().resolve() == null) {
                holder.createErrorAnnotation(atNonNumericOperation, "Unresolved module attribute");
            }
        }
    });
}