org.kodein.mock.ksp.utils.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mockmp-processor Show documentation

MocKMP KSP processor

There is a newer version: 2.0.0

package org.kodein.mock.ksp

import com.google.devtools.ksp.getConstructors
import com.google.devtools.ksp.isPublic
import com.google.devtools.ksp.symbol.*
import com.squareup.kotlinpoet.*
import com.squareup.kotlinpoet.ksp.TypeParameterResolver
import com.squareup.kotlinpoet.ksp.toTypeName
import com.squareup.kotlinpoet.ksp.toTypeParameterResolver


internal fun String.withNonEmptyPrefix(p: String) = if (isEmpty()) "" else "$p$this"

internal fun KSClassDeclaration.firstPublicConstructor() = (sequenceOf(primaryConstructor) + getConstructors()).firstOrNull { it?.isPublic() ?: false }


internal fun KSTypeReference.toRealTypeName(typeParamResolver: TypeParameterResolver = TypeParameterResolver.EMPTY): TypeName {

    val type = resolve()
    val decl = type.declaration

    if (decl is KSTypeAlias) {
        return decl.type.toRealTypeName(decl.typeParameters.toTypeParameterResolver(typeParamResolver))
    }

    return toTypeName(typeParamResolver)
}

internal fun KSType.toRealTypeName(typeParamResolver: TypeParameterResolver = TypeParameterResolver.EMPTY): TypeName {

    val decl = declaration

    if (decl is KSTypeAlias) {
        return decl.type.toRealTypeName(decl.typeParameters.toTypeParameterResolver(typeParamResolver))
    }

    return toTypeName(typeParamResolver)
}

internal fun TypeName.qualified(): String =
    when (this) {
        is ClassName -> canonicalName
        is ParameterizedTypeName -> rawType.canonicalName
        else -> error("Unsupported type: $this")
    }

internal fun KSType.realDeclaration(): KSDeclaration {
    val decl = declaration
    return when (decl) {
        is KSTypeAlias -> decl.type.resolve().realDeclaration()
        else -> decl
    }
}