com.ancientlightstudios.quarkus.kotlin.openapi.emitter.EmitterUtils.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quarkus-kotlin-openapi-maven-plugin Show documentation
A Maven plugin to use the OpenAPI generator.
There is a newer version: 0.4.14
Show newest version
package com.ancientlightstudios.quarkus.kotlin.openapi.emitter

import com.ancientlightstudios.quarkus.kotlin.openapi.models.kotlin.*
import com.ancientlightstudios.quarkus.kotlin.openapi.models.kotlin.TypeName.GenericTypeName.Companion.of
import com.ancientlightstudios.quarkus.kotlin.openapi.models.kotlin.TypeName.SimpleTypeName.Companion.typeName
import com.ancientlightstudios.quarkus.kotlin.openapi.models.kotlin.VariableName.Companion.variableName
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformable.ParameterKind
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformable.RequestMethod
import com.ancientlightstudios.quarkus.kotlin.openapi.models.types.*

fun AnnotationAware.addPathAnnotation(path: String) {
    kotlinAnnotation(Jakarta.PathAnnotationClass, "value".variableName() to path.literal())
}

fun AnnotationAware.addRequestMethodAnnotation(method: RequestMethod) {
    val className = when (method) {
        RequestMethod.Get -> Jakarta.GetAnnotationClass
        RequestMethod.Put -> Jakarta.PutAnnotationClass
        RequestMethod.Post -> Jakarta.PostAnnotationClass
        RequestMethod.Delete -> Jakarta.DeleteAnnotationClass
        RequestMethod.Options -> Jakarta.OptionsAnnotationClass
        RequestMethod.Head -> Jakarta.HeadAnnotationClass
        RequestMethod.Patch -> Jakarta.PatchAnnotationClass
        RequestMethod.Trace -> Jakarta.TraceAnnotationClass
    }
    kotlinAnnotation(className)
}

fun AnnotationAware.addConsumesAnnotation(vararg contentTypes: String) {
    kotlinAnnotation(Jakarta.ConsumesAnnotationClass,
        "value".variableName() to contentTypes.toList().arrayLiteral { it.literal() })
}

fun getSourceAnnotation(source: ParameterKind, name: String): KotlinAnnotation {
    val annotationClass = when (source) {
        ParameterKind.Path -> Jakarta.PathParamAnnotationClass
        ParameterKind.Query -> Jakarta.QueryParamAnnotationClass
        ParameterKind.Header -> Jakarta.HeaderParamAnnotationClass
        ParameterKind.Cookie -> Jakarta.CookieParamAnnotationClass
    }
    return KotlinAnnotation(annotationClass, null to name.literal())
}

fun TypeUsage.buildValidType(): TypeName {
    return when (val safeType = this.type) {
        is PrimitiveTypeDefinition -> safeType.baseType.typeName(isNullable())
        is EnumTypeDefinition -> safeType.modelName.typeName(isNullable())
        is ObjectTypeDefinition -> safeType.modelName.typeName(isNullable())
        is OneOfTypeDefinition -> safeType.modelName.typeName(isNullable())
        is CollectionTypeDefinition -> Kotlin.ListClass.typeName(isNullable())
            .of(safeType.items.buildValidType())
    }
}

fun TypeUsage.buildUnsafeJsonType(): TypeName {
    return when (val safeType = this.type) {
        is PrimitiveTypeDefinition -> safeType.baseType.typeName(true)
        is EnumTypeDefinition -> safeType.modelName.typeName(true)
        is ObjectTypeDefinition -> Library.UnsafeJsonClass.typeName(true).of(safeType.modelName.typeName())
        is OneOfTypeDefinition -> Library.UnsafeJsonClass.typeName(true).of(safeType.modelName.typeName())
        is CollectionTypeDefinition -> Kotlin.ListClass.typeName(true)
            .of(safeType.items.buildUnsafeJsonType())
    }
}

fun TypeUsage.isNullable() : Boolean {
    if (required && !type.nullable) {
        // null values are just not allowed
        return false
    }

    val hasDefault = when(val safeType = type) {
        is PrimitiveTypeDefinition -> safeType.defaultValue != null
        is EnumTypeDefinition -> safeType.defaultValue != null
        else -> false
    }

    if (hasDefault) {
        // if there is a default value set, this type never accepts null values
        return false
    }
    
    return type.nullable || !required
}