diff --git a/src/main/kotlin/com/mapk/core/KFunctionForCall.kt b/src/main/kotlin/com/mapk/core/KFunctionForCall.kt index 3be3f43..7586e05 100644 --- a/src/main/kotlin/com/mapk/core/KFunctionForCall.kt +++ b/src/main/kotlin/com/mapk/core/KFunctionForCall.kt @@ -1,19 +1,17 @@ package com.mapk.core -import com.mapk.annotations.KConstructor import com.mapk.annotations.KParameterFlatten import com.mapk.core.internal.ArgumentBinder import com.mapk.core.internal.BucketGenerator import com.mapk.core.internal.ParameterNameConverter import com.mapk.core.internal.getAliasOrName +import com.mapk.core.internal.getKConstructor import com.mapk.core.internal.isUseDefaultArgument import kotlin.reflect.KClass import kotlin.reflect.KFunction import kotlin.reflect.KParameter import kotlin.reflect.full.findAnnotation -import kotlin.reflect.full.primaryConstructor import kotlin.reflect.jvm.isAccessible -import kotlin.reflect.jvm.jvmName import org.jetbrains.annotations.TestOnly class KFunctionForCall internal constructor( @@ -86,30 +84,10 @@ class KFunctionForCall internal constructor( } } -@Suppress("UNCHECKED_CAST") -internal fun KClass.toKConstructor(parameterNameConverter: ParameterNameConverter): KFunctionForCall { - val constructors = ArrayList>() - - this.getAnnotatedFunctionsFromCompanionObject()?.let { (instance, functions) -> - functions.forEach { - constructors.add(KFunctionForCall(it as KFunction, parameterNameConverter, instance)) - } - } - - this.constructors.getAnnotatedFunctions().forEach { - constructors.add(KFunctionForCall(it, parameterNameConverter)) - } - - if (constructors.size == 1) return constructors.single() - - if (constructors.isEmpty()) return KFunctionForCall(this.primaryConstructor!!, parameterNameConverter) - - throw IllegalArgumentException("${this.jvmName} has multiple ${KConstructor::class.jvmName}.") -} - -@Suppress("UNCHECKED_CAST") fun KClass.toKConstructor(parameterNameConverter: ((String) -> String)?): KFunctionForCall = - this.toKConstructor(ParameterNameConverter.Simple(parameterNameConverter)) + this.getKConstructor().let { (instance, function) -> + KFunctionForCall(function, ParameterNameConverter.Simple(parameterNameConverter), instance) + } private fun KParameter.toArgumentBinder(parameterNameConverter: ParameterNameConverter): ArgumentBinder { val name = getAliasOrName()!! @@ -124,7 +102,9 @@ private fun KParameter.toArgumentBinder(parameterNameConverter: ParameterNameCon parameterNameConverter.toSimple() } - ArgumentBinder.Function(getKClass().toKConstructor(converter), index, annotations) + getKClass().getKConstructor().let { (instance, function) -> + ArgumentBinder.Function(KFunctionForCall(function, converter, instance), index, annotations) + } } ?: ArgumentBinder.Value( index, annotations, diff --git a/src/main/kotlin/com/mapk/core/internal/Functions.kt b/src/main/kotlin/com/mapk/core/internal/Functions.kt index 47bf781..d8bbd8a 100644 --- a/src/main/kotlin/com/mapk/core/internal/Functions.kt +++ b/src/main/kotlin/com/mapk/core/internal/Functions.kt @@ -1,10 +1,16 @@ package com.mapk.core.internal +import com.mapk.annotations.KConstructor import com.mapk.annotations.KParameterAlias import com.mapk.annotations.KUseDefaultArgument +import com.mapk.core.getAnnotatedFunctions +import com.mapk.core.getAnnotatedFunctionsFromCompanionObject import java.lang.IllegalArgumentException +import kotlin.reflect.KClass +import kotlin.reflect.KFunction import kotlin.reflect.KParameter import kotlin.reflect.full.findAnnotation +import kotlin.reflect.full.primaryConstructor import kotlin.reflect.jvm.jvmName /** @@ -24,3 +30,24 @@ internal fun KParameter.isUseDefaultArgument(): Boolean { } return false } + +@Suppress("UNCHECKED_CAST") +internal fun KClass.getKConstructor(): Pair> { + val constructors = ArrayList>>() + + this.getAnnotatedFunctionsFromCompanionObject()?.let { (instance, functions) -> + functions.forEach { + constructors.add(instance to it as KFunction) + } + } + + this.constructors.getAnnotatedFunctions().forEach { + constructors.add(null to it) + } + + if (constructors.size == 1) return constructors.single() + + if (constructors.isEmpty()) return null to this.primaryConstructor!! + + throw IllegalArgumentException("${this.jvmName} has multiple ${KConstructor::class.jvmName}.") +}