main.de.jensklingenberg.ktorfit.model.ParameterData.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ktorfit-ksp Show documentation

Ktorfit

There is a newer version: 2.2.0-1.0.28

package de.jensklingenberg.ktorfit.model

import com.google.devtools.ksp.processing.KSPLogger
import com.google.devtools.ksp.symbol.KSValueParameter
import de.jensklingenberg.ktorfit.model.annotations.ParameterAnnotation
import de.jensklingenberg.ktorfit.model.annotations.ParameterAnnotation.RequestBuilder
import de.jensklingenberg.ktorfit.model.annotations.getParamAnnotationList
import de.jensklingenberg.ktorfit.utils.anyInstance
import de.jensklingenberg.ktorfit.utils.resolveTypeName

data class ParameterData(
    val name: String,
    val type: ReturnTypeData,
    val annotations: List = emptyList(),
) {
    inline fun  findAnnotationOrNull(): T? = this.annotations.firstOrNull { it is T } as? T

    inline fun  hasAnnotation(): Boolean = this.findAnnotationOrNull() != null
}

fun KSValueParameter.createParameterData(logger: KSPLogger): ParameterData {
    val ksValueParameter = this
    if (ksValueParameter.isVararg) {
        logger.error(KtorfitError.VARARG_NOT_SUPPORTED_USE_LIST_OR_ARRAY, ksValueParameter)
    }

    val parameterAnnotations = ksValueParameter.getParamAnnotationList(logger)

    val parameterName = ksValueParameter.name?.asString().orEmpty()
    val parameterType = ksValueParameter.type.resolve()
    val hasRequestBuilderAnno = parameterAnnotations.anyInstance()

    if (parameterAnnotations.isEmpty() && !hasRequestBuilderAnno) {
        logger.error(
            KtorfitError.noKtorfitAnnotationFoundAt(parameterName),
            ksValueParameter,
        )
    }

    if (hasRequestBuilderAnno && parameterType.resolveTypeName() != "[@kotlin.ExtensionFunctionType] Function1") {
        logger.error(
            KtorfitError.REQ_BUILDER_PARAMETER_TYPE_NEEDS_TO_BE_HTTP_REQUEST_BUILDER,
            ksValueParameter,
        )
    }

    val type =
        if (hasRequestBuilderAnno) {
            ReturnTypeData(
                "HttpRequestBuilder.()->Unit",
                parameterType,
            )
        } else {
            ReturnTypeData(
                parameterType.resolveTypeName(),
                parameterType,
            )
        }

    return ParameterData(
        parameterName,
        type,
        parameterAnnotations,
    )
}