KotlinScope.kt

class KotlinScope: LanguageScope {
    override fun createTokenId(element: PsiElement): Short = when {
        isKotlinMethodName(element) -> nonAnonimizedIndex(element)
        else -> anonimizedIndex(element)
    }

    override fun isIndexedFile(fileType: FileType): Boolean {
        return fileType.name == "Kotlin"
    }

    override fun findIndexableScope(file: PsiFile): List<PsiElement> {
        return file.findChildren(::isScopeElement).map { it.lastChild }
    }

    override fun isIndexedElement(element: PsiElement): Boolean {
        return !isCommonNoise(element)
    }

    private fun isScopeElement(psiElement: PsiElement): Boolean {
        val typeString = psiElement.node.elementType.toString()
        return typeString == "FUN" || typeString == "PROPERTY_ACCESSOR"
    }

    private fun isKotlinMethodName(element: PsiElement): Boolean {
        return  element.type == "IDENTIFIER" &&
                element.parent?.type == "REFERENCE_EXPRESSION" &&
                element.grandparent?.type == "CALL_EXPRESSION"
    }
}