main.io.github.tabilzad.ktor.k2.JsonNameResolver.kt Maven / Gradle / Ivy

Go to download

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

Open API (Swagger) specification Generator for Ktor

There is a newer version: 0.6.5-alpha

package io.github.tabilzad.ktor.k2

import org.jetbrains.kotlin.fir.FirSession
import org.jetbrains.kotlin.fir.declarations.FirProperty
import org.jetbrains.kotlin.fir.declarations.getStringArgument
import org.jetbrains.kotlin.fir.declarations.primaryConstructorIfAny
import org.jetbrains.kotlin.fir.declarations.utils.isData
import org.jetbrains.kotlin.fir.expressions.FirAnnotation
import org.jetbrains.kotlin.fir.resolve.fqName
import org.jetbrains.kotlin.fir.resolve.getContainingClass

object JsonNameResolver {

    fun getCustomNameFromAnnotation(
        property: FirProperty,
        session: FirSession,
    ): String? = getCustomNameFromPropertyAnnotation(property, session)
        ?: getMoshiNameFromDataClassConstructorParamAnnotation(property, session)

    private fun getCustomNameFromPropertyAnnotation(
        property: FirProperty,
        session: FirSession,
    ): String? = property.annotations.getCustomNameFromAnnotation(session)

    private fun List?.getCustomNameFromAnnotation(
        session: FirSession
    ): String? = this?.let { annotations ->
        SerializationFramework.entries.mapNotNull {
            annotations
                .find { annotation -> annotation.fqName(session) == it.fqName }
                ?.getStringArgument(it.identifier, session)
        }
    }?.firstOrNull()

    private fun getMoshiNameFromDataClassConstructorParamAnnotation(
        property: FirProperty,
        session: FirSession,
    ): String? = property.getContainingClass(session)
        ?.takeIf { it.isData }
        ?.primaryConstructorIfAny(session)
        ?.valueParameterSymbols
        ?.find { it.name == property.name }
        ?.annotations
        ?.getCustomNameFromAnnotation(session)
}