org.jetbrains.kotlin.fir.resolve.FirJavaSyntheticNamesProvider.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.fir.resolve

import org.jetbrains.kotlin.fir.NoMutableState
import org.jetbrains.kotlin.fir.resolve.calls.FirSyntheticNamesProvider
import org.jetbrains.kotlin.load.java.propertyNameByGetMethodName
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.util.capitalizeDecapitalize.capitalizeAsciiOnly
import org.jetbrains.kotlin.util.capitalizeDecapitalize.capitalizeFirstWord

@NoMutableState
object FirJavaSyntheticNamesProvider : FirSyntheticNamesProvider() {
    private const val GETTER_PREFIX = "get"
    private const val SETTER_PREFIX = "set"
    private const val IS_PREFIX = "is"

    override fun possibleGetterNamesByPropertyName(name: Name): List {
        if (name.isSpecial) return emptyList()
        val identifier = name.identifier
        val capitalizedAsciiName = identifier.capitalizeAsciiOnly()
        val capitalizedFirstWordName = identifier.capitalizeFirstWord(asciiOnly = true)
        return listOfNotNull(
            Name.identifier(GETTER_PREFIX + capitalizedAsciiName),
            if (capitalizedFirstWordName == capitalizedAsciiName) null else Name.identifier(GETTER_PREFIX + capitalizedFirstWordName),
            name.takeIf { identifier.startsWith(IS_PREFIX) }
        ).filter {
            propertyNameByGetMethodName(it) == name
        }
    }

    override fun setterNameByGetterName(name: Name): Name? {
        val identifier = name.identifier
        val prefix = when {
            identifier.startsWith(GETTER_PREFIX) -> GETTER_PREFIX
            identifier.startsWith(IS_PREFIX) -> IS_PREFIX
            else -> return null
        }
        return Name.identifier(SETTER_PREFIX + identifier.removePrefix(prefix))
    }

    override fun getterNameBySetterName(name: Name): Name? {
        val identifier = name.identifier
        val prefix = when {
            identifier.startsWith(SETTER_PREFIX) -> SETTER_PREFIX
            else -> return null
        }
        return Name.identifier(GETTER_PREFIX + identifier.removePrefix(prefix))
    }

    override fun possiblePropertyNamesByAccessorName(name: Name): List {
        if (name.isSpecial) return emptyList()
        val identifier = name.identifier
        val prefix = when {
            identifier.startsWith(GETTER_PREFIX) -> GETTER_PREFIX
            identifier.startsWith(IS_PREFIX) -> ""
            identifier.startsWith(SETTER_PREFIX) -> SETTER_PREFIX
            else -> return emptyList()
        }
        val withoutPrefix = identifier.removePrefix(prefix)
        val withoutPrefixName = Name.identifier(withoutPrefix.decapitalize())
        return if (prefix == SETTER_PREFIX) {
            listOf(withoutPrefixName, Name.identifier(IS_PREFIX + withoutPrefix))
        } else {
            listOf(withoutPrefixName)
        }
    }
}