package cz.k2.eshop

import com.intellij.patterns.PlatformPatterns.psiElement
import com.intellij.patterns.PsiNamePatternCondition
import com.intellij.patterns.StandardPatterns
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiReferenceContributor
import com.intellij.psi.PsiReferenceRegistrar
import com.jetbrains.php.lang.parser.PhpElementTypes
import com.jetbrains.php.lang.psi.elements.FunctionReference

class GetTemplateReferenceContributor : PsiReferenceContributor() {
    override fun registerReferenceProviders(registrar: PsiReferenceRegistrar) {
        val templateMethodPattern = object : PsiNamePatternCondition<PsiElement>("withFunctionName", StandardPatterns.string().matches("GetTemplate")) {
            override fun getPropertyValue(o: Any): String? {
                return if (o is FunctionReference) o.name else null
            }
        }

        val elementPattern = psiElement().withElementType(PhpElementTypes.STRING).
                withParent(psiElement().withElementType(PhpElementTypes.PARAMETER_LIST).withParent(psiElement().
                        withElementType(PhpElementTypes.FUNCTION_CALL).with(templateMethodPattern)))

        registrar.registerReferenceProvider(elementPattern, GetTemplateReferenceProvider(), PsiReferenceRegistrar.DEFAULT_PRIORITY)
    }
}