All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.jetbrains.kotlin.resolve.DataClassDescriptorResolver.kt Maven / Gradle / Ivy

There is a newer version: 2.1.0-Beta1
Show newest version
/*
 * Copyright 2010-2016 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.jetbrains.kotlin.resolve

import org.jetbrains.kotlin.descriptors.*
import org.jetbrains.kotlin.descriptors.annotations.Annotations
import org.jetbrains.kotlin.descriptors.impl.SimpleFunctionDescriptorImpl
import org.jetbrains.kotlin.descriptors.impl.ValueParameterDescriptorImpl
import org.jetbrains.kotlin.incremental.components.NoLookupLocation
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.resolve.descriptorUtil.builtIns
import org.jetbrains.kotlin.util.OperatorNameConventions

object DataClassDescriptorResolver {
    val EQUALS_METHOD_NAME = OperatorNameConventions.EQUALS
    val HASH_CODE_METHOD_NAME = Name.identifier("hashCode")
    val TO_STRING_METHOD_NAME = Name.identifier("toString")

    val COPY_METHOD_NAME = Name.identifier("copy")

    private val COMPONENT_FUNCTION_NAME_PREFIX = "component"

    fun createComponentName(index: Int): Name
            = Name.identifier(COMPONENT_FUNCTION_NAME_PREFIX + index)

    fun getComponentIndex(componentName: String): Int
            = componentName.substring(COMPONENT_FUNCTION_NAME_PREFIX.length).toInt()

    fun isComponentLike(name: Name): Boolean
            = isComponentLike(name.asString())

    private fun isComponentLike(name: String): Boolean {
        if (!name.startsWith(COMPONENT_FUNCTION_NAME_PREFIX)) return false

        try {
            getComponentIndex(name)
        }
        catch (e: NumberFormatException) {
            return false
        }

        return true
    }

    fun createEqualsFunctionDescriptor(classDescriptor: ClassDescriptor): SimpleFunctionDescriptor =
            doCreateFunctionFromAny(classDescriptor, EQUALS_METHOD_NAME)

    fun createHashCodeFunctionDescriptor(classDescriptor: ClassDescriptor): SimpleFunctionDescriptor =
            doCreateFunctionFromAny(classDescriptor, HASH_CODE_METHOD_NAME)

    fun createToStringFunctionDescriptor(classDescriptor: ClassDescriptor): SimpleFunctionDescriptor =
            doCreateFunctionFromAny(classDescriptor, TO_STRING_METHOD_NAME)

    private fun doCreateFunctionFromAny(classDescriptor: ClassDescriptor, name: Name): SimpleFunctionDescriptor {
        val functionDescriptor = SimpleFunctionDescriptorImpl.create(
                classDescriptor, Annotations.EMPTY, name, CallableMemberDescriptor.Kind.SYNTHESIZED, classDescriptor.source
        )

        val functionFromAny = classDescriptor.builtIns.any.getMemberScope(emptyList())
                .getContributedFunctions(name, NoLookupLocation.FROM_BUILTINS).single()

        functionDescriptor.initialize(
                null,
                classDescriptor.thisAsReceiverParameter,
                functionFromAny.typeParameters,
                functionFromAny.valueParameters.map { it.copy(functionDescriptor, it.name, it.index) },
                functionFromAny.returnType,
                Modality.OPEN,
                Visibilities.PUBLIC
        )

        return functionDescriptor
    }

    fun createComponentFunctionDescriptor(
            parameterIndex: Int,
            property: PropertyDescriptor,
            parameter: ValueParameterDescriptor,
            classDescriptor: ClassDescriptor,
            trace: BindingTrace
    ): SimpleFunctionDescriptor {
        val functionDescriptor = SimpleFunctionDescriptorImpl.create(
                classDescriptor,
                Annotations.EMPTY,
                createComponentName(parameterIndex),
                CallableMemberDescriptor.Kind.SYNTHESIZED,
                parameter.source
        )

        functionDescriptor.initialize(
                null,
                classDescriptor.thisAsReceiverParameter,
                emptyList(),
                emptyList(),
                property.type,
                Modality.FINAL,
                property.visibility
        )
        functionDescriptor.isOperator = true

        trace.record(BindingContext.DATA_CLASS_COMPONENT_FUNCTION, parameter, functionDescriptor)
        return functionDescriptor
    }

    fun createCopyFunctionDescriptor(
            constructorParameters: Collection,
            classDescriptor: ClassDescriptor,
            trace: BindingTrace
    ): SimpleFunctionDescriptor {
        val functionDescriptor = SimpleFunctionDescriptorImpl.create(
                classDescriptor,
                Annotations.EMPTY,
                COPY_METHOD_NAME,
                CallableMemberDescriptor.Kind.SYNTHESIZED,
                classDescriptor.source
        )

        val parameterDescriptors = arrayListOf()

        for (parameter in constructorParameters) {
            val propertyDescriptor = trace.bindingContext.get(BindingContext.VALUE_PARAMETER_AS_PROPERTY, parameter)
            // If a parameter doesn't have the corresponding property, it must not have a default value in the 'copy' function
            val declaresDefaultValue = propertyDescriptor != null
            val parameterDescriptor = ValueParameterDescriptorImpl(
                    functionDescriptor, null, parameter.index, parameter.annotations, parameter.name, parameter.type, declaresDefaultValue,
                    parameter.isCrossinline, parameter.isNoinline, parameter.varargElementType, parameter.source
            )
            parameterDescriptors.add(parameterDescriptor)
            if (declaresDefaultValue) {
                trace.record(BindingContext.VALUE_PARAMETER_AS_PROPERTY, parameterDescriptor, propertyDescriptor)
            }
        }

        functionDescriptor.initialize(
                null,
                classDescriptor.thisAsReceiverParameter,
                emptyList(),
                parameterDescriptors,
                classDescriptor.defaultType,
                Modality.FINAL,
                Visibilities.PUBLIC
        )

        trace.record(BindingContext.DATA_CLASS_COPY_FUNCTION, classDescriptor, functionDescriptor)
        return functionDescriptor
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy