PhpJobMethodReferenceProvider.java

/*
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

package com.magento.idea.magento2plugin.reference.provider;

import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiReference;
import com.intellij.psi.PsiReferenceProvider;
import com.intellij.psi.xml.XmlElement;
import com.intellij.util.ProcessingContext;
import com.jetbrains.php.lang.psi.elements.Method;
import com.jetbrains.php.lang.psi.elements.PhpClass;
import com.magento.idea.magento2plugin.indexes.DiIndex;
import com.magento.idea.magento2plugin.reference.xml.PolyVariantReferenceBase;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import org.jetbrains.annotations.NotNull;

/**
 * TODO: enable style checks after decomposition.
 */
@SuppressWarnings({"PMD", "checkstyle:all"})
public class PhpJobMethodReferenceProvider extends PsiReferenceProvider {

    @NotNull
    @Override
    public PsiReference[] getReferencesByElement(
            @NotNull final PsiElement element,
            @NotNull final ProcessingContext context
    ) {
        final List<PsiReference> psiReferences = new ArrayList<>();
        final String methodName = StringUtil.unquoteString(element.getText());
        final PhpClass phpClass = DiIndex.getPhpClassOfJobMethod((XmlElement) element);

        if (phpClass != null) {
            final Collection<Method> methods = phpClass.getMethods();
            methods.removeIf(method -> !method.getName().matches(methodName));
            if (!methods.isEmpty()) {
                psiReferences.add(new PolyVariantReferenceBase(element, methods));
            }
        }

        return psiReferences.toArray(new PsiReference[0]);
    }
}