dev.mokkery.plugin.ir.IrFunction.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mokkery-plugin Show documentation

Mokkery is a mocking library for Kotlin Multiplatform, easy to use, boilerplate-free and compiler plugin driven.

The newest version!

package dev.mokkery.plugin.ir

import org.jetbrains.kotlin.backend.jvm.fullValueParameterList
import org.jetbrains.kotlin.backend.jvm.ir.eraseTypeParameters
import org.jetbrains.kotlin.backend.jvm.ir.isCompiledToJvmDefault
import org.jetbrains.kotlin.config.JvmDefaultMode
import org.jetbrains.kotlin.descriptors.Modality
import org.jetbrains.kotlin.ir.declarations.IrFunction
import org.jetbrains.kotlin.ir.declarations.IrSimpleFunction
import org.jetbrains.kotlin.ir.util.copyTo
import org.jetbrains.kotlin.ir.util.defaultType
import org.jetbrains.kotlin.ir.util.isFromJava
import org.jetbrains.kotlin.ir.util.isInterface
import org.jetbrains.kotlin.ir.util.parentClassOrNull
import org.jetbrains.kotlin.ir.util.remapTypeParameters
import org.jetbrains.kotlin.utils.memoryOptimizedMap

fun IrSimpleFunction.copyReturnTypeFrom(function: IrSimpleFunction) {
    returnType = function.returnType.remapTypeParameters(function, this, mapOf())
}

fun IrSimpleFunction.copyParametersFrom(function: IrSimpleFunction) {
    extensionReceiverParameter = function.extensionReceiverParameter?.copyTo(this)
    valueParameters = function.valueParameters
        .memoryOptimizedMap { it.copyTo(this, defaultValue = null) }
}

fun IrFunction.eraseFullValueParametersList() = fullValueParameterList.forEach { param ->
    val consumedParams = param.type.extractAllConsumedTypeParameters()
    if (consumedParams.any { it in typeParameters }) return@forEach
    param.type = param.type.eraseTypeParameters()
}

fun IrSimpleFunction.isJvmBinarySafeSuperCall(
    originalFunction: IrFunction,
    jvmDefaultMode: JvmDefaultMode,
    allowIndirectSuperCalls: Boolean
): Boolean {
    if (modality != Modality.OPEN) return false
    val parent = parentClassOrNull ?: return false
    val originalFunctionParentSupertypes = originalFunction.parentClassOrNull?.superTypes.orEmpty()
    if (parent.defaultType in originalFunctionParentSupertypes) return true
    if (!allowIndirectSuperCalls) return false
    if (isFakeOverride) return false
    if (isFromJava()) return false
    if (parent.isInterface && !isFakeOverride && isCompiledToJvmDefault(jvmDefaultMode)) return false
    return true
}